mirrors/gstreamer
mirrors/gstreamer
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gstreamer.git synced 2024-11-14 13:21:28 +00:00
Code Issues 1 Projects Releases Wiki Activity

simple_server: Abstract out ssl context generation

Browse source
This commit is contained in:
Nirbheek Chauhan 2020-05-25 18:33:32 +00:00 committed by Matthew Waters
parent 7b96b06752
commit 4761396d87
1 changed files with 28 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

49
webrtc/signalling/simple_server.py
View file

@ -222,28 +222,33 @@ class WebRTCSimpleServer(object):
await ws.send('HELLO') await ws.send('HELLO')
return uid return uid
def get_ssl_certs(self):
if 'letsencrypt' in self.cert_path:
chain_pem = os.path.join(self.cert_path, 'fullchain.pem')
key_pem = os.path.join(self.cert_path, 'privkey.pem')
else:
chain_pem = os.path.join(self.cert_path, 'cert.pem')
key_pem = os.path.join(self.cert_path, 'key.pem')
return chain_pem, key_pem
def get_ssl_ctx(self):
if self.disable_ssl:
return None
# Create an SSL context to be used by the websocket server
print('Using TLS with keys in {!r}'.format(self.cert_path))
chain_pem, key_pem = self.get_ssl_certs()
sslctx = ssl.create_default_context()
try:
sslctx.load_cert_chain(chain_pem, keyfile=key_pem)
except FileNotFoundError:
print("Certificates not found, did you run generate_cert.sh?")
sys.exit(1)
# FIXME
sslctx.check_hostname = False
sslctx.verify_mode = ssl.CERT_NONE
return sslctx
def run(self): def run(self):
sslctx = None
if not self.disable_ssl:
# Create an SSL context to be used by the websocket server
print('Using TLS with keys in {!r}'.format(self.certpath))
if 'letsencrypt' in self.certpath:
chain_pem = os.path.join(self.certpath, 'fullchain.pem')
key_pem = os.path.join(self.certpath, 'privkey.pem')
else:
chain_pem = os.path.join(self.certpath, 'cert.pem')
key_pem = os.path.join(self.certpath, 'key.pem')
sslctx = ssl.create_default_context()
try:
sslctx.load_cert_chain(chain_pem, keyfile=key_pem)
except FileNotFoundError:
print("Certificates not found, did you run generate_cert.sh?")
sys.exit(1)
# FIXME
sslctx.check_hostname = False
sslctx.verify_mode = ssl.CERT_NONE
async def handler(ws, path): async def handler(ws, path):
''' '''
All incoming messages are handled here. @path is unused. All incoming messages are handled here. @path is unused.
@ -258,6 +263,8 @@ class WebRTCSimpleServer(object):
finally: finally:
await self.remove_peer(peer_id) await self.remove_peer(peer_id)
sslctx = self.get_ssl_ctx()
print("Listening on https://{}:{}".format(self.addr, self.port)) print("Listening on https://{}:{}".format(self.addr, self.port))
# Websocket server # Websocket server
wsd = websockets.serve(handler, self.addr, self.port, ssl=sslctx, process_request=self.health_check if self.health_path else None, wsd = websockets.serve(handler, self.addr, self.port, ssl=sslctx, process_request=self.health_check if self.health_path else None,