1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0-or-later
2*4882a593Smuzhiyun /*
3*4882a593Smuzhiyun * Context.c. Python interfaces for perf script.
4*4882a593Smuzhiyun *
5*4882a593Smuzhiyun * Copyright (C) 2010 Tom Zanussi <tzanussi@gmail.com>
6*4882a593Smuzhiyun */
7*4882a593Smuzhiyun
8*4882a593Smuzhiyun #include <Python.h>
9*4882a593Smuzhiyun #include "../../../util/trace-event.h"
10*4882a593Smuzhiyun
11*4882a593Smuzhiyun #if PY_MAJOR_VERSION < 3
12*4882a593Smuzhiyun #define _PyCapsule_GetPointer(arg1, arg2) \
13*4882a593Smuzhiyun PyCObject_AsVoidPtr(arg1)
14*4882a593Smuzhiyun
15*4882a593Smuzhiyun PyMODINIT_FUNC initperf_trace_context(void);
16*4882a593Smuzhiyun #else
17*4882a593Smuzhiyun #define _PyCapsule_GetPointer(arg1, arg2) \
18*4882a593Smuzhiyun PyCapsule_GetPointer((arg1), (arg2))
19*4882a593Smuzhiyun
20*4882a593Smuzhiyun PyMODINIT_FUNC PyInit_perf_trace_context(void);
21*4882a593Smuzhiyun #endif
22*4882a593Smuzhiyun
perf_trace_context_common_pc(PyObject * obj,PyObject * args)23*4882a593Smuzhiyun static PyObject *perf_trace_context_common_pc(PyObject *obj, PyObject *args)
24*4882a593Smuzhiyun {
25*4882a593Smuzhiyun static struct scripting_context *scripting_context;
26*4882a593Smuzhiyun PyObject *context;
27*4882a593Smuzhiyun int retval;
28*4882a593Smuzhiyun
29*4882a593Smuzhiyun if (!PyArg_ParseTuple(args, "O", &context))
30*4882a593Smuzhiyun return NULL;
31*4882a593Smuzhiyun
32*4882a593Smuzhiyun scripting_context = _PyCapsule_GetPointer(context, NULL);
33*4882a593Smuzhiyun retval = common_pc(scripting_context);
34*4882a593Smuzhiyun
35*4882a593Smuzhiyun return Py_BuildValue("i", retval);
36*4882a593Smuzhiyun }
37*4882a593Smuzhiyun
perf_trace_context_common_flags(PyObject * obj,PyObject * args)38*4882a593Smuzhiyun static PyObject *perf_trace_context_common_flags(PyObject *obj,
39*4882a593Smuzhiyun PyObject *args)
40*4882a593Smuzhiyun {
41*4882a593Smuzhiyun static struct scripting_context *scripting_context;
42*4882a593Smuzhiyun PyObject *context;
43*4882a593Smuzhiyun int retval;
44*4882a593Smuzhiyun
45*4882a593Smuzhiyun if (!PyArg_ParseTuple(args, "O", &context))
46*4882a593Smuzhiyun return NULL;
47*4882a593Smuzhiyun
48*4882a593Smuzhiyun scripting_context = _PyCapsule_GetPointer(context, NULL);
49*4882a593Smuzhiyun retval = common_flags(scripting_context);
50*4882a593Smuzhiyun
51*4882a593Smuzhiyun return Py_BuildValue("i", retval);
52*4882a593Smuzhiyun }
53*4882a593Smuzhiyun
perf_trace_context_common_lock_depth(PyObject * obj,PyObject * args)54*4882a593Smuzhiyun static PyObject *perf_trace_context_common_lock_depth(PyObject *obj,
55*4882a593Smuzhiyun PyObject *args)
56*4882a593Smuzhiyun {
57*4882a593Smuzhiyun static struct scripting_context *scripting_context;
58*4882a593Smuzhiyun PyObject *context;
59*4882a593Smuzhiyun int retval;
60*4882a593Smuzhiyun
61*4882a593Smuzhiyun if (!PyArg_ParseTuple(args, "O", &context))
62*4882a593Smuzhiyun return NULL;
63*4882a593Smuzhiyun
64*4882a593Smuzhiyun scripting_context = _PyCapsule_GetPointer(context, NULL);
65*4882a593Smuzhiyun retval = common_lock_depth(scripting_context);
66*4882a593Smuzhiyun
67*4882a593Smuzhiyun return Py_BuildValue("i", retval);
68*4882a593Smuzhiyun }
69*4882a593Smuzhiyun
70*4882a593Smuzhiyun static PyMethodDef ContextMethods[] = {
71*4882a593Smuzhiyun { "common_pc", perf_trace_context_common_pc, METH_VARARGS,
72*4882a593Smuzhiyun "Get the common preempt count event field value."},
73*4882a593Smuzhiyun { "common_flags", perf_trace_context_common_flags, METH_VARARGS,
74*4882a593Smuzhiyun "Get the common flags event field value."},
75*4882a593Smuzhiyun { "common_lock_depth", perf_trace_context_common_lock_depth,
76*4882a593Smuzhiyun METH_VARARGS, "Get the common lock depth event field value."},
77*4882a593Smuzhiyun { NULL, NULL, 0, NULL}
78*4882a593Smuzhiyun };
79*4882a593Smuzhiyun
80*4882a593Smuzhiyun #if PY_MAJOR_VERSION < 3
initperf_trace_context(void)81*4882a593Smuzhiyun PyMODINIT_FUNC initperf_trace_context(void)
82*4882a593Smuzhiyun {
83*4882a593Smuzhiyun (void) Py_InitModule("perf_trace_context", ContextMethods);
84*4882a593Smuzhiyun }
85*4882a593Smuzhiyun #else
PyInit_perf_trace_context(void)86*4882a593Smuzhiyun PyMODINIT_FUNC PyInit_perf_trace_context(void)
87*4882a593Smuzhiyun {
88*4882a593Smuzhiyun static struct PyModuleDef moduledef = {
89*4882a593Smuzhiyun PyModuleDef_HEAD_INIT,
90*4882a593Smuzhiyun "perf_trace_context", /* m_name */
91*4882a593Smuzhiyun "", /* m_doc */
92*4882a593Smuzhiyun -1, /* m_size */
93*4882a593Smuzhiyun ContextMethods, /* m_methods */
94*4882a593Smuzhiyun NULL, /* m_reload */
95*4882a593Smuzhiyun NULL, /* m_traverse */
96*4882a593Smuzhiyun NULL, /* m_clear */
97*4882a593Smuzhiyun NULL, /* m_free */
98*4882a593Smuzhiyun };
99*4882a593Smuzhiyun return PyModule_Create(&moduledef);
100*4882a593Smuzhiyun }
101*4882a593Smuzhiyun #endif
102