Viewing file:  DistUpgradeFetcherCore.py (7.94 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
# DistUpgradeFetcherCore.py
#
# Copyright (c) 2006 Canonical
#
# Author: Michael Vogt <michael.vogt@ubuntu.com>
#
# 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 2 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, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
# USA
from string import Template
import os
import apt_pkg
import apt
import tarfile
import socket
import urlparse
import urllib2
import tempfile
import shutil
import sys
import GnuPGInterface
from gettext import gettext as _
def country_mirror():
" helper to get the country mirror from the current locale "
# special cases go here
lang_mirror = { 'C' : '',
}
# no lang, no mirror
if not os.environ.has_key('LANG'):
return ''
lang = os.environ['LANG'].lower()
# check if it is a special case
if lang_mirror.has_key(lang[:5]):
return lang_mirror[lang[:5]]
# now check for the most comon form (en_US.UTF-8)
if "_" in lang:
country = lang.split(".")[0].split("_")[1]
if "@" in country:
country = country.split("@")[0]
return country+"."
else:
return lang[:2]+"."
return ''
class DistUpgradeFetcherCore(object):
" base class (without GUI) for the upgrade fetcher "
def __init__(self, new_dist, progress):
self.new_dist = new_dist
self._progress = progress
# options to pass to the release upgrader when it is run
self.run_options = []
def showReleaseNotes(self):
return True
def error(self, summary, message):
""" dummy implementation for error display, should be overwriten
by subclasses that want to more fancy method
"""
print summary
print message
return False
def authenticate(self):
if self.new_dist.upgradeToolSig:
f = self.tmpdir+"/"+os.path.basename(self.new_dist.upgradeTool)
sig = self.tmpdir+"/"+os.path.basename(self.new_dist.upgradeToolSig)
print "authenticate '%s' against '%s' " % (os.path.basename(f),os.path.basename(sig))
if not self.gpgauthenticate(f, sig):
return False
# we may return False here by default if we want to make a sig
# mandatory
return True
def gpgauthenticate(self, file, signature,
keyring='/etc/apt/trusted.gpg'):
""" authenticated a file against a given signature, if no keyring
is given use the apt default keyring
"""
gpg = GnuPGInterface.GnuPG()
gpg.options.extra_args = ['--no-options',
'--homedir',self.tmpdir,
'--no-default-keyring',
'--ignore-time-conflict',
'--keyring', keyring]
proc = gpg.run(['--verify', signature, file],
create_fhs=['status','logger','stderr'])
gpgres = proc.handles['status'].read()
try:
proc.wait()
except IOError,e:
# gnupg returned a problem (non-zero exit)
print "exception from gpg: %s" % e
print "Debug information: "
print proc.handles['status'].read()
print proc.handles['stderr'].read()
print proc.handles['logger'].read()
return False
if "VALIDSIG" in gpgres:
return True
print "invalid result from gpg:"
print gpgres
return False
def extractDistUpgrader(self):
# extract the tarbal
fname = os.path.join(self.tmpdir,os.path.basename(self.uri))
print "extracting '%s'" % os.path.basename(fname)
if not os.path.exists(fname):
return False
tar = tarfile.open(self.tmpdir+"/"+os.path.basename(self.uri),"r")
for tarinfo in tar:
tar.extract(tarinfo)
tar.close()
return True
def verifyDistUprader(self):
pass
# FIXME: check a internal dependency file to make sure
# that the script will run correctly
# see if we have a script file that we can run
self.script = script = "%s/%s" % (self.tmpdir, self.new_dist.name)
if not os.path.exists(script):
return error(_("Could not run the upgrade tool"),
_("This is most likely a bug in the upgrade tool. "
"Please report it as a bug"))
return True
def _expandUri(self, uri):
uri_template = Template(uri)
m = country_mirror()
new_uri = uri_template.safe_substitute(countrymirror=m)
# be paranoid and check if the given uri actually exists
host = urlparse.urlparse(new_uri)[1]
try:
socket.gethostbyname(host)
except socket.gaierror,e:
print >> sys.stderr, "host '%s' could not be resolved" % host
new_uri = uri_template.safe_substitute(countrymirror='')
return new_uri
def fetchDistUpgrader(self):
" download the tarball with the upgrade script "
self.tmpdir = tmpdir = tempfile.mkdtemp()
os.chdir(tmpdir)
# turn debugging on here (if required)
#apt_pkg.Config.Set("Debug::Acquire::http","1")
fetcher = apt_pkg.GetAcquire(self._progress)
if self.new_dist.upgradeToolSig != None:
uri = self._expandUri(self.new_dist.upgradeToolSig)
af = apt_pkg.GetPkgAcqFile(fetcher,uri, descr=_("Upgrade tool signature"))
if self.new_dist.upgradeTool != None:
self.uri = self._expandUri(self.new_dist.upgradeTool)
af = apt_pkg.GetPkgAcqFile(fetcher,self.uri, descr=_("Upgrade tool"))
if fetcher.Run() != fetcher.ResultContinue:
return False
return True
return False
def runDistUpgrader(self):
#print "runing: %s" % script
args = [self.script]+self.run_options
if os.getuid() != 0:
os.execv("/usr/bin/sudo",["sudo"]+args)
else:
os.execv(self.script,args)
def cleanup(self):
# cleanup
os.chdir("..")
# del tmpdir
shutil.rmtree(self.tmpdir)
def run(self):
# see if we have release notes
if not self.showReleaseNotes():
return
if not self.fetchDistUpgrader():
self.error(_("Failed to fetch"),
_("Fetching the upgrade failed. There may be a network "
"problem. "))
return
if not self.extractDistUpgrader():
self.error(_("Failed to extract"),
_("Extracting the upgrade failed. There may be a problem "
"with the network or with the server. "))
return
if not self.verifyDistUprader():
self.error(_("Verfication failed"),
_("Verifying the upgrade failed. There may be a problem "
"with the network or with the server. "))
self.cleanup()
return
if not self.authenticate():
self.error(_("Authentication failed"),
_("Authenticating the upgrade failed. There may be a problem "
"with the network or with the server. "))
self.cleanup()
return
self.runDistUpgrader()
if __name__ == "__main__":
self.error("summary","message")
d = DistUpgradeFetcher(None,None)
print d.authenticate('/tmp/Release','/tmp/Release.gpg')
|