adding chan out

aquasecurity · Jun 28, 2021 · 2579020 · 2579020
1 parent 741e7bb
commit 2579020
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/tracee-ebpf/tracee/pipeline.go b/tracee-ebpf/tracee/pipeline.go
@@ -200,8 +200,12 @@ func (t *Tracee) printEvent(done <-chan struct{}, in <-chan external.Event) (<-c
 	go func() {
 		defer close(errc)
 		for printEvent := range in {
-			t.stats.eventCounter.Increment()
-			t.printer.Print(printEvent)
+			if t.config.ChanEvents != nil {
+				t.config.ChanEvents <- printEvent
+			} else {
+				t.stats.eventCounter.Increment()
+				t.printer.Print(printEvent)
+			}
 		}
 	}()
 	return errc, nil

diff --git a/tracee-ebpf/tracee/tracee.go b/tracee-ebpf/tracee/tracee.go
@@ -33,6 +33,7 @@ type Config struct {
 	SecurityAlerts     bool
 	maxPidsCache       int // maximum number of pids to cache per mnt ns (in Tracee.pidsInMntns)
 	BPFObjPath         string
+	ChanEvents         chan external.Event
 }
 
 type Filter struct {