xref: /illumos-gate/usr/src/cmd/plockstat/plockstat.c (revision 6bc6f338)
17c478bd9Sstevel@tonic-gate /*
27c478bd9Sstevel@tonic-gate  * CDDL HEADER START
37c478bd9Sstevel@tonic-gate  *
47c478bd9Sstevel@tonic-gate  * The contents of this file are subject to the terms of the
5351346dbSahl  * Common Development and Distribution License (the "License").
6351346dbSahl  * You may not use this file except in compliance with the License.
77c478bd9Sstevel@tonic-gate  *
87c478bd9Sstevel@tonic-gate  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
97c478bd9Sstevel@tonic-gate  * or http://www.opensolaris.org/os/licensing.
107c478bd9Sstevel@tonic-gate  * See the License for the specific language governing permissions
117c478bd9Sstevel@tonic-gate  * and limitations under the License.
127c478bd9Sstevel@tonic-gate  *
137c478bd9Sstevel@tonic-gate  * When distributing Covered Code, include this CDDL HEADER in each
147c478bd9Sstevel@tonic-gate  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
157c478bd9Sstevel@tonic-gate  * If applicable, add the following below this CDDL HEADER, with the
167c478bd9Sstevel@tonic-gate  * fields enclosed by brackets "[]" replaced with your own identifying
177c478bd9Sstevel@tonic-gate  * information: Portions Copyright [yyyy] [name of copyright owner]
187c478bd9Sstevel@tonic-gate  *
197c478bd9Sstevel@tonic-gate  * CDDL HEADER END
207c478bd9Sstevel@tonic-gate  */
21351346dbSahl 
227c478bd9Sstevel@tonic-gate /*
23351346dbSahl  * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
247c478bd9Sstevel@tonic-gate  * Use is subject to license terms.
257c478bd9Sstevel@tonic-gate  */
267c478bd9Sstevel@tonic-gate 
277c478bd9Sstevel@tonic-gate #include <assert.h>
287c478bd9Sstevel@tonic-gate #include <dtrace.h>
297c478bd9Sstevel@tonic-gate #include <limits.h>
307c478bd9Sstevel@tonic-gate #include <link.h>
317c478bd9Sstevel@tonic-gate #include <priv.h>
327c478bd9Sstevel@tonic-gate #include <signal.h>
337c478bd9Sstevel@tonic-gate #include <stdlib.h>
347c478bd9Sstevel@tonic-gate #include <stdarg.h>
357c478bd9Sstevel@tonic-gate #include <stdio.h>
367c478bd9Sstevel@tonic-gate #include <string.h>
377c478bd9Sstevel@tonic-gate #include <strings.h>
387c478bd9Sstevel@tonic-gate #include <errno.h>
397c478bd9Sstevel@tonic-gate #include <sys/wait.h>
40351346dbSahl #include <libgen.h>
417c478bd9Sstevel@tonic-gate #include <libproc.h>
427c478bd9Sstevel@tonic-gate 
437c478bd9Sstevel@tonic-gate static char *g_pname;
447c478bd9Sstevel@tonic-gate static dtrace_hdl_t *g_dtp;
457c478bd9Sstevel@tonic-gate struct ps_prochandle *g_pr;
467c478bd9Sstevel@tonic-gate 
477c478bd9Sstevel@tonic-gate #define	E_SUCCESS	0
487c478bd9Sstevel@tonic-gate #define	E_ERROR		1
497c478bd9Sstevel@tonic-gate #define	E_USAGE		2
507c478bd9Sstevel@tonic-gate 
517c478bd9Sstevel@tonic-gate /*
527c478bd9Sstevel@tonic-gate  * For hold times we use a global associative array since for mutexes, in
537c478bd9Sstevel@tonic-gate  * user-land, it's not invalid to release a sychonization primitive that
547c478bd9Sstevel@tonic-gate  * another thread acquired; rwlocks require a thread-local associative array
557c478bd9Sstevel@tonic-gate  * since multiple thread can hold the same lock for reading. Note that we
567c478bd9Sstevel@tonic-gate  * ignore recursive mutex acquisitions and releases as they don't truly
577c478bd9Sstevel@tonic-gate  * affect lock contention.
587c478bd9Sstevel@tonic-gate  */
597c478bd9Sstevel@tonic-gate static const char *g_hold_init =
607c478bd9Sstevel@tonic-gate "plockstat$target:::rw-acquire\n"
617c478bd9Sstevel@tonic-gate "{\n"
627c478bd9Sstevel@tonic-gate "	self->rwhold[arg0] = timestamp;\n"
637c478bd9Sstevel@tonic-gate "}\n"
647c478bd9Sstevel@tonic-gate "plockstat$target:::mutex-acquire\n"
657c478bd9Sstevel@tonic-gate "/arg1 == 0/\n"
667c478bd9Sstevel@tonic-gate "{\n"
677c478bd9Sstevel@tonic-gate "	mtxhold[arg0] = timestamp;\n"
687c478bd9Sstevel@tonic-gate "}\n";
697c478bd9Sstevel@tonic-gate 
707c478bd9Sstevel@tonic-gate static const char *g_hold_histogram =
717c478bd9Sstevel@tonic-gate "plockstat$target:::rw-release\n"
727c478bd9Sstevel@tonic-gate "/self->rwhold[arg0] && arg1 == 1/\n"
737c478bd9Sstevel@tonic-gate "{\n"
747c478bd9Sstevel@tonic-gate "	@rw_w_hold[arg0, ustack()] =\n"
757c478bd9Sstevel@tonic-gate "	    quantize(timestamp - self->rwhold[arg0]);\n"
767c478bd9Sstevel@tonic-gate "	self->rwhold[arg0] = 0;\n"
77351346dbSahl "	rw_w_hold_found = 1;\n"
787c478bd9Sstevel@tonic-gate "}\n"
797c478bd9Sstevel@tonic-gate "plockstat$target:::rw-release\n"
807c478bd9Sstevel@tonic-gate "/self->rwhold[arg0]/\n"
817c478bd9Sstevel@tonic-gate "{\n"
827c478bd9Sstevel@tonic-gate "	@rw_r_hold[arg0, ustack()] =\n"
837c478bd9Sstevel@tonic-gate "	    quantize(timestamp - self->rwhold[arg0]);\n"
847c478bd9Sstevel@tonic-gate "	self->rwhold[arg0] = 0;\n"
85351346dbSahl "	rw_r_hold_found = 1;\n"
867c478bd9Sstevel@tonic-gate "}\n"
877c478bd9Sstevel@tonic-gate "plockstat$target:::mutex-release\n"
887c478bd9Sstevel@tonic-gate "/mtxhold[arg0] && arg1 == 0/\n"
897c478bd9Sstevel@tonic-gate "{\n"
907c478bd9Sstevel@tonic-gate "	@mtx_hold[arg0, ustack()] = quantize(timestamp - mtxhold[arg0]);\n"
917c478bd9Sstevel@tonic-gate "	mtxhold[arg0] = 0;\n"
92351346dbSahl "	mtx_hold_found = 1;\n"
93351346dbSahl "}\n"
94351346dbSahl "\n"
95351346dbSahl "END\n"
96351346dbSahl "/mtx_hold_found/\n"
97351346dbSahl "{\n"
98351346dbSahl "	trace(\"Mutex hold\");\n"
99351346dbSahl "	printa(@mtx_hold);\n"
100351346dbSahl "}\n"
101351346dbSahl "END\n"
102351346dbSahl "/rw_r_hold_found/\n"
103351346dbSahl "{\n"
104351346dbSahl "	trace(\"R/W reader hold\");\n"
105351346dbSahl "	printa(@rw_r_hold);\n"
106351346dbSahl "}\n"
107351346dbSahl "END\n"
108351346dbSahl "/rw_w_hold_found/\n"
109351346dbSahl "{\n"
110351346dbSahl "	trace(\"R/W writer hold\");\n"
111351346dbSahl "	printa(@rw_w_hold);\n"
1127c478bd9Sstevel@tonic-gate "}\n";
1137c478bd9Sstevel@tonic-gate 
1147c478bd9Sstevel@tonic-gate static const char *g_hold_times =
1157c478bd9Sstevel@tonic-gate "plockstat$target:::rw-release\n"
1167c478bd9Sstevel@tonic-gate "/self->rwhold[arg0] && arg1 == 1/\n"
1177c478bd9Sstevel@tonic-gate "{\n"
118351346dbSahl "	@rw_w_hold[arg0, ustack(5)] = sum(timestamp - self->rwhold[arg0]);\n"
119351346dbSahl "	@rw_w_hold_count[arg0, ustack(5)] = count();\n"
1207c478bd9Sstevel@tonic-gate "	self->rwhold[arg0] = 0;\n"
121351346dbSahl "	rw_w_hold_found = 1;\n"
1227c478bd9Sstevel@tonic-gate "}\n"
1237c478bd9Sstevel@tonic-gate "plockstat$target:::rw-release\n"
1247c478bd9Sstevel@tonic-gate "/self->rwhold[arg0]/\n"
1257c478bd9Sstevel@tonic-gate "{\n"
126351346dbSahl "	@rw_r_hold[arg0, ustack(5)] = sum(timestamp - self->rwhold[arg0]);\n"
127351346dbSahl "	@rw_r_hold_count[arg0, ustack(5)] = count();\n"
1287c478bd9Sstevel@tonic-gate "	self->rwhold[arg0] = 0;\n"
129351346dbSahl "	rw_r_hold_found = 1;\n"
1307c478bd9Sstevel@tonic-gate "}\n"
1317c478bd9Sstevel@tonic-gate "plockstat$target:::mutex-release\n"
1327c478bd9Sstevel@tonic-gate "/mtxhold[arg0] && arg1 == 0/\n"
1337c478bd9Sstevel@tonic-gate "{\n"
134351346dbSahl "	@mtx_hold[arg0, ustack(5)] = sum(timestamp - mtxhold[arg0]);\n"
135351346dbSahl "	@mtx_hold_count[arg0, ustack(5)] = count();\n"
1367c478bd9Sstevel@tonic-gate "	mtxhold[arg0] = 0;\n"
137351346dbSahl "	mtx_hold_found = 1;\n"
138351346dbSahl "}\n"
139351346dbSahl "\n"
140351346dbSahl "END\n"
141351346dbSahl "/mtx_hold_found/\n"
142351346dbSahl "{\n"
143351346dbSahl "	trace(\"Mutex hold\");\n"
144351346dbSahl "	printa(@mtx_hold, @mtx_hold_count);\n"
145351346dbSahl "}\n"
146351346dbSahl "END\n"
147351346dbSahl "/rw_r_hold_found/\n"
148351346dbSahl "{\n"
149351346dbSahl "	trace(\"R/W reader hold\");\n"
150351346dbSahl "	printa(@rw_r_hold, @rw_r_hold_count);\n"
151351346dbSahl "}\n"
152351346dbSahl "END\n"
153351346dbSahl "/rw_w_hold_found/\n"
154351346dbSahl "{\n"
155351346dbSahl "	trace(\"R/W writer hold\");\n"
156351346dbSahl "	printa(@rw_w_hold, @rw_w_hold_count);\n"
1577c478bd9Sstevel@tonic-gate "}\n";
1587c478bd9Sstevel@tonic-gate 
1597c478bd9Sstevel@tonic-gate 
1607c478bd9Sstevel@tonic-gate /*
1617c478bd9Sstevel@tonic-gate  * For contention, we use thread-local associative arrays since we're tracing
1627c478bd9Sstevel@tonic-gate  * a single thread's activity in libc and multiple threads can be blocking or
1637c478bd9Sstevel@tonic-gate  * spinning on the same sychonization primitive.
1647c478bd9Sstevel@tonic-gate  */
1657c478bd9Sstevel@tonic-gate static const char *g_ctnd_init =
1667c478bd9Sstevel@tonic-gate "plockstat$target:::rw-block\n"
1677c478bd9Sstevel@tonic-gate "{\n"
1687c478bd9Sstevel@tonic-gate "	self->rwblock[arg0] = timestamp;\n"
1697c478bd9Sstevel@tonic-gate "}\n"
1707c478bd9Sstevel@tonic-gate "plockstat$target:::mutex-block\n"
1717c478bd9Sstevel@tonic-gate "{\n"
1727c478bd9Sstevel@tonic-gate "	self->mtxblock[arg0] = timestamp;\n"
1737c478bd9Sstevel@tonic-gate "}\n"
1747c478bd9Sstevel@tonic-gate "plockstat$target:::mutex-spin\n"
1757c478bd9Sstevel@tonic-gate "{\n"
1767c478bd9Sstevel@tonic-gate "	self->mtxspin[arg0] = timestamp;\n"
1777c478bd9Sstevel@tonic-gate "}\n";
1787c478bd9Sstevel@tonic-gate 
1797c478bd9Sstevel@tonic-gate static const char *g_ctnd_histogram =
1807c478bd9Sstevel@tonic-gate "plockstat$target:::rw-blocked\n"
1817c478bd9Sstevel@tonic-gate "/self->rwblock[arg0] && arg1 == 1 && arg2 != 0/\n"
1827c478bd9Sstevel@tonic-gate "{\n"
1837c478bd9Sstevel@tonic-gate "	@rw_w_block[arg0, ustack()] =\n"
1847c478bd9Sstevel@tonic-gate "	    quantize(timestamp - self->rwblock[arg0]);\n"
1857c478bd9Sstevel@tonic-gate "	self->rwblock[arg0] = 0;\n"
186351346dbSahl "	rw_w_block_found = 1;\n"
1877c478bd9Sstevel@tonic-gate "}\n"
1887c478bd9Sstevel@tonic-gate "plockstat$target:::rw-blocked\n"
1897c478bd9Sstevel@tonic-gate "/self->rwblock[arg0] && arg2 != 0/\n"
1907c478bd9Sstevel@tonic-gate "{\n"
1917c478bd9Sstevel@tonic-gate "	@rw_r_block[arg0, ustack()] =\n"
1927c478bd9Sstevel@tonic-gate "	    quantize(timestamp - self->rwblock[arg0]);\n"
1937c478bd9Sstevel@tonic-gate "	self->rwblock[arg0] = 0;\n"
194351346dbSahl "	rw_r_block_found = 1;\n"
1957c478bd9Sstevel@tonic-gate "}\n"
1967c478bd9Sstevel@tonic-gate "plockstat$target:::rw-blocked\n"
1977c478bd9Sstevel@tonic-gate "/self->rwblock[arg0]/\n"
1987c478bd9Sstevel@tonic-gate "{\n"
1997c478bd9Sstevel@tonic-gate "	self->rwblock[arg0] = 0;\n"
2007c478bd9Sstevel@tonic-gate "}\n"
2017c478bd9Sstevel@tonic-gate "plockstat$target:::mutex-spun\n"
2027c478bd9Sstevel@tonic-gate "/self->mtxspin[arg0] && arg1 != 0/\n"
2037c478bd9Sstevel@tonic-gate "{\n"
2047c478bd9Sstevel@tonic-gate "	@mtx_spin[arg0, ustack()] =\n"
2057c478bd9Sstevel@tonic-gate "	    quantize(timestamp - self->mtxspin[arg0]);\n"
2067c478bd9Sstevel@tonic-gate "	self->mtxspin[arg0] = 0;\n"
207351346dbSahl "	mtx_spin_found = 1;\n"
2087c478bd9Sstevel@tonic-gate "}\n"
2097c478bd9Sstevel@tonic-gate "plockstat$target:::mutex-spun\n"
2107c478bd9Sstevel@tonic-gate "/self->mtxspin[arg0]/\n"
2117c478bd9Sstevel@tonic-gate "{\n"
2127c478bd9Sstevel@tonic-gate "	@mtx_vain_spin[arg0, ustack()] =\n"
2137c478bd9Sstevel@tonic-gate "	    quantize(timestamp - self->mtxspin[arg0]);\n"
2147c478bd9Sstevel@tonic-gate "	self->mtxspin[arg0] = 0;\n"
215351346dbSahl "	mtx_vain_spin_found = 1;\n"
2167c478bd9Sstevel@tonic-gate "}\n"
2177c478bd9Sstevel@tonic-gate "plockstat$target:::mutex-blocked\n"
2187c478bd9Sstevel@tonic-gate "/self->mtxblock[arg0] && arg1 != 0/\n"
2197c478bd9Sstevel@tonic-gate "{\n"
2207c478bd9Sstevel@tonic-gate "	@mtx_block[arg0, ustack()] =\n"
2217c478bd9Sstevel@tonic-gate "	    quantize(timestamp - self->mtxblock[arg0]);\n"
2227c478bd9Sstevel@tonic-gate "	self->mtxblock[arg0] = 0;\n"
223351346dbSahl "	mtx_block_found = 1;\n"
2247c478bd9Sstevel@tonic-gate "}\n"
2257c478bd9Sstevel@tonic-gate "plockstat$target:::mutex-blocked\n"
2267c478bd9Sstevel@tonic-gate "/self->mtxblock[arg0]/\n"
2277c478bd9Sstevel@tonic-gate "{\n"
2287c478bd9Sstevel@tonic-gate "	self->mtxblock[arg0] = 0;\n"
229351346dbSahl "}\n"
230351346dbSahl "\n"
231351346dbSahl "END\n"
232351346dbSahl "/mtx_block_found/\n"
233351346dbSahl "{\n"
234351346dbSahl "	trace(\"Mutex block\");\n"
235351346dbSahl "	printa(@mtx_block);\n"
236351346dbSahl "}\n"
237351346dbSahl "END\n"
238351346dbSahl "/mtx_spin_found/\n"
239351346dbSahl "{\n"
240351346dbSahl "	trace(\"Mutex spin\");\n"
241351346dbSahl "	printa(@mtx_spin);\n"
242351346dbSahl "}\n"
243351346dbSahl "END\n"
244351346dbSahl "/mtx_vain_spin_found/\n"
245351346dbSahl "{\n"
246351346dbSahl "	trace(\"Mutex unsuccessful spin\");\n"
247351346dbSahl "	printa(@mtx_vain_spin);\n"
248351346dbSahl "}\n"
249351346dbSahl "END\n"
250351346dbSahl "/rw_r_block_found/\n"
251351346dbSahl "{\n"
252351346dbSahl "	trace(\"R/W reader block\");\n"
253351346dbSahl "	printa(@rw_r_block);\n"
254351346dbSahl "}\n"
255351346dbSahl "END\n"
256351346dbSahl "/rw_w_block_found/\n"
257351346dbSahl "{\n"
258351346dbSahl "	trace(\"R/W writer block\");\n"
259351346dbSahl "	printa(@rw_w_block);\n"
2607c478bd9Sstevel@tonic-gate "}\n";
2617c478bd9Sstevel@tonic-gate 
2627c478bd9Sstevel@tonic-gate 
2637c478bd9Sstevel@tonic-gate static const char *g_ctnd_times =
2647c478bd9Sstevel@tonic-gate "plockstat$target:::rw-blocked\n"
2657c478bd9Sstevel@tonic-gate "/self->rwblock[arg0] && arg1 == 1 && arg2 != 0/\n"
2667c478bd9Sstevel@tonic-gate "{\n"
2677c478bd9Sstevel@tonic-gate "	@rw_w_block[arg0, ustack(5)] =\n"
268351346dbSahl "	    sum(timestamp - self->rwblock[arg0]);\n"
269351346dbSahl "	@rw_w_block_count[arg0, ustack(5)] = count();\n"
2707c478bd9Sstevel@tonic-gate "	self->rwblock[arg0] = 0;\n"
271351346dbSahl "	rw_w_block_found = 1;\n"
2727c478bd9Sstevel@tonic-gate "}\n"
2737c478bd9Sstevel@tonic-gate "plockstat$target:::rw-blocked\n"
2747c478bd9Sstevel@tonic-gate "/self->rwblock[arg0] && arg2 != 0/\n"
2757c478bd9Sstevel@tonic-gate "{\n"
2767c478bd9Sstevel@tonic-gate "	@rw_r_block[arg0, ustack(5)] =\n"
277351346dbSahl "	    sum(timestamp - self->rwblock[arg0]);\n"
278351346dbSahl "	@rw_r_block_count[arg0, ustack(5)] = count();\n"
2797c478bd9Sstevel@tonic-gate "	self->rwblock[arg0] = 0;\n"
280351346dbSahl "	rw_r_block_found = 1;\n"
2817c478bd9Sstevel@tonic-gate "}\n"
2827c478bd9Sstevel@tonic-gate "plockstat$target:::rw-blocked\n"
2837c478bd9Sstevel@tonic-gate "/self->rwblock[arg0]/\n"
2847c478bd9Sstevel@tonic-gate "{\n"
2857c478bd9Sstevel@tonic-gate "	self->rwblock[arg0] = 0;\n"
2867c478bd9Sstevel@tonic-gate "}\n"
2877c478bd9Sstevel@tonic-gate "plockstat$target:::mutex-spun\n"
2887c478bd9Sstevel@tonic-gate "/self->mtxspin[arg0] && arg1 != 0/\n"
2897c478bd9Sstevel@tonic-gate "{\n"
2907c478bd9Sstevel@tonic-gate "	@mtx_spin[arg0, ustack(5)] =\n"
291351346dbSahl "	    sum(timestamp - self->mtxspin[arg0]);\n"
292351346dbSahl "	@mtx_spin_count[arg0, ustack(5)] = count();\n"
2937c478bd9Sstevel@tonic-gate "	self->mtxspin[arg0] = 0;\n"
294351346dbSahl "	mtx_spin_found = 1;\n"
2957c478bd9Sstevel@tonic-gate "}\n"
2967c478bd9Sstevel@tonic-gate "plockstat$target:::mutex-spun\n"
2977c478bd9Sstevel@tonic-gate "/self->mtxspin[arg0]/\n"
2987c478bd9Sstevel@tonic-gate "{\n"
2997c478bd9Sstevel@tonic-gate "	@mtx_vain_spin[arg0, ustack(5)] =\n"
300351346dbSahl "	    sum(timestamp - self->mtxspin[arg0]);\n"
301351346dbSahl "	@mtx_vain_spin_count[arg0, ustack(5)] = count();\n"
3027c478bd9Sstevel@tonic-gate "	self->mtxspin[arg0] = 0;\n"
303351346dbSahl "	mtx_vain_spin_found = 1;\n"
3047c478bd9Sstevel@tonic-gate "}\n"
3057c478bd9Sstevel@tonic-gate "plockstat$target:::mutex-blocked\n"
3067c478bd9Sstevel@tonic-gate "/self->mtxblock[arg0] && arg1 != 0/\n"
3077c478bd9Sstevel@tonic-gate "{\n"
3087c478bd9Sstevel@tonic-gate "	@mtx_block[arg0, ustack(5)] =\n"
309351346dbSahl "	    sum(timestamp - self->mtxblock[arg0]);\n"
310351346dbSahl "	@mtx_block_count[arg0, ustack(5)] = count();\n"
3117c478bd9Sstevel@tonic-gate "	self->mtxblock[arg0] = 0;\n"
312351346dbSahl "	mtx_block_found = 1;\n"
3137c478bd9Sstevel@tonic-gate "}\n"
3147c478bd9Sstevel@tonic-gate "plockstat$target:::mutex-blocked\n"
3157c478bd9Sstevel@tonic-gate "/self->mtxblock[arg0]/\n"
3167c478bd9Sstevel@tonic-gate "{\n"
3177c478bd9Sstevel@tonic-gate "	self->mtxblock[arg0] = 0;\n"
318351346dbSahl "}\n"
319351346dbSahl "\n"
320351346dbSahl "END\n"
321351346dbSahl "/mtx_block_found/\n"
322351346dbSahl "{\n"
323351346dbSahl "	trace(\"Mutex block\");\n"
324351346dbSahl "	printa(@mtx_block, @mtx_block_count);\n"
325351346dbSahl "}\n"
326351346dbSahl "END\n"
327351346dbSahl "/mtx_spin_found/\n"
328351346dbSahl "{\n"
329351346dbSahl "	trace(\"Mutex spin\");\n"
330351346dbSahl "	printa(@mtx_spin, @mtx_spin_count);\n"
331351346dbSahl "}\n"
332351346dbSahl "END\n"
333351346dbSahl "/mtx_vain_spin_found/\n"
334351346dbSahl "{\n"
335351346dbSahl "	trace(\"Mutex unsuccessful spin\");\n"
336351346dbSahl "	printa(@mtx_vain_spin, @mtx_vain_spin_count);\n"
337351346dbSahl "}\n"
338351346dbSahl "END\n"
339351346dbSahl "/rw_r_block_found/\n"
340351346dbSahl "{\n"
341351346dbSahl "	trace(\"R/W reader block\");\n"
342351346dbSahl "	printa(@rw_r_block, @rw_r_block_count);\n"
343351346dbSahl "}\n"
344351346dbSahl "END\n"
345351346dbSahl "/rw_w_block_found/\n"
346351346dbSahl "{\n"
347351346dbSahl "	trace(\"R/W writer block\");\n"
348351346dbSahl "	printa(@rw_w_block, @rw_w_block_count);\n"
3497c478bd9Sstevel@tonic-gate "}\n";
3507c478bd9Sstevel@tonic-gate 
351351346dbSahl static char g_prog[4096];
3527c478bd9Sstevel@tonic-gate static size_t g_proglen;
3537c478bd9Sstevel@tonic-gate static int g_opt_V, g_opt_s;
3547c478bd9Sstevel@tonic-gate static int g_intr;
355351346dbSahl static int g_exited;
3567c478bd9Sstevel@tonic-gate static dtrace_optval_t g_nframes;
3577c478bd9Sstevel@tonic-gate static ulong_t g_nent = ULONG_MAX;
3587c478bd9Sstevel@tonic-gate 
3597c478bd9Sstevel@tonic-gate #define	PLOCKSTAT_OPTSTR	"n:ps:e:vx:ACHV"
3607c478bd9Sstevel@tonic-gate 
3617c478bd9Sstevel@tonic-gate static void
usage(void)3627c478bd9Sstevel@tonic-gate usage(void)
3637c478bd9Sstevel@tonic-gate {
3647c478bd9Sstevel@tonic-gate 	(void) fprintf(stderr, "Usage:\n"
3657c478bd9Sstevel@tonic-gate 	    "\t%s [-vACHV] [-n count] [-s depth] [-e secs] [-x opt[=val]]\n"
3667c478bd9Sstevel@tonic-gate 	    "\t    command [arg...]\n"
3677c478bd9Sstevel@tonic-gate 	    "\t%s [-vACHV] [-n count] [-s depth] [-e secs] [-x opt[=val]]\n"
3687c478bd9Sstevel@tonic-gate 	    "\t    -p pid\n", g_pname, g_pname);
3697c478bd9Sstevel@tonic-gate 
3707c478bd9Sstevel@tonic-gate 	exit(E_USAGE);
3717c478bd9Sstevel@tonic-gate }
3727c478bd9Sstevel@tonic-gate 
3737c478bd9Sstevel@tonic-gate static void
verror(const char * fmt,va_list ap)3747c478bd9Sstevel@tonic-gate verror(const char *fmt, va_list ap)
3757c478bd9Sstevel@tonic-gate {
3767c478bd9Sstevel@tonic-gate 	int error = errno;
3777c478bd9Sstevel@tonic-gate 
3787c478bd9Sstevel@tonic-gate 	(void) fprintf(stderr, "%s: ", g_pname);
3797c478bd9Sstevel@tonic-gate 	(void) vfprintf(stderr, fmt, ap);
3807c478bd9Sstevel@tonic-gate 
3817c478bd9Sstevel@tonic-gate 	if (fmt[strlen(fmt) - 1] != '\n')
3827c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr, ": %s\n", strerror(error));
3837c478bd9Sstevel@tonic-gate }
3847c478bd9Sstevel@tonic-gate 
3857c478bd9Sstevel@tonic-gate /*PRINTFLIKE1*/
3867c478bd9Sstevel@tonic-gate static void
fatal(const char * fmt,...)3877c478bd9Sstevel@tonic-gate fatal(const char *fmt, ...)
3887c478bd9Sstevel@tonic-gate {
3897c478bd9Sstevel@tonic-gate 	va_list ap;
3907c478bd9Sstevel@tonic-gate 
3917c478bd9Sstevel@tonic-gate 	va_start(ap, fmt);
3927c478bd9Sstevel@tonic-gate 	verror(fmt, ap);
3937c478bd9Sstevel@tonic-gate 	va_end(ap);
3947c478bd9Sstevel@tonic-gate 
3957c478bd9Sstevel@tonic-gate 	if (g_pr != NULL && g_dtp != NULL)
3967c478bd9Sstevel@tonic-gate 		dtrace_proc_release(g_dtp, g_pr);
3977c478bd9Sstevel@tonic-gate 
3987c478bd9Sstevel@tonic-gate 	exit(E_ERROR);
3997c478bd9Sstevel@tonic-gate }
4007c478bd9Sstevel@tonic-gate 
4017c478bd9Sstevel@tonic-gate /*PRINTFLIKE1*/
4027c478bd9Sstevel@tonic-gate static void
dfatal(const char * fmt,...)4037c478bd9Sstevel@tonic-gate dfatal(const char *fmt, ...)
4047c478bd9Sstevel@tonic-gate {
4057c478bd9Sstevel@tonic-gate 	va_list ap;
4067c478bd9Sstevel@tonic-gate 
4077c478bd9Sstevel@tonic-gate 	va_start(ap, fmt);
4087c478bd9Sstevel@tonic-gate 
4097c478bd9Sstevel@tonic-gate 	(void) fprintf(stderr, "%s: ", g_pname);
4107c478bd9Sstevel@tonic-gate 	if (fmt != NULL)
4117c478bd9Sstevel@tonic-gate 		(void) vfprintf(stderr, fmt, ap);
4127c478bd9Sstevel@tonic-gate 
4137c478bd9Sstevel@tonic-gate 	va_end(ap);
4147c478bd9Sstevel@tonic-gate 
4157c478bd9Sstevel@tonic-gate 	if (fmt != NULL && fmt[strlen(fmt) - 1] != '\n') {
4167c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr, ": %s\n",
4177c478bd9Sstevel@tonic-gate 		    dtrace_errmsg(g_dtp, dtrace_errno(g_dtp)));
4187c478bd9Sstevel@tonic-gate 	} else if (fmt == NULL) {
4197c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr, "%s\n",
4207c478bd9Sstevel@tonic-gate 		    dtrace_errmsg(g_dtp, dtrace_errno(g_dtp)));
4217c478bd9Sstevel@tonic-gate 	}
4227c478bd9Sstevel@tonic-gate 
4237c478bd9Sstevel@tonic-gate 	if (g_pr != NULL) {
4247c478bd9Sstevel@tonic-gate 		dtrace_proc_continue(g_dtp, g_pr);
4257c478bd9Sstevel@tonic-gate 		dtrace_proc_release(g_dtp, g_pr);
4267c478bd9Sstevel@tonic-gate 	}
4277c478bd9Sstevel@tonic-gate 
4287c478bd9Sstevel@tonic-gate 	exit(E_ERROR);
4297c478bd9Sstevel@tonic-gate }
4307c478bd9Sstevel@tonic-gate 
4317c478bd9Sstevel@tonic-gate /*PRINTFLIKE1*/
4327c478bd9Sstevel@tonic-gate static void
notice(const char * fmt,...)4337c478bd9Sstevel@tonic-gate notice(const char *fmt, ...)
4347c478bd9Sstevel@tonic-gate {
4357c478bd9Sstevel@tonic-gate 	va_list ap;
4367c478bd9Sstevel@tonic-gate 
4377c478bd9Sstevel@tonic-gate 	va_start(ap, fmt);
4387c478bd9Sstevel@tonic-gate 	verror(fmt, ap);
4397c478bd9Sstevel@tonic-gate 	va_end(ap);
4407c478bd9Sstevel@tonic-gate }
4417c478bd9Sstevel@tonic-gate 
4427c478bd9Sstevel@tonic-gate static void
dprog_add(const char * prog)4437c478bd9Sstevel@tonic-gate dprog_add(const char *prog)
4447c478bd9Sstevel@tonic-gate {
4457c478bd9Sstevel@tonic-gate 	size_t len = strlen(prog);
4467c478bd9Sstevel@tonic-gate 	bcopy(prog, g_prog + g_proglen, len + 1);
4477c478bd9Sstevel@tonic-gate 	g_proglen += len;
448351346dbSahl 	assert(g_proglen < sizeof (g_prog));
4497c478bd9Sstevel@tonic-gate }
4507c478bd9Sstevel@tonic-gate 
4517c478bd9Sstevel@tonic-gate static void
dprog_compile(void)4527c478bd9Sstevel@tonic-gate dprog_compile(void)
4537c478bd9Sstevel@tonic-gate {
4547c478bd9Sstevel@tonic-gate 	dtrace_prog_t *prog;
4557c478bd9Sstevel@tonic-gate 	dtrace_proginfo_t info;
4567c478bd9Sstevel@tonic-gate 
4577c478bd9Sstevel@tonic-gate 	if (g_opt_V) {
4587c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr, "%s: vvvv D program vvvv\n", g_pname);
4597c478bd9Sstevel@tonic-gate 		(void) fputs(g_prog, stderr);
4607c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr, "%s: ^^^^ D program ^^^^\n", g_pname);
4617c478bd9Sstevel@tonic-gate 	}
4627c478bd9Sstevel@tonic-gate 
4637c478bd9Sstevel@tonic-gate 	if ((prog = dtrace_program_strcompile(g_dtp, g_prog,
4647c478bd9Sstevel@tonic-gate 	    DTRACE_PROBESPEC_NAME, 0, 0, NULL)) == NULL)
4657c478bd9Sstevel@tonic-gate 		dfatal("failed to compile program");
4667c478bd9Sstevel@tonic-gate 
4677c478bd9Sstevel@tonic-gate 	if (dtrace_program_exec(g_dtp, prog, &info) == -1)
4687c478bd9Sstevel@tonic-gate 		dfatal("failed to enable probes");
4697c478bd9Sstevel@tonic-gate }
4707c478bd9Sstevel@tonic-gate 
4717c478bd9Sstevel@tonic-gate void
print_legend(void)4727c478bd9Sstevel@tonic-gate print_legend(void)
4737c478bd9Sstevel@tonic-gate {
4747c478bd9Sstevel@tonic-gate 	(void) printf("%5s %8s %-28s %s\n", "Count", "nsec", "Lock", "Caller");
4757c478bd9Sstevel@tonic-gate }
4767c478bd9Sstevel@tonic-gate 
4777c478bd9Sstevel@tonic-gate void
print_bar(void)4787c478bd9Sstevel@tonic-gate print_bar(void)
4797c478bd9Sstevel@tonic-gate {
4807c478bd9Sstevel@tonic-gate 	(void) printf("---------------------------------------"
4817c478bd9Sstevel@tonic-gate 	    "----------------------------------------\n");
4827c478bd9Sstevel@tonic-gate }
4837c478bd9Sstevel@tonic-gate 
4847c478bd9Sstevel@tonic-gate void
print_histogram_header(void)4857c478bd9Sstevel@tonic-gate print_histogram_header(void)
4867c478bd9Sstevel@tonic-gate {
4877c478bd9Sstevel@tonic-gate 	(void) printf("\n%10s ---- Time Distribution --- %5s %s\n",
4887c478bd9Sstevel@tonic-gate 	    "nsec", "count", "Stack");
4897c478bd9Sstevel@tonic-gate }
4907c478bd9Sstevel@tonic-gate 
4917c478bd9Sstevel@tonic-gate /*
4927c478bd9Sstevel@tonic-gate  * Convert an address to a symbolic string or a numeric string. If nolocks
4937c478bd9Sstevel@tonic-gate  * is set, we return an error code if this symbol appears to be a mutex- or
4947c478bd9Sstevel@tonic-gate  * rwlock-related symbol in libc so the caller has a chance to find a more
4957c478bd9Sstevel@tonic-gate  * helpful symbol.
4967c478bd9Sstevel@tonic-gate  */
4977c478bd9Sstevel@tonic-gate static int
getsym(struct ps_prochandle * P,uintptr_t addr,char * buf,size_t size,int nolocks)4987c478bd9Sstevel@tonic-gate getsym(struct ps_prochandle *P, uintptr_t addr, char *buf, size_t size,
4997c478bd9Sstevel@tonic-gate     int nolocks)
5007c478bd9Sstevel@tonic-gate {
5017c478bd9Sstevel@tonic-gate 	char name[256];
5027c478bd9Sstevel@tonic-gate 	GElf_Sym sym;
5037c478bd9Sstevel@tonic-gate 	prsyminfo_t info;
5047c478bd9Sstevel@tonic-gate 	size_t len;
5057c478bd9Sstevel@tonic-gate 
5067c478bd9Sstevel@tonic-gate 	if (P == NULL || Pxlookup_by_addr(P, addr, name, sizeof (name),
5077c478bd9Sstevel@tonic-gate 	    &sym, &info) != 0) {
5087c478bd9Sstevel@tonic-gate 		(void) snprintf(buf, size, "%#lx", addr);
5097c478bd9Sstevel@tonic-gate 		return (0);
5107c478bd9Sstevel@tonic-gate 	}
51112e72dbbSrh 	if (info.prs_object == NULL)
51212e72dbbSrh 		info.prs_object = "<unknown>";
5137c478bd9Sstevel@tonic-gate 
5147c478bd9Sstevel@tonic-gate 	if (info.prs_lmid != LM_ID_BASE) {
5157c478bd9Sstevel@tonic-gate 		len = snprintf(buf, size, "LM%lu`", info.prs_lmid);
5167c478bd9Sstevel@tonic-gate 		buf += len;
5177c478bd9Sstevel@tonic-gate 		size -= len;
5187c478bd9Sstevel@tonic-gate 	}
5197c478bd9Sstevel@tonic-gate 
5207c478bd9Sstevel@tonic-gate 	len = snprintf(buf, size, "%s`%s", info.prs_object, info.prs_name);
5217c478bd9Sstevel@tonic-gate 	buf += len;
5227c478bd9Sstevel@tonic-gate 	size -= len;
5237c478bd9Sstevel@tonic-gate 
5247c478bd9Sstevel@tonic-gate 	if (sym.st_value != addr)
5257c478bd9Sstevel@tonic-gate 		len = snprintf(buf, size, "+%#lx", addr - sym.st_value);
5267c478bd9Sstevel@tonic-gate 
5277c478bd9Sstevel@tonic-gate 	if (nolocks && strcmp("libc.so.1", info.prs_object) == 0 &&
5287c478bd9Sstevel@tonic-gate 	    (strstr("mutex", info.prs_name) == 0 ||
5297c478bd9Sstevel@tonic-gate 	    strstr("rw", info.prs_name) == 0))
5307c478bd9Sstevel@tonic-gate 		return (-1);
5317c478bd9Sstevel@tonic-gate 
5327c478bd9Sstevel@tonic-gate 	return (0);
5337c478bd9Sstevel@tonic-gate }
5347c478bd9Sstevel@tonic-gate 
5357c478bd9Sstevel@tonic-gate /*ARGSUSED*/
5367c478bd9Sstevel@tonic-gate static int
process_aggregate(const dtrace_aggdata_t ** aggsdata,int naggvars,void * arg)537351346dbSahl process_aggregate(const dtrace_aggdata_t **aggsdata, int naggvars, void *arg)
5387c478bd9Sstevel@tonic-gate {
539a1b5e537Sbmc 	const dtrace_recdesc_t *rec;
5407c478bd9Sstevel@tonic-gate 	uintptr_t lock;
5417c478bd9Sstevel@tonic-gate 	uint64_t *stack;
542351346dbSahl 	caddr_t data;
543351346dbSahl 	pid_t pid;
5447c478bd9Sstevel@tonic-gate 	struct ps_prochandle *P;
545351346dbSahl 	char buf[256];
5467c478bd9Sstevel@tonic-gate 	int i, j;
547351346dbSahl 	uint64_t sum, count, avg;
5487c478bd9Sstevel@tonic-gate 
549351346dbSahl 	if ((*(uint_t *)arg)++ >= g_nent)
550351346dbSahl 		return (DTRACE_AGGWALK_NEXT);
5517c478bd9Sstevel@tonic-gate 
552351346dbSahl 	rec = aggsdata[0]->dtada_desc->dtagd_rec;
553351346dbSahl 	data = aggsdata[0]->dtada_data;
5547c478bd9Sstevel@tonic-gate 
5557c478bd9Sstevel@tonic-gate 	/*LINTED - alignment*/
5567c478bd9Sstevel@tonic-gate 	lock = (uintptr_t)*(uint64_t *)(data + rec[1].dtrd_offset);
5577c478bd9Sstevel@tonic-gate 	/*LINTED - alignment*/
5587c478bd9Sstevel@tonic-gate 	stack = (uint64_t *)(data + rec[2].dtrd_offset);
5597c478bd9Sstevel@tonic-gate 
5607c478bd9Sstevel@tonic-gate 	if (!g_opt_s) {
561351346dbSahl 		/*LINTED - alignment*/
562351346dbSahl 		sum = *(uint64_t *)(aggsdata[1]->dtada_data +
563351346dbSahl 		    aggsdata[1]->dtada_desc->dtagd_rec[3].dtrd_offset);
564351346dbSahl 		/*LINTED - alignment*/
565351346dbSahl 		count = *(uint64_t *)(aggsdata[2]->dtada_data +
566351346dbSahl 		    aggsdata[2]->dtada_desc->dtagd_rec[3].dtrd_offset);
5677c478bd9Sstevel@tonic-gate 	} else {
568351346dbSahl 		uint64_t *a;
569351346dbSahl 
570351346dbSahl 		/*LINTED - alignment*/
571351346dbSahl 		a = (uint64_t *)(aggsdata[1]->dtada_data +
572351346dbSahl 		    aggsdata[1]->dtada_desc->dtagd_rec[3].dtrd_offset);
573351346dbSahl 
5747c478bd9Sstevel@tonic-gate 		print_bar();
5757c478bd9Sstevel@tonic-gate 		print_legend();
5767c478bd9Sstevel@tonic-gate 
577351346dbSahl 		for (count = sum = 0, i = DTRACE_QUANTIZE_ZEROBUCKET, j = 0;
5787c478bd9Sstevel@tonic-gate 		    i < DTRACE_QUANTIZE_NBUCKETS; i++, j++) {
5797c478bd9Sstevel@tonic-gate 			count += a[i];
580351346dbSahl 			sum += a[i] << (j - 64);
5817c478bd9Sstevel@tonic-gate 		}
5827c478bd9Sstevel@tonic-gate 	}
5837c478bd9Sstevel@tonic-gate 
584351346dbSahl 	avg = sum / count;
5857c478bd9Sstevel@tonic-gate 	(void) printf("%5llu %8llu ", (u_longlong_t)count, (u_longlong_t)avg);
5867c478bd9Sstevel@tonic-gate 
5877c478bd9Sstevel@tonic-gate 	pid = stack[0];
5887c478bd9Sstevel@tonic-gate 	P = dtrace_proc_grab(g_dtp, pid, PGRAB_RDONLY);
5897c478bd9Sstevel@tonic-gate 
5907c478bd9Sstevel@tonic-gate 	(void) getsym(P, lock, buf, sizeof (buf), 0);
5917c478bd9Sstevel@tonic-gate 	(void) printf("%-28s ", buf);
5927c478bd9Sstevel@tonic-gate 
5937c478bd9Sstevel@tonic-gate 	for (i = 2; i <= 5; i++) {
5947c478bd9Sstevel@tonic-gate 		if (getsym(P, stack[i], buf, sizeof (buf), 1) == 0)
5957c478bd9Sstevel@tonic-gate 			break;
5967c478bd9Sstevel@tonic-gate 	}
5977c478bd9Sstevel@tonic-gate 	(void) printf("%s\n", buf);
5987c478bd9Sstevel@tonic-gate 
5997c478bd9Sstevel@tonic-gate 	if (g_opt_s) {
6007c478bd9Sstevel@tonic-gate 		int stack_done = 0;
6017c478bd9Sstevel@tonic-gate 		int quant_done = 0;
6027c478bd9Sstevel@tonic-gate 		int first_bin, last_bin;
603351346dbSahl 		uint64_t bin_size, *a;
604351346dbSahl 
605351346dbSahl 		/*LINTED - alignment*/
606351346dbSahl 		a = (uint64_t *)(aggsdata[1]->dtada_data +
607351346dbSahl 		    aggsdata[1]->dtada_desc->dtagd_rec[3].dtrd_offset);
6087c478bd9Sstevel@tonic-gate 
6097c478bd9Sstevel@tonic-gate 		print_histogram_header();
6107c478bd9Sstevel@tonic-gate 
6117c478bd9Sstevel@tonic-gate 		for (first_bin = DTRACE_QUANTIZE_ZEROBUCKET;
6127c478bd9Sstevel@tonic-gate 		    a[first_bin] == 0; first_bin++)
6137c478bd9Sstevel@tonic-gate 			continue;
6147c478bd9Sstevel@tonic-gate 		for (last_bin = DTRACE_QUANTIZE_ZEROBUCKET + 63;
6157c478bd9Sstevel@tonic-gate 		    a[last_bin] == 0; last_bin--)
6167c478bd9Sstevel@tonic-gate 			continue;
6177c478bd9Sstevel@tonic-gate 
6187c478bd9Sstevel@tonic-gate 		for (i = 0; !stack_done || !quant_done; i++) {
6197c478bd9Sstevel@tonic-gate 			if (!stack_done) {
6207c478bd9Sstevel@tonic-gate 				(void) getsym(P, stack[i + 2], buf,
6217c478bd9Sstevel@tonic-gate 				    sizeof (buf), 0);
6227c478bd9Sstevel@tonic-gate 			} else {
6237c478bd9Sstevel@tonic-gate 				buf[0] = '\0';
6247c478bd9Sstevel@tonic-gate 			}
6257c478bd9Sstevel@tonic-gate 
6267c478bd9Sstevel@tonic-gate 			if (!quant_done) {
6277c478bd9Sstevel@tonic-gate 				bin_size = a[first_bin];
6287c478bd9Sstevel@tonic-gate 
6297c478bd9Sstevel@tonic-gate 				(void) printf("%10llu |%-24.*s| %5llu %s\n",
6307c478bd9Sstevel@tonic-gate 				    1ULL <<
6317c478bd9Sstevel@tonic-gate 				    (first_bin - DTRACE_QUANTIZE_ZEROBUCKET),
6327c478bd9Sstevel@tonic-gate 				    (int)(24.0 * bin_size / count),
6337c478bd9Sstevel@tonic-gate 				    "@@@@@@@@@@@@@@@@@@@@@@@@@@",
6347c478bd9Sstevel@tonic-gate 				    (u_longlong_t)bin_size, buf);
6357c478bd9Sstevel@tonic-gate 			} else {
6367c478bd9Sstevel@tonic-gate 				(void) printf("%43s %s\n", "", buf);
6377c478bd9Sstevel@tonic-gate 			}
6387c478bd9Sstevel@tonic-gate 
6397c478bd9Sstevel@tonic-gate 			if (i + 1 >= g_nframes || stack[i + 3] == 0)
6407c478bd9Sstevel@tonic-gate 				stack_done = 1;
6417c478bd9Sstevel@tonic-gate 
6427c478bd9Sstevel@tonic-gate 			if (first_bin++ == last_bin)
6437c478bd9Sstevel@tonic-gate 				quant_done = 1;
6447c478bd9Sstevel@tonic-gate 		}
6457c478bd9Sstevel@tonic-gate 	}
6467c478bd9Sstevel@tonic-gate 
6477c478bd9Sstevel@tonic-gate 	dtrace_proc_release(g_dtp, P);
6487c478bd9Sstevel@tonic-gate 
6497c478bd9Sstevel@tonic-gate 	return (DTRACE_AGGWALK_NEXT);
6507c478bd9Sstevel@tonic-gate }
6517c478bd9Sstevel@tonic-gate 
652351346dbSahl /*ARGSUSED*/
653351346dbSahl static void
prochandler(struct ps_prochandle * P,const char * msg,void * arg)654351346dbSahl prochandler(struct ps_prochandle *P, const char *msg, void *arg)
6557c478bd9Sstevel@tonic-gate {
6567c478bd9Sstevel@tonic-gate 	const psinfo_t *prp = Ppsinfo(P);
6577c478bd9Sstevel@tonic-gate 	int pid = Pstatus(P)->pr_pid;
6587c478bd9Sstevel@tonic-gate 	char name[SIG2STR_MAX];
6597c478bd9Sstevel@tonic-gate 
660351346dbSahl 	if (msg != NULL) {
661351346dbSahl 		notice("pid %d: %s\n", pid, msg);
662351346dbSahl 		return;
663351346dbSahl 	}
664351346dbSahl 
6657c478bd9Sstevel@tonic-gate 	switch (Pstate(P)) {
6667c478bd9Sstevel@tonic-gate 	case PS_UNDEAD:
6677c478bd9Sstevel@tonic-gate 		/*
6687c478bd9Sstevel@tonic-gate 		 * Ideally we would like to always report pr_wstat here, but it
6697c478bd9Sstevel@tonic-gate 		 * isn't possible given current /proc semantics.  If we grabbed
6707c478bd9Sstevel@tonic-gate 		 * the process, Ppsinfo() will either fail or return a zeroed
6717c478bd9Sstevel@tonic-gate 		 * psinfo_t depending on how far the parent is in reaping it.
6727c478bd9Sstevel@tonic-gate 		 * When /proc provides a stable pr_wstat in the status file,
6737c478bd9Sstevel@tonic-gate 		 * this code can be improved by examining this new pr_wstat.
6747c478bd9Sstevel@tonic-gate 		 */
6757c478bd9Sstevel@tonic-gate 		if (prp != NULL && WIFSIGNALED(prp->pr_wstat)) {
6767c478bd9Sstevel@tonic-gate 			notice("pid %d terminated by %s\n", pid,
6777c478bd9Sstevel@tonic-gate 			    proc_signame(WTERMSIG(prp->pr_wstat),
6787c478bd9Sstevel@tonic-gate 			    name, sizeof (name)));
6797c478bd9Sstevel@tonic-gate 		} else if (prp != NULL && WEXITSTATUS(prp->pr_wstat) != 0) {
6807c478bd9Sstevel@tonic-gate 			notice("pid %d exited with status %d\n",
6817c478bd9Sstevel@tonic-gate 			    pid, WEXITSTATUS(prp->pr_wstat));
6827c478bd9Sstevel@tonic-gate 		} else {
6837c478bd9Sstevel@tonic-gate 			notice("pid %d has exited\n", pid);
6847c478bd9Sstevel@tonic-gate 		}
685351346dbSahl 		g_exited = 1;
686351346dbSahl 		break;
6877c478bd9Sstevel@tonic-gate 
6887c478bd9Sstevel@tonic-gate 	case PS_LOST:
6897c478bd9Sstevel@tonic-gate 		notice("pid %d exec'd a set-id or unobservable program\n", pid);
690351346dbSahl 		g_exited = 1;
691351346dbSahl 		break;
6927c478bd9Sstevel@tonic-gate 	}
693351346dbSahl }
6947c478bd9Sstevel@tonic-gate 
695351346dbSahl /*ARGSUSED*/
696351346dbSahl static int
chewrec(const dtrace_probedata_t * data,const dtrace_recdesc_t * rec,void * arg)697351346dbSahl chewrec(const dtrace_probedata_t *data, const dtrace_recdesc_t *rec, void *arg)
698351346dbSahl {
699351346dbSahl 	dtrace_eprobedesc_t *epd = data->dtpda_edesc;
700351346dbSahl 	dtrace_aggvarid_t aggvars[2];
701351346dbSahl 	const void *buf;
702351346dbSahl 	int i, nagv;
703351346dbSahl 
704351346dbSahl 	/*
705351346dbSahl 	 * A NULL rec indicates that we've processed the last record.
706351346dbSahl 	 */
707351346dbSahl 	if (rec == NULL)
708351346dbSahl 		return (DTRACE_CONSUME_NEXT);
709351346dbSahl 
710351346dbSahl 	buf = data->dtpda_data - rec->dtrd_offset;
711351346dbSahl 
712351346dbSahl 	switch (rec->dtrd_action) {
713351346dbSahl 	case DTRACEACT_DIFEXPR:
714351346dbSahl 		(void) printf("\n%s\n\n", (char *)buf + rec->dtrd_offset);
715351346dbSahl 		if (!g_opt_s) {
716351346dbSahl 			print_legend();
717351346dbSahl 			print_bar();
718351346dbSahl 		}
719351346dbSahl 		return (DTRACE_CONSUME_NEXT);
720351346dbSahl 
721351346dbSahl 	case DTRACEACT_PRINTA:
722351346dbSahl 		for (nagv = 0, i = 0; i < epd->dtepd_nrecs - 1; i++) {
723351346dbSahl 			const dtrace_recdesc_t *nrec = &rec[i];
724351346dbSahl 
725351346dbSahl 			if (nrec->dtrd_uarg != rec->dtrd_uarg)
726351346dbSahl 				break;
727351346dbSahl 
728351346dbSahl 			/*LINTED - alignment*/
729351346dbSahl 			aggvars[nagv++] = *(dtrace_aggvarid_t *)((caddr_t)buf +
730351346dbSahl 			    nrec->dtrd_offset);
731351346dbSahl 		}
732351346dbSahl 
733351346dbSahl 		if (nagv == (g_opt_s ? 1 : 2)) {
734351346dbSahl 			uint_t nent = 0;
735351346dbSahl 			if (dtrace_aggregate_walk_joined(g_dtp, aggvars, nagv,
736351346dbSahl 			    process_aggregate, &nent) != 0)
737351346dbSahl 				dfatal("failed to walk aggregate");
738351346dbSahl 		}
739351346dbSahl 
740351346dbSahl 		return (DTRACE_CONSUME_NEXT);
741351346dbSahl 	}
742351346dbSahl 
743351346dbSahl 	return (DTRACE_CONSUME_THIS);
7447c478bd9Sstevel@tonic-gate }
7457c478bd9Sstevel@tonic-gate 
7467c478bd9Sstevel@tonic-gate /*ARGSUSED*/
7477c478bd9Sstevel@tonic-gate static void
intr(int signo)7487c478bd9Sstevel@tonic-gate intr(int signo)
7497c478bd9Sstevel@tonic-gate {
7507c478bd9Sstevel@tonic-gate 	g_intr = 1;
7517c478bd9Sstevel@tonic-gate }
7527c478bd9Sstevel@tonic-gate 
7537c478bd9Sstevel@tonic-gate int
main(int argc,char ** argv)7547c478bd9Sstevel@tonic-gate main(int argc, char **argv)
7557c478bd9Sstevel@tonic-gate {
7567c478bd9Sstevel@tonic-gate 	ucred_t *ucp;
7577c478bd9Sstevel@tonic-gate 	int err;
7587c478bd9Sstevel@tonic-gate 	int opt_C = 0, opt_H = 0, opt_p = 0, opt_v = 0;
7597c478bd9Sstevel@tonic-gate 	char c, *p, *end;
7607c478bd9Sstevel@tonic-gate 	struct sigaction act;
7617c478bd9Sstevel@tonic-gate 	int done = 0;
7627c478bd9Sstevel@tonic-gate 
763351346dbSahl 	g_pname = basename(argv[0]);
764351346dbSahl 	argv[0] = g_pname; /* rewrite argv[0] for getopt errors */
7657c478bd9Sstevel@tonic-gate 
7667c478bd9Sstevel@tonic-gate 	/*
7677c478bd9Sstevel@tonic-gate 	 * Make sure we have the required dtrace_proc privilege.
7687c478bd9Sstevel@tonic-gate 	 */
7697c478bd9Sstevel@tonic-gate 	if ((ucp = ucred_get(getpid())) != NULL) {
7707c478bd9Sstevel@tonic-gate 		const priv_set_t *psp;
7717c478bd9Sstevel@tonic-gate 		if ((psp = ucred_getprivset(ucp, PRIV_EFFECTIVE)) != NULL &&
7727c478bd9Sstevel@tonic-gate 		    !priv_ismember(psp, PRIV_DTRACE_PROC)) {
7737c478bd9Sstevel@tonic-gate 			fatal("dtrace_proc privilege required\n");
7747c478bd9Sstevel@tonic-gate 		}
7757c478bd9Sstevel@tonic-gate 
7767c478bd9Sstevel@tonic-gate 		ucred_free(ucp);
7777c478bd9Sstevel@tonic-gate 	}
7787c478bd9Sstevel@tonic-gate 
7797c478bd9Sstevel@tonic-gate 	while ((c = getopt(argc, argv, PLOCKSTAT_OPTSTR)) != EOF) {
7807c478bd9Sstevel@tonic-gate 		switch (c) {
7817c478bd9Sstevel@tonic-gate 		case 'n':
7827c478bd9Sstevel@tonic-gate 			errno = 0;
7837c478bd9Sstevel@tonic-gate 			g_nent = strtoul(optarg, &end, 10);
7847c478bd9Sstevel@tonic-gate 			if (*end != '\0' || errno != 0) {
7857c478bd9Sstevel@tonic-gate 				(void) fprintf(stderr, "%s: invalid count "
7867c478bd9Sstevel@tonic-gate 				    "'%s'\n", g_pname, optarg);
7877c478bd9Sstevel@tonic-gate 				usage();
7887c478bd9Sstevel@tonic-gate 			}
7897c478bd9Sstevel@tonic-gate 			break;
7907c478bd9Sstevel@tonic-gate 
7917c478bd9Sstevel@tonic-gate 		case 'p':
7927c478bd9Sstevel@tonic-gate 			opt_p = 1;
7937c478bd9Sstevel@tonic-gate 			break;
7947c478bd9Sstevel@tonic-gate 
7957c478bd9Sstevel@tonic-gate 		case 'v':
7967c478bd9Sstevel@tonic-gate 			opt_v = 1;
7977c478bd9Sstevel@tonic-gate 			break;
7987c478bd9Sstevel@tonic-gate 
7997c478bd9Sstevel@tonic-gate 		case 'A':
8007c478bd9Sstevel@tonic-gate 			opt_C = opt_H = 1;
8017c478bd9Sstevel@tonic-gate 			break;
8027c478bd9Sstevel@tonic-gate 
8037c478bd9Sstevel@tonic-gate 		case 'C':
8047c478bd9Sstevel@tonic-gate 			opt_C = 1;
8057c478bd9Sstevel@tonic-gate 			break;
8067c478bd9Sstevel@tonic-gate 
8077c478bd9Sstevel@tonic-gate 		case 'H':
8087c478bd9Sstevel@tonic-gate 			opt_H = 1;
8097c478bd9Sstevel@tonic-gate 			break;
8107c478bd9Sstevel@tonic-gate 
8117c478bd9Sstevel@tonic-gate 		case 'V':
8127c478bd9Sstevel@tonic-gate 			g_opt_V = 1;
8137c478bd9Sstevel@tonic-gate 			break;
8147c478bd9Sstevel@tonic-gate 
8157c478bd9Sstevel@tonic-gate 		default:
8167c478bd9Sstevel@tonic-gate 			if (strchr(PLOCKSTAT_OPTSTR, c) == NULL)
8177c478bd9Sstevel@tonic-gate 				usage();
8187c478bd9Sstevel@tonic-gate 		}
8197c478bd9Sstevel@tonic-gate 	}
8207c478bd9Sstevel@tonic-gate 
8217c478bd9Sstevel@tonic-gate 	/*
8227c478bd9Sstevel@tonic-gate 	 * We need a command or at least one pid.
8237c478bd9Sstevel@tonic-gate 	 */
8247c478bd9Sstevel@tonic-gate 	if (argc == optind)
8257c478bd9Sstevel@tonic-gate 		usage();
8267c478bd9Sstevel@tonic-gate 
8277c478bd9Sstevel@tonic-gate 	if (opt_C == 0 && opt_H == 0)
8287c478bd9Sstevel@tonic-gate 		opt_C = 1;
8297c478bd9Sstevel@tonic-gate 
8307c478bd9Sstevel@tonic-gate 	if ((g_dtp = dtrace_open(DTRACE_VERSION, 0, &err)) == NULL)
8317c478bd9Sstevel@tonic-gate 		fatal("failed to initialize dtrace: %s\n",
8327c478bd9Sstevel@tonic-gate 		    dtrace_errmsg(NULL, err));
8337c478bd9Sstevel@tonic-gate 
834351346dbSahl 	/*
835351346dbSahl 	 * The longest string we trace is 23 bytes long -- so 32 is plenty.
836351346dbSahl 	 */
837351346dbSahl 	if (dtrace_setopt(g_dtp, "strsize", "32") == -1)
838351346dbSahl 		dfatal("failed to set 'strsize'");
839351346dbSahl 
840351346dbSahl 	/*
841351346dbSahl 	 * 1k should be more than enough for all trace() and printa() actions.
842351346dbSahl 	 */
843351346dbSahl 	if (dtrace_setopt(g_dtp, "bufsize", "1k") == -1)
844351346dbSahl 		dfatal("failed to set 'bufsize'");
845351346dbSahl 
846351346dbSahl 	/*
847351346dbSahl 	 * The table we produce has the hottest locks at the top.
848351346dbSahl 	 */
849351346dbSahl 	if (dtrace_setopt(g_dtp, "aggsortrev", NULL) == -1)
850351346dbSahl 		dfatal("failed to set 'aggsortrev'");
851351346dbSahl 
852351346dbSahl 	/*
853351346dbSahl 	 * These are two reasonable defaults which should suffice.
854351346dbSahl 	 */
8557c478bd9Sstevel@tonic-gate 	if (dtrace_setopt(g_dtp, "aggsize", "256k") == -1)
8567c478bd9Sstevel@tonic-gate 		dfatal("failed to set 'aggsize'");
8577c478bd9Sstevel@tonic-gate 	if (dtrace_setopt(g_dtp, "aggrate", "1sec") == -1)
8587c478bd9Sstevel@tonic-gate 		dfatal("failed to set 'aggrate'");
8597c478bd9Sstevel@tonic-gate 
8607c478bd9Sstevel@tonic-gate 	/*
8617c478bd9Sstevel@tonic-gate 	 * Take a second pass through to look for options that set options now
8627c478bd9Sstevel@tonic-gate 	 * that we have an open dtrace handle.
8637c478bd9Sstevel@tonic-gate 	 */
8647c478bd9Sstevel@tonic-gate 	optind = 1;
8657c478bd9Sstevel@tonic-gate 	while ((c = getopt(argc, argv, PLOCKSTAT_OPTSTR)) != EOF) {
8667c478bd9Sstevel@tonic-gate 		switch (c) {
8677c478bd9Sstevel@tonic-gate 		case 's':
8687c478bd9Sstevel@tonic-gate 			g_opt_s = 1;
8697c478bd9Sstevel@tonic-gate 			if (dtrace_setopt(g_dtp, "ustackframes", optarg) == -1)
8707c478bd9Sstevel@tonic-gate 				dfatal("failed to set 'ustackframes'");
8717c478bd9Sstevel@tonic-gate 			break;
8727c478bd9Sstevel@tonic-gate 
8737c478bd9Sstevel@tonic-gate 		case 'x':
8747c478bd9Sstevel@tonic-gate 			if ((p = strchr(optarg, '=')) != NULL)
8757c478bd9Sstevel@tonic-gate 				*p++ = '\0';
8767c478bd9Sstevel@tonic-gate 
8777c478bd9Sstevel@tonic-gate 			if (dtrace_setopt(g_dtp, optarg, p) != 0)
8787c478bd9Sstevel@tonic-gate 				dfatal("failed to set -x %s", optarg);
8797c478bd9Sstevel@tonic-gate 			break;
8807c478bd9Sstevel@tonic-gate 
8817c478bd9Sstevel@tonic-gate 		case 'e':
8827c478bd9Sstevel@tonic-gate 			errno = 0;
8837c478bd9Sstevel@tonic-gate 			(void) strtoul(optarg, &end, 10);
8847c478bd9Sstevel@tonic-gate 			if (*optarg == '-' || *end != '\0' || errno != 0) {
8857c478bd9Sstevel@tonic-gate 				(void) fprintf(stderr, "%s: invalid timeout "
8867c478bd9Sstevel@tonic-gate 				    "'%s'\n", g_pname, optarg);
8877c478bd9Sstevel@tonic-gate 				usage();
8887c478bd9Sstevel@tonic-gate 			}
8897c478bd9Sstevel@tonic-gate 
8907c478bd9Sstevel@tonic-gate 			/*
8917c478bd9Sstevel@tonic-gate 			 * Construct a DTrace enabling that will exit after
8927c478bd9Sstevel@tonic-gate 			 * the specified number of seconds.
8937c478bd9Sstevel@tonic-gate 			 */
8947c478bd9Sstevel@tonic-gate 			dprog_add("BEGIN\n{\n\tend = timestamp + ");
8957c478bd9Sstevel@tonic-gate 			dprog_add(optarg);
8967c478bd9Sstevel@tonic-gate 			dprog_add(" * 1000000000;\n}\n");
8977c478bd9Sstevel@tonic-gate 			dprog_add("tick-10hz\n/timestamp >= end/\n");
8987c478bd9Sstevel@tonic-gate 			dprog_add("{\n\texit(0);\n}\n");
8997c478bd9Sstevel@tonic-gate 			break;
9007c478bd9Sstevel@tonic-gate 		}
9017c478bd9Sstevel@tonic-gate 	}
9027c478bd9Sstevel@tonic-gate 
9037c478bd9Sstevel@tonic-gate 	argc -= optind;
9047c478bd9Sstevel@tonic-gate 	argv += optind;
9057c478bd9Sstevel@tonic-gate 
9067c478bd9Sstevel@tonic-gate 	if (opt_H) {
9077c478bd9Sstevel@tonic-gate 		dprog_add(g_hold_init);
908*6bc6f338SToomas Soome 		if (g_opt_s == 0)
9097c478bd9Sstevel@tonic-gate 			dprog_add(g_hold_times);
9107c478bd9Sstevel@tonic-gate 		else
9117c478bd9Sstevel@tonic-gate 			dprog_add(g_hold_histogram);
9127c478bd9Sstevel@tonic-gate 	}
9137c478bd9Sstevel@tonic-gate 
9147c478bd9Sstevel@tonic-gate 	if (opt_C) {
9157c478bd9Sstevel@tonic-gate 		dprog_add(g_ctnd_init);
916*6bc6f338SToomas Soome 		if (g_opt_s == 0)
9177c478bd9Sstevel@tonic-gate 			dprog_add(g_ctnd_times);
9187c478bd9Sstevel@tonic-gate 		else
9197c478bd9Sstevel@tonic-gate 			dprog_add(g_ctnd_histogram);
9207c478bd9Sstevel@tonic-gate 	}
9217c478bd9Sstevel@tonic-gate 
9227c478bd9Sstevel@tonic-gate 	if (opt_p) {
9237c478bd9Sstevel@tonic-gate 		ulong_t pid;
9247c478bd9Sstevel@tonic-gate 
9257c478bd9Sstevel@tonic-gate 		if (argc > 1) {
9267c478bd9Sstevel@tonic-gate 			(void) fprintf(stderr, "%s: only one pid is allowed\n",
9277c478bd9Sstevel@tonic-gate 			    g_pname);
9287c478bd9Sstevel@tonic-gate 			usage();
9297c478bd9Sstevel@tonic-gate 		}
9307c478bd9Sstevel@tonic-gate 
9317c478bd9Sstevel@tonic-gate 		errno = 0;
9327c478bd9Sstevel@tonic-gate 		pid = strtoul(argv[0], &end, 10);
9337c478bd9Sstevel@tonic-gate 		if (*end != '\0' || errno != 0 || (pid_t)pid != pid) {
9347c478bd9Sstevel@tonic-gate 			(void) fprintf(stderr, "%s: invalid pid '%s'\n",
9357c478bd9Sstevel@tonic-gate 			    g_pname, argv[0]);
9367c478bd9Sstevel@tonic-gate 			usage();
9377c478bd9Sstevel@tonic-gate 		}
9387c478bd9Sstevel@tonic-gate 
9397c478bd9Sstevel@tonic-gate 		if ((g_pr = dtrace_proc_grab(g_dtp, (pid_t)pid, 0)) == NULL)
9407c478bd9Sstevel@tonic-gate 			dfatal(NULL);
9417c478bd9Sstevel@tonic-gate 	} else {
9427c478bd9Sstevel@tonic-gate 		if ((g_pr = dtrace_proc_create(g_dtp, argv[0], argv)) == NULL)
9437c478bd9Sstevel@tonic-gate 			dfatal(NULL);
9447c478bd9Sstevel@tonic-gate 	}
9457c478bd9Sstevel@tonic-gate 
9467c478bd9Sstevel@tonic-gate 	dprog_compile();
9477c478bd9Sstevel@tonic-gate 
948351346dbSahl 	if (dtrace_handle_proc(g_dtp, &prochandler, NULL) == -1)
949351346dbSahl 		dfatal("failed to establish proc handler");
950351346dbSahl 
9517c478bd9Sstevel@tonic-gate 	(void) sigemptyset(&act.sa_mask);
9527c478bd9Sstevel@tonic-gate 	act.sa_flags = 0;
9537c478bd9Sstevel@tonic-gate 	act.sa_handler = intr;
9547c478bd9Sstevel@tonic-gate 	(void) sigaction(SIGINT, &act, NULL);
9557c478bd9Sstevel@tonic-gate 	(void) sigaction(SIGTERM, &act, NULL);
9567c478bd9Sstevel@tonic-gate 
9577c478bd9Sstevel@tonic-gate 	if (dtrace_go(g_dtp) != 0)
9587c478bd9Sstevel@tonic-gate 		dfatal("dtrace_go()");
9597c478bd9Sstevel@tonic-gate 
9607c478bd9Sstevel@tonic-gate 	if (dtrace_getopt(g_dtp, "ustackframes", &g_nframes) != 0)
9617c478bd9Sstevel@tonic-gate 		dfatal("failed to get 'ustackframes'");
9627c478bd9Sstevel@tonic-gate 
9637c478bd9Sstevel@tonic-gate 	dtrace_proc_continue(g_dtp, g_pr);
9647c478bd9Sstevel@tonic-gate 
9657c478bd9Sstevel@tonic-gate 	if (opt_v)
9667c478bd9Sstevel@tonic-gate 		(void) printf("%s: tracing enabled for pid %d\n", g_pname,
9677c478bd9Sstevel@tonic-gate 		    (int)Pstatus(g_pr)->pr_pid);
9687c478bd9Sstevel@tonic-gate 
969351346dbSahl 	do {
970351346dbSahl 		if (!g_intr && !done)
971351346dbSahl 			dtrace_sleep(g_dtp);
9727c478bd9Sstevel@tonic-gate 
973351346dbSahl 		if (done || g_intr || g_exited) {
9747c478bd9Sstevel@tonic-gate 			done = 1;
975351346dbSahl 			if (dtrace_stop(g_dtp) == -1)
976351346dbSahl 				dfatal("couldn't stop tracing");
977351346dbSahl 		}
9787c478bd9Sstevel@tonic-gate 
979351346dbSahl 		switch (dtrace_work(g_dtp, stdout, NULL, chewrec, NULL)) {
980351346dbSahl 		case DTRACE_WORKSTATUS_DONE:
9817c478bd9Sstevel@tonic-gate 			done = 1;
982351346dbSahl 			break;
983351346dbSahl 		case DTRACE_WORKSTATUS_OKAY:
984351346dbSahl 			break;
985351346dbSahl 		default:
986351346dbSahl 			dfatal("processing aborted");
987351346dbSahl 		}
9887c478bd9Sstevel@tonic-gate 
989351346dbSahl 	} while (!done);
9907c478bd9Sstevel@tonic-gate 
9917c478bd9Sstevel@tonic-gate 	dtrace_close(g_dtp);
9927c478bd9Sstevel@tonic-gate 
9937c478bd9Sstevel@tonic-gate 	return (0);
9947c478bd9Sstevel@tonic-gate }
995