Home | History | Annotate | Line # | Download | only in Python 
      1 #!/usr/sbin/dtrace -Zs
      2 /*
      3  * py_syscolors.d - trace Python function flow plus syscalls, in color.
      4  *                  Written for the Python DTrace provider.
      5  *
      6  * $Id: py_syscolors.d,v 1.1.1.1 2015/09/30 22:01:09 christos Exp $
      7  *
      8  * USAGE: py_syscolors.d { -p PID | -c cmd }	# hit Ctrl-C to end
      9  *
     10  * This watches Python function entries and returns, and indents child
     11  * function calls.
     12  *
     13  * FIELDS:
     14  *		C		CPU-id
     15  *		PID		Process ID
     16  *		DELTA(us)	Elapsed time from previous line to this line
     17  *		FILE		Filename of the Python program
     18  *		LINE		Line number of filename
     19  *		TYPE		Type of call (func/syscall)
     20  *		NAME		Python function or syscall name
     21  *
     22  * Filename and function names are printed if available.
     23  *
     24  * WARNING: Watch the first column carefully, it prints the CPU-id. If it
     25  * changes, then it is very likely that the output has been shuffled.
     26  *
     27  * COPYRIGHT: Copyright (c) 2007 Brendan Gregg.
     28  *
     29  * CDDL HEADER START
     30  *
     31  *  The contents of this file are subject to the terms of the
     32  *  Common Development and Distribution License, Version 1.0 only
     33  *  (the "License").  You may not use this file except in compliance
     34  *  with the License.
     35  *
     36  *  You can obtain a copy of the license at Docs/cddl1.txt
     37  *  or http://www.opensolaris.org/os/licensing.
     38  *  See the License for the specific language governing permissions
     39  *  and limitations under the License.
     40  *
     41  * CDDL HEADER END
     42  *
     43  * 09-Sep-2007	Brendan Gregg	Created this.
     44  */
     45 
     46 #pragma D option quiet
     47 #pragma D option switchrate=10
     48 
     49 self int depth;
     50 
     51 dtrace:::BEGIN
     52 {
     53         color_python = "\033[2;35m";		/* violet, faint */
     54         color_syscall = "\033[2;32m";		/* green, faint */
     55         color_off = "\033[0m";			/* default */
     56 
     57 	self->depth = 0;
     58 	printf("%s %6s %10s  %16s:%-4s %-8s -- %s\n", "C", "PID", "DELTA(us)",
     59 	    "FILE", "LINE", "TYPE", "NAME");
     60 }
     61 
     62 python$target:::function-entry,
     63 python$target:::function-return,
     64 syscall:::entry,
     65 syscall:::return
     66 /self->last == 0 && pid == $target/
     67 {
     68 	self->last = timestamp;
     69 }
     70 
     71 python$target:::function-entry
     72 {
     73 	this->delta = (timestamp - self->last) / 1000;
     74 	printf("%s%d %6d %10d  %16s:%-4d %-8s %*s-> %s%s\n", color_python,
     75 	    cpu, pid, this->delta, basename(copyinstr(arg0)), arg2, "func",
     76 	    self->depth * 2, "", copyinstr(arg1), color_off);
     77 	self->depth++;
     78 	self->last = timestamp;
     79 }
     80 
     81 python$target:::function-return
     82 {
     83 	this->delta = (timestamp - self->last) / 1000;
     84 	this->name = strjoin(strjoin(copyinstr(arg0), "::"), copyinstr(arg1));
     85 	self->depth -= self->depth > 0 ? 1 : 0;
     86 	printf("%s%d %6d %10d  %16s:%-4d %-8s %*s<- %s%s\n", color_python,
     87 	    cpu, pid, this->delta, basename(copyinstr(arg0)), arg2, "func",
     88 	    self->depth * 2, "", copyinstr(arg1), color_off);
     89 	self->last = timestamp;
     90 }
     91 
     92 syscall:::entry
     93 /pid == $target/
     94 {
     95 	this->delta = (timestamp - self->last) / 1000;
     96 	printf("%s%d %6d %10d  %16s:-    %-8s %*s-> %s%s\n", color_syscall,
     97 	    cpu, pid, this->delta, "\"", "syscall", self->depth * 2, "",
     98 	    probefunc, color_off);
     99 	self->last = timestamp;
    100 }
    101 
    102 syscall:::return
    103 /pid == $target/
    104 {
    105 	this->delta = (timestamp - self->last) / 1000;
    106 	printf("%s%d %6d %10d  %16s:-    %-8s %*s<- %s%s\n", color_syscall,
    107 	    cpu, pid, this->delta, "\"", "syscall", self->depth * 2, "",
    108 	    probefunc, color_off);
    109 	self->last = timestamp;
    110 }
    111 
    112 proc:::exit
    113 /pid == $target/
    114 {
    115 	exit(0);
    116 }
    117