1 #!/bin/ksh -p
   2 #
   3 # CDDL HEADER START
   4 #
   5 # The contents of this file are subject to the terms of the
   6 # Common Development and Distribution License (the "License").
   7 # You may not use this file except in compliance with the License.
   8 #
   9 # You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
  10 # or http://www.opensolaris.org/os/licensing.
  11 # See the License for the specific language governing permissions
  12 # and limitations under the License.
  13 #
  14 # When distributing Covered Code, include this CDDL HEADER in each
  15 # file and include the License file at usr/src/OPENSOLARIS.LICENSE.
  16 # If applicable, add the following below this CDDL HEADER, with the
  17 # fields enclosed by brackets "[]" replaced with your own identifying
  18 # information: Portions Copyright [yyyy] [name of copyright owner]
  19 #
  20 # CDDL HEADER END
  21 #
  22 
  23 #
  24 # Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
  25 # Use is subject to license terms.
  26 #
  27 
  28 if [ $# != 1 ]; then
  29         echo expected one argument: '<'dtrace-path'>'
  30         exit 2
  31 fi
  32 
  33 dtrace=$1
  34 DIR=/var/tmp/dtest.$$
  35 
  36 mkdir $DIR
  37 cd $DIR
  38 
  39 cat > prov.d <<EOF
  40 provider test_prov {
  41         probe go();
  42 };
  43 EOF
  44 
  45 $dtrace -h -s prov.d
  46 if [ $? -ne 0 ]; then
  47         print -u2 "failed to generate header file"
  48         exit 1
  49 fi
  50 
  51 cat > test.c <<EOF
  52 #include <sys/types.h>
  53 #include "prov.h"
  54 
  55 int
  56 main(int argc, char **argv)
  57 {
  58         TEST_PROV_GO();
  59         TEST_PROV_GO();
  60         TEST_PROV_GO();
  61         TEST_PROV_GO();
  62 
  63         return (0);
  64 }
  65 EOF
  66 
  67 gcc -c test.c
  68 if [ $? -ne 0 ]; then
  69         print -u2 "failed to compile test.c"
  70         exit 1
  71 fi
  72 $dtrace -G -32 -s prov.d test.o
  73 if [ $? -ne 0 ]; then
  74         print -u2 "failed to create DOF"
  75         exit 1
  76 fi
  77 gcc -o test test.o prov.o
  78 if [ $? -ne 0 ]; then
  79         print -u2 "failed to link final executable"
  80         exit 1
  81 fi
  82 
  83 script() {
  84         $dtrace -c ./test -qs /dev/stdin <<EOF
  85         test_prov\$target:::
  86         {
  87                 printf("%s:%s:%s\n", probemod, probefunc, probename);
  88         }
  89 EOF
  90 }
  91 
  92 script
  93 status=$?
  94 
  95 cd /
  96 /usr/bin/rm -rf $DIR
  97 
  98 exit $status