19cd928feSAlan Maguire#!/usr/bin/ksh 29cd928feSAlan Maguire# 39cd928feSAlan Maguire# CDDL HEADER START 49cd928feSAlan Maguire# 59cd928feSAlan Maguire# The contents of this file are subject to the terms of the 69cd928feSAlan Maguire# Common Development and Distribution License (the "License"). 79cd928feSAlan Maguire# You may not use this file except in compliance with the License. 89cd928feSAlan Maguire# 99cd928feSAlan Maguire# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE 109cd928feSAlan Maguire# or http://www.opensolaris.org/os/licensing. 119cd928feSAlan Maguire# See the License for the specific language governing permissions 129cd928feSAlan Maguire# and limitations under the License. 139cd928feSAlan Maguire# 149cd928feSAlan Maguire# When distributing Covered Code, include this CDDL HEADER in each 159cd928feSAlan Maguire# file and include the License file at usr/src/OPENSOLARIS.LICENSE. 169cd928feSAlan Maguire# If applicable, add the following below this CDDL HEADER, with the 179cd928feSAlan Maguire# fields enclosed by brackets "[]" replaced with your own identifying 189cd928feSAlan Maguire# information: Portions Copyright [yyyy] [name of copyright owner] 199cd928feSAlan Maguire# 209cd928feSAlan Maguire# CDDL HEADER END 219cd928feSAlan Maguire# 229cd928feSAlan Maguire 239cd928feSAlan Maguire# 249cd928feSAlan Maguire# Copyright (c) 2010, Oracle and/or its affiliates. All rights reserved. 259cd928feSAlan Maguire# 269cd928feSAlan Maguire 279cd928feSAlan Maguire# 289cd928feSAlan Maguire# Test tcp:::state-change and tcp:::{send,receive} by connecting to 299cd928feSAlan Maguire# the local ssh service and sending a test message. This should result 309cd928feSAlan Maguire# in a "Protocol mismatch" response and a close of the connection. 319cd928feSAlan Maguire# A number of state transition events along with tcp fusion send and 329cd928feSAlan Maguire# receive events for the message should result. 339cd928feSAlan Maguire# 349cd928feSAlan Maguire# This may fail due to: 359cd928feSAlan Maguire# 369cd928feSAlan Maguire# 1. A change to the ip stack breaking expected probe behavior, 379cd928feSAlan Maguire# which is the reason we are testing. 389cd928feSAlan Maguire# 2. The lo0 interface missing or not up. 399cd928feSAlan Maguire# 3. The local ssh service is not online. 409cd928feSAlan Maguire# 4. An unlikely race causes the unlocked global send/receive 419cd928feSAlan Maguire# variables to be corrupted. 429cd928feSAlan Maguire# 439cd928feSAlan Maguire# This test performs a TCP connection to the ssh service (port 22) and 449cd928feSAlan Maguire# checks that at least the following packet counts were traced: 459cd928feSAlan Maguire# 469cd928feSAlan Maguire# 3 x ip:::send (2 during the TCP handshake, then a FIN) 479cd928feSAlan Maguire# 4 x tcp:::send (2 during the TCP handshake, 1 message then a FIN) 489cd928feSAlan Maguire# 2 x ip:::receive (1 during the TCP handshake, then the FIN ACK) 499cd928feSAlan Maguire# 3 x tcp:::receive (1 during the TCP handshake, 1 message then the FIN ACK) 509cd928feSAlan Maguire# 519cd928feSAlan Maguire# The actual ip count tested is 5 each way, since we are tracing both 529cd928feSAlan Maguire# source and destination events. The actual tcp count tested is 7 539cd928feSAlan Maguire# each way, since the TCP fusion send/receive events will not reach IP. 549cd928feSAlan Maguire# 559cd928feSAlan Maguire# For this test to work, we are assuming that the TCP handshake and 569cd928feSAlan Maguire# TCP close will enter the IP code path and not use tcp fusion. 579cd928feSAlan Maguire# 589cd928feSAlan Maguire 599cd928feSAlan Maguireif (( $# != 1 )); then 609cd928feSAlan Maguire print -u2 "expected one argument: <dtrace-path>" 619cd928feSAlan Maguire exit 2 629cd928feSAlan Maguirefi 639cd928feSAlan Maguire 649cd928feSAlan Maguiredtrace=$1 659cd928feSAlan Maguirelocal=127.0.0.1 669cd928feSAlan Maguiretcpport=22 679cd928feSAlan MaguireDIR=/var/tmp/dtest.$$ 689cd928feSAlan Maguire 699cd928feSAlan Maguiremkdir $DIR 709cd928feSAlan Maguirecd $DIR 719cd928feSAlan Maguire 729cd928feSAlan Maguirecat > test.pl <<-EOPERL 739cd928feSAlan Maguire use IO::Socket; 749cd928feSAlan Maguire my \$s = IO::Socket::INET->new( 759cd928feSAlan Maguire Proto => "tcp", 769cd928feSAlan Maguire PeerAddr => "$local", 779cd928feSAlan Maguire PeerPort => $tcpport, 789cd928feSAlan Maguire Timeout => 3); 799cd928feSAlan Maguire die "Could not connect to host $local port $tcpport" unless \$s; 809cd928feSAlan Maguire print \$s "testing state machine transitions"; 819cd928feSAlan Maguire close \$s; 829cd928feSAlan MaguireEOPERL 839cd928feSAlan Maguire 84*c090e5dfSBryan Cantrill$dtrace -c 'perl test.pl' -qs /dev/stdin <<EODTRACE 859cd928feSAlan MaguireBEGIN 869cd928feSAlan Maguire{ 879cd928feSAlan Maguire ipsend = tcpsend = ipreceive = tcpreceive = 0; 889cd928feSAlan Maguire connreq = connest = connaccept = 0; 899cd928feSAlan Maguire} 909cd928feSAlan Maguire 919cd928feSAlan Maguireip:::send 929cd928feSAlan Maguire/args[2]->ip_saddr == "$local" && args[2]->ip_daddr == "$local" && 939cd928feSAlan Maguire args[4]->ipv4_protocol == IPPROTO_TCP/ 949cd928feSAlan Maguire{ 959cd928feSAlan Maguire ipsend++; 969cd928feSAlan Maguire} 979cd928feSAlan Maguire 989cd928feSAlan Maguiretcp:::send 999cd928feSAlan Maguire/args[2]->ip_saddr == "$local" && args[2]->ip_daddr == "$local" && 1009cd928feSAlan Maguire (args[4]->tcp_sport == $tcpport || args[4]->tcp_dport == $tcpport)/ 1019cd928feSAlan Maguire{ 1029cd928feSAlan Maguire tcpsend++; 1039cd928feSAlan Maguire} 1049cd928feSAlan Maguire 1059cd928feSAlan Maguireip:::receive 1069cd928feSAlan Maguire/args[2]->ip_saddr == "$local" && args[2]->ip_daddr == "$local" && 1079cd928feSAlan Maguire args[4]->ipv4_protocol == IPPROTO_TCP/ 1089cd928feSAlan Maguire{ 1099cd928feSAlan Maguire ipreceive++; 1109cd928feSAlan Maguire} 1119cd928feSAlan Maguire 1129cd928feSAlan Maguiretcp:::receive 1139cd928feSAlan Maguire/args[2]->ip_saddr == "$local" && args[2]->ip_daddr == "$local" && 1149cd928feSAlan Maguire (args[4]->tcp_sport == $tcpport || args[4]->tcp_dport == $tcpport)/ 1159cd928feSAlan Maguire{ 1169cd928feSAlan Maguire tcpreceive++; 1179cd928feSAlan Maguire} 1189cd928feSAlan Maguire 1199cd928feSAlan Maguiretcp:::state-change 1209cd928feSAlan Maguire{ 1219cd928feSAlan Maguire state_event[args[3]->tcps_state]++; 1229cd928feSAlan Maguire} 1239cd928feSAlan Maguire 1249cd928feSAlan Maguiretcp:::connect-request 1259cd928feSAlan Maguire/args[2]->ip_saddr == "$local" && args[2]->ip_daddr == "$local" && 1269cd928feSAlan Maguire args[4]->tcp_dport == $tcpport/ 1279cd928feSAlan Maguire{ 1289cd928feSAlan Maguire connreq++; 1299cd928feSAlan Maguire} 1309cd928feSAlan Maguire 1319cd928feSAlan Maguiretcp:::connect-established 1329cd928feSAlan Maguire/args[2]->ip_saddr == "$local" && args[2]->ip_daddr == "$local" && 1339cd928feSAlan Maguire args[4]->tcp_sport == $tcpport/ 1349cd928feSAlan Maguire{ 1359cd928feSAlan Maguire connest++; 1369cd928feSAlan Maguire} 1379cd928feSAlan Maguire 1389cd928feSAlan Maguiretcp:::accept-established 1399cd928feSAlan Maguire/args[2]->ip_saddr == "$local" && args[2]->ip_daddr == "$local" && 1409cd928feSAlan Maguire args[4]->tcp_dport == $tcpport/ 1419cd928feSAlan Maguire{ 1429cd928feSAlan Maguire connaccept++; 1439cd928feSAlan Maguire} 1449cd928feSAlan Maguire 1459cd928feSAlan MaguireEND 1469cd928feSAlan Maguire{ 1479cd928feSAlan Maguire printf("Minimum TCP events seen\n\n"); 1489cd928feSAlan Maguire printf("ip:::send - %s\n", ipsend >= 5 ? "yes" : "no"); 1499cd928feSAlan Maguire printf("ip:::receive - %s\n", ipreceive >= 5 ? "yes" : "no"); 1509cd928feSAlan Maguire printf("tcp:::send - %s\n", tcpsend >= 7 ? "yes" : "no"); 1519cd928feSAlan Maguire printf("tcp:::receive - %s\n", tcpreceive >= 7 ? "yes" : "no"); 1529cd928feSAlan Maguire printf("tcp:::state-change to syn-sent - %s\n", 1539cd928feSAlan Maguire state_event[TCP_STATE_SYN_SENT] >=1 ? "yes" : "no"); 1549cd928feSAlan Maguire printf("tcp:::state-change to syn-received - %s\n", 1559cd928feSAlan Maguire state_event[TCP_STATE_SYN_RECEIVED] >=1 ? "yes" : "no"); 1569cd928feSAlan Maguire printf("tcp:::state-change to established - %s\n", 1579cd928feSAlan Maguire state_event[TCP_STATE_ESTABLISHED] >= 2 ? "yes" : "no"); 1589cd928feSAlan Maguire printf("tcp:::state-change to fin-wait-1 - %s\n", 1599cd928feSAlan Maguire state_event[TCP_STATE_FIN_WAIT_1] >= 1 ? "yes" : "no"); 1609cd928feSAlan Maguire printf("tcp:::state-change to close-wait - %s\n", 1619cd928feSAlan Maguire state_event[TCP_STATE_CLOSE_WAIT] >= 1 ? "yes" : "no"); 1629cd928feSAlan Maguire printf("tcp:::state-change to fin-wait-2 - %s\n", 1639cd928feSAlan Maguire state_event[TCP_STATE_FIN_WAIT_2] >= 1 ? "yes" : "no"); 1649cd928feSAlan Maguire printf("tcp:::state-change to last-ack - %s\n", 1659cd928feSAlan Maguire state_event[TCP_STATE_LAST_ACK] >= 1 ? "yes" : "no"); 1669cd928feSAlan Maguire printf("tcp:::state-change to time-wait - %s\n", 1679cd928feSAlan Maguire state_event[TCP_STATE_TIME_WAIT] >= 1 ? "yes" : "no"); 1689cd928feSAlan Maguire printf("tcp:::connect-request - %s\n", 1699cd928feSAlan Maguire connreq >=1 ? "yes" : "no"); 1709cd928feSAlan Maguire printf("tcp:::connect-established - %s\n", 1719cd928feSAlan Maguire connest >=1 ? "yes" : "no"); 1729cd928feSAlan Maguire printf("tcp:::accept-established - %s\n", 1739cd928feSAlan Maguire connaccept >=1 ? "yes" : "no"); 1749cd928feSAlan Maguire} 1759cd928feSAlan MaguireEODTRACE 1769cd928feSAlan Maguire 1779cd928feSAlan Maguirestatus=$? 1789cd928feSAlan Maguire 1799cd928feSAlan Maguirecd / 1809cd928feSAlan Maguire/usr/bin/rm -rf $DIR 1819cd928feSAlan Maguire 1829cd928feSAlan Maguireexit $status 183