def run(scrwin, errlist, params):
doneflag = threading.Event()
d = CursesDisplayer(scrwin, errlist, doneflag)
try:
while 1:
configdir = ConfigDir('downloadcurses')
defaultsToIgnore = ['responsefile', 'url', 'priority']
configdir.setDefaults(defaults, defaultsToIgnore)
configdefaults = configdir.loadConfig()
defaults.append(
('save_options', 0, 'whether to save the current options as '
'the new default configuration (only for btdownloadcurses.py)'
))
try:
config = parse_params(params, configdefaults)
except ValueError as e:
d.error('error: {}\nrun with no args for parameter '
'explanations'.format(e))
break
if not config:
d.error(get_usage(defaults, d.fieldw, configdefaults))
break
if config['save_options']:
configdir.saveConfig(config)
configdir.deleteOldCacheData(config['expire_cache_data'])
myid = createPeerID()
random.seed(myid)
rawserver = RawServer(doneflag,
config['timeout_check_interval'],
config['timeout'],
ipv6_enable=config['ipv6_enabled'],
failfunc=d.failed,
errorfunc=d.error)
upnp_type = UPnP_test(config['upnp_nat_access'])
while True:
try:
listen_port = rawserver.find_and_bind(
config['minport'],
config['maxport'],
config['bind'],
ipv6_socket_style=config['ipv6_binds_v4'],
upnp=upnp_type,
randomizer=config['random_port'])
break
except socket.error as e:
if upnp_type and e == UPnP_ERROR:
d.error('WARNING: COULD NOT FORWARD VIA UPnP')
upnp_type = 0
continue
d.error("Couldn't listen - " + str(e))
d.failed()
return
response = get_response(config['responsefile'], config['url'],
d.error)
if not response:
break
infohash = hashlib.sha1(bencode(response['info'])).digest()
dow = BT1Download(d.display, d.finished, d.error, d.error,
doneflag, config, response, infohash, myid,
rawserver, listen_port, configdir)
if not dow.saveAs(d.chooseFile):
break
if not dow.initFiles(old_style=True):
break
if not dow.startEngine():
dow.shutdown()
break
dow.startRerequester()
dow.autoStats()
if not dow.am_I_finished():
d.display(activity='connecting to peers')
rawserver.listen_forever(dow.getPortHandler())
d.display(activity='shutting down')
dow.shutdown()
break
except KeyboardInterrupt:
# ^C to exit...
pass
try:
rawserver.shutdown()
except:
pass
if not d.done:
d.failed()
def run(params):
h = HeadlessDisplayer()
while 1:
configdir = ConfigDir('downloadheadless')
defaultsToIgnore = ['responsefile', 'url', 'priority']
configdir.setDefaults(defaults, defaultsToIgnore)
configdefaults = configdir.loadConfig()
defaults.append(
('save_options', 0, 'whether to save the current options as the '
'new default configuration (only for btdownloadheadless.py)'))
try:
config = parse_params(params, configdefaults)
except ValueError as e:
print 'error: {}\nrun with no args for parameter explanations' \
''.format(e)
break
if not config:
print get_usage(defaults, 80, configdefaults)
break
if config['save_options']:
configdir.saveConfig(config)
configdir.deleteOldCacheData(config['expire_cache_data'])
myid = createPeerID()
random.seed(myid)
doneflag = threading.Event()
def disp_exception(text):
print text
rawserver = RawServer(
doneflag, config['timeout_check_interval'], config['timeout'],
ipv6_enable=config['ipv6_enabled'], failfunc=h.failed,
errorfunc=disp_exception)
upnp_type = UPnP_test(config['upnp_nat_access'])
while True:
try:
listen_port = rawserver.find_and_bind(
config['minport'], config['maxport'], config['bind'],
ipv6_socket_style=config['ipv6_binds_v4'],
upnp=upnp_type, randomizer=config['random_port'])
break
except socket.error as e:
if upnp_type and e == UPnP_ERROR:
print 'WARNING: COULD NOT FORWARD VIA UPnP'
upnp_type = 0
continue
print "error: Couldn't listen - " + str(e)
h.failed()
return
response = get_response(config['responsefile'], config['url'], h.error)
if not response:
break
infohash = hashlib.sha1(bencode(response['info'])).digest()
dow = BT1Download(
h.display, h.finished, h.error, disp_exception, doneflag, config,
response, infohash, myid, rawserver, listen_port, configdir)
if not dow.saveAs(h.chooseFile, h.newpath):
break
if not dow.initFiles(old_style=True):
break
if not dow.startEngine():
dow.shutdown()
break
dow.startRerequester()
dow.autoStats()
if not dow.am_I_finished():
h.display(activity='connecting to peers')
rawserver.listen_forever(dow.getPortHandler())
h.display(activity='shutting down')
dow.shutdown()
break
try:
rawserver.shutdown()
except:
pass
if not h.done:
h.failed()
def start(self):
rawserver = None
try:
config = dict([(v[0], v[1]) for v in defaults])
config['ip'] = self.ip
config['responsefile'] = self.torrent_file
config['saveas'] = self.destination_file
random.seed(self.id)
rawserver = RawServer(
doneflag=self.done_flag,
timeout_check_interval=config['timeout_check_interval'],
timeout=config['timeout'],
ipv6_enable=config['ipv6_enabled'],
failfunc=self.on_fail,
errorfunc=self.on_exception)
rawserver.bind(
port=self.port,
bind=config['bind'],
reuse=True,
ipv6_socket_style=config['ipv6_binds_v4'])
# Download torrent metadata.
response = get_response(
file=config['responsefile'],
url=config['url'],
errorfunc=self.on_error)
# Bail if tracker is done.
if not response:
return
dow = BT1Download(
statusfunc=self.on_status,
finfunc=self.on_finish,
errorfunc=self.on_error,
excfunc=self.on_exception,
doneflag=self.done_flag,
config=config,
response=response,
infohash=hashlib.sha1(bencode(response['info'])).digest(),
id=self.id,
rawserver=rawserver,
port=self.port)
if not dow.saveAs(lambda default, size, saveas, dir: saveas if saveas else default):
return
if not dow.initFiles(old_style=True):
return
if not dow.startEngine():
dow.shutdown()
return
dow.startRerequester()
dow.autoStats()
if not dow.am_I_finished():
self.on_status(activity='connecting to peers')
rawserver.listen_forever(dow.getPortHandler())
self.on_status(activity='shutting down')
dow.shutdown()
finally:
if rawserver:
rawserver.shutdown()
if not self.finished_at:
self.on_fail()
def start(self):
rawserver = None
try:
config = dict([(v[0], v[1]) for v in defaults])
config['ip'] = self.ip
config['responsefile'] = self.torrent_file
config['saveas'] = self.destination_file
random.seed(self.id)
rawserver = RawServer(
doneflag=self.done_flag,
timeout_check_interval=config['timeout_check_interval'],
timeout=config['timeout'],
ipv6_enable=config['ipv6_enabled'],
failfunc=self.on_fail,
errorfunc=self.on_exception)
rawserver.bind(port=self.port,
bind=config['bind'],
reuse=True,
ipv6_socket_style=config['ipv6_binds_v4'])
# Download torrent metadata.
response = get_response(file=config['responsefile'],
url=config['url'],
errorfunc=self.on_error)
# Bail if tracker is done.
if not response:
return
dow = BT1Download(statusfunc=self.on_status,
finfunc=self.on_finish,
errorfunc=self.on_error,
excfunc=self.on_exception,
doneflag=self.done_flag,
config=config,
response=response,
infohash=hashlib.sha1(bencode(
response['info'])).digest(),
id=self.id,
rawserver=rawserver,
port=self.port)
if not dow.saveAs(lambda default, size, saveas, dir: saveas
if saveas else default):
return
if not dow.initFiles(old_style=True):
return
if not dow.startEngine():
dow.shutdown()
return
dow.startRerequester()
dow.autoStats()
if not dow.am_I_finished():
self.on_status(activity='connecting to peers')
rawserver.listen_forever(dow.getPortHandler())
self.on_status(activity='shutting down')
dow.shutdown()
finally:
if rawserver:
rawserver.shutdown()
if not self.finished_at:
self.on_fail()
class LaunchMany(object):
def __init__(self, configdir, config, Output):
try:
self.configdir = configdir
self.config = config
self.Output = Output
self.torrent_dir = config['torrent_dir']
self.torrent_cache = {}
self.file_cache = {}
self.blocked_files = {}
self.scan_period = config['parse_dir_interval']
self.stats_period = config['display_interval']
if not _natpmp_supported:
config["natpmp"] = 0
self.torrent_list = []
self.downloads = {}
self.counter = 0
self.doneflag = Event()
self.hashcheck_queue = []
self.hashcheck_current = None
# BT RawServer
self.rawserver = RawServer(self.doneflag, config['timeout_check_interval'],
config['timeout'], ipv6_enable = config['ipv6_enabled'],
failfunc = self.failed, errorfunc = self.exchandler)
self.upnp_type = 0
self.listen_port = self.getPort(config)
# Rate limiter
self.ratelimiter = RateLimiter(self.rawserver.add_task,
config['upload_unit_size'],
config['average_ping'])
self.ratelimiter.set_upload_rate(config['max_upload_rate'])
# Multi Handler
self.handler = MultiHandler(self.rawserver, self.doneflag, config)
self.rawserver.add_task(self.scan, 0)
self.rawserver.add_task(self.stats, 0)
# DHT
if config["dht"] and self.listen_port:
self.dht = Khashmir(config['bind'], self.listen_port, self.configdir.getDatabaseDir(),
ipv6_enable = config['ipv6_enabled'],
upnp = self.upnp_type, natpmp = config['natpmp'])
self.dht.start()
else:
self.dht = None
OPTION_PATTERN[63] = False
# Run
self.handler.listen_forever()
# Shutdown
self.Output.message('shutting down')
self.hashcheck_queue = []
for hash in self.torrent_list:
self.Output.message('dropped "'+self.torrent_cache[hash]['path']+'"')
self.downloads[hash].shutdown()
self.rawserver.shutdown()
if self.dht:
self.dht.close()
except IOError:
data = StringIO()
print_exc(file = data)
self.Output.exception(data.getvalue())
def getPort(self, config):
listen_port = None
self.upnp_type = UPnP_test(config['upnp_nat_access'])
while True:
try:
listen_port = self.rawserver.find_and_bind(config['minport'],
config['maxport'],
config['bind'],
ipv6_socket_style = config['ipv6_binds_v4'],
upnp = self.upnp_type,
randomizer = False,
natpmp = config['natpmp'])
self.listen_port = listen_port
break
except socketerror, e:
self.failed(_('listening port not available'))
break
return listen_port
