cleanup when unexpected packet received

pin · Oct 22, 2024 · 4234ee6 · 4234ee6
1 parent e69c06d
commit 4234ee6
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/single_port.go b/single_port.go
@@ -35,10 +35,13 @@ func (s *Server) singlePortProcessRequests() error {
 				s.Unlock()
 				go func() {
 					err := s.handlePacket(localAddr, srcAddr, buf, cnt, maxSz, lc)
-					if err != nil && s.hook != nil {
-						s.hook.OnFailure(TransferStats{
-							SenderAnticipateEnabled: s.sendAEnable,
-						}, err)
+					if err != nil {
+						if s.hook != nil {
+							s.hook.OnFailure(TransferStats{
+								SenderAnticipateEnabled: s.sendAEnable,
+							}, err)
+						}
+						delete(s.handlers, srcAddr.String())
 					}
 				}()
 			}