xref: /illumos-gate/usr/src/cmd/dtrace/test/tst/common/scripting/tst.uid.ksh (revision 45ede40b2394db7967e59f19288fae9b62efd4aa)
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#ident	"%Z%%M%	%I%	%E% SMI"
29
30############################################################################
31# ASSERTION:
32#	To verify uid of current process.
33#
34# SECTION: Scripting
35#
36############################################################################
37
38if [ $# != 1 ]; then
39	echo expected one argument: '<'dtrace-path'>'
40	exit 2
41fi
42
43dtrace=$1
44bname=`/bin/basename $0`
45dfilename=/var/tmp/$bname.$$
46
47## Create .d file
48##########################################################################
49cat > $dfilename <<-EOF
50#!$dtrace -qs
51
52
53BEGIN
54/\$uid != \$1/
55{
56	exit(1);
57}
58
59BEGIN
60/\$uid == \$1/
61{
62	exit(0);
63}
64EOF
65##########################################################################
66
67
68#Call dtrace -C -s <.d>
69
70chmod 555 $dfilename
71
72userid=`ps -o pid,uid | grep "$$ " | awk '{print $2}' 2>/dev/null`
73if [ $? -ne 0 ]; then
74	print -u2 "unable to get uid of the current process with pid = $$"
75	exit 1
76fi
77
78$dfilename $userid >/dev/null 2>&1
79
80if [ $? -ne 0 ]; then
81	print -u2 "Error in executing $dfilename"
82	exit 1
83fi
84
85#/usr/bin/rm -f $dfilename
86exit 0
87