1178476Sjb#!/bin/ksh -p
2178476Sjb#
3178476Sjb# CDDL HEADER START
4178476Sjb#
5178476Sjb# The contents of this file are subject to the terms of the
6178476Sjb# Common Development and Distribution License (the "License").
7178476Sjb# You may not use this file except in compliance with the License.
8178476Sjb#
9178476Sjb# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10178476Sjb# or http://www.opensolaris.org/os/licensing.
11178476Sjb# See the License for the specific language governing permissions
12178476Sjb# and limitations under the License.
13178476Sjb#
14178476Sjb# When distributing Covered Code, include this CDDL HEADER in each
15178476Sjb# file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16178476Sjb# If applicable, add the following below this CDDL HEADER, with the
17178476Sjb# fields enclosed by brackets "[]" replaced with your own identifying
18178476Sjb# information: Portions Copyright [yyyy] [name of copyright owner]
19178476Sjb#
20178476Sjb# CDDL HEADER END
21178476Sjb#
22178476Sjb
23178476Sjb#
24178476Sjb# Copyright 2008 Sun Microsystems, Inc.  All rights reserved.
25178476Sjb# Use is subject to license terms.
26178476Sjb#
27178476Sjb# ident	"%Z%%M%	%I%	%E% SMI"
28178476Sjb
29178476Sjb#
30178476Sjb# This test verifies that probes will be picked up after a dlopen(3C)
31178476Sjb# when the pid provider is specified as a glob (e.g., p*d$target.)
32178476Sjb#
33178476Sjb
34178476Sjbif [ $# != 1 ]; then
35178476Sjb	echo expected one argument: '<'dtrace-path'>'
36178476Sjb	exit 2
37178476Sjbfi
38178476Sjb
39178476Sjbdtrace=$1
40178476SjbDIR=${TMPDIR:-/tmp}/dtest.$$
41178476Sjb
42178476Sjbmkdir $DIR
43178476Sjbcd $DIR
44178476Sjb
45178476Sjbcat > Makefile <<EOF
46178476Sjball: main altlib.so
47178476Sjb
48178476Sjbmain: main.o
49178476Sjb	cc -o main main.o
50178476Sjb
51178476Sjbmain.o: main.c
52178476Sjb	cc -c main.c
53178476Sjb
54178476Sjbaltlib.so: altlib.o
55279417Smarkj	cc -shared -o altlib.so altlib.o -lc
56178476Sjb
57178476Sjbaltlib.o: altlib.c
58178476Sjb	cc -c altlib.c
59178476SjbEOF
60178476Sjb
61178476Sjbcat > altlib.c <<EOF
62178476Sjbvoid
63178476Sjbgo(void)
64178476Sjb{
65178476Sjb}
66178476SjbEOF
67178476Sjb
68178476Sjbcat > main.c <<EOF
69178476Sjb#include <dlfcn.h>
70178476Sjb#include <unistd.h>
71178476Sjb#include <stdio.h>
72178476Sjb
73178476Sjbvoid
74178476Sjbgo(void)
75178476Sjb{
76178476Sjb}
77178476Sjb
78178476Sjbint
79178476Sjbmain(int argc, char **argv)
80178476Sjb{
81178476Sjb	void *alt;
82178476Sjb	void *alt_go;
83178476Sjb
84178476Sjb	go();
85178476Sjb
86178476Sjb	if ((alt = dlopen("./altlib.so", RTLD_LAZY | RTLD_LOCAL)) 
87178476Sjb	    == NULL) {
88178476Sjb		printf("dlopen of altlib.so failed: %s\n", dlerror());
89178476Sjb		return (1);
90178476Sjb	}
91178476Sjb
92178476Sjb	if ((alt_go = dlsym(alt, "go")) == NULL) {
93178476Sjb		printf("failed to lookup 'go' in altlib.so\n");
94178476Sjb		return (1);
95178476Sjb	}
96178476Sjb
97178476Sjb	((void (*)(void))alt_go)();
98178476Sjb
99178476Sjb	return (0);
100178476Sjb}
101178476SjbEOF
102178476Sjb
103178476Sjbmake > /dev/null
104178476Sjbif [ $? -ne 0 ]; then
105178476Sjb	print -u2 "failed to build"
106178476Sjb	exit 1
107178476Sjbfi
108178476Sjb
109178476Sjbcat > main.d <<'EOF'
110178476Sjbp*d$target::go:entry
111178476Sjb{
112178476Sjb	@foo[probemod, probefunc, probename] = count();
113178476Sjb}
114178476Sjb
115178476SjbEND
116178476Sjb{
117178476Sjb	printa("%s:%s:%s %@u\n",@foo);
118178476Sjb}
119178476SjbEOF
120178476Sjb
121178476Sjbscript() {
122178476Sjb	$dtrace -q -s ./main.d -c ./main
123178476Sjb}
124178476Sjb
125178476Sjbscript
126178476Sjbstatus=$?
127178476Sjb
128178476Sjbcd /tmp
129211545Srpaulo/bin/rm -rf $DIR
130178476Sjb
131178476Sjbexit $status
132