Home | History | Annotate | Download | only in pid
      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 2008 Sun Microsystems, Inc.  All rights reserved.
     25 # Use is subject to license terms.
     26 #
     27 # ident	"%Z%%M%	%I%	%E% SMI"
     28 
     29 #
     30 # This test verifies that USDT probes will be picked up after a dlopen(3C)
     31 # when a regex in the provider name matches both USDT probes and pid probes
     32 # (e.g., p*d$target matches both pid$target and pyramid$target.)
     33 #
     34 
     35 if [ $# != 1 ]; then
     36 	echo expected one argument: '<'dtrace-path'>'
     37 	exit 2
     38 fi
     39 
     40 dtrace=$1
     41 DIR=${TMPDIR:-/tmp}/dtest.$$
     42 
     43 mkdir $DIR
     44 cd $DIR
     45 
     46 cat > Makefile <<EOF
     47 all: main altlib.so
     48 
     49 main: main.o provmain.o
     50 	cc -o main main.o provmain.o
     51 
     52 main.o: main.c prov.h
     53 	cc -c main.c
     54 
     55 prov.h: prov.d
     56 	$dtrace -h -s prov.d
     57 
     58 provmain.o: prov.d main.o
     59 	$dtrace -G -32 -o provmain.o -s prov.d main.o
     60 
     61 altlib.so: altlib.o provalt.o
     62 	cc -z defs -G -o altlib.so altlib.o provalt.o -lc
     63 
     64 altlib.o: altlib.c prov.h
     65 	cc -c altlib.c
     66 
     67 provalt.o: prov.d altlib.o
     68 	$dtrace -G -32 -o provalt.o -s prov.d altlib.o
     69 EOF
     70 
     71 cat > prov.d <<EOF
     72 provider pyramid {
     73 	probe entry();
     74 };
     75 EOF
     76 
     77 cat > altlib.c <<EOF
     78 #include <sys/sdt.h>
     79 #include "prov.h"
     80 
     81 void
     82 go(void)
     83 {
     84 	PYRAMID_ENTRY();
     85 }
     86 EOF
     87 
     88 cat > main.c <<EOF
     89 #include <dlfcn.h>
     90 #include <unistd.h>
     91 #include <stdio.h>
     92 #include <sys/sdt.h>
     93 #include "prov.h"
     94 
     95 void
     96 go(void)
     97 {
     98 	PYRAMID_ENTRY();
     99 }
    100 
    101 int
    102 main(int argc, char **argv)
    103 {
    104 	void *alt;
    105 	void *alt_go;
    106 
    107 	go();
    108 
    109 	if ((alt = dlopen("./altlib.so", RTLD_LAZY | RTLD_LOCAL)) 
    110 	    == NULL) {
    111 		printf("dlopen of altlib.so failed: %s\n", dlerror());
    112 		return (1);
    113 	}
    114 
    115 	if ((alt_go = dlsym(alt, "go")) == NULL) {
    116 		printf("failed to lookup 'go' in altlib.so\n");
    117 		return (1);
    118 	}
    119 
    120 	((void (*)(void))alt_go)();
    121 
    122 	return (0);
    123 }
    124 EOF
    125 
    126 make > /dev/null
    127 if [ $? -ne 0 ]; then
    128 	print -u2 "failed to build"
    129 	exit 1
    130 fi
    131 
    132 cat > main.d <<'EOF'
    133 p*d$target::go:entry
    134 {
    135 	@foo[probemod, probefunc, probename] = count();
    136 }
    137 
    138 END
    139 {
    140 	printa("%s:%s:%s %@u\n",@foo);
    141 }
    142 EOF
    143 
    144 script() {
    145 	$dtrace -q -s ./main.d -c ./main
    146 }
    147 
    148 script
    149 status=$?
    150 
    151 cd /tmp
    152 /usr/bin/rm -rf $DIR
    153 
    154 exit $status
    155