Commit e79457a5 authored by Adrian Hunter's avatar Adrian Hunter Committed by Arnaldo Carvalho de Melo
Browse files

perf scripting python: Add perf_sample_srcline() and perf_sample_srccode() 



Add perf_sample_srcline() and perf_sample_srccode() to the
perf_trace_context module so that a script can get the srcline or srccode
information.

Signed-off-by: default avatarAdrian Hunter <adrian.hunter@intel.com>
Cc: Andi Kleen <ak@linux.intel.com>
Cc: Jiri Olsa <jolsa@redhat.com>
Link: https://lore.kernel.org/r/20210530192308.7382-11-adrian.hunter@intel.com


Signed-off-by: default avatarArnaldo Carvalho de Melo <acme@redhat.com>
parent 7d00540d

tools/perf/scripts/python/Perf-Trace-Util/Context.c

+56 −0
Original line number Diff line number Diff line
@@ -5,14 +5,23 @@ 
 * Copyright (C) 2010 Tom Zanussi <tzanussi@gmail.com>

 */



/*

 * Use Py_ssize_t for '#' formats to avoid DeprecationWarning: PY_SSIZE_T_CLEAN

 * will be required for '#' formats.

 */

#define PY_SSIZE_T_CLEAN



#include <Python.h>

#include "../../../util/trace-event.h"

#include "../../../util/event.h"

#include "../../../util/symbol.h"

#include "../../../util/thread.h"

#include "../../../util/map.h"

#include "../../../util/maps.h"

#include "../../../util/auxtrace.h"

#include "../../../util/session.h"

#include "../../../util/srcline.h"

#include "../../../util/srccode.h"



#if PY_MAJOR_VERSION < 3

#define _PyCapsule_GetPointer(arg1, arg2) \
@@ -125,6 +134,49 @@ static PyObject *perf_set_itrace_options(PyObject *obj, PyObject *args) 
	return Py_BuildValue("i", retval);

}



static PyObject *perf_sample_src(PyObject *obj, PyObject *args, bool get_srccode)

{

	struct scripting_context *c = get_scripting_context(args);

	unsigned int line = 0;

	char *srcfile = NULL;

	char *srccode = NULL;

	PyObject *result;

	struct map *map;

	int len = 0;

	u64 addr;



	if (!c)

		return NULL;



	map = c->al->map;

	addr = c->al->addr;



	if (map && map->dso)

		srcfile = get_srcline_split(map->dso, map__rip_2objdump(map, addr), &line);



	if (get_srccode) {

		if (srcfile)

			srccode = find_sourceline(srcfile, line, &len);

		result = Py_BuildValue("(sIs#)", srcfile, line, srccode, (Py_ssize_t)len);

	} else {

		result = Py_BuildValue("(sI)", srcfile, line);

	}



	free(srcfile);



	return result;

}



static PyObject *perf_sample_srcline(PyObject *obj, PyObject *args)

{

	return perf_sample_src(obj, args, false);

}



static PyObject *perf_sample_srccode(PyObject *obj, PyObject *args)

{

	return perf_sample_src(obj, args, true);

}



static PyMethodDef ContextMethods[] = {

	{ "common_pc", perf_trace_context_common_pc, METH_VARARGS,

	  "Get the common preempt count event field value."},
@@ -136,6 +188,10 @@ static PyMethodDef ContextMethods[] = { 
	  METH_VARARGS,	"Get the machine code instruction."},

	{ "perf_set_itrace_options", perf_set_itrace_options,

	  METH_VARARGS,	"Set --itrace options."},

	{ "perf_sample_srcline", perf_sample_srcline,

	  METH_VARARGS,	"Get source file name and line number."},

	{ "perf_sample_srccode", perf_sample_srccode,

	  METH_VARARGS,	"Get source file name, line number and line."},

	{ NULL, NULL, 0, NULL}

};