Skip to content

Commit 94cdac8

Browse files
committed
use ssl contexts to enable privilege dropping
1 parent 650a7d8 commit 94cdac8

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

SimpleWebSocketServer/SimpleWebSocketServer.py

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -718,19 +718,14 @@ def __init__(self, host, port, websocketclass, certfile, keyfile, version = ssl.
718718

719719
SimpleWebSocketServer.__init__(self, host, port, websocketclass)
720720

721-
self.cerfile = certfile
722-
self.keyfile = keyfile
723-
self.version = version
721+
self.context = ssl.SSLContext(version)
722+
self.context.load_cert_chain(certfile, keyfile)
724723

725724
def close(self):
726725
super(SimpleSSLWebSocketServer, self).close()
727726

728727
def _decorateSocket(self, sock):
729-
sslsock = ssl.wrap_socket(sock,
730-
server_side=True,
731-
certfile=self.cerfile,
732-
keyfile=self.keyfile,
733-
ssl_version=self.version)
728+
sslsock = self.context.wrap_socket(sock, server_side=True)
734729
return sslsock
735730

736731
def _constructWebSocket(self, sock, address):

0 commit comments

Comments
 (0)