cddl/contrib/opensolaris/cmd/dtrace/test/tst/common/usdt/tst.fork.ksh

   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 #ident  "%Z%%M% %I%     %E% SMI"
  28
  29 if [ $# != 1 ]; then
  30         echo expected one argument: '<'dtrace-path'>'
  31         exit 2
  32 fi
  33
  34 dtrace=$1
  35 DIR=/var/tmp/dtest.$$
  36
  37 mkdir $DIR
  38 cd $DIR
  39
  40 cat > prov.d <<EOF
  41 provider test_prov {
  42         probe go();
  43 };
  44 EOF
  45
  46 $dtrace -h -s prov.d
  47 if [ $? -ne 0 ]; then
  48         print -u2 "failed to generate header file"
  49         exit 1
  50 fi
  51
  52 cat > test.c <<EOF
  53 #include <sys/types.h>
  54 #include <sys/wait.h>
  55 #include <unistd.h>
  56 #include "prov.h"
  57
  58 int
  59 main(int argc, char **argv)
  60 {
  61         TEST_PROV_GO();
  62         if (fork() == 0) {
  63                 TEST_PROV_GO();
  64                 return (0);
  65         }
  66
  67         (void) wait(NULL);
  68         TEST_PROV_GO();
  69
  70         return (0);
  71 }
  72 EOF
  73
  74 cc -c test.c
  75 if [ $? -ne 0 ]; then
  76         print -u2 "failed to compile test.c"
  77         exit 1
  78 fi
  79 $dtrace -G -32 -s prov.d test.o
  80 if [ $? -ne 0 ]; then
  81         print -u2 "failed to create DOF"
  82         exit 1
  83 fi
  84 cc -o test test.o prov.o
  85 if [ $? -ne 0 ]; then
  86         print -u2 "failed to link final executable"
  87         exit 1
  88 fi
  89
  90 script() {
  91         $dtrace -c ./test -Zqs /dev/stdin <<EOF
  92         test_prov*:::
  93         {
  94                 printf("%s:%s:%s\n", probemod, probefunc, probename);
  95         }
  96 EOF
  97 }
  98
  99 script
 100 status=$?
 101
 102 cd /
 103 /bin/rm -rf $DIR
 104
 105 exit $status