Skip to content

Commit ecba035

Browse files
committed
Stop being silly about ssl/tls
1 parent 4f538d3 commit ecba035

4 files changed

Lines changed: 15 additions & 16 deletions

File tree

receptor/config.py

Lines changed: 12 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import importlib
44
import logging
55
import os
6-
import ssl as tls
6+
import ssl
77

88
from .entrypoints import run_as_node, run_as_controller, run_as_ping, run_as_send
99
from .exceptions import ReceptorRuntimeError, ReceptorConfigError
@@ -58,7 +58,6 @@ def __init__(self, args=None):
5858
self.add_config_option(
5959
section='default',
6060
key='node_id',
61-
#long_option='--node-id',
6261
default_value='',
6362
value_type='str',
6463
hint='Set/override node identifier. If unspecified here or in a config file, one will be automatically generated.',
@@ -91,19 +90,19 @@ def __init__(self, args=None):
9190
# so all of these options use `subparse=False`.
9291
self.add_config_option(
9392
section='auth',
94-
key='tls_cert',
93+
key='ssl_cert',
9594
default_value='',
9695
value_type='str',
9796
subparse=False,
98-
hint='Path to the TLS certificate chain file.',
97+
hint='Path to the SSL/TLS certificate chain file.',
9998
)
10099
self.add_config_option(
101100
section='auth',
102-
key='tls_key',
101+
key='ssl_key',
103102
default_value='',
104103
value_type='str',
105104
subparse=False,
106-
hint='Path to the TLS certificate key file.',
105+
hint='Path to the SSL/TLS certificate key file.',
107106
)
108107
# Receptor node options
109108
self.add_config_option(
@@ -375,18 +374,18 @@ def go(self):
375374
self._parsed_args.func(self)
376375

377376

378-
def get_client_tls_context(self):
379-
if self.auth_tls_cert:
377+
def get_client_ssl_context(self):
378+
if self.auth_ssl_cert:
380379
logger.debug("Loading SSL Client Context")
381-
return tls.create_default_context(tls.Purpose.SERVER_AUTH, cafile=self.auth_tls_cert)
380+
return ssl.create_default_context(ssl.Purpose.SERVER_AUTH, cafile=self.auth_ssl_cert)
382381
else:
383382
return None
384383

385-
def get_server_tls_context(self):
386-
if self.auth_tls_cert and self.auth_tls_key:
384+
def get_server_ssl_context(self):
385+
if self.auth_ssl_cert and self.auth_ssl_key:
387386
logger.debug("Loading SSL Server Context")
388-
sc = tls.create_default_context(tls.Purpose.CLIENT_AUTH)
389-
sc.load_cert_chain(self.auth_tls_cert, self.auth_tls_key)
387+
sc = ssl.create_default_context(ssl.Purpose.CLIENT_AUTH)
388+
sc.load_cert_chain(self.auth_ssl_cert, self.auth_ssl_key)
390389
return sc
391390
else:
392391
return None

receptor/controller.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def mainloop(receptor, socket_path, loop=asyncio.get_event_loop()):
2626
config = receptor.config
2727
listener = loop.create_server(
2828
lambda: protocol.BasicProtocol(receptor, loop),
29-
config.controller_listen_address, config.controller_listen_port, ssl=config.get_server_tls_context())
29+
config.controller_listen_address, config.controller_listen_port, ssl=config.get_server_ssl_context())
3030
logger.info("Serving on %s:%s", config.controller_listen_address, config.controller_listen_port)
3131
loop.create_task(listener)
3232
control_listener = loop.create_unix_server(

receptor/node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ def mainloop(receptor, ping_interval=None, loop=asyncio.get_event_loop(), skip_r
1212
if not config.node_server_disable:
1313
listener = loop.create_server(
1414
lambda: BasicProtocol(receptor, loop),
15-
config.node_listen_address, config.node_listen_port, ssl=config.get_server_tls_context())
15+
config.node_listen_address, config.node_listen_port, ssl=config.get_server_ssl_context())
1616
loop.create_task(listener)
1717
logger.info("Serving on %s:%s", config.node_listen_address, config.node_listen_port)
1818
for peer in config.node_peers:

receptor/protocol.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ async def create_peer(receptor, loop, host, port):
125125
while True:
126126
try:
127127
await loop.create_connection(
128-
lambda: BasicClientProtocol(receptor, loop), host, port, ssl=receptor.config.get_client_tls_context())
128+
lambda: BasicClientProtocol(receptor, loop), host, port, ssl=receptor.config.get_client_ssl_context())
129129
break
130130
except Exception:
131131
logger.exception("Connection Refused: {}:{}".format(host, port))

0 commit comments

Comments
 (0)