From 02707dde82ed3030aa66d8155928f364cefe143d Mon Sep 17 00:00:00 2001 From: Paul Buetow Date: Fri, 16 Feb 2024 00:47:08 +0200 Subject: move flags to types --- internal/flags/flags.go | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) (limited to 'internal/flags') diff --git a/internal/flags/flags.go b/internal/flags/flags.go index ac7abd4..d50410b 100644 --- a/internal/flags/flags.go +++ b/internal/flags/flags.go @@ -3,6 +3,7 @@ package flags import ( "flag" "fmt" + "ioriotng/internal/types" "unsafe" bpf "github.com/aquasecurity/libbpfgo" @@ -17,6 +18,7 @@ func New() (flags Flags) { flag.IntVar(&flags.UidFilter, "uid", 0, "Filter for processes with UID") flag.IntVar(&flags.EventMapSize, "mapSize", 4096, "BPF FD event ring buffer map size") flag.Parse() + return flags } @@ -26,13 +28,10 @@ func (flags Flags) SetBPF(bpfModule *bpf.Module) error { return err } - flagsValues := struct { - UidFilter int32 - }{ - UidFilter: int32(flags.UidFilter), - } - - key := uint32(1) + var ( + key = uint32(1) + flagsValues = types.FlagValues{uint32(flags.UidFilter)} + ) return flagsMap.Update(unsafe.Pointer(&key), unsafe.Pointer(&flagsValues)) } -- cgit v1.2.3