added flood fill work, misc bug fixes and removing communicator timers

master
Kevin Froman 2020-11-13 08:17:48 +00:00
parent 04105a2b70
commit bbd76da333
15 changed files with 98 additions and 82 deletions

View File

@ -5,7 +5,7 @@ This file registers blueprints for the private api server
from threading import Thread from threading import Thread
from gevent import sleep from gevent import sleep
from httpapi import security, friendsapi, profilesapi, configapi, insertblock from httpapi import security, friendsapi, configapi, insertblock
from httpapi import miscclientapi, onionrsitesapi, apiutils from httpapi import miscclientapi, onionrsitesapi, apiutils
from httpapi import directconnections from httpapi import directconnections
from httpapi import themeapi from httpapi import themeapi
@ -33,7 +33,6 @@ def register_private_blueprints(private_api, app):
app.register_blueprint(security.client.ClientAPISecurity( app.register_blueprint(security.client.ClientAPISecurity(
private_api).client_api_security_bp) private_api).client_api_security_bp)
app.register_blueprint(friendsapi.friends) app.register_blueprint(friendsapi.friends)
app.register_blueprint(profilesapi.profile_BP)
app.register_blueprint(configapi.config_BP) app.register_blueprint(configapi.config_BP)
app.register_blueprint(insertblock.ib) app.register_blueprint(insertblock.ib)
app.register_blueprint(miscclientapi.getblocks.client_get_blocks) app.register_blueprint(miscclientapi.getblocks.client_get_blocks)

View File

@ -86,12 +86,10 @@ class OnionrCommunicatorDaemon:
# extends our upload list and saves our list when Onionr exits # extends our upload list and saves our list when Onionr exits
uploadqueue.UploadQueue(self) uploadqueue.UploadQueue(self)
# Timers to periodically lookup new blocks and download them add_onionr_thread(
lookup_blocks_timer = OnionrCommunicatorTimers(
self,
lookupblocks.lookup_blocks_from_communicator, lookupblocks.lookup_blocks_from_communicator,
config.get('timers.lookupBlocks', 25), [self.shared_state], 25, 3)
my_args=[self], requires_peer=True, max_threads=1)
"""The block download timer is accessed by the block lookup function """The block download timer is accessed by the block lookup function
to trigger faster download starts""" to trigger faster download starts"""
@ -99,11 +97,7 @@ class OnionrCommunicatorDaemon:
self, self.getBlocks, config.get('timers.getBlocks', 10), self, self.getBlocks, config.get('timers.getBlocks', 10),
requires_peer=True, max_threads=5) requires_peer=True, max_threads=5)
# Timer to reset the longest offline peer add_onionr_thread(onlinepeers.clear_offline_peer, [self.kv], 58)
# so contact can be attempted again
OnionrCommunicatorTimers(
self, onlinepeers.clear_offline_peer, 58, my_args=[self],
max_threads=1)
# Timer to cleanup old blocks # Timer to cleanup old blocks
blockCleanupTimer = OnionrCommunicatorTimers( blockCleanupTimer = OnionrCommunicatorTimers(
@ -180,7 +174,6 @@ class OnionrCommunicatorDaemon:
# Adjust initial timer triggers # Adjust initial timer triggers
cleanupTimer.count = (cleanupTimer.frequency - 60) cleanupTimer.count = (cleanupTimer.frequency - 60)
blockCleanupTimer.count = (blockCleanupTimer.frequency - 2) blockCleanupTimer.count = (blockCleanupTimer.frequency - 2)
lookup_blocks_timer = (lookup_blocks_timer.frequency - 2)
shared_state.add(self) shared_state.add(self)

View File

@ -23,9 +23,8 @@ if TYPE_CHECKING:
""" """
def clear_offline_peer(comm_inst: 'OnionrCommunicatorDaemon'): def clear_offline_peer(kv: 'DeadSimpleKV'):
"""Remove the longest offline peer to retry later.""" """Remove the longest offline peer to retry later."""
kv: "DeadSimpleKV" = comm_inst.shared_state.get_by_string("DeadSimpleKV")
try: try:
removed = kv.get('offlinePeers').pop(0) removed = kv.get('offlinePeers').pop(0)
except IndexError: except IndexError:
@ -33,4 +32,4 @@ def clear_offline_peer(comm_inst: 'OnionrCommunicatorDaemon'):
else: else:
logger.debug('Removed ' + removed + logger.debug('Removed ' + removed +
' from offline list, will try them again.') ' from offline list, will try them again.')
comm_inst.decrementThreadCount('clear_offline_peer')

View File

@ -39,7 +39,7 @@ blacklist = onionrblacklist.OnionrBlackList()
storage_counter = StorageCounter() storage_counter = StorageCounter()
def lookup_blocks_from_communicator(comm_inst): def lookup_blocks_from_communicator(shared_state: 'TooMany'):
logger.info('Looking up new blocks') logger.info('Looking up new blocks')
tryAmount = 2 tryAmount = 2
newBlocks = '' newBlocks = ''
@ -50,7 +50,7 @@ def lookup_blocks_from_communicator(comm_inst):
maxBacklog = 1560 maxBacklog = 1560
lastLookupTime = 0 # Last time we looked up a particular peer's list lastLookupTime = 0 # Last time we looked up a particular peer's list
new_block_count = 0 new_block_count = 0
kv: "DeadSimpleKV" = comm_inst.shared_state.get_by_string("DeadSimpleKV") kv: "DeadSimpleKV" = shared_state.get_by_string("DeadSimpleKV")
for i in range(tryAmount): for i in range(tryAmount):
# Defined here to reset it each time, time offset is added later # Defined here to reset it each time, time offset is added later
listLookupCommand = 'getblocklist' listLookupCommand = 'getblocklist'
@ -87,7 +87,7 @@ def lookup_blocks_from_communicator(comm_inst):
listLookupCommand += '?date=%s' % (lastLookupTime,) listLookupCommand += '?date=%s' % (lastLookupTime,)
try: try:
newBlocks = peeraction.peer_action( newBlocks = peeraction.peer_action(
comm_inst.shared_state, shared_state,
peer, listLookupCommand) # get list of new block hashes peer, listLookupCommand) # get list of new block hashes
except Exception as error: except Exception as error:
logger.warn( logger.warn(
@ -124,6 +124,3 @@ def lookup_blocks_from_communicator(comm_inst):
logger.info( logger.info(
f'Discovered {new_block_count} new block{block_string}', f'Discovered {new_block_count} new block{block_string}',
terminal=True) terminal=True)
comm_inst.download_blocks_timer.count = \
int(comm_inst.download_blocks_timer.frequency * 0.99)
comm_inst.decrementThreadCount('lookup_blocks_from_communicator')

View File

@ -8,6 +8,7 @@ import platform
from flask import Response, Blueprint, request, send_from_directory, abort from flask import Response, Blueprint, request, send_from_directory, abort
from flask import g from flask import g
from gevent import sleep
import unpaddedbase32 import unpaddedbase32
from httpapi import apiutils from httpapi import apiutils
@ -138,10 +139,6 @@ class PrivateEndpoints:
return Response( return Response(
bytesconverter.bytes_to_str(mnemonickeys.get_base32(words))) bytesconverter.bytes_to_str(mnemonickeys.get_base32(words)))
@private_endpoints_bp.route('/gettorsocks')
def get_tor_socks():
return Response(str(g.too_many.get(NetController).socksPort))
@private_endpoints_bp.route('/setonboarding', methods=['POST']) @private_endpoints_bp.route('/setonboarding', methods=['POST'])
def set_onboarding(): def set_onboarding():
return Response( return Response(
@ -151,11 +148,25 @@ class PrivateEndpoints:
def get_os_system(): def get_os_system():
return Response(platform.system().lower()) return Response(platform.system().lower())
@private_endpoints_bp.route('/gettorsocks')
def get_tor_socks():
while True:
try:
return Response(
str(
g.too_many.get_by_string(
'NetController').socksPort))
except KeyError:
sleep(0.1)
@private_endpoints_bp.route('/torready') @private_endpoints_bp.route('/torready')
def is_tor_ready(): def is_tor_ready():
"""If Tor is starting up, the web UI is not ready to be used.""" """If Tor is starting up, the web UI is not ready to be used."""
try:
return Response( return Response(
str(g.too_many.get(NetController).readyState).lower()) str(g.too_many.get_by_string('NetController').readyState).lower())
except KeyError:
return Response("false")
@private_endpoints_bp.route('/gettoraddress') @private_endpoints_bp.route('/gettoraddress')
def get_tor_address(): def get_tor_address():

View File

@ -1,27 +0,0 @@
'''
Onionr - Private P2P Communication
This file creates http endpoints for user profile pages
'''
'''
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/>.
'''
from flask import Blueprint, Response, request, abort
from . import profiles
profile_BP = Blueprint('profile_BP', __name__)
@profile_BP.route('/profile/get/<pubkey>', endpoint='profiles')
def get_profile_page(pubkey):
return Response(pubkey)

View File

@ -1,2 +0,0 @@
def get_latest_user_profile(pubkey):
return ''

View File

@ -10,6 +10,7 @@ from onionrtypes import LANIP
import logger import logger
from coredb.blockmetadb import get_block_list from coredb.blockmetadb import get_block_list
from onionrblocks.blockimporter import import_block_from_data from onionrblocks.blockimporter import import_block_from_data
import onionrexceptions
from ..server import ports from ..server import ports
from threading import Thread from threading import Thread
@ -37,9 +38,12 @@ def _lan_work(peer: LANIP):
blocks = requests.get(url + 'blist/0').text.splitlines() blocks = requests.get(url + 'blist/0').text.splitlines()
for block in blocks: for block in blocks:
if block not in our_blocks: if block not in our_blocks:
try:
import_block_from_data( import_block_from_data(
requests.get( requests.get(
url + f'get/{block}', stream=True).raw.read(6000000)) url + f'get/{block}', stream=True).raw.read(6000000))
except onionrexceptions.InvalidMetadata:
logger.warn(f"Could not get {block} from lan peer")
for port in ports: for port in ports:
try: try:

View File

@ -1,17 +1,25 @@
from typing import Callable from typing import Callable
from typing import Iterable from typing import Iterable
import traceback
from threading import Thread from threading import Thread
from time import sleep from time import sleep
import logger
def _onionr_thread(func: Callable, args: Iterable, def _onionr_thread(func: Callable, args: Iterable,
sleep_secs: int, initial_sleep): sleep_secs: int, initial_sleep):
if initial_sleep: if initial_sleep:
sleep(initial_sleep) sleep(initial_sleep)
while True: while True:
try:
func(*args) func(*args)
except Exception as _: # noqa
logger.warn(
"Onionr thread exception \n" + traceback.format_exc(),
terminal=True)
sleep(sleep_secs) sleep(sleep_secs)

View File

@ -44,14 +44,14 @@ def test_clearnet_tor_request(testmanager):
try: try:
leak_result: str = do_get_request( leak_result: str = do_get_request(
'https://onionr.net/404', 'https://example.com/notvalidpage',
port=socks_port, ignoreAPI=True).lower() port=socks_port, ignoreAPI=True).lower()
except AttributeError: except AttributeError:
leak_result = "" leak_result = ""
except Exception as e: except Exception as e:
logger.warn(str(e)) logger.warn(str(e))
try: try:
if 'not found' in leak_result: if 'example' in leak_result:
logger.error('Tor was able to request a clearnet site') logger.error('Tor was able to request a clearnet site')
raise ValueError('Tor was able to request a clearnet site') raise ValueError('Tor was able to request a clearnet site')
except TypeError: except TypeError:

View File

@ -45,7 +45,9 @@ class _Importer(FileSystemEventHandler):
os.remove(event.src_path) os.remove(event.src_path)
try: try:
import_block_from_data(block_data) import_block_from_data(block_data)
except onionrexceptions.DataExists: except(
onionrexceptions.DataExists,
onionrexceptions.BlockMetaEntryExists) as _:
return return
if block_data_location in event.src_path: if block_data_location in event.src_path:
try: try:

10
src/streamfill/pool.py Normal file
View File

@ -0,0 +1,10 @@
from typing import List
from onionrutils.localcommand import local_command
from .neighbors import identify_neighbors
def stream_pool():
peers = lioc

View File

@ -0,0 +1,38 @@
from secrets import token_bytes
from typing import TYPE_CHECKING
import socket
if TYPE_CHECKING:
from stem.control import Controller
from onionrtypes import OnionAddressString
import yam
def peer_tunnel(tor_controller: Controller, peer):
socks_port = tor_controller.get_conf('SocksPort')
class Connected:
connected = False
send_buffer = []
rec_buffer = []
rec_address = None
yam.client(1, peer, socks_port, send_buffer, rec_buffer, Connected)
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
ip = '127.0.0.1'
s.bind((ip, 0))
s.listen(1)
port = s.getsockname()[1]
serv = tor_controller.create_ephemeral_hidden_service(
{1337: '127.0.0.1:' + str(port)},
key_content='ED25519-V3',
await_publication=True,
)
rec_address = serv.service_id
conn, addr = s.accept()
yam.server(1, tor_controller, conn, send_buffer, rec_buffer, Connected)

View File

@ -1,16 +0,0 @@
import sys, os
sys.path.append(".")
sys.path.append("src/")
import unittest, uuid
TEST_DIR = 'testdata/%s-%s' % (uuid.uuid4(), os.path.basename(__file__)) + '/'
print("Test directory:", TEST_DIR)
os.environ["ONIONR_HOME"] = TEST_DIR
from onionrsetup import setup_config
setup_config()
import config
class TestRandomBindIP(unittest.TestCase):
def test_random_bind_ip_default_setting(self):
self.assertTrue(config.get('general.random_bind_ip'))
unittest.main()