usr/src/cmd/dtrace/test/tst/common/usdt/tst.enabled2.ksh

   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 2008 Sun Microsystems, Inc.  All rights reserved.
  24 # Use is subject to license terms.
  25 #
  26
  27 #
  28 # This test is primarily intended to verify a fix for SPARC, but there's no
  29 # harm in running it on other platforms. Here, we verify that is-enabled
  30 # probes don't interfere with return values from previously invoked functions.
  31 #
  32
  33 if [ $# != 1 ]; then
  34         echo expected one argument: '<'dtrace-path'>'
  35         exit 2
  36 fi
  37
  38 dtrace=$1
  39 DIR=/var/tmp/dtest.$$
  40
  41 mkdir $DIR
  42 cd $DIR
  43
  44 cat > prov.d <<EOF
  45 provider test_prov {
  46         probe go();
  47 };
  48 EOF
  49
  50 $dtrace -h -s prov.d
  51 if [ $? -ne 0 ]; then
  52         print -u2 "failed to generate header file"
  53         exit 1
  54 fi
  55
  56 cat > test.c <<EOF
  57 #include <stdio.h>
  58 #include "prov.h"
  59
  60 int
  61 foo(void)
  62 {
  63         return (24);
  64 }
  65
  66 int
  67 main(int argc, char **argv)
  68 {
  69         int a = foo();
  70         if (TEST_PROV_GO_ENABLED()) {
  71                 TEST_PROV_GO();
  72         }
  73         (void) printf("%d %d %d\n", a, a, a);
  74
  75         return (0);
  76 }
  77 EOF
  78
  79 gcc -m32 -c test.c
  80 if [ $? -ne 0 ]; then
  81         print -u2 "failed to compile test.c"
  82         exit 1
  83 fi
  84 $dtrace -G -32 -s prov.d test.o
  85 if [ $? -ne 0 ]; then
  86         print -u2 "failed to create DOF"
  87         exit 1
  88 fi
  89
  90 gcc -m32 -o test test.o prov.o
  91 if [ $? -ne 0 ]; then
  92         print -u2 "failed to link final executable"
  93         exit 1
  94 fi
  95
  96 script()
  97 {
  98         ./test
  99
 100         $dtrace -c ./test -qs /dev/stdin <<EOF
 101         test_prov\$target:::
 102         {
 103         }
 104 EOF
 105 }
 106
 107 script
 108 status=$?
 109
 110 cd /
 111 /usr/bin/rm -rf $DIR
 112
 113 exit $status