Don't close UDP socket on bad packet magic

2017-10-19 09:00:02 +02:00 · 2017-10-19 09:00:02 +02:00 · f785558ca5
parent a090eea9b0
commit f785558ca5
1 changed files with 3 additions and 1 deletions
--- a/src/network/bmproto.py
+++ b/src/network/bmproto.py
@ -74,7 +74,9 @@ class BMProto(AdvancedDispatcher, ObjectTracker):
            self.set_state("bm_header", length=1)
            self.bm_proto_reset()
            logger.debug("Bad magic")
-            self.handle_close("Bad magic")
+            if self.socket.type == socket.SOCK_STREAM:
+                self.close_reason = "Bad magic"
+                self.set_state("close")
            return False
        if self.payloadLength > BMProto.maxMessageSize:
            self.invalid = True