1#!/usr/bin/env ksh
2#
3# CDDL HEADER START
4#
5# The contents of this file are subject to the terms of the
6# Common Development and Distribution License (the "License").
7# You may not use this file except in compliance with the License.
8#
9# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10# or http://www.opensolaris.org/os/licensing.
11# See the License for the specific language governing permissions
12# and limitations under the License.
13#
14# When distributing Covered Code, include this CDDL HEADER in each
15# file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16# If applicable, add the following below this CDDL HEADER, with the
17# fields enclosed by brackets "[]" replaced with your own identifying
18# information: Portions Copyright [yyyy] [name of copyright owner]
19#
20# CDDL HEADER END
21#
22
23#
24# Copyright (c) 2008, 2010, Oracle and/or its affiliates. All rights reserved.
25#
26
27#
28# Test {ip,udp}:::{send,receive} of IPv4 UDP to a local address.
29#
30# This may fail due to:
31#
32# 1. A change to the ip stack breaking expected probe behavior,
33#    which is the reason we are testing.
34# 2. No physical network interface is plumbed and up.
35# 3. No other hosts on this subnet are reachable and listening on rpcbind.
36# 4. An unlikely race causes the unlocked global send/receive
37#    variables to be corrupted.
38#
39# This test sends a UDP message using perl and checks that at least the
40# following counts were traced:
41#
42# 1 x ip:::send (UDP sent to UDP port 33434)
43# 1 x udp:::send (UDP sent to UDP port 33434)
44# 1 x ip:::receive (UDP received)
45# 1 x udp:::receive (UDP received)
46#
47# A udp:::receive event is expected even if the received UDP packet
48# elicits an ICMP PORT_UNREACHABLE message since there is no UDP
49# socket for receiving the packet.
50#
51
52if (( $# != 1 )); then
53	print -u2 "expected one argument: <dtrace-path>"
54	exit 2
55fi
56
57dtrace=$1
58local=127.0.0.1
59port=33434
60DIR=/var/tmp/dtest.$$
61
62mkdir $DIR
63cd $DIR
64
65cat > test.pl <<-EOPERL
66	use IO::Socket;
67	my \$s = IO::Socket::INET->new(
68	    Proto => "udp",
69	    PeerAddr => "$local",
70	    PeerPort => $port);
71	die "Could not create UDP socket $local port $port" unless \$s;
72	send \$s, "Hello", 0;
73	close \$s;
74	sleep(2);
75EOPERL
76
77$dtrace -c 'perl test.pl' -qs /dev/stdin <<EODTRACE
78BEGIN
79{
80	ipsend = udpsend = ipreceive = udpreceive = 0;
81}
82
83ip:::send
84/args[2]->ip_saddr == "$local" && args[2]->ip_daddr == "$local" &&
85    args[4]->ipv4_protocol == IPPROTO_UDP/
86{
87	ipsend++;
88}
89
90udp:::send
91/args[2]->ip_saddr == "$local" && args[2]->ip_daddr == "$local"/
92{
93	udpsend++;
94}
95
96ip:::receive
97/args[2]->ip_saddr == "$local" && args[2]->ip_daddr == "$local" &&
98    args[4]->ipv4_protocol == IPPROTO_UDP/
99{
100	ipreceive++;
101}
102
103udp:::receive
104/args[2]->ip_saddr == "$local" && args[2]->ip_daddr == "$local"/
105{
106	udpreceive++;
107}
108
109END
110{
111	printf("Minimum UDP events seen\n\n");
112	printf("ip:::send - %s\n", ipsend >= 1 ? "yes" : "no");
113	printf("ip:::receive - %s\n", ipreceive >= 1 ? "yes" : "no");
114	printf("udp:::send - %s\n", udpsend >= 1 ? "yes" : "no");
115	printf("udp:::receive - %s\n", udpreceive >= 1 ? "yes" : "no");
116}
117EODTRACE
118
119status=$?
120
121cd /
122/bin/rm -rf $DIR
123
124exit $status
125