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 2007 Sun Microsystems, Inc.  All rights reserved.
  24 # Use is subject to license terms.
  25 #
  26 
  27 if [ $# != 1 ]; then
  28         echo expected one argument: '<'dtrace-path'>'
  29         exit 2
  30 fi
  31 
  32 dtrace=$1
  33 DIR=/var/tmp/dtest.$$
  34 
  35 mkdir $DIR
  36 cd $DIR
  37 
  38 cat > test.c <<EOF
  39 #include <sys/sdt.h>
  40 
  41 int
  42 main(int argc, char **argv)
  43 {
  44         DTRACE_PROBE(test_prov, entry);
  45         DTRACE_PROBE(test_prov, __entry);
  46         DTRACE_PROBE(test_prov, foo__entry);
  47         DTRACE_PROBE(test_prov, carpentry);
  48         DTRACE_PROBE(test_prov, miniatureturn);
  49         DTRACE_PROBE(test_prov, foo__return);
  50         DTRACE_PROBE(test_prov, __return);
  51         /*
  52          * Unfortunately, a "return" probe is not currently possible due to
  53          * the conflict with a reserved word.
  54          */
  55         DTRACE_PROBE(test_prov, done);
  56 }
  57 EOF
  58 
  59 cat > prov.d <<EOF
  60 provider test_prov {
  61         probe entry();
  62         probe __entry();
  63         probe foo__entry();
  64         probe carpentry();
  65         probe miniatureturn();
  66         probe foo__return();
  67         probe __return();
  68         probe done();
  69 };
  70 EOF
  71 
  72 gcc -c test.c
  73 if [ $? -ne 0 ]; then
  74         print -u2 "failed to compile test.c"
  75         exit 1
  76 fi
  77 $dtrace -G -32 -s prov.d test.o
  78 if [ $? -ne 0 ]; then
  79         print -u2 "failed to create DOF"
  80         exit 1
  81 fi
  82 gcc -o test test.o prov.o
  83 if [ $? -ne 0 ]; then
  84         print -u2 "failed to link final executable"
  85         exit 1
  86 fi
  87 
  88 script()
  89 {
  90         $dtrace -wqZFs /dev/stdin <<EOF
  91         BEGIN
  92         {
  93                 system("$DIR/test");
  94                 printf("\n");
  95         }
  96 
  97         test_prov*:::done
  98         /progenyof(\$pid)/
  99         {
 100                 exit(0);
 101         }
 102 
 103         test_prov*:::
 104         /progenyof(\$pid)/
 105         {
 106                 printf("\n");
 107         }
 108 EOF
 109 }
 110 
 111 script | cut -c5-
 112 status=$?
 113 
 114 cd /
 115 /usr/bin/rm -rf $DIR
 116 
 117 exit $status