79 lines
2.2 KiB
Python
79 lines
2.2 KiB
Python
import shelve
|
|
from threading import Thread
|
|
from time import sleep
|
|
import os
|
|
import dbm
|
|
import traceback
|
|
from typing import Callable
|
|
|
|
from gossip.peer import Peer
|
|
from gossip.peerset import gossip_peer_set
|
|
from utils.identifyhome import identify_home
|
|
import logger
|
|
import config
|
|
from getsocks import get_socks
|
|
|
|
from torpeer import TorPeer
|
|
from torfilepaths import control_socket, peer_database_file
|
|
|
|
bootstrap_file = f'{os.path.dirname(os.path.realpath(__file__))}/bootstrap.txt'
|
|
|
|
|
|
def load_existing_peers(callback: Callable):
|
|
"""Load peers saved to disk"""
|
|
peer_address: str = ''
|
|
|
|
# the peers here are saved on clean shutdown
|
|
with shelve.open(peer_database_file, 'r') as tor_peer_db:
|
|
peer_address: str = tor_peer_db.nextkey()
|
|
while peer_address:
|
|
Thread(
|
|
target=callback,
|
|
args=[tor_peer_db[peer_address]],
|
|
name=f'{peer_address} connection attempt',
|
|
daemon=True).start()
|
|
|
|
|
|
def on_bootstrap(api, data):
|
|
|
|
callback_func = data['callback']
|
|
|
|
try:
|
|
load_existing_peers(callback_func)
|
|
except dbm.error:
|
|
try:
|
|
with open(bootstrap_file, 'r') as bootstrap_file_obj:
|
|
bootstrap_nodes = set(bootstrap_file_obj.read().split(','))
|
|
except FileNotFoundError:
|
|
bootstrap_nodes = set()
|
|
except Exception as _:
|
|
logger.warn(traceback.format_exc(), terminal=True)
|
|
return
|
|
else:
|
|
return
|
|
|
|
while not os.path.exists(control_socket):
|
|
sleep(0.1)
|
|
|
|
while not config.get('tor.transport_address'):
|
|
sleep(1)
|
|
try:
|
|
config.reload()
|
|
except Exception:
|
|
logger.error(traceback.format_exc(), terminal=True)
|
|
|
|
socks_address, socks_port = get_socks()[0]
|
|
|
|
for address in bootstrap_nodes:
|
|
if address == config.get('tor.transport_address').replace('.onion', '') or not address:
|
|
continue
|
|
assert not address.endswith('.onion')
|
|
address += '.onion'
|
|
# Tell the gossip logic that this peer is ready to connect
|
|
# it will add it to data['peer_set'] if it responds to ping
|
|
Thread(
|
|
target=callback_func,
|
|
args=[TorPeer(socks_address, socks_port, address)],
|
|
daemon=True).start()
|
|
|