move flags to types

2 files changed, 9 insertions, 8 deletions

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))
 }
 
diff --git a/internal/types/types.go b/internal/types/types.go
index 473339f..2339727 100644
--- a/internal/types/types.go
+++ b/internal/types/types.go
@@ -29,6 +29,8 @@ type OpenatEnterEvent struct {
 	Comm     [MAX_PROGNAME_LENGTH]byte
 }
 
-// TODO: Move Flags type struct to here, too
+type FlagValues struct {
+	UidFilter uint32
+}
 
 // duration := float64(e.ExitTime-e.EnterTime) / float64(1_000_000)