78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
"""
|
|
Select which node to connect to
|
|
"""
|
|
# pylint: disable=too-many-branches
|
|
import logging
|
|
import random # nosec
|
|
|
|
import knownnodes
|
|
import protocol
|
|
import state
|
|
from bmconfigparser import BMConfigParser
|
|
from queues import Queue, portCheckerQueue
|
|
|
|
logger = logging.getLogger('default')
|
|
|
|
|
|
def getDiscoveredPeer():
|
|
"""Get a peer from the local peer discovery list"""
|
|
try:
|
|
peer = random.choice(state.discoveredPeers.keys())
|
|
except (IndexError, KeyError):
|
|
raise ValueError
|
|
try:
|
|
del state.discoveredPeers[peer]
|
|
except KeyError:
|
|
pass
|
|
return peer
|
|
|
|
|
|
def chooseConnection(stream):
|
|
"""Returns an appropriate connection"""
|
|
haveOnion = BMConfigParser().safeGet(
|
|
"bitmessagesettings", "socksproxytype")[0:5] == 'SOCKS'
|
|
onionOnly = BMConfigParser().safeGetBoolean(
|
|
"bitmessagesettings", "onionservicesonly")
|
|
try:
|
|
retval = portCheckerQueue.get(False)
|
|
portCheckerQueue.task_done()
|
|
return retval
|
|
except Queue.Empty:
|
|
pass
|
|
# with a probability of 0.5, connect to a discovered peer
|
|
if random.choice((False, True)) and not haveOnion:
|
|
# discovered peers are already filtered by allowed streams
|
|
return getDiscoveredPeer()
|
|
for _ in range(50):
|
|
peer = random.choice(knownnodes.knownNodes[stream].keys())
|
|
try:
|
|
peer_info = knownnodes.knownNodes[stream][peer]
|
|
if peer_info.get('self'):
|
|
continue
|
|
rating = peer_info["rating"]
|
|
except TypeError:
|
|
logger.warning('Error in %s', peer)
|
|
rating = 0
|
|
if haveOnion:
|
|
# do not connect to raw IP addresses
|
|
# --keep all traffic within Tor overlay
|
|
if onionOnly and not peer.host.endswith('.onion'):
|
|
continue
|
|
# onion addresses have a higher priority when SOCKS
|
|
if peer.host.endswith('.onion') and rating > 0:
|
|
rating = 1
|
|
# TODO: need better check
|
|
elif not peer.host.startswith('bootstrap'):
|
|
encodedAddr = protocol.encodeHost(peer.host)
|
|
# don't connect to local IPs when using SOCKS
|
|
if not protocol.checkIPAddress(encodedAddr, False):
|
|
continue
|
|
if rating > 1:
|
|
rating = 1
|
|
try:
|
|
if 0.05 / (1.0 - rating) > random.random():
|
|
return peer
|
|
except ZeroDivisionError:
|
|
return peer
|
|
raise ValueError
|