1#
2# CDDL HEADER START
3#
4# The contents of this file are subject to the terms of the
5# Common Development and Distribution License (the "License").
6# You may not use this file except in compliance with the License.
7#
8# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9# or http://www.opensolaris.org/os/licensing.
10# See the License for the specific language governing permissions
11# and limitations under the License.
12#
13# When distributing Covered Code, include this CDDL HEADER in each
14# file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15# If applicable, add the following below this CDDL HEADER, with the
16# fields enclosed by brackets "[]" replaced with your own identifying
17# information: Portions Copyright [yyyy] [name of copyright owner]
18#
19# CDDL HEADER END
20#
21
22#
23# Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
24# Use is subject to license terms.
25#
26
27if [ $# != 1 ]; then
28	echo expected one argument: '<'dtrace-path'>'
29	exit 2
30fi
31
32dtrace=$1
33DIR=/var/tmp/dtest.$$
34
35mkdir $DIR
36cd $DIR
37
38cat > test.c <<EOF
39#include <sys/sdt.h>
40
41int
42main(int argc, char **argv)
43{
44	DTRACE_PROBE(test_prov, entry);
45	DTRACE_PROBE(test_prov, __entry);
46	DTRACE_PROBE(test_prov, foo__entry);
47	DTRACE_PROBE(test_prov, carpentry);
48	DTRACE_PROBE(test_prov, miniatureturn);
49	DTRACE_PROBE(test_prov, foo__return);
50	DTRACE_PROBE(test_prov, __return);
51	/*
52	 * Unfortunately, a "return" probe is not currently possible due to
53	 * the conflict with a reserved word.
54	 */
55	DTRACE_PROBE(test_prov, done);
56}
57EOF
58
59cat > prov.d <<EOF
60provider test_prov {
61	probe entry();
62	probe __entry();
63	probe foo__entry();
64	probe carpentry();
65	probe miniatureturn();
66	probe foo__return();
67	probe __return();
68	probe done();
69};
70EOF
71
72gcc -m32 -c test.c
73if [ $? -ne 0 ]; then
74	print -u2 "failed to compile test.c"
75	exit 1
76fi
77$dtrace -G -32 -s prov.d test.o
78if [ $? -ne 0 ]; then
79	print -u2 "failed to create DOF"
80	exit 1
81fi
82gcc -m32 -o test test.o prov.o
83if [ $? -ne 0 ]; then
84	print -u2 "failed to link final executable"
85	exit 1
86fi
87
88script()
89{
90	$dtrace -wqZFs /dev/stdin <<EOF
91	BEGIN
92	{
93		system("$DIR/test");
94		printf("\n");
95	}
96
97	test_prov*:::done
98	/progenyof(\$pid)/
99	{
100		exit(0);
101	}
102
103	test_prov*:::
104	/progenyof(\$pid)/
105	{
106		printf("\n");
107	}
108EOF
109}
110
111script | cut -c5-
112status=$?
113
114cd /
115/usr/bin/rm -rf $DIR
116
117exit $status
118