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