aboutsummaryrefslogtreecommitdiffstats
path: root/tools
diff options
context:
space:
mode:
Diffstat (limited to 'tools')
-rw-r--r--tools/perf/Makefile21
-rw-r--r--tools/perf/builtin-trace.c1
-rw-r--r--tools/perf/scripts/python/Perf-Trace-Util/Context.c88
-rw-r--r--tools/perf/scripts/python/Perf-Trace-Util/lib/Perf/Trace/Core.py91
-rw-r--r--tools/perf/scripts/python/Perf-Trace-Util/lib/Perf/Trace/Util.py25
-rw-r--r--tools/perf/util/scripting-engines/trace-event-python.c576
-rw-r--r--tools/perf/util/trace-event-scripting.c61
-rw-r--r--tools/perf/util/trace-event.h1
8 files changed, 864 insertions, 0 deletions
diff --git a/tools/perf/Makefile b/tools/perf/Makefile
index 0a3c0c8b3fc0..14273164db04 100644
--- a/tools/perf/Makefile
+++ b/tools/perf/Makefile
@@ -522,6 +522,19 @@ else
522 LIB_OBJS += scripts/perl/Perf-Trace-Util/Context.o 522 LIB_OBJS += scripts/perl/Perf-Trace-Util/Context.o
523endif 523endif
524 524
525ifndef NO_LIBPYTHON
526PYTHON_EMBED_LDOPTS = `python-config --ldflags 2>/dev/null`
527PYTHON_EMBED_CCOPTS = `python-config --cflags 2>/dev/null`
528endif
529
530ifneq ($(shell sh -c "(echo '\#include <Python.h>'; echo 'int main(void) { Py_Initialize(); return 0; }') | $(CC) -x c - $(PYTHON_EMBED_CCOPTS) -o /dev/null $(PYTHON_EMBED_LDOPTS) > /dev/null 2>&1 && echo y"), y)
531 BASIC_CFLAGS += -DNO_LIBPYTHON
532else
533 ALL_LDFLAGS += $(PYTHON_EMBED_LDOPTS)
534 LIB_OBJS += util/scripting-engines/trace-event-python.o
535 LIB_OBJS += scripts/python/Perf-Trace-Util/Context.o
536endif
537
525ifdef NO_DEMANGLE 538ifdef NO_DEMANGLE
526 BASIC_CFLAGS += -DNO_DEMANGLE 539 BASIC_CFLAGS += -DNO_DEMANGLE
527else 540else
@@ -899,6 +912,12 @@ util/scripting-engines/trace-event-perl.o: util/scripting-engines/trace-event-pe
899scripts/perl/Perf-Trace-Util/Context.o: scripts/perl/Perf-Trace-Util/Context.c PERF-CFLAGS 912scripts/perl/Perf-Trace-Util/Context.o: scripts/perl/Perf-Trace-Util/Context.c PERF-CFLAGS
900 $(QUIET_CC)$(CC) -o scripts/perl/Perf-Trace-Util/Context.o -c $(ALL_CFLAGS) $(PERL_EMBED_CCOPTS) -Wno-redundant-decls -Wno-strict-prototypes -Wno-unused-parameter -Wno-nested-externs $< 913 $(QUIET_CC)$(CC) -o scripts/perl/Perf-Trace-Util/Context.o -c $(ALL_CFLAGS) $(PERL_EMBED_CCOPTS) -Wno-redundant-decls -Wno-strict-prototypes -Wno-unused-parameter -Wno-nested-externs $<
901 914
915util/scripting-engines/trace-event-python.o: util/scripting-engines/trace-event-python.c PERF-CFLAGS
916 $(QUIET_CC)$(CC) -o util/scripting-engines/trace-event-python.o -c $(ALL_CFLAGS) $(PYTHON_EMBED_CCOPTS) -Wno-redundant-decls -Wno-strict-prototypes -Wno-unused-parameter -Wno-shadow $<
917
918scripts/python/Perf-Trace-Util/Context.o: scripts/python/Perf-Trace-Util/Context.c PERF-CFLAGS
919 $(QUIET_CC)$(CC) -o scripts/python/Perf-Trace-Util/Context.o -c $(ALL_CFLAGS) $(PYTHON_EMBED_CCOPTS) -Wno-redundant-decls -Wno-strict-prototypes -Wno-unused-parameter -Wno-nested-externs $<
920
902perf-%$X: %.o $(PERFLIBS) 921perf-%$X: %.o $(PERFLIBS)
903 $(QUIET_LINK)$(CC) $(ALL_CFLAGS) -o $@ $(ALL_LDFLAGS) $(filter %.o,$^) $(LIBS) 922 $(QUIET_LINK)$(CC) $(ALL_CFLAGS) -o $@ $(ALL_LDFLAGS) $(filter %.o,$^) $(LIBS)
904 923
@@ -1012,6 +1031,8 @@ install: all
1012 $(INSTALL) scripts/perl/Perf-Trace-Util/lib/Perf/Trace/* -t '$(DESTDIR_SQ)$(perfexec_instdir_SQ)/scripts/perl/Perf-Trace-Util/lib/Perf/Trace' 1031 $(INSTALL) scripts/perl/Perf-Trace-Util/lib/Perf/Trace/* -t '$(DESTDIR_SQ)$(perfexec_instdir_SQ)/scripts/perl/Perf-Trace-Util/lib/Perf/Trace'
1013 $(INSTALL) scripts/perl/*.pl -t '$(DESTDIR_SQ)$(perfexec_instdir_SQ)/scripts/perl' 1032 $(INSTALL) scripts/perl/*.pl -t '$(DESTDIR_SQ)$(perfexec_instdir_SQ)/scripts/perl'
1014 $(INSTALL) scripts/perl/bin/* -t '$(DESTDIR_SQ)$(perfexec_instdir_SQ)/scripts/perl/bin' 1033 $(INSTALL) scripts/perl/bin/* -t '$(DESTDIR_SQ)$(perfexec_instdir_SQ)/scripts/perl/bin'
1034 $(INSTALL) -d -m 755 '$(DESTDIR_SQ)$(perfexec_instdir_SQ)/scripts/python/Perf-Trace-Util/lib/Perf/Trace'
1035 $(INSTALL) scripts/python/Perf-Trace-Util/lib/Perf/Trace/* -t '$(DESTDIR_SQ)$(perfexec_instdir_SQ)/scripts/python/Perf-Trace-Util/lib/Perf/Trace'
1015 1036
1016ifdef BUILT_INS 1037ifdef BUILT_INS
1017 $(INSTALL) -d -m 755 '$(DESTDIR_SQ)$(perfexec_instdir_SQ)' 1038 $(INSTALL) -d -m 755 '$(DESTDIR_SQ)$(perfexec_instdir_SQ)'
diff --git a/tools/perf/builtin-trace.c b/tools/perf/builtin-trace.c
index d5d20c34e221..5db687fc13de 100644
--- a/tools/perf/builtin-trace.c
+++ b/tools/perf/builtin-trace.c
@@ -44,6 +44,7 @@ static void setup_scripting(void)
44 perf_set_argv_exec_path(perf_exec_path()); 44 perf_set_argv_exec_path(perf_exec_path());
45 45
46 setup_perl_scripting(); 46 setup_perl_scripting();
47 setup_python_scripting();
47 48
48 scripting_ops = &default_scripting_ops; 49 scripting_ops = &default_scripting_ops;
49} 50}
diff --git a/tools/perf/scripts/python/Perf-Trace-Util/Context.c b/tools/perf/scripts/python/Perf-Trace-Util/Context.c
new file mode 100644
index 000000000000..957085dd5d8d
--- /dev/null
+++ b/tools/perf/scripts/python/Perf-Trace-Util/Context.c
@@ -0,0 +1,88 @@
1/*
2 * Context.c. Python interfaces for perf trace.
3 *
4 * Copyright (C) 2010 Tom Zanussi <tzanussi@gmail.com>
5 *
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19 *
20 */
21
22#include <Python.h>
23#include "../../../perf.h"
24#include "../../../util/trace-event.h"
25
26PyMODINIT_FUNC initperf_trace_context(void);
27
28static PyObject *perf_trace_context_common_pc(PyObject *self, PyObject *args)
29{
30 static struct scripting_context *scripting_context;
31 PyObject *context;
32 int retval;
33
34 if (!PyArg_ParseTuple(args, "O", &context))
35 return NULL;
36
37 scripting_context = PyCObject_AsVoidPtr(context);
38 retval = common_pc(scripting_context);
39
40 return Py_BuildValue("i", retval);
41}
42
43static PyObject *perf_trace_context_common_flags(PyObject *self,
44 PyObject *args)
45{
46 static struct scripting_context *scripting_context;
47 PyObject *context;
48 int retval;
49
50 if (!PyArg_ParseTuple(args, "O", &context))
51 return NULL;
52
53 scripting_context = PyCObject_AsVoidPtr(context);
54 retval = common_flags(scripting_context);
55
56 return Py_BuildValue("i", retval);
57}
58
59static PyObject *perf_trace_context_common_lock_depth(PyObject *self,
60 PyObject *args)
61{
62 static struct scripting_context *scripting_context;
63 PyObject *context;
64 int retval;
65
66 if (!PyArg_ParseTuple(args, "O", &context))
67 return NULL;
68
69 scripting_context = PyCObject_AsVoidPtr(context);
70 retval = common_lock_depth(scripting_context);
71
72 return Py_BuildValue("i", retval);
73}
74
75static PyMethodDef ContextMethods[] = {
76 { "common_pc", perf_trace_context_common_pc, METH_VARARGS,
77 "Get the common preempt count event field value."},
78 { "common_flags", perf_trace_context_common_flags, METH_VARARGS,
79 "Get the common flags event field value."},
80 { "common_lock_depth", perf_trace_context_common_lock_depth,
81 METH_VARARGS, "Get the common lock depth event field value."},
82 { NULL, NULL, 0, NULL}
83};
84
85PyMODINIT_FUNC initperf_trace_context(void)
86{
87 (void) Py_InitModule("perf_trace_context", ContextMethods);
88}
diff --git a/tools/perf/scripts/python/Perf-Trace-Util/lib/Perf/Trace/Core.py b/tools/perf/scripts/python/Perf-Trace-Util/lib/Perf/Trace/Core.py
new file mode 100644
index 000000000000..1dc464ee2ca8
--- /dev/null
+++ b/tools/perf/scripts/python/Perf-Trace-Util/lib/Perf/Trace/Core.py
@@ -0,0 +1,91 @@
1# Core.py - Python extension for perf trace, core functions
2#
3# Copyright (C) 2010 by Tom Zanussi <tzanussi@gmail.com>
4#
5# This software may be distributed under the terms of the GNU General
6# Public License ("GPL") version 2 as published by the Free Software
7# Foundation.
8
9from collections import defaultdict
10
11def autodict():
12 return defaultdict(autodict)
13
14flag_fields = autodict()
15symbolic_fields = autodict()
16
17def define_flag_field(event_name, field_name, delim):
18 flag_fields[event_name][field_name]['delim'] = delim
19
20def define_flag_value(event_name, field_name, value, field_str):
21 flag_fields[event_name][field_name]['values'][value] = field_str
22
23def define_symbolic_field(event_name, field_name):
24 # nothing to do, really
25 pass
26
27def define_symbolic_value(event_name, field_name, value, field_str):
28 symbolic_fields[event_name][field_name]['values'][value] = field_str
29
30def flag_str(event_name, field_name, value):
31 string = ""
32
33 if flag_fields[event_name][field_name]:
34 print_delim = 0
35 keys = flag_fields[event_name][field_name]['values'].keys()
36 keys.sort()
37 for idx in keys:
38 if not value and not idx:
39 string += flag_fields[event_name][field_name]['values'][idx]
40 break
41 if idx and (value & idx) == idx:
42 if print_delim and flag_fields[event_name][field_name]['delim']:
43 string += " " + flag_fields[event_name][field_name]['delim'] + " "
44 string += flag_fields[event_name][field_name]['values'][idx]
45 print_delim = 1
46 value &= ~idx
47
48 return string
49
50def symbol_str(event_name, field_name, value):
51 string = ""
52
53 if symbolic_fields[event_name][field_name]:
54 keys = symbolic_fields[event_name][field_name]['values'].keys()
55 keys.sort()
56 for idx in keys:
57 if not value and not idx:
58 string = symbolic_fields[event_name][field_name]['values'][idx]
59 break
60 if (value == idx):
61 string = symbolic_fields[event_name][field_name]['values'][idx]
62 break
63
64 return string
65
66trace_flags = { 0x00: "NONE", \
67 0x01: "IRQS_OFF", \
68 0x02: "IRQS_NOSUPPORT", \
69 0x04: "NEED_RESCHED", \
70 0x08: "HARDIRQ", \
71 0x10: "SOFTIRQ" }
72
73def trace_flag_str(value):
74 string = ""
75 print_delim = 0
76
77 keys = trace_flags.keys()
78
79 for idx in keys:
80 if not value and not idx:
81 string += "NONE"
82 break
83
84 if idx and (value & idx) == idx:
85 if print_delim:
86 string += " | ";
87 string += trace_flags[idx]
88 print_delim = 1
89 value &= ~idx
90
91 return string
diff --git a/tools/perf/scripts/python/Perf-Trace-Util/lib/Perf/Trace/Util.py b/tools/perf/scripts/python/Perf-Trace-Util/lib/Perf/Trace/Util.py
new file mode 100644
index 000000000000..83e91435ed09
--- /dev/null
+++ b/tools/perf/scripts/python/Perf-Trace-Util/lib/Perf/Trace/Util.py
@@ -0,0 +1,25 @@
1# Util.py - Python extension for perf trace, miscellaneous utility code
2#
3# Copyright (C) 2010 by Tom Zanussi <tzanussi@gmail.com>
4#
5# This software may be distributed under the terms of the GNU General
6# Public License ("GPL") version 2 as published by the Free Software
7# Foundation.
8
9NSECS_PER_SEC = 1000000000
10
11def avg(total, n):
12 return total / n
13
14def nsecs(secs, nsecs):
15 return secs * NSECS_PER_SEC + nsecs
16
17def nsecs_secs(nsecs):
18 return nsecs / NSECS_PER_SEC
19
20def nsecs_nsecs(nsecs):
21 return nsecs % NSECS_PER_SEC
22
23def nsecs_str(nsecs):
24 str = "%5u.%09u" % (nsecs_secs(nsecs), nsecs_nsecs(nsecs)),
25 return str
diff --git a/tools/perf/util/scripting-engines/trace-event-python.c b/tools/perf/util/scripting-engines/trace-event-python.c
new file mode 100644
index 000000000000..d402f64f9b46
--- /dev/null
+++ b/tools/perf/util/scripting-engines/trace-event-python.c
@@ -0,0 +1,576 @@
1/*
2 * trace-event-python. Feed trace events to an embedded Python interpreter.
3 *
4 * Copyright (C) 2010 Tom Zanussi <tzanussi@gmail.com>
5 *
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19 *
20 */
21
22#include <Python.h>
23
24#include <stdio.h>
25#include <stdlib.h>
26#include <string.h>
27#include <ctype.h>
28#include <errno.h>
29
30#include "../../perf.h"
31#include "../util.h"
32#include "../trace-event.h"
33
34PyMODINIT_FUNC initperf_trace_context(void);
35
36#define FTRACE_MAX_EVENT \
37 ((1 << (sizeof(unsigned short) * 8)) - 1)
38
39struct event *events[FTRACE_MAX_EVENT];
40
41#define MAX_FIELDS 64
42#define N_COMMON_FIELDS 7
43
44extern struct scripting_context *scripting_context;
45
46static char *cur_field_name;
47static int zero_flag_atom;
48
49static PyObject *main_module, *main_dict;
50
51static void handler_call_die(const char *handler_name)
52{
53 PyErr_Print();
54 Py_FatalError("problem in Python trace event handler");
55}
56
57static void define_value(enum print_arg_type field_type,
58 const char *ev_name,
59 const char *field_name,
60 const char *field_value,
61 const char *field_str)
62{
63 const char *handler_name = "define_flag_value";
64 PyObject *handler, *t, *retval;
65 unsigned long long value;
66 unsigned n = 0;
67
68 if (field_type == PRINT_SYMBOL)
69 handler_name = "define_symbolic_value";
70
71 t = PyTuple_New(MAX_FIELDS);
72 if (!t)
73 Py_FatalError("couldn't create Python tuple");
74
75 value = eval_flag(field_value);
76
77 PyTuple_SetItem(t, n++, PyString_FromString(ev_name));
78 PyTuple_SetItem(t, n++, PyString_FromString(field_name));
79 PyTuple_SetItem(t, n++, PyInt_FromLong(value));
80 PyTuple_SetItem(t, n++, PyString_FromString(field_str));
81
82 if (_PyTuple_Resize(&t, n) == -1)
83 Py_FatalError("error resizing Python tuple");
84
85 handler = PyDict_GetItemString(main_dict, handler_name);
86 if (handler && PyCallable_Check(handler)) {
87 retval = PyObject_CallObject(handler, t);
88 if (retval == NULL)
89 handler_call_die(handler_name);
90 }
91
92 Py_DECREF(t);
93}
94
95static void define_values(enum print_arg_type field_type,
96 struct print_flag_sym *field,
97 const char *ev_name,
98 const char *field_name)
99{
100 define_value(field_type, ev_name, field_name, field->value,
101 field->str);
102
103 if (field->next)
104 define_values(field_type, field->next, ev_name, field_name);
105}
106
107static void define_field(enum print_arg_type field_type,
108 const char *ev_name,
109 const char *field_name,
110 const char *delim)
111{
112 const char *handler_name = "define_flag_field";
113 PyObject *handler, *t, *retval;
114 unsigned n = 0;
115
116 if (field_type == PRINT_SYMBOL)
117 handler_name = "define_symbolic_field";
118
119 t = PyTuple_New(MAX_FIELDS);
120 if (!t)
121 Py_FatalError("couldn't create Python tuple");
122
123 PyTuple_SetItem(t, n++, PyString_FromString(ev_name));
124 PyTuple_SetItem(t, n++, PyString_FromString(field_name));
125 if (field_type == PRINT_FLAGS)
126 PyTuple_SetItem(t, n++, PyString_FromString(delim));
127
128 if (_PyTuple_Resize(&t, n) == -1)
129 Py_FatalError("error resizing Python tuple");
130
131 handler = PyDict_GetItemString(main_dict, handler_name);
132 if (handler && PyCallable_Check(handler)) {
133 retval = PyObject_CallObject(handler, t);
134 if (retval == NULL)
135 handler_call_die(handler_name);
136 }
137
138 Py_DECREF(t);
139}
140
141static void define_event_symbols(struct event *event,
142 const char *ev_name,
143 struct print_arg *args)
144{
145 switch (args->type) {
146 case PRINT_NULL:
147 break;
148 case PRINT_ATOM:
149 define_value(PRINT_FLAGS, ev_name, cur_field_name, "0",
150 args->atom.atom);
151 zero_flag_atom = 0;
152 break;
153 case PRINT_FIELD:
154 if (cur_field_name)
155 free(cur_field_name);
156 cur_field_name = strdup(args->field.name);
157 break;
158 case PRINT_FLAGS:
159 define_event_symbols(event, ev_name, args->flags.field);
160 define_field(PRINT_FLAGS, ev_name, cur_field_name,
161 args->flags.delim);
162 define_values(PRINT_FLAGS, args->flags.flags, ev_name,
163 cur_field_name);
164 break;
165 case PRINT_SYMBOL:
166 define_event_symbols(event, ev_name, args->symbol.field);
167 define_field(PRINT_SYMBOL, ev_name, cur_field_name, NULL);
168 define_values(PRINT_SYMBOL, args->symbol.symbols, ev_name,
169 cur_field_name);
170 break;
171 case PRINT_STRING:
172 break;
173 case PRINT_TYPE:
174 define_event_symbols(event, ev_name, args->typecast.item);
175 break;
176 case PRINT_OP:
177 if (strcmp(args->op.op, ":") == 0)
178 zero_flag_atom = 1;
179 define_event_symbols(event, ev_name, args->op.left);
180 define_event_symbols(event, ev_name, args->op.right);
181 break;
182 default:
183 /* we should warn... */
184 return;
185 }
186
187 if (args->next)
188 define_event_symbols(event, ev_name, args->next);
189}
190
191static inline struct event *find_cache_event(int type)
192{
193 static char ev_name[256];
194 struct event *event;
195
196 if (events[type])
197 return events[type];
198
199 events[type] = event = trace_find_event(type);
200 if (!event)
201 return NULL;
202
203 sprintf(ev_name, "%s__%s", event->system, event->name);
204
205 define_event_symbols(event, ev_name, event->print_fmt.args);
206
207 return event;
208}
209
210static void python_process_event(int cpu, void *data,
211 int size __unused,
212 unsigned long long nsecs, char *comm)
213{
214 PyObject *handler, *retval, *context, *t;
215 static char handler_name[256];
216 struct format_field *field;
217 unsigned long long val;
218 unsigned long s, ns;
219 struct event *event;
220 unsigned n = 0;
221 int type;
222 int pid;
223
224 t = PyTuple_New(MAX_FIELDS);
225 if (!t)
226 Py_FatalError("couldn't create Python tuple");
227
228 type = trace_parse_common_type(data);
229
230 event = find_cache_event(type);
231 if (!event)
232 die("ug! no event found for type %d", type);
233
234 pid = trace_parse_common_pid(data);
235
236 sprintf(handler_name, "%s__%s", event->system, event->name);
237
238 s = nsecs / NSECS_PER_SEC;
239 ns = nsecs - s * NSECS_PER_SEC;
240
241 scripting_context->event_data = data;
242
243 context = PyCObject_FromVoidPtr(scripting_context, NULL);
244
245 PyTuple_SetItem(t, n++, PyString_FromString(handler_name));
246 PyTuple_SetItem(t, n++,
247 PyCObject_FromVoidPtr(scripting_context, NULL));
248 PyTuple_SetItem(t, n++, PyInt_FromLong(cpu));
249 PyTuple_SetItem(t, n++, PyInt_FromLong(s));
250 PyTuple_SetItem(t, n++, PyInt_FromLong(ns));
251 PyTuple_SetItem(t, n++, PyInt_FromLong(pid));
252 PyTuple_SetItem(t, n++, PyString_FromString(comm));
253
254 for (field = event->format.fields; field; field = field->next) {
255 if (field->flags & FIELD_IS_STRING) {
256 int offset;
257 if (field->flags & FIELD_IS_DYNAMIC) {
258 offset = *(int *)(data + field->offset);
259 offset &= 0xffff;
260 } else
261 offset = field->offset;
262 PyTuple_SetItem(t, n++,
263 PyString_FromString((char *)data + offset));
264 } else { /* FIELD_IS_NUMERIC */
265 val = read_size(data + field->offset, field->size);
266 if (field->flags & FIELD_IS_SIGNED) {
267 PyTuple_SetItem(t, n++, PyInt_FromLong(val));
268 } else {
269 PyTuple_SetItem(t, n++, PyInt_FromLong(val));
270 }
271 }
272 }
273
274 if (_PyTuple_Resize(&t, n) == -1)
275 Py_FatalError("error resizing Python tuple");
276
277 handler = PyDict_GetItemString(main_dict, handler_name);
278 if (handler && PyCallable_Check(handler)) {
279 retval = PyObject_CallObject(handler, t);
280 if (retval == NULL)
281 handler_call_die(handler_name);
282 } else {
283 handler = PyDict_GetItemString(main_dict, "trace_unhandled");
284 if (handler && PyCallable_Check(handler)) {
285 if (_PyTuple_Resize(&t, N_COMMON_FIELDS) == -1)
286 Py_FatalError("error resizing Python tuple");
287
288 retval = PyObject_CallObject(handler, t);
289 if (retval == NULL)
290 handler_call_die("trace_unhandled");
291 }
292 }
293
294 Py_DECREF(t);
295}
296
297static int run_start_sub(void)
298{
299 PyObject *handler, *retval;
300 int err = 0;
301
302 main_module = PyImport_AddModule("__main__");
303 if (main_module == NULL)
304 return -1;
305 Py_INCREF(main_module);
306
307 main_dict = PyModule_GetDict(main_module);
308 if (main_dict == NULL) {
309 err = -1;
310 goto error;
311 }
312 Py_INCREF(main_dict);
313
314 handler = PyDict_GetItemString(main_dict, "trace_begin");
315 if (handler == NULL || !PyCallable_Check(handler))
316 goto out;
317
318 retval = PyObject_CallObject(handler, NULL);
319 if (retval == NULL)
320 handler_call_die("trace_begin");
321
322 Py_DECREF(retval);
323 return err;
324error:
325 Py_XDECREF(main_dict);
326 Py_XDECREF(main_module);
327out:
328 return err;
329}
330
331/*
332 * Start trace script
333 */
334static int python_start_script(const char *script, int argc, const char **argv)
335{
336 const char **command_line;
337 char buf[PATH_MAX];
338 int i, err = 0;
339 FILE *fp;
340
341 command_line = malloc((argc + 1) * sizeof(const char *));
342 command_line[0] = script;
343 for (i = 1; i < argc + 1; i++)
344 command_line[i] = argv[i - 1];
345
346 Py_Initialize();
347
348 initperf_trace_context();
349
350 PySys_SetArgv(argc + 1, (char **)command_line);
351
352 fp = fopen(script, "r");
353 if (!fp) {
354 sprintf(buf, "Can't open python script \"%s\"", script);
355 perror(buf);
356 err = -1;
357 goto error;
358 }
359
360 err = PyRun_SimpleFile(fp, script);
361 if (err) {
362 fprintf(stderr, "Error running python script %s\n", script);
363 goto error;
364 }
365
366 err = run_start_sub();
367 if (err) {
368 fprintf(stderr, "Error starting python script %s\n", script);
369 goto error;
370 }
371
372 free(command_line);
373 fprintf(stderr, "perf trace started with Python script %s\n\n",
374 script);
375
376 return err;
377error:
378 Py_Finalize();
379 free(command_line);
380
381 return err;
382}
383
384/*
385 * Stop trace script
386 */
387static int python_stop_script(void)
388{
389 PyObject *handler, *retval;
390 int err = 0;
391
392 handler = PyDict_GetItemString(main_dict, "trace_end");
393 if (handler == NULL || !PyCallable_Check(handler))
394 goto out;
395
396 retval = PyObject_CallObject(handler, NULL);
397 if (retval == NULL)
398 handler_call_die("trace_end");
399 else
400 Py_DECREF(retval);
401out:
402 Py_XDECREF(main_dict);
403 Py_XDECREF(main_module);
404 Py_Finalize();
405
406 fprintf(stderr, "\nperf trace Python script stopped\n");
407
408 return err;
409}
410
411static int python_generate_script(const char *outfile)
412{
413 struct event *event = NULL;
414 struct format_field *f;
415 char fname[PATH_MAX];
416 int not_first, count;
417 FILE *ofp;
418
419 sprintf(fname, "%s.py", outfile);
420 ofp = fopen(fname, "w");
421 if (ofp == NULL) {
422 fprintf(stderr, "couldn't open %s\n", fname);
423 return -1;
424 }
425 fprintf(ofp, "# perf trace event handlers, "
426 "generated by perf trace -g python\n");
427
428 fprintf(ofp, "# Licensed under the terms of the GNU GPL"
429 " License version 2\n\n");
430
431 fprintf(ofp, "# The common_* event handler fields are the most useful "
432 "fields common to\n");
433
434 fprintf(ofp, "# all events. They don't necessarily correspond to "
435 "the 'common_*' fields\n");
436
437 fprintf(ofp, "# in the format files. Those fields not available as "
438 "handler params can\n");
439
440 fprintf(ofp, "# be retrieved using Python functions of the form "
441 "common_*(context).\n");
442
443 fprintf(ofp, "# See the perf-trace-python Documentation for the list "
444 "of available functions.\n\n");
445
446 fprintf(ofp, "import os\n");
447 fprintf(ofp, "import sys\n\n");
448
449 fprintf(ofp, "sys.path.append(os.environ['PERF_EXEC_PATH'] + \\\n");
450 fprintf(ofp, "\t'/scripts/python/Perf-Trace-Util/lib/Perf/Trace')\n");
451 fprintf(ofp, "\nfrom perf_trace_context import *\n");
452 fprintf(ofp, "from Core import *\n\n\n");
453
454 fprintf(ofp, "def trace_begin():\n");
455 fprintf(ofp, "\tprint \"in trace_begin\"\n\n");
456
457 fprintf(ofp, "def trace_end():\n");
458 fprintf(ofp, "\tprint \"in trace_end\"\n\n");
459
460 while ((event = trace_find_next_event(event))) {
461 fprintf(ofp, "def %s__%s(", event->system, event->name);
462 fprintf(ofp, "event_name, ");
463 fprintf(ofp, "context, ");
464 fprintf(ofp, "common_cpu,\n");
465 fprintf(ofp, "\tcommon_secs, ");
466 fprintf(ofp, "common_nsecs, ");
467 fprintf(ofp, "common_pid, ");
468 fprintf(ofp, "common_comm,\n\t");
469
470 not_first = 0;
471 count = 0;
472
473 for (f = event->format.fields; f; f = f->next) {
474 if (not_first++)
475 fprintf(ofp, ", ");
476 if (++count % 5 == 0)
477 fprintf(ofp, "\n\t");
478
479 fprintf(ofp, "%s", f->name);
480 }
481 fprintf(ofp, "):\n");
482
483 fprintf(ofp, "\t\tprint_header(event_name, common_cpu, "
484 "common_secs, common_nsecs,\n\t\t\t"
485 "common_pid, common_comm)\n\n");
486
487 fprintf(ofp, "\t\tprint \"");
488
489 not_first = 0;
490 count = 0;
491
492 for (f = event->format.fields; f; f = f->next) {
493 if (not_first++)
494 fprintf(ofp, ", ");
495 if (count && count % 3 == 0) {
496 fprintf(ofp, "\" \\\n\t\t\"");
497 }
498 count++;
499
500 fprintf(ofp, "%s=", f->name);
501 if (f->flags & FIELD_IS_STRING ||
502 f->flags & FIELD_IS_FLAG ||
503 f->flags & FIELD_IS_SYMBOLIC)
504 fprintf(ofp, "%%s");
505 else if (f->flags & FIELD_IS_SIGNED)
506 fprintf(ofp, "%%d");
507 else
508 fprintf(ofp, "%%u");
509 }
510
511 fprintf(ofp, "\\n\" %% \\\n\t\t(");
512
513 not_first = 0;
514 count = 0;
515
516 for (f = event->format.fields; f; f = f->next) {
517 if (not_first++)
518 fprintf(ofp, ", ");
519
520 if (++count % 5 == 0)
521 fprintf(ofp, "\n\t\t");
522
523 if (f->flags & FIELD_IS_FLAG) {
524 if ((count - 1) % 5 != 0) {
525 fprintf(ofp, "\n\t\t");
526 count = 4;
527 }
528 fprintf(ofp, "flag_str(\"");
529 fprintf(ofp, "%s__%s\", ", event->system,
530 event->name);
531 fprintf(ofp, "\"%s\", %s)", f->name,
532 f->name);
533 } else if (f->flags & FIELD_IS_SYMBOLIC) {
534 if ((count - 1) % 5 != 0) {
535 fprintf(ofp, "\n\t\t");
536 count = 4;
537 }
538 fprintf(ofp, "symbol_str(\"");
539 fprintf(ofp, "%s__%s\", ", event->system,
540 event->name);
541 fprintf(ofp, "\"%s\", %s)", f->name,
542 f->name);
543 } else
544 fprintf(ofp, "%s", f->name);
545 }
546
547 fprintf(ofp, "),\n\n");
548 }
549
550 fprintf(ofp, "def trace_unhandled(event_name, context, "
551 "common_cpu, common_secs, common_nsecs,\n\t\t"
552 "common_pid, common_comm):\n");
553
554 fprintf(ofp, "\t\tprint_header(event_name, common_cpu, "
555 "common_secs, common_nsecs,\n\t\tcommon_pid, "
556 "common_comm)\n\n");
557
558 fprintf(ofp, "def print_header("
559 "event_name, cpu, secs, nsecs, pid, comm):\n"
560 "\tprint \"%%-20s %%5u %%05u.%%09u %%8u %%-20s \" %% \\\n\t"
561 "(event_name, cpu, secs, nsecs, pid, comm),\n");
562
563 fclose(ofp);
564
565 fprintf(stderr, "generated Python script: %s\n", fname);
566
567 return 0;
568}
569
570struct scripting_ops python_scripting_ops = {
571 .name = "Python",
572 .start_script = python_start_script,
573 .stop_script = python_stop_script,
574 .process_event = python_process_event,
575 .generate_script = python_generate_script,
576};
diff --git a/tools/perf/util/trace-event-scripting.c b/tools/perf/util/trace-event-scripting.c
index 9e371965c034..7ea983acfaea 100644
--- a/tools/perf/util/trace-event-scripting.c
+++ b/tools/perf/util/trace-event-scripting.c
@@ -44,6 +44,67 @@ static void process_event_unsupported(int cpu __unused,
44{ 44{
45} 45}
46 46
47static void print_python_unsupported_msg(void)
48{
49 fprintf(stderr, "Python scripting not supported."
50 " Install libpython and rebuild perf to enable it.\n"
51 "For example:\n # apt-get install python-dev (ubuntu)"
52 "\n # yum install python-devel (Fedora)"
53 "\n etc.\n");
54}
55
56static int python_start_script_unsupported(const char *script __unused,
57 int argc __unused,
58 const char **argv __unused)
59{
60 print_python_unsupported_msg();
61
62 return -1;
63}
64
65static int python_generate_script_unsupported(const char *outfile __unused)
66{
67 print_python_unsupported_msg();
68
69 return -1;
70}
71
72struct scripting_ops python_scripting_unsupported_ops = {
73 .name = "Python",
74 .start_script = python_start_script_unsupported,
75 .stop_script = stop_script_unsupported,
76 .process_event = process_event_unsupported,
77 .generate_script = python_generate_script_unsupported,
78};
79
80static void register_python_scripting(struct scripting_ops *scripting_ops)
81{
82 int err;
83 err = script_spec_register("Python", scripting_ops);
84 if (err)
85 die("error registering Python script extension");
86
87 err = script_spec_register("py", scripting_ops);
88 if (err)
89 die("error registering py script extension");
90
91 scripting_context = malloc(sizeof(struct scripting_context));
92}
93
94#ifdef NO_LIBPYTHON
95void setup_python_scripting(void)
96{
97 register_python_scripting(&python_scripting_unsupported_ops);
98}
99#else
100struct scripting_ops python_scripting_ops;
101
102void setup_python_scripting(void)
103{
104 register_python_scripting(&python_scripting_ops);
105}
106#endif
107
47static void print_perl_unsupported_msg(void) 108static void print_perl_unsupported_msg(void)
48{ 109{
49 fprintf(stderr, "Perl scripting not supported." 110 fprintf(stderr, "Perl scripting not supported."
diff --git a/tools/perf/util/trace-event.h b/tools/perf/util/trace-event.h
index aaf2da2d21e5..c3269b937db4 100644
--- a/tools/perf/util/trace-event.h
+++ b/tools/perf/util/trace-event.h
@@ -280,6 +280,7 @@ struct scripting_ops {
280int script_spec_register(const char *spec, struct scripting_ops *ops); 280int script_spec_register(const char *spec, struct scripting_ops *ops);
281 281
282void setup_perl_scripting(void); 282void setup_perl_scripting(void);
283void setup_python_scripting(void);
283 284
284struct scripting_context { 285struct scripting_context {
285 void *event_data; 286 void *event_data;