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 
  74         BEGIN
  75         {
  76                 spec = speculation();
  77                 speculate(spec);
  78                 printf("this is speculative!\n");
  79         }
  80 
  81         test_prov*:::
  82         {
  83                 probeid = id;
  84         }
  85 
  86         tick-1sec
  87         /probeid == 0/
  88         {
  89                 printf("launching test\n");
  90                 system("./test");
  91         }
  92 
  93         tick-1sec
  94         /probeid != 0/
  95         {
  96                 printf("attempting re-enabling\n");
  97                 system("dtrace -e -x errtags -i %d", probeid);
  98                 attempts++;
  99         }
 100 
 101         tick-1sec
 102         /attempts > 10/
 103         {
 104                 exit(0);
 105         }
 106 EOF
 107 }
 108 
 109 script 2>&1 | tee test.out
 110 
 111 #
 112 # It should be true that our probe was not reaped after the provider was made
 113 # defunct: the speculative tracing action prevents reaping of any ECB in the
 114 # enabling.
 115 # 
 116 status=0
 117 
 118 if grep D_PDESC_INVAL test.out 2> /dev/null 1>&2 ; then
 119         status=1
 120 else
 121         grep D_PROC_GRAB test.out 2> /dev/null 1>&2
 122         status=$?
 123 fi
 124 
 125 cd /
 126 /usr/bin/rm -rf $DIR
 127 
 128 exit $status