Checking cookie error in websocket handshake.

rawpython · Nov 27, 2018 · 5b97d51 · 5b97d51
1 parent 015f8be
commit 5b97d51
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/remi/server.py b/remi/server.py
@@ -195,7 +195,10 @@ def send_message(self, message):
     def handshake(self):
         self._log.debug('handshake')
         key = self.headers['Sec-WebSocket-Key']
-        self.session = parse_session_cookie(self.headers['cookie'])
+        self.session = None
+        if 'cookie' in self.headers:
+            if self.headers['cookie']!=None:
+                self.session = parse_session_cookie(self.headers['cookie'])
         if self.session == None:
             return False
         if not self.session in clients.keys():