1235368Sgnn#!/usr/sbin/dtrace -Zs
2/*
3 * py_flowtime.d - snoop Python functions with flow and delta times.
4 *                 Written for the Python DTrace provider.
5 *
6 * $Id: py_flowtime.d 41 2007-09-17 02:20:10Z brendan $
7 *
8 * This traces shell activity from Python programs on the system that are
9 * running with Python provider support.
10 *
11 * USAGE: py_flowtime.d			# hit Ctrl-C to end
12 *
13 * This watches Python function entries and returns, and indents child
14 * function calls.
15 *
16 * FIELDS:
17 *		C		CPU-id
18 *		TIME(us)	Time since boot, us
19 *		FILE		Filename that this function belongs to
20 *		DELTA(us)	Elapsed time from previous line to this line
21 *		FUNC		Python function name
22 *
23 * LEGEND:
24 *		->		function entry
25 *		<-		function return
26 *
27 * Filename and function names are printed if available.
28 *
29 * WARNING: Watch the first column carefully, it prints the CPU-id. If it
30 * changes, then it is very likely that the output has been shuffled.
31 *
32 * COPYRIGHT: Copyright (c) 2007 Brendan Gregg.
33 *
34 * CDDL HEADER START
35 *
36 *  The contents of this file are subject to the terms of the
37 *  Common Development and Distribution License, Version 1.0 only
38 *  (the "License").  You may not use this file except in compliance
39 *  with the License.
40 *
41 *  You can obtain a copy of the license at Docs/cddl1.txt
42 *  or http://www.opensolaris.org/os/licensing.
43 *  See the License for the specific language governing permissions
44 *  and limitations under the License.
45 *
46 * CDDL HEADER END
47 *
48 * 09-Sep-2007	Brendan Gregg	Created this.
49 */
50
51#pragma D option quiet
52#pragma D option switchrate=10
53
54self int depth;
55
56self int last;
57
58dtrace:::BEGIN
59{
60	printf("%3s %-16s %-16s %9s  -- %s\n", "C", "TIME(us)", "FILE",
61	    "DELTA(us)", "FUNC");
62}
63
64python*:::function-entry,
65python*:::function-return
66/self->last == 0/
67{
68	self->last = timestamp;
69}
70
71python*:::function-entry
72{
73	this->delta = (timestamp - self->last) / 1000;
74	printf("%3d %-16d %-16s %9d %*s-> %s\n", cpu, timestamp / 1000,
75	    basename(copyinstr(arg0)), this->delta, self->depth * 2, "",
76	    copyinstr(arg1));
77	self->depth++;
78	self->last = timestamp;
79}
80
81python*:::function-return
82{
83	this->delta = (timestamp - self->last) / 1000;
84	self->depth -= self->depth > 0 ? 1 : 0;
85	printf("%3d %-16d %-16s %9d %*s<- %s\n", cpu, timestamp / 1000,
86	    basename(copyinstr(arg0)), this->delta, self->depth * 2, "",
87	    copyinstr(arg1));
88	self->last = timestamp;
89}
90