kvm tools: Add ivshmem device
[linux-2.6/next.git] / tools / perf / util / python.c
bloba9ac0504aabd25f3dc4a528f5e86d291eed4da05
1 #include <Python.h>
2 #include <structmember.h>
3 #include <inttypes.h>
4 #include <poll.h>
5 #include "evlist.h"
6 #include "evsel.h"
7 #include "event.h"
8 #include "cpumap.h"
9 #include "thread_map.h"
11 /* Define PyVarObject_HEAD_INIT for python 2.5 */
12 #ifndef PyVarObject_HEAD_INIT
13 # define PyVarObject_HEAD_INIT(type, size) PyObject_HEAD_INIT(type) size,
14 #endif
16 struct throttle_event {
17 struct perf_event_header header;
18 u64 time;
19 u64 id;
20 u64 stream_id;
23 PyMODINIT_FUNC initperf(void);
25 #define member_def(type, member, ptype, help) \
26 { #member, ptype, \
27 offsetof(struct pyrf_event, event) + offsetof(struct type, member), \
28 0, help }
30 #define sample_member_def(name, member, ptype, help) \
31 { #name, ptype, \
32 offsetof(struct pyrf_event, sample) + offsetof(struct perf_sample, member), \
33 0, help }
35 struct pyrf_event {
36 PyObject_HEAD
37 struct perf_sample sample;
38 union perf_event event;
41 #define sample_members \
42 sample_member_def(sample_ip, ip, T_ULONGLONG, "event type"), \
43 sample_member_def(sample_pid, pid, T_INT, "event pid"), \
44 sample_member_def(sample_tid, tid, T_INT, "event tid"), \
45 sample_member_def(sample_time, time, T_ULONGLONG, "event timestamp"), \
46 sample_member_def(sample_addr, addr, T_ULONGLONG, "event addr"), \
47 sample_member_def(sample_id, id, T_ULONGLONG, "event id"), \
48 sample_member_def(sample_stream_id, stream_id, T_ULONGLONG, "event stream id"), \
49 sample_member_def(sample_period, period, T_ULONGLONG, "event period"), \
50 sample_member_def(sample_cpu, cpu, T_UINT, "event cpu"),
52 static char pyrf_mmap_event__doc[] = PyDoc_STR("perf mmap event object.");
54 static PyMemberDef pyrf_mmap_event__members[] = {
55 sample_members
56 member_def(perf_event_header, type, T_UINT, "event type"),
57 member_def(mmap_event, pid, T_UINT, "event pid"),
58 member_def(mmap_event, tid, T_UINT, "event tid"),
59 member_def(mmap_event, start, T_ULONGLONG, "start of the map"),
60 member_def(mmap_event, len, T_ULONGLONG, "map length"),
61 member_def(mmap_event, pgoff, T_ULONGLONG, "page offset"),
62 member_def(mmap_event, filename, T_STRING_INPLACE, "backing store"),
63 { .name = NULL, },
66 static PyObject *pyrf_mmap_event__repr(struct pyrf_event *pevent)
68 PyObject *ret;
69 char *s;
71 if (asprintf(&s, "{ type: mmap, pid: %u, tid: %u, start: %#" PRIx64 ", "
72 "length: %#" PRIx64 ", offset: %#" PRIx64 ", "
73 "filename: %s }",
74 pevent->event.mmap.pid, pevent->event.mmap.tid,
75 pevent->event.mmap.start, pevent->event.mmap.len,
76 pevent->event.mmap.pgoff, pevent->event.mmap.filename) < 0) {
77 ret = PyErr_NoMemory();
78 } else {
79 ret = PyString_FromString(s);
80 free(s);
82 return ret;
85 static PyTypeObject pyrf_mmap_event__type = {
86 PyVarObject_HEAD_INIT(NULL, 0)
87 .tp_name = "perf.mmap_event",
88 .tp_basicsize = sizeof(struct pyrf_event),
89 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
90 .tp_doc = pyrf_mmap_event__doc,
91 .tp_members = pyrf_mmap_event__members,
92 .tp_repr = (reprfunc)pyrf_mmap_event__repr,
95 static char pyrf_task_event__doc[] = PyDoc_STR("perf task (fork/exit) event object.");
97 static PyMemberDef pyrf_task_event__members[] = {
98 sample_members
99 member_def(perf_event_header, type, T_UINT, "event type"),
100 member_def(fork_event, pid, T_UINT, "event pid"),
101 member_def(fork_event, ppid, T_UINT, "event ppid"),
102 member_def(fork_event, tid, T_UINT, "event tid"),
103 member_def(fork_event, ptid, T_UINT, "event ptid"),
104 member_def(fork_event, time, T_ULONGLONG, "timestamp"),
105 { .name = NULL, },
108 static PyObject *pyrf_task_event__repr(struct pyrf_event *pevent)
110 return PyString_FromFormat("{ type: %s, pid: %u, ppid: %u, tid: %u, "
111 "ptid: %u, time: %" PRIu64 "}",
112 pevent->event.header.type == PERF_RECORD_FORK ? "fork" : "exit",
113 pevent->event.fork.pid,
114 pevent->event.fork.ppid,
115 pevent->event.fork.tid,
116 pevent->event.fork.ptid,
117 pevent->event.fork.time);
120 static PyTypeObject pyrf_task_event__type = {
121 PyVarObject_HEAD_INIT(NULL, 0)
122 .tp_name = "perf.task_event",
123 .tp_basicsize = sizeof(struct pyrf_event),
124 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
125 .tp_doc = pyrf_task_event__doc,
126 .tp_members = pyrf_task_event__members,
127 .tp_repr = (reprfunc)pyrf_task_event__repr,
130 static char pyrf_comm_event__doc[] = PyDoc_STR("perf comm event object.");
132 static PyMemberDef pyrf_comm_event__members[] = {
133 sample_members
134 member_def(perf_event_header, type, T_UINT, "event type"),
135 member_def(comm_event, pid, T_UINT, "event pid"),
136 member_def(comm_event, tid, T_UINT, "event tid"),
137 member_def(comm_event, comm, T_STRING_INPLACE, "process name"),
138 { .name = NULL, },
141 static PyObject *pyrf_comm_event__repr(struct pyrf_event *pevent)
143 return PyString_FromFormat("{ type: comm, pid: %u, tid: %u, comm: %s }",
144 pevent->event.comm.pid,
145 pevent->event.comm.tid,
146 pevent->event.comm.comm);
149 static PyTypeObject pyrf_comm_event__type = {
150 PyVarObject_HEAD_INIT(NULL, 0)
151 .tp_name = "perf.comm_event",
152 .tp_basicsize = sizeof(struct pyrf_event),
153 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
154 .tp_doc = pyrf_comm_event__doc,
155 .tp_members = pyrf_comm_event__members,
156 .tp_repr = (reprfunc)pyrf_comm_event__repr,
159 static char pyrf_throttle_event__doc[] = PyDoc_STR("perf throttle event object.");
161 static PyMemberDef pyrf_throttle_event__members[] = {
162 sample_members
163 member_def(perf_event_header, type, T_UINT, "event type"),
164 member_def(throttle_event, time, T_ULONGLONG, "timestamp"),
165 member_def(throttle_event, id, T_ULONGLONG, "event id"),
166 member_def(throttle_event, stream_id, T_ULONGLONG, "event stream id"),
167 { .name = NULL, },
170 static PyObject *pyrf_throttle_event__repr(struct pyrf_event *pevent)
172 struct throttle_event *te = (struct throttle_event *)(&pevent->event.header + 1);
174 return PyString_FromFormat("{ type: %sthrottle, time: %" PRIu64 ", id: %" PRIu64
175 ", stream_id: %" PRIu64 " }",
176 pevent->event.header.type == PERF_RECORD_THROTTLE ? "" : "un",
177 te->time, te->id, te->stream_id);
180 static PyTypeObject pyrf_throttle_event__type = {
181 PyVarObject_HEAD_INIT(NULL, 0)
182 .tp_name = "perf.throttle_event",
183 .tp_basicsize = sizeof(struct pyrf_event),
184 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
185 .tp_doc = pyrf_throttle_event__doc,
186 .tp_members = pyrf_throttle_event__members,
187 .tp_repr = (reprfunc)pyrf_throttle_event__repr,
190 static int pyrf_event__setup_types(void)
192 int err;
193 pyrf_mmap_event__type.tp_new =
194 pyrf_task_event__type.tp_new =
195 pyrf_comm_event__type.tp_new =
196 pyrf_throttle_event__type.tp_new = PyType_GenericNew;
197 err = PyType_Ready(&pyrf_mmap_event__type);
198 if (err < 0)
199 goto out;
200 err = PyType_Ready(&pyrf_task_event__type);
201 if (err < 0)
202 goto out;
203 err = PyType_Ready(&pyrf_comm_event__type);
204 if (err < 0)
205 goto out;
206 err = PyType_Ready(&pyrf_throttle_event__type);
207 if (err < 0)
208 goto out;
209 out:
210 return err;
213 static PyTypeObject *pyrf_event__type[] = {
214 [PERF_RECORD_MMAP] = &pyrf_mmap_event__type,
215 [PERF_RECORD_LOST] = &pyrf_mmap_event__type,
216 [PERF_RECORD_COMM] = &pyrf_comm_event__type,
217 [PERF_RECORD_EXIT] = &pyrf_task_event__type,
218 [PERF_RECORD_THROTTLE] = &pyrf_throttle_event__type,
219 [PERF_RECORD_UNTHROTTLE] = &pyrf_throttle_event__type,
220 [PERF_RECORD_FORK] = &pyrf_task_event__type,
221 [PERF_RECORD_READ] = &pyrf_mmap_event__type,
222 [PERF_RECORD_SAMPLE] = &pyrf_mmap_event__type,
225 static PyObject *pyrf_event__new(union perf_event *event)
227 struct pyrf_event *pevent;
228 PyTypeObject *ptype;
230 if (event->header.type < PERF_RECORD_MMAP ||
231 event->header.type > PERF_RECORD_SAMPLE)
232 return NULL;
234 ptype = pyrf_event__type[event->header.type];
235 pevent = PyObject_New(struct pyrf_event, ptype);
236 if (pevent != NULL)
237 memcpy(&pevent->event, event, event->header.size);
238 return (PyObject *)pevent;
241 struct pyrf_cpu_map {
242 PyObject_HEAD
244 struct cpu_map *cpus;
247 static int pyrf_cpu_map__init(struct pyrf_cpu_map *pcpus,
248 PyObject *args, PyObject *kwargs)
250 static char *kwlist[] = { "cpustr", NULL, NULL, };
251 char *cpustr = NULL;
253 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|s",
254 kwlist, &cpustr))
255 return -1;
257 pcpus->cpus = cpu_map__new(cpustr);
258 if (pcpus->cpus == NULL)
259 return -1;
260 return 0;
263 static void pyrf_cpu_map__delete(struct pyrf_cpu_map *pcpus)
265 cpu_map__delete(pcpus->cpus);
266 pcpus->ob_type->tp_free((PyObject*)pcpus);
269 static Py_ssize_t pyrf_cpu_map__length(PyObject *obj)
271 struct pyrf_cpu_map *pcpus = (void *)obj;
273 return pcpus->cpus->nr;
276 static PyObject *pyrf_cpu_map__item(PyObject *obj, Py_ssize_t i)
278 struct pyrf_cpu_map *pcpus = (void *)obj;
280 if (i >= pcpus->cpus->nr)
281 return NULL;
283 return Py_BuildValue("i", pcpus->cpus->map[i]);
286 static PySequenceMethods pyrf_cpu_map__sequence_methods = {
287 .sq_length = pyrf_cpu_map__length,
288 .sq_item = pyrf_cpu_map__item,
291 static char pyrf_cpu_map__doc[] = PyDoc_STR("cpu map object.");
293 static PyTypeObject pyrf_cpu_map__type = {
294 PyVarObject_HEAD_INIT(NULL, 0)
295 .tp_name = "perf.cpu_map",
296 .tp_basicsize = sizeof(struct pyrf_cpu_map),
297 .tp_dealloc = (destructor)pyrf_cpu_map__delete,
298 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
299 .tp_doc = pyrf_cpu_map__doc,
300 .tp_as_sequence = &pyrf_cpu_map__sequence_methods,
301 .tp_init = (initproc)pyrf_cpu_map__init,
304 static int pyrf_cpu_map__setup_types(void)
306 pyrf_cpu_map__type.tp_new = PyType_GenericNew;
307 return PyType_Ready(&pyrf_cpu_map__type);
310 struct pyrf_thread_map {
311 PyObject_HEAD
313 struct thread_map *threads;
316 static int pyrf_thread_map__init(struct pyrf_thread_map *pthreads,
317 PyObject *args, PyObject *kwargs)
319 static char *kwlist[] = { "pid", "tid", NULL, NULL, };
320 int pid = -1, tid = -1;
322 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ii",
323 kwlist, &pid, &tid))
324 return -1;
326 pthreads->threads = thread_map__new(pid, tid);
327 if (pthreads->threads == NULL)
328 return -1;
329 return 0;
332 static void pyrf_thread_map__delete(struct pyrf_thread_map *pthreads)
334 thread_map__delete(pthreads->threads);
335 pthreads->ob_type->tp_free((PyObject*)pthreads);
338 static Py_ssize_t pyrf_thread_map__length(PyObject *obj)
340 struct pyrf_thread_map *pthreads = (void *)obj;
342 return pthreads->threads->nr;
345 static PyObject *pyrf_thread_map__item(PyObject *obj, Py_ssize_t i)
347 struct pyrf_thread_map *pthreads = (void *)obj;
349 if (i >= pthreads->threads->nr)
350 return NULL;
352 return Py_BuildValue("i", pthreads->threads->map[i]);
355 static PySequenceMethods pyrf_thread_map__sequence_methods = {
356 .sq_length = pyrf_thread_map__length,
357 .sq_item = pyrf_thread_map__item,
360 static char pyrf_thread_map__doc[] = PyDoc_STR("thread map object.");
362 static PyTypeObject pyrf_thread_map__type = {
363 PyVarObject_HEAD_INIT(NULL, 0)
364 .tp_name = "perf.thread_map",
365 .tp_basicsize = sizeof(struct pyrf_thread_map),
366 .tp_dealloc = (destructor)pyrf_thread_map__delete,
367 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
368 .tp_doc = pyrf_thread_map__doc,
369 .tp_as_sequence = &pyrf_thread_map__sequence_methods,
370 .tp_init = (initproc)pyrf_thread_map__init,
373 static int pyrf_thread_map__setup_types(void)
375 pyrf_thread_map__type.tp_new = PyType_GenericNew;
376 return PyType_Ready(&pyrf_thread_map__type);
379 struct pyrf_evsel {
380 PyObject_HEAD
382 struct perf_evsel evsel;
385 static int pyrf_evsel__init(struct pyrf_evsel *pevsel,
386 PyObject *args, PyObject *kwargs)
388 struct perf_event_attr attr = {
389 .type = PERF_TYPE_HARDWARE,
390 .config = PERF_COUNT_HW_CPU_CYCLES,
391 .sample_type = PERF_SAMPLE_PERIOD | PERF_SAMPLE_TID,
393 static char *kwlist[] = {
394 "type",
395 "config",
396 "sample_freq",
397 "sample_period",
398 "sample_type",
399 "read_format",
400 "disabled",
401 "inherit",
402 "pinned",
403 "exclusive",
404 "exclude_user",
405 "exclude_kernel",
406 "exclude_hv",
407 "exclude_idle",
408 "mmap",
409 "comm",
410 "freq",
411 "inherit_stat",
412 "enable_on_exec",
413 "task",
414 "watermark",
415 "precise_ip",
416 "mmap_data",
417 "sample_id_all",
418 "wakeup_events",
419 "bp_type",
420 "bp_addr",
421 "bp_len", NULL, NULL, };
422 u64 sample_period = 0;
423 u32 disabled = 0,
424 inherit = 0,
425 pinned = 0,
426 exclusive = 0,
427 exclude_user = 0,
428 exclude_kernel = 0,
429 exclude_hv = 0,
430 exclude_idle = 0,
431 mmap = 0,
432 comm = 0,
433 freq = 1,
434 inherit_stat = 0,
435 enable_on_exec = 0,
436 task = 0,
437 watermark = 0,
438 precise_ip = 0,
439 mmap_data = 0,
440 sample_id_all = 1;
441 int idx = 0;
443 if (!PyArg_ParseTupleAndKeywords(args, kwargs,
444 "|iKiKKiiiiiiiiiiiiiiiiiiiiiKK", kwlist,
445 &attr.type, &attr.config, &attr.sample_freq,
446 &sample_period, &attr.sample_type,
447 &attr.read_format, &disabled, &inherit,
448 &pinned, &exclusive, &exclude_user,
449 &exclude_kernel, &exclude_hv, &exclude_idle,
450 &mmap, &comm, &freq, &inherit_stat,
451 &enable_on_exec, &task, &watermark,
452 &precise_ip, &mmap_data, &sample_id_all,
453 &attr.wakeup_events, &attr.bp_type,
454 &attr.bp_addr, &attr.bp_len, &idx))
455 return -1;
457 /* union... */
458 if (sample_period != 0) {
459 if (attr.sample_freq != 0)
460 return -1; /* FIXME: throw right exception */
461 attr.sample_period = sample_period;
464 /* Bitfields */
465 attr.disabled = disabled;
466 attr.inherit = inherit;
467 attr.pinned = pinned;
468 attr.exclusive = exclusive;
469 attr.exclude_user = exclude_user;
470 attr.exclude_kernel = exclude_kernel;
471 attr.exclude_hv = exclude_hv;
472 attr.exclude_idle = exclude_idle;
473 attr.mmap = mmap;
474 attr.comm = comm;
475 attr.freq = freq;
476 attr.inherit_stat = inherit_stat;
477 attr.enable_on_exec = enable_on_exec;
478 attr.task = task;
479 attr.watermark = watermark;
480 attr.precise_ip = precise_ip;
481 attr.mmap_data = mmap_data;
482 attr.sample_id_all = sample_id_all;
484 perf_evsel__init(&pevsel->evsel, &attr, idx);
485 return 0;
488 static void pyrf_evsel__delete(struct pyrf_evsel *pevsel)
490 perf_evsel__exit(&pevsel->evsel);
491 pevsel->ob_type->tp_free((PyObject*)pevsel);
494 static PyObject *pyrf_evsel__open(struct pyrf_evsel *pevsel,
495 PyObject *args, PyObject *kwargs)
497 struct perf_evsel *evsel = &pevsel->evsel;
498 struct cpu_map *cpus = NULL;
499 struct thread_map *threads = NULL;
500 PyObject *pcpus = NULL, *pthreads = NULL;
501 int group = 0, inherit = 0;
502 static char *kwlist[] = {"cpus", "threads", "group", "inherit", NULL, NULL};
504 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|OOii", kwlist,
505 &pcpus, &pthreads, &group, &inherit))
506 return NULL;
508 if (pthreads != NULL)
509 threads = ((struct pyrf_thread_map *)pthreads)->threads;
511 if (pcpus != NULL)
512 cpus = ((struct pyrf_cpu_map *)pcpus)->cpus;
514 evsel->attr.inherit = inherit;
515 if (perf_evsel__open(evsel, cpus, threads, group) < 0) {
516 PyErr_SetFromErrno(PyExc_OSError);
517 return NULL;
520 Py_INCREF(Py_None);
521 return Py_None;
524 static PyMethodDef pyrf_evsel__methods[] = {
526 .ml_name = "open",
527 .ml_meth = (PyCFunction)pyrf_evsel__open,
528 .ml_flags = METH_VARARGS | METH_KEYWORDS,
529 .ml_doc = PyDoc_STR("open the event selector file descriptor table.")
531 { .ml_name = NULL, }
534 static char pyrf_evsel__doc[] = PyDoc_STR("perf event selector list object.");
536 static PyTypeObject pyrf_evsel__type = {
537 PyVarObject_HEAD_INIT(NULL, 0)
538 .tp_name = "perf.evsel",
539 .tp_basicsize = sizeof(struct pyrf_evsel),
540 .tp_dealloc = (destructor)pyrf_evsel__delete,
541 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
542 .tp_doc = pyrf_evsel__doc,
543 .tp_methods = pyrf_evsel__methods,
544 .tp_init = (initproc)pyrf_evsel__init,
547 static int pyrf_evsel__setup_types(void)
549 pyrf_evsel__type.tp_new = PyType_GenericNew;
550 return PyType_Ready(&pyrf_evsel__type);
553 struct pyrf_evlist {
554 PyObject_HEAD
556 struct perf_evlist evlist;
559 static int pyrf_evlist__init(struct pyrf_evlist *pevlist,
560 PyObject *args, PyObject *kwargs __used)
562 PyObject *pcpus = NULL, *pthreads = NULL;
563 struct cpu_map *cpus;
564 struct thread_map *threads;
566 if (!PyArg_ParseTuple(args, "OO", &pcpus, &pthreads))
567 return -1;
569 threads = ((struct pyrf_thread_map *)pthreads)->threads;
570 cpus = ((struct pyrf_cpu_map *)pcpus)->cpus;
571 perf_evlist__init(&pevlist->evlist, cpus, threads);
572 return 0;
575 static void pyrf_evlist__delete(struct pyrf_evlist *pevlist)
577 perf_evlist__exit(&pevlist->evlist);
578 pevlist->ob_type->tp_free((PyObject*)pevlist);
581 static PyObject *pyrf_evlist__mmap(struct pyrf_evlist *pevlist,
582 PyObject *args, PyObject *kwargs)
584 struct perf_evlist *evlist = &pevlist->evlist;
585 static char *kwlist[] = {"pages", "overwrite",
586 NULL, NULL};
587 int pages = 128, overwrite = false;
589 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ii", kwlist,
590 &pages, &overwrite))
591 return NULL;
593 if (perf_evlist__mmap(evlist, pages, overwrite) < 0) {
594 PyErr_SetFromErrno(PyExc_OSError);
595 return NULL;
598 Py_INCREF(Py_None);
599 return Py_None;
602 static PyObject *pyrf_evlist__poll(struct pyrf_evlist *pevlist,
603 PyObject *args, PyObject *kwargs)
605 struct perf_evlist *evlist = &pevlist->evlist;
606 static char *kwlist[] = {"timeout", NULL, NULL};
607 int timeout = -1, n;
609 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|i", kwlist, &timeout))
610 return NULL;
612 n = poll(evlist->pollfd, evlist->nr_fds, timeout);
613 if (n < 0) {
614 PyErr_SetFromErrno(PyExc_OSError);
615 return NULL;
618 return Py_BuildValue("i", n);
621 static PyObject *pyrf_evlist__get_pollfd(struct pyrf_evlist *pevlist,
622 PyObject *args __used, PyObject *kwargs __used)
624 struct perf_evlist *evlist = &pevlist->evlist;
625 PyObject *list = PyList_New(0);
626 int i;
628 for (i = 0; i < evlist->nr_fds; ++i) {
629 PyObject *file;
630 FILE *fp = fdopen(evlist->pollfd[i].fd, "r");
632 if (fp == NULL)
633 goto free_list;
635 file = PyFile_FromFile(fp, "perf", "r", NULL);
636 if (file == NULL)
637 goto free_list;
639 if (PyList_Append(list, file) != 0) {
640 Py_DECREF(file);
641 goto free_list;
644 Py_DECREF(file);
647 return list;
648 free_list:
649 return PyErr_NoMemory();
653 static PyObject *pyrf_evlist__add(struct pyrf_evlist *pevlist,
654 PyObject *args, PyObject *kwargs __used)
656 struct perf_evlist *evlist = &pevlist->evlist;
657 PyObject *pevsel;
658 struct perf_evsel *evsel;
660 if (!PyArg_ParseTuple(args, "O", &pevsel))
661 return NULL;
663 Py_INCREF(pevsel);
664 evsel = &((struct pyrf_evsel *)pevsel)->evsel;
665 evsel->idx = evlist->nr_entries;
666 perf_evlist__add(evlist, evsel);
668 return Py_BuildValue("i", evlist->nr_entries);
671 static PyObject *pyrf_evlist__read_on_cpu(struct pyrf_evlist *pevlist,
672 PyObject *args, PyObject *kwargs)
674 struct perf_evlist *evlist = &pevlist->evlist;
675 union perf_event *event;
676 int sample_id_all = 1, cpu;
677 static char *kwlist[] = {"cpu", "sample_id_all", NULL, NULL};
678 int err;
680 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "i|i", kwlist,
681 &cpu, &sample_id_all))
682 return NULL;
684 event = perf_evlist__mmap_read(evlist, cpu);
685 if (event != NULL) {
686 struct perf_evsel *first;
687 PyObject *pyevent = pyrf_event__new(event);
688 struct pyrf_event *pevent = (struct pyrf_event *)pyevent;
690 if (pyevent == NULL)
691 return PyErr_NoMemory();
693 first = list_entry(evlist->entries.next, struct perf_evsel, node);
694 err = perf_event__parse_sample(event, first->attr.sample_type,
695 perf_evsel__sample_size(first),
696 sample_id_all, &pevent->sample);
697 if (err)
698 return PyErr_Format(PyExc_OSError,
699 "perf: can't parse sample, err=%d", err);
700 return pyevent;
703 Py_INCREF(Py_None);
704 return Py_None;
707 static PyMethodDef pyrf_evlist__methods[] = {
709 .ml_name = "mmap",
710 .ml_meth = (PyCFunction)pyrf_evlist__mmap,
711 .ml_flags = METH_VARARGS | METH_KEYWORDS,
712 .ml_doc = PyDoc_STR("mmap the file descriptor table.")
715 .ml_name = "poll",
716 .ml_meth = (PyCFunction)pyrf_evlist__poll,
717 .ml_flags = METH_VARARGS | METH_KEYWORDS,
718 .ml_doc = PyDoc_STR("poll the file descriptor table.")
721 .ml_name = "get_pollfd",
722 .ml_meth = (PyCFunction)pyrf_evlist__get_pollfd,
723 .ml_flags = METH_VARARGS | METH_KEYWORDS,
724 .ml_doc = PyDoc_STR("get the poll file descriptor table.")
727 .ml_name = "add",
728 .ml_meth = (PyCFunction)pyrf_evlist__add,
729 .ml_flags = METH_VARARGS | METH_KEYWORDS,
730 .ml_doc = PyDoc_STR("adds an event selector to the list.")
733 .ml_name = "read_on_cpu",
734 .ml_meth = (PyCFunction)pyrf_evlist__read_on_cpu,
735 .ml_flags = METH_VARARGS | METH_KEYWORDS,
736 .ml_doc = PyDoc_STR("reads an event.")
738 { .ml_name = NULL, }
741 static Py_ssize_t pyrf_evlist__length(PyObject *obj)
743 struct pyrf_evlist *pevlist = (void *)obj;
745 return pevlist->evlist.nr_entries;
748 static PyObject *pyrf_evlist__item(PyObject *obj, Py_ssize_t i)
750 struct pyrf_evlist *pevlist = (void *)obj;
751 struct perf_evsel *pos;
753 if (i >= pevlist->evlist.nr_entries)
754 return NULL;
756 list_for_each_entry(pos, &pevlist->evlist.entries, node)
757 if (i-- == 0)
758 break;
760 return Py_BuildValue("O", container_of(pos, struct pyrf_evsel, evsel));
763 static PySequenceMethods pyrf_evlist__sequence_methods = {
764 .sq_length = pyrf_evlist__length,
765 .sq_item = pyrf_evlist__item,
768 static char pyrf_evlist__doc[] = PyDoc_STR("perf event selector list object.");
770 static PyTypeObject pyrf_evlist__type = {
771 PyVarObject_HEAD_INIT(NULL, 0)
772 .tp_name = "perf.evlist",
773 .tp_basicsize = sizeof(struct pyrf_evlist),
774 .tp_dealloc = (destructor)pyrf_evlist__delete,
775 .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
776 .tp_as_sequence = &pyrf_evlist__sequence_methods,
777 .tp_doc = pyrf_evlist__doc,
778 .tp_methods = pyrf_evlist__methods,
779 .tp_init = (initproc)pyrf_evlist__init,
782 static int pyrf_evlist__setup_types(void)
784 pyrf_evlist__type.tp_new = PyType_GenericNew;
785 return PyType_Ready(&pyrf_evlist__type);
788 static struct {
789 const char *name;
790 int value;
791 } perf__constants[] = {
792 { "TYPE_HARDWARE", PERF_TYPE_HARDWARE },
793 { "TYPE_SOFTWARE", PERF_TYPE_SOFTWARE },
794 { "TYPE_TRACEPOINT", PERF_TYPE_TRACEPOINT },
795 { "TYPE_HW_CACHE", PERF_TYPE_HW_CACHE },
796 { "TYPE_RAW", PERF_TYPE_RAW },
797 { "TYPE_BREAKPOINT", PERF_TYPE_BREAKPOINT },
799 { "COUNT_HW_CPU_CYCLES", PERF_COUNT_HW_CPU_CYCLES },
800 { "COUNT_HW_INSTRUCTIONS", PERF_COUNT_HW_INSTRUCTIONS },
801 { "COUNT_HW_CACHE_REFERENCES", PERF_COUNT_HW_CACHE_REFERENCES },
802 { "COUNT_HW_CACHE_MISSES", PERF_COUNT_HW_CACHE_MISSES },
803 { "COUNT_HW_BRANCH_INSTRUCTIONS", PERF_COUNT_HW_BRANCH_INSTRUCTIONS },
804 { "COUNT_HW_BRANCH_MISSES", PERF_COUNT_HW_BRANCH_MISSES },
805 { "COUNT_HW_BUS_CYCLES", PERF_COUNT_HW_BUS_CYCLES },
806 { "COUNT_HW_CACHE_L1D", PERF_COUNT_HW_CACHE_L1D },
807 { "COUNT_HW_CACHE_L1I", PERF_COUNT_HW_CACHE_L1I },
808 { "COUNT_HW_CACHE_LL", PERF_COUNT_HW_CACHE_LL },
809 { "COUNT_HW_CACHE_DTLB", PERF_COUNT_HW_CACHE_DTLB },
810 { "COUNT_HW_CACHE_ITLB", PERF_COUNT_HW_CACHE_ITLB },
811 { "COUNT_HW_CACHE_BPU", PERF_COUNT_HW_CACHE_BPU },
812 { "COUNT_HW_CACHE_OP_READ", PERF_COUNT_HW_CACHE_OP_READ },
813 { "COUNT_HW_CACHE_OP_WRITE", PERF_COUNT_HW_CACHE_OP_WRITE },
814 { "COUNT_HW_CACHE_OP_PREFETCH", PERF_COUNT_HW_CACHE_OP_PREFETCH },
815 { "COUNT_HW_CACHE_RESULT_ACCESS", PERF_COUNT_HW_CACHE_RESULT_ACCESS },
816 { "COUNT_HW_CACHE_RESULT_MISS", PERF_COUNT_HW_CACHE_RESULT_MISS },
818 { "COUNT_HW_STALLED_CYCLES_FRONTEND", PERF_COUNT_HW_STALLED_CYCLES_FRONTEND },
819 { "COUNT_HW_STALLED_CYCLES_BACKEND", PERF_COUNT_HW_STALLED_CYCLES_BACKEND },
821 { "COUNT_SW_CPU_CLOCK", PERF_COUNT_SW_CPU_CLOCK },
822 { "COUNT_SW_TASK_CLOCK", PERF_COUNT_SW_TASK_CLOCK },
823 { "COUNT_SW_PAGE_FAULTS", PERF_COUNT_SW_PAGE_FAULTS },
824 { "COUNT_SW_CONTEXT_SWITCHES", PERF_COUNT_SW_CONTEXT_SWITCHES },
825 { "COUNT_SW_CPU_MIGRATIONS", PERF_COUNT_SW_CPU_MIGRATIONS },
826 { "COUNT_SW_PAGE_FAULTS_MIN", PERF_COUNT_SW_PAGE_FAULTS_MIN },
827 { "COUNT_SW_PAGE_FAULTS_MAJ", PERF_COUNT_SW_PAGE_FAULTS_MAJ },
828 { "COUNT_SW_ALIGNMENT_FAULTS", PERF_COUNT_SW_ALIGNMENT_FAULTS },
829 { "COUNT_SW_EMULATION_FAULTS", PERF_COUNT_SW_EMULATION_FAULTS },
831 { "SAMPLE_IP", PERF_SAMPLE_IP },
832 { "SAMPLE_TID", PERF_SAMPLE_TID },
833 { "SAMPLE_TIME", PERF_SAMPLE_TIME },
834 { "SAMPLE_ADDR", PERF_SAMPLE_ADDR },
835 { "SAMPLE_READ", PERF_SAMPLE_READ },
836 { "SAMPLE_CALLCHAIN", PERF_SAMPLE_CALLCHAIN },
837 { "SAMPLE_ID", PERF_SAMPLE_ID },
838 { "SAMPLE_CPU", PERF_SAMPLE_CPU },
839 { "SAMPLE_PERIOD", PERF_SAMPLE_PERIOD },
840 { "SAMPLE_STREAM_ID", PERF_SAMPLE_STREAM_ID },
841 { "SAMPLE_RAW", PERF_SAMPLE_RAW },
843 { "FORMAT_TOTAL_TIME_ENABLED", PERF_FORMAT_TOTAL_TIME_ENABLED },
844 { "FORMAT_TOTAL_TIME_RUNNING", PERF_FORMAT_TOTAL_TIME_RUNNING },
845 { "FORMAT_ID", PERF_FORMAT_ID },
846 { "FORMAT_GROUP", PERF_FORMAT_GROUP },
848 { "RECORD_MMAP", PERF_RECORD_MMAP },
849 { "RECORD_LOST", PERF_RECORD_LOST },
850 { "RECORD_COMM", PERF_RECORD_COMM },
851 { "RECORD_EXIT", PERF_RECORD_EXIT },
852 { "RECORD_THROTTLE", PERF_RECORD_THROTTLE },
853 { "RECORD_UNTHROTTLE", PERF_RECORD_UNTHROTTLE },
854 { "RECORD_FORK", PERF_RECORD_FORK },
855 { "RECORD_READ", PERF_RECORD_READ },
856 { "RECORD_SAMPLE", PERF_RECORD_SAMPLE },
857 { .name = NULL, },
860 static PyMethodDef perf__methods[] = {
861 { .ml_name = NULL, }
864 PyMODINIT_FUNC initperf(void)
866 PyObject *obj;
867 int i;
868 PyObject *dict, *module = Py_InitModule("perf", perf__methods);
870 if (module == NULL ||
871 pyrf_event__setup_types() < 0 ||
872 pyrf_evlist__setup_types() < 0 ||
873 pyrf_evsel__setup_types() < 0 ||
874 pyrf_thread_map__setup_types() < 0 ||
875 pyrf_cpu_map__setup_types() < 0)
876 return;
878 Py_INCREF(&pyrf_evlist__type);
879 PyModule_AddObject(module, "evlist", (PyObject*)&pyrf_evlist__type);
881 Py_INCREF(&pyrf_evsel__type);
882 PyModule_AddObject(module, "evsel", (PyObject*)&pyrf_evsel__type);
884 Py_INCREF(&pyrf_thread_map__type);
885 PyModule_AddObject(module, "thread_map", (PyObject*)&pyrf_thread_map__type);
887 Py_INCREF(&pyrf_cpu_map__type);
888 PyModule_AddObject(module, "cpu_map", (PyObject*)&pyrf_cpu_map__type);
890 dict = PyModule_GetDict(module);
891 if (dict == NULL)
892 goto error;
894 for (i = 0; perf__constants[i].name != NULL; i++) {
895 obj = PyInt_FromLong(perf__constants[i].value);
896 if (obj == NULL)
897 goto error;
898 PyDict_SetItemString(dict, perf__constants[i].name, obj);
899 Py_DECREF(obj);
902 error:
903 if (PyErr_Occurred())
904 PyErr_SetString(PyExc_ImportError, "perf: Init failed!");