fix network consumer registration

DataDog · Dec 12, 2023 · f2fef52 · f2fef52
1 parent c7f41fd
commit f2fef52
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/cmd/system-probe/modules/eventmonitor.go b/cmd/system-probe/modules/eventmonitor.go
@@ -12,6 +12,7 @@ import (
 	"github.com/DataDog/datadog-agent/cmd/system-probe/config"
 	"github.com/DataDog/datadog-agent/pkg/eventmonitor"
 	emconfig "github.com/DataDog/datadog-agent/pkg/eventmonitor/config"
+	"github.com/DataDog/datadog-agent/pkg/network/events"
 	procconsumer "github.com/DataDog/datadog-agent/pkg/process/events/consumer"
 	secconfig "github.com/DataDog/datadog-agent/pkg/security/config"
 	secmodule "github.com/DataDog/datadog-agent/pkg/security/module"
@@ -56,6 +57,15 @@ var EventMonitor = module.Factory{
 			log.Info("event monitoring cws consumer initialized")
 		}
 
+		if emconfig.NetworkConsumerEnabled {
+			network, err := events.NewNetworkConsumer(evm)
+			if err != nil {
+				return nil, err
+			}
+			evm.RegisterEventConsumer(network)
+			log.Info("event monitoring network consumer initialized")
+		}
+
 		if emconfig.ProcessConsumerEnabled {
 			process, err := procconsumer.NewProcessConsumer(evm)
 			if err != nil {

diff --git a/...etwork/events/network_consumer_windows.go → ...network/events/network_consumer_others.go b/...etwork/events/network_consumer_windows.go → ...network/events/network_consumer_others.go
@@ -3,13 +3,14 @@
 // This product includes software developed at Datadog (https://www.datadoghq.com/).
 // Copyright 2016-present Datadog, Inc.
 
-//go:build windows
+//go:build !linux
 
 //nolint:revive // TODO(NET) Fix revive linter
 package events
 
 import (
 	"fmt"
+
 	"github.com/DataDog/datadog-agent/pkg/eventmonitor"
 )