tetragon: Add sensor PostLoadHook hook

Adding sensor PostLoadHook hook, it will be used in following changes in kprobe sensor to kill the stats timer. Signed-off-by: Jiri Olsa <[email protected]>
cilium · Apr 9, 2024 · b9b9488 · b9b9488
1 parent d4ce92a
commit b9b9488
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/pkg/sensors/load.go b/pkg/sensors/load.go
@@ -114,6 +114,11 @@ func (s *Sensor) Load(bpfDir string) error {
 		p.LoadState.RefInc()
 		l.WithField("prog", p.Name).WithField("label", p.Label).Debugf("BPF prog was loaded")
 	}
+	if s.PostLoadHook != nil {
+		if err := s.PostLoadHook(); err != nil {
+			logger.GetLogger().WithError(err).WithField("sensor", s.Name).Warn("Post load hook failed")
+		}
+	}
 	l.WithField("sensor", s.Name).Infof("Loaded BPF maps and events for sensor successfully")
 	s.Loaded = true
 	return nil

diff --git a/pkg/sensors/sensors.go b/pkg/sensors/sensors.go
@@ -44,6 +44,8 @@ type Sensor struct {
 	Loaded bool
 	// Destroyed indicates whether the sensor had been destroyed.
 	Destroyed bool
+	// PostLoadHook hook
+	PostLoadHook SensorHook
 	// PreUnloadHook can optionally contain a pointer to a function to be
 	// called during sensor unloading, prior to the programs and maps being
 	// unloaded.