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 -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 gcc -o test test.o prov.o
  65 if [ $? -ne 0 ]; then
  66         print -u2 "failed to link final executable"
  67         exit 1
  68 fi
  69 
  70 script()
  71 {
  72         $dtrace -Zwqs /dev/stdin <<EOF
  73         test_prov*:::
  74         {
  75                 probeid = id;
  76         }
  77 
  78         tick-1sec
  79         /probeid == 0/
  80         {
  81                 printf("launching test\n");
  82                 system("./test");
  83         }
  84 
  85         tick-1sec
  86         /probeid != 0/
  87         {
  88                 printf("attempting re-enabling\n");
  89                 system("dtrace -e -x errtags -i %d", probeid);
  90                 attempts++;
  91         }
  92 
  93         tick-1sec
  94         /attempts > 10/
  95         {
  96                 exit(0);
  97         }
  98 EOF
  99 }
 100 
 101 script 2>&1 | tee test.out
 102 
 103 #
 104 # It should be true that our probe was reaped over the course of the enabling,
 105 # causing the embedded DTrace invocation to fail on an invalid probe (that is,
 106 # D_PDESC_INVAL) instead of an inability to grab the underlying process
 107 # (D_PROC_GRAB).
 108 # 
 109 grep D_PDESC_INVAL test.out 2> /dev/null 1>&2
 110 status=$?
 111 
 112 cd /
 113 /usr/bin/rm -rf $DIR
 114 
 115 exit $status