From 98f68e89bff0ea0ad0d27d55cd7e4fa10bf365c1 Mon Sep 17 00:00:00 2001
From: ksc <ksc010@qq.com>
Date: Wed, 17 Jan 2018 14:26:27 +0800
Subject: [PATCH] the size of handshake HTTP request header may bigger than
 1024

Like appends some Cookies
---
 websocket_server/websocket_server.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/websocket_server/websocket_server.py b/websocket_server/websocket_server.py
index fc50b8e..9891022 100644
--- a/websocket_server/websocket_server.py
+++ b/websocket_server/websocket_server.py
@@ -293,7 +293,9 @@ def send_text(self, message, opcode=OPCODE_TEXT):
         self.request.send(header + payload)
 
     def handshake(self):
-        message = self.request.recv(1024).decode().strip()
+        message = self.request.recv(1024).decode()
+        if not message.endswith("\r\n\r\n"):
+            message +=self.request.recv(1024).decode()
         upgrade = re.search('\nupgrade[\s]*:[\s]*websocket', message.lower())
         if not upgrade:
             self.keep_alive = False