tetragon: Use program.SetTailCall for lsm probes

Using program.SetTailCall for lsm probes and getting rid of the map name check for tail call setup. Signed-off-by: Jiri Olsa <[email protected]>
cilium · Aug 1, 2024 · 253cc56 · 253cc56
1 parent 9d463fc
commit 253cc56
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 14 deletions.
diff --git a/pkg/sensors/program/loader.go b/pkg/sensors/program/loader.go
@@ -26,11 +26,6 @@ type AttachFunc func(*ebpf.Collection, *ebpf.CollectionSpec, *ebpf.Program, *ebp
 
 type OpenFunc func(*ebpf.CollectionSpec) error
 
-type tailCall struct {
-	m      *Map
-	prefix string
-}
-
 type LoadOpts struct {
 	Attach AttachFunc
 	Open   OpenFunc
@@ -660,18 +655,11 @@ func LoadTracingProgram(bpfDir string, load *Program, verbose int) error {
 }
 
 func LoadLSMProgram(bpfDir string, load *Program, verbose int) error {
-	var tc tailCall
-	for mName, m := range load.PinMap {
-		if mName == "lsm_calls" {
-			tc = tailCall{m, "lsm"}
-			break
-		}
-	}
 	opts := &LoadOpts{
 		Attach:   LSMAttach(),
 		Open:     LSMOpen(load),
-		TcMap:    tc.m,
-		TcPrefix: tc.prefix,
+		TcMap:    load.TcMap,
+		TcPrefix: load.TcPrefix,
 	}
 	return loadProgram(bpfDir, load, opts, verbose)
 }

diff --git a/pkg/sensors/tracing/genericlsm.go b/pkg/sensors/tracing/genericlsm.go
@@ -372,6 +372,8 @@ func createLsmSensorFromEntry(lsmEntry *genericLsm,
 	tailCalls := program.MapBuilderPin("lsm_calls", sensors.PathJoin(pinPath, "lsm_calls"), load)
 	maps = append(maps, tailCalls)
 
+	load.SetTailCall("lsm", tailCalls)
+
 	filterMap := program.MapBuilderPin("filter_map", sensors.PathJoin(pinPath, "filter_map"), load)
 	maps = append(maps, filterMap)