Merge pull request #59 from slayer321/limit-flag

feat: add limit flag
kubearmor · May 4, 2022 · b61b333 · b61b333
2 parents 6da4191 + b52b3a9
commit b61b333
Show file tree

Hide file tree

Showing 3 changed files with 223 additions and 166 deletions.
diff --git a/cmd/log.go b/cmd/log.go
@@ -39,4 +39,5 @@ func init() {
  logCmd.Flags().StringVar(&logOptions.PodName, "pod", "", "name of the pod ")
  logCmd.Flags().StringVar(&logOptions.Resource, "resource", "", "command used by the user")
  logCmd.Flags().StringVar(&logOptions.Source, "source", "", "binary used by the system ")
+ logCmd.Flags().Uint32Var(&logOptions.Limit, "limit", 0, "number of logs you want to see")
 }
diff --git a/log/log.go b/log/log.go
@@ -40,6 +40,7 @@ type Options struct {
  PodName string
  Source string
  Resource string
+ Limit uint32
 }
 
 // StopChan Channel
@@ -120,7 +121,7 @@ func StartObserver(o Options) error {
  }
 
  // create a client
- logClient := NewClient(gRPC, o.MsgPath, o.LogPath, o.LogFilter)
+ logClient := NewClient(gRPC, o.MsgPath, o.LogPath, o.LogFilter, o.Limit)
  if logClient == nil {
  return errors.New("failed to connect to the gRPC server\nPossible troubleshooting:\n- Check if Kubearmor is running\n- Create a portforward to KubeArmor relay service using\n\t\033[1mkubectl -n kube-system port-forward service/kubearmor --address 0.0.0.0 --address :: 32767:32767\033[0m\n- Configure grpc server information using\n\t\033[1mkarmor log --grpc <info>\033[0m")
  }
@@ -143,6 +144,7 @@ func StartObserver(o Options) error {
  return err
  }
 
+ Limitchan = make(chan bool, 2)
  if o.LogPath != "none" {
  if o.LogFilter == "all" || o.LogFilter == "policy" {
  // watch alerts
@@ -157,9 +159,18 @@ func StartObserver(o Options) error {
  }
  }
 
- // listen for interrupt signals
- sigChan := GetOSSigChannel()
- <-sigChan
+ if o.Limit != 0 {
+ if o.LogFilter == "all" {
+ <-Limitchan
+ <-Limitchan
+ } else {
+ <-Limitchan
+ }
+ } else {
+ // listen for interrupt signals
+ sigChan := GetOSSigChannel()
+ <-sigChan
+ }
  close(StopChan)
 
  logClient.Running = false