Implement dt_bpf_raw_tracepoint_open()

Signed-off-by: Kris Van Hees <kris.van.hees@oracle.com> Reviewed-by: Nick Alcock <nick.alcock@oracle.com>
oracle · Feb 27, 2023 · 209ee79 · 209ee79
1 parent 31f2a0a
commit 209ee79
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/libdtrace/dt_bpf.c b/libdtrace/dt_bpf.c
@@ -300,6 +300,21 @@ dt_bpf_map_update_inner(int fd, const void *okey, const void *ikey,
 	return rc;
 }
 
+/*
+ * Associate a BPF program (by fd) with a raw tracepoint.
+ */
+int
+dt_bpf_raw_tracepoint_open(const void *tp, int fd)
+{
+	union bpf_attr	attr;
+
+	memset(&attr, 0, sizeof(attr));
+	attr.raw_tracepoint.name = (uint64_t)(unsigned long)tp;
+	attr.raw_tracepoint.prog_fd = fd;
+
+	return bpf(BPF_RAW_TRACEPOINT_OPEN, &attr);
+}
+
 static int
 have_helper(uint32_t func_id)
 {

diff --git a/libdtrace/dt_bpf.h b/libdtrace/dt_bpf.h
@@ -56,6 +56,7 @@ extern int dt_bpf_map_lookup_inner(int fd, const void *okey, const void *ikey,
 				   void *val);
 extern int dt_bpf_map_update_inner(int fd, const void *okey, const void *ikey,
 				   const void *val);
+extern int dt_bpf_raw_tracepoint_open(const void *tp, int fd);
 extern int dt_bpf_load_progs(struct dtrace_hdl *, uint_t);
 extern void dt_bpf_init_helpers(struct dtrace_hdl *dtp);