Trace-cruncher low level wrappers for tracefs library APIs for
uprobe and uretprobe allocation.
Signed-off-by: Tzvetomir Stoyanov (VMware) <tz.stoyanov@xxxxxxxxx>
---
src/ftracepy-utils.c | 48 ++++++++++++++++++++++++++++++++++++++++++++
src/ftracepy-utils.h | 4 ++++
src/ftracepy.c | 10 +++++++++
3 files changed, 62 insertions(+)
diff --git a/src/ftracepy-utils.c b/src/ftracepy-utils.c
index e8411ae..b39459b 100644
--- a/src/ftracepy-utils.c
+++ b/src/ftracepy-utils.c
@@ -2449,6 +2449,54 @@ PyObject *PyFtrace_eprobe(PyObject *self, PyObject *args, PyObject *kwargs)
return py_dyn;
}
+static PyObject *alloc_uprobe(PyObject *self, PyObject *args, PyObject *kwargs, bool pret)
+{
+ static char *kwlist[] = {"event", "file", "offset", "fetch_args", NULL};
+ const char *event, *file, *fetchargs = NULL;
+ unsigned long long offset;
+ struct tracefs_dynevent *uprobe;
+ PyObject *py_dyn;
+
+ if (!PyArg_ParseTupleAndKeywords(args,
+ kwargs,
+ "ssK|s",
+ kwlist,
+ &event,
+ &file,
+ &offset,
+ &fetchargs)) {
+ return NULL;
+ }
+
+ if (pret)
+ uprobe = tracefs_uretprobe_alloc(TC_SYS, event, file, offset, fetchargs);
+ else
+ uprobe = tracefs_uprobe_alloc(TC_SYS, event, file, offset, fetchargs);
+ if (!uprobe) {
+ MEM_ERROR;
+ return NULL;
+ }
+
+ py_dyn = PyDynevent_New(uprobe);
+ /*
+ * Here we only allocated and initializes a dynamic event object.
+ * However, no dynamic event is added to the system yet. Hence,
+ * there is no need to 'destroy' this event at exit.
+ */
+ set_destroy_flag(py_dyn, false);
+ return py_dyn;
+}
+
+PyObject *PyFtrace_uprobe(PyObject *self, PyObject *args, PyObject *kwargs)
+{
+ return alloc_uprobe(self, args, kwargs, false);
+}
+
+PyObject *PyFtrace_uretprobe(PyObject *self, PyObject *args, PyObject *kwargs)
+{
+ return alloc_uprobe(self, args, kwargs, true);
+}
+
static PyObject *set_filter(PyObject *args, PyObject *kwargs,
struct tep_handle *tep,
struct tep_event *event)
diff --git a/src/ftracepy-utils.h b/src/ftracepy-utils.h
index 9491b18..e6fab69 100644
--- a/src/ftracepy-utils.h
+++ b/src/ftracepy-utils.h
@@ -257,6 +257,10 @@ PyObject *PyFtrace_kretprobe(PyObject *self, PyObject *args, PyObject *kwargs);
PyObject *PyFtrace_eprobe(PyObject *self, PyObject *args, PyObject *kwargs);
+PyObject *PyFtrace_uprobe(PyObject *self, PyObject *args, PyObject *kwargs);
+
+PyObject *PyFtrace_uretprobe(PyObject *self, PyObject *args, PyObject *kwargs);
+
PyObject *PyFtrace_hist(PyObject *self, PyObject *args,
PyObject *kwargs);
diff --git a/src/ftracepy.c b/src/ftracepy.c
index 2819da2..29d273f 100644
--- a/src/ftracepy.c
+++ b/src/ftracepy.c
@@ -483,6 +483,16 @@ static PyMethodDef ftracepy_methods[] = {
METH_VARARGS | METH_KEYWORDS,
PyFtrace_eprobe_doc,
},
+ {"uprobe",
+ (PyCFunction) PyFtrace_uprobe,
+ METH_VARARGS | METH_KEYWORDS,
+ "Define a uprobe."
+ },
+ {"uretprobe",
+ (PyCFunction) PyFtrace_uretprobe,
+ METH_VARARGS | METH_KEYWORDS,
+ "Define a uretprobe."
+ },
{"hist",
(PyCFunction) PyFtrace_hist,
METH_VARARGS | METH_KEYWORDS,
--
2.35.3
|