2022-02-14 23:47:54 +00:00
|
|
|
"""Onionr - Private P2P Communication.
|
|
|
|
|
|
|
|
Dandelion ++ Gossip client logic
|
|
|
|
"""
|
2022-02-22 20:34:19 +00:00
|
|
|
import traceback
|
2022-06-05 20:11:53 +00:00
|
|
|
from threading import Thread
|
2022-02-11 06:56:19 +00:00
|
|
|
from typing import TYPE_CHECKING
|
2022-03-11 17:15:18 +00:00
|
|
|
from typing import Set, Tuple
|
2022-02-22 20:34:19 +00:00
|
|
|
from time import sleep
|
2022-03-20 23:05:44 +00:00
|
|
|
import asyncio
|
2022-02-11 06:56:19 +00:00
|
|
|
|
|
|
|
from queue import Queue
|
2022-03-05 00:05:12 +00:00
|
|
|
|
2022-03-10 07:10:13 +00:00
|
|
|
|
|
|
|
from onionrblocks import Block
|
|
|
|
|
2022-03-05 00:05:12 +00:00
|
|
|
from ..constants import DANDELION_EPOCH_LENGTH
|
2022-03-02 13:29:59 +00:00
|
|
|
from ..connectpeer import connect_peer
|
2022-02-11 06:56:19 +00:00
|
|
|
|
|
|
|
if TYPE_CHECKING:
|
2022-02-23 17:47:01 +00:00
|
|
|
from ..peer import Peer
|
2022-03-11 17:15:18 +00:00
|
|
|
from ordered_set import OrderedSet
|
2022-02-14 23:47:54 +00:00
|
|
|
|
2022-02-22 20:34:19 +00:00
|
|
|
import logger
|
2022-02-14 23:47:54 +00:00
|
|
|
import onionrplugins
|
2022-02-23 17:47:01 +00:00
|
|
|
from ..commands import GossipCommands
|
2022-03-05 00:05:12 +00:00
|
|
|
from gossip.dandelion.phase import DandelionPhase
|
2022-02-23 17:47:01 +00:00
|
|
|
from onionrthreads import add_onionr_thread
|
2022-03-13 01:28:18 +00:00
|
|
|
from blockdb import add_block_to_db
|
2022-03-10 07:10:13 +00:00
|
|
|
|
2022-02-23 17:47:01 +00:00
|
|
|
|
2022-05-09 17:38:03 +00:00
|
|
|
from .storeblocks import store_blocks
|
2022-02-23 17:47:01 +00:00
|
|
|
from .announce import do_announce
|
2022-03-02 13:29:59 +00:00
|
|
|
from .dandelionstem import stem_out
|
|
|
|
from .peerexchange import get_new_peers
|
2022-03-21 06:03:53 +00:00
|
|
|
from ..peerset import gossip_peer_set
|
2022-05-05 05:07:58 +00:00
|
|
|
from .streamblocks import stream_from_peers
|
2022-03-02 13:29:59 +00:00
|
|
|
|
2022-02-14 23:47:54 +00:00
|
|
|
"""
|
|
|
|
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/>.
|
|
|
|
"""
|
|
|
|
|
2022-05-09 17:38:03 +00:00
|
|
|
dandelion_phase = DandelionPhase(DANDELION_EPOCH_LENGTH)
|
|
|
|
|
|
|
|
|
|
|
|
def block_queue_processing():
|
|
|
|
|
|
|
|
|
|
|
|
while not len(gossip_peer_set):
|
|
|
|
sleep(0.2)
|
|
|
|
if dandelion_phase.remaining_time() <= 15:
|
|
|
|
logger.debug("Sleeping", terminal=True)
|
|
|
|
sleep(dandelion_phase.remaining_time())
|
|
|
|
if dandelion_phase.is_stem_phase():
|
|
|
|
logger.debug("Entering stem phase", terminal=True)
|
|
|
|
try:
|
|
|
|
# Stem out blocks for (roughly) remaining epoch time
|
|
|
|
asyncio.run(stem_out(dandelion_phase))
|
|
|
|
except TimeoutError:
|
|
|
|
pass
|
|
|
|
except Exception:
|
|
|
|
logger.error(traceback.format_exc(), terminal=True)
|
|
|
|
pass
|
|
|
|
else:
|
|
|
|
logger.debug("Entering fluff phase", terminal=True)
|
|
|
|
# Add block to primary block db, where the diffuser can read it
|
|
|
|
store_blocks(dandelion_phase)
|
2022-02-11 06:56:19 +00:00
|
|
|
|
2022-05-09 17:38:03 +00:00
|
|
|
|
|
|
|
def start_gossip_client():
|
2022-02-22 20:34:19 +00:00
|
|
|
"""
|
|
|
|
Gossip client does the following:
|
|
|
|
|
|
|
|
Stem new blocks we created or downloaded *during stem phase*
|
|
|
|
Stream new blocks
|
|
|
|
"""
|
2022-03-10 07:10:13 +00:00
|
|
|
bl: Block
|
2022-06-05 20:11:53 +00:00
|
|
|
|
|
|
|
def _start_announce():
|
|
|
|
sleep(60)
|
|
|
|
do_announce()
|
|
|
|
Thread(target=_start_announce, daemon=True).start()
|
2022-02-23 17:47:01 +00:00
|
|
|
|
2022-03-10 07:10:13 +00:00
|
|
|
# Start a thread that runs every 1200 secs to
|
|
|
|
# Ask peers for a subset for their peer set
|
|
|
|
# The transport addresses for said peers will
|
|
|
|
# be passed to a plugin event where the respective
|
|
|
|
# transport plugin handles the new peer
|
2022-02-23 17:47:01 +00:00
|
|
|
add_onionr_thread(
|
|
|
|
get_new_peers,
|
2022-05-13 05:17:06 +00:00
|
|
|
60, initial_sleep=120)
|
2022-02-22 20:34:19 +00:00
|
|
|
|
2022-03-28 05:13:36 +00:00
|
|
|
# Start a new thread to stream blocks from peers
|
2022-05-13 05:17:06 +00:00
|
|
|
# These blocks are being diffused and are stored in
|
|
|
|
# the peer's block database
|
2022-05-09 17:38:03 +00:00
|
|
|
add_onionr_thread(
|
|
|
|
stream_from_peers,
|
|
|
|
3, initial_sleep=10
|
|
|
|
)
|
2022-03-02 13:29:59 +00:00
|
|
|
|
2022-05-13 05:17:06 +00:00
|
|
|
# Blocks we receive or create through all means except
|
|
|
|
# Diffusal are put into block queues, we decide to either
|
|
|
|
# stem or diffuse a block from the queue based on the current
|
|
|
|
# dandelion++ phase
|
2022-02-22 20:34:19 +00:00
|
|
|
while True:
|
2022-05-09 17:38:03 +00:00
|
|
|
block_queue_processing()
|