Add --output-bpfmap

Signed-off-by: gray <[email protected]>

Author: jschwinger233

internal/pwru/output.go

@@ -458,8 +458,8 @@ func (o *output) Print(event *Event) {
 		fmt.Fprintf(o.writer, "%s", getShinfoData(event, o))
 	}
 
-	if event.PrintBpfmapId > 0 {
-		fmt.Fprintf(o.writer, "\n%s", getBpfMapData(event, o))
+	if o.flags.OutputBpfmap {
+		fmt.Fprintf(o.writer, "%s", getBpfMapData(event, o))
 	}
 
 	fmt.Fprintln(o.writer)

internal/pwru/types.go

@@ -47,6 +47,7 @@ type Flags struct {
 	OutputSkb        bool
 	OutputShinfo     bool
 	OutputStack      bool
+	OutputBpfmap     bool
 	OutputCaller     bool
 	OutputLimitLines uint64
 	OutputSkbCB      bool
@@ -84,6 +85,7 @@ func (f *Flags) SetFlags() {
 	flag.BoolVar(&f.OutputSkb, "output-skb", false, "print skb")
 	flag.BoolVar(&f.OutputShinfo, "output-skb-shared-info", false, "print skb shared info")
 	flag.BoolVar(&f.OutputStack, "output-stack", false, "print stack")
+	flag.BoolVar(&f.OutputBpfmap, "output-bpfmap", false, "print function arguments related to bpf maps")
 	flag.BoolVar(&f.OutputCaller, "output-caller", false, "print caller function name")
 	flag.Uint64Var(&f.OutputLimitLines, "output-limit-lines", 0, "exit the program after the number of events has been received/printed")
 	flag.BoolVar(&f.OutputSkbCB, "output-skb-cb", false, "print skb->cb")

main.go

@@ -98,7 +98,7 @@ func main() {
 		useKprobeMulti = true
 	}
 
-	funcs, bpfmapFuncs, err := pwru.GetFuncs(flags.FilterFunc, btfSpec, flags.KMods, useKprobeMulti, false)
+	funcs, bpfmapFuncs, err := pwru.GetFuncs(flags.FilterFunc, btfSpec, flags.KMods, useKprobeMulti, flags.OutputBpfmap)
 	if err != nil {
 		log.Fatalf("Failed to get skb-accepting functions: %s", err)
 	}