usr/src/cmd/dtrace/test/tst/common/usdt/tst.dlclose3.ksh

   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 2007 Sun Microsystems, Inc.  All rights reserved.
  25 # Use is subject to license terms.
  26 #
  27
  28 #
  29 # This test verifies that performing a dlclose(3dl) on a library doesn't
  30 # cause existing pid provider probes to become invalid.
  31 #
  32
  33 if [ $# != 1 ]; then
  34         echo expected one argument: '<'dtrace-path'>'
  35         exit 2
  36 fi
  37
  38 dtrace=$1
  39 DIR=/var/tmp/dtest.$$
  40
  41 mkdir $DIR
  42 cd $DIR
  43
  44 cat > Makefile <<EOF
  45 all: main livelib.so deadlib.so
  46
  47 main: main.o prov.o
  48         gcc -m32 -o main main.o
  49
  50 main.o: main.c
  51         gcc -m32 -c main.c
  52
  53
  54 livelib.so: livelib.o prov.o
  55         gcc -m32 -shared -o livelib.so livelib.o prov.o -lc
  56
  57 livelib.o: livelib.c prov.h
  58         gcc -m32 -fPIC -c livelib.c
  59
  60 prov.o: livelib.o prov.d
  61         $dtrace -G -s prov.d livelib.o
  62
  63 prov.h: prov.d
  64         $dtrace -h -s prov.d
  65
  66
  67 deadlib.so: deadlib.o
  68         gcc -m32 -shared -o deadlib.so deadlib.o -lc
  69
  70 deadlib.o: deadlib.c
  71         gcc -m32 -fPIC -c deadlib.c
  72
  73 clean:
  74         rm -f main.o livelib.o prov.o prov.h deadlib.o
  75
  76 clobber: clean
  77         rm -f main livelib.so deadlib.so
  78 EOF
  79
  80 cat > prov.d <<EOF
  81 provider test_prov {
  82         probe go();
  83 };
  84 EOF
  85
  86 cat > livelib.c <<EOF
  87 #include "prov.h"
  88
  89 void
  90 go(void)
  91 {
  92         TEST_PROV_GO();
  93 }
  94 EOF
  95
  96 cat > deadlib.c <<EOF
  97 void
  98 go(void)
  99 {
 100 }
 101 EOF
 102
 103
 104 cat > main.c <<EOF
 105 #include <dlfcn.h>
 106 #include <unistd.h>
 107 #include <stdio.h>
 108
 109 static void
 110 foo(void)
 111 {
 112         (void) close(-1);
 113 }
 114
 115 int
 116 main(int argc, char **argv)
 117 {
 118         void *live;
 119
 120         if ((live = dlopen("./livelib.so", RTLD_LAZY | RTLD_LOCAL)) == NULL) {
 121                 printf("dlopen of livelib.so failed: %s\n", dlerror());
 122                 return (1);
 123         }
 124
 125         (void) dlclose(live);
 126
 127         foo();
 128
 129         return (0);
 130 }
 131 EOF
 132
 133 make > /dev/null
 134 if [ $? -ne 0 ]; then
 135         print -u2 "failed to build"
 136         exit 1
 137 fi
 138
 139 script() {
 140         $dtrace -c ./main -s /dev/stdin <<EOF
 141         pid\$target:a.out:foo:entry
 142         {
 143                 gotit = 1;
 144                 exit(0);
 145         }
 146
 147         tick-1s
 148         /i++ == 5/
 149         {
 150                 printf("test timed out");
 151                 exit(1);
 152         }
 153
 154         END
 155         /!gotit/
 156         {
 157                 printf("program ended without hitting probe");
 158                 exit(1);
 159         }
 160 EOF
 161 }
 162
 163 script
 164 status=$?
 165
 166 cd /
 167 /usr/bin/rm -rf $DIR
 168
 169 exit $status