xref: /linux/tools/perf/util/python.c (revision 0019dc87b928380e2c85fcd8694e512079d7b580)
1b2441318SGreg Kroah-Hartman // SPDX-License-Identifier: GPL-2.0
2877108e4SArnaldo Carvalho de Melo #include <Python.h>
3877108e4SArnaldo Carvalho de Melo #include <structmember.h>
4877108e4SArnaldo Carvalho de Melo #include <inttypes.h>
5877108e4SArnaldo Carvalho de Melo #include <poll.h>
61075fbb2SJiri Olsa #include <linux/err.h>
7877108e4SArnaldo Carvalho de Melo #include "evlist.h"
856e2e056SArnaldo Carvalho de Melo #include "callchain.h"
9877108e4SArnaldo Carvalho de Melo #include "evsel.h"
10877108e4SArnaldo Carvalho de Melo #include "event.h"
11877108e4SArnaldo Carvalho de Melo #include "cpumap.h"
12fea01392SArnaldo Carvalho de Melo #include "print_binary.h"
13877108e4SArnaldo Carvalho de Melo #include "thread_map.h"
14877108e4SArnaldo Carvalho de Melo 
1566dfdff0SJaroslav Škarvada #if PY_MAJOR_VERSION < 3
1666dfdff0SJaroslav Škarvada #define _PyUnicode_FromString(arg) \
1766dfdff0SJaroslav Škarvada   PyString_FromString(arg)
1866dfdff0SJaroslav Škarvada #define _PyUnicode_AsString(arg) \
1966dfdff0SJaroslav Škarvada   PyString_AsString(arg)
2066dfdff0SJaroslav Škarvada #define _PyUnicode_FromFormat(...) \
2166dfdff0SJaroslav Škarvada   PyString_FromFormat(__VA_ARGS__)
2266dfdff0SJaroslav Škarvada #define _PyLong_FromLong(arg) \
2366dfdff0SJaroslav Škarvada   PyInt_FromLong(arg)
2466dfdff0SJaroslav Škarvada 
2566dfdff0SJaroslav Škarvada #else
2666dfdff0SJaroslav Škarvada 
2766dfdff0SJaroslav Škarvada #define _PyUnicode_FromString(arg) \
2866dfdff0SJaroslav Škarvada   PyUnicode_FromString(arg)
2966dfdff0SJaroslav Škarvada #define _PyUnicode_FromFormat(...) \
3066dfdff0SJaroslav Škarvada   PyUnicode_FromFormat(__VA_ARGS__)
3166dfdff0SJaroslav Škarvada #define _PyLong_FromLong(arg) \
3266dfdff0SJaroslav Škarvada   PyLong_FromLong(arg)
3366dfdff0SJaroslav Škarvada #endif
3466dfdff0SJaroslav Škarvada 
3566dfdff0SJaroslav Škarvada #ifndef Py_TYPE
3666dfdff0SJaroslav Škarvada #define Py_TYPE(ob) (((PyObject*)(ob))->ob_type)
3766dfdff0SJaroslav Škarvada #endif
3866dfdff0SJaroslav Škarvada 
398afb4c01SAdrian Hunter /*
4056e2e056SArnaldo Carvalho de Melo  * Provide these two so that we don't have to link against callchain.c and
4156e2e056SArnaldo Carvalho de Melo  * start dragging hist.c, etc.
4256e2e056SArnaldo Carvalho de Melo  */
4356e2e056SArnaldo Carvalho de Melo struct callchain_param callchain_param;
4456e2e056SArnaldo Carvalho de Melo 
4556e2e056SArnaldo Carvalho de Melo int parse_callchain_record(const char *arg __maybe_unused,
4656e2e056SArnaldo Carvalho de Melo 			   struct callchain_param *param __maybe_unused)
4756e2e056SArnaldo Carvalho de Melo {
4856e2e056SArnaldo Carvalho de Melo 	return 0;
4956e2e056SArnaldo Carvalho de Melo }
5056e2e056SArnaldo Carvalho de Melo 
5156e2e056SArnaldo Carvalho de Melo /*
528afb4c01SAdrian Hunter  * Support debug printing even though util/debug.c is not linked.  That means
538afb4c01SAdrian Hunter  * implementing 'verbose' and 'eprintf'.
548afb4c01SAdrian Hunter  */
558afb4c01SAdrian Hunter int verbose;
568afb4c01SAdrian Hunter 
57c95688aaSJiri Olsa int eprintf(int level, int var, const char *fmt, ...)
588afb4c01SAdrian Hunter {
598afb4c01SAdrian Hunter 	va_list args;
608afb4c01SAdrian Hunter 	int ret = 0;
618afb4c01SAdrian Hunter 
62c95688aaSJiri Olsa 	if (var >= level) {
638afb4c01SAdrian Hunter 		va_start(args, fmt);
648afb4c01SAdrian Hunter 		ret = vfprintf(stderr, fmt, args);
658afb4c01SAdrian Hunter 		va_end(args);
668afb4c01SAdrian Hunter 	}
678afb4c01SAdrian Hunter 
688afb4c01SAdrian Hunter 	return ret;
698afb4c01SAdrian Hunter }
708afb4c01SAdrian Hunter 
71cfff2d90SFrederic Weisbecker /* Define PyVarObject_HEAD_INIT for python 2.5 */
72cfff2d90SFrederic Weisbecker #ifndef PyVarObject_HEAD_INIT
73cfff2d90SFrederic Weisbecker # define PyVarObject_HEAD_INIT(type, size) PyObject_HEAD_INIT(type) size,
74cfff2d90SFrederic Weisbecker #endif
75cfff2d90SFrederic Weisbecker 
7666dfdff0SJaroslav Škarvada #if PY_MAJOR_VERSION < 3
77f6bbc1daSArnaldo Carvalho de Melo PyMODINIT_FUNC initperf(void);
7866dfdff0SJaroslav Škarvada #else
7966dfdff0SJaroslav Škarvada PyMODINIT_FUNC PyInit_perf(void);
8066dfdff0SJaroslav Škarvada #endif
81f6bbc1daSArnaldo Carvalho de Melo 
82877108e4SArnaldo Carvalho de Melo #define member_def(type, member, ptype, help) \
83877108e4SArnaldo Carvalho de Melo 	{ #member, ptype, \
84877108e4SArnaldo Carvalho de Melo 	  offsetof(struct pyrf_event, event) + offsetof(struct type, member), \
85877108e4SArnaldo Carvalho de Melo 	  0, help }
86877108e4SArnaldo Carvalho de Melo 
87877108e4SArnaldo Carvalho de Melo #define sample_member_def(name, member, ptype, help) \
88877108e4SArnaldo Carvalho de Melo 	{ #name, ptype, \
89877108e4SArnaldo Carvalho de Melo 	  offsetof(struct pyrf_event, sample) + offsetof(struct perf_sample, member), \
90877108e4SArnaldo Carvalho de Melo 	  0, help }
91877108e4SArnaldo Carvalho de Melo 
92877108e4SArnaldo Carvalho de Melo struct pyrf_event {
93877108e4SArnaldo Carvalho de Melo 	PyObject_HEAD
94377f698dSJiri Olsa 	struct perf_evsel *evsel;
95877108e4SArnaldo Carvalho de Melo 	struct perf_sample sample;
96877108e4SArnaldo Carvalho de Melo 	union perf_event   event;
97877108e4SArnaldo Carvalho de Melo };
98877108e4SArnaldo Carvalho de Melo 
99877108e4SArnaldo Carvalho de Melo #define sample_members \
100f6bbc1daSArnaldo Carvalho de Melo 	sample_member_def(sample_ip, ip, T_ULONGLONG, "event type"),			 \
101877108e4SArnaldo Carvalho de Melo 	sample_member_def(sample_pid, pid, T_INT, "event pid"),			 \
102877108e4SArnaldo Carvalho de Melo 	sample_member_def(sample_tid, tid, T_INT, "event tid"),			 \
103f6bbc1daSArnaldo Carvalho de Melo 	sample_member_def(sample_time, time, T_ULONGLONG, "event timestamp"),		 \
104f6bbc1daSArnaldo Carvalho de Melo 	sample_member_def(sample_addr, addr, T_ULONGLONG, "event addr"),		 \
105f6bbc1daSArnaldo Carvalho de Melo 	sample_member_def(sample_id, id, T_ULONGLONG, "event id"),			 \
106f6bbc1daSArnaldo Carvalho de Melo 	sample_member_def(sample_stream_id, stream_id, T_ULONGLONG, "event stream id"), \
107f6bbc1daSArnaldo Carvalho de Melo 	sample_member_def(sample_period, period, T_ULONGLONG, "event period"),		 \
108877108e4SArnaldo Carvalho de Melo 	sample_member_def(sample_cpu, cpu, T_UINT, "event cpu"),
109877108e4SArnaldo Carvalho de Melo 
110877108e4SArnaldo Carvalho de Melo static char pyrf_mmap_event__doc[] = PyDoc_STR("perf mmap event object.");
111877108e4SArnaldo Carvalho de Melo 
112877108e4SArnaldo Carvalho de Melo static PyMemberDef pyrf_mmap_event__members[] = {
113877108e4SArnaldo Carvalho de Melo 	sample_members
114877108e4SArnaldo Carvalho de Melo 	member_def(perf_event_header, type, T_UINT, "event type"),
115ae938802SArnaldo Carvalho de Melo 	member_def(perf_event_header, misc, T_UINT, "event misc"),
116877108e4SArnaldo Carvalho de Melo 	member_def(mmap_event, pid, T_UINT, "event pid"),
117877108e4SArnaldo Carvalho de Melo 	member_def(mmap_event, tid, T_UINT, "event tid"),
118f6bbc1daSArnaldo Carvalho de Melo 	member_def(mmap_event, start, T_ULONGLONG, "start of the map"),
119f6bbc1daSArnaldo Carvalho de Melo 	member_def(mmap_event, len, T_ULONGLONG, "map length"),
120f6bbc1daSArnaldo Carvalho de Melo 	member_def(mmap_event, pgoff, T_ULONGLONG, "page offset"),
121877108e4SArnaldo Carvalho de Melo 	member_def(mmap_event, filename, T_STRING_INPLACE, "backing store"),
122f6bbc1daSArnaldo Carvalho de Melo 	{ .name = NULL, },
123877108e4SArnaldo Carvalho de Melo };
124877108e4SArnaldo Carvalho de Melo 
125877108e4SArnaldo Carvalho de Melo static PyObject *pyrf_mmap_event__repr(struct pyrf_event *pevent)
126877108e4SArnaldo Carvalho de Melo {
127877108e4SArnaldo Carvalho de Melo 	PyObject *ret;
128877108e4SArnaldo Carvalho de Melo 	char *s;
129877108e4SArnaldo Carvalho de Melo 
130877108e4SArnaldo Carvalho de Melo 	if (asprintf(&s, "{ type: mmap, pid: %u, tid: %u, start: %#" PRIx64 ", "
131877108e4SArnaldo Carvalho de Melo 			 "length: %#" PRIx64 ", offset: %#" PRIx64 ", "
132877108e4SArnaldo Carvalho de Melo 			 "filename: %s }",
133877108e4SArnaldo Carvalho de Melo 		     pevent->event.mmap.pid, pevent->event.mmap.tid,
134877108e4SArnaldo Carvalho de Melo 		     pevent->event.mmap.start, pevent->event.mmap.len,
135877108e4SArnaldo Carvalho de Melo 		     pevent->event.mmap.pgoff, pevent->event.mmap.filename) < 0) {
136877108e4SArnaldo Carvalho de Melo 		ret = PyErr_NoMemory();
137877108e4SArnaldo Carvalho de Melo 	} else {
13866dfdff0SJaroslav Škarvada 		ret = _PyUnicode_FromString(s);
139877108e4SArnaldo Carvalho de Melo 		free(s);
140877108e4SArnaldo Carvalho de Melo 	}
141877108e4SArnaldo Carvalho de Melo 	return ret;
142877108e4SArnaldo Carvalho de Melo }
143877108e4SArnaldo Carvalho de Melo 
144877108e4SArnaldo Carvalho de Melo static PyTypeObject pyrf_mmap_event__type = {
145877108e4SArnaldo Carvalho de Melo 	PyVarObject_HEAD_INIT(NULL, 0)
146877108e4SArnaldo Carvalho de Melo 	.tp_name	= "perf.mmap_event",
147877108e4SArnaldo Carvalho de Melo 	.tp_basicsize	= sizeof(struct pyrf_event),
148877108e4SArnaldo Carvalho de Melo 	.tp_flags	= Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
149877108e4SArnaldo Carvalho de Melo 	.tp_doc		= pyrf_mmap_event__doc,
150877108e4SArnaldo Carvalho de Melo 	.tp_members	= pyrf_mmap_event__members,
151877108e4SArnaldo Carvalho de Melo 	.tp_repr	= (reprfunc)pyrf_mmap_event__repr,
152877108e4SArnaldo Carvalho de Melo };
153877108e4SArnaldo Carvalho de Melo 
154877108e4SArnaldo Carvalho de Melo static char pyrf_task_event__doc[] = PyDoc_STR("perf task (fork/exit) event object.");
155877108e4SArnaldo Carvalho de Melo 
156877108e4SArnaldo Carvalho de Melo static PyMemberDef pyrf_task_event__members[] = {
157877108e4SArnaldo Carvalho de Melo 	sample_members
158877108e4SArnaldo Carvalho de Melo 	member_def(perf_event_header, type, T_UINT, "event type"),
159877108e4SArnaldo Carvalho de Melo 	member_def(fork_event, pid, T_UINT, "event pid"),
160877108e4SArnaldo Carvalho de Melo 	member_def(fork_event, ppid, T_UINT, "event ppid"),
161877108e4SArnaldo Carvalho de Melo 	member_def(fork_event, tid, T_UINT, "event tid"),
162877108e4SArnaldo Carvalho de Melo 	member_def(fork_event, ptid, T_UINT, "event ptid"),
163f6bbc1daSArnaldo Carvalho de Melo 	member_def(fork_event, time, T_ULONGLONG, "timestamp"),
164f6bbc1daSArnaldo Carvalho de Melo 	{ .name = NULL, },
165877108e4SArnaldo Carvalho de Melo };
166877108e4SArnaldo Carvalho de Melo 
167877108e4SArnaldo Carvalho de Melo static PyObject *pyrf_task_event__repr(struct pyrf_event *pevent)
168877108e4SArnaldo Carvalho de Melo {
16966dfdff0SJaroslav Škarvada 	return _PyUnicode_FromFormat("{ type: %s, pid: %u, ppid: %u, tid: %u, "
170877108e4SArnaldo Carvalho de Melo 				   "ptid: %u, time: %" PRIu64 "}",
171877108e4SArnaldo Carvalho de Melo 				   pevent->event.header.type == PERF_RECORD_FORK ? "fork" : "exit",
172877108e4SArnaldo Carvalho de Melo 				   pevent->event.fork.pid,
173877108e4SArnaldo Carvalho de Melo 				   pevent->event.fork.ppid,
174877108e4SArnaldo Carvalho de Melo 				   pevent->event.fork.tid,
175877108e4SArnaldo Carvalho de Melo 				   pevent->event.fork.ptid,
176877108e4SArnaldo Carvalho de Melo 				   pevent->event.fork.time);
177877108e4SArnaldo Carvalho de Melo }
178877108e4SArnaldo Carvalho de Melo 
179877108e4SArnaldo Carvalho de Melo static PyTypeObject pyrf_task_event__type = {
180877108e4SArnaldo Carvalho de Melo 	PyVarObject_HEAD_INIT(NULL, 0)
181877108e4SArnaldo Carvalho de Melo 	.tp_name	= "perf.task_event",
182877108e4SArnaldo Carvalho de Melo 	.tp_basicsize	= sizeof(struct pyrf_event),
183877108e4SArnaldo Carvalho de Melo 	.tp_flags	= Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
184877108e4SArnaldo Carvalho de Melo 	.tp_doc		= pyrf_task_event__doc,
185877108e4SArnaldo Carvalho de Melo 	.tp_members	= pyrf_task_event__members,
186877108e4SArnaldo Carvalho de Melo 	.tp_repr	= (reprfunc)pyrf_task_event__repr,
187877108e4SArnaldo Carvalho de Melo };
188877108e4SArnaldo Carvalho de Melo 
189877108e4SArnaldo Carvalho de Melo static char pyrf_comm_event__doc[] = PyDoc_STR("perf comm event object.");
190877108e4SArnaldo Carvalho de Melo 
191877108e4SArnaldo Carvalho de Melo static PyMemberDef pyrf_comm_event__members[] = {
192877108e4SArnaldo Carvalho de Melo 	sample_members
193877108e4SArnaldo Carvalho de Melo 	member_def(perf_event_header, type, T_UINT, "event type"),
194877108e4SArnaldo Carvalho de Melo 	member_def(comm_event, pid, T_UINT, "event pid"),
195877108e4SArnaldo Carvalho de Melo 	member_def(comm_event, tid, T_UINT, "event tid"),
196877108e4SArnaldo Carvalho de Melo 	member_def(comm_event, comm, T_STRING_INPLACE, "process name"),
197f6bbc1daSArnaldo Carvalho de Melo 	{ .name = NULL, },
198877108e4SArnaldo Carvalho de Melo };
199877108e4SArnaldo Carvalho de Melo 
200877108e4SArnaldo Carvalho de Melo static PyObject *pyrf_comm_event__repr(struct pyrf_event *pevent)
201877108e4SArnaldo Carvalho de Melo {
20266dfdff0SJaroslav Škarvada 	return _PyUnicode_FromFormat("{ type: comm, pid: %u, tid: %u, comm: %s }",
203877108e4SArnaldo Carvalho de Melo 				   pevent->event.comm.pid,
204877108e4SArnaldo Carvalho de Melo 				   pevent->event.comm.tid,
205877108e4SArnaldo Carvalho de Melo 				   pevent->event.comm.comm);
206877108e4SArnaldo Carvalho de Melo }
207877108e4SArnaldo Carvalho de Melo 
208877108e4SArnaldo Carvalho de Melo static PyTypeObject pyrf_comm_event__type = {
209877108e4SArnaldo Carvalho de Melo 	PyVarObject_HEAD_INIT(NULL, 0)
210877108e4SArnaldo Carvalho de Melo 	.tp_name	= "perf.comm_event",
211877108e4SArnaldo Carvalho de Melo 	.tp_basicsize	= sizeof(struct pyrf_event),
212877108e4SArnaldo Carvalho de Melo 	.tp_flags	= Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
213877108e4SArnaldo Carvalho de Melo 	.tp_doc		= pyrf_comm_event__doc,
214877108e4SArnaldo Carvalho de Melo 	.tp_members	= pyrf_comm_event__members,
215877108e4SArnaldo Carvalho de Melo 	.tp_repr	= (reprfunc)pyrf_comm_event__repr,
216877108e4SArnaldo Carvalho de Melo };
217877108e4SArnaldo Carvalho de Melo 
218877108e4SArnaldo Carvalho de Melo static char pyrf_throttle_event__doc[] = PyDoc_STR("perf throttle event object.");
219877108e4SArnaldo Carvalho de Melo 
220877108e4SArnaldo Carvalho de Melo static PyMemberDef pyrf_throttle_event__members[] = {
221877108e4SArnaldo Carvalho de Melo 	sample_members
222877108e4SArnaldo Carvalho de Melo 	member_def(perf_event_header, type, T_UINT, "event type"),
223f6bbc1daSArnaldo Carvalho de Melo 	member_def(throttle_event, time, T_ULONGLONG, "timestamp"),
224f6bbc1daSArnaldo Carvalho de Melo 	member_def(throttle_event, id, T_ULONGLONG, "event id"),
225f6bbc1daSArnaldo Carvalho de Melo 	member_def(throttle_event, stream_id, T_ULONGLONG, "event stream id"),
226f6bbc1daSArnaldo Carvalho de Melo 	{ .name = NULL, },
227877108e4SArnaldo Carvalho de Melo };
228877108e4SArnaldo Carvalho de Melo 
229877108e4SArnaldo Carvalho de Melo static PyObject *pyrf_throttle_event__repr(struct pyrf_event *pevent)
230877108e4SArnaldo Carvalho de Melo {
231877108e4SArnaldo Carvalho de Melo 	struct throttle_event *te = (struct throttle_event *)(&pevent->event.header + 1);
232877108e4SArnaldo Carvalho de Melo 
23366dfdff0SJaroslav Škarvada 	return _PyUnicode_FromFormat("{ type: %sthrottle, time: %" PRIu64 ", id: %" PRIu64
234877108e4SArnaldo Carvalho de Melo 				   ", stream_id: %" PRIu64 " }",
235877108e4SArnaldo Carvalho de Melo 				   pevent->event.header.type == PERF_RECORD_THROTTLE ? "" : "un",
236877108e4SArnaldo Carvalho de Melo 				   te->time, te->id, te->stream_id);
237877108e4SArnaldo Carvalho de Melo }
238877108e4SArnaldo Carvalho de Melo 
239877108e4SArnaldo Carvalho de Melo static PyTypeObject pyrf_throttle_event__type = {
240877108e4SArnaldo Carvalho de Melo 	PyVarObject_HEAD_INIT(NULL, 0)
241877108e4SArnaldo Carvalho de Melo 	.tp_name	= "perf.throttle_event",
242877108e4SArnaldo Carvalho de Melo 	.tp_basicsize	= sizeof(struct pyrf_event),
243877108e4SArnaldo Carvalho de Melo 	.tp_flags	= Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
244877108e4SArnaldo Carvalho de Melo 	.tp_doc		= pyrf_throttle_event__doc,
245877108e4SArnaldo Carvalho de Melo 	.tp_members	= pyrf_throttle_event__members,
246877108e4SArnaldo Carvalho de Melo 	.tp_repr	= (reprfunc)pyrf_throttle_event__repr,
247877108e4SArnaldo Carvalho de Melo };
248877108e4SArnaldo Carvalho de Melo 
2493e9f45a7SArnaldo Carvalho de Melo static char pyrf_lost_event__doc[] = PyDoc_STR("perf lost event object.");
2503e9f45a7SArnaldo Carvalho de Melo 
2513e9f45a7SArnaldo Carvalho de Melo static PyMemberDef pyrf_lost_event__members[] = {
2523e9f45a7SArnaldo Carvalho de Melo 	sample_members
2533e9f45a7SArnaldo Carvalho de Melo 	member_def(lost_event, id, T_ULONGLONG, "event id"),
2543e9f45a7SArnaldo Carvalho de Melo 	member_def(lost_event, lost, T_ULONGLONG, "number of lost events"),
2553e9f45a7SArnaldo Carvalho de Melo 	{ .name = NULL, },
2563e9f45a7SArnaldo Carvalho de Melo };
2573e9f45a7SArnaldo Carvalho de Melo 
2583e9f45a7SArnaldo Carvalho de Melo static PyObject *pyrf_lost_event__repr(struct pyrf_event *pevent)
2593e9f45a7SArnaldo Carvalho de Melo {
2603e9f45a7SArnaldo Carvalho de Melo 	PyObject *ret;
2613e9f45a7SArnaldo Carvalho de Melo 	char *s;
2623e9f45a7SArnaldo Carvalho de Melo 
2633e9f45a7SArnaldo Carvalho de Melo 	if (asprintf(&s, "{ type: lost, id: %#" PRIx64 ", "
2643e9f45a7SArnaldo Carvalho de Melo 			 "lost: %#" PRIx64 " }",
2653e9f45a7SArnaldo Carvalho de Melo 		     pevent->event.lost.id, pevent->event.lost.lost) < 0) {
2663e9f45a7SArnaldo Carvalho de Melo 		ret = PyErr_NoMemory();
2673e9f45a7SArnaldo Carvalho de Melo 	} else {
26866dfdff0SJaroslav Škarvada 		ret = _PyUnicode_FromString(s);
2693e9f45a7SArnaldo Carvalho de Melo 		free(s);
2703e9f45a7SArnaldo Carvalho de Melo 	}
2713e9f45a7SArnaldo Carvalho de Melo 	return ret;
2723e9f45a7SArnaldo Carvalho de Melo }
2733e9f45a7SArnaldo Carvalho de Melo 
2743e9f45a7SArnaldo Carvalho de Melo static PyTypeObject pyrf_lost_event__type = {
2753e9f45a7SArnaldo Carvalho de Melo 	PyVarObject_HEAD_INIT(NULL, 0)
2763e9f45a7SArnaldo Carvalho de Melo 	.tp_name	= "perf.lost_event",
2773e9f45a7SArnaldo Carvalho de Melo 	.tp_basicsize	= sizeof(struct pyrf_event),
2783e9f45a7SArnaldo Carvalho de Melo 	.tp_flags	= Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
2793e9f45a7SArnaldo Carvalho de Melo 	.tp_doc		= pyrf_lost_event__doc,
2803e9f45a7SArnaldo Carvalho de Melo 	.tp_members	= pyrf_lost_event__members,
2813e9f45a7SArnaldo Carvalho de Melo 	.tp_repr	= (reprfunc)pyrf_lost_event__repr,
2823e9f45a7SArnaldo Carvalho de Melo };
2833e9f45a7SArnaldo Carvalho de Melo 
2843e9f45a7SArnaldo Carvalho de Melo static char pyrf_read_event__doc[] = PyDoc_STR("perf read event object.");
2853e9f45a7SArnaldo Carvalho de Melo 
2863e9f45a7SArnaldo Carvalho de Melo static PyMemberDef pyrf_read_event__members[] = {
2873e9f45a7SArnaldo Carvalho de Melo 	sample_members
2883e9f45a7SArnaldo Carvalho de Melo 	member_def(read_event, pid, T_UINT, "event pid"),
2893e9f45a7SArnaldo Carvalho de Melo 	member_def(read_event, tid, T_UINT, "event tid"),
2903e9f45a7SArnaldo Carvalho de Melo 	{ .name = NULL, },
2913e9f45a7SArnaldo Carvalho de Melo };
2923e9f45a7SArnaldo Carvalho de Melo 
2933e9f45a7SArnaldo Carvalho de Melo static PyObject *pyrf_read_event__repr(struct pyrf_event *pevent)
2943e9f45a7SArnaldo Carvalho de Melo {
29566dfdff0SJaroslav Škarvada 	return _PyUnicode_FromFormat("{ type: read, pid: %u, tid: %u }",
2963e9f45a7SArnaldo Carvalho de Melo 				   pevent->event.read.pid,
2973e9f45a7SArnaldo Carvalho de Melo 				   pevent->event.read.tid);
2983e9f45a7SArnaldo Carvalho de Melo 	/*
2993e9f45a7SArnaldo Carvalho de Melo  	 * FIXME: return the array of read values,
3003e9f45a7SArnaldo Carvalho de Melo  	 * making this method useful ;-)
3013e9f45a7SArnaldo Carvalho de Melo  	 */
3023e9f45a7SArnaldo Carvalho de Melo }
3033e9f45a7SArnaldo Carvalho de Melo 
3043e9f45a7SArnaldo Carvalho de Melo static PyTypeObject pyrf_read_event__type = {
3053e9f45a7SArnaldo Carvalho de Melo 	PyVarObject_HEAD_INIT(NULL, 0)
3063e9f45a7SArnaldo Carvalho de Melo 	.tp_name	= "perf.read_event",
3073e9f45a7SArnaldo Carvalho de Melo 	.tp_basicsize	= sizeof(struct pyrf_event),
3083e9f45a7SArnaldo Carvalho de Melo 	.tp_flags	= Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
3093e9f45a7SArnaldo Carvalho de Melo 	.tp_doc		= pyrf_read_event__doc,
3103e9f45a7SArnaldo Carvalho de Melo 	.tp_members	= pyrf_read_event__members,
3113e9f45a7SArnaldo Carvalho de Melo 	.tp_repr	= (reprfunc)pyrf_read_event__repr,
3123e9f45a7SArnaldo Carvalho de Melo };
3133e9f45a7SArnaldo Carvalho de Melo 
3143e9f45a7SArnaldo Carvalho de Melo static char pyrf_sample_event__doc[] = PyDoc_STR("perf sample event object.");
3153e9f45a7SArnaldo Carvalho de Melo 
3163e9f45a7SArnaldo Carvalho de Melo static PyMemberDef pyrf_sample_event__members[] = {
3173e9f45a7SArnaldo Carvalho de Melo 	sample_members
3183e9f45a7SArnaldo Carvalho de Melo 	member_def(perf_event_header, type, T_UINT, "event type"),
3193e9f45a7SArnaldo Carvalho de Melo 	{ .name = NULL, },
3203e9f45a7SArnaldo Carvalho de Melo };
3213e9f45a7SArnaldo Carvalho de Melo 
3223e9f45a7SArnaldo Carvalho de Melo static PyObject *pyrf_sample_event__repr(struct pyrf_event *pevent)
3233e9f45a7SArnaldo Carvalho de Melo {
3243e9f45a7SArnaldo Carvalho de Melo 	PyObject *ret;
3253e9f45a7SArnaldo Carvalho de Melo 	char *s;
3263e9f45a7SArnaldo Carvalho de Melo 
3273e9f45a7SArnaldo Carvalho de Melo 	if (asprintf(&s, "{ type: sample }") < 0) {
3283e9f45a7SArnaldo Carvalho de Melo 		ret = PyErr_NoMemory();
3293e9f45a7SArnaldo Carvalho de Melo 	} else {
33066dfdff0SJaroslav Škarvada 		ret = _PyUnicode_FromString(s);
3313e9f45a7SArnaldo Carvalho de Melo 		free(s);
3323e9f45a7SArnaldo Carvalho de Melo 	}
3333e9f45a7SArnaldo Carvalho de Melo 	return ret;
3343e9f45a7SArnaldo Carvalho de Melo }
3353e9f45a7SArnaldo Carvalho de Melo 
336bae57e38SJiri Olsa static bool is_tracepoint(struct pyrf_event *pevent)
337bae57e38SJiri Olsa {
338bae57e38SJiri Olsa 	return pevent->evsel->attr.type == PERF_TYPE_TRACEPOINT;
339bae57e38SJiri Olsa }
340bae57e38SJiri Olsa 
341bae57e38SJiri Olsa static PyObject*
342bae57e38SJiri Olsa tracepoint_field(struct pyrf_event *pe, struct format_field *field)
343bae57e38SJiri Olsa {
344bae57e38SJiri Olsa 	struct pevent *pevent = field->event->pevent;
345bae57e38SJiri Olsa 	void *data = pe->sample.raw_data;
346bae57e38SJiri Olsa 	PyObject *ret = NULL;
347bae57e38SJiri Olsa 	unsigned long long val;
348bae57e38SJiri Olsa 	unsigned int offset, len;
349bae57e38SJiri Olsa 
350bae57e38SJiri Olsa 	if (field->flags & FIELD_IS_ARRAY) {
351bae57e38SJiri Olsa 		offset = field->offset;
352bae57e38SJiri Olsa 		len    = field->size;
353bae57e38SJiri Olsa 		if (field->flags & FIELD_IS_DYNAMIC) {
354bae57e38SJiri Olsa 			val     = pevent_read_number(pevent, data + offset, len);
355bae57e38SJiri Olsa 			offset  = val;
356bae57e38SJiri Olsa 			len     = offset >> 16;
357bae57e38SJiri Olsa 			offset &= 0xffff;
358bae57e38SJiri Olsa 		}
359bae57e38SJiri Olsa 		if (field->flags & FIELD_IS_STRING &&
360bae57e38SJiri Olsa 		    is_printable_array(data + offset, len)) {
36166dfdff0SJaroslav Škarvada 			ret = _PyUnicode_FromString((char *)data + offset);
362bae57e38SJiri Olsa 		} else {
363bae57e38SJiri Olsa 			ret = PyByteArray_FromStringAndSize((const char *) data + offset, len);
364bae57e38SJiri Olsa 			field->flags &= ~FIELD_IS_STRING;
365bae57e38SJiri Olsa 		}
366bae57e38SJiri Olsa 	} else {
367bae57e38SJiri Olsa 		val = pevent_read_number(pevent, data + field->offset,
368bae57e38SJiri Olsa 					 field->size);
369bae57e38SJiri Olsa 		if (field->flags & FIELD_IS_POINTER)
370bae57e38SJiri Olsa 			ret = PyLong_FromUnsignedLong((unsigned long) val);
371bae57e38SJiri Olsa 		else if (field->flags & FIELD_IS_SIGNED)
372bae57e38SJiri Olsa 			ret = PyLong_FromLong((long) val);
373bae57e38SJiri Olsa 		else
374bae57e38SJiri Olsa 			ret = PyLong_FromUnsignedLong((unsigned long) val);
375bae57e38SJiri Olsa 	}
376bae57e38SJiri Olsa 
377bae57e38SJiri Olsa 	return ret;
378bae57e38SJiri Olsa }
379bae57e38SJiri Olsa 
380bae57e38SJiri Olsa static PyObject*
381bae57e38SJiri Olsa get_tracepoint_field(struct pyrf_event *pevent, PyObject *attr_name)
382bae57e38SJiri Olsa {
38366dfdff0SJaroslav Škarvada 	const char *str = _PyUnicode_AsString(PyObject_Str(attr_name));
384bae57e38SJiri Olsa 	struct perf_evsel *evsel = pevent->evsel;
385bae57e38SJiri Olsa 	struct format_field *field;
386bae57e38SJiri Olsa 
387bae57e38SJiri Olsa 	if (!evsel->tp_format) {
388bae57e38SJiri Olsa 		struct event_format *tp_format;
389bae57e38SJiri Olsa 
390bae57e38SJiri Olsa 		tp_format = trace_event__tp_format_id(evsel->attr.config);
391bae57e38SJiri Olsa 		if (!tp_format)
392bae57e38SJiri Olsa 			return NULL;
393bae57e38SJiri Olsa 
394bae57e38SJiri Olsa 		evsel->tp_format = tp_format;
395bae57e38SJiri Olsa 	}
396bae57e38SJiri Olsa 
397bae57e38SJiri Olsa 	field = pevent_find_any_field(evsel->tp_format, str);
398bae57e38SJiri Olsa 	if (!field)
399bae57e38SJiri Olsa 		return NULL;
400bae57e38SJiri Olsa 
401bae57e38SJiri Olsa 	return tracepoint_field(pevent, field);
402bae57e38SJiri Olsa }
403bae57e38SJiri Olsa 
404bae57e38SJiri Olsa static PyObject*
405bae57e38SJiri Olsa pyrf_sample_event__getattro(struct pyrf_event *pevent, PyObject *attr_name)
406bae57e38SJiri Olsa {
407bae57e38SJiri Olsa 	PyObject *obj = NULL;
408bae57e38SJiri Olsa 
409bae57e38SJiri Olsa 	if (is_tracepoint(pevent))
410bae57e38SJiri Olsa 		obj = get_tracepoint_field(pevent, attr_name);
411bae57e38SJiri Olsa 
412bae57e38SJiri Olsa 	return obj ?: PyObject_GenericGetAttr((PyObject *) pevent, attr_name);
413bae57e38SJiri Olsa }
414bae57e38SJiri Olsa 
4153e9f45a7SArnaldo Carvalho de Melo static PyTypeObject pyrf_sample_event__type = {
4163e9f45a7SArnaldo Carvalho de Melo 	PyVarObject_HEAD_INIT(NULL, 0)
4173e9f45a7SArnaldo Carvalho de Melo 	.tp_name	= "perf.sample_event",
4183e9f45a7SArnaldo Carvalho de Melo 	.tp_basicsize	= sizeof(struct pyrf_event),
4193e9f45a7SArnaldo Carvalho de Melo 	.tp_flags	= Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
4203e9f45a7SArnaldo Carvalho de Melo 	.tp_doc		= pyrf_sample_event__doc,
4213e9f45a7SArnaldo Carvalho de Melo 	.tp_members	= pyrf_sample_event__members,
4223e9f45a7SArnaldo Carvalho de Melo 	.tp_repr	= (reprfunc)pyrf_sample_event__repr,
423bae57e38SJiri Olsa 	.tp_getattro	= (getattrofunc) pyrf_sample_event__getattro,
4243e9f45a7SArnaldo Carvalho de Melo };
4253e9f45a7SArnaldo Carvalho de Melo 
426ae938802SArnaldo Carvalho de Melo static char pyrf_context_switch_event__doc[] = PyDoc_STR("perf context_switch event object.");
427ae938802SArnaldo Carvalho de Melo 
428ae938802SArnaldo Carvalho de Melo static PyMemberDef pyrf_context_switch_event__members[] = {
429ae938802SArnaldo Carvalho de Melo 	sample_members
430ae938802SArnaldo Carvalho de Melo 	member_def(perf_event_header, type, T_UINT, "event type"),
431ae938802SArnaldo Carvalho de Melo 	member_def(context_switch_event, next_prev_pid, T_UINT, "next/prev pid"),
432ae938802SArnaldo Carvalho de Melo 	member_def(context_switch_event, next_prev_tid, T_UINT, "next/prev tid"),
433ae938802SArnaldo Carvalho de Melo 	{ .name = NULL, },
434ae938802SArnaldo Carvalho de Melo };
435ae938802SArnaldo Carvalho de Melo 
436ae938802SArnaldo Carvalho de Melo static PyObject *pyrf_context_switch_event__repr(struct pyrf_event *pevent)
437ae938802SArnaldo Carvalho de Melo {
438ae938802SArnaldo Carvalho de Melo 	PyObject *ret;
439ae938802SArnaldo Carvalho de Melo 	char *s;
440ae938802SArnaldo Carvalho de Melo 
441ae938802SArnaldo Carvalho de Melo 	if (asprintf(&s, "{ type: context_switch, next_prev_pid: %u, next_prev_tid: %u, switch_out: %u }",
442ae938802SArnaldo Carvalho de Melo 		     pevent->event.context_switch.next_prev_pid,
443ae938802SArnaldo Carvalho de Melo 		     pevent->event.context_switch.next_prev_tid,
444ae938802SArnaldo Carvalho de Melo 		     !!(pevent->event.header.misc & PERF_RECORD_MISC_SWITCH_OUT)) < 0) {
445ae938802SArnaldo Carvalho de Melo 		ret = PyErr_NoMemory();
446ae938802SArnaldo Carvalho de Melo 	} else {
44766dfdff0SJaroslav Škarvada 		ret = _PyUnicode_FromString(s);
448ae938802SArnaldo Carvalho de Melo 		free(s);
449ae938802SArnaldo Carvalho de Melo 	}
450ae938802SArnaldo Carvalho de Melo 	return ret;
451ae938802SArnaldo Carvalho de Melo }
452ae938802SArnaldo Carvalho de Melo 
453ae938802SArnaldo Carvalho de Melo static PyTypeObject pyrf_context_switch_event__type = {
454ae938802SArnaldo Carvalho de Melo 	PyVarObject_HEAD_INIT(NULL, 0)
455ae938802SArnaldo Carvalho de Melo 	.tp_name	= "perf.context_switch_event",
456ae938802SArnaldo Carvalho de Melo 	.tp_basicsize	= sizeof(struct pyrf_event),
457ae938802SArnaldo Carvalho de Melo 	.tp_flags	= Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
458ae938802SArnaldo Carvalho de Melo 	.tp_doc		= pyrf_context_switch_event__doc,
459ae938802SArnaldo Carvalho de Melo 	.tp_members	= pyrf_context_switch_event__members,
460ae938802SArnaldo Carvalho de Melo 	.tp_repr	= (reprfunc)pyrf_context_switch_event__repr,
461ae938802SArnaldo Carvalho de Melo };
462ae938802SArnaldo Carvalho de Melo 
463877108e4SArnaldo Carvalho de Melo static int pyrf_event__setup_types(void)
464877108e4SArnaldo Carvalho de Melo {
465877108e4SArnaldo Carvalho de Melo 	int err;
466877108e4SArnaldo Carvalho de Melo 	pyrf_mmap_event__type.tp_new =
467877108e4SArnaldo Carvalho de Melo 	pyrf_task_event__type.tp_new =
468877108e4SArnaldo Carvalho de Melo 	pyrf_comm_event__type.tp_new =
4693e9f45a7SArnaldo Carvalho de Melo 	pyrf_lost_event__type.tp_new =
4703e9f45a7SArnaldo Carvalho de Melo 	pyrf_read_event__type.tp_new =
4713e9f45a7SArnaldo Carvalho de Melo 	pyrf_sample_event__type.tp_new =
472ae938802SArnaldo Carvalho de Melo 	pyrf_context_switch_event__type.tp_new =
473877108e4SArnaldo Carvalho de Melo 	pyrf_throttle_event__type.tp_new = PyType_GenericNew;
474877108e4SArnaldo Carvalho de Melo 	err = PyType_Ready(&pyrf_mmap_event__type);
475877108e4SArnaldo Carvalho de Melo 	if (err < 0)
476877108e4SArnaldo Carvalho de Melo 		goto out;
4773e9f45a7SArnaldo Carvalho de Melo 	err = PyType_Ready(&pyrf_lost_event__type);
4783e9f45a7SArnaldo Carvalho de Melo 	if (err < 0)
4793e9f45a7SArnaldo Carvalho de Melo 		goto out;
480877108e4SArnaldo Carvalho de Melo 	err = PyType_Ready(&pyrf_task_event__type);
481877108e4SArnaldo Carvalho de Melo 	if (err < 0)
482877108e4SArnaldo Carvalho de Melo 		goto out;
483877108e4SArnaldo Carvalho de Melo 	err = PyType_Ready(&pyrf_comm_event__type);
484877108e4SArnaldo Carvalho de Melo 	if (err < 0)
485877108e4SArnaldo Carvalho de Melo 		goto out;
486877108e4SArnaldo Carvalho de Melo 	err = PyType_Ready(&pyrf_throttle_event__type);
487877108e4SArnaldo Carvalho de Melo 	if (err < 0)
488877108e4SArnaldo Carvalho de Melo 		goto out;
4893e9f45a7SArnaldo Carvalho de Melo 	err = PyType_Ready(&pyrf_read_event__type);
4903e9f45a7SArnaldo Carvalho de Melo 	if (err < 0)
4913e9f45a7SArnaldo Carvalho de Melo 		goto out;
4923e9f45a7SArnaldo Carvalho de Melo 	err = PyType_Ready(&pyrf_sample_event__type);
4933e9f45a7SArnaldo Carvalho de Melo 	if (err < 0)
4943e9f45a7SArnaldo Carvalho de Melo 		goto out;
495ae938802SArnaldo Carvalho de Melo 	err = PyType_Ready(&pyrf_context_switch_event__type);
496ae938802SArnaldo Carvalho de Melo 	if (err < 0)
497ae938802SArnaldo Carvalho de Melo 		goto out;
498877108e4SArnaldo Carvalho de Melo out:
499877108e4SArnaldo Carvalho de Melo 	return err;
500877108e4SArnaldo Carvalho de Melo }
501877108e4SArnaldo Carvalho de Melo 
502877108e4SArnaldo Carvalho de Melo static PyTypeObject *pyrf_event__type[] = {
503877108e4SArnaldo Carvalho de Melo 	[PERF_RECORD_MMAP]	 = &pyrf_mmap_event__type,
5043e9f45a7SArnaldo Carvalho de Melo 	[PERF_RECORD_LOST]	 = &pyrf_lost_event__type,
505877108e4SArnaldo Carvalho de Melo 	[PERF_RECORD_COMM]	 = &pyrf_comm_event__type,
506877108e4SArnaldo Carvalho de Melo 	[PERF_RECORD_EXIT]	 = &pyrf_task_event__type,
507877108e4SArnaldo Carvalho de Melo 	[PERF_RECORD_THROTTLE]	 = &pyrf_throttle_event__type,
508877108e4SArnaldo Carvalho de Melo 	[PERF_RECORD_UNTHROTTLE] = &pyrf_throttle_event__type,
509877108e4SArnaldo Carvalho de Melo 	[PERF_RECORD_FORK]	 = &pyrf_task_event__type,
5103e9f45a7SArnaldo Carvalho de Melo 	[PERF_RECORD_READ]	 = &pyrf_read_event__type,
5113e9f45a7SArnaldo Carvalho de Melo 	[PERF_RECORD_SAMPLE]	 = &pyrf_sample_event__type,
512ae938802SArnaldo Carvalho de Melo 	[PERF_RECORD_SWITCH]	 = &pyrf_context_switch_event__type,
513ae938802SArnaldo Carvalho de Melo 	[PERF_RECORD_SWITCH_CPU_WIDE]  = &pyrf_context_switch_event__type,
514877108e4SArnaldo Carvalho de Melo };
515877108e4SArnaldo Carvalho de Melo 
516877108e4SArnaldo Carvalho de Melo static PyObject *pyrf_event__new(union perf_event *event)
517877108e4SArnaldo Carvalho de Melo {
518877108e4SArnaldo Carvalho de Melo 	struct pyrf_event *pevent;
519877108e4SArnaldo Carvalho de Melo 	PyTypeObject *ptype;
520877108e4SArnaldo Carvalho de Melo 
521ae938802SArnaldo Carvalho de Melo 	if ((event->header.type < PERF_RECORD_MMAP ||
522ae938802SArnaldo Carvalho de Melo 	     event->header.type > PERF_RECORD_SAMPLE) &&
523ae938802SArnaldo Carvalho de Melo 	    !(event->header.type == PERF_RECORD_SWITCH ||
524ae938802SArnaldo Carvalho de Melo 	      event->header.type == PERF_RECORD_SWITCH_CPU_WIDE))
525877108e4SArnaldo Carvalho de Melo 		return NULL;
526877108e4SArnaldo Carvalho de Melo 
527877108e4SArnaldo Carvalho de Melo 	ptype = pyrf_event__type[event->header.type];
528877108e4SArnaldo Carvalho de Melo 	pevent = PyObject_New(struct pyrf_event, ptype);
529877108e4SArnaldo Carvalho de Melo 	if (pevent != NULL)
530877108e4SArnaldo Carvalho de Melo 		memcpy(&pevent->event, event, event->header.size);
531877108e4SArnaldo Carvalho de Melo 	return (PyObject *)pevent;
532877108e4SArnaldo Carvalho de Melo }
533877108e4SArnaldo Carvalho de Melo 
534877108e4SArnaldo Carvalho de Melo struct pyrf_cpu_map {
535877108e4SArnaldo Carvalho de Melo 	PyObject_HEAD
536877108e4SArnaldo Carvalho de Melo 
537877108e4SArnaldo Carvalho de Melo 	struct cpu_map *cpus;
538877108e4SArnaldo Carvalho de Melo };
539877108e4SArnaldo Carvalho de Melo 
540877108e4SArnaldo Carvalho de Melo static int pyrf_cpu_map__init(struct pyrf_cpu_map *pcpus,
541877108e4SArnaldo Carvalho de Melo 			      PyObject *args, PyObject *kwargs)
542877108e4SArnaldo Carvalho de Melo {
54364348153SFrederic Weisbecker 	static char *kwlist[] = { "cpustr", NULL };
544877108e4SArnaldo Carvalho de Melo 	char *cpustr = NULL;
545877108e4SArnaldo Carvalho de Melo 
546877108e4SArnaldo Carvalho de Melo 	if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|s",
547877108e4SArnaldo Carvalho de Melo 					 kwlist, &cpustr))
548877108e4SArnaldo Carvalho de Melo 		return -1;
549877108e4SArnaldo Carvalho de Melo 
550877108e4SArnaldo Carvalho de Melo 	pcpus->cpus = cpu_map__new(cpustr);
551877108e4SArnaldo Carvalho de Melo 	if (pcpus->cpus == NULL)
552877108e4SArnaldo Carvalho de Melo 		return -1;
553877108e4SArnaldo Carvalho de Melo 	return 0;
554877108e4SArnaldo Carvalho de Melo }
555877108e4SArnaldo Carvalho de Melo 
556877108e4SArnaldo Carvalho de Melo static void pyrf_cpu_map__delete(struct pyrf_cpu_map *pcpus)
557877108e4SArnaldo Carvalho de Melo {
558f30a79b0SJiri Olsa 	cpu_map__put(pcpus->cpus);
55966dfdff0SJaroslav Škarvada 	Py_TYPE(pcpus)->tp_free((PyObject*)pcpus);
560877108e4SArnaldo Carvalho de Melo }
561877108e4SArnaldo Carvalho de Melo 
562877108e4SArnaldo Carvalho de Melo static Py_ssize_t pyrf_cpu_map__length(PyObject *obj)
563877108e4SArnaldo Carvalho de Melo {
564877108e4SArnaldo Carvalho de Melo 	struct pyrf_cpu_map *pcpus = (void *)obj;
565877108e4SArnaldo Carvalho de Melo 
566877108e4SArnaldo Carvalho de Melo 	return pcpus->cpus->nr;
567877108e4SArnaldo Carvalho de Melo }
568877108e4SArnaldo Carvalho de Melo 
569877108e4SArnaldo Carvalho de Melo static PyObject *pyrf_cpu_map__item(PyObject *obj, Py_ssize_t i)
570877108e4SArnaldo Carvalho de Melo {
571877108e4SArnaldo Carvalho de Melo 	struct pyrf_cpu_map *pcpus = (void *)obj;
572877108e4SArnaldo Carvalho de Melo 
573877108e4SArnaldo Carvalho de Melo 	if (i >= pcpus->cpus->nr)
574877108e4SArnaldo Carvalho de Melo 		return NULL;
575877108e4SArnaldo Carvalho de Melo 
576877108e4SArnaldo Carvalho de Melo 	return Py_BuildValue("i", pcpus->cpus->map[i]);
577877108e4SArnaldo Carvalho de Melo }
578877108e4SArnaldo Carvalho de Melo 
579877108e4SArnaldo Carvalho de Melo static PySequenceMethods pyrf_cpu_map__sequence_methods = {
580877108e4SArnaldo Carvalho de Melo 	.sq_length = pyrf_cpu_map__length,
581877108e4SArnaldo Carvalho de Melo 	.sq_item   = pyrf_cpu_map__item,
582877108e4SArnaldo Carvalho de Melo };
583877108e4SArnaldo Carvalho de Melo 
584877108e4SArnaldo Carvalho de Melo static char pyrf_cpu_map__doc[] = PyDoc_STR("cpu map object.");
585877108e4SArnaldo Carvalho de Melo 
586877108e4SArnaldo Carvalho de Melo static PyTypeObject pyrf_cpu_map__type = {
587877108e4SArnaldo Carvalho de Melo 	PyVarObject_HEAD_INIT(NULL, 0)
588877108e4SArnaldo Carvalho de Melo 	.tp_name	= "perf.cpu_map",
589877108e4SArnaldo Carvalho de Melo 	.tp_basicsize	= sizeof(struct pyrf_cpu_map),
590877108e4SArnaldo Carvalho de Melo 	.tp_dealloc	= (destructor)pyrf_cpu_map__delete,
591877108e4SArnaldo Carvalho de Melo 	.tp_flags	= Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
592877108e4SArnaldo Carvalho de Melo 	.tp_doc		= pyrf_cpu_map__doc,
593877108e4SArnaldo Carvalho de Melo 	.tp_as_sequence	= &pyrf_cpu_map__sequence_methods,
594877108e4SArnaldo Carvalho de Melo 	.tp_init	= (initproc)pyrf_cpu_map__init,
595877108e4SArnaldo Carvalho de Melo };
596877108e4SArnaldo Carvalho de Melo 
597877108e4SArnaldo Carvalho de Melo static int pyrf_cpu_map__setup_types(void)
598877108e4SArnaldo Carvalho de Melo {
599877108e4SArnaldo Carvalho de Melo 	pyrf_cpu_map__type.tp_new = PyType_GenericNew;
600877108e4SArnaldo Carvalho de Melo 	return PyType_Ready(&pyrf_cpu_map__type);
601877108e4SArnaldo Carvalho de Melo }
602877108e4SArnaldo Carvalho de Melo 
603877108e4SArnaldo Carvalho de Melo struct pyrf_thread_map {
604877108e4SArnaldo Carvalho de Melo 	PyObject_HEAD
605877108e4SArnaldo Carvalho de Melo 
606877108e4SArnaldo Carvalho de Melo 	struct thread_map *threads;
607877108e4SArnaldo Carvalho de Melo };
608877108e4SArnaldo Carvalho de Melo 
609877108e4SArnaldo Carvalho de Melo static int pyrf_thread_map__init(struct pyrf_thread_map *pthreads,
610877108e4SArnaldo Carvalho de Melo 				 PyObject *args, PyObject *kwargs)
611877108e4SArnaldo Carvalho de Melo {
6120d37aa34SArnaldo Carvalho de Melo 	static char *kwlist[] = { "pid", "tid", "uid", NULL };
6130d37aa34SArnaldo Carvalho de Melo 	int pid = -1, tid = -1, uid = UINT_MAX;
614877108e4SArnaldo Carvalho de Melo 
6150d37aa34SArnaldo Carvalho de Melo 	if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|iii",
6160d37aa34SArnaldo Carvalho de Melo 					 kwlist, &pid, &tid, &uid))
617877108e4SArnaldo Carvalho de Melo 		return -1;
618877108e4SArnaldo Carvalho de Melo 
6190d37aa34SArnaldo Carvalho de Melo 	pthreads->threads = thread_map__new(pid, tid, uid);
620877108e4SArnaldo Carvalho de Melo 	if (pthreads->threads == NULL)
621877108e4SArnaldo Carvalho de Melo 		return -1;
622877108e4SArnaldo Carvalho de Melo 	return 0;
623877108e4SArnaldo Carvalho de Melo }
624877108e4SArnaldo Carvalho de Melo 
625877108e4SArnaldo Carvalho de Melo static void pyrf_thread_map__delete(struct pyrf_thread_map *pthreads)
626877108e4SArnaldo Carvalho de Melo {
627186fbb74SJiri Olsa 	thread_map__put(pthreads->threads);
62866dfdff0SJaroslav Škarvada 	Py_TYPE(pthreads)->tp_free((PyObject*)pthreads);
629877108e4SArnaldo Carvalho de Melo }
630877108e4SArnaldo Carvalho de Melo 
631877108e4SArnaldo Carvalho de Melo static Py_ssize_t pyrf_thread_map__length(PyObject *obj)
632877108e4SArnaldo Carvalho de Melo {
633877108e4SArnaldo Carvalho de Melo 	struct pyrf_thread_map *pthreads = (void *)obj;
634877108e4SArnaldo Carvalho de Melo 
635877108e4SArnaldo Carvalho de Melo 	return pthreads->threads->nr;
636877108e4SArnaldo Carvalho de Melo }
637877108e4SArnaldo Carvalho de Melo 
638877108e4SArnaldo Carvalho de Melo static PyObject *pyrf_thread_map__item(PyObject *obj, Py_ssize_t i)
639877108e4SArnaldo Carvalho de Melo {
640877108e4SArnaldo Carvalho de Melo 	struct pyrf_thread_map *pthreads = (void *)obj;
641877108e4SArnaldo Carvalho de Melo 
642877108e4SArnaldo Carvalho de Melo 	if (i >= pthreads->threads->nr)
643877108e4SArnaldo Carvalho de Melo 		return NULL;
644877108e4SArnaldo Carvalho de Melo 
645877108e4SArnaldo Carvalho de Melo 	return Py_BuildValue("i", pthreads->threads->map[i]);
646877108e4SArnaldo Carvalho de Melo }
647877108e4SArnaldo Carvalho de Melo 
648877108e4SArnaldo Carvalho de Melo static PySequenceMethods pyrf_thread_map__sequence_methods = {
649877108e4SArnaldo Carvalho de Melo 	.sq_length = pyrf_thread_map__length,
650877108e4SArnaldo Carvalho de Melo 	.sq_item   = pyrf_thread_map__item,
651877108e4SArnaldo Carvalho de Melo };
652877108e4SArnaldo Carvalho de Melo 
653877108e4SArnaldo Carvalho de Melo static char pyrf_thread_map__doc[] = PyDoc_STR("thread map object.");
654877108e4SArnaldo Carvalho de Melo 
655877108e4SArnaldo Carvalho de Melo static PyTypeObject pyrf_thread_map__type = {
656877108e4SArnaldo Carvalho de Melo 	PyVarObject_HEAD_INIT(NULL, 0)
657877108e4SArnaldo Carvalho de Melo 	.tp_name	= "perf.thread_map",
658877108e4SArnaldo Carvalho de Melo 	.tp_basicsize	= sizeof(struct pyrf_thread_map),
659877108e4SArnaldo Carvalho de Melo 	.tp_dealloc	= (destructor)pyrf_thread_map__delete,
660877108e4SArnaldo Carvalho de Melo 	.tp_flags	= Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
661877108e4SArnaldo Carvalho de Melo 	.tp_doc		= pyrf_thread_map__doc,
662877108e4SArnaldo Carvalho de Melo 	.tp_as_sequence	= &pyrf_thread_map__sequence_methods,
663877108e4SArnaldo Carvalho de Melo 	.tp_init	= (initproc)pyrf_thread_map__init,
664877108e4SArnaldo Carvalho de Melo };
665877108e4SArnaldo Carvalho de Melo 
666877108e4SArnaldo Carvalho de Melo static int pyrf_thread_map__setup_types(void)
667877108e4SArnaldo Carvalho de Melo {
668877108e4SArnaldo Carvalho de Melo 	pyrf_thread_map__type.tp_new = PyType_GenericNew;
669877108e4SArnaldo Carvalho de Melo 	return PyType_Ready(&pyrf_thread_map__type);
670877108e4SArnaldo Carvalho de Melo }
671877108e4SArnaldo Carvalho de Melo 
672877108e4SArnaldo Carvalho de Melo struct pyrf_evsel {
673877108e4SArnaldo Carvalho de Melo 	PyObject_HEAD
674877108e4SArnaldo Carvalho de Melo 
675877108e4SArnaldo Carvalho de Melo 	struct perf_evsel evsel;
676877108e4SArnaldo Carvalho de Melo };
677877108e4SArnaldo Carvalho de Melo 
678877108e4SArnaldo Carvalho de Melo static int pyrf_evsel__init(struct pyrf_evsel *pevsel,
679877108e4SArnaldo Carvalho de Melo 			    PyObject *args, PyObject *kwargs)
680877108e4SArnaldo Carvalho de Melo {
681877108e4SArnaldo Carvalho de Melo 	struct perf_event_attr attr = {
682877108e4SArnaldo Carvalho de Melo 		.type = PERF_TYPE_HARDWARE,
683877108e4SArnaldo Carvalho de Melo 		.config = PERF_COUNT_HW_CPU_CYCLES,
684877108e4SArnaldo Carvalho de Melo 		.sample_type = PERF_SAMPLE_PERIOD | PERF_SAMPLE_TID,
685877108e4SArnaldo Carvalho de Melo 	};
686877108e4SArnaldo Carvalho de Melo 	static char *kwlist[] = {
687877108e4SArnaldo Carvalho de Melo 		"type",
688877108e4SArnaldo Carvalho de Melo 		"config",
689877108e4SArnaldo Carvalho de Melo 		"sample_freq",
690877108e4SArnaldo Carvalho de Melo 		"sample_period",
691877108e4SArnaldo Carvalho de Melo 		"sample_type",
692877108e4SArnaldo Carvalho de Melo 		"read_format",
693877108e4SArnaldo Carvalho de Melo 		"disabled",
694877108e4SArnaldo Carvalho de Melo 		"inherit",
695877108e4SArnaldo Carvalho de Melo 		"pinned",
696877108e4SArnaldo Carvalho de Melo 		"exclusive",
697877108e4SArnaldo Carvalho de Melo 		"exclude_user",
698877108e4SArnaldo Carvalho de Melo 		"exclude_kernel",
699877108e4SArnaldo Carvalho de Melo 		"exclude_hv",
700877108e4SArnaldo Carvalho de Melo 		"exclude_idle",
701877108e4SArnaldo Carvalho de Melo 		"mmap",
702ae938802SArnaldo Carvalho de Melo 		"context_switch",
703877108e4SArnaldo Carvalho de Melo 		"comm",
704877108e4SArnaldo Carvalho de Melo 		"freq",
705877108e4SArnaldo Carvalho de Melo 		"inherit_stat",
706877108e4SArnaldo Carvalho de Melo 		"enable_on_exec",
707877108e4SArnaldo Carvalho de Melo 		"task",
708877108e4SArnaldo Carvalho de Melo 		"watermark",
709877108e4SArnaldo Carvalho de Melo 		"precise_ip",
710877108e4SArnaldo Carvalho de Melo 		"mmap_data",
711877108e4SArnaldo Carvalho de Melo 		"sample_id_all",
712877108e4SArnaldo Carvalho de Melo 		"wakeup_events",
713877108e4SArnaldo Carvalho de Melo 		"bp_type",
714877108e4SArnaldo Carvalho de Melo 		"bp_addr",
71564348153SFrederic Weisbecker 		"bp_len",
71664348153SFrederic Weisbecker 		 NULL
71764348153SFrederic Weisbecker 	};
718877108e4SArnaldo Carvalho de Melo 	u64 sample_period = 0;
719877108e4SArnaldo Carvalho de Melo 	u32 disabled = 0,
720877108e4SArnaldo Carvalho de Melo 	    inherit = 0,
721877108e4SArnaldo Carvalho de Melo 	    pinned = 0,
722877108e4SArnaldo Carvalho de Melo 	    exclusive = 0,
723877108e4SArnaldo Carvalho de Melo 	    exclude_user = 0,
724877108e4SArnaldo Carvalho de Melo 	    exclude_kernel = 0,
725877108e4SArnaldo Carvalho de Melo 	    exclude_hv = 0,
726877108e4SArnaldo Carvalho de Melo 	    exclude_idle = 0,
727877108e4SArnaldo Carvalho de Melo 	    mmap = 0,
728ae938802SArnaldo Carvalho de Melo 	    context_switch = 0,
729877108e4SArnaldo Carvalho de Melo 	    comm = 0,
730877108e4SArnaldo Carvalho de Melo 	    freq = 1,
731877108e4SArnaldo Carvalho de Melo 	    inherit_stat = 0,
732877108e4SArnaldo Carvalho de Melo 	    enable_on_exec = 0,
733877108e4SArnaldo Carvalho de Melo 	    task = 0,
734877108e4SArnaldo Carvalho de Melo 	    watermark = 0,
735877108e4SArnaldo Carvalho de Melo 	    precise_ip = 0,
736877108e4SArnaldo Carvalho de Melo 	    mmap_data = 0,
737877108e4SArnaldo Carvalho de Melo 	    sample_id_all = 1;
738877108e4SArnaldo Carvalho de Melo 	int idx = 0;
739877108e4SArnaldo Carvalho de Melo 
740877108e4SArnaldo Carvalho de Melo 	if (!PyArg_ParseTupleAndKeywords(args, kwargs,
741ae938802SArnaldo Carvalho de Melo 					 "|iKiKKiiiiiiiiiiiiiiiiiiiiiiKK", kwlist,
742877108e4SArnaldo Carvalho de Melo 					 &attr.type, &attr.config, &attr.sample_freq,
743877108e4SArnaldo Carvalho de Melo 					 &sample_period, &attr.sample_type,
744877108e4SArnaldo Carvalho de Melo 					 &attr.read_format, &disabled, &inherit,
745877108e4SArnaldo Carvalho de Melo 					 &pinned, &exclusive, &exclude_user,
746877108e4SArnaldo Carvalho de Melo 					 &exclude_kernel, &exclude_hv, &exclude_idle,
747ae938802SArnaldo Carvalho de Melo 					 &mmap, &context_switch, &comm, &freq, &inherit_stat,
748877108e4SArnaldo Carvalho de Melo 					 &enable_on_exec, &task, &watermark,
749877108e4SArnaldo Carvalho de Melo 					 &precise_ip, &mmap_data, &sample_id_all,
750877108e4SArnaldo Carvalho de Melo 					 &attr.wakeup_events, &attr.bp_type,
751877108e4SArnaldo Carvalho de Melo 					 &attr.bp_addr, &attr.bp_len, &idx))
752877108e4SArnaldo Carvalho de Melo 		return -1;
753877108e4SArnaldo Carvalho de Melo 
754877108e4SArnaldo Carvalho de Melo 	/* union... */
755877108e4SArnaldo Carvalho de Melo 	if (sample_period != 0) {
756877108e4SArnaldo Carvalho de Melo 		if (attr.sample_freq != 0)
757877108e4SArnaldo Carvalho de Melo 			return -1; /* FIXME: throw right exception */
758877108e4SArnaldo Carvalho de Melo 		attr.sample_period = sample_period;
759877108e4SArnaldo Carvalho de Melo 	}
760877108e4SArnaldo Carvalho de Melo 
761877108e4SArnaldo Carvalho de Melo 	/* Bitfields */
762877108e4SArnaldo Carvalho de Melo 	attr.disabled	    = disabled;
763877108e4SArnaldo Carvalho de Melo 	attr.inherit	    = inherit;
764877108e4SArnaldo Carvalho de Melo 	attr.pinned	    = pinned;
765877108e4SArnaldo Carvalho de Melo 	attr.exclusive	    = exclusive;
766877108e4SArnaldo Carvalho de Melo 	attr.exclude_user   = exclude_user;
767877108e4SArnaldo Carvalho de Melo 	attr.exclude_kernel = exclude_kernel;
768877108e4SArnaldo Carvalho de Melo 	attr.exclude_hv	    = exclude_hv;
769877108e4SArnaldo Carvalho de Melo 	attr.exclude_idle   = exclude_idle;
770877108e4SArnaldo Carvalho de Melo 	attr.mmap	    = mmap;
771ae938802SArnaldo Carvalho de Melo 	attr.context_switch = context_switch;
772877108e4SArnaldo Carvalho de Melo 	attr.comm	    = comm;
773877108e4SArnaldo Carvalho de Melo 	attr.freq	    = freq;
774877108e4SArnaldo Carvalho de Melo 	attr.inherit_stat   = inherit_stat;
775877108e4SArnaldo Carvalho de Melo 	attr.enable_on_exec = enable_on_exec;
776877108e4SArnaldo Carvalho de Melo 	attr.task	    = task;
777877108e4SArnaldo Carvalho de Melo 	attr.watermark	    = watermark;
778877108e4SArnaldo Carvalho de Melo 	attr.precise_ip	    = precise_ip;
779877108e4SArnaldo Carvalho de Melo 	attr.mmap_data	    = mmap_data;
780877108e4SArnaldo Carvalho de Melo 	attr.sample_id_all  = sample_id_all;
781ad4e3c04SJiri Olsa 	attr.size	    = sizeof(attr);
782877108e4SArnaldo Carvalho de Melo 
783877108e4SArnaldo Carvalho de Melo 	perf_evsel__init(&pevsel->evsel, &attr, idx);
784877108e4SArnaldo Carvalho de Melo 	return 0;
785877108e4SArnaldo Carvalho de Melo }
786877108e4SArnaldo Carvalho de Melo 
787877108e4SArnaldo Carvalho de Melo static void pyrf_evsel__delete(struct pyrf_evsel *pevsel)
788877108e4SArnaldo Carvalho de Melo {
789877108e4SArnaldo Carvalho de Melo 	perf_evsel__exit(&pevsel->evsel);
79066dfdff0SJaroslav Škarvada 	Py_TYPE(pevsel)->tp_free((PyObject*)pevsel);
791877108e4SArnaldo Carvalho de Melo }
792877108e4SArnaldo Carvalho de Melo 
793877108e4SArnaldo Carvalho de Melo static PyObject *pyrf_evsel__open(struct pyrf_evsel *pevsel,
794877108e4SArnaldo Carvalho de Melo 				  PyObject *args, PyObject *kwargs)
795877108e4SArnaldo Carvalho de Melo {
796877108e4SArnaldo Carvalho de Melo 	struct perf_evsel *evsel = &pevsel->evsel;
797877108e4SArnaldo Carvalho de Melo 	struct cpu_map *cpus = NULL;
798877108e4SArnaldo Carvalho de Melo 	struct thread_map *threads = NULL;
799877108e4SArnaldo Carvalho de Melo 	PyObject *pcpus = NULL, *pthreads = NULL;
8005d2cd909SArnaldo Carvalho de Melo 	int group = 0, inherit = 0;
80164348153SFrederic Weisbecker 	static char *kwlist[] = { "cpus", "threads", "group", "inherit", NULL };
802877108e4SArnaldo Carvalho de Melo 
803877108e4SArnaldo Carvalho de Melo 	if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|OOii", kwlist,
8045d2cd909SArnaldo Carvalho de Melo 					 &pcpus, &pthreads, &group, &inherit))
805877108e4SArnaldo Carvalho de Melo 		return NULL;
806877108e4SArnaldo Carvalho de Melo 
807877108e4SArnaldo Carvalho de Melo 	if (pthreads != NULL)
808877108e4SArnaldo Carvalho de Melo 		threads = ((struct pyrf_thread_map *)pthreads)->threads;
809877108e4SArnaldo Carvalho de Melo 
810877108e4SArnaldo Carvalho de Melo 	if (pcpus != NULL)
811877108e4SArnaldo Carvalho de Melo 		cpus = ((struct pyrf_cpu_map *)pcpus)->cpus;
812877108e4SArnaldo Carvalho de Melo 
8135d2cd909SArnaldo Carvalho de Melo 	evsel->attr.inherit = inherit;
814727ab04eSArnaldo Carvalho de Melo 	/*
815727ab04eSArnaldo Carvalho de Melo 	 * This will group just the fds for this single evsel, to group
816727ab04eSArnaldo Carvalho de Melo 	 * multiple events, use evlist.open().
817727ab04eSArnaldo Carvalho de Melo 	 */
8186a4bb04cSJiri Olsa 	if (perf_evsel__open(evsel, cpus, threads) < 0) {
819877108e4SArnaldo Carvalho de Melo 		PyErr_SetFromErrno(PyExc_OSError);
820877108e4SArnaldo Carvalho de Melo 		return NULL;
821877108e4SArnaldo Carvalho de Melo 	}
822877108e4SArnaldo Carvalho de Melo 
823877108e4SArnaldo Carvalho de Melo 	Py_INCREF(Py_None);
824877108e4SArnaldo Carvalho de Melo 	return Py_None;
825877108e4SArnaldo Carvalho de Melo }
826877108e4SArnaldo Carvalho de Melo 
827877108e4SArnaldo Carvalho de Melo static PyMethodDef pyrf_evsel__methods[] = {
828877108e4SArnaldo Carvalho de Melo 	{
829877108e4SArnaldo Carvalho de Melo 		.ml_name  = "open",
830877108e4SArnaldo Carvalho de Melo 		.ml_meth  = (PyCFunction)pyrf_evsel__open,
831877108e4SArnaldo Carvalho de Melo 		.ml_flags = METH_VARARGS | METH_KEYWORDS,
832877108e4SArnaldo Carvalho de Melo 		.ml_doc	  = PyDoc_STR("open the event selector file descriptor table.")
833877108e4SArnaldo Carvalho de Melo 	},
834f6bbc1daSArnaldo Carvalho de Melo 	{ .ml_name = NULL, }
835877108e4SArnaldo Carvalho de Melo };
836877108e4SArnaldo Carvalho de Melo 
837877108e4SArnaldo Carvalho de Melo static char pyrf_evsel__doc[] = PyDoc_STR("perf event selector list object.");
838877108e4SArnaldo Carvalho de Melo 
839877108e4SArnaldo Carvalho de Melo static PyTypeObject pyrf_evsel__type = {
840877108e4SArnaldo Carvalho de Melo 	PyVarObject_HEAD_INIT(NULL, 0)
841877108e4SArnaldo Carvalho de Melo 	.tp_name	= "perf.evsel",
842877108e4SArnaldo Carvalho de Melo 	.tp_basicsize	= sizeof(struct pyrf_evsel),
843877108e4SArnaldo Carvalho de Melo 	.tp_dealloc	= (destructor)pyrf_evsel__delete,
844877108e4SArnaldo Carvalho de Melo 	.tp_flags	= Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
845877108e4SArnaldo Carvalho de Melo 	.tp_doc		= pyrf_evsel__doc,
846877108e4SArnaldo Carvalho de Melo 	.tp_methods	= pyrf_evsel__methods,
847877108e4SArnaldo Carvalho de Melo 	.tp_init	= (initproc)pyrf_evsel__init,
848877108e4SArnaldo Carvalho de Melo };
849877108e4SArnaldo Carvalho de Melo 
850877108e4SArnaldo Carvalho de Melo static int pyrf_evsel__setup_types(void)
851877108e4SArnaldo Carvalho de Melo {
852877108e4SArnaldo Carvalho de Melo 	pyrf_evsel__type.tp_new = PyType_GenericNew;
853877108e4SArnaldo Carvalho de Melo 	return PyType_Ready(&pyrf_evsel__type);
854877108e4SArnaldo Carvalho de Melo }
855877108e4SArnaldo Carvalho de Melo 
856877108e4SArnaldo Carvalho de Melo struct pyrf_evlist {
857877108e4SArnaldo Carvalho de Melo 	PyObject_HEAD
858877108e4SArnaldo Carvalho de Melo 
859877108e4SArnaldo Carvalho de Melo 	struct perf_evlist evlist;
860877108e4SArnaldo Carvalho de Melo };
861877108e4SArnaldo Carvalho de Melo 
862877108e4SArnaldo Carvalho de Melo static int pyrf_evlist__init(struct pyrf_evlist *pevlist,
8631d037ca1SIrina Tirdea 			     PyObject *args, PyObject *kwargs __maybe_unused)
864877108e4SArnaldo Carvalho de Melo {
8657e2ed097SArnaldo Carvalho de Melo 	PyObject *pcpus = NULL, *pthreads = NULL;
8667e2ed097SArnaldo Carvalho de Melo 	struct cpu_map *cpus;
8677e2ed097SArnaldo Carvalho de Melo 	struct thread_map *threads;
8687e2ed097SArnaldo Carvalho de Melo 
8697e2ed097SArnaldo Carvalho de Melo 	if (!PyArg_ParseTuple(args, "OO", &pcpus, &pthreads))
8707e2ed097SArnaldo Carvalho de Melo 		return -1;
8717e2ed097SArnaldo Carvalho de Melo 
8727e2ed097SArnaldo Carvalho de Melo 	threads = ((struct pyrf_thread_map *)pthreads)->threads;
8737e2ed097SArnaldo Carvalho de Melo 	cpus = ((struct pyrf_cpu_map *)pcpus)->cpus;
8747e2ed097SArnaldo Carvalho de Melo 	perf_evlist__init(&pevlist->evlist, cpus, threads);
875877108e4SArnaldo Carvalho de Melo 	return 0;
876877108e4SArnaldo Carvalho de Melo }
877877108e4SArnaldo Carvalho de Melo 
878877108e4SArnaldo Carvalho de Melo static void pyrf_evlist__delete(struct pyrf_evlist *pevlist)
879877108e4SArnaldo Carvalho de Melo {
880877108e4SArnaldo Carvalho de Melo 	perf_evlist__exit(&pevlist->evlist);
88166dfdff0SJaroslav Škarvada 	Py_TYPE(pevlist)->tp_free((PyObject*)pevlist);
882877108e4SArnaldo Carvalho de Melo }
883877108e4SArnaldo Carvalho de Melo 
884877108e4SArnaldo Carvalho de Melo static PyObject *pyrf_evlist__mmap(struct pyrf_evlist *pevlist,
885877108e4SArnaldo Carvalho de Melo 				   PyObject *args, PyObject *kwargs)
886877108e4SArnaldo Carvalho de Melo {
887877108e4SArnaldo Carvalho de Melo 	struct perf_evlist *evlist = &pevlist->evlist;
88864348153SFrederic Weisbecker 	static char *kwlist[] = { "pages", "overwrite", NULL };
889877108e4SArnaldo Carvalho de Melo 	int pages = 128, overwrite = false;
890877108e4SArnaldo Carvalho de Melo 
8917e2ed097SArnaldo Carvalho de Melo 	if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ii", kwlist,
8927e2ed097SArnaldo Carvalho de Melo 					 &pages, &overwrite))
893877108e4SArnaldo Carvalho de Melo 		return NULL;
894877108e4SArnaldo Carvalho de Melo 
895f74b9d3aSWang Nan 	if (perf_evlist__mmap(evlist, pages) < 0) {
896877108e4SArnaldo Carvalho de Melo 		PyErr_SetFromErrno(PyExc_OSError);
897877108e4SArnaldo Carvalho de Melo 		return NULL;
898877108e4SArnaldo Carvalho de Melo 	}
899877108e4SArnaldo Carvalho de Melo 
900877108e4SArnaldo Carvalho de Melo 	Py_INCREF(Py_None);
901877108e4SArnaldo Carvalho de Melo 	return Py_None;
902877108e4SArnaldo Carvalho de Melo }
903877108e4SArnaldo Carvalho de Melo 
904877108e4SArnaldo Carvalho de Melo static PyObject *pyrf_evlist__poll(struct pyrf_evlist *pevlist,
905877108e4SArnaldo Carvalho de Melo 				   PyObject *args, PyObject *kwargs)
906877108e4SArnaldo Carvalho de Melo {
907877108e4SArnaldo Carvalho de Melo 	struct perf_evlist *evlist = &pevlist->evlist;
90864348153SFrederic Weisbecker 	static char *kwlist[] = { "timeout", NULL };
909877108e4SArnaldo Carvalho de Melo 	int timeout = -1, n;
910877108e4SArnaldo Carvalho de Melo 
911877108e4SArnaldo Carvalho de Melo 	if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|i", kwlist, &timeout))
912877108e4SArnaldo Carvalho de Melo 		return NULL;
913877108e4SArnaldo Carvalho de Melo 
914f66a889dSArnaldo Carvalho de Melo 	n = perf_evlist__poll(evlist, timeout);
915877108e4SArnaldo Carvalho de Melo 	if (n < 0) {
916877108e4SArnaldo Carvalho de Melo 		PyErr_SetFromErrno(PyExc_OSError);
917877108e4SArnaldo Carvalho de Melo 		return NULL;
918877108e4SArnaldo Carvalho de Melo 	}
919877108e4SArnaldo Carvalho de Melo 
920877108e4SArnaldo Carvalho de Melo 	return Py_BuildValue("i", n);
921877108e4SArnaldo Carvalho de Melo }
922877108e4SArnaldo Carvalho de Melo 
923877108e4SArnaldo Carvalho de Melo static PyObject *pyrf_evlist__get_pollfd(struct pyrf_evlist *pevlist,
9241d037ca1SIrina Tirdea 					 PyObject *args __maybe_unused,
9251d037ca1SIrina Tirdea 					 PyObject *kwargs __maybe_unused)
926877108e4SArnaldo Carvalho de Melo {
927877108e4SArnaldo Carvalho de Melo 	struct perf_evlist *evlist = &pevlist->evlist;
928877108e4SArnaldo Carvalho de Melo         PyObject *list = PyList_New(0);
929877108e4SArnaldo Carvalho de Melo 	int i;
930877108e4SArnaldo Carvalho de Melo 
9311b85337dSArnaldo Carvalho de Melo 	for (i = 0; i < evlist->pollfd.nr; ++i) {
932877108e4SArnaldo Carvalho de Melo 		PyObject *file;
93366dfdff0SJaroslav Škarvada #if PY_MAJOR_VERSION < 3
9341b85337dSArnaldo Carvalho de Melo 		FILE *fp = fdopen(evlist->pollfd.entries[i].fd, "r");
935877108e4SArnaldo Carvalho de Melo 
936877108e4SArnaldo Carvalho de Melo 		if (fp == NULL)
937877108e4SArnaldo Carvalho de Melo 			goto free_list;
938877108e4SArnaldo Carvalho de Melo 
939877108e4SArnaldo Carvalho de Melo 		file = PyFile_FromFile(fp, "perf", "r", NULL);
94066dfdff0SJaroslav Škarvada #else
94166dfdff0SJaroslav Škarvada 		file = PyFile_FromFd(evlist->pollfd.entries[i].fd, "perf", "r", -1, NULL, NULL, NULL, 1);
94266dfdff0SJaroslav Škarvada #endif
943877108e4SArnaldo Carvalho de Melo 		if (file == NULL)
944877108e4SArnaldo Carvalho de Melo 			goto free_list;
945877108e4SArnaldo Carvalho de Melo 
946877108e4SArnaldo Carvalho de Melo 		if (PyList_Append(list, file) != 0) {
947877108e4SArnaldo Carvalho de Melo 			Py_DECREF(file);
948877108e4SArnaldo Carvalho de Melo 			goto free_list;
949877108e4SArnaldo Carvalho de Melo 		}
950877108e4SArnaldo Carvalho de Melo 
951877108e4SArnaldo Carvalho de Melo 		Py_DECREF(file);
952877108e4SArnaldo Carvalho de Melo 	}
953877108e4SArnaldo Carvalho de Melo 
954877108e4SArnaldo Carvalho de Melo 	return list;
955877108e4SArnaldo Carvalho de Melo free_list:
956877108e4SArnaldo Carvalho de Melo 	return PyErr_NoMemory();
957877108e4SArnaldo Carvalho de Melo }
958877108e4SArnaldo Carvalho de Melo 
959877108e4SArnaldo Carvalho de Melo 
960877108e4SArnaldo Carvalho de Melo static PyObject *pyrf_evlist__add(struct pyrf_evlist *pevlist,
9611d037ca1SIrina Tirdea 				  PyObject *args,
9621d037ca1SIrina Tirdea 				  PyObject *kwargs __maybe_unused)
963877108e4SArnaldo Carvalho de Melo {
964877108e4SArnaldo Carvalho de Melo 	struct perf_evlist *evlist = &pevlist->evlist;
965877108e4SArnaldo Carvalho de Melo 	PyObject *pevsel;
966877108e4SArnaldo Carvalho de Melo 	struct perf_evsel *evsel;
967877108e4SArnaldo Carvalho de Melo 
968877108e4SArnaldo Carvalho de Melo 	if (!PyArg_ParseTuple(args, "O", &pevsel))
969877108e4SArnaldo Carvalho de Melo 		return NULL;
970877108e4SArnaldo Carvalho de Melo 
971877108e4SArnaldo Carvalho de Melo 	Py_INCREF(pevsel);
972877108e4SArnaldo Carvalho de Melo 	evsel = &((struct pyrf_evsel *)pevsel)->evsel;
973877108e4SArnaldo Carvalho de Melo 	evsel->idx = evlist->nr_entries;
974877108e4SArnaldo Carvalho de Melo 	perf_evlist__add(evlist, evsel);
975877108e4SArnaldo Carvalho de Melo 
976877108e4SArnaldo Carvalho de Melo 	return Py_BuildValue("i", evlist->nr_entries);
977877108e4SArnaldo Carvalho de Melo }
978877108e4SArnaldo Carvalho de Melo 
979877108e4SArnaldo Carvalho de Melo static PyObject *pyrf_evlist__read_on_cpu(struct pyrf_evlist *pevlist,
980877108e4SArnaldo Carvalho de Melo 					  PyObject *args, PyObject *kwargs)
981877108e4SArnaldo Carvalho de Melo {
982877108e4SArnaldo Carvalho de Melo 	struct perf_evlist *evlist = &pevlist->evlist;
983877108e4SArnaldo Carvalho de Melo 	union perf_event *event;
984877108e4SArnaldo Carvalho de Melo 	int sample_id_all = 1, cpu;
98564348153SFrederic Weisbecker 	static char *kwlist[] = { "cpu", "sample_id_all", NULL };
98635b7cdc6SKan Liang 	struct perf_mmap *md;
98735b7cdc6SKan Liang 	u64 end, start;
9885538becaSFrederic Weisbecker 	int err;
989877108e4SArnaldo Carvalho de Melo 
990877108e4SArnaldo Carvalho de Melo 	if (!PyArg_ParseTupleAndKeywords(args, kwargs, "i|i", kwlist,
991877108e4SArnaldo Carvalho de Melo 					 &cpu, &sample_id_all))
992877108e4SArnaldo Carvalho de Melo 		return NULL;
993877108e4SArnaldo Carvalho de Melo 
99435b7cdc6SKan Liang 	md = &evlist->mmap[cpu];
99535b7cdc6SKan Liang 	if (perf_mmap__read_init(md, false, &start, &end) < 0)
99635b7cdc6SKan Liang 		goto end;
99735b7cdc6SKan Liang 
998*0019dc87SKan Liang 	event = perf_mmap__read_event(md);
999877108e4SArnaldo Carvalho de Melo 	if (event != NULL) {
1000877108e4SArnaldo Carvalho de Melo 		PyObject *pyevent = pyrf_event__new(event);
1001877108e4SArnaldo Carvalho de Melo 		struct pyrf_event *pevent = (struct pyrf_event *)pyevent;
1002377f698dSJiri Olsa 		struct perf_evsel *evsel;
1003877108e4SArnaldo Carvalho de Melo 
1004877108e4SArnaldo Carvalho de Melo 		if (pyevent == NULL)
1005877108e4SArnaldo Carvalho de Melo 			return PyErr_NoMemory();
1006877108e4SArnaldo Carvalho de Melo 
1007377f698dSJiri Olsa 		evsel = perf_evlist__event2evsel(evlist, event);
1008377f698dSJiri Olsa 		if (!evsel)
1009377f698dSJiri Olsa 			return Py_None;
1010377f698dSJiri Olsa 
1011377f698dSJiri Olsa 		pevent->evsel = evsel;
1012377f698dSJiri Olsa 
1013377f698dSJiri Olsa 		err = perf_evsel__parse_sample(evsel, event, &pevent->sample);
1014e8968e65SJiri Olsa 
1015e8968e65SJiri Olsa 		/* Consume the even only after we parsed it out. */
1016d6ace3dfSKan Liang 		perf_mmap__consume(md);
1017e8968e65SJiri Olsa 
10185c6970afSArnaldo Carvalho de Melo 		if (err)
10195c6970afSArnaldo Carvalho de Melo 			return PyErr_Format(PyExc_OSError,
10205c6970afSArnaldo Carvalho de Melo 					    "perf: can't parse sample, err=%d", err);
10215538becaSFrederic Weisbecker 		return pyevent;
10225538becaSFrederic Weisbecker 	}
102335b7cdc6SKan Liang end:
1024877108e4SArnaldo Carvalho de Melo 	Py_INCREF(Py_None);
1025877108e4SArnaldo Carvalho de Melo 	return Py_None;
1026877108e4SArnaldo Carvalho de Melo }
1027877108e4SArnaldo Carvalho de Melo 
1028727ab04eSArnaldo Carvalho de Melo static PyObject *pyrf_evlist__open(struct pyrf_evlist *pevlist,
1029727ab04eSArnaldo Carvalho de Melo 				   PyObject *args, PyObject *kwargs)
1030727ab04eSArnaldo Carvalho de Melo {
1031727ab04eSArnaldo Carvalho de Melo 	struct perf_evlist *evlist = &pevlist->evlist;
1032727ab04eSArnaldo Carvalho de Melo 	int group = 0;
1033727ab04eSArnaldo Carvalho de Melo 	static char *kwlist[] = { "group", NULL };
1034727ab04eSArnaldo Carvalho de Melo 
1035727ab04eSArnaldo Carvalho de Melo 	if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|OOii", kwlist, &group))
1036727ab04eSArnaldo Carvalho de Melo 		return NULL;
1037727ab04eSArnaldo Carvalho de Melo 
10386a4bb04cSJiri Olsa 	if (group)
103963dab225SArnaldo Carvalho de Melo 		perf_evlist__set_leader(evlist);
10406a4bb04cSJiri Olsa 
10416a4bb04cSJiri Olsa 	if (perf_evlist__open(evlist) < 0) {
1042727ab04eSArnaldo Carvalho de Melo 		PyErr_SetFromErrno(PyExc_OSError);
1043727ab04eSArnaldo Carvalho de Melo 		return NULL;
1044727ab04eSArnaldo Carvalho de Melo 	}
1045727ab04eSArnaldo Carvalho de Melo 
1046727ab04eSArnaldo Carvalho de Melo 	Py_INCREF(Py_None);
1047727ab04eSArnaldo Carvalho de Melo 	return Py_None;
1048727ab04eSArnaldo Carvalho de Melo }
1049727ab04eSArnaldo Carvalho de Melo 
1050877108e4SArnaldo Carvalho de Melo static PyMethodDef pyrf_evlist__methods[] = {
1051877108e4SArnaldo Carvalho de Melo 	{
1052877108e4SArnaldo Carvalho de Melo 		.ml_name  = "mmap",
1053877108e4SArnaldo Carvalho de Melo 		.ml_meth  = (PyCFunction)pyrf_evlist__mmap,
1054877108e4SArnaldo Carvalho de Melo 		.ml_flags = METH_VARARGS | METH_KEYWORDS,
1055877108e4SArnaldo Carvalho de Melo 		.ml_doc	  = PyDoc_STR("mmap the file descriptor table.")
1056877108e4SArnaldo Carvalho de Melo 	},
1057877108e4SArnaldo Carvalho de Melo 	{
1058727ab04eSArnaldo Carvalho de Melo 		.ml_name  = "open",
1059727ab04eSArnaldo Carvalho de Melo 		.ml_meth  = (PyCFunction)pyrf_evlist__open,
1060727ab04eSArnaldo Carvalho de Melo 		.ml_flags = METH_VARARGS | METH_KEYWORDS,
1061727ab04eSArnaldo Carvalho de Melo 		.ml_doc	  = PyDoc_STR("open the file descriptors.")
1062727ab04eSArnaldo Carvalho de Melo 	},
1063727ab04eSArnaldo Carvalho de Melo 	{
1064877108e4SArnaldo Carvalho de Melo 		.ml_name  = "poll",
1065877108e4SArnaldo Carvalho de Melo 		.ml_meth  = (PyCFunction)pyrf_evlist__poll,
1066877108e4SArnaldo Carvalho de Melo 		.ml_flags = METH_VARARGS | METH_KEYWORDS,
1067877108e4SArnaldo Carvalho de Melo 		.ml_doc	  = PyDoc_STR("poll the file descriptor table.")
1068877108e4SArnaldo Carvalho de Melo 	},
1069877108e4SArnaldo Carvalho de Melo 	{
1070877108e4SArnaldo Carvalho de Melo 		.ml_name  = "get_pollfd",
1071877108e4SArnaldo Carvalho de Melo 		.ml_meth  = (PyCFunction)pyrf_evlist__get_pollfd,
1072877108e4SArnaldo Carvalho de Melo 		.ml_flags = METH_VARARGS | METH_KEYWORDS,
1073877108e4SArnaldo Carvalho de Melo 		.ml_doc	  = PyDoc_STR("get the poll file descriptor table.")
1074877108e4SArnaldo Carvalho de Melo 	},
1075877108e4SArnaldo Carvalho de Melo 	{
1076877108e4SArnaldo Carvalho de Melo 		.ml_name  = "add",
1077877108e4SArnaldo Carvalho de Melo 		.ml_meth  = (PyCFunction)pyrf_evlist__add,
1078877108e4SArnaldo Carvalho de Melo 		.ml_flags = METH_VARARGS | METH_KEYWORDS,
1079877108e4SArnaldo Carvalho de Melo 		.ml_doc	  = PyDoc_STR("adds an event selector to the list.")
1080877108e4SArnaldo Carvalho de Melo 	},
1081877108e4SArnaldo Carvalho de Melo 	{
1082877108e4SArnaldo Carvalho de Melo 		.ml_name  = "read_on_cpu",
1083877108e4SArnaldo Carvalho de Melo 		.ml_meth  = (PyCFunction)pyrf_evlist__read_on_cpu,
1084877108e4SArnaldo Carvalho de Melo 		.ml_flags = METH_VARARGS | METH_KEYWORDS,
1085877108e4SArnaldo Carvalho de Melo 		.ml_doc	  = PyDoc_STR("reads an event.")
1086877108e4SArnaldo Carvalho de Melo 	},
1087f6bbc1daSArnaldo Carvalho de Melo 	{ .ml_name = NULL, }
1088877108e4SArnaldo Carvalho de Melo };
1089877108e4SArnaldo Carvalho de Melo 
1090877108e4SArnaldo Carvalho de Melo static Py_ssize_t pyrf_evlist__length(PyObject *obj)
1091877108e4SArnaldo Carvalho de Melo {
1092877108e4SArnaldo Carvalho de Melo 	struct pyrf_evlist *pevlist = (void *)obj;
1093877108e4SArnaldo Carvalho de Melo 
1094877108e4SArnaldo Carvalho de Melo 	return pevlist->evlist.nr_entries;
1095877108e4SArnaldo Carvalho de Melo }
1096877108e4SArnaldo Carvalho de Melo 
1097877108e4SArnaldo Carvalho de Melo static PyObject *pyrf_evlist__item(PyObject *obj, Py_ssize_t i)
1098877108e4SArnaldo Carvalho de Melo {
1099877108e4SArnaldo Carvalho de Melo 	struct pyrf_evlist *pevlist = (void *)obj;
1100877108e4SArnaldo Carvalho de Melo 	struct perf_evsel *pos;
1101877108e4SArnaldo Carvalho de Melo 
1102877108e4SArnaldo Carvalho de Melo 	if (i >= pevlist->evlist.nr_entries)
1103877108e4SArnaldo Carvalho de Melo 		return NULL;
1104877108e4SArnaldo Carvalho de Melo 
1105e5cadb93SArnaldo Carvalho de Melo 	evlist__for_each_entry(&pevlist->evlist, pos) {
1106877108e4SArnaldo Carvalho de Melo 		if (i-- == 0)
1107877108e4SArnaldo Carvalho de Melo 			break;
11080050f7aaSArnaldo Carvalho de Melo 	}
1109877108e4SArnaldo Carvalho de Melo 
1110877108e4SArnaldo Carvalho de Melo 	return Py_BuildValue("O", container_of(pos, struct pyrf_evsel, evsel));
1111877108e4SArnaldo Carvalho de Melo }
1112877108e4SArnaldo Carvalho de Melo 
1113877108e4SArnaldo Carvalho de Melo static PySequenceMethods pyrf_evlist__sequence_methods = {
1114877108e4SArnaldo Carvalho de Melo 	.sq_length = pyrf_evlist__length,
1115877108e4SArnaldo Carvalho de Melo 	.sq_item   = pyrf_evlist__item,
1116877108e4SArnaldo Carvalho de Melo };
1117877108e4SArnaldo Carvalho de Melo 
1118877108e4SArnaldo Carvalho de Melo static char pyrf_evlist__doc[] = PyDoc_STR("perf event selector list object.");
1119877108e4SArnaldo Carvalho de Melo 
1120877108e4SArnaldo Carvalho de Melo static PyTypeObject pyrf_evlist__type = {
1121877108e4SArnaldo Carvalho de Melo 	PyVarObject_HEAD_INIT(NULL, 0)
1122877108e4SArnaldo Carvalho de Melo 	.tp_name	= "perf.evlist",
1123877108e4SArnaldo Carvalho de Melo 	.tp_basicsize	= sizeof(struct pyrf_evlist),
1124877108e4SArnaldo Carvalho de Melo 	.tp_dealloc	= (destructor)pyrf_evlist__delete,
1125877108e4SArnaldo Carvalho de Melo 	.tp_flags	= Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,
1126877108e4SArnaldo Carvalho de Melo 	.tp_as_sequence	= &pyrf_evlist__sequence_methods,
1127877108e4SArnaldo Carvalho de Melo 	.tp_doc		= pyrf_evlist__doc,
1128877108e4SArnaldo Carvalho de Melo 	.tp_methods	= pyrf_evlist__methods,
1129877108e4SArnaldo Carvalho de Melo 	.tp_init	= (initproc)pyrf_evlist__init,
1130877108e4SArnaldo Carvalho de Melo };
1131877108e4SArnaldo Carvalho de Melo 
1132877108e4SArnaldo Carvalho de Melo static int pyrf_evlist__setup_types(void)
1133877108e4SArnaldo Carvalho de Melo {
1134877108e4SArnaldo Carvalho de Melo 	pyrf_evlist__type.tp_new = PyType_GenericNew;
1135877108e4SArnaldo Carvalho de Melo 	return PyType_Ready(&pyrf_evlist__type);
1136877108e4SArnaldo Carvalho de Melo }
1137877108e4SArnaldo Carvalho de Melo 
11385865fe36SArnaldo Carvalho de Melo #define PERF_CONST(name) { #name, PERF_##name }
11395865fe36SArnaldo Carvalho de Melo 
1140877108e4SArnaldo Carvalho de Melo static struct {
1141877108e4SArnaldo Carvalho de Melo 	const char *name;
1142877108e4SArnaldo Carvalho de Melo 	int	    value;
1143877108e4SArnaldo Carvalho de Melo } perf__constants[] = {
11445865fe36SArnaldo Carvalho de Melo 	PERF_CONST(TYPE_HARDWARE),
11455865fe36SArnaldo Carvalho de Melo 	PERF_CONST(TYPE_SOFTWARE),
11465865fe36SArnaldo Carvalho de Melo 	PERF_CONST(TYPE_TRACEPOINT),
11475865fe36SArnaldo Carvalho de Melo 	PERF_CONST(TYPE_HW_CACHE),
11485865fe36SArnaldo Carvalho de Melo 	PERF_CONST(TYPE_RAW),
11495865fe36SArnaldo Carvalho de Melo 	PERF_CONST(TYPE_BREAKPOINT),
1150877108e4SArnaldo Carvalho de Melo 
11515865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_CPU_CYCLES),
11525865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_INSTRUCTIONS),
11535865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_CACHE_REFERENCES),
11545865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_CACHE_MISSES),
11555865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_BRANCH_INSTRUCTIONS),
11565865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_BRANCH_MISSES),
11575865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_BUS_CYCLES),
11585865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_CACHE_L1D),
11595865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_CACHE_L1I),
11605865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_CACHE_LL),
11615865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_CACHE_DTLB),
11625865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_CACHE_ITLB),
11635865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_CACHE_BPU),
11645865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_CACHE_OP_READ),
11655865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_CACHE_OP_WRITE),
11665865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_CACHE_OP_PREFETCH),
11675865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_CACHE_RESULT_ACCESS),
11685865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_CACHE_RESULT_MISS),
1169877108e4SArnaldo Carvalho de Melo 
11705865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_STALLED_CYCLES_FRONTEND),
11715865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_HW_STALLED_CYCLES_BACKEND),
1172129c04cbSIngo Molnar 
11735865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_SW_CPU_CLOCK),
11745865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_SW_TASK_CLOCK),
11755865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_SW_PAGE_FAULTS),
11765865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_SW_CONTEXT_SWITCHES),
11775865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_SW_CPU_MIGRATIONS),
11785865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_SW_PAGE_FAULTS_MIN),
11795865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_SW_PAGE_FAULTS_MAJ),
11805865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_SW_ALIGNMENT_FAULTS),
11815865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_SW_EMULATION_FAULTS),
11825865fe36SArnaldo Carvalho de Melo 	PERF_CONST(COUNT_SW_DUMMY),
1183877108e4SArnaldo Carvalho de Melo 
11845865fe36SArnaldo Carvalho de Melo 	PERF_CONST(SAMPLE_IP),
11855865fe36SArnaldo Carvalho de Melo 	PERF_CONST(SAMPLE_TID),
11865865fe36SArnaldo Carvalho de Melo 	PERF_CONST(SAMPLE_TIME),
11875865fe36SArnaldo Carvalho de Melo 	PERF_CONST(SAMPLE_ADDR),
11885865fe36SArnaldo Carvalho de Melo 	PERF_CONST(SAMPLE_READ),
11895865fe36SArnaldo Carvalho de Melo 	PERF_CONST(SAMPLE_CALLCHAIN),
11905865fe36SArnaldo Carvalho de Melo 	PERF_CONST(SAMPLE_ID),
11915865fe36SArnaldo Carvalho de Melo 	PERF_CONST(SAMPLE_CPU),
11925865fe36SArnaldo Carvalho de Melo 	PERF_CONST(SAMPLE_PERIOD),
11935865fe36SArnaldo Carvalho de Melo 	PERF_CONST(SAMPLE_STREAM_ID),
11945865fe36SArnaldo Carvalho de Melo 	PERF_CONST(SAMPLE_RAW),
1195877108e4SArnaldo Carvalho de Melo 
11965865fe36SArnaldo Carvalho de Melo 	PERF_CONST(FORMAT_TOTAL_TIME_ENABLED),
11975865fe36SArnaldo Carvalho de Melo 	PERF_CONST(FORMAT_TOTAL_TIME_RUNNING),
11985865fe36SArnaldo Carvalho de Melo 	PERF_CONST(FORMAT_ID),
11995865fe36SArnaldo Carvalho de Melo 	PERF_CONST(FORMAT_GROUP),
1200877108e4SArnaldo Carvalho de Melo 
12015865fe36SArnaldo Carvalho de Melo 	PERF_CONST(RECORD_MMAP),
12025865fe36SArnaldo Carvalho de Melo 	PERF_CONST(RECORD_LOST),
12035865fe36SArnaldo Carvalho de Melo 	PERF_CONST(RECORD_COMM),
12045865fe36SArnaldo Carvalho de Melo 	PERF_CONST(RECORD_EXIT),
12055865fe36SArnaldo Carvalho de Melo 	PERF_CONST(RECORD_THROTTLE),
12065865fe36SArnaldo Carvalho de Melo 	PERF_CONST(RECORD_UNTHROTTLE),
12075865fe36SArnaldo Carvalho de Melo 	PERF_CONST(RECORD_FORK),
12085865fe36SArnaldo Carvalho de Melo 	PERF_CONST(RECORD_READ),
12095865fe36SArnaldo Carvalho de Melo 	PERF_CONST(RECORD_SAMPLE),
121084576da2SArnaldo Carvalho de Melo 	PERF_CONST(RECORD_MMAP2),
121184576da2SArnaldo Carvalho de Melo 	PERF_CONST(RECORD_AUX),
121284576da2SArnaldo Carvalho de Melo 	PERF_CONST(RECORD_ITRACE_START),
121384576da2SArnaldo Carvalho de Melo 	PERF_CONST(RECORD_LOST_SAMPLES),
121484576da2SArnaldo Carvalho de Melo 	PERF_CONST(RECORD_SWITCH),
121584576da2SArnaldo Carvalho de Melo 	PERF_CONST(RECORD_SWITCH_CPU_WIDE),
1216ae938802SArnaldo Carvalho de Melo 
1217ae938802SArnaldo Carvalho de Melo 	PERF_CONST(RECORD_MISC_SWITCH_OUT),
1218f6bbc1daSArnaldo Carvalho de Melo 	{ .name = NULL, },
1219877108e4SArnaldo Carvalho de Melo };
1220877108e4SArnaldo Carvalho de Melo 
12211075fbb2SJiri Olsa static PyObject *pyrf__tracepoint(struct pyrf_evsel *pevsel,
12221075fbb2SJiri Olsa 				  PyObject *args, PyObject *kwargs)
12231075fbb2SJiri Olsa {
12241075fbb2SJiri Olsa 	struct event_format *tp_format;
12251075fbb2SJiri Olsa 	static char *kwlist[] = { "sys", "name", NULL };
12261075fbb2SJiri Olsa 	char *sys  = NULL;
12271075fbb2SJiri Olsa 	char *name = NULL;
12281075fbb2SJiri Olsa 
12291075fbb2SJiri Olsa 	if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ss", kwlist,
12301075fbb2SJiri Olsa 					 &sys, &name))
12311075fbb2SJiri Olsa 		return NULL;
12321075fbb2SJiri Olsa 
12331075fbb2SJiri Olsa 	tp_format = trace_event__tp_format(sys, name);
12341075fbb2SJiri Olsa 	if (IS_ERR(tp_format))
123566dfdff0SJaroslav Škarvada 		return _PyLong_FromLong(-1);
12361075fbb2SJiri Olsa 
123766dfdff0SJaroslav Škarvada 	return _PyLong_FromLong(tp_format->id);
12381075fbb2SJiri Olsa }
12391075fbb2SJiri Olsa 
1240877108e4SArnaldo Carvalho de Melo static PyMethodDef perf__methods[] = {
12411075fbb2SJiri Olsa 	{
12421075fbb2SJiri Olsa 		.ml_name  = "tracepoint",
12431075fbb2SJiri Olsa 		.ml_meth  = (PyCFunction) pyrf__tracepoint,
12441075fbb2SJiri Olsa 		.ml_flags = METH_VARARGS | METH_KEYWORDS,
12451075fbb2SJiri Olsa 		.ml_doc	  = PyDoc_STR("Get tracepoint config.")
12461075fbb2SJiri Olsa 	},
1247f6bbc1daSArnaldo Carvalho de Melo 	{ .ml_name = NULL, }
1248877108e4SArnaldo Carvalho de Melo };
1249877108e4SArnaldo Carvalho de Melo 
125066dfdff0SJaroslav Škarvada #if PY_MAJOR_VERSION < 3
1251877108e4SArnaldo Carvalho de Melo PyMODINIT_FUNC initperf(void)
125266dfdff0SJaroslav Škarvada #else
125366dfdff0SJaroslav Škarvada PyMODINIT_FUNC PyInit_perf(void)
125466dfdff0SJaroslav Škarvada #endif
1255877108e4SArnaldo Carvalho de Melo {
1256877108e4SArnaldo Carvalho de Melo 	PyObject *obj;
1257877108e4SArnaldo Carvalho de Melo 	int i;
125866dfdff0SJaroslav Škarvada 	PyObject *dict;
125966dfdff0SJaroslav Škarvada #if PY_MAJOR_VERSION < 3
126066dfdff0SJaroslav Škarvada 	PyObject *module = Py_InitModule("perf", perf__methods);
126166dfdff0SJaroslav Škarvada #else
126266dfdff0SJaroslav Škarvada 	static struct PyModuleDef moduledef = {
126366dfdff0SJaroslav Škarvada 		PyModuleDef_HEAD_INIT,
126466dfdff0SJaroslav Škarvada 		"perf",			/* m_name */
126566dfdff0SJaroslav Škarvada 		"",			/* m_doc */
126666dfdff0SJaroslav Škarvada 		-1,			/* m_size */
126766dfdff0SJaroslav Škarvada 		perf__methods,		/* m_methods */
126866dfdff0SJaroslav Škarvada 		NULL,			/* m_reload */
126966dfdff0SJaroslav Škarvada 		NULL,			/* m_traverse */
127066dfdff0SJaroslav Škarvada 		NULL,			/* m_clear */
127166dfdff0SJaroslav Škarvada 		NULL,			/* m_free */
127266dfdff0SJaroslav Škarvada 	};
127366dfdff0SJaroslav Škarvada 	PyObject *module = PyModule_Create(&moduledef);
127466dfdff0SJaroslav Škarvada #endif
1275877108e4SArnaldo Carvalho de Melo 
1276877108e4SArnaldo Carvalho de Melo 	if (module == NULL ||
1277877108e4SArnaldo Carvalho de Melo 	    pyrf_event__setup_types() < 0 ||
1278877108e4SArnaldo Carvalho de Melo 	    pyrf_evlist__setup_types() < 0 ||
1279877108e4SArnaldo Carvalho de Melo 	    pyrf_evsel__setup_types() < 0 ||
1280877108e4SArnaldo Carvalho de Melo 	    pyrf_thread_map__setup_types() < 0 ||
1281877108e4SArnaldo Carvalho de Melo 	    pyrf_cpu_map__setup_types() < 0)
128266dfdff0SJaroslav Škarvada #if PY_MAJOR_VERSION < 3
1283877108e4SArnaldo Carvalho de Melo 		return;
128466dfdff0SJaroslav Škarvada #else
128566dfdff0SJaroslav Škarvada 		return module;
128666dfdff0SJaroslav Škarvada #endif
1287877108e4SArnaldo Carvalho de Melo 
1288918512b4SJiri Olsa 	/* The page_size is placed in util object. */
12890da2e9c2SArnaldo Carvalho de Melo 	page_size = sysconf(_SC_PAGE_SIZE);
12900da2e9c2SArnaldo Carvalho de Melo 
1291877108e4SArnaldo Carvalho de Melo 	Py_INCREF(&pyrf_evlist__type);
1292877108e4SArnaldo Carvalho de Melo 	PyModule_AddObject(module, "evlist", (PyObject*)&pyrf_evlist__type);
1293877108e4SArnaldo Carvalho de Melo 
1294877108e4SArnaldo Carvalho de Melo 	Py_INCREF(&pyrf_evsel__type);
1295877108e4SArnaldo Carvalho de Melo 	PyModule_AddObject(module, "evsel", (PyObject*)&pyrf_evsel__type);
1296877108e4SArnaldo Carvalho de Melo 
129785e37de3SJiri Olsa 	Py_INCREF(&pyrf_mmap_event__type);
129885e37de3SJiri Olsa 	PyModule_AddObject(module, "mmap_event", (PyObject *)&pyrf_mmap_event__type);
129985e37de3SJiri Olsa 
130085e37de3SJiri Olsa 	Py_INCREF(&pyrf_lost_event__type);
130185e37de3SJiri Olsa 	PyModule_AddObject(module, "lost_event", (PyObject *)&pyrf_lost_event__type);
130285e37de3SJiri Olsa 
130385e37de3SJiri Olsa 	Py_INCREF(&pyrf_comm_event__type);
130485e37de3SJiri Olsa 	PyModule_AddObject(module, "comm_event", (PyObject *)&pyrf_comm_event__type);
130585e37de3SJiri Olsa 
130685e37de3SJiri Olsa 	Py_INCREF(&pyrf_task_event__type);
130785e37de3SJiri Olsa 	PyModule_AddObject(module, "task_event", (PyObject *)&pyrf_task_event__type);
130885e37de3SJiri Olsa 
130985e37de3SJiri Olsa 	Py_INCREF(&pyrf_throttle_event__type);
131085e37de3SJiri Olsa 	PyModule_AddObject(module, "throttle_event", (PyObject *)&pyrf_throttle_event__type);
131185e37de3SJiri Olsa 
131285e37de3SJiri Olsa 	Py_INCREF(&pyrf_task_event__type);
131385e37de3SJiri Olsa 	PyModule_AddObject(module, "task_event", (PyObject *)&pyrf_task_event__type);
131485e37de3SJiri Olsa 
131585e37de3SJiri Olsa 	Py_INCREF(&pyrf_read_event__type);
131685e37de3SJiri Olsa 	PyModule_AddObject(module, "read_event", (PyObject *)&pyrf_read_event__type);
131785e37de3SJiri Olsa 
131885e37de3SJiri Olsa 	Py_INCREF(&pyrf_sample_event__type);
131985e37de3SJiri Olsa 	PyModule_AddObject(module, "sample_event", (PyObject *)&pyrf_sample_event__type);
132085e37de3SJiri Olsa 
132185e37de3SJiri Olsa 	Py_INCREF(&pyrf_context_switch_event__type);
132285e37de3SJiri Olsa 	PyModule_AddObject(module, "switch_event", (PyObject *)&pyrf_context_switch_event__type);
132385e37de3SJiri Olsa 
1324877108e4SArnaldo Carvalho de Melo 	Py_INCREF(&pyrf_thread_map__type);
1325877108e4SArnaldo Carvalho de Melo 	PyModule_AddObject(module, "thread_map", (PyObject*)&pyrf_thread_map__type);
1326877108e4SArnaldo Carvalho de Melo 
1327877108e4SArnaldo Carvalho de Melo 	Py_INCREF(&pyrf_cpu_map__type);
1328877108e4SArnaldo Carvalho de Melo 	PyModule_AddObject(module, "cpu_map", (PyObject*)&pyrf_cpu_map__type);
1329877108e4SArnaldo Carvalho de Melo 
1330877108e4SArnaldo Carvalho de Melo 	dict = PyModule_GetDict(module);
1331877108e4SArnaldo Carvalho de Melo 	if (dict == NULL)
1332877108e4SArnaldo Carvalho de Melo 		goto error;
1333877108e4SArnaldo Carvalho de Melo 
1334877108e4SArnaldo Carvalho de Melo 	for (i = 0; perf__constants[i].name != NULL; i++) {
133566dfdff0SJaroslav Škarvada 		obj = _PyLong_FromLong(perf__constants[i].value);
1336877108e4SArnaldo Carvalho de Melo 		if (obj == NULL)
1337877108e4SArnaldo Carvalho de Melo 			goto error;
1338877108e4SArnaldo Carvalho de Melo 		PyDict_SetItemString(dict, perf__constants[i].name, obj);
1339877108e4SArnaldo Carvalho de Melo 		Py_DECREF(obj);
1340877108e4SArnaldo Carvalho de Melo 	}
1341877108e4SArnaldo Carvalho de Melo 
1342877108e4SArnaldo Carvalho de Melo error:
1343877108e4SArnaldo Carvalho de Melo 	if (PyErr_Occurred())
1344877108e4SArnaldo Carvalho de Melo 		PyErr_SetString(PyExc_ImportError, "perf: Init failed!");
134566dfdff0SJaroslav Škarvada #if PY_MAJOR_VERSION >= 3
134666dfdff0SJaroslav Škarvada 	return module;
134766dfdff0SJaroslav Škarvada #endif
1348877108e4SArnaldo Carvalho de Melo }
13490c6332e9SArnaldo Carvalho de Melo 
13500c6332e9SArnaldo Carvalho de Melo /*
13510c6332e9SArnaldo Carvalho de Melo  * Dummy, to avoid dragging all the test_attr infrastructure in the python
13520c6332e9SArnaldo Carvalho de Melo  * binding.
13530c6332e9SArnaldo Carvalho de Melo  */
13540c6332e9SArnaldo Carvalho de Melo void test_attr__open(struct perf_event_attr *attr, pid_t pid, int cpu,
13550c6332e9SArnaldo Carvalho de Melo                      int fd, int group_fd, unsigned long flags)
13560c6332e9SArnaldo Carvalho de Melo {
13570c6332e9SArnaldo Carvalho de Melo }
1358