80 lines
3.2 KiB
Python
80 lines
3.2 KiB
Python
|
'''
|
||
|
|
Onionr - P2P Microblogging Platform & Social network
|
||
|
|
|
||
|
|
Connect a new peer to our communicator instance. Does so randomly if no peer is specified
|
||
|
|
'''
|
||
|
|
'''
|
||
|
|
This program is free software: you can redistribute it and/or modify
|
||
|
|
it under the terms of the GNU General Public License as published by
|
||
|
|
the Free Software Foundation, either version 3 of the License, or
|
||
|
|
(at your option) any later version.
|
||
|
|
|
||
|
|
This program is distributed in the hope that it will be useful,
|
||
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
GNU General Public License for more details.
|
||
|
|
|
||
|
|
You should have received a copy of the GNU General Public License
|
||
|
|
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||
|
|
'''
|
||
|
|
import time
|
||
|
|
import onionrexceptions, logger, onionrpeers
|
||
|
|
from utils import networkmerger
|
||
|
|
def connect_new_peer_to_communicator(comm_inst, peer='', useBootstrap=False):
|
||
|
|
config = comm_inst._core.config
|
||
|
|
retData = False
|
||
|
|
tried = comm_inst.offlinePeers
|
||
|
|
if peer != '':
|
||
|
|
if comm_inst._core._utils.validateID(peer):
|
||
|
|
peerList = [peer]
|
||
|
|
else:
|
||
|
|
raise onionrexceptions.InvalidAddress('Will not attempt connection test to invalid address')
|
||
|
|
else:
|
||
|
|
peerList = comm_inst._core.listAdders()
|
||
|
|
|
||
|
|
mainPeerList = comm_inst._core.listAdders()
|
||
|
|
peerList = onionrpeers.getScoreSortedPeerList(comm_inst._core)
|
||
|
|
|
||
|
|
if len(peerList) < 8 or secrets.randbelow(4) == 3:
|
||
|
|
tryingNew = []
|
||
|
|
for x in comm_inst.newPeers:
|
||
|
|
if x not in peerList:
|
||
|
|
peerList.append(x)
|
||
|
|
tryingNew.append(x)
|
||
|
|
for i in tryingNew:
|
||
|
|
comm_inst.newPeers.remove(i)
|
||
|
|
|
||
|
|
if len(peerList) == 0 or useBootstrap:
|
||
|
|
# Avoid duplicating bootstrap addresses in peerList
|
||
|
|
comm_inst.addBootstrapListToPeerList(peerList)
|
||
|
|
|
||
|
|
for address in peerList:
|
||
|
|
if not config.get('tor.v3onions') and len(address) == 62:
|
||
|
|
continue
|
||
|
|
if address == comm_inst._core.hsAddress:
|
||
|
|
continue
|
||
|
|
if len(address) == 0 or address in tried or address in comm_inst.onlinePeers or address in comm_inst.cooldownPeer:
|
||
|
|
continue
|
||
|
|
if comm_inst.shutdown:
|
||
|
|
return
|
||
|
|
if comm_inst.peerAction(address, 'ping') == 'pong!':
|
||
|
|
time.sleep(0.1)
|
||
|
|
if address not in mainPeerList:
|
||
|
|
networkmerger.mergeAdders(address, comm_inst._core)
|
||
|
|
if address not in comm_inst.onlinePeers:
|
||
|
|
logger.info('Connected to ' + address)
|
||
|
|
comm_inst.onlinePeers.append(address)
|
||
|
|
comm_inst.connectTimes[address] = comm_inst._core._utils.getEpoch()
|
||
|
|
retData = address
|
||
|
|
|
||
|
|
# add peer to profile list if they're not in it
|
||
|
|
for profile in comm_inst.peerProfiles:
|
||
|
|
if profile.address == address:
|
||
|
|
break
|
||
|
|
else:
|
||
|
|
comm_inst.peerProfiles.append(onionrpeers.PeerProfiles(address, comm_inst._core))
|
||
|
|
break
|
||
|
|
else:
|
||
|
|
tried.append(address)
|
||
|
|
logger.debug('Failed to connect to ' + address)
|
||
|
|
return retData
|