tetragon-bench: Add --rb-size/--rb-queue-size options

Adding --rb-size/--rb-queue-size options to configure the
perf ring buffer size and ring buffer queue size.

Signed-off-by: Jiri Olsa <jolsa@kernel.org>

cmd/tetragon-bench/bench.go

@@ -24,9 +24,14 @@ var (
 
 	traceBench *string
 	crd        *string
+
+	rbSize      *int
+	rbQueueSize *int
 )
 
 func init() {
+	rbSize = flag.Int("rb-size", 65535, "perf ring buffer size")
+	rbQueueSize = flag.Int("rb-queue-size", 65535, "ring buffer queue size")
 	debug = flag.Bool("debug", false, "enable debugging")
 	jsonEncode = flag.Bool("json-encode", false, "JSON encode the events and measure overhead")
 	baseline = flag.Bool("baseline", false, "run a baseline benchmark without tetragon")
@@ -89,6 +94,8 @@ func main() {
 		Trace:       bench.TraceBenchNameOrPanic(*traceBench),
 		Crd:         *crd,
 		CmdArgs:     cmdArgs,
+		RBSize:      *rbSize,
+		RBQueueSize: *rbQueueSize,
 	}
 
 	summary := bench.RunTraceBench(args)

pkg/bench/bench.go

@@ -49,6 +49,8 @@ type Arguments struct {
 	PrintEvents bool
 	JSONEncode  bool
 	Baseline    bool
+	RBSize      int
+	RBQueueSize int
 }
 
 func readConfig(file string) (*config.GenericTracingConf, error) {
@@ -98,6 +100,9 @@ func runTetragon(ctx context.Context, configFile string, args *Arguments, summar
 		}
 	}
 
+	option.Config.RBSize = args.RBSize
+	option.Config.RBQueueSize = args.RBQueueSize
+
 	option.Config.BpfDir = bpf.MapPrefixPath()
 	option.Config.MapDir = bpf.MapPrefixPath()
 	obs := observer.NewObserver(configFile)