fix: wait for first heartbeat before starting networking loops

cmd/bee/main.go

@@ -6,6 +6,7 @@ import (
 	"net/netip"
 	"os"
 	"os/signal"
+	"sync"
 	"syscall"
 	"time"
 
@@ -65,7 +66,8 @@ func run(bindAddress netip.Addr, disableNftables bool, beekeeperBasePath string)
 	}
 	defer forwarder.Close()
 
-	heartbeat := heartbeat.NewHeartbeat(bee, forwarder, bindAddress)
+	var configReceived sync.WaitGroup
+	heartbeat := heartbeat.NewHeartbeat(bee, forwarder, bindAddress, &configReceived)
 
 	signalChannel := make(chan os.Signal, 1)
 	signal.Notify(signalChannel, os.Interrupt, syscall.SIGTERM)
@@ -74,6 +76,7 @@ func run(bindAddress netip.Addr, disableNftables bool, beekeeperBasePath string)
 
 	go func() {
 		defer recoverPanic(signalChannel)
+		configReceived.Wait()
 		for {
 			if err := forwarder.AttackerToBeehiveLoop(ctx); err != nil {
 				log.WithError(err).Error("Attacker to Beehive loop failed. Restarting.")
@@ -88,6 +91,7 @@ func run(bindAddress netip.Addr, disableNftables bool, beekeeperBasePath string)
 
 	go func() {
 		defer recoverPanic(signalChannel)
+		configReceived.Wait()
 		for {
 			if err := forwarder.BeehiveToAttackerLoop(ctx); err != nil {
 				log.WithError(err).Error("Beehive to Attacker loop failed. Restarting.")

internal/heartbeat/heartbeat.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"net"
 	"net/netip"
+	"sync"
 	"time"
 
 	"github.com/Cybersecurity-and-Enterprise-Security/bee/internal/apibee"
@@ -18,23 +19,31 @@ const (
 )
 
 type Heartbeat struct {
-	bee         *apibee.Bee
-	forwarder   *forward.Forwarder
-	bindAddress netip.Addr
+	bee             *apibee.Bee
+	forwarder       *forward.Forwarder
+	bindAddress     netip.Addr
+	startupComplete bool
+	configReceived  *sync.WaitGroup
 }
 
-func NewHeartbeat(bee *apibee.Bee, forwarder *forward.Forwarder, bindAddress netip.Addr) *Heartbeat {
+func NewHeartbeat(bee *apibee.Bee, forwarder *forward.Forwarder, bindAddress netip.Addr, configReceived *sync.WaitGroup) *Heartbeat {
 	return &Heartbeat{
 		bee,
 		forwarder,
 		bindAddress,
+		false,
+		configReceived,
 	}
 }
 
 func (h *Heartbeat) Run(ctx context.Context) error {
 	for {
 		h.ReportStats(ctx)
 		h.UpdateForwardings(ctx)
+		if !h.startupComplete {
+			h.startupComplete = true
+			h.configReceived.Done()
+		}
 
 		select {
 		case <-ctx.Done():