forked from LBRYCommunity/lbry-sdk
87 lines
3.1 KiB
Python
87 lines
3.1 KiB
Python
import random
|
|
|
|
from twisted.internet import reactor, defer
|
|
from lbrynet import conf
|
|
from lbrynet.extras.reflector.client.client import EncryptedFileReflectorClientFactory
|
|
from lbrynet.extras.reflector.client.blob import BlobReflectorClientFactory
|
|
|
|
|
|
def _is_ip(host):
|
|
try:
|
|
if len(host.split(".")) == 4 and all([0 <= int(x) <= 255 for x in host.split(".")]):
|
|
return True
|
|
return False
|
|
except ValueError:
|
|
return False
|
|
|
|
|
|
@defer.inlineCallbacks
|
|
def resolve(host):
|
|
if _is_ip(host):
|
|
ip = host
|
|
else:
|
|
ip = yield reactor.resolve(host)
|
|
defer.returnValue(ip)
|
|
|
|
|
|
@defer.inlineCallbacks
|
|
def _reflect_stream(blob_manager, stream_hash, sd_hash, reflector_server):
|
|
reflector_address, reflector_port = reflector_server[0], reflector_server[1]
|
|
factory = EncryptedFileReflectorClientFactory(blob_manager, stream_hash, sd_hash)
|
|
ip = yield resolve(reflector_address)
|
|
yield reactor.connectTCP(ip, reflector_port, factory)
|
|
result = yield factory.finished_deferred
|
|
defer.returnValue(result)
|
|
|
|
|
|
def _reflect_file(lbry_file, reflector_server):
|
|
return _reflect_stream(lbry_file.blob_manager, lbry_file.stream_hash, lbry_file.sd_hash, reflector_server)
|
|
|
|
|
|
@defer.inlineCallbacks
|
|
def _reflect_blobs(blob_manager, blob_hashes, reflector_server):
|
|
reflector_address, reflector_port = reflector_server[0], reflector_server[1]
|
|
factory = BlobReflectorClientFactory(blob_manager, blob_hashes)
|
|
ip = yield resolve(reflector_address)
|
|
yield reactor.connectTCP(ip, reflector_port, factory)
|
|
result = yield factory.finished_deferred
|
|
return result
|
|
|
|
|
|
def reflect_file(lbry_file, reflector_server=None):
|
|
if reflector_server:
|
|
if len(reflector_server.split(":")) == 2:
|
|
host, port = tuple(reflector_server.split(":"))
|
|
reflector_server = host, int(port)
|
|
else:
|
|
reflector_server = reflector_server, 5566
|
|
else:
|
|
reflector_server = random.choice(conf.settings['reflector_servers'])
|
|
return _reflect_file(lbry_file, reflector_server)
|
|
|
|
|
|
@defer.inlineCallbacks
|
|
def reflect_stream(blob_manager, stream_hash, reflector_server=None):
|
|
if reflector_server:
|
|
if len(reflector_server.split(":")) == 2:
|
|
host, port = tuple(reflector_server.split(":"))
|
|
reflector_server = host, int(port)
|
|
else:
|
|
reflector_server = reflector_server, 5566
|
|
else:
|
|
reflector_server = random.choice(conf.settings['reflector_servers'])
|
|
sd_hash = yield blob_manager.storage.get_sd_blob_hash_for_stream(stream_hash)
|
|
result = yield _reflect_stream(blob_manager, stream_hash, sd_hash, reflector_server)
|
|
defer.returnValue(result)
|
|
|
|
|
|
def reflect_blob_hashes(blob_hashes, blob_manager, reflector_server=None):
|
|
if reflector_server:
|
|
if len(reflector_server.split(":")) == 2:
|
|
host, port = tuple(reflector_server.split(":"))
|
|
reflector_server = host, int(port)
|
|
else:
|
|
reflector_server = reflector_server, 5566
|
|
else:
|
|
reflector_server = random.choice(conf.settings['reflector_servers'])
|
|
return _reflect_blobs(blob_manager, blob_hashes, reflector_server)
|