aboutsummaryrefslogtreecommitdiff
path: root/cddl/contrib/dtracetoolkit/rwtop
blob: ed0a6faaa47553046ac64317268f8b500b7a6807 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
#!/usr/bin/ksh
#
# rwtop - display top read/write bytes by process.
#         Written using DTrace (Solaris 10 3/05).
#
# This is measuring reads and writes at the application level. This matches
# read and write system calls.
#
# $Id: rwtop 3 2007-08-01 10:50:08Z brendan $
#
# USAGE:	rwtop [-cC] [-j|-Z] [-n name] [-p pid]
#		      [-t top] [interval [count]]
# 
#		rwtop		# default output, 5 second samples
#
#		-C		# don't clear the screen
#		-c		# print counts
#		-j		# print project ID
#		-Z		# print zone ID
#		-n name		# this process name only
#		-p PID		# this PID only
#		-t top		# print top number only
#	eg,
#		rwtop 1		# 1 second samples
#		rwtop -t 10	# print top 10 only
#		rwtop -n bash	# monitor processes named "bash"
#		rwtop -C 5 12	# print 12 x 5 second samples
#
# FIELDS:
#		ZONE		Zone ID
#		PROJ		Project ID
#		UID		User ID
#		PID		Process ID
#		PPID		Parent Process ID
#		CMD		Process name
#		D		Direction, Read or Write
#		BYTES		Total bytes during sample
#		app_r		total reads during sample, Kbytes
#		app_w		total writes during sample, Kbytes
#
# SEE ALSO:	iotop
#
# INSPIRATION:  top(1) by William LeFebvre
#
# COPYRIGHT: Copyright (c) 2005, 2006 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]
#
# 24-Jul-2005   Brendan Gregg   Created this.
# 20-Apr-2006	   "      "	Last update.


##############################
# --- Process Arguments ---
#

### default variables
opt_name=0; opt_pid=0; opt_clear=1; opt_proj=0; opt_zone=0
opt_def=1; opt_bytes=1; filter=0; pname=.; pid=0
opt_top=0; opt_count=0; interval=5; count=-1; top=0

### process options
while getopts Cchn:p:jt:Z name
do
	case $name in
	C)	opt_clear=0 ;;
	c)	opt_count=1; opt_bytes=0 ;;
	n)	opt_name=1; pname=$OPTARG ;;
	p)	opt_pid=1; pid=$OPTARG ;;
	j)	opt_proj=1; opt_def=0 ;;
	t)	opt_top=1; top=$OPTARG ;;
	Z)	opt_zone=1; opt_def=0 ;;
	h|?)	cat <<-END >&2
		USAGE: rwtop [-cC] [-j|-Z] [-n name] [-p pid]
		             [-t top] [interval [count]]
 
		                -C        # don't clear the screen
		                -c        # print counts
		                -j        # print project ID
		                -Z        # print zone ID
		                -n name   # this process name only
		                -p PID    # this PID only
		                -t top    # print top number only
		   eg,
		        rwtop          # default output, 5 second samples
		        rwtop 1        # 1 second samples
		        rwtop -t 10    # print top 10 only
		        rwtop -n bash  # monitor processes named "bash"
		        rwtop -C 5 12  # print 12 x 5 second samples
		END
		exit 1
	esac
done

shift $(( $OPTIND - 1 ))

### option logic
if [[ "$1" > 0 ]]; then
        interval=$1; shift
fi
if [[ "$1" > 0 ]]; then
        count=$1; shift
fi
if (( opt_proj && opt_zone )); then
        opt_proj=0
fi
if (( opt_name || opt_pid )); then
	filter=1
fi
if (( opt_clear )); then
        clearstr=`clear`
else
        clearstr=.
fi



#################################
# --- Main Program, DTrace ---
#
/usr/sbin/dtrace -n '
 /*
  * Command line arguments
  */
 inline int OPT_def 	= '$opt_def';
 inline int OPT_proj 	= '$opt_proj';
 inline int OPT_zone 	= '$opt_zone';
 inline int OPT_clear 	= '$opt_clear';
 inline int OPT_bytes 	= '$opt_bytes';
 inline int OPT_count	= '$opt_count';
 inline int OPT_name 	= '$opt_name';
 inline int OPT_pid 	= '$opt_pid';
 inline int OPT_top 	= '$opt_top';
 inline int INTERVAL 	= '$interval';
 inline int COUNTER 	= '$count';
 inline int FILTER 	= '$filter';
 inline int TOP 	= '$top';
 inline int PID		= '$pid';
 inline string NAME 	= "'$pname'";
 inline string CLEAR 	= "'$clearstr'";
 
 #pragma D option quiet

 /*
  * Print header
  */
 dtrace:::BEGIN 
 {
        /* starting values */
        counts = COUNTER;
        secs = INTERVAL;
        app_r = 0;
        app_w = 0;

        printf("Tracing... Please wait.\n");
 }

 /*
  * Check event is being traced
  */
 sysinfo:::readch,
 sysinfo:::writech
 /pid != $pid/
 { 
	/* default is to trace unless filtering, */
	this->ok = FILTER ? 0 : 1;

	/* check each filter, */
	(OPT_name == 1 && NAME == execname)? this->ok = 1 : 1;
	(OPT_pid == 1 && PID == pid) ? this->ok = 1 : 1;
 }

 /*
  * Increment tallys
  */
 sysinfo:::readch
 /this->ok/
 {
	app_r += arg0;
 }
 sysinfo:::writech
 /this->ok/
 {
	app_w += arg0;
 }

 /*
  * Process event
  */
 sysinfo:::readch,
 sysinfo:::writech
 /this->ok/
 {
	/* choose statistic to track */
	this->value = OPT_bytes ? arg0 : 1;
	
	/*
	 * Save details
	 */
	OPT_def ? @out[uid, pid, ppid, execname,
	    probename == "readch" ? "R" : "W"] = sum(this->value) : 1;
	OPT_proj ? @out[curpsinfo->pr_projid, pid, ppid, execname,
	    probename == "readch" ? "R" : "W"] = sum(this->value) : 1;
	OPT_zone ? @out[curpsinfo->pr_zoneid, pid, ppid, execname,
	    probename == "readch" ? "R" : "W"] = sum(this->value) : 1;

	this->ok = 0;
 }

 /*
  * Timer
  */
 profile:::tick-1sec
 {
	secs--;
 }

 /*
  * Print Report
  */
 profile:::tick-1sec
 /secs == 0/
 {
	/* fetch 1 min load average */
	this->load1a  = `hp_avenrun[0] / 65536;
	this->load1b  = ((`hp_avenrun[0] % 65536) * 100) / 65536;

	/* convert counters to Kbytes */
	app_r /= 1024;
	app_w /= 1024;

	/* print status */
	OPT_clear ? printf("%s", CLEAR) : 1;
	printf("%Y,  load: %d.%02d,  app_r: %6d KB,  app_w: %6d KB\n\n",
	    walltimestamp, this->load1a, this->load1b, app_r, app_w);

	/* print headers */
	OPT_def  ? printf("  UID ") : 1;
	OPT_proj ? printf(" PROJ ") : 1;
	OPT_zone ? printf(" ZONE ") : 1;
	printf("%6s %6s %-16s %1s",
	    "PID", "PPID", "CMD", "D");
	OPT_bytes ? printf(" %16s\n", "BYTES") : 1;
	OPT_count ? printf(" %16s\n", "COUNT") : 1;

	/* truncate to top lines if needed */
	OPT_top ? trunc(@out, TOP) : 1;

	/* print data */
	printa("%5d %6d %6d %-16s %1s %16@d\n", @out);
	printf("\n");

	/* clear data */
	trunc(@out);
	app_r = 0;
	app_w = 0;
	secs = INTERVAL;
	counts--;
 }

 /*
  * End of program
  */
 profile:::tick-1sec
 /counts == 0/
 {
	exit(0);
 }

 /*
  * Cleanup for Ctrl-C
  */
 dtrace:::END
 {
	trunc(@out);
 }
'