1 #
   2 # CDDL HEADER START
   3 #
   4 # The contents of this file are subject to the terms of the
   5 # Common Development and Distribution License (the "License").
   6 # You may not use this file except in compliance with the License.
   7 #
   8 # You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
   9 # or http://www.opensolaris.org/os/licensing.
  10 # See the License for the specific language governing permissions
  11 # and limitations under the License.
  12 #
  13 # When distributing Covered Code, include this CDDL HEADER in each
  14 # file and include the License file at usr/src/OPENSOLARIS.LICENSE.
  15 # If applicable, add the following below this CDDL HEADER, with the
  16 # fields enclosed by brackets "[]" replaced with your own identifying
  17 # information: Portions Copyright [yyyy] [name of copyright owner]
  18 #
  19 # CDDL HEADER END
  20 #
  21 
  22 #
  23 # Copyright (c) 2011, Joyent, Inc. All rights reserved.
  24 #
  25 
  26 if [ $# != 1 ]; then
  27         echo expected one argument: '<'dtrace-path'>'
  28         exit 2
  29 fi
  30 
  31 dtrace=$1
  32 DIR=/var/tmp/dtest.$$
  33 
  34 mkdir $DIR
  35 cd $DIR
  36 
  37 cat > test.c <<EOF
  38 #include <unistd.h>
  39 #include <sys/sdt.h>
  40 
  41 int
  42 main(int argc, char **argv)
  43 {
  44         DTRACE_PROBE(test_prov, probe1);
  45 }
  46 EOF
  47 
  48 cat > prov.d <<EOF
  49 provider test_prov {
  50         probe probe1();
  51 };
  52 EOF
  53 
  54 gcc -m32 -c test.c
  55 if [ $? -ne 0 ]; then
  56         print -u2 "failed to compile test.c"
  57         exit 1
  58 fi
  59 $dtrace -G -32 -s prov.d test.o
  60 if [ $? -ne 0 ]; then
  61         print -u2 "failed to create DOF"
  62         exit 1
  63 fi
  64 
  65 gcc -m32 -o test test.o prov.o
  66 if [ $? -ne 0 ]; then
  67         print -u2 "failed to link final executable"
  68         exit 1
  69 fi
  70 
  71 script()
  72 {
  73         $dtrace -Zwqs /dev/stdin <<EOF
  74         test_prov*:::
  75         {
  76                 probeid = id;
  77         }
  78 
  79         tick-1sec
  80         /probeid == 0/
  81         {
  82                 printf("launching test\n");
  83                 system("./test");
  84         }
  85 
  86         tick-1sec
  87         /probeid != 0/
  88         {
  89                 printf("attempting re-enabling\n");
  90                 system("dtrace -e -x errtags -i %d", probeid);
  91                 attempts++;
  92         }
  93 
  94         tick-1sec
  95         /attempts > 10/
  96         {
  97                 exit(0);
  98         }
  99 EOF
 100 }
 101 
 102 script 2>&1 | tee test.out
 103 
 104 #
 105 # It should be true that our probe was reaped over the course of the enabling,
 106 # causing the embedded DTrace invocation to fail on an invalid probe (that is,
 107 # D_PDESC_INVAL) instead of an inability to grab the underlying process
 108 # (D_PROC_GRAB).
 109 # 
 110 grep D_PDESC_INVAL test.out 2> /dev/null 1>&2
 111 status=$?
 112 
 113 cd /
 114 /usr/bin/rm -rf $DIR
 115 
 116 exit $status