xref: /illumos-gate/usr/src/cmd/dtrace/test/tst/common/usdt/tst.fork.ksh (revision 581cede61ac9c14d8d4ea452562a567189eead78)
1#!/bin/ksh -p
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 2006 Sun Microsystems, Inc.  All rights reserved.
25# Use is subject to license terms.
26#
27#ident	"%Z%%M%	%I%	%E% SMI"
28
29if [ $# != 1 ]; then
30	echo expected one argument: '<'dtrace-path'>'
31	exit 2
32fi
33
34dtrace=$1
35DIR=/var/tmp/dtest.$$
36
37mkdir $DIR
38cd $DIR
39
40cat > prov.d <<EOF
41provider test_prov {
42	probe go();
43};
44EOF
45
46$dtrace -h -s prov.d
47if [ $? -ne 0 ]; then
48	print -u2 "failed to generate header file"
49	exit 1
50fi
51
52cat > test.c <<EOF
53#include <sys/types.h>
54#include <sys/wait.h>
55#include <unistd.h>
56#include "prov.h"
57
58int
59main(int argc, char **argv)
60{
61	TEST_PROV_GO();
62	if (fork() == 0) {
63		TEST_PROV_GO();
64		return (0);
65	}
66
67	(void) wait(NULL);
68	TEST_PROV_GO();
69
70	return (0);
71}
72EOF
73
74cc -c test.c
75if [ $? -ne 0 ]; then
76	print -u2 "failed to compile test.c"
77	exit 1
78fi
79$dtrace -G -32 -s prov.d test.o
80if [ $? -ne 0 ]; then
81	print -u2 "failed to create DOF"
82	exit 1
83fi
84cc -o test test.o prov.o
85if [ $? -ne 0 ]; then
86	print -u2 "failed to link final executable"
87	exit 1
88fi
89
90script() {
91	$dtrace -c ./test -Zqs /dev/stdin <<EOF
92	test_prov*:::
93	{
94		printf("%s:%s:%s\n", probemod, probefunc, probename);
95	}
96EOF
97}
98
99script
100status=$?
101
102cd /
103/usr/bin/rm -rf $DIR
104
105exit $status
106