From fe5bebbb09cd765f636ecf5088169930fb965fbb Mon Sep 17 00:00:00 2001
From: Jakub Stasiak <jakub@stasiak.at>
Date: Sat, 18 Jan 2014 21:54:42 +0000
Subject: [PATCH] websockets: handle HTTP_CONNECTION more flexibly

This fixes GitHub issue #73
---
 eventlet/websocket.py       |  5 ++++-
 tests/websocket_new_test.py | 29 ++++++++++++++++-------------
 2 files changed, 20 insertions(+), 14 deletions(-)

diff --git a/eventlet/websocket.py b/eventlet/websocket.py
index d99206e..20342db 100644
--- a/eventlet/websocket.py
+++ b/eventlet/websocket.py
@@ -97,7 +97,10 @@ class WebSocketWSGI(object):
         return decorator(handler)
 
     def __call__(self, environ, start_response):
-        if not (environ.get('HTTP_CONNECTION') == 'Upgrade' and
+        http_connection_parts = [
+            part.strip()
+            for part in environ.get('HTTP_CONNECTION', '').lower().split(',')]
+        if not ('upgrade' in http_connection_parts and
                 environ.get('HTTP_UPGRADE').lower() == 'websocket'):
             # need to check a few more things here for true compliance
             start_response('400 Bad Request', [('Connection', 'close')])
diff --git a/tests/websocket_new_test.py b/tests/websocket_new_test.py
index 8009197..8a84895 100644
--- a/tests/websocket_new_test.py
+++ b/tests/websocket_new_test.py
@@ -68,25 +68,28 @@ class TestWebSocket(_TestBase):
         self.assertEqual(resp.read(), '')
 
     def test_correct_upgrade_request_13(self):
-        connect = [
+        for http_connection in ['Upgrade', 'UpGrAdE', 'keep-alive, Upgrade']:
+            connect = [
                 "GET /echo HTTP/1.1",
                 "Upgrade: websocket",
-                "Connection: Upgrade",
+                "Connection: %s" % http_connection,
                 "Host: localhost:%s" % self.port,
                 "Origin: http://localhost:%s" % self.port,
                 "Sec-WebSocket-Version: 13",
-                "Sec-WebSocket-Key: d9MXuOzlVQ0h+qRllvSCIg==", ]
-        sock = eventlet.connect(
-            ('localhost', self.port))
+                "Sec-WebSocket-Key: d9MXuOzlVQ0h+qRllvSCIg==",
+            ]
+            sock = eventlet.connect(('localhost', self.port))
 
-        sock.sendall('\r\n'.join(connect) + '\r\n\r\n')
-        result = sock.recv(1024)
-        ## The server responds the correct Websocket handshake
-        self.assertEqual(result,
-                         '\r\n'.join(['HTTP/1.1 101 Switching Protocols',
-                                      'Upgrade: websocket',
-                                      'Connection: Upgrade',
-                                      'Sec-WebSocket-Accept: ywSyWXCPNsDxLrQdQrn5RFNRfBU=\r\n\r\n', ]))
+            sock.sendall('\r\n'.join(connect) + '\r\n\r\n')
+            result = sock.recv(1024)
+            ## The server responds the correct Websocket handshake
+            print('Connection string: %r' % http_connection)
+            self.assertEqual(result, '\r\n'.join([
+                'HTTP/1.1 101 Switching Protocols',
+                'Upgrade: websocket',
+                'Connection: Upgrade',
+                'Sec-WebSocket-Accept: ywSyWXCPNsDxLrQdQrn5RFNRfBU=\r\n\r\n',
+            ]))
 
     def test_send_recv_13(self):
         connect = [