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 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 16struct throttle_event { 17 struct perf_event_header header; 18 u64 time; 19 u64 id; 20 u64 stream_id; 21}; 22 23PyMODINIT_FUNC initperf(void); 24 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 35struct pyrf_event { 36 PyObject_HEAD 37 struct perf_sample sample; 38 union perf_event event; 39}; 40 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"), 51 52static char pyrf_mmap_event__doc[] = PyDoc_STR("perf mmap event object."); 53 54static 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, }, 64}; 65 66static 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 85static 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 95static char pyrf_task_event__doc[] = PyDoc_STR("perf task (fork/exit) event object."); 96 97static 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, }, 106}; 107 108static 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 120static 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 130static char pyrf_comm_event__doc[] = PyDoc_STR("perf comm event object."); 131 132static 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, }, 139}; 140 141static 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 149static 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 159static char pyrf_throttle_event__doc[] = PyDoc_STR("perf throttle event object."); 160 161static 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, }, 168}; 169 170static 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 180static 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 190static 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; 209out: 210 return err; 211} 212 213static 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 225static 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 241struct pyrf_cpu_map { 242 PyObject_HEAD 243 244 struct cpu_map *cpus; 245}; 246 247static 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 263static 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 269static 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 276static 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 286static PySequenceMethods pyrf_cpu_map__sequence_methods = { 287 .sq_length = pyrf_cpu_map__length, 288 .sq_item = pyrf_cpu_map__item, 289}; 290 291static char pyrf_cpu_map__doc[] = PyDoc_STR("cpu map object."); 292 293static 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 304static 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 310struct pyrf_thread_map { 311 PyObject_HEAD 312 313 struct thread_map *threads; 314}; 315 316static 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 332static 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 338static 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 345static 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 355static PySequenceMethods pyrf_thread_map__sequence_methods = { 356 .sq_length = pyrf_thread_map__length, 357 .sq_item = pyrf_thread_map__item, 358}; 359 360static char pyrf_thread_map__doc[] = PyDoc_STR("thread map object."); 361 362static 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 373static 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 379struct pyrf_evsel { 380 PyObject_HEAD 381 382 struct perf_evsel evsel; 383}; 384 385static 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 488static 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 494static 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; 501 int group = 0, inherit = 0; 502 static char *kwlist[] = {"cpus", "threads", "group", "inherit", NULL, NULL}; 503 504 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|OOii", kwlist, 505 &pcpus, &pthreads, &group, &inherit)) 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 514 evsel->attr.inherit = inherit; 515 if (perf_evsel__open(evsel, cpus, threads, group) < 0) { 516 PyErr_SetFromErrno(PyExc_OSError); 517 return NULL; 518 } 519 520 Py_INCREF(Py_None); 521 return Py_None; 522} 523 524static 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 }, 531 { .ml_name = NULL, } 532}; 533 534static char pyrf_evsel__doc[] = PyDoc_STR("perf event selector list object."); 535 536static 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 547static int pyrf_evsel__setup_types(void) 548{ 549 pyrf_evsel__type.tp_new = PyType_GenericNew; 550 return PyType_Ready(&pyrf_evsel__type); 551} 552 553struct pyrf_evlist { 554 PyObject_HEAD 555 556 struct perf_evlist evlist; 557}; 558 559static int pyrf_evlist__init(struct pyrf_evlist *pevlist, 560 PyObject *args, PyObject *kwargs __used) 561{ 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); 572 return 0; 573} 574 575static 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 581static PyObject *pyrf_evlist__mmap(struct pyrf_evlist *pevlist, 582 PyObject *args, PyObject *kwargs) 583{ 584 struct perf_evlist *evlist = &pevlist->evlist; 585 static char *kwlist[] = {"pages", "overwrite", 586 NULL, NULL}; 587 int pages = 128, overwrite = false; 588 589 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ii", kwlist, 590 &pages, &overwrite)) 591 return NULL; 592 593 if (perf_evlist__mmap(evlist, pages, overwrite) < 0) { 594 PyErr_SetFromErrno(PyExc_OSError); 595 return NULL; 596 } 597 598 Py_INCREF(Py_None); 599 return Py_None; 600} 601 602static 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 621static PyObject *pyrf_evlist__get_pollfd(struct pyrf_evlist *pevlist, 622 PyObject *args __used, PyObject *kwargs __used) 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; 648free_list: 649 return PyErr_NoMemory(); 650} 651 652 653static PyObject *pyrf_evlist__add(struct pyrf_evlist *pevlist, 654 PyObject *args, PyObject *kwargs __used) 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 671static 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[] = {"cpu", "sample_id_all", NULL, NULL}; 678 int err; 679 680 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "i|i", kwlist, 681 &cpu, &sample_id_all)) 682 return NULL; 683 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; 689 690 if (pyevent == NULL) 691 return PyErr_NoMemory(); 692 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; 701 } 702 703 Py_INCREF(Py_None); 704 return Py_None; 705} 706 707static PyMethodDef pyrf_evlist__methods[] = { 708 { 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.") 713 }, 714 { 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.") 719 }, 720 { 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.") 725 }, 726 { 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.") 731 }, 732 { 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.") 737 }, 738 { .ml_name = NULL, } 739}; 740 741static Py_ssize_t pyrf_evlist__length(PyObject *obj) 742{ 743 struct pyrf_evlist *pevlist = (void *)obj; 744 745 return pevlist->evlist.nr_entries; 746} 747 748static PyObject *pyrf_evlist__item(PyObject *obj, Py_ssize_t i) 749{ 750 struct pyrf_evlist *pevlist = (void *)obj; 751 struct perf_evsel *pos; 752 753 if (i >= pevlist->evlist.nr_entries) 754 return NULL; 755 756 list_for_each_entry(pos, &pevlist->evlist.entries, node) 757 if (i-- == 0) 758 break; 759 760 return Py_BuildValue("O", container_of(pos, struct pyrf_evsel, evsel)); 761} 762 763static PySequenceMethods pyrf_evlist__sequence_methods = { 764 .sq_length = pyrf_evlist__length, 765 .sq_item = pyrf_evlist__item, 766}; 767 768static char pyrf_evlist__doc[] = PyDoc_STR("perf event selector list object."); 769 770static 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, 780}; 781 782static int pyrf_evlist__setup_types(void) 783{ 784 pyrf_evlist__type.tp_new = PyType_GenericNew; 785 return PyType_Ready(&pyrf_evlist__type); 786} 787 788static 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 }, 798 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 }, 817 818 { "COUNT_HW_STALLED_CYCLES_FRONTEND", PERF_COUNT_HW_STALLED_CYCLES_FRONTEND }, 819 { "COUNT_HW_STALLED_CYCLES_BACKEND", PERF_COUNT_HW_STALLED_CYCLES_BACKEND }, 820 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 }, 830 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 }, 842 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 }, 847 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, }, 858}; 859 860static PyMethodDef perf__methods[] = { 861 { .ml_name = NULL, } 862}; 863 864PyMODINIT_FUNC initperf(void) 865{ 866 PyObject *obj; 867 int i; 868 PyObject *dict, *module = Py_InitModule("perf", perf__methods); 869 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; 877 878 Py_INCREF(&pyrf_evlist__type); 879 PyModule_AddObject(module, "evlist", (PyObject*)&pyrf_evlist__type); 880 881 Py_INCREF(&pyrf_evsel__type); 882 PyModule_AddObject(module, "evsel", (PyObject*)&pyrf_evsel__type); 883 884 Py_INCREF(&pyrf_thread_map__type); 885 PyModule_AddObject(module, "thread_map", (PyObject*)&pyrf_thread_map__type); 886 887 Py_INCREF(&pyrf_cpu_map__type); 888 PyModule_AddObject(module, "cpu_map", (PyObject*)&pyrf_cpu_map__type); 889 890 dict = PyModule_GetDict(module); 891 if (dict == NULL) 892 goto error; 893 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); 900 } 901 902error: 903 if (PyErr_Occurred()) 904 PyErr_SetString(PyExc_ImportError, "perf: Init failed!"); 905} 906