usr/src/cmd/dtrace/test/tst/common/scripting/tst.egid.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
  28 #
  29 # Copyright (c) 2012 by Delphix. All rights reserved.
  30 #
  31
  32 ############################################################################
  33 # ASSERTION:
  34 #
  35 #       To verify egid of current process
  36 #
  37 #
  38 # SECTION: Scripting
  39 #
  40 ############################################################################
  41
  42 if [ $# != 1 ]; then
  43         echo expected one argument: '<'dtrace-path'>'
  44         exit 2
  45 fi
  46
  47 dtrace=$1
  48 bname=`/bin/basename $0`
  49 dfilename=/var/tmp/$bname.$$.d
  50
  51 ## Create .d file
  52 ##########################################################################
  53 cat > $dfilename <<-EOF
  54 #!$dtrace -qs
  55
  56
  57 BEGIN
  58 /\$egid != \$1/
  59 {
  60         exit(1);
  61 }
  62
  63 BEGIN
  64 /\$egid == \$1/
  65 {
  66         exit(0);
  67 }
  68 EOF
  69 ##########################################################################
  70
  71
  72 #chmod 555 the .d file
  73
  74 chmod 555 $dfilename >/dev/null 2>&1
  75 if [ $? -ne 0 ]; then
  76         print -u2 "chmod $dfilename failed"
  77         exit 1
  78 fi
  79
  80 #Get the groupid of the calling process using ps
  81
  82 groupid=`ps -o pid,gid | grep "$$ " | awk '{print $2}' 2>/dev/null`
  83 if [ $? -ne 0 ]; then
  84         print -u2 "unable to get uid of the current process with pid = $$"
  85         exit 1
  86 fi
  87
  88 #Pass groupid as argument to .d file
  89 $dfilename $groupid >/dev/null 2>&1
  90
  91 if [ $? -ne 0 ]; then
  92         print -u2 "Error in executing $dfilename"
  93         exit 1
  94 fi
  95
  96 #Cleanup leftovers
  97
  98 /usr/bin/rm -f $dfilename
  99 exit 0