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: Adrian 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: Arnaldo Carvalho de Melo <acme@redhat.com>

diff --git a/tools/perf/scripts/python/Perf-Trace-Util/Context.c b/tools/perf/scripts/python/Perf-Trace-Util/Context.c
index 3c9bc12..895f5fc 100644 (file)
--- a/tools/perf/scripts/python/Perf-Trace-Util/Context.c
+++ b/tools/perf/scripts/python/Perf-Trace-Util/Context.c
@@ -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 @@ out:
        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}
 };