From aca5c8dd24fa5b51aa90456c647bbe43eaa51ed0 Mon Sep 17 00:00:00 2001
From: RouxAntoine <antoinroux@hotmail.fr>
Date: Tue, 17 May 2022 19:34:39 +0200
Subject: [PATCH] fix: apply comment from PR review

---
 pkg/encapsulation/cilium.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/pkg/encapsulation/cilium.go b/pkg/encapsulation/cilium.go
index 84065d6..9e34292 100644
--- a/pkg/encapsulation/cilium.go
+++ b/pkg/encapsulation/cilium.go
@@ -19,8 +19,9 @@ import (
 	"net"
 	"sync"
 
-	"github.com/squat/kilo/pkg/iptables"
 	"github.com/vishvananda/netlink"
+
+	"github.com/squat/kilo/pkg/iptables"
 )
 
 const ciliumDeviceName = "cilium_host"
@@ -43,7 +44,7 @@ func NewCilium(strategy Strategy) Encapsulator {
 	}
 }
 
-// CleanUp is a no-op.
+// CleanUp close done channel
 func (f *cilium) CleanUp() error {
 	close(f.done)
 	return nil
@@ -56,6 +57,8 @@ func (f *cilium) Gw(_, _ net.IP, subnet *net.IPNet) net.IP {
 
 // Index returns the index of the Cilium interface.
 func (f *cilium) Index() int {
+	f.mu.Lock()
+	defer f.mu.Unlock()
 	return f.iface
 }