#!/usr/bin/sh
 #
 # statsnoop - snoop file stats as they occur.
 #             Written using DTrace (Solaris 10 3/05).
 #
 # $Id: statsnoop 65 2007-10-04 11:09:40Z brendan $
 #
 # USAGE:	statsnoop [-a|-A|-ceghlsvxZ] [-f pathname] [-t syscall]
 #		          [-n name] [-p PID]
 #
 #		statsnoop	# default output
 #
 #		-a		# print most data
 #		-A		# dump all data, space delimited
 #		-c		# print cwd of process
 #		-e		# print errno value
 #		-g		# print command arguments
 #		-l		# print syscall type
 #		-s		# print start time, us
 #		-v		# print start time, string
 #		-x		# only print failed stats
 #		-Z		# print zonename
 #		-f pathname	# file pathname to snoop
 #		-n name		# command name to snoop
 #		-p PID		# process ID to snoop
 #		-t syscall	# stat syscall to trace
 #	eg,
 #		statsnoop -v			# human readable timestamps
 #		statsnoop -S			# syscall type
 #		statsnoop -e			# see error codes
 #		statsnoop -f /etc/passwd	# snoop this file only
 # 	
 # FIELDS:
 #		ZONE		Zone name
 #		UID		User ID
 #		PID		Process ID
 #		PPID		Parent Process ID
 #		FD		file descriptor (-1 for error)
 #		ERR		errno value (see /usr/include/sys/errno.h)
 #		TYPE		syscall type
 #		CWD		current working directory of process
 #		PATH		pathname for file stat
 #		COMM		command name for the process
 #		ARGS		argument listing for the process
 #		TIME		timestamp for the stat event, us
 #		STRTIME		timestamp for the stat event, string
 #
 # SEE ALSO: truss, BSM auditing.
 #
 # COPYRIGHT: Copyright (c) 2007 Brendan Gregg.
 #
 # CDDL HEADER START
 #
 #  The contents of this file are subject to the terms of the
 #  Common Development and Distribution License, Version 1.0 only
 #  (the "License").  You may not use this file except in compliance
 #  with the License.
 #
 #  You can obtain a copy of the license at Docs/cddl1.txt
 #  or http://www.opensolaris.org/os/licensing.
 #  See the License for the specific language governing permissions
 #  and limitations under the License.
 #
 # CDDL HEADER END
 #
 # Author: Brendan Gregg  [Sydney, Australia]
 #
 # 09-Sep-2007	Brendan Gregg	Created this.
 # 
 
 
 ##############################
 # --- Process Arguments ---
 #
 
 ### Default variables
 opt_dump=0; opt_file=0; opt_time=0; opt_timestr=0; opt_args=0
 opt_zone=0; opt_cwd=0; opt_failonly=0; opt_err=0; filter=0; pathname=.
 opt_name=0; opt_pid=0; opt_type=0; opt_trace=0; pname=.; pid=0; trace=.
 
 ### Process options
 while getopts aAcef:ghln:p:st:vxZ name
 do
 	case $name in
 	a)	opt_time=1; opt_timestr=1; opt_args=1; opt_err=1 ;;
 	A)	opt_dump=1 ;;
 	c)	opt_cwd=1 ;;
 	e)	opt_err=1 ;;
 	g)	opt_args=1 ;;
 	f)	opt_file=1; pathname=$OPTARG ;;
 	l)	opt_type=1 ;;
 	n)	opt_name=1; pname=$OPTARG ;;
 	p)	opt_pid=1; pid=$OPTARG ;;
 	s)	opt_time=1 ;;
 	t)	opt_trace=1; trace=$OPTARG ;;
 	v)	opt_timestr=1 ;;
 	x)	opt_failonly=1 ;;
 	Z)	opt_zone=1 ;;
 	h|?)	cat <<-END >&2
 		USAGE: statsnoop [-a|-A|-ceghlsvxZ] [-f pathname] [-t syscall]
 		                 [-n execname] [-p PID]
 		       statsnoop                # default output
 		                -a              # print most data
 		                -A              # dump all data, space delimited
 		                -c              # print cwd of process
 		                -e              # print errno value
 		                -g              # print command arguments
 		                -l              # print syscall type
 		                -s              # print start time, us
 		                -v              # print start time, string
 		                -x              # only print failed stats
 		                -Z              # print zonename
 		                -f pathname	# pathname name to snoop
 		                -n name		# process name to snoop
 		                -p PID		# process ID to snoop
 		                -t syscall	# stat syscall to trace
 		  eg,
 		       statsnoop -v             # human readable timestamps
 		       statsnoop -e             # see error codes
 		       statsnoop -f /etc/motd   # snoop this file only
 		END
 		exit 1
 	esac
 done
 
 ### Option logic
 if [ $opt_dump -eq 1 ]; then
 	opt_zone=0; opt_cwd=0; opt_time=0; opt_timestr=0; opt_type=0
 	opt_args=2 
 fi
 if [ $opt_name -eq 1 -o $opt_pid -eq 1 -o $opt_trace -eq 1 ]; then
 	filter=1
 fi
 
 
 #################################
 # --- Main Program, DTrace ---
 #
 /usr/sbin/dtrace -n '
  /*
   * Command line arguments
   */
  inline int OPT_dump 	 = '$opt_dump';
  inline int OPT_file 	 = '$opt_file';
  inline int OPT_args 	 = '$opt_args';
  inline int OPT_cwd	 = '$opt_cwd';
  inline int OPT_err	 = '$opt_err';
  inline int OPT_zone 	 = '$opt_zone';
  inline int OPT_time 	 = '$opt_time';
  inline int OPT_timestr	 = '$opt_timestr';
  inline int OPT_type	 = '$opt_type';
  inline int OPT_failonly = '$opt_failonly';
  inline int OPT_pid	 = '$opt_pid';
  inline int OPT_name	 = '$opt_name';
  inline int OPT_trace	 = '$opt_trace';
  inline int FILTER 	 = '$filter';
  inline int PID		 = '$pid';
  inline string PATHNAME	 = "'$pathname'";
  inline string NAME	 = "'$pname'";
  inline string TRACE	 = "'$trace'";
 
  #pragma D option quiet
  #pragma D option switchrate=10hz
 
  /*
   * Print header
   */
  dtrace:::BEGIN 
  {
 	/* print optional headers */
  	OPT_time ? printf("%-14s ", "TIME") : 1;
  	OPT_timestr ? printf("%-20s ", "STRTIME") : 1;
  	OPT_zone ? printf("%-10s ", "ZONE") : 1;
 
 	/* print dump headers */
 	OPT_dump ? printf("%s %s %s %s %s %s %s %s %s %s %s", "ZONE",
 	    "TIME", "UID", "PID", "PPID", "COMM", "FD", "ERR", "CWD", 
 	    "PATH", "ARGS") : printf("%5s %6s ","UID","PID");
 	
 	/* print main headers */
 	OPT_args == 0 ? printf("%-12s ", "COMM") : 1;
 	OPT_dump == 0 ? printf("%3s ", "FD") : 1;
 	OPT_err ? printf("%3s ", "ERR") : 1;
 	OPT_cwd ? printf("%-20s ", "CWD") : 1;
 	OPT_type ? printf("%-8s ", "TYPE") : 1;
 	OPT_dump == 0 ? printf("%-20s ", "PATH") : 1;
 	OPT_args == 1 ? printf("%s", "ARGS") : 1;
 	printf("\n");
  }
 
  /*
   * Print stat event
   */
  syscall::stat:entry, syscall::stat64:entry, syscall::xstat:entry,
  syscall::lstat:entry, syscall::lstat64:entry, syscall::lxstat:entry,
  syscall::fstat:entry, syscall::fstat64:entry, syscall::fxstat:entry
  {
 	/* default is to trace unless filtering */
 	self->ok = FILTER ? 0 : 1;
 
 	/* check each filter */
 	(OPT_name == 1 && NAME == execname) ? self->ok = 1 : 1;
 	(OPT_pid == 1 && PID == pid) ? self->ok = 1 : 1;
 	(OPT_trace == 1 && TRACE == probefunc) ? self->ok = 1 : 1;
  }
 
  syscall::stat:entry, syscall::stat64:entry,
  syscall::lstat:entry, syscall::lstat64:entry, syscall::lxstat:entry
  /self->ok/
  {
 	self->pathp = arg0;
  }
 
  syscall::xstat:entry
  /self->ok/
  {
 	self->pathp = arg1;
  }
 
  syscall::stat:return, syscall::stat64:return, syscall::xstat:return,
  syscall::lstat:return, syscall::lstat64:return, syscall::lxstat:return
  /self->ok/
  {
 	self->path = copyinstr(self->pathp);
 	self->pathp = 0;
  }
 
  syscall::fstat:return, syscall::fstat64:entry, syscall::fxstat:entry
  /self->ok/
  {
 	self->filep = curthread->t_procp->p_user.u_finfo.fi_list[arg0].uf_file;
  }
 
  syscall::fstat:return, syscall::fstat64:return, syscall::fxstat:return
  /self->ok/
  {
         this->vnodep = self->filep != 0 ? self->filep->f_vnode : 0;
         self->path = this->vnodep ? (this->vnodep->v_path != 0 ?
             cleanpath(this->vnodep->v_path) : "<unknown>") : "";
 	self->filep = 0;
  }
 
  syscall::stat:return, syscall::stat64:return, syscall::xstat:return,
  syscall::lstat:return, syscall::lstat64:return, syscall::lxstat:return,
  syscall::fstat:return, syscall::fstat64:return, syscall::fxstat:return
  /self->ok && (! OPT_failonly || (int)arg0 < 0) && 
      ((OPT_file == 0) || (OPT_file == 1 && PATHNAME == copyinstr(self->pathp)))/
  {
 	/* print optional fields */
  	OPT_time ? printf("%-14d ", timestamp/1000) : 1;
  	OPT_timestr ? printf("%-20Y ", walltimestamp) : 1;
  	OPT_zone ? printf("%-10s ", zonename) : 1;
 
 	/* print dump fields */
 	OPT_dump ? printf("%s %d %d %d %d %s %d %d %s %s %S", zonename,
 	    timestamp/1000, uid, pid, ppid, execname, (int)arg0, errno,
 	    cwd, self->path, curpsinfo->pr_psargs) :
 	    printf("%5d %6d ", uid, pid);
 
 	/* print main fields */
 	OPT_args == 0 ? printf("%-12.12s ", execname) : 1;
 	OPT_dump == 0 ? printf("%3d ", (int)arg0) : 1;
 	OPT_err ? printf("%3d ", errno) : 1;
 	OPT_cwd ? printf("%-20s ", cwd) : 1;
 	OPT_type ? printf("%-8s ", probefunc) : 1;
 	OPT_dump == 0 ? printf("%-20s ", self->path) : 1;
 	OPT_args == 1 ? printf("%S", curpsinfo->pr_psargs) : 1;
 	printf("\n");
 
 	/* cleanup */
 	self->path = 0;
 	self->ok = 0;
  }
 
  /* 
   * Cleanup 
   */
  syscall::stat:return, syscall::stat64:return, syscall::xstat:return,
  syscall::lstat:return, syscall::lstat64:return, syscall::lxstat:return,
  syscall::fstat:return, syscall::fstat64:return, syscall::fxstat:return
  /self->ok/
  {
 	self->path = 0;
 	self->ok = 0;
  }
 '