From f41b29c1741ddb81f9af304e7364dd7b257296c0 Mon Sep 17 00:00:00 2001
From: Junyeong Jeong <rhdxmr@gmail.com>
Date: Fri, 18 Feb 2022 01:25:41 +0900
Subject: [PATCH] Generate binding for struct unix_sock for probes

Signed-off-by: Junyeong Jeong <rhdxmr@gmail.com>
---
 redbpf-probes/build.rs                 | 2 ++
 redbpf-probes/include/redbpf_helpers.h | 1 +
 2 files changed, 3 insertions(+)

diff --git a/redbpf-probes/build.rs b/redbpf-probes/build.rs
index bf73ef4a..6504a575 100644
--- a/redbpf-probes/build.rs
+++ b/redbpf-probes/build.rs
@@ -83,6 +83,7 @@ fn generate_bindings_kernel_headers() -> Result<()> {
         "__sk_.*",
         "sk_.*",
         "inet_sock",
+        "unix_sock",
         "sockaddr",
         "sockaddr_in",
         "in_addr",
@@ -181,6 +182,7 @@ fn generate_bindings_vmlinux() -> Result<()> {
         "^__sk_.*",
         "^sk_.*",
         "^inet_sock$",
+        "^unix_sock$",
         "^sockaddr$",
         "^sockaddr_in$",
         "^in_addr$",
diff --git a/redbpf-probes/include/redbpf_helpers.h b/redbpf-probes/include/redbpf_helpers.h
index c0fd9509..c7e3c13c 100644
--- a/redbpf-probes/include/redbpf_helpers.h
+++ b/redbpf-probes/include/redbpf_helpers.h
@@ -19,6 +19,7 @@
 #include <uapi/linux/bpf.h>
 #include <net/sock.h>
 #include <net/inet_sock.h>
+#include <net/af_unix.h>
 #include <linux/ipv6.h>
 #include "xdp.h"
 #include "bpf_iter.h"