Do not duplicate SSLSocket instances (Fixes #26)

miguelgrinberg · Aug 13, 2023 · da42e98 · da42e98
1 parent abdfcb0
commit da42e98
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/simple_websocket/ws.py b/src/simple_websocket/ws.py
@@ -313,7 +313,11 @@ def __init__(self, environ, subprotocols=None, receive_bytes=4096,
             if not hasattr(wsgi_input, 'raw') and hasattr(wsgi_input, 'rfile'):
                 wsgi_input = wsgi_input.rfile
             if hasattr(wsgi_input, 'raw'):
-                sock = wsgi_input.raw._sock.dup()
+                sock = wsgi_input.raw._sock
+                try:
+                    sock = sock.dup()
+                except NotImplementedError:
+                    pass
                 self.mode = 'gevent'
         if sock is None:
             raise RuntimeError('Cannot obtain socket from WSGI environment.')