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