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