def _get_torrent(self, installer_url):
"""Get the .torrent file from the version site."""
torrentfile = None
try:
torrentfile = GetTorrent.get_url(installer_url)
except GetTorrent.GetTorrentException, e:
self.debug('_get_torrent() run#%d: failed to download torrent file %s: %s' %
(self.runs, installer_url, str_exc(e)))
pass
def _get_torrent(self, installer_url):
"""Get the .torrent file from the version site."""
torrentfile = None
try:
torrentfile = GetTorrent.get_url(installer_url)
except GetTorrent.GetTorrentException, e:
self.debug(
'_get_torrent() run#%d: failed to download torrent file %s: %s'
% (self.runs, installer_url, str_exc(e)))
pass
class Updater(object):
def __init__(self,
threadwrap,
newversionfunc,
startfunc,
installfunc,
errorfunc,
test_new_version='',
test_current_version=''):
self.threadwrap = threadwrap # for calling back to UI from thread
self.newversionfunc = newversionfunc # alert to new version UI function
self.startfunc = startfunc # start torrent UI function
self.installfunc = installfunc # install torrent UI function
self.errorfunc = errorfunc # report error UI function
self.infohash = None
self.version = currentversion
self.currentversion = currentversion
self.asked_for_install = False
self.version_site = version_host
if os.name == 'nt':
self.version_site += 'win32/'
if os_version not in ('XP', '2000', '2003'):
self.version_site += 'legacy/'
elif osx:
self.version_site += 'osx/'
self.debug_mode = DEBUG
if test_new_version:
test_new_version = Version.from_str(test_new_version)
self.debug_mode = True
def _hack_get_available(url):
return test_new_version
self._get_available = _hack_get_available
if test_current_version:
self.debug_mode = True
self.currentversion = Version.from_str(test_current_version)
def debug(self, message):
if self.debug_mode:
self.threadwrap(self.errorfunc, WARNING, message)
def _get_available(self, url):
self.debug('Updater.get_available() hitting url %s' % url)
try:
u = zurllib.urlopen(url)
s = u.read()
s = s.strip()
except:
raise BTFailure(_("Could not get latest version from %s") % url)
try:
assert len(s) == 5
availableversion = Version.from_str(s)
except:
raise BTFailure(
_("Could not parse new version string from %s") % url)
return availableversion
def get_available(self):
url = self.version_site + self.currentversion.name()
availableversion = self._get_available(url)
if availableversion.is_beta():
if availableversion[1] != self.currentversion[1]:
availableversion = self.currentversion
if self.currentversion.is_beta():
stable_url = self.version_site + 'stable'
available_stable_version = self._get_available(stable_url)
if available_stable_version > availableversion:
availableversion = available_stable_version
self.version = availableversion
self.debug('Updater.get_available() got %s' % str(self.version))
return self.version
def get(self):
try:
self.get_available()
except BTFailure, e:
self.threadwrap(self.errorfunc, WARNING, e)
return
if self.version <= self.currentversion:
self.debug('Updater.get() not updating old version %s' %
str(self.version))
return
if not self.can_install():
self.debug('Updater.get() cannot install on this os')
return
self.installer_name = self.calc_installer_name()
self.installer_url = self.version_site + self.installer_name + '.torrent'
self.installer_dir = self.calc_installer_dir()
self.torrentfile = None
torrentfile, terrors = GetTorrent.get_url(self.installer_url)
signature = None
try:
signfile = zurllib.urlopen(self.installer_url + '.sign')
except:
self.debug('Updater.get() failed to get signfile %s.sign' %
self.installer_url)
else:
try:
signature = pickle.load(signfile)
except:
self.debug('Updater.get() failed to load signfile %s' %
signfile)
if terrors:
self.threadwrap(self.errorfunc, WARNING, '\n'.join(terrors))
if torrentfile and signature:
public_key_file = open(os.path.join(doc_root, 'public.key'), 'rb')
public_key = pickle.load(public_key_file)
h = sha(torrentfile).digest()
if public_key.verify(h, signature):
self.torrentfile = torrentfile
b = bdecode(torrentfile)
self.infohash = sha(bencode(b['info'])).digest()
self.total_size = b['info']['length']
self.debug('Updater.get() got torrent file and signature')
else:
self.debug(
'Updater.get() torrent file signature failed to verify.')
pass
else:
self.debug(
'Updater.get() doesn\'t have torrentfile %s and signature %s' %
(str(type(torrentfile)), str(type(signature))))
