perf probe: Search SDT/cached event from all probe caches
[deliverable/linux.git] / tools / perf / util / python.c
CommitLineData
877108e4
ACM
1#include <Python.h>
2#include <structmember.h>
3#include <inttypes.h>
4#include <poll.h>
1075fbb2 5#include <linux/err.h>
877108e4
ACM
6#include "evlist.h"
7#include "evsel.h"
8#include "event.h"
9#include "cpumap.h"
10#include "thread_map.h"
11
8afb4c01
AH
12/*
13 * Support debug printing even though util/debug.c is not linked. That means
14 * implementing 'verbose' and 'eprintf'.
15 */
16int verbose;
17
c95688aa 18int eprintf(int level, int var, const char *fmt, ...)
8afb4c01
AH
19{
20 va_list args;
21 int ret = 0;
22
c95688aa 23 if (var >= level) {
8afb4c01
AH
24 va_start(args, fmt);
25 ret = vfprintf(stderr, fmt, args);
26 va_end(args);
27 }
28
29 return ret;
30}
31
cfff2d90
FW
32/* Define PyVarObject_HEAD_INIT for python 2.5 */
33#ifndef PyVarObject_HEAD_INIT
34# define PyVarObject_HEAD_INIT(type, size) PyObject_HEAD_INIT(type) size,
35#endif
36
f6bbc1da
ACM
37PyMODINIT_FUNC initperf(void);
38
877108e4
ACM
39#define member_def(type, member, ptype, help) \
40 { #member, ptype, \
41 offsetof(struct pyrf_event, event) + offsetof(struct type, member), \
42 0, help }
43
44#define sample_member_def(name, member, ptype, help) \
45 { #name, ptype, \
46 offsetof(struct pyrf_event, sample) + offsetof(struct perf_sample, member), \
47 0, help }
48
49struct pyrf_event {
50 PyObject_HEAD
377f698d 51 struct perf_evsel *evsel;
877108e4
ACM
52 struct perf_sample sample;
53 union perf_event event;
54};
55
877108e4 56#define sample_members \
f6bbc1da 57 sample_member_def(sample_ip, ip, T_ULONGLONG, "event type"), \
877108e4
ACM
58 sample_member_def(sample_pid, pid, T_INT, "event pid"), \
59 sample_member_def(sample_tid, tid, T_INT, "event tid"), \
f6bbc1da
ACM
60 sample_member_def(sample_time, time, T_ULONGLONG, "event timestamp"), \
61 sample_member_def(sample_addr, addr, T_ULONGLONG, "event addr"), \
62 sample_member_def(sample_id, id, T_ULONGLONG, "event id"), \
63 sample_member_def(sample_stream_id, stream_id, T_ULONGLONG, "event stream id"), \
64 sample_member_def(sample_period, period, T_ULONGLONG, "event period"), \
877108e4
ACM
65 sample_member_def(sample_cpu, cpu, T_UINT, "event cpu"),
66
67static char pyrf_mmap_event__doc[] = PyDoc_STR("perf mmap event object.");
68
69static PyMemberDef pyrf_mmap_event__members[] = {
70 sample_members
71 member_def(perf_event_header, type, T_UINT, "event type"),
ae938802 72 member_def(perf_event_header, misc, T_UINT, "event misc"),
877108e4
ACM
73 member_def(mmap_event, pid, T_UINT, "event pid"),
74 member_def(mmap_event, tid, T_UINT, "event tid"),
f6bbc1da
ACM
75 member_def(mmap_event, start, T_ULONGLONG, "start of the map"),
76 member_def(mmap_event, len, T_ULONGLONG, "map length"),
77 member_def(mmap_event, pgoff, T_ULONGLONG, "page offset"),
877108e4 78 member_def(mmap_event, filename, T_STRING_INPLACE, "backing store"),
f6bbc1da 79 { .name = NULL, },
877108e4
ACM
80};
81
82static PyObject *pyrf_mmap_event__repr(struct pyrf_event *pevent)
83{
84 PyObject *ret;
85 char *s;
86
87 if (asprintf(&s, "{ type: mmap, pid: %u, tid: %u, start: %#" PRIx64 ", "
88 "length: %#" PRIx64 ", offset: %#" PRIx64 ", "
89 "filename: %s }",
90 pevent->event.mmap.pid, pevent->event.mmap.tid,
91 pevent->event.mmap.start, pevent->event.mmap.len,
92 pevent->event.mmap.pgoff, pevent->event.mmap.filename) < 0) {
93 ret = PyErr_NoMemory();
94 } else {
95 ret = PyString_FromString(s);
96 free(s);
97 }
98 return ret;
99}
100
101static PyTypeObject pyrf_mmap_event__type = {
102 PyVarObject_HEAD_INIT(NULL, 0)
103 .tp_name = "perf.mmap_event",
104 .tp_basicsize = sizeof(struct pyrf_event),
105 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
106 .tp_doc = pyrf_mmap_event__doc,
107 .tp_members = pyrf_mmap_event__members,
108 .tp_repr = (reprfunc)pyrf_mmap_event__repr,
109};
110
111static char pyrf_task_event__doc[] = PyDoc_STR("perf task (fork/exit) event object.");
112
113static PyMemberDef pyrf_task_event__members[] = {
114 sample_members
115 member_def(perf_event_header, type, T_UINT, "event type"),
116 member_def(fork_event, pid, T_UINT, "event pid"),
117 member_def(fork_event, ppid, T_UINT, "event ppid"),
118 member_def(fork_event, tid, T_UINT, "event tid"),
119 member_def(fork_event, ptid, T_UINT, "event ptid"),
f6bbc1da
ACM
120 member_def(fork_event, time, T_ULONGLONG, "timestamp"),
121 { .name = NULL, },
877108e4
ACM
122};
123
124static PyObject *pyrf_task_event__repr(struct pyrf_event *pevent)
125{
126 return PyString_FromFormat("{ type: %s, pid: %u, ppid: %u, tid: %u, "
127 "ptid: %u, time: %" PRIu64 "}",
128 pevent->event.header.type == PERF_RECORD_FORK ? "fork" : "exit",
129 pevent->event.fork.pid,
130 pevent->event.fork.ppid,
131 pevent->event.fork.tid,
132 pevent->event.fork.ptid,
133 pevent->event.fork.time);
134}
135
136static PyTypeObject pyrf_task_event__type = {
137 PyVarObject_HEAD_INIT(NULL, 0)
138 .tp_name = "perf.task_event",
139 .tp_basicsize = sizeof(struct pyrf_event),
140 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
141 .tp_doc = pyrf_task_event__doc,
142 .tp_members = pyrf_task_event__members,
143 .tp_repr = (reprfunc)pyrf_task_event__repr,
144};
145
146static char pyrf_comm_event__doc[] = PyDoc_STR("perf comm event object.");
147
148static PyMemberDef pyrf_comm_event__members[] = {
149 sample_members
150 member_def(perf_event_header, type, T_UINT, "event type"),
151 member_def(comm_event, pid, T_UINT, "event pid"),
152 member_def(comm_event, tid, T_UINT, "event tid"),
153 member_def(comm_event, comm, T_STRING_INPLACE, "process name"),
f6bbc1da 154 { .name = NULL, },
877108e4
ACM
155};
156
157static PyObject *pyrf_comm_event__repr(struct pyrf_event *pevent)
158{
159 return PyString_FromFormat("{ type: comm, pid: %u, tid: %u, comm: %s }",
160 pevent->event.comm.pid,
161 pevent->event.comm.tid,
162 pevent->event.comm.comm);
163}
164
165static PyTypeObject pyrf_comm_event__type = {
166 PyVarObject_HEAD_INIT(NULL, 0)
167 .tp_name = "perf.comm_event",
168 .tp_basicsize = sizeof(struct pyrf_event),
169 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
170 .tp_doc = pyrf_comm_event__doc,
171 .tp_members = pyrf_comm_event__members,
172 .tp_repr = (reprfunc)pyrf_comm_event__repr,
173};
174
175static char pyrf_throttle_event__doc[] = PyDoc_STR("perf throttle event object.");
176
177static PyMemberDef pyrf_throttle_event__members[] = {
178 sample_members
179 member_def(perf_event_header, type, T_UINT, "event type"),
f6bbc1da
ACM
180 member_def(throttle_event, time, T_ULONGLONG, "timestamp"),
181 member_def(throttle_event, id, T_ULONGLONG, "event id"),
182 member_def(throttle_event, stream_id, T_ULONGLONG, "event stream id"),
183 { .name = NULL, },
877108e4
ACM
184};
185
186static PyObject *pyrf_throttle_event__repr(struct pyrf_event *pevent)
187{
188 struct throttle_event *te = (struct throttle_event *)(&pevent->event.header + 1);
189
190 return PyString_FromFormat("{ type: %sthrottle, time: %" PRIu64 ", id: %" PRIu64
191 ", stream_id: %" PRIu64 " }",
192 pevent->event.header.type == PERF_RECORD_THROTTLE ? "" : "un",
193 te->time, te->id, te->stream_id);
194}
195
196static PyTypeObject pyrf_throttle_event__type = {
197 PyVarObject_HEAD_INIT(NULL, 0)
198 .tp_name = "perf.throttle_event",
199 .tp_basicsize = sizeof(struct pyrf_event),
200 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
201 .tp_doc = pyrf_throttle_event__doc,
202 .tp_members = pyrf_throttle_event__members,
203 .tp_repr = (reprfunc)pyrf_throttle_event__repr,
204};
205
3e9f45a7
ACM
206static char pyrf_lost_event__doc[] = PyDoc_STR("perf lost event object.");
207
208static PyMemberDef pyrf_lost_event__members[] = {
209 sample_members
210 member_def(lost_event, id, T_ULONGLONG, "event id"),
211 member_def(lost_event, lost, T_ULONGLONG, "number of lost events"),
212 { .name = NULL, },
213};
214
215static PyObject *pyrf_lost_event__repr(struct pyrf_event *pevent)
216{
217 PyObject *ret;
218 char *s;
219
220 if (asprintf(&s, "{ type: lost, id: %#" PRIx64 ", "
221 "lost: %#" PRIx64 " }",
222 pevent->event.lost.id, pevent->event.lost.lost) < 0) {
223 ret = PyErr_NoMemory();
224 } else {
225 ret = PyString_FromString(s);
226 free(s);
227 }
228 return ret;
229}
230
231static PyTypeObject pyrf_lost_event__type = {
232 PyVarObject_HEAD_INIT(NULL, 0)
233 .tp_name = "perf.lost_event",
234 .tp_basicsize = sizeof(struct pyrf_event),
235 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
236 .tp_doc = pyrf_lost_event__doc,
237 .tp_members = pyrf_lost_event__members,
238 .tp_repr = (reprfunc)pyrf_lost_event__repr,
239};
240
241static char pyrf_read_event__doc[] = PyDoc_STR("perf read event object.");
242
243static PyMemberDef pyrf_read_event__members[] = {
244 sample_members
245 member_def(read_event, pid, T_UINT, "event pid"),
246 member_def(read_event, tid, T_UINT, "event tid"),
247 { .name = NULL, },
248};
249
250static PyObject *pyrf_read_event__repr(struct pyrf_event *pevent)
251{
252 return PyString_FromFormat("{ type: read, pid: %u, tid: %u }",
253 pevent->event.read.pid,
254 pevent->event.read.tid);
255 /*
256 * FIXME: return the array of read values,
257 * making this method useful ;-)
258 */
259}
260
261static PyTypeObject pyrf_read_event__type = {
262 PyVarObject_HEAD_INIT(NULL, 0)
263 .tp_name = "perf.read_event",
264 .tp_basicsize = sizeof(struct pyrf_event),
265 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
266 .tp_doc = pyrf_read_event__doc,
267 .tp_members = pyrf_read_event__members,
268 .tp_repr = (reprfunc)pyrf_read_event__repr,
269};
270
271static char pyrf_sample_event__doc[] = PyDoc_STR("perf sample event object.");
272
273static PyMemberDef pyrf_sample_event__members[] = {
274 sample_members
275 member_def(perf_event_header, type, T_UINT, "event type"),
276 { .name = NULL, },
277};
278
279static PyObject *pyrf_sample_event__repr(struct pyrf_event *pevent)
280{
281 PyObject *ret;
282 char *s;
283
284 if (asprintf(&s, "{ type: sample }") < 0) {
285 ret = PyErr_NoMemory();
286 } else {
287 ret = PyString_FromString(s);
288 free(s);
289 }
290 return ret;
291}
292
bae57e38
JO
293static bool is_tracepoint(struct pyrf_event *pevent)
294{
295 return pevent->evsel->attr.type == PERF_TYPE_TRACEPOINT;
296}
297
298static int is_printable_array(char *p, unsigned int len)
299{
300 unsigned int i;
301
302 for (i = 0; i < len; i++) {
303 if (!isprint(p[i]) && !isspace(p[i]))
304 return 0;
305 }
306
307 return 1;
308}
309
310static PyObject*
311tracepoint_field(struct pyrf_event *pe, struct format_field *field)
312{
313 struct pevent *pevent = field->event->pevent;
314 void *data = pe->sample.raw_data;
315 PyObject *ret = NULL;
316 unsigned long long val;
317 unsigned int offset, len;
318
319 if (field->flags & FIELD_IS_ARRAY) {
320 offset = field->offset;
321 len = field->size;
322 if (field->flags & FIELD_IS_DYNAMIC) {
323 val = pevent_read_number(pevent, data + offset, len);
324 offset = val;
325 len = offset >> 16;
326 offset &= 0xffff;
327 }
328 if (field->flags & FIELD_IS_STRING &&
329 is_printable_array(data + offset, len)) {
330 ret = PyString_FromString((char *)data + offset);
331 } else {
332 ret = PyByteArray_FromStringAndSize((const char *) data + offset, len);
333 field->flags &= ~FIELD_IS_STRING;
334 }
335 } else {
336 val = pevent_read_number(pevent, data + field->offset,
337 field->size);
338 if (field->flags & FIELD_IS_POINTER)
339 ret = PyLong_FromUnsignedLong((unsigned long) val);
340 else if (field->flags & FIELD_IS_SIGNED)
341 ret = PyLong_FromLong((long) val);
342 else
343 ret = PyLong_FromUnsignedLong((unsigned long) val);
344 }
345
346 return ret;
347}
348
349static PyObject*
350get_tracepoint_field(struct pyrf_event *pevent, PyObject *attr_name)
351{
352 const char *str = PyString_AsString(PyObject_Str(attr_name));
353 struct perf_evsel *evsel = pevent->evsel;
354 struct format_field *field;
355
356 if (!evsel->tp_format) {
357 struct event_format *tp_format;
358
359 tp_format = trace_event__tp_format_id(evsel->attr.config);
360 if (!tp_format)
361 return NULL;
362
363 evsel->tp_format = tp_format;
364 }
365
366 field = pevent_find_any_field(evsel->tp_format, str);
367 if (!field)
368 return NULL;
369
370 return tracepoint_field(pevent, field);
371}
372
373static PyObject*
374pyrf_sample_event__getattro(struct pyrf_event *pevent, PyObject *attr_name)
375{
376 PyObject *obj = NULL;
377
378 if (is_tracepoint(pevent))
379 obj = get_tracepoint_field(pevent, attr_name);
380
381 return obj ?: PyObject_GenericGetAttr((PyObject *) pevent, attr_name);
382}
383
3e9f45a7
ACM
384static PyTypeObject pyrf_sample_event__type = {
385 PyVarObject_HEAD_INIT(NULL, 0)
386 .tp_name = "perf.sample_event",
387 .tp_basicsize = sizeof(struct pyrf_event),
388 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
389 .tp_doc = pyrf_sample_event__doc,
390 .tp_members = pyrf_sample_event__members,
391 .tp_repr = (reprfunc)pyrf_sample_event__repr,
bae57e38 392 .tp_getattro = (getattrofunc) pyrf_sample_event__getattro,
3e9f45a7
ACM
393};
394
ae938802
ACM
395static char pyrf_context_switch_event__doc[] = PyDoc_STR("perf context_switch event object.");
396
397static PyMemberDef pyrf_context_switch_event__members[] = {
398 sample_members
399 member_def(perf_event_header, type, T_UINT, "event type"),
400 member_def(context_switch_event, next_prev_pid, T_UINT, "next/prev pid"),
401 member_def(context_switch_event, next_prev_tid, T_UINT, "next/prev tid"),
402 { .name = NULL, },
403};
404
405static PyObject *pyrf_context_switch_event__repr(struct pyrf_event *pevent)
406{
407 PyObject *ret;
408 char *s;
409
410 if (asprintf(&s, "{ type: context_switch, next_prev_pid: %u, next_prev_tid: %u, switch_out: %u }",
411 pevent->event.context_switch.next_prev_pid,
412 pevent->event.context_switch.next_prev_tid,
413 !!(pevent->event.header.misc & PERF_RECORD_MISC_SWITCH_OUT)) < 0) {
414 ret = PyErr_NoMemory();
415 } else {
416 ret = PyString_FromString(s);
417 free(s);
418 }
419 return ret;
420}
421
422static PyTypeObject pyrf_context_switch_event__type = {
423 PyVarObject_HEAD_INIT(NULL, 0)
424 .tp_name = "perf.context_switch_event",
425 .tp_basicsize = sizeof(struct pyrf_event),
426 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
427 .tp_doc = pyrf_context_switch_event__doc,
428 .tp_members = pyrf_context_switch_event__members,
429 .tp_repr = (reprfunc)pyrf_context_switch_event__repr,
430};
431
877108e4
ACM
432static int pyrf_event__setup_types(void)
433{
434 int err;
435 pyrf_mmap_event__type.tp_new =
436 pyrf_task_event__type.tp_new =
437 pyrf_comm_event__type.tp_new =
3e9f45a7
ACM
438 pyrf_lost_event__type.tp_new =
439 pyrf_read_event__type.tp_new =
440 pyrf_sample_event__type.tp_new =
ae938802 441 pyrf_context_switch_event__type.tp_new =
877108e4
ACM
442 pyrf_throttle_event__type.tp_new = PyType_GenericNew;
443 err = PyType_Ready(&pyrf_mmap_event__type);
3e9f45a7
ACM
444 if (err < 0)
445 goto out;
446 err = PyType_Ready(&pyrf_lost_event__type);
877108e4
ACM
447 if (err < 0)
448 goto out;
449 err = PyType_Ready(&pyrf_task_event__type);
450 if (err < 0)
451 goto out;
452 err = PyType_Ready(&pyrf_comm_event__type);
453 if (err < 0)
454 goto out;
455 err = PyType_Ready(&pyrf_throttle_event__type);
456 if (err < 0)
457 goto out;
3e9f45a7
ACM
458 err = PyType_Ready(&pyrf_read_event__type);
459 if (err < 0)
460 goto out;
461 err = PyType_Ready(&pyrf_sample_event__type);
462 if (err < 0)
463 goto out;
ae938802
ACM
464 err = PyType_Ready(&pyrf_context_switch_event__type);
465 if (err < 0)
466 goto out;
877108e4
ACM
467out:
468 return err;
469}
470
471static PyTypeObject *pyrf_event__type[] = {
472 [PERF_RECORD_MMAP] = &pyrf_mmap_event__type,
3e9f45a7 473 [PERF_RECORD_LOST] = &pyrf_lost_event__type,
877108e4
ACM
474 [PERF_RECORD_COMM] = &pyrf_comm_event__type,
475 [PERF_RECORD_EXIT] = &pyrf_task_event__type,
476 [PERF_RECORD_THROTTLE] = &pyrf_throttle_event__type,
477 [PERF_RECORD_UNTHROTTLE] = &pyrf_throttle_event__type,
478 [PERF_RECORD_FORK] = &pyrf_task_event__type,
3e9f45a7
ACM
479 [PERF_RECORD_READ] = &pyrf_read_event__type,
480 [PERF_RECORD_SAMPLE] = &pyrf_sample_event__type,
ae938802
ACM
481 [PERF_RECORD_SWITCH] = &pyrf_context_switch_event__type,
482 [PERF_RECORD_SWITCH_CPU_WIDE] = &pyrf_context_switch_event__type,
877108e4
ACM
483};
484
485static PyObject *pyrf_event__new(union perf_event *event)
486{
487 struct pyrf_event *pevent;
488 PyTypeObject *ptype;
489
ae938802
ACM
490 if ((event->header.type < PERF_RECORD_MMAP ||
491 event->header.type > PERF_RECORD_SAMPLE) &&
492 !(event->header.type == PERF_RECORD_SWITCH ||
493 event->header.type == PERF_RECORD_SWITCH_CPU_WIDE))
877108e4
ACM
494 return NULL;
495
496 ptype = pyrf_event__type[event->header.type];
497 pevent = PyObject_New(struct pyrf_event, ptype);
498 if (pevent != NULL)
499 memcpy(&pevent->event, event, event->header.size);
500 return (PyObject *)pevent;
501}
502
503struct pyrf_cpu_map {
504 PyObject_HEAD
505
506 struct cpu_map *cpus;
507};
508
509static int pyrf_cpu_map__init(struct pyrf_cpu_map *pcpus,
510 PyObject *args, PyObject *kwargs)
511{
64348153 512 static char *kwlist[] = { "cpustr", NULL };
877108e4
ACM
513 char *cpustr = NULL;
514
515 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|s",
516 kwlist, &cpustr))
517 return -1;
518
519 pcpus->cpus = cpu_map__new(cpustr);
520 if (pcpus->cpus == NULL)
521 return -1;
522 return 0;
523}
524
525static void pyrf_cpu_map__delete(struct pyrf_cpu_map *pcpus)
526{
f30a79b0 527 cpu_map__put(pcpus->cpus);
877108e4
ACM
528 pcpus->ob_type->tp_free((PyObject*)pcpus);
529}
530
531static Py_ssize_t pyrf_cpu_map__length(PyObject *obj)
532{
533 struct pyrf_cpu_map *pcpus = (void *)obj;
534
535 return pcpus->cpus->nr;
536}
537
538static PyObject *pyrf_cpu_map__item(PyObject *obj, Py_ssize_t i)
539{
540 struct pyrf_cpu_map *pcpus = (void *)obj;
541
542 if (i >= pcpus->cpus->nr)
543 return NULL;
544
545 return Py_BuildValue("i", pcpus->cpus->map[i]);
546}
547
548static PySequenceMethods pyrf_cpu_map__sequence_methods = {
549 .sq_length = pyrf_cpu_map__length,
550 .sq_item = pyrf_cpu_map__item,
551};
552
553static char pyrf_cpu_map__doc[] = PyDoc_STR("cpu map object.");
554
555static PyTypeObject pyrf_cpu_map__type = {
556 PyVarObject_HEAD_INIT(NULL, 0)
557 .tp_name = "perf.cpu_map",
558 .tp_basicsize = sizeof(struct pyrf_cpu_map),
559 .tp_dealloc = (destructor)pyrf_cpu_map__delete,
560 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
561 .tp_doc = pyrf_cpu_map__doc,
562 .tp_as_sequence = &pyrf_cpu_map__sequence_methods,
563 .tp_init = (initproc)pyrf_cpu_map__init,
564};
565
566static int pyrf_cpu_map__setup_types(void)
567{
568 pyrf_cpu_map__type.tp_new = PyType_GenericNew;
569 return PyType_Ready(&pyrf_cpu_map__type);
570}
571
572struct pyrf_thread_map {
573 PyObject_HEAD
574
575 struct thread_map *threads;
576};
577
578static int pyrf_thread_map__init(struct pyrf_thread_map *pthreads,
579 PyObject *args, PyObject *kwargs)
580{
0d37aa34
ACM
581 static char *kwlist[] = { "pid", "tid", "uid", NULL };
582 int pid = -1, tid = -1, uid = UINT_MAX;
877108e4 583
0d37aa34
ACM
584 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|iii",
585 kwlist, &pid, &tid, &uid))
877108e4
ACM
586 return -1;
587
0d37aa34 588 pthreads->threads = thread_map__new(pid, tid, uid);
877108e4
ACM
589 if (pthreads->threads == NULL)
590 return -1;
591 return 0;
592}
593
594static void pyrf_thread_map__delete(struct pyrf_thread_map *pthreads)
595{
186fbb74 596 thread_map__put(pthreads->threads);
877108e4
ACM
597 pthreads->ob_type->tp_free((PyObject*)pthreads);
598}
599
600static Py_ssize_t pyrf_thread_map__length(PyObject *obj)
601{
602 struct pyrf_thread_map *pthreads = (void *)obj;
603
604 return pthreads->threads->nr;
605}
606
607static PyObject *pyrf_thread_map__item(PyObject *obj, Py_ssize_t i)
608{
609 struct pyrf_thread_map *pthreads = (void *)obj;
610
611 if (i >= pthreads->threads->nr)
612 return NULL;
613
614 return Py_BuildValue("i", pthreads->threads->map[i]);
615}
616
617static PySequenceMethods pyrf_thread_map__sequence_methods = {
618 .sq_length = pyrf_thread_map__length,
619 .sq_item = pyrf_thread_map__item,
620};
621
622static char pyrf_thread_map__doc[] = PyDoc_STR("thread map object.");
623
624static PyTypeObject pyrf_thread_map__type = {
625 PyVarObject_HEAD_INIT(NULL, 0)
626 .tp_name = "perf.thread_map",
627 .tp_basicsize = sizeof(struct pyrf_thread_map),
628 .tp_dealloc = (destructor)pyrf_thread_map__delete,
629 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
630 .tp_doc = pyrf_thread_map__doc,
631 .tp_as_sequence = &pyrf_thread_map__sequence_methods,
632 .tp_init = (initproc)pyrf_thread_map__init,
633};
634
635static int pyrf_thread_map__setup_types(void)
636{
637 pyrf_thread_map__type.tp_new = PyType_GenericNew;
638 return PyType_Ready(&pyrf_thread_map__type);
639}
640
641struct pyrf_evsel {
642 PyObject_HEAD
643
644 struct perf_evsel evsel;
645};
646
647static int pyrf_evsel__init(struct pyrf_evsel *pevsel,
648 PyObject *args, PyObject *kwargs)
649{
650 struct perf_event_attr attr = {
651 .type = PERF_TYPE_HARDWARE,
652 .config = PERF_COUNT_HW_CPU_CYCLES,
653 .sample_type = PERF_SAMPLE_PERIOD | PERF_SAMPLE_TID,
654 };
655 static char *kwlist[] = {
656 "type",
657 "config",
658 "sample_freq",
659 "sample_period",
660 "sample_type",
661 "read_format",
662 "disabled",
663 "inherit",
664 "pinned",
665 "exclusive",
666 "exclude_user",
667 "exclude_kernel",
668 "exclude_hv",
669 "exclude_idle",
670 "mmap",
ae938802 671 "context_switch",
877108e4
ACM
672 "comm",
673 "freq",
674 "inherit_stat",
675 "enable_on_exec",
676 "task",
677 "watermark",
678 "precise_ip",
679 "mmap_data",
680 "sample_id_all",
681 "wakeup_events",
682 "bp_type",
683 "bp_addr",
64348153
FW
684 "bp_len",
685 NULL
686 };
877108e4
ACM
687 u64 sample_period = 0;
688 u32 disabled = 0,
689 inherit = 0,
690 pinned = 0,
691 exclusive = 0,
692 exclude_user = 0,
693 exclude_kernel = 0,
694 exclude_hv = 0,
695 exclude_idle = 0,
696 mmap = 0,
ae938802 697 context_switch = 0,
877108e4
ACM
698 comm = 0,
699 freq = 1,
700 inherit_stat = 0,
701 enable_on_exec = 0,
702 task = 0,
703 watermark = 0,
704 precise_ip = 0,
705 mmap_data = 0,
706 sample_id_all = 1;
707 int idx = 0;
708
709 if (!PyArg_ParseTupleAndKeywords(args, kwargs,
ae938802 710 "|iKiKKiiiiiiiiiiiiiiiiiiiiiiKK", kwlist,
877108e4
ACM
711 &attr.type, &attr.config, &attr.sample_freq,
712 &sample_period, &attr.sample_type,
713 &attr.read_format, &disabled, &inherit,
714 &pinned, &exclusive, &exclude_user,
715 &exclude_kernel, &exclude_hv, &exclude_idle,
ae938802 716 &mmap, &context_switch, &comm, &freq, &inherit_stat,
877108e4
ACM
717 &enable_on_exec, &task, &watermark,
718 &precise_ip, &mmap_data, &sample_id_all,
719 &attr.wakeup_events, &attr.bp_type,
720 &attr.bp_addr, &attr.bp_len, &idx))
721 return -1;
722
723 /* union... */
724 if (sample_period != 0) {
725 if (attr.sample_freq != 0)
726 return -1; /* FIXME: throw right exception */
727 attr.sample_period = sample_period;
728 }
729
730 /* Bitfields */
731 attr.disabled = disabled;
732 attr.inherit = inherit;
733 attr.pinned = pinned;
734 attr.exclusive = exclusive;
735 attr.exclude_user = exclude_user;
736 attr.exclude_kernel = exclude_kernel;
737 attr.exclude_hv = exclude_hv;
738 attr.exclude_idle = exclude_idle;
739 attr.mmap = mmap;
ae938802 740 attr.context_switch = context_switch;
877108e4
ACM
741 attr.comm = comm;
742 attr.freq = freq;
743 attr.inherit_stat = inherit_stat;
744 attr.enable_on_exec = enable_on_exec;
745 attr.task = task;
746 attr.watermark = watermark;
747 attr.precise_ip = precise_ip;
748 attr.mmap_data = mmap_data;
749 attr.sample_id_all = sample_id_all;
ad4e3c04 750 attr.size = sizeof(attr);
877108e4
ACM
751
752 perf_evsel__init(&pevsel->evsel, &attr, idx);
753 return 0;
754}
755
756static void pyrf_evsel__delete(struct pyrf_evsel *pevsel)
757{
758 perf_evsel__exit(&pevsel->evsel);
759 pevsel->ob_type->tp_free((PyObject*)pevsel);
760}
761
762static PyObject *pyrf_evsel__open(struct pyrf_evsel *pevsel,
763 PyObject *args, PyObject *kwargs)
764{
765 struct perf_evsel *evsel = &pevsel->evsel;
766 struct cpu_map *cpus = NULL;
767 struct thread_map *threads = NULL;
768 PyObject *pcpus = NULL, *pthreads = NULL;
5d2cd909 769 int group = 0, inherit = 0;
64348153 770 static char *kwlist[] = { "cpus", "threads", "group", "inherit", NULL };
877108e4
ACM
771
772 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|OOii", kwlist,
5d2cd909 773 &pcpus, &pthreads, &group, &inherit))
877108e4
ACM
774 return NULL;
775
776 if (pthreads != NULL)
777 threads = ((struct pyrf_thread_map *)pthreads)->threads;
778
779 if (pcpus != NULL)
780 cpus = ((struct pyrf_cpu_map *)pcpus)->cpus;
781
5d2cd909 782 evsel->attr.inherit = inherit;
727ab04e
ACM
783 /*
784 * This will group just the fds for this single evsel, to group
785 * multiple events, use evlist.open().
786 */
6a4bb04c 787 if (perf_evsel__open(evsel, cpus, threads) < 0) {
877108e4
ACM
788 PyErr_SetFromErrno(PyExc_OSError);
789 return NULL;
790 }
791
792 Py_INCREF(Py_None);
793 return Py_None;
794}
795
796static PyMethodDef pyrf_evsel__methods[] = {
797 {
798 .ml_name = "open",
799 .ml_meth = (PyCFunction)pyrf_evsel__open,
800 .ml_flags = METH_VARARGS | METH_KEYWORDS,
801 .ml_doc = PyDoc_STR("open the event selector file descriptor table.")
802 },
f6bbc1da 803 { .ml_name = NULL, }
877108e4
ACM
804};
805
806static char pyrf_evsel__doc[] = PyDoc_STR("perf event selector list object.");
807
808static PyTypeObject pyrf_evsel__type = {
809 PyVarObject_HEAD_INIT(NULL, 0)
810 .tp_name = "perf.evsel",
811 .tp_basicsize = sizeof(struct pyrf_evsel),
812 .tp_dealloc = (destructor)pyrf_evsel__delete,
813 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
814 .tp_doc = pyrf_evsel__doc,
815 .tp_methods = pyrf_evsel__methods,
816 .tp_init = (initproc)pyrf_evsel__init,
817};
818
819static int pyrf_evsel__setup_types(void)
820{
821 pyrf_evsel__type.tp_new = PyType_GenericNew;
822 return PyType_Ready(&pyrf_evsel__type);
823}
824
825struct pyrf_evlist {
826 PyObject_HEAD
827
828 struct perf_evlist evlist;
829};
830
831static int pyrf_evlist__init(struct pyrf_evlist *pevlist,
1d037ca1 832 PyObject *args, PyObject *kwargs __maybe_unused)
877108e4 833{
7e2ed097
ACM
834 PyObject *pcpus = NULL, *pthreads = NULL;
835 struct cpu_map *cpus;
836 struct thread_map *threads;
837
838 if (!PyArg_ParseTuple(args, "OO", &pcpus, &pthreads))
839 return -1;
840
841 threads = ((struct pyrf_thread_map *)pthreads)->threads;
842 cpus = ((struct pyrf_cpu_map *)pcpus)->cpus;
843 perf_evlist__init(&pevlist->evlist, cpus, threads);
877108e4
ACM
844 return 0;
845}
846
847static void pyrf_evlist__delete(struct pyrf_evlist *pevlist)
848{
849 perf_evlist__exit(&pevlist->evlist);
850 pevlist->ob_type->tp_free((PyObject*)pevlist);
851}
852
853static PyObject *pyrf_evlist__mmap(struct pyrf_evlist *pevlist,
854 PyObject *args, PyObject *kwargs)
855{
856 struct perf_evlist *evlist = &pevlist->evlist;
64348153 857 static char *kwlist[] = { "pages", "overwrite", NULL };
877108e4
ACM
858 int pages = 128, overwrite = false;
859
7e2ed097
ACM
860 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ii", kwlist,
861 &pages, &overwrite))
877108e4
ACM
862 return NULL;
863
7e2ed097 864 if (perf_evlist__mmap(evlist, pages, overwrite) < 0) {
877108e4
ACM
865 PyErr_SetFromErrno(PyExc_OSError);
866 return NULL;
867 }
868
869 Py_INCREF(Py_None);
870 return Py_None;
871}
872
873static PyObject *pyrf_evlist__poll(struct pyrf_evlist *pevlist,
874 PyObject *args, PyObject *kwargs)
875{
876 struct perf_evlist *evlist = &pevlist->evlist;
64348153 877 static char *kwlist[] = { "timeout", NULL };
877108e4
ACM
878 int timeout = -1, n;
879
880 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|i", kwlist, &timeout))
881 return NULL;
882
f66a889d 883 n = perf_evlist__poll(evlist, timeout);
877108e4
ACM
884 if (n < 0) {
885 PyErr_SetFromErrno(PyExc_OSError);
886 return NULL;
887 }
888
889 return Py_BuildValue("i", n);
890}
891
892static PyObject *pyrf_evlist__get_pollfd(struct pyrf_evlist *pevlist,
1d037ca1
IT
893 PyObject *args __maybe_unused,
894 PyObject *kwargs __maybe_unused)
877108e4
ACM
895{
896 struct perf_evlist *evlist = &pevlist->evlist;
897 PyObject *list = PyList_New(0);
898 int i;
899
1b85337d 900 for (i = 0; i < evlist->pollfd.nr; ++i) {
877108e4 901 PyObject *file;
1b85337d 902 FILE *fp = fdopen(evlist->pollfd.entries[i].fd, "r");
877108e4
ACM
903
904 if (fp == NULL)
905 goto free_list;
906
907 file = PyFile_FromFile(fp, "perf", "r", NULL);
908 if (file == NULL)
909 goto free_list;
910
911 if (PyList_Append(list, file) != 0) {
912 Py_DECREF(file);
913 goto free_list;
914 }
48000a1a 915
877108e4
ACM
916 Py_DECREF(file);
917 }
918
919 return list;
920free_list:
921 return PyErr_NoMemory();
922}
923
924
925static PyObject *pyrf_evlist__add(struct pyrf_evlist *pevlist,
1d037ca1
IT
926 PyObject *args,
927 PyObject *kwargs __maybe_unused)
877108e4
ACM
928{
929 struct perf_evlist *evlist = &pevlist->evlist;
930 PyObject *pevsel;
931 struct perf_evsel *evsel;
932
933 if (!PyArg_ParseTuple(args, "O", &pevsel))
934 return NULL;
935
936 Py_INCREF(pevsel);
937 evsel = &((struct pyrf_evsel *)pevsel)->evsel;
938 evsel->idx = evlist->nr_entries;
939 perf_evlist__add(evlist, evsel);
940
941 return Py_BuildValue("i", evlist->nr_entries);
942}
943
944static PyObject *pyrf_evlist__read_on_cpu(struct pyrf_evlist *pevlist,
945 PyObject *args, PyObject *kwargs)
946{
947 struct perf_evlist *evlist = &pevlist->evlist;
948 union perf_event *event;
949 int sample_id_all = 1, cpu;
64348153 950 static char *kwlist[] = { "cpu", "sample_id_all", NULL };
5538beca 951 int err;
877108e4
ACM
952
953 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "i|i", kwlist,
954 &cpu, &sample_id_all))
955 return NULL;
956
aece948f 957 event = perf_evlist__mmap_read(evlist, cpu);
877108e4 958 if (event != NULL) {
877108e4
ACM
959 PyObject *pyevent = pyrf_event__new(event);
960 struct pyrf_event *pevent = (struct pyrf_event *)pyevent;
377f698d 961 struct perf_evsel *evsel;
877108e4
ACM
962
963 if (pyevent == NULL)
964 return PyErr_NoMemory();
965
377f698d
JO
966 evsel = perf_evlist__event2evsel(evlist, event);
967 if (!evsel)
968 return Py_None;
969
970 pevent->evsel = evsel;
971
972 err = perf_evsel__parse_sample(evsel, event, &pevent->sample);
e8968e65
JO
973
974 /* Consume the even only after we parsed it out. */
975 perf_evlist__mmap_consume(evlist, cpu);
976
5c6970af
ACM
977 if (err)
978 return PyErr_Format(PyExc_OSError,
979 "perf: can't parse sample, err=%d", err);
877108e4
ACM
980 return pyevent;
981 }
5c6970af 982
877108e4
ACM
983 Py_INCREF(Py_None);
984 return Py_None;
985}
986
727ab04e
ACM
987static PyObject *pyrf_evlist__open(struct pyrf_evlist *pevlist,
988 PyObject *args, PyObject *kwargs)
989{
990 struct perf_evlist *evlist = &pevlist->evlist;
991 int group = 0;
992 static char *kwlist[] = { "group", NULL };
993
994 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|OOii", kwlist, &group))
995 return NULL;
996
6a4bb04c 997 if (group)
63dab225 998 perf_evlist__set_leader(evlist);
6a4bb04c
JO
999
1000 if (perf_evlist__open(evlist) < 0) {
727ab04e
ACM
1001 PyErr_SetFromErrno(PyExc_OSError);
1002 return NULL;
1003 }
1004
1005 Py_INCREF(Py_None);
1006 return Py_None;
1007}
1008
877108e4
ACM
1009static PyMethodDef pyrf_evlist__methods[] = {
1010 {
1011 .ml_name = "mmap",
1012 .ml_meth = (PyCFunction)pyrf_evlist__mmap,
1013 .ml_flags = METH_VARARGS | METH_KEYWORDS,
1014 .ml_doc = PyDoc_STR("mmap the file descriptor table.")
1015 },
727ab04e
ACM
1016 {
1017 .ml_name = "open",
1018 .ml_meth = (PyCFunction)pyrf_evlist__open,
1019 .ml_flags = METH_VARARGS | METH_KEYWORDS,
1020 .ml_doc = PyDoc_STR("open the file descriptors.")
1021 },
877108e4
ACM
1022 {
1023 .ml_name = "poll",
1024 .ml_meth = (PyCFunction)pyrf_evlist__poll,
1025 .ml_flags = METH_VARARGS | METH_KEYWORDS,
1026 .ml_doc = PyDoc_STR("poll the file descriptor table.")
1027 },
1028 {
1029 .ml_name = "get_pollfd",
1030 .ml_meth = (PyCFunction)pyrf_evlist__get_pollfd,
1031 .ml_flags = METH_VARARGS | METH_KEYWORDS,
1032 .ml_doc = PyDoc_STR("get the poll file descriptor table.")
1033 },
1034 {
1035 .ml_name = "add",
1036 .ml_meth = (PyCFunction)pyrf_evlist__add,
1037 .ml_flags = METH_VARARGS | METH_KEYWORDS,
1038 .ml_doc = PyDoc_STR("adds an event selector to the list.")
1039 },
1040 {
1041 .ml_name = "read_on_cpu",
1042 .ml_meth = (PyCFunction)pyrf_evlist__read_on_cpu,
1043 .ml_flags = METH_VARARGS | METH_KEYWORDS,
1044 .ml_doc = PyDoc_STR("reads an event.")
1045 },
f6bbc1da 1046 { .ml_name = NULL, }
877108e4
ACM
1047};
1048
1049static Py_ssize_t pyrf_evlist__length(PyObject *obj)
1050{
1051 struct pyrf_evlist *pevlist = (void *)obj;
1052
1053 return pevlist->evlist.nr_entries;
1054}
1055
1056static PyObject *pyrf_evlist__item(PyObject *obj, Py_ssize_t i)
1057{
1058 struct pyrf_evlist *pevlist = (void *)obj;
1059 struct perf_evsel *pos;
1060
1061 if (i >= pevlist->evlist.nr_entries)
1062 return NULL;
1063
e5cadb93 1064 evlist__for_each_entry(&pevlist->evlist, pos) {
877108e4
ACM
1065 if (i-- == 0)
1066 break;
0050f7aa 1067 }
877108e4
ACM
1068
1069 return Py_BuildValue("O", container_of(pos, struct pyrf_evsel, evsel));
1070}
1071
1072static PySequenceMethods pyrf_evlist__sequence_methods = {
1073 .sq_length = pyrf_evlist__length,
1074 .sq_item = pyrf_evlist__item,
1075};
1076
1077static char pyrf_evlist__doc[] = PyDoc_STR("perf event selector list object.");
1078
1079static PyTypeObject pyrf_evlist__type = {
1080 PyVarObject_HEAD_INIT(NULL, 0)
1081 .tp_name = "perf.evlist",
1082 .tp_basicsize = sizeof(struct pyrf_evlist),
1083 .tp_dealloc = (destructor)pyrf_evlist__delete,
1084 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
1085 .tp_as_sequence = &pyrf_evlist__sequence_methods,
1086 .tp_doc = pyrf_evlist__doc,
1087 .tp_methods = pyrf_evlist__methods,
1088 .tp_init = (initproc)pyrf_evlist__init,
1089};
1090
1091static int pyrf_evlist__setup_types(void)
1092{
1093 pyrf_evlist__type.tp_new = PyType_GenericNew;
1094 return PyType_Ready(&pyrf_evlist__type);
1095}
1096
5865fe36
ACM
1097#define PERF_CONST(name) { #name, PERF_##name }
1098
877108e4
ACM
1099static struct {
1100 const char *name;
1101 int value;
1102} perf__constants[] = {
5865fe36
ACM
1103 PERF_CONST(TYPE_HARDWARE),
1104 PERF_CONST(TYPE_SOFTWARE),
1105 PERF_CONST(TYPE_TRACEPOINT),
1106 PERF_CONST(TYPE_HW_CACHE),
1107 PERF_CONST(TYPE_RAW),
1108 PERF_CONST(TYPE_BREAKPOINT),
1109
1110 PERF_CONST(COUNT_HW_CPU_CYCLES),
1111 PERF_CONST(COUNT_HW_INSTRUCTIONS),
1112 PERF_CONST(COUNT_HW_CACHE_REFERENCES),
1113 PERF_CONST(COUNT_HW_CACHE_MISSES),
1114 PERF_CONST(COUNT_HW_BRANCH_INSTRUCTIONS),
1115 PERF_CONST(COUNT_HW_BRANCH_MISSES),
1116 PERF_CONST(COUNT_HW_BUS_CYCLES),
1117 PERF_CONST(COUNT_HW_CACHE_L1D),
1118 PERF_CONST(COUNT_HW_CACHE_L1I),
1119 PERF_CONST(COUNT_HW_CACHE_LL),
1120 PERF_CONST(COUNT_HW_CACHE_DTLB),
1121 PERF_CONST(COUNT_HW_CACHE_ITLB),
1122 PERF_CONST(COUNT_HW_CACHE_BPU),
1123 PERF_CONST(COUNT_HW_CACHE_OP_READ),
1124 PERF_CONST(COUNT_HW_CACHE_OP_WRITE),
1125 PERF_CONST(COUNT_HW_CACHE_OP_PREFETCH),
1126 PERF_CONST(COUNT_HW_CACHE_RESULT_ACCESS),
1127 PERF_CONST(COUNT_HW_CACHE_RESULT_MISS),
1128
1129 PERF_CONST(COUNT_HW_STALLED_CYCLES_FRONTEND),
1130 PERF_CONST(COUNT_HW_STALLED_CYCLES_BACKEND),
1131
1132 PERF_CONST(COUNT_SW_CPU_CLOCK),
1133 PERF_CONST(COUNT_SW_TASK_CLOCK),
1134 PERF_CONST(COUNT_SW_PAGE_FAULTS),
1135 PERF_CONST(COUNT_SW_CONTEXT_SWITCHES),
1136 PERF_CONST(COUNT_SW_CPU_MIGRATIONS),
1137 PERF_CONST(COUNT_SW_PAGE_FAULTS_MIN),
1138 PERF_CONST(COUNT_SW_PAGE_FAULTS_MAJ),
1139 PERF_CONST(COUNT_SW_ALIGNMENT_FAULTS),
1140 PERF_CONST(COUNT_SW_EMULATION_FAULTS),
1141 PERF_CONST(COUNT_SW_DUMMY),
1142
1143 PERF_CONST(SAMPLE_IP),
1144 PERF_CONST(SAMPLE_TID),
1145 PERF_CONST(SAMPLE_TIME),
1146 PERF_CONST(SAMPLE_ADDR),
1147 PERF_CONST(SAMPLE_READ),
1148 PERF_CONST(SAMPLE_CALLCHAIN),
1149 PERF_CONST(SAMPLE_ID),
1150 PERF_CONST(SAMPLE_CPU),
1151 PERF_CONST(SAMPLE_PERIOD),
1152 PERF_CONST(SAMPLE_STREAM_ID),
1153 PERF_CONST(SAMPLE_RAW),
1154
1155 PERF_CONST(FORMAT_TOTAL_TIME_ENABLED),
1156 PERF_CONST(FORMAT_TOTAL_TIME_RUNNING),
1157 PERF_CONST(FORMAT_ID),
1158 PERF_CONST(FORMAT_GROUP),
1159
1160 PERF_CONST(RECORD_MMAP),
1161 PERF_CONST(RECORD_LOST),
1162 PERF_CONST(RECORD_COMM),
1163 PERF_CONST(RECORD_EXIT),
1164 PERF_CONST(RECORD_THROTTLE),
1165 PERF_CONST(RECORD_UNTHROTTLE),
1166 PERF_CONST(RECORD_FORK),
1167 PERF_CONST(RECORD_READ),
1168 PERF_CONST(RECORD_SAMPLE),
84576da2
ACM
1169 PERF_CONST(RECORD_MMAP2),
1170 PERF_CONST(RECORD_AUX),
1171 PERF_CONST(RECORD_ITRACE_START),
1172 PERF_CONST(RECORD_LOST_SAMPLES),
1173 PERF_CONST(RECORD_SWITCH),
1174 PERF_CONST(RECORD_SWITCH_CPU_WIDE),
ae938802
ACM
1175
1176 PERF_CONST(RECORD_MISC_SWITCH_OUT),
f6bbc1da 1177 { .name = NULL, },
877108e4
ACM
1178};
1179
1075fbb2
JO
1180static PyObject *pyrf__tracepoint(struct pyrf_evsel *pevsel,
1181 PyObject *args, PyObject *kwargs)
1182{
1183 struct event_format *tp_format;
1184 static char *kwlist[] = { "sys", "name", NULL };
1185 char *sys = NULL;
1186 char *name = NULL;
1187
1188 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ss", kwlist,
1189 &sys, &name))
1190 return NULL;
1191
1192 tp_format = trace_event__tp_format(sys, name);
1193 if (IS_ERR(tp_format))
1194 return PyInt_FromLong(-1);
1195
1196 return PyInt_FromLong(tp_format->id);
1197}
1198
877108e4 1199static PyMethodDef perf__methods[] = {
1075fbb2
JO
1200 {
1201 .ml_name = "tracepoint",
1202 .ml_meth = (PyCFunction) pyrf__tracepoint,
1203 .ml_flags = METH_VARARGS | METH_KEYWORDS,
1204 .ml_doc = PyDoc_STR("Get tracepoint config.")
1205 },
f6bbc1da 1206 { .ml_name = NULL, }
877108e4
ACM
1207};
1208
1209PyMODINIT_FUNC initperf(void)
1210{
1211 PyObject *obj;
1212 int i;
1213 PyObject *dict, *module = Py_InitModule("perf", perf__methods);
1214
1215 if (module == NULL ||
1216 pyrf_event__setup_types() < 0 ||
1217 pyrf_evlist__setup_types() < 0 ||
1218 pyrf_evsel__setup_types() < 0 ||
1219 pyrf_thread_map__setup_types() < 0 ||
1220 pyrf_cpu_map__setup_types() < 0)
1221 return;
1222
918512b4 1223 /* The page_size is placed in util object. */
0da2e9c2
ACM
1224 page_size = sysconf(_SC_PAGE_SIZE);
1225
877108e4
ACM
1226 Py_INCREF(&pyrf_evlist__type);
1227 PyModule_AddObject(module, "evlist", (PyObject*)&pyrf_evlist__type);
1228
1229 Py_INCREF(&pyrf_evsel__type);
1230 PyModule_AddObject(module, "evsel", (PyObject*)&pyrf_evsel__type);
1231
85e37de3
JO
1232 Py_INCREF(&pyrf_mmap_event__type);
1233 PyModule_AddObject(module, "mmap_event", (PyObject *)&pyrf_mmap_event__type);
1234
1235 Py_INCREF(&pyrf_lost_event__type);
1236 PyModule_AddObject(module, "lost_event", (PyObject *)&pyrf_lost_event__type);
1237
1238 Py_INCREF(&pyrf_comm_event__type);
1239 PyModule_AddObject(module, "comm_event", (PyObject *)&pyrf_comm_event__type);
1240
1241 Py_INCREF(&pyrf_task_event__type);
1242 PyModule_AddObject(module, "task_event", (PyObject *)&pyrf_task_event__type);
1243
1244 Py_INCREF(&pyrf_throttle_event__type);
1245 PyModule_AddObject(module, "throttle_event", (PyObject *)&pyrf_throttle_event__type);
1246
1247 Py_INCREF(&pyrf_task_event__type);
1248 PyModule_AddObject(module, "task_event", (PyObject *)&pyrf_task_event__type);
1249
1250 Py_INCREF(&pyrf_read_event__type);
1251 PyModule_AddObject(module, "read_event", (PyObject *)&pyrf_read_event__type);
1252
1253 Py_INCREF(&pyrf_sample_event__type);
1254 PyModule_AddObject(module, "sample_event", (PyObject *)&pyrf_sample_event__type);
1255
1256 Py_INCREF(&pyrf_context_switch_event__type);
1257 PyModule_AddObject(module, "switch_event", (PyObject *)&pyrf_context_switch_event__type);
1258
877108e4
ACM
1259 Py_INCREF(&pyrf_thread_map__type);
1260 PyModule_AddObject(module, "thread_map", (PyObject*)&pyrf_thread_map__type);
1261
1262 Py_INCREF(&pyrf_cpu_map__type);
1263 PyModule_AddObject(module, "cpu_map", (PyObject*)&pyrf_cpu_map__type);
1264
1265 dict = PyModule_GetDict(module);
1266 if (dict == NULL)
1267 goto error;
1268
1269 for (i = 0; perf__constants[i].name != NULL; i++) {
1270 obj = PyInt_FromLong(perf__constants[i].value);
1271 if (obj == NULL)
1272 goto error;
1273 PyDict_SetItemString(dict, perf__constants[i].name, obj);
1274 Py_DECREF(obj);
1275 }
1276
1277error:
1278 if (PyErr_Occurred())
1279 PyErr_SetString(PyExc_ImportError, "perf: Init failed!");
1280}
0c6332e9
ACM
1281
1282/*
1283 * Dummy, to avoid dragging all the test_attr infrastructure in the python
1284 * binding.
1285 */
1286void test_attr__open(struct perf_event_attr *attr, pid_t pid, int cpu,
1287 int fd, int group_fd, unsigned long flags)
1288{
1289}
This page took 0.28441 seconds and 5 git commands to generate.