Allowlist some types related to perf event buffers

kxxt · Apr 19, 2022 · 11f9379 · 11f9379
1 parent f055b6f
commit 11f9379
Show file tree

Hide file tree

Showing 3 changed files with 1,063 additions and 51 deletions.
diff --git a/bindings.h b/bindings.h
@@ -1,11 +1,13 @@
 #ifdef __LIBBPF_SYS_NOVENDOR
 #include <linux/if_link.h>
+#include <linux/perf_event.h>
 #include <bpf/bpf.h>
 #include <bpf/btf.h>
 #include <bpf/libbpf.h>
 #include <bpf/xsk.h>
 #else
 #include "libbpf/include/uapi/linux/if_link.h"
+#include "libbpf/include/uapi/linux/perf_event.h"
 #include "libbpf/src/bpf.h"
 #include "libbpf/src/btf.h"
 #include "libbpf/src/libbpf.h"

diff --git a/build.rs b/build.rs
@@ -53,16 +53,20 @@ fn generate_bindings(src_dir: path::PathBuf) {
         .allowlist_function("xsk_.+")
         .allowlist_function("_xsk_.+")
         .allowlist_function("perf_buffer_.+")
+        .allowlist_function("perf_event_.+")
         .allowlist_function("ring_buffer_.+")
         .allowlist_function("vdprintf")
-        .allowlist_type("bpf_.*")
-        .allowlist_type("btf_.*")
-        .allowlist_type("xdp_.*")
-        .allowlist_type("xsk_.*")
+        .allowlist_type("bpf_.+")
+        .allowlist_type("btf_.+")
+        .allowlist_type("xdp_.+")
+        .allowlist_type("xsk_.+")
+        .allowlist_type("perf_event_.+")
+        .allowlist_type("perf_sample_.+")
         .allowlist_var("BPF_.+")
         .allowlist_var("BTF_.+")
         .allowlist_var("XSK_.+")
         .allowlist_var("XDP_.+")
+        .allowlist_var("PERF_RECORD_.+")
         .parse_callbacks(Box::new(ignored_macros))
         .header("bindings.h")
         .clang_arg(format!("-I{}", src_dir.join("libbpf/include").display()))