From 161e55dc76ee0d3ac52fea8473de5cffebd2bfd9 Mon Sep 17 00:00:00 2001 From: Paul Buetow Date: Tue, 13 Feb 2024 19:20:18 +0200 Subject: forgot to add the flags package --- internal/flags/flags.go | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 internal/flags/flags.go (limited to 'internal/flags/flags.go') 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)) +} -- cgit v1.2.3