1*10e6dadfSbrendan#!/usr/bin/ksh
2*10e6dadfSbrendan#
3*10e6dadfSbrendan# CDDL HEADER START
4*10e6dadfSbrendan#
5*10e6dadfSbrendan# The contents of this file are subject to the terms of the
6*10e6dadfSbrendan# Common Development and Distribution License (the "License").
7*10e6dadfSbrendan# You may not use this file except in compliance with the License.
8*10e6dadfSbrendan#
9*10e6dadfSbrendan# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10*10e6dadfSbrendan# or http://www.opensolaris.org/os/licensing.
11*10e6dadfSbrendan# See the License for the specific language governing permissions
12*10e6dadfSbrendan# and limitations under the License.
13*10e6dadfSbrendan#
14*10e6dadfSbrendan# When distributing Covered Code, include this CDDL HEADER in each
15*10e6dadfSbrendan# file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16*10e6dadfSbrendan# If applicable, add the following below this CDDL HEADER, with the
17*10e6dadfSbrendan# fields enclosed by brackets "[]" replaced with your own identifying
18*10e6dadfSbrendan# information: Portions Copyright [yyyy] [name of copyright owner]
19*10e6dadfSbrendan#
20*10e6dadfSbrendan# CDDL HEADER END
21*10e6dadfSbrendan#
22*10e6dadfSbrendan
23*10e6dadfSbrendan#
24*10e6dadfSbrendan# Copyright 2008 Sun Microsystems, Inc.  All rights reserved.
25*10e6dadfSbrendan# Use is subject to license terms.
26*10e6dadfSbrendan#
27*10e6dadfSbrendan#pragma ident	"%Z%%M%	%I%	%E% SMI"
28*10e6dadfSbrendan
29*10e6dadfSbrendan#
30*10e6dadfSbrendan# Test ip:::{send,receive} of IPv4 TCP to a remote host.
31*10e6dadfSbrendan#
32*10e6dadfSbrendan# This may fail due to:
33*10e6dadfSbrendan#
34*10e6dadfSbrendan# 1. A change to the ip stack breaking expected probe behavior,
35*10e6dadfSbrendan#    which is the reason we are testing.
36*10e6dadfSbrendan# 2. No physical network interface is plumbed and up.
37*10e6dadfSbrendan# 3. No other hosts on this subnet are reachable and listening on ssh.
38*10e6dadfSbrendan# 4. An unlikely race causes the unlocked global send/receive
39*10e6dadfSbrendan#    variables to be corrupted.
40*10e6dadfSbrendan#
41*10e6dadfSbrendan# This test performs a TCP connection and checks that at least the
42*10e6dadfSbrendan# following packet counts were traced:
43*10e6dadfSbrendan#
44*10e6dadfSbrendan# 3 x ip:::send (2 during the TCP handshake, then a FIN)
45*10e6dadfSbrendan# 2 x ip:::receive (1 during the TCP handshake, then the FIN ACK)
46*10e6dadfSbrendan#
47*10e6dadfSbrendan
48*10e6dadfSbrendanif (( $# != 1 )); then
49*10e6dadfSbrendan	print -u2 "expected one argument: <dtrace-path>"
50*10e6dadfSbrendan	exit 2
51*10e6dadfSbrendanfi
52*10e6dadfSbrendan
53*10e6dadfSbrendandtrace=$1
54*10e6dadfSbrendangetaddr=./get.ipv4remote.pl
55*10e6dadfSbrendantcpport=22
56*10e6dadfSbrendanDIR=/var/tmp/dtest.$$
57*10e6dadfSbrendan
58*10e6dadfSbrendanif [[ ! -x $getaddr ]]; then
59*10e6dadfSbrendan        print -u2 "could not find or execute sub program: $getaddr"
60*10e6dadfSbrendan        exit 3
61*10e6dadfSbrendanfi
62*10e6dadfSbrendan$getaddr $tcpport | read source dest
63*10e6dadfSbrendanif (( $? != 0 )); then
64*10e6dadfSbrendan        exit 4
65*10e6dadfSbrendanfi
66*10e6dadfSbrendan
67*10e6dadfSbrendanmkdir $DIR
68*10e6dadfSbrendancd $DIR
69*10e6dadfSbrendan
70*10e6dadfSbrendancat > test.pl <<-EOPERL
71*10e6dadfSbrendan	use IO::Socket;
72*10e6dadfSbrendan	my \$s = IO::Socket::INET->new(
73*10e6dadfSbrendan	    Proto => "tcp",
74*10e6dadfSbrendan	    PeerAddr => "$dest",
75*10e6dadfSbrendan	    PeerPort => $tcpport,
76*10e6dadfSbrendan	    Timeout => 3);
77*10e6dadfSbrendan	die "Could not connect to host $dest port $tcpport" unless \$s;
78*10e6dadfSbrendan	close \$s;
79*10e6dadfSbrendanEOPERL
80*10e6dadfSbrendan
81*10e6dadfSbrendan$dtrace -c '/usr/bin/perl test.pl' -qs /dev/stdin <<EODTRACE
82*10e6dadfSbrendanBEGIN
83*10e6dadfSbrendan{
84*10e6dadfSbrendan	send = receive = 0;
85*10e6dadfSbrendan}
86*10e6dadfSbrendan
87*10e6dadfSbrendanip:::send
88*10e6dadfSbrendan/args[2]->ip_saddr == "$source" && args[2]->ip_daddr == "$dest" &&
89*10e6dadfSbrendan    args[4]->ipv4_protocol == IPPROTO_TCP/
90*10e6dadfSbrendan{
91*10e6dadfSbrendan	send++;
92*10e6dadfSbrendan}
93*10e6dadfSbrendan
94*10e6dadfSbrendanip:::receive
95*10e6dadfSbrendan/args[2]->ip_saddr == "$dest" && args[2]->ip_daddr == "$source" &&
96*10e6dadfSbrendan    args[4]->ipv4_protocol == IPPROTO_TCP/
97*10e6dadfSbrendan{
98*10e6dadfSbrendan	receive++;
99*10e6dadfSbrendan}
100*10e6dadfSbrendan
101*10e6dadfSbrendanEND
102*10e6dadfSbrendan{
103*10e6dadfSbrendan	printf("Minimum TCP events seen\n\n");
104*10e6dadfSbrendan	printf("ip:::send - %s\n", send >= 3 ? "yes" : "no");
105*10e6dadfSbrendan	printf("ip:::receive - %s\n", receive >= 2 ? "yes" : "no");
106*10e6dadfSbrendan}
107*10e6dadfSbrendanEODTRACE
108*10e6dadfSbrendan
109*10e6dadfSbrendanstatus=$?
110*10e6dadfSbrendan
111*10e6dadfSbrendancd /
112*10e6dadfSbrendan/usr/bin/rm -rf $DIR
113*10e6dadfSbrendan
114*10e6dadfSbrendanexit $?
115