79 lines
2.9 KiB
Python
79 lines
2.9 KiB
Python
"""Onionr - Private P2P Communication.
|
|
|
|
This file implements logic for performing requests to Onionr peers
|
|
"""
|
|
from typing import TYPE_CHECKING
|
|
|
|
import streamedrequests
|
|
import logger
|
|
from onionrutils import epoch, basicrequests
|
|
from coredb import keydb
|
|
from . import onlinepeers
|
|
from onionrtypes import OnionAddressString
|
|
from onionrpeers.peerprofiles import PeerProfiles
|
|
from etc.waitforsetvar import wait_for_set_var
|
|
"""
|
|
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/>.
|
|
"""
|
|
|
|
|
|
def get_peer_profile(kv, address: OnionAddressString) -> 'PeerProfiles':
|
|
profile_inst_list = kv.get('peerProfiles')
|
|
for profile in profile_inst_list:
|
|
if profile.address == address:
|
|
return profile
|
|
p = PeerProfiles(address)
|
|
profile_inst_list.append(p)
|
|
return p
|
|
|
|
|
|
def peer_action(shared_state, peer, action,
|
|
returnHeaders=False, max_resp_size=5242880):
|
|
"""Perform a get request to a peer."""
|
|
penalty_score = -10
|
|
kv: "DeadSimpleKV" = shared_state.get_by_string("DeadSimpleKV")
|
|
if len(peer) == 0:
|
|
return False
|
|
url = 'http://%s/%s' % (peer, action)
|
|
|
|
try:
|
|
ret_data = basicrequests.do_get_request(url, port=kv.get('proxyPort'),
|
|
max_size=max_resp_size)
|
|
except streamedrequests.exceptions.ResponseLimitReached:
|
|
logger.warn(
|
|
'Request failed due to max response size being overflowed',
|
|
terminal=True)
|
|
ret_data = False
|
|
penalty_score = -100
|
|
# if request failed, (error), mark peer offline
|
|
if ret_data is False:
|
|
try:
|
|
get_peer_profile(kv, peer).addScore(penalty_score)
|
|
onlinepeers.remove_online_peer(kv, peer)
|
|
keydb.transportinfo.set_address_info(
|
|
peer, 'lastConnectAttempt', epoch.get_epoch())
|
|
if action != 'ping' and not kv.get('shutdown'):
|
|
logger.warn(f'Lost connection to {peer}', terminal=True)
|
|
# Will only add a new peer to pool if needed
|
|
onlinepeers.get_online_peers(kv)
|
|
except ValueError:
|
|
pass
|
|
else:
|
|
peer_profile = get_peer_profile(kv, peer)
|
|
peer_profile.update_connect_time()
|
|
peer_profile.addScore(1)
|
|
# If returnHeaders, returns tuple of data, headers.
|
|
# If not, just data string
|
|
return ret_data
|