From c1b4bee6c93265139974b1e5cf065d3f82a7154b Mon Sep 17 00:00:00 2001 From: Paul Buetow Date: Thu, 22 Feb 2024 22:44:44 +0200 Subject: PID filter --- internal/flags/flags.go | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) (limited to 'internal/flags') 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 } -- cgit v1.2.3