blob: 674f22f4f9c662403fb2c98eb109d7f655057241 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
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))
}
|
