documented communicator2 better and bootstrap better

This commit is contained in:
Kevin Froman 2018-07-01 16:01:19 -05:00
parent a2ad90f2b9
commit 5c49f544fe
2 changed files with 37 additions and 7 deletions

View File

@ -27,22 +27,35 @@ class OnionrCommunicatorDaemon:
def __init__(self, debug, developmentMode): def __init__(self, debug, developmentMode):
logger.warn('New (unstable) communicator is being used.') logger.warn('New (unstable) communicator is being used.')
# list of timer instances
self.timers = [] self.timers = []
self._core = core.Core(torPort=sys.argv[2])
# initalize core with Tor socks port being 3rd argument
self.proxyPort = sys.argv[2]
self._core = core.Core(torPort=self.proxyPort)
# intalize NIST beacon salt and time
self.nistSaltTimestamp = 0 self.nistSaltTimestamp = 0
self.powSalt = 0 self.powSalt = 0
# loop time.sleep delay in seconds
self.delay = 1 self.delay = 1
self.proxyPort = sys.argv[2]
# time app started running for info/statistics purposes
self.startTime = self._core._utils.getEpoch() self.startTime = self._core._utils.getEpoch()
# lists of connected peers and peers we know we can't reach currently
self.onlinePeers = [] self.onlinePeers = []
self.offlinePeers = [] self.offlinePeers = []
# amount of threads running by name, used to prevent too many
self.threadCounts = {} self.threadCounts = {}
# set true when shutdown command recieved
self.shutdown = False self.shutdown = False
self.blockQueue = [] # list of new blocks to download # list of new blocks to download, added to when new block lists are fetched from peers
self.blockQueue = []
# Clear the daemon queue for any dead messages # Clear the daemon queue for any dead messages
if os.path.exists(self._core.queueDB): if os.path.exists(self._core.queueDB):
@ -195,8 +208,16 @@ class OnionrCommunicatorDaemon:
for i in range(needed): for i in range(needed):
self.connectNewPeer() self.connectNewPeer()
if len(self.onlinePeers) == 0:
self.addBootstrapListToPeerList()
self.decrementThreadCount('getOnlinePeers') self.decrementThreadCount('getOnlinePeers')
def addBootstrapListToPeerList(self):
'''Add the bootstrap list to the peer list (no duplicates)'''
for i in self._core.bootstrapList:
if i not in peerList:
peerList.append(i)
def connectNewPeer(self, peer=''): def connectNewPeer(self, peer=''):
'''Adds a new random online peer to self.onlinePeers''' '''Adds a new random online peer to self.onlinePeers'''
retData = False retData = False
@ -210,19 +231,20 @@ class OnionrCommunicatorDaemon:
peerList = self._core.listAdders() peerList = self._core.listAdders()
if len(peerList) == 0: if len(peerList) == 0:
peerList.extend(self._core.bootstrapList) # Avoid duplicating bootstrap addresses in peerList
self.addBootstrapListToPeerList()
for address in peerList: for address in peerList:
if len(address) == 0 or address in tried or address in self.onlinePeers: if len(address) == 0 or address in tried or address in self.onlinePeers:
continue continue
if self.peerAction(address, 'ping') == 'pong!': if self.peerAction(address, 'ping') == 'pong!':
logger.info('connected to ' + address) logger.info('Connected to ' + address)
self.onlinePeers.append(address) self.onlinePeers.append(address)
retData = address retData = address
break break
else: else:
tried.append(address) tried.append(address)
logger.debug('failed to connect to ' + address) logger.debug('Failed to connect to ' + address)
else: else:
if len(self.onlinePeers) == 0: if len(self.onlinePeers) == 0:
logger.warn('Could not connect to any peer') logger.warn('Could not connect to any peer')
@ -273,6 +295,8 @@ class OnionrCommunicatorDaemon:
open('data/.runcheck', 'w+').close() open('data/.runcheck', 'w+').close()
elif cmd[0] == 'connectedPeers': elif cmd[0] == 'connectedPeers':
self.printOnlinePeers() self.printOnlinePeers()
elif cmd[0] == 'kex':
self.timers['lookupKeys'].count = (self.timers['lookupKeys'].frequency - 1)
else: else:
logger.info('Recieved daemonQueue command:' + cmd[0]) logger.info('Recieved daemonQueue command:' + cmd[0])
self.decrementThreadCount('daemonCommands') self.decrementThreadCount('daemonCommands')

View File

@ -196,6 +196,7 @@ class Onionr:
'add-addr': self.addAddress, 'add-addr': self.addAddress,
'addaddr': self.addAddress, 'addaddr': self.addAddress,
'addaddress': self.addAddress, 'addaddress': self.addAddress,
'list-peers': self.listPeers,
'add-file': self.addFile, 'add-file': self.addFile,
'addfile': self.addFile, 'addfile': self.addFile,
@ -260,6 +261,11 @@ class Onionr:
def listConn(self): def listConn(self):
self.onionrCore.daemonQueueAdd('connectedPeers') self.onionrCore.daemonQueueAdd('connectedPeers')
def listPeers(self):
logger.info('Peer transport address list:')
for i in self.onionrCore.listAdders():
logger.info(i)
def getWebPassword(self): def getWebPassword(self):
return config.get('client.hmac') return config.get('client.hmac')