]>
Commit | Line | Data |
---|---|---|
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 | */ | |
16 | int verbose; | |
17 | ||
c95688aa | 18 | int 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 |
37 | PyMODINIT_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 | ||
49 | struct 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 | ||
67 | static char pyrf_mmap_event__doc[] = PyDoc_STR("perf mmap event object."); | |
68 | ||
69 | static 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 | ||
82 | static 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 | ||
101 | static 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 | ||
111 | static char pyrf_task_event__doc[] = PyDoc_STR("perf task (fork/exit) event object."); | |
112 | ||
113 | static 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 | ||
124 | static 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 | ||
136 | static 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 | ||
146 | static char pyrf_comm_event__doc[] = PyDoc_STR("perf comm event object."); | |
147 | ||
148 | static 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 | ||
157 | static 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 | ||
165 | static 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 | ||
175 | static char pyrf_throttle_event__doc[] = PyDoc_STR("perf throttle event object."); | |
176 | ||
177 | static 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 | ||
186 | static 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 | ||
196 | static 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 |
206 | static char pyrf_lost_event__doc[] = PyDoc_STR("perf lost event object."); |
207 | ||
208 | static 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 | ||
215 | static 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 | ||
231 | static 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 | ||
241 | static char pyrf_read_event__doc[] = PyDoc_STR("perf read event object."); | |
242 | ||
243 | static 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 | ||
250 | static 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 | ||
261 | static 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 | ||
271 | static char pyrf_sample_event__doc[] = PyDoc_STR("perf sample event object."); | |
272 | ||
273 | static PyMemberDef pyrf_sample_event__members[] = { | |
274 | sample_members | |
275 | member_def(perf_event_header, type, T_UINT, "event type"), | |
276 | { .name = NULL, }, | |
277 | }; | |
278 | ||
279 | static 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 | ||
293 | static PyTypeObject pyrf_sample_event__type = { | |
294 | PyVarObject_HEAD_INIT(NULL, 0) | |
295 | .tp_name = "perf.sample_event", | |
296 | .tp_basicsize = sizeof(struct pyrf_event), | |
297 | .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE, | |
298 | .tp_doc = pyrf_sample_event__doc, | |
299 | .tp_members = pyrf_sample_event__members, | |
300 | .tp_repr = (reprfunc)pyrf_sample_event__repr, | |
301 | }; | |
302 | ||
ae938802 ACM |
303 | static char pyrf_context_switch_event__doc[] = PyDoc_STR("perf context_switch event object."); |
304 | ||
305 | static PyMemberDef pyrf_context_switch_event__members[] = { | |
306 | sample_members | |
307 | member_def(perf_event_header, type, T_UINT, "event type"), | |
308 | member_def(context_switch_event, next_prev_pid, T_UINT, "next/prev pid"), | |
309 | member_def(context_switch_event, next_prev_tid, T_UINT, "next/prev tid"), | |
310 | { .name = NULL, }, | |
311 | }; | |
312 | ||
313 | static PyObject *pyrf_context_switch_event__repr(struct pyrf_event *pevent) | |
314 | { | |
315 | PyObject *ret; | |
316 | char *s; | |
317 | ||
318 | if (asprintf(&s, "{ type: context_switch, next_prev_pid: %u, next_prev_tid: %u, switch_out: %u }", | |
319 | pevent->event.context_switch.next_prev_pid, | |
320 | pevent->event.context_switch.next_prev_tid, | |
321 | !!(pevent->event.header.misc & PERF_RECORD_MISC_SWITCH_OUT)) < 0) { | |
322 | ret = PyErr_NoMemory(); | |
323 | } else { | |
324 | ret = PyString_FromString(s); | |
325 | free(s); | |
326 | } | |
327 | return ret; | |
328 | } | |
329 | ||
330 | static PyTypeObject pyrf_context_switch_event__type = { | |
331 | PyVarObject_HEAD_INIT(NULL, 0) | |
332 | .tp_name = "perf.context_switch_event", | |
333 | .tp_basicsize = sizeof(struct pyrf_event), | |
334 | .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE, | |
335 | .tp_doc = pyrf_context_switch_event__doc, | |
336 | .tp_members = pyrf_context_switch_event__members, | |
337 | .tp_repr = (reprfunc)pyrf_context_switch_event__repr, | |
338 | }; | |
339 | ||
877108e4 ACM |
340 | static int pyrf_event__setup_types(void) |
341 | { | |
342 | int err; | |
343 | pyrf_mmap_event__type.tp_new = | |
344 | pyrf_task_event__type.tp_new = | |
345 | pyrf_comm_event__type.tp_new = | |
3e9f45a7 ACM |
346 | pyrf_lost_event__type.tp_new = |
347 | pyrf_read_event__type.tp_new = | |
348 | pyrf_sample_event__type.tp_new = | |
ae938802 | 349 | pyrf_context_switch_event__type.tp_new = |
877108e4 ACM |
350 | pyrf_throttle_event__type.tp_new = PyType_GenericNew; |
351 | err = PyType_Ready(&pyrf_mmap_event__type); | |
3e9f45a7 ACM |
352 | if (err < 0) |
353 | goto out; | |
354 | err = PyType_Ready(&pyrf_lost_event__type); | |
877108e4 ACM |
355 | if (err < 0) |
356 | goto out; | |
357 | err = PyType_Ready(&pyrf_task_event__type); | |
358 | if (err < 0) | |
359 | goto out; | |
360 | err = PyType_Ready(&pyrf_comm_event__type); | |
361 | if (err < 0) | |
362 | goto out; | |
363 | err = PyType_Ready(&pyrf_throttle_event__type); | |
364 | if (err < 0) | |
365 | goto out; | |
3e9f45a7 ACM |
366 | err = PyType_Ready(&pyrf_read_event__type); |
367 | if (err < 0) | |
368 | goto out; | |
369 | err = PyType_Ready(&pyrf_sample_event__type); | |
370 | if (err < 0) | |
371 | goto out; | |
ae938802 ACM |
372 | err = PyType_Ready(&pyrf_context_switch_event__type); |
373 | if (err < 0) | |
374 | goto out; | |
877108e4 ACM |
375 | out: |
376 | return err; | |
377 | } | |
378 | ||
379 | static PyTypeObject *pyrf_event__type[] = { | |
380 | [PERF_RECORD_MMAP] = &pyrf_mmap_event__type, | |
3e9f45a7 | 381 | [PERF_RECORD_LOST] = &pyrf_lost_event__type, |
877108e4 ACM |
382 | [PERF_RECORD_COMM] = &pyrf_comm_event__type, |
383 | [PERF_RECORD_EXIT] = &pyrf_task_event__type, | |
384 | [PERF_RECORD_THROTTLE] = &pyrf_throttle_event__type, | |
385 | [PERF_RECORD_UNTHROTTLE] = &pyrf_throttle_event__type, | |
386 | [PERF_RECORD_FORK] = &pyrf_task_event__type, | |
3e9f45a7 ACM |
387 | [PERF_RECORD_READ] = &pyrf_read_event__type, |
388 | [PERF_RECORD_SAMPLE] = &pyrf_sample_event__type, | |
ae938802 ACM |
389 | [PERF_RECORD_SWITCH] = &pyrf_context_switch_event__type, |
390 | [PERF_RECORD_SWITCH_CPU_WIDE] = &pyrf_context_switch_event__type, | |
877108e4 ACM |
391 | }; |
392 | ||
393 | static PyObject *pyrf_event__new(union perf_event *event) | |
394 | { | |
395 | struct pyrf_event *pevent; | |
396 | PyTypeObject *ptype; | |
397 | ||
ae938802 ACM |
398 | if ((event->header.type < PERF_RECORD_MMAP || |
399 | event->header.type > PERF_RECORD_SAMPLE) && | |
400 | !(event->header.type == PERF_RECORD_SWITCH || | |
401 | event->header.type == PERF_RECORD_SWITCH_CPU_WIDE)) | |
877108e4 ACM |
402 | return NULL; |
403 | ||
404 | ptype = pyrf_event__type[event->header.type]; | |
405 | pevent = PyObject_New(struct pyrf_event, ptype); | |
406 | if (pevent != NULL) | |
407 | memcpy(&pevent->event, event, event->header.size); | |
408 | return (PyObject *)pevent; | |
409 | } | |
410 | ||
411 | struct pyrf_cpu_map { | |
412 | PyObject_HEAD | |
413 | ||
414 | struct cpu_map *cpus; | |
415 | }; | |
416 | ||
417 | static int pyrf_cpu_map__init(struct pyrf_cpu_map *pcpus, | |
418 | PyObject *args, PyObject *kwargs) | |
419 | { | |
64348153 | 420 | static char *kwlist[] = { "cpustr", NULL }; |
877108e4 ACM |
421 | char *cpustr = NULL; |
422 | ||
423 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|s", | |
424 | kwlist, &cpustr)) | |
425 | return -1; | |
426 | ||
427 | pcpus->cpus = cpu_map__new(cpustr); | |
428 | if (pcpus->cpus == NULL) | |
429 | return -1; | |
430 | return 0; | |
431 | } | |
432 | ||
433 | static void pyrf_cpu_map__delete(struct pyrf_cpu_map *pcpus) | |
434 | { | |
f30a79b0 | 435 | cpu_map__put(pcpus->cpus); |
877108e4 ACM |
436 | pcpus->ob_type->tp_free((PyObject*)pcpus); |
437 | } | |
438 | ||
439 | static Py_ssize_t pyrf_cpu_map__length(PyObject *obj) | |
440 | { | |
441 | struct pyrf_cpu_map *pcpus = (void *)obj; | |
442 | ||
443 | return pcpus->cpus->nr; | |
444 | } | |
445 | ||
446 | static PyObject *pyrf_cpu_map__item(PyObject *obj, Py_ssize_t i) | |
447 | { | |
448 | struct pyrf_cpu_map *pcpus = (void *)obj; | |
449 | ||
450 | if (i >= pcpus->cpus->nr) | |
451 | return NULL; | |
452 | ||
453 | return Py_BuildValue("i", pcpus->cpus->map[i]); | |
454 | } | |
455 | ||
456 | static PySequenceMethods pyrf_cpu_map__sequence_methods = { | |
457 | .sq_length = pyrf_cpu_map__length, | |
458 | .sq_item = pyrf_cpu_map__item, | |
459 | }; | |
460 | ||
461 | static char pyrf_cpu_map__doc[] = PyDoc_STR("cpu map object."); | |
462 | ||
463 | static PyTypeObject pyrf_cpu_map__type = { | |
464 | PyVarObject_HEAD_INIT(NULL, 0) | |
465 | .tp_name = "perf.cpu_map", | |
466 | .tp_basicsize = sizeof(struct pyrf_cpu_map), | |
467 | .tp_dealloc = (destructor)pyrf_cpu_map__delete, | |
468 | .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE, | |
469 | .tp_doc = pyrf_cpu_map__doc, | |
470 | .tp_as_sequence = &pyrf_cpu_map__sequence_methods, | |
471 | .tp_init = (initproc)pyrf_cpu_map__init, | |
472 | }; | |
473 | ||
474 | static int pyrf_cpu_map__setup_types(void) | |
475 | { | |
476 | pyrf_cpu_map__type.tp_new = PyType_GenericNew; | |
477 | return PyType_Ready(&pyrf_cpu_map__type); | |
478 | } | |
479 | ||
480 | struct pyrf_thread_map { | |
481 | PyObject_HEAD | |
482 | ||
483 | struct thread_map *threads; | |
484 | }; | |
485 | ||
486 | static int pyrf_thread_map__init(struct pyrf_thread_map *pthreads, | |
487 | PyObject *args, PyObject *kwargs) | |
488 | { | |
0d37aa34 ACM |
489 | static char *kwlist[] = { "pid", "tid", "uid", NULL }; |
490 | int pid = -1, tid = -1, uid = UINT_MAX; | |
877108e4 | 491 | |
0d37aa34 ACM |
492 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|iii", |
493 | kwlist, &pid, &tid, &uid)) | |
877108e4 ACM |
494 | return -1; |
495 | ||
0d37aa34 | 496 | pthreads->threads = thread_map__new(pid, tid, uid); |
877108e4 ACM |
497 | if (pthreads->threads == NULL) |
498 | return -1; | |
499 | return 0; | |
500 | } | |
501 | ||
502 | static void pyrf_thread_map__delete(struct pyrf_thread_map *pthreads) | |
503 | { | |
186fbb74 | 504 | thread_map__put(pthreads->threads); |
877108e4 ACM |
505 | pthreads->ob_type->tp_free((PyObject*)pthreads); |
506 | } | |
507 | ||
508 | static Py_ssize_t pyrf_thread_map__length(PyObject *obj) | |
509 | { | |
510 | struct pyrf_thread_map *pthreads = (void *)obj; | |
511 | ||
512 | return pthreads->threads->nr; | |
513 | } | |
514 | ||
515 | static PyObject *pyrf_thread_map__item(PyObject *obj, Py_ssize_t i) | |
516 | { | |
517 | struct pyrf_thread_map *pthreads = (void *)obj; | |
518 | ||
519 | if (i >= pthreads->threads->nr) | |
520 | return NULL; | |
521 | ||
522 | return Py_BuildValue("i", pthreads->threads->map[i]); | |
523 | } | |
524 | ||
525 | static PySequenceMethods pyrf_thread_map__sequence_methods = { | |
526 | .sq_length = pyrf_thread_map__length, | |
527 | .sq_item = pyrf_thread_map__item, | |
528 | }; | |
529 | ||
530 | static char pyrf_thread_map__doc[] = PyDoc_STR("thread map object."); | |
531 | ||
532 | static PyTypeObject pyrf_thread_map__type = { | |
533 | PyVarObject_HEAD_INIT(NULL, 0) | |
534 | .tp_name = "perf.thread_map", | |
535 | .tp_basicsize = sizeof(struct pyrf_thread_map), | |
536 | .tp_dealloc = (destructor)pyrf_thread_map__delete, | |
537 | .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE, | |
538 | .tp_doc = pyrf_thread_map__doc, | |
539 | .tp_as_sequence = &pyrf_thread_map__sequence_methods, | |
540 | .tp_init = (initproc)pyrf_thread_map__init, | |
541 | }; | |
542 | ||
543 | static int pyrf_thread_map__setup_types(void) | |
544 | { | |
545 | pyrf_thread_map__type.tp_new = PyType_GenericNew; | |
546 | return PyType_Ready(&pyrf_thread_map__type); | |
547 | } | |
548 | ||
549 | struct pyrf_evsel { | |
550 | PyObject_HEAD | |
551 | ||
552 | struct perf_evsel evsel; | |
553 | }; | |
554 | ||
555 | static int pyrf_evsel__init(struct pyrf_evsel *pevsel, | |
556 | PyObject *args, PyObject *kwargs) | |
557 | { | |
558 | struct perf_event_attr attr = { | |
559 | .type = PERF_TYPE_HARDWARE, | |
560 | .config = PERF_COUNT_HW_CPU_CYCLES, | |
561 | .sample_type = PERF_SAMPLE_PERIOD | PERF_SAMPLE_TID, | |
562 | }; | |
563 | static char *kwlist[] = { | |
564 | "type", | |
565 | "config", | |
566 | "sample_freq", | |
567 | "sample_period", | |
568 | "sample_type", | |
569 | "read_format", | |
570 | "disabled", | |
571 | "inherit", | |
572 | "pinned", | |
573 | "exclusive", | |
574 | "exclude_user", | |
575 | "exclude_kernel", | |
576 | "exclude_hv", | |
577 | "exclude_idle", | |
578 | "mmap", | |
ae938802 | 579 | "context_switch", |
877108e4 ACM |
580 | "comm", |
581 | "freq", | |
582 | "inherit_stat", | |
583 | "enable_on_exec", | |
584 | "task", | |
585 | "watermark", | |
586 | "precise_ip", | |
587 | "mmap_data", | |
588 | "sample_id_all", | |
589 | "wakeup_events", | |
590 | "bp_type", | |
591 | "bp_addr", | |
64348153 FW |
592 | "bp_len", |
593 | NULL | |
594 | }; | |
877108e4 ACM |
595 | u64 sample_period = 0; |
596 | u32 disabled = 0, | |
597 | inherit = 0, | |
598 | pinned = 0, | |
599 | exclusive = 0, | |
600 | exclude_user = 0, | |
601 | exclude_kernel = 0, | |
602 | exclude_hv = 0, | |
603 | exclude_idle = 0, | |
604 | mmap = 0, | |
ae938802 | 605 | context_switch = 0, |
877108e4 ACM |
606 | comm = 0, |
607 | freq = 1, | |
608 | inherit_stat = 0, | |
609 | enable_on_exec = 0, | |
610 | task = 0, | |
611 | watermark = 0, | |
612 | precise_ip = 0, | |
613 | mmap_data = 0, | |
614 | sample_id_all = 1; | |
615 | int idx = 0; | |
616 | ||
617 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, | |
ae938802 | 618 | "|iKiKKiiiiiiiiiiiiiiiiiiiiiiKK", kwlist, |
877108e4 ACM |
619 | &attr.type, &attr.config, &attr.sample_freq, |
620 | &sample_period, &attr.sample_type, | |
621 | &attr.read_format, &disabled, &inherit, | |
622 | &pinned, &exclusive, &exclude_user, | |
623 | &exclude_kernel, &exclude_hv, &exclude_idle, | |
ae938802 | 624 | &mmap, &context_switch, &comm, &freq, &inherit_stat, |
877108e4 ACM |
625 | &enable_on_exec, &task, &watermark, |
626 | &precise_ip, &mmap_data, &sample_id_all, | |
627 | &attr.wakeup_events, &attr.bp_type, | |
628 | &attr.bp_addr, &attr.bp_len, &idx)) | |
629 | return -1; | |
630 | ||
631 | /* union... */ | |
632 | if (sample_period != 0) { | |
633 | if (attr.sample_freq != 0) | |
634 | return -1; /* FIXME: throw right exception */ | |
635 | attr.sample_period = sample_period; | |
636 | } | |
637 | ||
638 | /* Bitfields */ | |
639 | attr.disabled = disabled; | |
640 | attr.inherit = inherit; | |
641 | attr.pinned = pinned; | |
642 | attr.exclusive = exclusive; | |
643 | attr.exclude_user = exclude_user; | |
644 | attr.exclude_kernel = exclude_kernel; | |
645 | attr.exclude_hv = exclude_hv; | |
646 | attr.exclude_idle = exclude_idle; | |
647 | attr.mmap = mmap; | |
ae938802 | 648 | attr.context_switch = context_switch; |
877108e4 ACM |
649 | attr.comm = comm; |
650 | attr.freq = freq; | |
651 | attr.inherit_stat = inherit_stat; | |
652 | attr.enable_on_exec = enable_on_exec; | |
653 | attr.task = task; | |
654 | attr.watermark = watermark; | |
655 | attr.precise_ip = precise_ip; | |
656 | attr.mmap_data = mmap_data; | |
657 | attr.sample_id_all = sample_id_all; | |
ad4e3c04 | 658 | attr.size = sizeof(attr); |
877108e4 ACM |
659 | |
660 | perf_evsel__init(&pevsel->evsel, &attr, idx); | |
661 | return 0; | |
662 | } | |
663 | ||
664 | static void pyrf_evsel__delete(struct pyrf_evsel *pevsel) | |
665 | { | |
666 | perf_evsel__exit(&pevsel->evsel); | |
667 | pevsel->ob_type->tp_free((PyObject*)pevsel); | |
668 | } | |
669 | ||
670 | static PyObject *pyrf_evsel__open(struct pyrf_evsel *pevsel, | |
671 | PyObject *args, PyObject *kwargs) | |
672 | { | |
673 | struct perf_evsel *evsel = &pevsel->evsel; | |
674 | struct cpu_map *cpus = NULL; | |
675 | struct thread_map *threads = NULL; | |
676 | PyObject *pcpus = NULL, *pthreads = NULL; | |
5d2cd909 | 677 | int group = 0, inherit = 0; |
64348153 | 678 | static char *kwlist[] = { "cpus", "threads", "group", "inherit", NULL }; |
877108e4 ACM |
679 | |
680 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|OOii", kwlist, | |
5d2cd909 | 681 | &pcpus, &pthreads, &group, &inherit)) |
877108e4 ACM |
682 | return NULL; |
683 | ||
684 | if (pthreads != NULL) | |
685 | threads = ((struct pyrf_thread_map *)pthreads)->threads; | |
686 | ||
687 | if (pcpus != NULL) | |
688 | cpus = ((struct pyrf_cpu_map *)pcpus)->cpus; | |
689 | ||
5d2cd909 | 690 | evsel->attr.inherit = inherit; |
727ab04e ACM |
691 | /* |
692 | * This will group just the fds for this single evsel, to group | |
693 | * multiple events, use evlist.open(). | |
694 | */ | |
6a4bb04c | 695 | if (perf_evsel__open(evsel, cpus, threads) < 0) { |
877108e4 ACM |
696 | PyErr_SetFromErrno(PyExc_OSError); |
697 | return NULL; | |
698 | } | |
699 | ||
700 | Py_INCREF(Py_None); | |
701 | return Py_None; | |
702 | } | |
703 | ||
704 | static PyMethodDef pyrf_evsel__methods[] = { | |
705 | { | |
706 | .ml_name = "open", | |
707 | .ml_meth = (PyCFunction)pyrf_evsel__open, | |
708 | .ml_flags = METH_VARARGS | METH_KEYWORDS, | |
709 | .ml_doc = PyDoc_STR("open the event selector file descriptor table.") | |
710 | }, | |
f6bbc1da | 711 | { .ml_name = NULL, } |
877108e4 ACM |
712 | }; |
713 | ||
714 | static char pyrf_evsel__doc[] = PyDoc_STR("perf event selector list object."); | |
715 | ||
716 | static PyTypeObject pyrf_evsel__type = { | |
717 | PyVarObject_HEAD_INIT(NULL, 0) | |
718 | .tp_name = "perf.evsel", | |
719 | .tp_basicsize = sizeof(struct pyrf_evsel), | |
720 | .tp_dealloc = (destructor)pyrf_evsel__delete, | |
721 | .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE, | |
722 | .tp_doc = pyrf_evsel__doc, | |
723 | .tp_methods = pyrf_evsel__methods, | |
724 | .tp_init = (initproc)pyrf_evsel__init, | |
725 | }; | |
726 | ||
727 | static int pyrf_evsel__setup_types(void) | |
728 | { | |
729 | pyrf_evsel__type.tp_new = PyType_GenericNew; | |
730 | return PyType_Ready(&pyrf_evsel__type); | |
731 | } | |
732 | ||
733 | struct pyrf_evlist { | |
734 | PyObject_HEAD | |
735 | ||
736 | struct perf_evlist evlist; | |
737 | }; | |
738 | ||
739 | static int pyrf_evlist__init(struct pyrf_evlist *pevlist, | |
1d037ca1 | 740 | PyObject *args, PyObject *kwargs __maybe_unused) |
877108e4 | 741 | { |
7e2ed097 ACM |
742 | PyObject *pcpus = NULL, *pthreads = NULL; |
743 | struct cpu_map *cpus; | |
744 | struct thread_map *threads; | |
745 | ||
746 | if (!PyArg_ParseTuple(args, "OO", &pcpus, &pthreads)) | |
747 | return -1; | |
748 | ||
749 | threads = ((struct pyrf_thread_map *)pthreads)->threads; | |
750 | cpus = ((struct pyrf_cpu_map *)pcpus)->cpus; | |
751 | perf_evlist__init(&pevlist->evlist, cpus, threads); | |
877108e4 ACM |
752 | return 0; |
753 | } | |
754 | ||
755 | static void pyrf_evlist__delete(struct pyrf_evlist *pevlist) | |
756 | { | |
757 | perf_evlist__exit(&pevlist->evlist); | |
758 | pevlist->ob_type->tp_free((PyObject*)pevlist); | |
759 | } | |
760 | ||
761 | static PyObject *pyrf_evlist__mmap(struct pyrf_evlist *pevlist, | |
762 | PyObject *args, PyObject *kwargs) | |
763 | { | |
764 | struct perf_evlist *evlist = &pevlist->evlist; | |
64348153 | 765 | static char *kwlist[] = { "pages", "overwrite", NULL }; |
877108e4 ACM |
766 | int pages = 128, overwrite = false; |
767 | ||
7e2ed097 ACM |
768 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ii", kwlist, |
769 | &pages, &overwrite)) | |
877108e4 ACM |
770 | return NULL; |
771 | ||
7e2ed097 | 772 | if (perf_evlist__mmap(evlist, pages, overwrite) < 0) { |
877108e4 ACM |
773 | PyErr_SetFromErrno(PyExc_OSError); |
774 | return NULL; | |
775 | } | |
776 | ||
777 | Py_INCREF(Py_None); | |
778 | return Py_None; | |
779 | } | |
780 | ||
781 | static PyObject *pyrf_evlist__poll(struct pyrf_evlist *pevlist, | |
782 | PyObject *args, PyObject *kwargs) | |
783 | { | |
784 | struct perf_evlist *evlist = &pevlist->evlist; | |
64348153 | 785 | static char *kwlist[] = { "timeout", NULL }; |
877108e4 ACM |
786 | int timeout = -1, n; |
787 | ||
788 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|i", kwlist, &timeout)) | |
789 | return NULL; | |
790 | ||
f66a889d | 791 | n = perf_evlist__poll(evlist, timeout); |
877108e4 ACM |
792 | if (n < 0) { |
793 | PyErr_SetFromErrno(PyExc_OSError); | |
794 | return NULL; | |
795 | } | |
796 | ||
797 | return Py_BuildValue("i", n); | |
798 | } | |
799 | ||
800 | static PyObject *pyrf_evlist__get_pollfd(struct pyrf_evlist *pevlist, | |
1d037ca1 IT |
801 | PyObject *args __maybe_unused, |
802 | PyObject *kwargs __maybe_unused) | |
877108e4 ACM |
803 | { |
804 | struct perf_evlist *evlist = &pevlist->evlist; | |
805 | PyObject *list = PyList_New(0); | |
806 | int i; | |
807 | ||
1b85337d | 808 | for (i = 0; i < evlist->pollfd.nr; ++i) { |
877108e4 | 809 | PyObject *file; |
1b85337d | 810 | FILE *fp = fdopen(evlist->pollfd.entries[i].fd, "r"); |
877108e4 ACM |
811 | |
812 | if (fp == NULL) | |
813 | goto free_list; | |
814 | ||
815 | file = PyFile_FromFile(fp, "perf", "r", NULL); | |
816 | if (file == NULL) | |
817 | goto free_list; | |
818 | ||
819 | if (PyList_Append(list, file) != 0) { | |
820 | Py_DECREF(file); | |
821 | goto free_list; | |
822 | } | |
48000a1a | 823 | |
877108e4 ACM |
824 | Py_DECREF(file); |
825 | } | |
826 | ||
827 | return list; | |
828 | free_list: | |
829 | return PyErr_NoMemory(); | |
830 | } | |
831 | ||
832 | ||
833 | static PyObject *pyrf_evlist__add(struct pyrf_evlist *pevlist, | |
1d037ca1 IT |
834 | PyObject *args, |
835 | PyObject *kwargs __maybe_unused) | |
877108e4 ACM |
836 | { |
837 | struct perf_evlist *evlist = &pevlist->evlist; | |
838 | PyObject *pevsel; | |
839 | struct perf_evsel *evsel; | |
840 | ||
841 | if (!PyArg_ParseTuple(args, "O", &pevsel)) | |
842 | return NULL; | |
843 | ||
844 | Py_INCREF(pevsel); | |
845 | evsel = &((struct pyrf_evsel *)pevsel)->evsel; | |
846 | evsel->idx = evlist->nr_entries; | |
847 | perf_evlist__add(evlist, evsel); | |
848 | ||
849 | return Py_BuildValue("i", evlist->nr_entries); | |
850 | } | |
851 | ||
852 | static PyObject *pyrf_evlist__read_on_cpu(struct pyrf_evlist *pevlist, | |
853 | PyObject *args, PyObject *kwargs) | |
854 | { | |
855 | struct perf_evlist *evlist = &pevlist->evlist; | |
856 | union perf_event *event; | |
857 | int sample_id_all = 1, cpu; | |
64348153 | 858 | static char *kwlist[] = { "cpu", "sample_id_all", NULL }; |
5538beca | 859 | int err; |
877108e4 ACM |
860 | |
861 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "i|i", kwlist, | |
862 | &cpu, &sample_id_all)) | |
863 | return NULL; | |
864 | ||
aece948f | 865 | event = perf_evlist__mmap_read(evlist, cpu); |
877108e4 | 866 | if (event != NULL) { |
877108e4 ACM |
867 | PyObject *pyevent = pyrf_event__new(event); |
868 | struct pyrf_event *pevent = (struct pyrf_event *)pyevent; | |
377f698d | 869 | struct perf_evsel *evsel; |
877108e4 ACM |
870 | |
871 | if (pyevent == NULL) | |
872 | return PyErr_NoMemory(); | |
873 | ||
377f698d JO |
874 | evsel = perf_evlist__event2evsel(evlist, event); |
875 | if (!evsel) | |
876 | return Py_None; | |
877 | ||
878 | pevent->evsel = evsel; | |
879 | ||
880 | err = perf_evsel__parse_sample(evsel, event, &pevent->sample); | |
e8968e65 JO |
881 | |
882 | /* Consume the even only after we parsed it out. */ | |
883 | perf_evlist__mmap_consume(evlist, cpu); | |
884 | ||
5c6970af ACM |
885 | if (err) |
886 | return PyErr_Format(PyExc_OSError, | |
887 | "perf: can't parse sample, err=%d", err); | |
877108e4 ACM |
888 | return pyevent; |
889 | } | |
5c6970af | 890 | |
877108e4 ACM |
891 | Py_INCREF(Py_None); |
892 | return Py_None; | |
893 | } | |
894 | ||
727ab04e ACM |
895 | static PyObject *pyrf_evlist__open(struct pyrf_evlist *pevlist, |
896 | PyObject *args, PyObject *kwargs) | |
897 | { | |
898 | struct perf_evlist *evlist = &pevlist->evlist; | |
899 | int group = 0; | |
900 | static char *kwlist[] = { "group", NULL }; | |
901 | ||
902 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|OOii", kwlist, &group)) | |
903 | return NULL; | |
904 | ||
6a4bb04c | 905 | if (group) |
63dab225 | 906 | perf_evlist__set_leader(evlist); |
6a4bb04c JO |
907 | |
908 | if (perf_evlist__open(evlist) < 0) { | |
727ab04e ACM |
909 | PyErr_SetFromErrno(PyExc_OSError); |
910 | return NULL; | |
911 | } | |
912 | ||
913 | Py_INCREF(Py_None); | |
914 | return Py_None; | |
915 | } | |
916 | ||
877108e4 ACM |
917 | static PyMethodDef pyrf_evlist__methods[] = { |
918 | { | |
919 | .ml_name = "mmap", | |
920 | .ml_meth = (PyCFunction)pyrf_evlist__mmap, | |
921 | .ml_flags = METH_VARARGS | METH_KEYWORDS, | |
922 | .ml_doc = PyDoc_STR("mmap the file descriptor table.") | |
923 | }, | |
727ab04e ACM |
924 | { |
925 | .ml_name = "open", | |
926 | .ml_meth = (PyCFunction)pyrf_evlist__open, | |
927 | .ml_flags = METH_VARARGS | METH_KEYWORDS, | |
928 | .ml_doc = PyDoc_STR("open the file descriptors.") | |
929 | }, | |
877108e4 ACM |
930 | { |
931 | .ml_name = "poll", | |
932 | .ml_meth = (PyCFunction)pyrf_evlist__poll, | |
933 | .ml_flags = METH_VARARGS | METH_KEYWORDS, | |
934 | .ml_doc = PyDoc_STR("poll the file descriptor table.") | |
935 | }, | |
936 | { | |
937 | .ml_name = "get_pollfd", | |
938 | .ml_meth = (PyCFunction)pyrf_evlist__get_pollfd, | |
939 | .ml_flags = METH_VARARGS | METH_KEYWORDS, | |
940 | .ml_doc = PyDoc_STR("get the poll file descriptor table.") | |
941 | }, | |
942 | { | |
943 | .ml_name = "add", | |
944 | .ml_meth = (PyCFunction)pyrf_evlist__add, | |
945 | .ml_flags = METH_VARARGS | METH_KEYWORDS, | |
946 | .ml_doc = PyDoc_STR("adds an event selector to the list.") | |
947 | }, | |
948 | { | |
949 | .ml_name = "read_on_cpu", | |
950 | .ml_meth = (PyCFunction)pyrf_evlist__read_on_cpu, | |
951 | .ml_flags = METH_VARARGS | METH_KEYWORDS, | |
952 | .ml_doc = PyDoc_STR("reads an event.") | |
953 | }, | |
f6bbc1da | 954 | { .ml_name = NULL, } |
877108e4 ACM |
955 | }; |
956 | ||
957 | static Py_ssize_t pyrf_evlist__length(PyObject *obj) | |
958 | { | |
959 | struct pyrf_evlist *pevlist = (void *)obj; | |
960 | ||
961 | return pevlist->evlist.nr_entries; | |
962 | } | |
963 | ||
964 | static PyObject *pyrf_evlist__item(PyObject *obj, Py_ssize_t i) | |
965 | { | |
966 | struct pyrf_evlist *pevlist = (void *)obj; | |
967 | struct perf_evsel *pos; | |
968 | ||
969 | if (i >= pevlist->evlist.nr_entries) | |
970 | return NULL; | |
971 | ||
e5cadb93 | 972 | evlist__for_each_entry(&pevlist->evlist, pos) { |
877108e4 ACM |
973 | if (i-- == 0) |
974 | break; | |
0050f7aa | 975 | } |
877108e4 ACM |
976 | |
977 | return Py_BuildValue("O", container_of(pos, struct pyrf_evsel, evsel)); | |
978 | } | |
979 | ||
980 | static PySequenceMethods pyrf_evlist__sequence_methods = { | |
981 | .sq_length = pyrf_evlist__length, | |
982 | .sq_item = pyrf_evlist__item, | |
983 | }; | |
984 | ||
985 | static char pyrf_evlist__doc[] = PyDoc_STR("perf event selector list object."); | |
986 | ||
987 | static PyTypeObject pyrf_evlist__type = { | |
988 | PyVarObject_HEAD_INIT(NULL, 0) | |
989 | .tp_name = "perf.evlist", | |
990 | .tp_basicsize = sizeof(struct pyrf_evlist), | |
991 | .tp_dealloc = (destructor)pyrf_evlist__delete, | |
992 | .tp_flags = Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE, | |
993 | .tp_as_sequence = &pyrf_evlist__sequence_methods, | |
994 | .tp_doc = pyrf_evlist__doc, | |
995 | .tp_methods = pyrf_evlist__methods, | |
996 | .tp_init = (initproc)pyrf_evlist__init, | |
997 | }; | |
998 | ||
999 | static int pyrf_evlist__setup_types(void) | |
1000 | { | |
1001 | pyrf_evlist__type.tp_new = PyType_GenericNew; | |
1002 | return PyType_Ready(&pyrf_evlist__type); | |
1003 | } | |
1004 | ||
5865fe36 ACM |
1005 | #define PERF_CONST(name) { #name, PERF_##name } |
1006 | ||
877108e4 ACM |
1007 | static struct { |
1008 | const char *name; | |
1009 | int value; | |
1010 | } perf__constants[] = { | |
5865fe36 ACM |
1011 | PERF_CONST(TYPE_HARDWARE), |
1012 | PERF_CONST(TYPE_SOFTWARE), | |
1013 | PERF_CONST(TYPE_TRACEPOINT), | |
1014 | PERF_CONST(TYPE_HW_CACHE), | |
1015 | PERF_CONST(TYPE_RAW), | |
1016 | PERF_CONST(TYPE_BREAKPOINT), | |
1017 | ||
1018 | PERF_CONST(COUNT_HW_CPU_CYCLES), | |
1019 | PERF_CONST(COUNT_HW_INSTRUCTIONS), | |
1020 | PERF_CONST(COUNT_HW_CACHE_REFERENCES), | |
1021 | PERF_CONST(COUNT_HW_CACHE_MISSES), | |
1022 | PERF_CONST(COUNT_HW_BRANCH_INSTRUCTIONS), | |
1023 | PERF_CONST(COUNT_HW_BRANCH_MISSES), | |
1024 | PERF_CONST(COUNT_HW_BUS_CYCLES), | |
1025 | PERF_CONST(COUNT_HW_CACHE_L1D), | |
1026 | PERF_CONST(COUNT_HW_CACHE_L1I), | |
1027 | PERF_CONST(COUNT_HW_CACHE_LL), | |
1028 | PERF_CONST(COUNT_HW_CACHE_DTLB), | |
1029 | PERF_CONST(COUNT_HW_CACHE_ITLB), | |
1030 | PERF_CONST(COUNT_HW_CACHE_BPU), | |
1031 | PERF_CONST(COUNT_HW_CACHE_OP_READ), | |
1032 | PERF_CONST(COUNT_HW_CACHE_OP_WRITE), | |
1033 | PERF_CONST(COUNT_HW_CACHE_OP_PREFETCH), | |
1034 | PERF_CONST(COUNT_HW_CACHE_RESULT_ACCESS), | |
1035 | PERF_CONST(COUNT_HW_CACHE_RESULT_MISS), | |
1036 | ||
1037 | PERF_CONST(COUNT_HW_STALLED_CYCLES_FRONTEND), | |
1038 | PERF_CONST(COUNT_HW_STALLED_CYCLES_BACKEND), | |
1039 | ||
1040 | PERF_CONST(COUNT_SW_CPU_CLOCK), | |
1041 | PERF_CONST(COUNT_SW_TASK_CLOCK), | |
1042 | PERF_CONST(COUNT_SW_PAGE_FAULTS), | |
1043 | PERF_CONST(COUNT_SW_CONTEXT_SWITCHES), | |
1044 | PERF_CONST(COUNT_SW_CPU_MIGRATIONS), | |
1045 | PERF_CONST(COUNT_SW_PAGE_FAULTS_MIN), | |
1046 | PERF_CONST(COUNT_SW_PAGE_FAULTS_MAJ), | |
1047 | PERF_CONST(COUNT_SW_ALIGNMENT_FAULTS), | |
1048 | PERF_CONST(COUNT_SW_EMULATION_FAULTS), | |
1049 | PERF_CONST(COUNT_SW_DUMMY), | |
1050 | ||
1051 | PERF_CONST(SAMPLE_IP), | |
1052 | PERF_CONST(SAMPLE_TID), | |
1053 | PERF_CONST(SAMPLE_TIME), | |
1054 | PERF_CONST(SAMPLE_ADDR), | |
1055 | PERF_CONST(SAMPLE_READ), | |
1056 | PERF_CONST(SAMPLE_CALLCHAIN), | |
1057 | PERF_CONST(SAMPLE_ID), | |
1058 | PERF_CONST(SAMPLE_CPU), | |
1059 | PERF_CONST(SAMPLE_PERIOD), | |
1060 | PERF_CONST(SAMPLE_STREAM_ID), | |
1061 | PERF_CONST(SAMPLE_RAW), | |
1062 | ||
1063 | PERF_CONST(FORMAT_TOTAL_TIME_ENABLED), | |
1064 | PERF_CONST(FORMAT_TOTAL_TIME_RUNNING), | |
1065 | PERF_CONST(FORMAT_ID), | |
1066 | PERF_CONST(FORMAT_GROUP), | |
1067 | ||
1068 | PERF_CONST(RECORD_MMAP), | |
1069 | PERF_CONST(RECORD_LOST), | |
1070 | PERF_CONST(RECORD_COMM), | |
1071 | PERF_CONST(RECORD_EXIT), | |
1072 | PERF_CONST(RECORD_THROTTLE), | |
1073 | PERF_CONST(RECORD_UNTHROTTLE), | |
1074 | PERF_CONST(RECORD_FORK), | |
1075 | PERF_CONST(RECORD_READ), | |
1076 | PERF_CONST(RECORD_SAMPLE), | |
84576da2 ACM |
1077 | PERF_CONST(RECORD_MMAP2), |
1078 | PERF_CONST(RECORD_AUX), | |
1079 | PERF_CONST(RECORD_ITRACE_START), | |
1080 | PERF_CONST(RECORD_LOST_SAMPLES), | |
1081 | PERF_CONST(RECORD_SWITCH), | |
1082 | PERF_CONST(RECORD_SWITCH_CPU_WIDE), | |
ae938802 ACM |
1083 | |
1084 | PERF_CONST(RECORD_MISC_SWITCH_OUT), | |
f6bbc1da | 1085 | { .name = NULL, }, |
877108e4 ACM |
1086 | }; |
1087 | ||
1075fbb2 JO |
1088 | static PyObject *pyrf__tracepoint(struct pyrf_evsel *pevsel, |
1089 | PyObject *args, PyObject *kwargs) | |
1090 | { | |
1091 | struct event_format *tp_format; | |
1092 | static char *kwlist[] = { "sys", "name", NULL }; | |
1093 | char *sys = NULL; | |
1094 | char *name = NULL; | |
1095 | ||
1096 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ss", kwlist, | |
1097 | &sys, &name)) | |
1098 | return NULL; | |
1099 | ||
1100 | tp_format = trace_event__tp_format(sys, name); | |
1101 | if (IS_ERR(tp_format)) | |
1102 | return PyInt_FromLong(-1); | |
1103 | ||
1104 | return PyInt_FromLong(tp_format->id); | |
1105 | } | |
1106 | ||
877108e4 | 1107 | static PyMethodDef perf__methods[] = { |
1075fbb2 JO |
1108 | { |
1109 | .ml_name = "tracepoint", | |
1110 | .ml_meth = (PyCFunction) pyrf__tracepoint, | |
1111 | .ml_flags = METH_VARARGS | METH_KEYWORDS, | |
1112 | .ml_doc = PyDoc_STR("Get tracepoint config.") | |
1113 | }, | |
f6bbc1da | 1114 | { .ml_name = NULL, } |
877108e4 ACM |
1115 | }; |
1116 | ||
1117 | PyMODINIT_FUNC initperf(void) | |
1118 | { | |
1119 | PyObject *obj; | |
1120 | int i; | |
1121 | PyObject *dict, *module = Py_InitModule("perf", perf__methods); | |
1122 | ||
1123 | if (module == NULL || | |
1124 | pyrf_event__setup_types() < 0 || | |
1125 | pyrf_evlist__setup_types() < 0 || | |
1126 | pyrf_evsel__setup_types() < 0 || | |
1127 | pyrf_thread_map__setup_types() < 0 || | |
1128 | pyrf_cpu_map__setup_types() < 0) | |
1129 | return; | |
1130 | ||
918512b4 | 1131 | /* The page_size is placed in util object. */ |
0da2e9c2 ACM |
1132 | page_size = sysconf(_SC_PAGE_SIZE); |
1133 | ||
877108e4 ACM |
1134 | Py_INCREF(&pyrf_evlist__type); |
1135 | PyModule_AddObject(module, "evlist", (PyObject*)&pyrf_evlist__type); | |
1136 | ||
1137 | Py_INCREF(&pyrf_evsel__type); | |
1138 | PyModule_AddObject(module, "evsel", (PyObject*)&pyrf_evsel__type); | |
1139 | ||
85e37de3 JO |
1140 | Py_INCREF(&pyrf_mmap_event__type); |
1141 | PyModule_AddObject(module, "mmap_event", (PyObject *)&pyrf_mmap_event__type); | |
1142 | ||
1143 | Py_INCREF(&pyrf_lost_event__type); | |
1144 | PyModule_AddObject(module, "lost_event", (PyObject *)&pyrf_lost_event__type); | |
1145 | ||
1146 | Py_INCREF(&pyrf_comm_event__type); | |
1147 | PyModule_AddObject(module, "comm_event", (PyObject *)&pyrf_comm_event__type); | |
1148 | ||
1149 | Py_INCREF(&pyrf_task_event__type); | |
1150 | PyModule_AddObject(module, "task_event", (PyObject *)&pyrf_task_event__type); | |
1151 | ||
1152 | Py_INCREF(&pyrf_throttle_event__type); | |
1153 | PyModule_AddObject(module, "throttle_event", (PyObject *)&pyrf_throttle_event__type); | |
1154 | ||
1155 | Py_INCREF(&pyrf_task_event__type); | |
1156 | PyModule_AddObject(module, "task_event", (PyObject *)&pyrf_task_event__type); | |
1157 | ||
1158 | Py_INCREF(&pyrf_read_event__type); | |
1159 | PyModule_AddObject(module, "read_event", (PyObject *)&pyrf_read_event__type); | |
1160 | ||
1161 | Py_INCREF(&pyrf_sample_event__type); | |
1162 | PyModule_AddObject(module, "sample_event", (PyObject *)&pyrf_sample_event__type); | |
1163 | ||
1164 | Py_INCREF(&pyrf_context_switch_event__type); | |
1165 | PyModule_AddObject(module, "switch_event", (PyObject *)&pyrf_context_switch_event__type); | |
1166 | ||
877108e4 ACM |
1167 | Py_INCREF(&pyrf_thread_map__type); |
1168 | PyModule_AddObject(module, "thread_map", (PyObject*)&pyrf_thread_map__type); | |
1169 | ||
1170 | Py_INCREF(&pyrf_cpu_map__type); | |
1171 | PyModule_AddObject(module, "cpu_map", (PyObject*)&pyrf_cpu_map__type); | |
1172 | ||
1173 | dict = PyModule_GetDict(module); | |
1174 | if (dict == NULL) | |
1175 | goto error; | |
1176 | ||
1177 | for (i = 0; perf__constants[i].name != NULL; i++) { | |
1178 | obj = PyInt_FromLong(perf__constants[i].value); | |
1179 | if (obj == NULL) | |
1180 | goto error; | |
1181 | PyDict_SetItemString(dict, perf__constants[i].name, obj); | |
1182 | Py_DECREF(obj); | |
1183 | } | |
1184 | ||
1185 | error: | |
1186 | if (PyErr_Occurred()) | |
1187 | PyErr_SetString(PyExc_ImportError, "perf: Init failed!"); | |
1188 | } | |
0c6332e9 ACM |
1189 | |
1190 | /* | |
1191 | * Dummy, to avoid dragging all the test_attr infrastructure in the python | |
1192 | * binding. | |
1193 | */ | |
1194 | void test_attr__open(struct perf_event_attr *attr, pid_t pid, int cpu, | |
1195 | int fd, int group_fd, unsigned long flags) | |
1196 | { | |
1197 | } |