From 5381cf96db71d92f537032990884cf5ce7a9f957 Mon Sep 17 00:00:00 2001 From: sbruens Date: Fri, 28 Jun 2024 15:03:23 -0400 Subject: [PATCH] Revert key string change. --- service/udp.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/service/udp.go b/service/udp.go index a6a46b50..dbf02e59 100644 --- a/service/udp.go +++ b/service/udp.go @@ -316,7 +316,7 @@ func (m *natmap) Get(key string) *natconn { return m.keyConn[key] } -func (m *natmap) set(clientAddr net.Addr, pc net.PacketConn, clientInfo ipinfo.IPInfo) *natconn { +func (m *natmap) set(key string, pc net.PacketConn, clientInfo ipinfo.IPInfo) *natconn { entry := &natconn{ PacketConn: pc, clientInfo: clientInfo, @@ -326,7 +326,7 @@ func (m *natmap) set(clientAddr net.Addr, pc net.PacketConn, clientInfo ipinfo.I m.Lock() defer m.Unlock() - m.keyConn[clientAddr.String()] = entry + m.keyConn[key] = entry return entry } @@ -343,7 +343,7 @@ func (m *natmap) del(key string) net.PacketConn { } func (m *natmap) Add(clientAddr net.Addr, clientConn net.PacketConn, cryptoKey *shadowsocks.EncryptionKey, targetConn net.PacketConn, clientInfo ipinfo.IPInfo, keyID string) *natconn { - entry := m.set(clientAddr, targetConn, clientInfo) + entry := m.set(clientAddr.String(), targetConn, clientInfo) m.metrics.AddUDPNatEntry(clientAddr, keyID) m.running.Add(1)