Merge pull request #303 from YouShengLiu/bugfix/race-condition

fix: add lock at WriteTo
wmnsk · Oct 23, 2024 · f103b7b · f103b7b
2 parents 5414bc8 + cceb5ea
commit f103b7b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/gtpv1/u-conn.go b/gtpv1/u-conn.go
@@ -476,6 +476,8 @@ func (u *UPlaneConn) ReadFromGTP(p []byte) (n int, addr net.Addr, teid uint32, e
 // see SetDeadline and SetWriteDeadline.
 // On packet-oriented connections, write timeouts are rare.
 func (u *UPlaneConn) WriteTo(p []byte, addr net.Addr) (n int, err error) {
+	u.mu.Lock()
+	defer u.mu.Unlock()
 	return u.pktConn.WriteToWithDSCPECN(p, addr, 0)
 }