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