perf script: Add general python handler to process non-tracepoint events

This patch just follows Robert Richter's idea and the commit 37a058ea0
	"perf script: Add generic perl handler to process events"
to similarly add a python handler for general events other than tracepoints.

For non-tracepoint events, this patch will try to find a function named
"process_event" in the python script, and pass the event attribute,
perf_sample, raw_data in format of raw string. And the python script can
use "struct" module's unpack function to disasemble the needed info and process.

Signed-off-by: Feng Tang <feng.tang@intel.com>
Cc: Andi Kleen <andi@firstfloor.org>
Cc: David Ahern <dsahern@gmail.com>
Cc: Ingo Molnar <mingo@elte.hu>
Cc: Peter Zijlstra <peterz@infradead.org>
Cc: Robert Richter <robert.richter@amd.com>
Cc: Stephane Eranian <eranian@google.com>
Link: http://lkml.kernel.org/r/1344419875-21665-2-git-send-email-feng.tang@intel.com
[ committer note: Fixed up wrt da37896, i.e. pevent parm in script event handlers ]
Signed-off-by: Arnaldo Carvalho de Melo <acme@redhat.com>

diff --git a/tools/perf/util/scripting-engines/trace-event-python.c b/tools/perf/util/scripting-engines/trace-event-python.c
index df7d33d1de0f2b1f65db51861456d67735631436..b9010d878b4bf0a749a34fff7900423783ce4701 100644 (file)
--- a/tools/perf/util/scripting-engines/trace-event-python.c
+++ b/tools/perf/util/scripting-engines/trace-event-python.c
@@ -32,6 +32,7 @@
 #include "../event.h"
 #include "../thread.h"
 #include "../trace-event.h"
+#include "../evsel.h"
 
 PyMODINIT_FUNC initperf_trace_context(void);
 
@@ -220,7 +221,7 @@ static inline struct event_format *find_cache_event(struct perf_evsel *evsel)
        return event;
 }
 
-static void python_process_event(union perf_event *perf_event __unused,
+static void python_process_tracepoint(union perf_event *perf_event __unused,
                                 struct perf_sample *sample,
                                 struct perf_evsel *evsel,
                                 struct machine *machine __unused,
@@ -337,6 +338,62 @@ static void python_process_event(union perf_event *perf_event __unused,
        Py_DECREF(t);
 }
 
+static void python_process_general_event(union perf_event *perf_event __unused,
+                                        struct perf_sample *sample,
+                                        struct perf_evsel *evsel,
+                                        struct machine *machine __unused,
+                                        struct thread *thread __unused)
+{
+       PyObject *handler, *retval, *t;
+       static char handler_name[64];
+       unsigned n = 0;
+       void *data = sample->raw_data;
+
+       t = PyTuple_New(MAX_FIELDS);
+       if (!t)
+               Py_FatalError("couldn't create Python tuple");
+
+       snprintf(handler_name, sizeof(handler_name), "%s", "process_event");
+
+       handler = PyDict_GetItemString(main_dict, handler_name);
+       if (handler && !PyCallable_Check(handler)) {
+               handler = NULL;
+               goto exit;
+       }
+
+       /* Pass 3 parameters: event_attr, perf_sample, raw data */
+       PyTuple_SetItem(t, n++, PyString_FromStringAndSize((void *)&evsel->attr, sizeof(evsel->attr)));
+       PyTuple_SetItem(t, n++, PyString_FromStringAndSize((void *)sample, sizeof(*sample)));
+       PyTuple_SetItem(t, n++, PyString_FromStringAndSize(data, sample->raw_size));
+
+       if (_PyTuple_Resize(&t, n) == -1)
+               Py_FatalError("error resizing Python tuple");
+
+       retval = PyObject_CallObject(handler, t);
+       if (retval == NULL)
+               handler_call_die(handler_name);
+exit:
+       Py_DECREF(t);
+}
+
+static void python_process_event(union perf_event *perf_event,
+                                struct perf_sample *sample,
+                                struct perf_evsel *evsel,
+                                struct machine *machine,
+                                struct thread *thread)
+{
+       switch (evsel->attr.type) {
+       case PERF_TYPE_TRACEPOINT:
+               python_process_tracepoint(perf_event, sample, evsel,
+                                         machine, thread);
+               break;
+       /* Reserve for future process_hw/sw/raw APIs */
+       default:
+               python_process_general_event(perf_event, sample, evsel,
+                                            machine, thread);
+       }
+}
+
 static int run_start_sub(void)
 {
        PyObject *handler, *retval;