do not rause KeyError for unknown clients

miguelgrinberg · May 27, 2016 · 82ffcfb · 82ffcfb
1 parent 7fbc56d
commit 82ffcfb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/flask_socketio/__init__.py b/flask_socketio/__init__.py
@@ -174,6 +174,9 @@ def handle_my_custom_event(json):
 
         def decorator(handler):
             def _handler(sid, *args):
+                if sid not in self.server.environ:
+                    # we don't have record of this client, ignore this event
+                    return '', 400
                 app = self.server.environ[sid]['flask.app']
                 with app.request_context(self.server.environ[sid]):
                     if 'saved_session' in self.server.environ[sid]: