Use SSLContext.wrap_context() for SSL/TLS sockets

ssl.wrap_context() has been deprecated for some time, and has now been
removed. Use the newer API instead.

Author: CendioOssman

websockify/websocket.py

@@ -139,7 +139,8 @@ def connect(self, uri, origin=None, protocols=[]):
             self.socket = socket.create_connection((uri.hostname, port))
 
             if uri.scheme in ("wss", "https"):
-                self.socket = ssl.wrap_socket(self.socket)
+                context = ssl.create_default_context()
+                self.socket = context.wrap_socket(self.socket)
                 self._state = "ssl_handshake"
             else:
                 self._state = "headers"

websockify/websockifyserver.py

@@ -470,7 +470,8 @@ def socket(host, port=None, connect=False, prefer_ipv6=False,
             if connect:
                 sock.connect(addrs[0][4])
                 if use_ssl:
-                    sock = ssl.wrap_socket(sock)
+                    context = ssl.create_default_context()
+                    sock = context.wrap_socket(sock)
             else:
                 sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
                 sock.bind(addrs[0][4])