def _request_from_node(teacher,
client: Optional[NucypherMiddlewareClient] = None,
timeout: int = 2,
log: Logger = IP_DETECTION_LOGGER) -> Union[str, None]:
if not client:
client = NucypherMiddlewareClient()
try:
response = client.get(node_or_sprout=teacher,
path=f"ping",
timeout=timeout) # TLS certificate logic within
except RestMiddleware.UnexpectedResponse:
# 404, 405, 500, All server response codes handled by will be caught here.
return # Default teacher does not support this request - just move on.
except NodeSeemsToBeDown:
# This node is unreachable. Move on.
return
if response.status_code == 200:
try:
ip = str(ip_address(response.text))
except ValueError:
error = f'Teacher {teacher} returned an invalid IP response; Got {response.text}'
raise UnknownIPAddress(error)
log.info(
f'Fetched external IP address ({ip}) from teacher ({teacher}).')
return ip
else:
# Something strange happened... move on anyways.
log.debug(
f'Failed to get external IP from teacher node ({teacher} returned {response.status_code})'
)
def get_external_ip_from_known_nodes(known_nodes: FleetSensor,
sample_size: int = 3,
log: Logger = IP_DETECTION_LOGGER
) -> Union[str, None]:
"""
Randomly select a sample of peers to determine the external IP address
of this host. The first node to reply successfully will be used.
# TODO: Parallelize the requests and compare results.
"""
if len(known_nodes) < sample_size:
return # There are too few known nodes
sample = random.sample(list(known_nodes), sample_size)
client = NucypherMiddlewareClient()
for node in sample:
ip = _request_from_node(teacher=node, client=client)
if ip:
log.info(f'Fetched external IP address ({ip}) from randomly selected known nodes.')
return ip