Profiling Python imports using DTrace probes

module_import_trace.d

/* run using:

 $ sudo dtrace -q -s module_import_trace.d -c './python.exe -c True'
 
*/
self int indent;
self int traceit;

python$target:::module-exec-start
{
        printf("%d\t%*s:", timestamp, 15, probename);
        printf("%*s", self->indent, "");
        printf("start %s\n", copyinstr(arg0));
        self->indent++;
}

python$target:::module-exec-done
{
        printf("%d\t%*s:", timestamp, 15, probename);
        printf("%*s", self->indent, "");
        printf("done %s\n", copyinstr(arg0));
        self->indent--;
}

python$target:::module-exec-start
/copyinstr(arg0) == "mymodule"/
{
	self->traceit = 1;
}

python$target:::module-exec-done
/copyinstr(arg0) == "mymodule"/
{
	self->traceit = 0;
}

python$target:::function-entry
/self->traceit/
{
        printf("%d\t%*s:", timestamp, 15, probename);
        printf("%*s", self->indent, "");
        printf("%s:%s:%d\n", copyinstr(arg0), copyinstr(arg1), arg2);
}