tetragon: Add keep-sensors-on-exit option

Signed-off-by: Jiri Olsa <[email protected]>

cmd/tetragon/main.go

@@ -220,6 +220,10 @@ func tetragonExecute() error {
 		log.Info("Force loading smallprograms")
 	}
 
+	if option.Config.KeepSensorsOnExit {
+		log.Info("Not unloading sensors on exit")
+	}
+
 	if viper.IsSet(option.KeyNetnsDir) {
 		defaults.NetnsDir = viper.GetString(option.KeyNetnsDir)
 	}

pkg/option/config.go

@@ -94,6 +94,8 @@ type config struct {
 
 	HealthServerAddress  string
 	HealthServerInterval int
+
+	KeepSensorsOnExit bool
 }
 
 var (

pkg/option/flags.go

@@ -105,6 +105,8 @@ const (
 	KeyHealthTimeInterval  = "health-server-interval"
 
 	KeyBpfDir = "bpf-dir"
+
+	KeyKeepSensorsOnExit = "keep-sensors-on-exit"
 )
 
 type UsernameMetadaCode int
@@ -222,6 +224,8 @@ func ReadAndSetFlags() error {
 	Config.HealthServerInterval = viper.GetInt(KeyHealthTimeInterval)
 
 	Config.BpfDir = viper.GetString(KeyBpfDir)
+
+	Config.KeepSensorsOnExit = viper.GetBool(KeyKeepSensorsOnExit)
 	return nil
 }
 
@@ -378,4 +382,6 @@ func AddFlags(flags *pflag.FlagSet) {
 	flags.Int(KeyHealthTimeInterval, 10, "Health server interval in seconds")
 
 	flags.String(KeyBpfDir, defaults.DefaultMapPrefix, "Set tetragon bpf directory (default 'tetragon')")
+
+	flags.Bool(KeyKeepSensorsOnExit, false, "Do not unload sensors on exit")
 }

pkg/sensors/program/loader.go

@@ -60,7 +60,7 @@ func linkPinPath(bpfDir string, load *Program, extra ...string) string {
 
 func linkPin(lnk link.Link, bpfDir string, load *Program, extra ...string) error {
 	// pinned link is not configured
-	if !load.PinLink {
+	if !option.Config.KeepSensorsOnExit && !load.PinLink {
 		return nil
 	}
 	// pinned link is not supported