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 26if [ $# != 1 ]; then 27 echo expected one argument: '<'dtrace-path'>' 28 exit 2 29fi 30 31dtrace=$1 32DIR=/var/tmp/dtest.$$ 33 34mkdir $DIR 35cd $DIR 36 37cat > test.c <<EOF 38#include <unistd.h> 39#include <sys/sdt.h> 40 41int 42main(int argc, char **argv) 43{ 44 DTRACE_PROBE(test_prov, probe1); 45} 46EOF 47 48cat > prov.d <<EOF 49provider test_prov { 50 probe probe1(); 51}; 52EOF 53 54gcc -m32 -c test.c 55if [ $? -ne 0 ]; then 56 print -u2 "failed to compile test.c" 57 exit 1 58fi 59$dtrace -G -32 -s prov.d test.o 60if [ $? -ne 0 ]; then 61 print -u2 "failed to create DOF" 62 exit 1 63fi 64 65gcc -m32 -o test test.o prov.o 66if [ $? -ne 0 ]; then 67 print -u2 "failed to link final executable" 68 exit 1 69fi 70 71script() 72{ 73 $dtrace -Zwqs /dev/stdin <<EOF 74 test_prov*::: 75 { 76 probeid = id; 77 } 78 79 tick-1sec 80 /probeid == 0/ 81 { 82 printf("launching test\n"); 83 system("./test"); 84 } 85 86 tick-1sec 87 /probeid != 0/ 88 { 89 printf("attempting re-enabling\n"); 90 system("dtrace -e -x errtags -i %d", probeid); 91 attempts++; 92 } 93 94 tick-1sec 95 /attempts > 10/ 96 { 97 exit(0); 98 } 99EOF 100} 101 102script 2>&1 | tee test.out 103 104# 105# It should be true that our probe was reaped over the course of the enabling, 106# causing the embedded DTrace invocation to fail on an invalid probe (that is, 107# D_PDESC_INVAL) instead of an inability to grab the underlying process 108# (D_PROC_GRAB). 109# 110grep D_PDESC_INVAL test.out 2> /dev/null 1>&2 111status=$? 112 113cd / 114/usr/bin/rm -rf $DIR 115 116exit $status 117