tetragon: Add keep-sensors-on-exit option

Signed-off-by: Jiri Olsa <[email protected]>
cilium · Jun 24, 2024 · 2b70b05 · 2b70b05
1 parent 23364bb
commit 2b70b05
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 2 deletions.
diff --git a/cmd/tetragon/main.go b/cmd/tetragon/main.go
@@ -221,6 +221,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)
  }
@@ -457,7 +461,9 @@ func tetragonExecute() error {
  return err
  }
  defer func() {
- initialSensor.Unload()
+ if !option.Config.KeepSensorsOnExit {
+ initialSensor.Unload()
+ }
  }()
 
  cgrouprate.NewCgroupRate(ctx, pm, base.CgroupRateMap, &option.Config.CgroupRate)
@@ -468,7 +474,9 @@ func tetragonExecute() error {
  sensorMgWait = nil
  observer.GetSensorManager().LogSensorsAndProbes(ctx)
  defer func() {
- observer.RemoveSensors(ctx)
+ if !option.Config.KeepSensorsOnExit {
+ observer.RemoveSensors(ctx)
+ }
  }()
 
  err = loadTpFromDir(ctx, option.Config.TracingPolicyDir)

diff --git a/docs/data/tetragon_flags.yaml b/docs/data/tetragon_flags.yaml
diff --git a/pkg/option/config.go b/pkg/option/config.go
@@ -95,6 +95,8 @@ type config struct {
 
  HealthServerAddress string
  HealthServerInterval int
+
+ KeepSensorsOnExit bool
 }
 
 var (

diff --git a/pkg/option/flags.go b/pkg/option/flags.go
@@ -105,6 +105,8 @@ const (
  KeyHealthTimeInterval = "health-server-interval"
 
  KeyBpfDir = "bpf-dir"
+
+ KeyKeepSensorsOnExit = "keep-sensors-on-exit"
 )
 
 type UsernameMetadaCode int
@@ -220,6 +222,8 @@ func ReadAndSetFlags() error {
  Config.HealthServerInterval = viper.GetInt(KeyHealthTimeInterval)
 
  Config.BpfDir = viper.GetString(KeyBpfDir)
+
+ Config.KeepSensorsOnExit = viper.GetBool(KeyKeepSensorsOnExit)
  return nil
 }
 
@@ -376,4 +380,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")
 }