lbry-sdk/lbrynet/core/Session.py

293 lines
12 KiB
Python
Raw Normal View History

2015-08-20 17:27:15 +02:00
import logging
import miniupnpc
from twisted.internet import threads, defer
from lbrynet.core.BlobManager import DiskBlobManager
from lbrynet.dht import node
from lbrynet.database.storage import SQLiteStorage
2015-08-20 17:27:15 +02:00
from lbrynet.core.RateLimiter import RateLimiter
from lbrynet.core.utils import generate_id
from lbrynet.core.PaymentRateManager import BasePaymentRateManager, NegotiatedPaymentRateManager
from lbrynet.core.BlobAvailability import BlobAvailabilityTracker
2015-08-20 17:27:15 +02:00
log = logging.getLogger(__name__)
class Session(object):
2016-11-03 20:22:16 +01:00
"""This class manages all important services common to any application that uses the network.
the hash announcer, which informs other peers that this peer is
associated with some hash. Usually, this means this peer has a
blob identified by the hash in question, but it can be used for
other purposes.
the peer finder, which finds peers that are associated with some
hash.
the blob manager, which keeps track of which blobs have been
downloaded and provides access to them,
the rate limiter, which attempts to ensure download and upload
rates stay below a set maximum
upnp, which opens holes in compatible firewalls so that remote
peers can connect to this peer.
"""
2017-04-10 16:51:49 +02:00
2016-11-03 20:22:16 +01:00
def __init__(self, blob_data_payment_rate, db_dir=None,
2017-10-10 19:15:25 +02:00
node_id=None, peer_manager=None, dht_node_port=None,
2016-11-03 20:22:16 +01:00
known_dht_nodes=None, peer_finder=None,
hash_announcer=None, blob_dir=None,
blob_manager=None, peer_port=None, use_upnp=True,
rate_limiter=None, wallet=None,
dht_node_class=node.Node, blob_tracker_class=None,
payment_rate_manager_class=None, is_generous=True, external_ip=None, storage=None):
2016-11-03 20:22:16 +01:00
"""@param blob_data_payment_rate: The default payment rate for blob data
2015-08-20 17:27:15 +02:00
@param db_dir: The directory in which levelDB files should be stored
2017-10-10 19:15:25 +02:00
@param node_id: The unique ID of this node
2015-08-20 17:27:15 +02:00
2016-11-03 20:22:16 +01:00
@param peer_manager: An object which keeps track of all known
peers. If None, a PeerManager will be created
2015-08-20 17:27:15 +02:00
2016-11-03 20:22:16 +01:00
@param dht_node_port: The port on which the dht node should
listen for incoming connections
2015-08-20 17:27:15 +02:00
2016-11-03 20:22:16 +01:00
@param known_dht_nodes: A list of nodes which the dht node
should use to bootstrap into the dht
2015-08-20 17:27:15 +02:00
2016-11-03 20:22:16 +01:00
@param peer_finder: An object which is used to look up peers
that are associated with some hash. If None, a
DHTPeerFinder will be used, which looks for peers in the
distributed hash table.
2015-08-20 17:27:15 +02:00
2016-11-03 20:22:16 +01:00
@param hash_announcer: An object which announces to other
peers that this peer is associated with some hash. If
None, and peer_port is not None, a DHTHashAnnouncer will
be used. If None and peer_port is None, a
DummyHashAnnouncer will be used, which will not actually
announce anything.
2015-08-20 17:27:15 +02:00
2016-11-03 20:22:16 +01:00
@param blob_dir: The directory in which blobs will be
stored. If None and blob_manager is None, blobs will be
stored in memory only.
2015-08-20 17:27:15 +02:00
2016-11-03 20:22:16 +01:00
@param blob_manager: An object which keeps track of downloaded
blobs and provides access to them. If None, and blob_dir
is not None, a DiskBlobManager will be used, with the
given blob_dir. If None and blob_dir is None, a
TempBlobManager will be used, which stores blobs in memory
only.
2015-08-20 17:27:15 +02:00
2016-11-03 20:22:16 +01:00
@param peer_port: The port on which other peers should connect
to this peer
2015-08-20 17:27:15 +02:00
2016-11-03 20:22:16 +01:00
@param use_upnp: Whether or not to try to open a hole in the
firewall so that outside peers can connect to this peer's
peer_port and dht_node_port
2015-08-20 17:27:15 +02:00
2016-11-03 20:22:16 +01:00
@param rate_limiter: An object which keeps track of the amount
of data transferred to and from this peer, and can limit
that rate if desired
2015-08-20 17:27:15 +02:00
2016-11-03 20:22:16 +01:00
@param wallet: An object which will be used to keep track of
expected payments and which will pay peers. If None, a
wallet which uses the Point Trader system will be used,
which is meant for testing only
2015-08-20 17:27:15 +02:00
"""
self.db_dir = db_dir
2017-10-10 19:15:25 +02:00
self.node_id = node_id
2015-08-20 17:27:15 +02:00
self.peer_manager = peer_manager
self.peer_finder = peer_finder
self.hash_announcer = hash_announcer
2015-08-20 17:27:15 +02:00
self.dht_node_port = dht_node_port
self.known_dht_nodes = known_dht_nodes
if self.known_dht_nodes is None:
self.known_dht_nodes = []
self.blob_dir = blob_dir
self.blob_manager = blob_manager
self.blob_tracker = None
self.blob_tracker_class = blob_tracker_class or BlobAvailabilityTracker
2015-08-20 17:27:15 +02:00
self.peer_port = peer_port
self.use_upnp = use_upnp
self.rate_limiter = rate_limiter
self.external_ip = external_ip
self.upnp_redirects = []
2015-08-20 17:27:15 +02:00
self.wallet = wallet
2016-06-04 20:28:32 +02:00
self.dht_node_class = dht_node_class
2015-08-20 17:27:15 +02:00
self.dht_node = None
self.base_payment_rate_manager = BasePaymentRateManager(blob_data_payment_rate)
self.payment_rate_manager = None
2016-10-13 19:35:55 +02:00
self.payment_rate_manager_class = payment_rate_manager_class or NegotiatedPaymentRateManager
self.is_generous = is_generous
self.storage = storage or SQLiteStorage(self.db_dir)
self._join_dht_deferred = None
2015-08-20 17:27:15 +02:00
def setup(self):
"""Create the blob directory and database if necessary, start all desired services"""
2017-08-02 21:48:07 +02:00
log.debug("Starting session.")
2015-08-20 17:27:15 +02:00
2017-10-10 19:15:25 +02:00
if self.node_id is None:
self.node_id = generate_id()
2015-08-20 17:27:15 +02:00
if self.wallet is None:
from lbrynet.core.PTCWallet import PTCWallet
2015-08-20 17:27:15 +02:00
self.wallet = PTCWallet(self.db_dir)
if self.use_upnp is True:
d = self._try_upnp()
else:
d = defer.succeed(True)
d.addCallback(lambda _: self._setup_dht())
2015-08-20 17:27:15 +02:00
d.addCallback(lambda _: self._setup_other_components())
return d
def shut_down(self):
"""Stop all services"""
2017-08-02 21:48:07 +02:00
log.info('Stopping session.')
2015-08-20 17:27:15 +02:00
ds = []
if self.blob_tracker is not None:
ds.append(defer.maybeDeferred(self.blob_tracker.stop))
2015-08-20 17:27:15 +02:00
if self.dht_node is not None:
ds.append(defer.maybeDeferred(self.dht_node.stop))
if self.rate_limiter is not None:
ds.append(defer.maybeDeferred(self.rate_limiter.stop))
if self.wallet is not None:
ds.append(defer.maybeDeferred(self.wallet.stop))
if self.blob_manager is not None:
ds.append(defer.maybeDeferred(self.blob_manager.stop))
if self.use_upnp is True:
ds.append(defer.maybeDeferred(self._unset_upnp))
2015-08-20 17:27:15 +02:00
return defer.DeferredList(ds)
def _try_upnp(self):
log.debug("In _try_upnp")
2015-08-20 17:27:15 +02:00
2017-10-25 01:11:19 +02:00
def get_free_port(upnp, port, protocol):
# returns an existing mapping if it exists
mapping = upnp.getspecificportmapping(port, protocol)
if not mapping:
return port
if upnp.lanaddr == mapping[0]:
return mapping
return get_free_port(upnp, port + 1, protocol)
def get_port_mapping(upnp, internal_port, protocol, description):
# try to map to the requested port, if there is already a mapping use the next external
# port available
if protocol not in ['UDP', 'TCP']:
raise Exception("invalid protocol")
external_port = get_free_port(upnp, internal_port, protocol)
if isinstance(external_port, tuple):
log.info("Found existing UPnP redirect %s:%i (%s) to %s:%i, using it",
self.external_ip, external_port[1], protocol, upnp.lanaddr, internal_port)
return external_port[1], protocol
upnp.addportmapping(external_port, protocol, upnp.lanaddr, internal_port,
description, '')
log.info("Set UPnP redirect %s:%i (%s) to %s:%i", self.external_ip, external_port,
protocol, upnp.lanaddr, internal_port)
return external_port, protocol
2015-08-20 17:27:15 +02:00
def threaded_try_upnp():
if self.use_upnp is False:
log.debug("Not using upnp")
2015-08-20 17:27:15 +02:00
return False
u = miniupnpc.UPnP()
num_devices_found = u.discover()
if num_devices_found > 0:
u.selectigd()
external_ip = u.externalipaddress()
2017-10-25 01:11:19 +02:00
if external_ip != '0.0.0.0' and not self.external_ip:
# best not to rely on this external ip, the router can be behind layers of NATs
2015-08-20 17:27:15 +02:00
self.external_ip = external_ip
2017-10-25 01:11:19 +02:00
if self.peer_port:
2018-02-12 20:16:43 +01:00
self.upnp_redirects.append(
get_port_mapping(u, self.peer_port, 'TCP', 'LBRY peer port')
)
2017-10-25 01:11:19 +02:00
if self.dht_node_port:
2018-02-12 20:16:43 +01:00
self.upnp_redirects.append(
get_port_mapping(u, self.dht_node_port, 'UDP', 'LBRY DHT port')
)
2015-08-20 17:27:15 +02:00
return True
return False
def upnp_failed(err):
log.warning("UPnP failed. Reason: %s", err.getErrorMessage())
2015-08-20 17:27:15 +02:00
return False
d = threads.deferToThread(threaded_try_upnp)
d.addErrback(upnp_failed)
return d
def _setup_dht(self): # does not block startup, the dht will re-attempt if necessary
2016-06-04 20:28:32 +02:00
self.dht_node = self.dht_node_class(
self.hash_announcer,
2016-06-04 20:28:32 +02:00
udpPort=self.dht_node_port,
2017-10-10 19:15:25 +02:00
node_id=self.node_id,
externalIP=self.external_ip,
peerPort=self.peer_port,
peer_manager=self.peer_manager,
peer_finder=self.peer_finder,
2016-06-04 20:28:32 +02:00
)
self.peer_manager = self.dht_node.peer_manager
self.peer_finder = self.dht_node.peer_finder
self.hash_announcer = self.dht_node.hash_announcer
self._join_dht_deferred = self.dht_node.joinNetwork(self.known_dht_nodes)
self._join_dht_deferred.addCallback(lambda _: log.info("Joined the dht"))
2015-08-20 17:27:15 +02:00
def _setup_other_components(self):
log.debug("Setting up the rest of the components")
2015-08-20 17:27:15 +02:00
if self.rate_limiter is None:
self.rate_limiter = RateLimiter()
if self.blob_manager is None:
if self.blob_dir is None:
raise Exception(
"TempBlobManager is no longer supported, specify BlobManager or db_dir")
2015-08-20 17:27:15 +02:00
else:
self.blob_manager = DiskBlobManager(
self.dht_node.hash_announcer, self.blob_dir, self.storage
)
2015-08-20 17:27:15 +02:00
if self.blob_tracker is None:
2018-02-12 20:16:43 +01:00
self.blob_tracker = self.blob_tracker_class(
self.blob_manager, self.dht_node.peer_finder, self.dht_node
2018-02-12 20:16:43 +01:00
)
if self.payment_rate_manager is None:
2016-11-03 20:22:16 +01:00
self.payment_rate_manager = self.payment_rate_manager_class(
2018-02-12 20:16:43 +01:00
self.base_payment_rate_manager, self.blob_tracker, self.is_generous
)
self.rate_limiter.start()
d = self.storage.setup()
2018-03-08 22:46:39 +01:00
d.addCallback(lambda _: self.blob_manager.setup())
d.addCallback(lambda _: self.wallet.start())
d.addCallback(lambda _: self.blob_tracker.start())
return d
2015-08-20 17:27:15 +02:00
def _unset_upnp(self):
2017-08-02 21:48:07 +02:00
log.info("Unsetting upnp for session")
2017-04-10 16:51:49 +02:00
2015-08-20 17:27:15 +02:00
def threaded_unset_upnp():
u = miniupnpc.UPnP()
num_devices_found = u.discover()
if num_devices_found > 0:
u.selectigd()
for port, protocol in self.upnp_redirects:
if u.getspecificportmapping(port, protocol) is None:
2016-11-03 20:22:16 +01:00
log.warning(
"UPnP redirect for %s %d was removed by something else.",
protocol, port)
else:
u.deleteportmapping(port, protocol)
log.info("Removed UPnP redirect for %s %d.", protocol, port)
self.upnp_redirects = []
2015-08-20 17:27:15 +02:00
d = threads.deferToThread(threaded_unset_upnp)
d.addErrback(lambda err: str(err))
2016-06-04 20:28:32 +02:00
return d