Commit | Line | Data |
---|---|---|
877108e4 ACM |
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" | |
10 | ||
cfff2d90 FW |
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 | |
15 | ||
877108e4 ACM |
16 | struct throttle_event { |
17 | struct perf_event_header header; | |
18 | u64 time; | |
19 | u64 id; | |
20 | u64 stream_id; | |
21 | }; | |
22 | ||
f6bbc1da ACM |
23 | PyMODINIT_FUNC initperf(void); |
24 | ||
877108e4 ACM |
25 | #define member_def(type, member, ptype, help) \ |
26 | { #member, ptype, \ | |
27 | offsetof(struct pyrf_event, event) + offsetof(struct type, member), \ | |
28 | 0, help } | |
29 | ||
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 } | |
34 | ||
35 | struct pyrf_event { | |
36 | PyObject_HEAD | |
37 | struct perf_sample sample; | |
38 | union perf_event event; | |
39 | }; | |
40 | ||
877108e4 | 41 | #define sample_members \ |
f6bbc1da | 42 | sample_member_def(sample_ip, ip, T_ULONGLONG, "event type"), \ |
877108e4 ACM |
43 | sample_member_def(sample_pid, pid, T_INT, "event pid"), \ |
44 | sample_member_def(sample_tid, tid, T_INT, "event tid"), \ | |
f6bbc1da ACM |
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"), \ | |
877108e4 ACM |
50 | sample_member_def(sample_cpu, cpu, T_UINT, "event cpu"), |
51 | ||
52 | static char pyrf_mmap_event__doc[] = PyDoc_STR("perf mmap event object."); | |
53 | ||
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"), | |
f6bbc1da ACM |
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"), | |
877108e4 | 62 | member_def(mmap_event, filename, T_STRING_INPLACE, "backing store"), |
f6bbc1da | 63 | { .name = NULL, }, |
877108e4 ACM |
64 | }; |
65 | ||
66 | static PyObject *pyrf_mmap_event__repr(struct pyrf_event *pevent) | |
67 | { | |
68 | PyObject *ret; | |
69 | char *s; | |
70 | ||
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); | |
81 | } | |
82 | return ret; | |
83 | } | |
84 | ||
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, | |
93 | }; | |
94 | ||
95 | static char pyrf_task_event__doc[] = PyDoc_STR("perf task (fork/exit) event object."); | |
96 | ||
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"), | |
f6bbc1da ACM |
104 | member_def(fork_event, time, T_ULONGLONG, "timestamp"), |
105 | { .name = NULL, }, | |
877108e4 ACM |
106 | }; |
107 | ||
108 | static PyObject *pyrf_task_event__repr(struct pyrf_event *pevent) | |
109 | { | |
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); | |
118 | } | |
119 | ||
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, | |
128 | }; | |
129 | ||
130 | static char pyrf_comm_event__doc[] = PyDoc_STR("perf comm event object."); | |
131 | ||
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"), | |
f6bbc1da | 138 | { .name = NULL, }, |
877108e4 ACM |
139 | }; |
140 | ||
141 | static PyObject *pyrf_comm_event__repr(struct pyrf_event *pevent) | |
142 | { | |
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); | |
147 | } | |
148 | ||
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, | |
157 | }; | |
158 | ||
159 | static char pyrf_throttle_event__doc[] = PyDoc_STR("perf throttle event object."); | |
160 | ||
161 | static PyMemberDef pyrf_throttle_event__members[] = { | |
162 | sample_members | |
163 | member_def(perf_event_header, type, T_UINT, "event type"), | |
f6bbc1da ACM |
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, }, | |
877108e4 ACM |
168 | }; |
169 | ||
170 | static PyObject *pyrf_throttle_event__repr(struct pyrf_event *pevent) | |
171 | { | |
172 | struct throttle_event *te = (struct throttle_event *)(&pevent->event.header + 1); | |
173 | ||
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); | |
178 | } | |
179 | ||
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, | |
188 | }; | |
189 | ||
190 | static int pyrf_event__setup_types(void) | |
191 | { | |
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; | |
211 | } | |
212 | ||
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, | |
223 | }; | |
224 | ||
225 | static PyObject *pyrf_event__new(union perf_event *event) | |
226 | { | |
227 | struct pyrf_event *pevent; | |
228 | PyTypeObject *ptype; | |
229 | ||
230 | if (event->header.type < PERF_RECORD_MMAP || | |
231 | event->header.type > PERF_RECORD_SAMPLE) | |
232 | return NULL; | |
233 | ||
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; | |
239 | } | |
240 | ||
241 | struct pyrf_cpu_map { | |
242 | PyObject_HEAD | |
243 | ||
244 | struct cpu_map *cpus; | |
245 | }; | |
246 | ||
247 | static int pyrf_cpu_map__init(struct pyrf_cpu_map *pcpus, | |
248 | PyObject *args, PyObject *kwargs) | |
249 | { | |
250 | static char *kwlist[] = { "cpustr", NULL, NULL, }; | |
251 | char *cpustr = NULL; | |
252 | ||
253 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|s", | |
254 | kwlist, &cpustr)) | |
255 | return -1; | |
256 | ||
257 | pcpus->cpus = cpu_map__new(cpustr); | |
258 | if (pcpus->cpus == NULL) | |
259 | return -1; | |
260 | return 0; | |
261 | } | |
262 | ||
263 | static void pyrf_cpu_map__delete(struct pyrf_cpu_map *pcpus) | |
264 | { | |
265 | cpu_map__delete(pcpus->cpus); | |
266 | pcpus->ob_type->tp_free((PyObject*)pcpus); | |
267 | } | |
268 | ||
269 | static Py_ssize_t pyrf_cpu_map__length(PyObject *obj) | |
270 | { | |
271 | struct pyrf_cpu_map *pcpus = (void *)obj; | |
272 | ||
273 | return pcpus->cpus->nr; | |
274 | } | |
275 | ||
276 | static PyObject *pyrf_cpu_map__item(PyObject *obj, Py_ssize_t i) | |
277 | { | |
278 | struct pyrf_cpu_map *pcpus = (void *)obj; | |
279 | ||
280 | if (i >= pcpus->cpus->nr) | |
281 | return NULL; | |
282 | ||
283 | return Py_BuildValue("i", pcpus->cpus->map[i]); | |
284 | } | |
285 | ||
286 | static PySequenceMethods pyrf_cpu_map__sequence_methods = { | |
287 | .sq_length = pyrf_cpu_map__length, | |
288 | .sq_item = pyrf_cpu_map__item, | |
289 | }; | |
290 | ||
291 | static char pyrf_cpu_map__doc[] = PyDoc_STR("cpu map object."); | |
292 | ||
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, | |
302 | }; | |
303 | ||
304 | static int pyrf_cpu_map__setup_types(void) | |
305 | { | |
306 | pyrf_cpu_map__type.tp_new = PyType_GenericNew; | |
307 | return PyType_Ready(&pyrf_cpu_map__type); | |
308 | } | |
309 | ||
310 | struct pyrf_thread_map { | |
311 | PyObject_HEAD | |
312 | ||
313 | struct thread_map *threads; | |
314 | }; | |
315 | ||
316 | static int pyrf_thread_map__init(struct pyrf_thread_map *pthreads, | |
317 | PyObject *args, PyObject *kwargs) | |
318 | { | |
319 | static char *kwlist[] = { "pid", "tid", NULL, NULL, }; | |
320 | int pid = -1, tid = -1; | |
321 | ||
322 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ii", | |
323 | kwlist, &pid, &tid)) | |
324 | return -1; | |
325 | ||
326 | pthreads->threads = thread_map__new(pid, tid); | |
327 | if (pthreads->threads == NULL) | |
328 | return -1; | |
329 | return 0; | |
330 | } | |
331 | ||
332 | static void pyrf_thread_map__delete(struct pyrf_thread_map *pthreads) | |
333 | { | |
334 | thread_map__delete(pthreads->threads); | |
335 | pthreads->ob_type->tp_free((PyObject*)pthreads); | |
336 | } | |
337 | ||
338 | static Py_ssize_t pyrf_thread_map__length(PyObject *obj) | |
339 | { | |
340 | struct pyrf_thread_map *pthreads = (void *)obj; | |
341 | ||
342 | return pthreads->threads->nr; | |
343 | } | |
344 | ||
345 | static PyObject *pyrf_thread_map__item(PyObject *obj, Py_ssize_t i) | |
346 | { | |
347 | struct pyrf_thread_map *pthreads = (void *)obj; | |
348 | ||
349 | if (i >= pthreads->threads->nr) | |
350 | return NULL; | |
351 | ||
352 | return Py_BuildValue("i", pthreads->threads->map[i]); | |
353 | } | |
354 | ||
355 | static PySequenceMethods pyrf_thread_map__sequence_methods = { | |
356 | .sq_length = pyrf_thread_map__length, | |
357 | .sq_item = pyrf_thread_map__item, | |
358 | }; | |
359 | ||
360 | static char pyrf_thread_map__doc[] = PyDoc_STR("thread map object."); | |
361 | ||
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, | |
371 | }; | |
372 | ||
373 | static int pyrf_thread_map__setup_types(void) | |
374 | { | |
375 | pyrf_thread_map__type.tp_new = PyType_GenericNew; | |
376 | return PyType_Ready(&pyrf_thread_map__type); | |
377 | } | |
378 | ||
379 | struct pyrf_evsel { | |
380 | PyObject_HEAD | |
381 | ||
382 | struct perf_evsel evsel; | |
383 | }; | |
384 | ||
385 | static int pyrf_evsel__init(struct pyrf_evsel *pevsel, | |
386 | PyObject *args, PyObject *kwargs) | |
387 | { | |
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, | |
392 | }; | |
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; | |
442 | ||
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; | |
456 | ||
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; | |
462 | } | |
463 | ||
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; | |
483 | ||
484 | perf_evsel__init(&pevsel->evsel, &attr, idx); | |
485 | return 0; | |
486 | } | |
487 | ||
488 | static void pyrf_evsel__delete(struct pyrf_evsel *pevsel) | |
489 | { | |
490 | perf_evsel__exit(&pevsel->evsel); | |
491 | pevsel->ob_type->tp_free((PyObject*)pevsel); | |
492 | } | |
493 | ||
494 | static PyObject *pyrf_evsel__open(struct pyrf_evsel *pevsel, | |
495 | PyObject *args, PyObject *kwargs) | |
496 | { | |
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; | |
5d2cd909 ACM |
501 | int group = 0, inherit = 0; |
502 | static char *kwlist[] = {"cpus", "threads", "group", "inherit", NULL, NULL}; | |
877108e4 ACM |
503 | |
504 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|OOii", kwlist, | |
5d2cd909 | 505 | &pcpus, &pthreads, &group, &inherit)) |
877108e4 ACM |
506 | return NULL; |
507 | ||
508 | if (pthreads != NULL) | |
509 | threads = ((struct pyrf_thread_map *)pthreads)->threads; | |
510 | ||
511 | if (pcpus != NULL) | |
512 | cpus = ((struct pyrf_cpu_map *)pcpus)->cpus; | |
513 | ||
5d2cd909 ACM |
514 | evsel->attr.inherit = inherit; |
515 | if (perf_evsel__open(evsel, cpus, threads, group) < 0) { | |
877108e4 ACM |
516 | PyErr_SetFromErrno(PyExc_OSError); |
517 | return NULL; | |
518 | } | |
519 | ||
520 | Py_INCREF(Py_None); | |
521 | return Py_None; | |
522 | } | |
523 | ||
524 | static PyMethodDef pyrf_evsel__methods[] = { | |
525 | { | |
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.") | |
530 | }, | |
f6bbc1da | 531 | { .ml_name = NULL, } |
877108e4 ACM |
532 | }; |
533 | ||
534 | static char pyrf_evsel__doc[] = PyDoc_STR("perf event selector list object."); | |
535 | ||
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, | |
545 | }; | |
546 | ||
547 | static int pyrf_evsel__setup_types(void) | |
548 | { | |
549 | pyrf_evsel__type.tp_new = PyType_GenericNew; | |
550 | return PyType_Ready(&pyrf_evsel__type); | |
551 | } | |
552 | ||
553 | struct pyrf_evlist { | |
554 | PyObject_HEAD | |
555 | ||
556 | struct perf_evlist evlist; | |
557 | }; | |
558 | ||
559 | static int pyrf_evlist__init(struct pyrf_evlist *pevlist, | |
f6bbc1da | 560 | PyObject *args, PyObject *kwargs __used) |
877108e4 | 561 | { |
7e2ed097 ACM |
562 | PyObject *pcpus = NULL, *pthreads = NULL; |
563 | struct cpu_map *cpus; | |
564 | struct thread_map *threads; | |
565 | ||
566 | if (!PyArg_ParseTuple(args, "OO", &pcpus, &pthreads)) | |
567 | return -1; | |
568 | ||
569 | threads = ((struct pyrf_thread_map *)pthreads)->threads; | |
570 | cpus = ((struct pyrf_cpu_map *)pcpus)->cpus; | |
571 | perf_evlist__init(&pevlist->evlist, cpus, threads); | |
877108e4 ACM |
572 | return 0; |
573 | } | |
574 | ||
575 | static void pyrf_evlist__delete(struct pyrf_evlist *pevlist) | |
576 | { | |
577 | perf_evlist__exit(&pevlist->evlist); | |
578 | pevlist->ob_type->tp_free((PyObject*)pevlist); | |
579 | } | |
580 | ||
581 | static PyObject *pyrf_evlist__mmap(struct pyrf_evlist *pevlist, | |
582 | PyObject *args, PyObject *kwargs) | |
583 | { | |
584 | struct perf_evlist *evlist = &pevlist->evlist; | |
7e2ed097 | 585 | static char *kwlist[] = {"pages", "overwrite", |
877108e4 ACM |
586 | NULL, NULL}; |
587 | int pages = 128, overwrite = false; | |
588 | ||
7e2ed097 ACM |
589 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ii", kwlist, |
590 | &pages, &overwrite)) | |
877108e4 ACM |
591 | return NULL; |
592 | ||
7e2ed097 | 593 | if (perf_evlist__mmap(evlist, pages, overwrite) < 0) { |
877108e4 ACM |
594 | PyErr_SetFromErrno(PyExc_OSError); |
595 | return NULL; | |
596 | } | |
597 | ||
598 | Py_INCREF(Py_None); | |
599 | return Py_None; | |
600 | } | |
601 | ||
602 | static PyObject *pyrf_evlist__poll(struct pyrf_evlist *pevlist, | |
603 | PyObject *args, PyObject *kwargs) | |
604 | { | |
605 | struct perf_evlist *evlist = &pevlist->evlist; | |
606 | static char *kwlist[] = {"timeout", NULL, NULL}; | |
607 | int timeout = -1, n; | |
608 | ||
609 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|i", kwlist, &timeout)) | |
610 | return NULL; | |
611 | ||
612 | n = poll(evlist->pollfd, evlist->nr_fds, timeout); | |
613 | if (n < 0) { | |
614 | PyErr_SetFromErrno(PyExc_OSError); | |
615 | return NULL; | |
616 | } | |
617 | ||
618 | return Py_BuildValue("i", n); | |
619 | } | |
620 | ||
621 | static PyObject *pyrf_evlist__get_pollfd(struct pyrf_evlist *pevlist, | |
f6bbc1da | 622 | PyObject *args __used, PyObject *kwargs __used) |
877108e4 ACM |
623 | { |
624 | struct perf_evlist *evlist = &pevlist->evlist; | |
625 | PyObject *list = PyList_New(0); | |
626 | int i; | |
627 | ||
628 | for (i = 0; i < evlist->nr_fds; ++i) { | |
629 | PyObject *file; | |
630 | FILE *fp = fdopen(evlist->pollfd[i].fd, "r"); | |
631 | ||
632 | if (fp == NULL) | |
633 | goto free_list; | |
634 | ||
635 | file = PyFile_FromFile(fp, "perf", "r", NULL); | |
636 | if (file == NULL) | |
637 | goto free_list; | |
638 | ||
639 | if (PyList_Append(list, file) != 0) { | |
640 | Py_DECREF(file); | |
641 | goto free_list; | |
642 | } | |
643 | ||
644 | Py_DECREF(file); | |
645 | } | |
646 | ||
647 | return list; | |
648 | free_list: | |
649 | return PyErr_NoMemory(); | |
650 | } | |
651 | ||
652 | ||
653 | static PyObject *pyrf_evlist__add(struct pyrf_evlist *pevlist, | |
f6bbc1da | 654 | PyObject *args, PyObject *kwargs __used) |
877108e4 ACM |
655 | { |
656 | struct perf_evlist *evlist = &pevlist->evlist; | |
657 | PyObject *pevsel; | |
658 | struct perf_evsel *evsel; | |
659 | ||
660 | if (!PyArg_ParseTuple(args, "O", &pevsel)) | |
661 | return NULL; | |
662 | ||
663 | Py_INCREF(pevsel); | |
664 | evsel = &((struct pyrf_evsel *)pevsel)->evsel; | |
665 | evsel->idx = evlist->nr_entries; | |
666 | perf_evlist__add(evlist, evsel); | |
667 | ||
668 | return Py_BuildValue("i", evlist->nr_entries); | |
669 | } | |
670 | ||
671 | static PyObject *pyrf_evlist__read_on_cpu(struct pyrf_evlist *pevlist, | |
672 | PyObject *args, PyObject *kwargs) | |
673 | { | |
674 | struct perf_evlist *evlist = &pevlist->evlist; | |
675 | union perf_event *event; | |
676 | int sample_id_all = 1, cpu; | |
677 | static char *kwlist[] = {"sample_id_all", NULL, NULL}; | |
678 | ||
679 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "i|i", kwlist, | |
680 | &cpu, &sample_id_all)) | |
681 | return NULL; | |
682 | ||
683 | event = perf_evlist__read_on_cpu(evlist, cpu); | |
684 | if (event != NULL) { | |
685 | struct perf_evsel *first; | |
686 | PyObject *pyevent = pyrf_event__new(event); | |
687 | struct pyrf_event *pevent = (struct pyrf_event *)pyevent; | |
688 | ||
689 | if (pyevent == NULL) | |
690 | return PyErr_NoMemory(); | |
691 | ||
692 | first = list_entry(evlist->entries.next, struct perf_evsel, node); | |
a2854124 FW |
693 | perf_event__parse_sample(event, first->attr.sample_type, |
694 | perf_sample_size(first->attr.sample_type), | |
695 | sample_id_all, &pevent->sample); | |
877108e4 ACM |
696 | return pyevent; |
697 | } | |
698 | ||
699 | Py_INCREF(Py_None); | |
700 | return Py_None; | |
701 | } | |
702 | ||
703 | static PyMethodDef pyrf_evlist__methods[] = { | |
704 | { | |
705 | .ml_name = "mmap", | |
706 | .ml_meth = (PyCFunction)pyrf_evlist__mmap, | |
707 | .ml_flags = METH_VARARGS | METH_KEYWORDS, | |
708 | .ml_doc = PyDoc_STR("mmap the file descriptor table.") | |
709 | }, | |
710 | { | |
711 | .ml_name = "poll", | |
712 | .ml_meth = (PyCFunction)pyrf_evlist__poll, | |
713 | .ml_flags = METH_VARARGS | METH_KEYWORDS, | |
714 | .ml_doc = PyDoc_STR("poll the file descriptor table.") | |
715 | }, | |
716 | { | |
717 | .ml_name = "get_pollfd", | |
718 | .ml_meth = (PyCFunction)pyrf_evlist__get_pollfd, | |
719 | .ml_flags = METH_VARARGS | METH_KEYWORDS, | |
720 | .ml_doc = PyDoc_STR("get the poll file descriptor table.") | |
721 | }, | |
722 | { | |
723 | .ml_name = "add", | |
724 | .ml_meth = (PyCFunction)pyrf_evlist__add, | |
725 | .ml_flags = METH_VARARGS | METH_KEYWORDS, | |
726 | .ml_doc = PyDoc_STR("adds an event selector to the list.") | |
727 | }, | |
728 | { | |
729 | .ml_name = "read_on_cpu", | |
730 | .ml_meth = (PyCFunction)pyrf_evlist__read_on_cpu, | |
731 | .ml_flags = METH_VARARGS | METH_KEYWORDS, | |
732 | .ml_doc = PyDoc_STR("reads an event.") | |
733 | }, | |
f6bbc1da | 734 | { .ml_name = NULL, } |
877108e4 ACM |
735 | }; |
736 | ||
737 | static Py_ssize_t pyrf_evlist__length(PyObject *obj) | |
738 | { | |
739 | struct pyrf_evlist *pevlist = (void *)obj; | |
740 | ||
741 | return pevlist->evlist.nr_entries; | |
742 | } | |
743 | ||
744 | static PyObject *pyrf_evlist__item(PyObject *obj, Py_ssize_t i) | |
745 | { | |
746 | struct pyrf_evlist *pevlist = (void *)obj; | |
747 | struct perf_evsel *pos; | |
748 | ||
749 | if (i >= pevlist->evlist.nr_entries) | |
750 | return NULL; | |
751 | ||
752 | list_for_each_entry(pos, &pevlist->evlist.entries, node) | |
753 | if (i-- == 0) | |
754 | break; | |
755 | ||
756 | return Py_BuildValue("O", container_of(pos, struct pyrf_evsel, evsel)); | |
757 | } | |
758 | ||
759 | static PySequenceMethods pyrf_evlist__sequence_methods = { | |
760 | .sq_length = pyrf_evlist__length, | |
761 | .sq_item = pyrf_evlist__item, | |
762 | }; | |
763 | ||
764 | static char pyrf_evlist__doc[] = PyDoc_STR("perf event selector list object."); | |
765 | ||
766 | static PyTypeObject pyrf_evlist__type = { | |
767 | PyVarObject_HEAD_INIT(NULL, 0) | |
768 | .tp_name = "perf.evlist", | |
769 | .tp_basicsize = sizeof(struct pyrf_evlist), | |
770 | .tp_dealloc = (destructor)pyrf_evlist__delete, | |
771 | .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE, | |
772 | .tp_as_sequence = &pyrf_evlist__sequence_methods, | |
773 | .tp_doc = pyrf_evlist__doc, | |
774 | .tp_methods = pyrf_evlist__methods, | |
775 | .tp_init = (initproc)pyrf_evlist__init, | |
776 | }; | |
777 | ||
778 | static int pyrf_evlist__setup_types(void) | |
779 | { | |
780 | pyrf_evlist__type.tp_new = PyType_GenericNew; | |
781 | return PyType_Ready(&pyrf_evlist__type); | |
782 | } | |
783 | ||
784 | static struct { | |
785 | const char *name; | |
786 | int value; | |
787 | } perf__constants[] = { | |
788 | { "TYPE_HARDWARE", PERF_TYPE_HARDWARE }, | |
789 | { "TYPE_SOFTWARE", PERF_TYPE_SOFTWARE }, | |
790 | { "TYPE_TRACEPOINT", PERF_TYPE_TRACEPOINT }, | |
791 | { "TYPE_HW_CACHE", PERF_TYPE_HW_CACHE }, | |
792 | { "TYPE_RAW", PERF_TYPE_RAW }, | |
793 | { "TYPE_BREAKPOINT", PERF_TYPE_BREAKPOINT }, | |
794 | ||
795 | { "COUNT_HW_CPU_CYCLES", PERF_COUNT_HW_CPU_CYCLES }, | |
796 | { "COUNT_HW_INSTRUCTIONS", PERF_COUNT_HW_INSTRUCTIONS }, | |
797 | { "COUNT_HW_CACHE_REFERENCES", PERF_COUNT_HW_CACHE_REFERENCES }, | |
798 | { "COUNT_HW_CACHE_MISSES", PERF_COUNT_HW_CACHE_MISSES }, | |
799 | { "COUNT_HW_BRANCH_INSTRUCTIONS", PERF_COUNT_HW_BRANCH_INSTRUCTIONS }, | |
800 | { "COUNT_HW_BRANCH_MISSES", PERF_COUNT_HW_BRANCH_MISSES }, | |
801 | { "COUNT_HW_BUS_CYCLES", PERF_COUNT_HW_BUS_CYCLES }, | |
802 | { "COUNT_HW_CACHE_L1D", PERF_COUNT_HW_CACHE_L1D }, | |
803 | { "COUNT_HW_CACHE_L1I", PERF_COUNT_HW_CACHE_L1I }, | |
804 | { "COUNT_HW_CACHE_LL", PERF_COUNT_HW_CACHE_LL }, | |
805 | { "COUNT_HW_CACHE_DTLB", PERF_COUNT_HW_CACHE_DTLB }, | |
806 | { "COUNT_HW_CACHE_ITLB", PERF_COUNT_HW_CACHE_ITLB }, | |
807 | { "COUNT_HW_CACHE_BPU", PERF_COUNT_HW_CACHE_BPU }, | |
808 | { "COUNT_HW_CACHE_OP_READ", PERF_COUNT_HW_CACHE_OP_READ }, | |
809 | { "COUNT_HW_CACHE_OP_WRITE", PERF_COUNT_HW_CACHE_OP_WRITE }, | |
810 | { "COUNT_HW_CACHE_OP_PREFETCH", PERF_COUNT_HW_CACHE_OP_PREFETCH }, | |
811 | { "COUNT_HW_CACHE_RESULT_ACCESS", PERF_COUNT_HW_CACHE_RESULT_ACCESS }, | |
812 | { "COUNT_HW_CACHE_RESULT_MISS", PERF_COUNT_HW_CACHE_RESULT_MISS }, | |
813 | ||
129c04cb IM |
814 | { "COUNT_HW_STALLED_CYCLES_FRONTEND", PERF_COUNT_HW_STALLED_CYCLES_FRONTEND }, |
815 | { "COUNT_HW_STALLED_CYCLES_BACKEND", PERF_COUNT_HW_STALLED_CYCLES_BACKEND }, | |
816 | ||
877108e4 ACM |
817 | { "COUNT_SW_CPU_CLOCK", PERF_COUNT_SW_CPU_CLOCK }, |
818 | { "COUNT_SW_TASK_CLOCK", PERF_COUNT_SW_TASK_CLOCK }, | |
819 | { "COUNT_SW_PAGE_FAULTS", PERF_COUNT_SW_PAGE_FAULTS }, | |
820 | { "COUNT_SW_CONTEXT_SWITCHES", PERF_COUNT_SW_CONTEXT_SWITCHES }, | |
821 | { "COUNT_SW_CPU_MIGRATIONS", PERF_COUNT_SW_CPU_MIGRATIONS }, | |
822 | { "COUNT_SW_PAGE_FAULTS_MIN", PERF_COUNT_SW_PAGE_FAULTS_MIN }, | |
823 | { "COUNT_SW_PAGE_FAULTS_MAJ", PERF_COUNT_SW_PAGE_FAULTS_MAJ }, | |
824 | { "COUNT_SW_ALIGNMENT_FAULTS", PERF_COUNT_SW_ALIGNMENT_FAULTS }, | |
825 | { "COUNT_SW_EMULATION_FAULTS", PERF_COUNT_SW_EMULATION_FAULTS }, | |
826 | ||
827 | { "SAMPLE_IP", PERF_SAMPLE_IP }, | |
828 | { "SAMPLE_TID", PERF_SAMPLE_TID }, | |
829 | { "SAMPLE_TIME", PERF_SAMPLE_TIME }, | |
830 | { "SAMPLE_ADDR", PERF_SAMPLE_ADDR }, | |
831 | { "SAMPLE_READ", PERF_SAMPLE_READ }, | |
832 | { "SAMPLE_CALLCHAIN", PERF_SAMPLE_CALLCHAIN }, | |
833 | { "SAMPLE_ID", PERF_SAMPLE_ID }, | |
834 | { "SAMPLE_CPU", PERF_SAMPLE_CPU }, | |
835 | { "SAMPLE_PERIOD", PERF_SAMPLE_PERIOD }, | |
836 | { "SAMPLE_STREAM_ID", PERF_SAMPLE_STREAM_ID }, | |
837 | { "SAMPLE_RAW", PERF_SAMPLE_RAW }, | |
838 | ||
839 | { "FORMAT_TOTAL_TIME_ENABLED", PERF_FORMAT_TOTAL_TIME_ENABLED }, | |
840 | { "FORMAT_TOTAL_TIME_RUNNING", PERF_FORMAT_TOTAL_TIME_RUNNING }, | |
841 | { "FORMAT_ID", PERF_FORMAT_ID }, | |
842 | { "FORMAT_GROUP", PERF_FORMAT_GROUP }, | |
843 | ||
844 | { "RECORD_MMAP", PERF_RECORD_MMAP }, | |
845 | { "RECORD_LOST", PERF_RECORD_LOST }, | |
846 | { "RECORD_COMM", PERF_RECORD_COMM }, | |
847 | { "RECORD_EXIT", PERF_RECORD_EXIT }, | |
848 | { "RECORD_THROTTLE", PERF_RECORD_THROTTLE }, | |
849 | { "RECORD_UNTHROTTLE", PERF_RECORD_UNTHROTTLE }, | |
850 | { "RECORD_FORK", PERF_RECORD_FORK }, | |
851 | { "RECORD_READ", PERF_RECORD_READ }, | |
852 | { "RECORD_SAMPLE", PERF_RECORD_SAMPLE }, | |
f6bbc1da | 853 | { .name = NULL, }, |
877108e4 ACM |
854 | }; |
855 | ||
856 | static PyMethodDef perf__methods[] = { | |
f6bbc1da | 857 | { .ml_name = NULL, } |
877108e4 ACM |
858 | }; |
859 | ||
860 | PyMODINIT_FUNC initperf(void) | |
861 | { | |
862 | PyObject *obj; | |
863 | int i; | |
864 | PyObject *dict, *module = Py_InitModule("perf", perf__methods); | |
865 | ||
866 | if (module == NULL || | |
867 | pyrf_event__setup_types() < 0 || | |
868 | pyrf_evlist__setup_types() < 0 || | |
869 | pyrf_evsel__setup_types() < 0 || | |
870 | pyrf_thread_map__setup_types() < 0 || | |
871 | pyrf_cpu_map__setup_types() < 0) | |
872 | return; | |
873 | ||
874 | Py_INCREF(&pyrf_evlist__type); | |
875 | PyModule_AddObject(module, "evlist", (PyObject*)&pyrf_evlist__type); | |
876 | ||
877 | Py_INCREF(&pyrf_evsel__type); | |
878 | PyModule_AddObject(module, "evsel", (PyObject*)&pyrf_evsel__type); | |
879 | ||
880 | Py_INCREF(&pyrf_thread_map__type); | |
881 | PyModule_AddObject(module, "thread_map", (PyObject*)&pyrf_thread_map__type); | |
882 | ||
883 | Py_INCREF(&pyrf_cpu_map__type); | |
884 | PyModule_AddObject(module, "cpu_map", (PyObject*)&pyrf_cpu_map__type); | |
885 | ||
886 | dict = PyModule_GetDict(module); | |
887 | if (dict == NULL) | |
888 | goto error; | |
889 | ||
890 | for (i = 0; perf__constants[i].name != NULL; i++) { | |
891 | obj = PyInt_FromLong(perf__constants[i].value); | |
892 | if (obj == NULL) | |
893 | goto error; | |
894 | PyDict_SetItemString(dict, perf__constants[i].name, obj); | |
895 | Py_DECREF(obj); | |
896 | } | |
897 | ||
898 | error: | |
899 | if (PyErr_Occurred()) | |
900 | PyErr_SetString(PyExc_ImportError, "perf: Init failed!"); | |
901 | } |