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