summaryrefslogtreecommitdiff
path: root/usr/src/cmd/dtrace/test/tst/common/usdt/tst.sameprovmulti.ksh
blob: 04b76207ea7acd029a5e3a8a8303f321cbe61f94 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
#
# This file and its contents are supplied under the terms of the
# Common Development and Distribution License ("CDDL"), version 1.0.
# You may only use this file in accordance with the terms of version
# 1.0 of the CDDL.
#
# A full copy of the text of the CDDL should have accompanied this
# source.  A copy of the CDDL is also available via the Internet at
# http://www.illumos.org/license/CDDL.
#

#
# Copyright (c) 2015, Joyent, Inc. All rights reserved.
#

#
# This test assures that we can have the same provider name across multiple
# probe definitions, and that the result will be the union of those
# definitions.  In particular, libusdt depends on this when (for example)
# node modules that create a provider are loaded multiple times due to
# being included by different modules.
#

if [ $# != 1 ]; then
	echo expected one argument: '<'dtrace-path'>'
	exit 2
fi

dtrace=$1
DIR=/var/tmp/dtest.$$

mkdir $DIR
cd $DIR

cat > test.c <<EOF
#include <unistd.h>

void
main()
{
EOF

objs=

for oogle in bagnoogle stalloogle cockoogle; do
	cat > $oogle.c <<EOF
#include <sys/sdt.h>

void
$oogle()
{
	DTRACE_PROBE(doogle, $oogle);
}
EOF

	cat > $oogle.d <<EOF
provider doogle {
	probe $oogle();
};
EOF

	gcc -m32 -c $oogle.c

	if [ $? -ne 0 ]; then
		print -u2 "failed to compile $oogle.c"
		exit 1
	fi

	$dtrace -G -32 -s $oogle.d $oogle.o -o $oogle.d.o

	if [ $? -ne 0 ]; then
		print -u2 "failed to process $oogle.d"
		exit 1
	fi

	objs="$objs $oogle.o $oogle.d.o"
	echo $oogle'();' >> test.c
done

echo "}" >> test.c

gcc -m32 -o test test.c $objs

if [ $? -ne 0 ]; then
	print -u2 "failed to compile test.c"
	exit 1
fi

$dtrace -n 'doogle$target:::{@[probename] = count()}' \
    -n 'END{printa("%-10s %@d\n", @)}' -x quiet -x aggsortkey -Zc ./test

if [ $? -ne 0 ]; then
	print -u2 "failed to execute test"
	exit 1
fi

cd /
/usr/bin/rm -rf $DIR
exit 0