fix: don't close client sockets too early

[saartuer.git] / tysock.py

diff --git a/tysock.py b/tysock.py
index 676f57398b28b73f81e5882b79fefd831e84e6ca..2a5e193ad1ab71b967dc8b375647235eababb36b 100644 (file)
--- a/tysock.py
+++ b/tysock.py
@@ -27,6 +27,9 @@ class TySocket():
        CMDs = {
                b'buzz': StateMachine.CMD_BUZZ,
                b'unlock': StateMachine.CMD_UNLOCK,
+               b'lock': StateMachine.CMD_LOCK,
+               b'fallback_mode_on': StateMachine.CMD_FALLBACK_ON,
+               b'fallback_mode_off': StateMachine.CMD_FALLBACK_OFF,
        }
        
        def __init__(self, sm):
@@ -69,6 +72,7 @@ class TySocket():
                                if data in self.CMDs:
                                        self._sm.callback(self.CMDs[data], self._answer(conn))
                                        # _answer will be called, and it will close the connection
+                                       continue # make sure we break so we don't close it
                                else:
                                        waynesend(conn, 'Command not found')
                        except KeyboardInterrupt: