PID filter

1 files changed, 10 insertions, 3 deletions

diff --git a/internal/flags/flags.go b/internal/flags/flags.go
index f139654..fbb0569 100644
--- a/internal/flags/flags.go
+++ b/internal/flags/flags.go
@@ -9,11 +9,13 @@ import (
 
 type Flags struct {
 	UidFilter    int
+	PidFilter    int
 	EventMapSize int
 }
 
 func New() (flags Flags) {
-	flag.IntVar(&flags.UidFilter, "uid", 0, "Filter for processes with UID")
+	// flag.IntVar(&flags.UidFilter, "uid", 0, "Filter for user ID")
+	flag.IntVar(&flags.PidFilter, "pid", 0, "Filter for processes ID")
 	flag.IntVar(&flags.EventMapSize, "mapSize", 4096*16, "BPF FD event ring buffer map size")
 	flag.Parse()
 
@@ -21,8 +23,13 @@ func New() (flags Flags) {
 }
 
 func (flags Flags) SetBPF(bpfModule *bpf.Module) error {
-	if err := bpfModule.InitGlobalVariable("UID_FILTER", uint32(flags.UidFilter)); err != nil {
-		return fmt.Errorf("unable to set up UID_FILTER global variable: %w", err)
+	/*
+		if err := bpfModule.InitGlobalVariable("UID_FILTER", uint32(flags.UidFilter)); err != nil {
+			return fmt.Errorf("unable to set up UID_FILTER global variable: %w", err)
+		}
+	*/
+	if err := bpfModule.InitGlobalVariable("PID_FILTER", uint32(flags.PidFilter)); err != nil {
+		return fmt.Errorf("unable to set up PID_FILTER global variable: %w", err)
 	}
 	return nil
 }