internal: embed BPF object into ior binary

1 files changed, 31 insertions, 0 deletions

diff --git a/internal/bpfembed.go b/internal/bpfembed.go
new file mode 100644
index 0000000..fce784c
--- /dev/null
+++ b/internal/bpfembed.go
@@ -0,0 +1,31 @@
+package internal
+
+import (
+	_ "embed"
+	"os"
+
+	bpf "github.com/aquasecurity/libbpfgo"
+)
+
+const (
+	bpfObjectOverrideEnv  = "IOR_BPF_OBJECT"
+	embeddedBPFObjectName = "ior.bpf.o"
+)
+
+//go:embed c/ior.bpf.o
+var embeddedBPFObject []byte
+
+var (
+	newBPFModuleFromFile   = bpf.NewModuleFromFile
+	newBPFModuleFromBuffer = bpf.NewModuleFromBuffer
+)
+
+func loadBPFModule() (*bpf.Module, string, error) {
+	if path := os.Getenv(bpfObjectOverrideEnv); path != "" {
+		module, err := newBPFModuleFromFile(path)
+		return module, "load module from override file", err
+	}
+
+	module, err := newBPFModuleFromBuffer(embeddedBPFObject, embeddedBPFObjectName)
+	return module, "load embedded module", err
+}