forgot to add the flags package

1 files changed, 34 insertions, 0 deletions

diff --git a/internal/flags/flags.go b/internal/flags/flags.go
new file mode 100644
index 0000000..674f22f
--- /dev/null
+++ b/internal/flags/flags.go
@@ -0,0 +1,34 @@
+package flags
+
+import (
+	"flag"
+	"unsafe"
+
+	bpf "github.com/aquasecurity/libbpfgo"
+)
+
+type Flags struct {
+	UidFilter int
+}
+
+func New() (flags Flags) {
+	flag.IntVar(&flags.UidFilter, "uid", 0, "Filter for processes with UID")
+	flag.Parse()
+	return flags
+}
+
+func (flags Flags) SetBPF(bpfModule *bpf.Module) error {
+	flagsMap, err := bpfModule.GetMap("flags_map")
+	if err != nil {
+		return err
+	}
+
+	flagsValues := struct {
+		UidFilter int32
+	}{
+		UidFilter: int32(flags.UidFilter),
+	}
+
+	key := uint32(1)
+	return flagsMap.Update(unsafe.Pointer(&key), unsafe.Pointer(&flagsValues))
+}