Cast the cached UDPConn

Signed-off-by: Vasyl Gello <[email protected]>
yggdrasil-network · Jul 15, 2024 · 7f23c6f · 7f23c6f
1 parent 28636a5
commit 7f23c6f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cmd/yggstack/main.go b/cmd/yggstack/main.go
@@ -369,7 +369,7 @@ func main() {
 
 					var udpFwdConn *net.UDPConn = nil
 
-					udpFwdConn, ok := remoteUdpConnections.Load(remoteUdpAddr)
+					connVal, ok := remoteUdpConnections.Load(remoteUdpAddr)
 
 					if !ok {
 						udpFwdConn, err = net.DialUDP("udp", nil, mapping.Mapped)
@@ -379,6 +379,8 @@ func main() {
 						}
 						remoteUdpConnections.Store(remoteUdpAddr, &udpFwdConn)
 						go types.ReverseProxyUDP(mtu, udpListenConn, remoteUdpAddr, udpFwdConn)
+					} else {
+						udpFwdConn = connVal.(net.UDPConn)
 					}
 
 					_, err = udpFwdConn.Write(udpBuffer[:bytesRead])