Protect WritePackets with mutex

Introducing mutex without implementing buffered channel may halt reads until replywithReset is sent. Buffering smooths the reads till the chan capacity is exceeded Signed-off-by: Vasyl Gello <[email protected]>
yggdrasil-network · Jul 17, 2024 · b3b5337 · b3b5337
1 parent b8879a4
commit b3b5337
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/netstack/yggdrasil.go b/src/netstack/yggdrasil.go
@@ -3,6 +3,7 @@ package netstack
 import (
 	"log"
 	"net"
+	"sync"
 
 	"github.com/yggdrasil-network/yggdrasil-go/src/core"
 	"github.com/yggdrasil-network/yggdrasil-go/src/ipv6rwc"
@@ -19,6 +20,7 @@ type YggdrasilNIC struct {
 	ipv6rwc      *ipv6rwc.ReadWriteCloser
 	dispatcher   stack.NetworkDispatcher
 	readBufChan  chan *stack.PacketBuffer
+	writeLock    sync.Mutex
 	writeBuf     []byte
 }
 
@@ -102,6 +104,8 @@ func (*YggdrasilNIC) Wait() {}
 func (e *YggdrasilNIC) WritePackets(
 	list stack.PacketBufferList,
 ) (int, tcpip.Error) {
+	e.writeLock.Lock()
+	defer e.writeLock.Unlock()
 	var i int = 0
 	for i, pkt := range list.AsSlice() {
 		vv := pkt.ToView()