def local_address():
local_ip = socket.gethostbyname('localhost')
local_port = 56789
yield PeerAddress(
network=Network.IPV4,
host=local_ip,
port=local_port,
)
def gen_squeak_peer(peer_name):
host = "random_host_{}".format(uuid.uuid1())
port = random.randint(1, 10000)
peer_address = PeerAddress(
network=Network.IPV4,
host=host,
port=port,
)
return create_saved_peer(
peer_name,
peer_address,
)
def handle_get_offer(self, squeak_hash_str, client_host,
client_port) -> Offer:
squeak_hash = bytes.fromhex(squeak_hash_str)
client_addr = PeerAddress(
network=Network.IPV4,
host=client_host,
port=client_port,
)
offer = self.squeak_controller.get_packaged_offer(
squeak_hash,
client_addr,
)
if not offer:
raise NotFoundError()
return offer
def accept_connections(self):
try:
self.listen_socket.bind(('', self.port))
self.listen_socket.listen()
while True:
peer_socket, address = self.listen_socket.accept()
host, port = address
peer_address = PeerAddress(
network=Network.IPV4,
host=host,
port=port,
)
peer_socket.setblocking(True)
self.handle_connection(
peer_socket,
peer_address,
)
except Exception:
logger.exception("Accept peer connections failed.")
def peer_address_with_zero():
return PeerAddress(
network=Network.IPV4,
host="fake_host",
port=0,
)
def peer_address():
return PeerAddress(
network=Network.IPV4,
host="fake_host",
port=5678,
)
def peer_address_with_no_port():
yield PeerAddress(
network=Network.IPV4,
host="fake_host",
port=0,
)
def peer_address_with_tor():
yield PeerAddress(
network=Network.TORV3,
host="fake_host",
port=1234,
)
def peer_address():
yield PeerAddress(
network=Network.IPV4,
host="fake_host",
port=8765,
)
def message_to_peer_address(msg: squeak_admin_pb2.PeerAddress) -> PeerAddress:
return PeerAddress(
network=Network[msg.network],
host=msg.host,
port=msg.port,
)
def get_external_address(self) -> PeerAddress:
return PeerAddress(
network=Network.IPV4,
host=self.config.server.external_address or '',
port=self.config.server.external_port or 0,
)
def outbound_local_address():
yield PeerAddress(
network=Network.IPV4,
host='outbound.com',
port=4321,
)
def inbound_local_address():
yield PeerAddress(
network=Network.IPV4,
host='inbound.com',
port=56789,
)
def external_address(self) -> PeerAddress:
return PeerAddress(
network=Network.IPV4,
host=self.external_host or self.local_ip,
port=self.local_port,
)