#!/usr/sbin/dtrace -Zs /* * py_malloc.d - Python libc malloc analysis. * Written for the Python DTrace provider. * * $Id: py_malloc.d,v 1.1.1.1 2015/09/30 22:01:07 christos Exp $ * * This is an expiremental script to identify who is calling malloc() for * memory allocation, and to print distribution plots of the requested bytes. * If a malloc() occured while in a Python function, then that function is * identified as responsible; else the caller of malloc() is identified as * responsible - which will be a function from the Python engine. * * USAGE: py_malloc.d { -p PID | -c cmd } # hit Ctrl-C to end * * Filename and function names are printed if available. * * COPYRIGHT: Copyright (c) 2007 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 * * 09-Sep-2007 Brendan Gregg Created this. */ #pragma D option quiet dtrace:::BEGIN { printf("Tracing... Hit Ctrl-C to end.\n"); } python$target:::function-entry { self->file = basename(copyinstr(arg0)); self->name = copyinstr(arg1); } python$target:::function-return { self->file = 0; self->name = 0; } pid$target:libc:malloc:entry /self->file != NULL/ { @malloc_func_size[self->file, self->name] = sum(arg0); @malloc_func_dist[self->file, self->name] = quantize(arg0); } pid$target:libc:malloc:entry /self->name == NULL/ { @malloc_lib_size[usym(ucaller)] = sum(arg0); @malloc_lib_dist[usym(ucaller)] = quantize(arg0); } dtrace:::END { printf("\nPython malloc byte distributions by engine caller,\n\n"); printa(" %A, total bytes = %@d %@d\n", @malloc_lib_size, @malloc_lib_dist); printf("\nPython malloc byte distributions by Python file and "); printf("function,\n\n"); printa(" %s, %s, bytes total = %@d %@d\n", @malloc_func_size, @malloc_func_dist); }