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