fixed udp memory leak

txthinking · Aug 31, 2020 · d947c5b · d947c5b
1 parent 884ad52
commit d947c5b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/packetserver.go b/packetserver.go
@@ -56,6 +56,9 @@ func (s *PacketServer) RemoteToClient(remote net.Conn, timeout int, dst []byte,
 			}
 		}
 		l, err := remote.Read(s.wb[12+len(dst) : 65507-16])
+		if err != nil {
+			return nil
+		}
 		if _, err := io.ReadFull(rand.Reader, s.wb[:12]); err != nil {
 			return err
 		}