X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=dak%2Fgenerate_releases.py;h=ba7976873c6ef0b853695450338ae56d92d13d57;hb=6f863f95e3d1bae18aa5579e958fd8a121b06545;hp=9de4614da4d97530678125ae0580a16c86529572;hpb=2d4f734d1d24765254687338a386be1e742f713b;p=dak.git diff --git a/dak/generate_releases.py b/dak/generate_releases.py index 9de4614d..ba797687 100755 --- a/dak/generate_releases.py +++ b/dak/generate_releases.py @@ -1,8 +1,14 @@ #!/usr/bin/env python -""" Create all the Release files """ +""" +Create all the Release files -# Copyright (C) 2001, 2002, 2006 Anthony Towns +@contact: Debian FTPMaster +@copyright: 2011 Joerg Jaspert +@copyright: 2011 Mark Hymers +@license: GNU General Public License version 2 or later + +""" # 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 @@ -18,367 +24,358 @@ # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA -# ``Bored now'' +################################################################################ + +# I wish they wouldnt leave biscuits out, thats just tempting. Damnit. ################################################################################ -import sys, os, stat, time -import gzip, bz2 +import sys +import os +import os.path +import stat +import time +import gzip +import bz2 import apt_pkg +import subprocess +from tempfile import mkstemp, mkdtemp +import commands +from sqlalchemy.orm import object_session -from daklib import utils +from daklib import utils, daklog +from daklib.regexes import re_gensubrelease, re_includeinrelease from daklib.dak_exceptions import * from daklib.dbconn import * +from daklib.config import Config +from daklib.dakmultiprocessing import DakProcessPool, PROC_STATUS_SUCCESS +import daklib.daksubprocess ################################################################################ - -Cnf = None -out = None -AptCnf = None +Logger = None #: Our logging object ################################################################################ def usage (exit_code=0): - print """Usage: dak generate-releases [OPTION]... [SUITE]... -Generate Release files (for SUITE). + """ Usage information""" - -h, --help show this help and exit - -a, --apt-conf FILE use FILE instead of default apt.conf - -f, --force-touch ignore Untouchable directives in dak.conf + print """Usage: dak generate-releases [OPTIONS] +Generate the Release files -If no SUITE is given Release files are generated for all suites.""" + -a, --archive=ARCHIVE process suites in ARCHIVE + -s, --suite=SUITE(s) process this suite + Default: All suites not marked 'untouchable' + -f, --force Allow processing of untouchable suites + CAREFUL: Only to be used at (point) release time! + -h, --help show this help and exit + -q, --quiet Don't output progress +SUITE can be a space seperated list, e.g. + --suite=unstable testing + """ sys.exit(exit_code) -################################################################################ +######################################################################## -def add_tiffani (files, path, indexstem): - index = "%s.diff/Index" % (indexstem) - filepath = "%s/%s" % (path, index) - if os.path.exists(filepath): - #print "ALERT: there was a tiffani file %s" % (filepath) - files.append(index) - -def gen_i18n_index (files, tree, sec): - path = Cnf["Dir::Root"] + tree + "/" - i18n_path = "%s/i18n" % (sec) - if os.path.exists("%s/%s" % (path, i18n_path)): - index = "%s/Index" % (i18n_path) - out = open("%s/%s" % (path, index), "w") - out.write("SHA1:\n") - for x in os.listdir("%s/%s" % (path, i18n_path)): - if x.startswith('Translation-'): - f = open("%s/%s/%s" % (path, i18n_path, x), "r") - size = os.fstat(f.fileno())[6] - f.seek(0) - sha1sum = apt_pkg.sha1sum(f) - f.close() - out.write(" %s %7d %s\n" % (sha1sum, size, x)) - out.close() - files.append(index) - -def compressnames (tree,type,file): - compress = AptCnf.get("%s::%s::Compress" % (tree,type), AptCnf.get("Default::%s::Compress" % (type), ". gzip")) - result = [] - cl = compress.split() - uncompress = ("." not in cl) - for mode in compress.split(): - if mode == ".": - result.append(file) - elif mode == "gzip": - if uncompress: - result.append("" + file) - uncompress = 0 - result.append(file + ".gz") - elif mode == "bzip2": - if uncompress: - result.append("" + file) - uncompress = 0 - result.append(file + ".bz2") - return result - -decompressors = { 'zcat' : gzip.GzipFile, - 'bzip2' : bz2.BZ2File } - -def print_md5sha_files (tree, files, hashop): - path = Cnf["Dir::Root"] + tree + "/" - for name in files: - hashvalue = "" - hashlen = 0 - try: - if name[0] == "<": - j = name.index("/") - k = name.index(">") - (cat, ext, name) = (name[1:j], name[j+1:k], name[k+1:]) - file_handle = decompressors[ cat ]( "%s%s%s" % (path, name, ext) ) - contents = file_handle.read() - hashvalue = hashop(contents) - hashlen = len(contents) - else: - try: - file_handle = utils.open_file(path + name) - hashvalue = hashop(file_handle) - hashlen = os.stat(path + name).st_size - except: - raise - else: - if file_handle: - file_handle.close() - - except CantOpenError: - print "ALERT: Couldn't open " + path + name - else: - out.write(" %s %8d %s\n" % (hashvalue, hashlen, name)) - -def print_md5_files (tree, files): - print_md5sha_files (tree, files, apt_pkg.md5sum) - -def print_sha1_files (tree, files): - print_md5sha_files (tree, files, apt_pkg.sha1sum) - -def print_sha256_files (tree, files): - print_md5sha_files (tree, files, apt_pkg.sha256sum) - -def write_release_file (relpath, suite, component, origin, label, arch, version="", suite_suffix="", notautomatic=""): - try: - if os.access(relpath, os.F_OK): - if os.stat(relpath).st_nlink > 1: - os.unlink(relpath) - release = open(relpath, "w") - except IOError: - utils.fubar("Couldn't write to " + relpath) - - release.write("Archive: %s\n" % (suite)) - if version != "": - release.write("Version: %s\n" % (version)) - - if suite_suffix: - release.write("Component: %s/%s\n" % (suite_suffix,component)) - else: - release.write("Component: %s\n" % (component)) +def sign_release_dir(suite, dirname): + cnf = Config() - release.write("Origin: %s\n" % (origin)) - release.write("Label: %s\n" % (label)) - if notautomatic != "": - release.write("NotAutomatic: %s\n" % (notautomatic)) - release.write("Architecture: %s\n" % (arch)) - release.close() + if cnf.has_key("Dinstall::SigningKeyring"): + keyring = "--secret-keyring \"%s\"" % cnf["Dinstall::SigningKeyring"] + if cnf.has_key("Dinstall::SigningPubKeyring"): + keyring += " --keyring \"%s\"" % cnf["Dinstall::SigningPubKeyring"] -################################################################################ + arguments = "--no-options --batch --no-tty --armour --personal-digest-preferences=SHA256" -def main (): - global Cnf, AptCnf, out - out = sys.stdout + relname = os.path.join(dirname, 'Release') - Cnf = utils.get_conf() + dest = os.path.join(dirname, 'Release.gpg') + if os.path.exists(dest): + os.unlink(dest) - Arguments = [('h',"help","Generate-Releases::Options::Help"), - ('a',"apt-conf","Generate-Releases::Options::Apt-Conf", "HasArg"), - ('f',"force-touch","Generate-Releases::Options::Force-Touch"), - ] - for i in [ "help", "apt-conf", "force-touch" ]: - if not Cnf.has_key("Generate-Releases::Options::%s" % (i)): - Cnf["Generate-Releases::Options::%s" % (i)] = "" + inlinedest = os.path.join(dirname, 'InRelease') + if os.path.exists(inlinedest): + os.unlink(inlinedest) - suites = apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv) - Options = Cnf.SubTree("Generate-Releases::Options") + defkeyid="" + for keyid in suite.signingkeys or []: + defkeyid += "--local-user %s " % keyid - if Options["Help"]: - usage() + os.system("gpg %s %s %s --detach-sign <%s >>%s" % + (keyring, defkeyid, arguments, relname, dest)) + os.system("gpg %s %s %s --clearsign <%s >>%s" % + (keyring, defkeyid, arguments, relname, inlinedest)) - if not Options["Apt-Conf"]: - Options["Apt-Conf"] = utils.which_apt_conf_file() +class XzFile(object): + def __init__(self, filename, mode='r'): + self.filename = filename + def read(self): + cmd = ("xz", "-d") + with open(self.filename, 'r') as stdin: + process = daklib.daksubprocess.Popen(cmd, stdin=stdin, stdout=subprocess.PIPE) + (stdout, stderr) = process.communicate() + return stdout - AptCnf = apt_pkg.newConfiguration() - apt_pkg.ReadConfigFileISC(AptCnf, Options["Apt-Conf"]) +class ReleaseWriter(object): + def __init__(self, suite): + self.suite = suite - if not suites: - suites = Cnf.SubTree("Suite").List() + def generate_release_files(self): + """ + Generate Release files for the given suite - for suitename in suites: - print "Processing: " + suitename - SuiteBlock = Cnf.SubTree("Suite::" + suitename) - suiteobj = get_suite(suitename.lower()) - if not suiteobj: - print "ALERT: Cannot find suite %s!" % (suitename.lower()) - continue + @type suite: string + @param suite: Suite name + """ - # Use the canonical name - suite = suiteobj.suite_name.lower() + suite = self.suite + session = object_session(suite) - if suiteobj.untouchable and not Options["Force-Touch"]: - print "Skipping: " + suite + " (untouchable)" - continue + architectures = get_suite_architectures(suite.suite_name, skipall=True, skipsrc=True, session=session) - origin = suiteobj.origin - label = suiteobj.label or suiteobj.origin - codename = suiteobj.codename or "" - version = "" - if suiteobj.version and suiteobj.version != '-': - version = suiteobj.version - description = suiteobj.description or "" - - architectures = get_suite_architectures(suite, skipall=True, skipsrc=True) - - if SuiteBlock.has_key("NotAutomatic"): - notautomatic = "yes" - else: - notautomatic = "" - - if SuiteBlock.has_key("Components"): - components = SuiteBlock.ValueList("Components") - else: - components = [] - - suite_suffix = Cnf.Find("Dinstall::SuiteSuffix") - if components and suite_suffix: - longsuite = suite + "/" + suite_suffix - else: - longsuite = suite - - tree = SuiteBlock.get("Tree", "dists/%s" % (longsuite)) - - if AptCnf.has_key("tree::%s" % (tree)): - pass - elif AptCnf.has_key("bindirectory::%s" % (tree)): - pass - else: - aptcnf_filename = os.path.basename(utils.which_apt_conf_file()) - print "ALERT: suite %s not in %s, nor untouchable!" % (suite, aptcnf_filename) - continue + # Attribs contains a tuple of field names and the database names to use to + # fill them in + attribs = ( ('Origin', 'origin'), + ('Label', 'label'), + ('Suite', 'release_suite_output'), + ('Version', 'version'), + ('Codename', 'codename') ) + + # A "Sub" Release file has slightly different fields + subattribs = ( ('Archive', 'suite_name'), + ('Origin', 'origin'), + ('Label', 'label'), + ('Version', 'version') ) + + # Boolean stuff. If we find it true in database, write out "yes" into the release file + boolattrs = ( ('NotAutomatic', 'notautomatic'), + ('ButAutomaticUpgrades', 'butautomaticupgrades') ) + + cnf = Config() - print Cnf["Dir::Root"] + tree + "/Release" - out = open(Cnf["Dir::Root"] + tree + "/Release", "w") + suite_suffix = cnf.find("Dinstall::SuiteSuffix", "") + + outfile = os.path.join(suite.archive.path, 'dists', suite.suite_name, suite_suffix, "Release") + out = open(outfile + ".new", "w") + + for key, dbfield in attribs: + # Hack to skip NULL Version fields as we used to do this + # We should probably just always ignore anything which is None + if key == "Version" and getattr(suite, dbfield) is None: + continue + + out.write("%s: %s\n" % (key, getattr(suite, dbfield))) - out.write("Origin: %s\n" % (suiteobj.origin)) - out.write("Label: %s\n" % (label)) - out.write("Suite: %s\n" % (suite)) - if version != "": - out.write("Version: %s\n" % (version)) - if codename != "": - out.write("Codename: %s\n" % (codename)) out.write("Date: %s\n" % (time.strftime("%a, %d %b %Y %H:%M:%S UTC", time.gmtime(time.time())))) - if SuiteBlock.has_key("ValidTime"): - validtime=float(SuiteBlock["ValidTime"]) + if suite.validtime: + validtime=float(suite.validtime) out.write("Valid-Until: %s\n" % (time.strftime("%a, %d %b %Y %H:%M:%S UTC", time.gmtime(time.time()+validtime)))) - if notautomatic != "": - out.write("NotAutomatic: %s\n" % (notautomatic)) + for key, dbfield in boolattrs: + if getattr(suite, dbfield, False): + out.write("%s: yes\n" % (key)) + out.write("Architectures: %s\n" % (" ".join([a.arch_string for a in architectures]))) - if components: - out.write("Components: %s\n" % (" ".join(components))) - if description: - out.write("Description: %s\n" % (description)) + components = [ c.component_name for c in suite.components ] - files = [] + out.write("Components: %s\n" % (" ".join(components))) - if AptCnf.has_key("tree::%s" % (tree)): - if AptCnf.has_key("tree::%s::Contents" % (tree)): - pass - else: - for x in os.listdir("%s/%s" % (Cnf["Dir::Root"], tree)): - if x.startswith('Contents-'): - if x.endswith('.diff'): - files.append("%s/Index" % (x)) - else: - files.append(x) - - for sec in AptCnf["tree::%s::Sections" % (tree)].split(): - for arch in AptCnf["tree::%s::Architectures" % (tree)].split(): - if arch == "source": - filepath = "%s/%s/Sources" % (sec, arch) - for cfile in compressnames("tree::%s" % (tree), "Sources", filepath): - files.append(cfile) - add_tiffani(files, Cnf["Dir::Root"] + tree, filepath) - else: - disks = "%s/disks-%s" % (sec, arch) - diskspath = Cnf["Dir::Root"]+tree+"/"+disks - if os.path.exists(diskspath): - for dir in os.listdir(diskspath): - if os.path.exists("%s/%s/md5sum.txt" % (diskspath, dir)): - files.append("%s/%s/md5sum.txt" % (disks, dir)) - - filepath = "%s/binary-%s/Packages" % (sec, arch) - for cfile in compressnames("tree::%s" % (tree), "Packages", filepath): - files.append(cfile) - add_tiffani(files, Cnf["Dir::Root"] + tree, filepath) - - if arch == "source": - rel = "%s/%s/Release" % (sec, arch) - else: - rel = "%s/binary-%s/Release" % (sec, arch) - relpath = Cnf["Dir::Root"]+tree+"/"+rel - write_release_file(relpath, suite, sec, origin, label, arch, version, suite_suffix, notautomatic) - files.append(rel) - gen_i18n_index(files, tree, sec) - - if AptCnf.has_key("tree::%s/main" % (tree)): - for dis in ["main", "contrib", "non-free"]: - if not AptCnf.has_key("tree::%s/%s" % (tree, dis)): continue - sec = AptCnf["tree::%s/%s::Sections" % (tree,dis)].split()[0] - if sec != "debian-installer": - print "ALERT: weird non debian-installer section in %s" % (tree) - - for arch in AptCnf["tree::%s/%s::Architectures" % (tree,dis)].split(): - if arch != "source": # always true - rel = "%s/%s/binary-%s/Release" % (dis, sec, arch) - relpath = Cnf["Dir::Root"]+tree+"/"+rel - write_release_file(relpath, suite, dis, origin, label, arch, version, suite_suffix, notautomatic) - files.append(rel) - for cfile in compressnames("tree::%s/%s" % (tree,dis), - "Packages", - "%s/%s/binary-%s/Packages" % (dis, sec, arch)): - files.append(cfile) - elif AptCnf.has_key("tree::%s::FakeDI" % (tree)): - usetree = AptCnf["tree::%s::FakeDI" % (tree)] - sec = AptCnf["tree::%s/main::Sections" % (usetree)].split()[0] - if sec != "debian-installer": - print "ALERT: weird non debian-installer section in %s" % (usetree) - - for arch in AptCnf["tree::%s/main::Architectures" % (usetree)].split(): - if arch != "source": # always true - for cfile in compressnames("tree::%s/main" % (usetree), "Packages", "main/%s/binary-%s/Packages" % (sec, arch)): - files.append(cfile) - - elif AptCnf.has_key("bindirectory::%s" % (tree)): - for cfile in compressnames("bindirectory::%s" % (tree), "Packages", AptCnf["bindirectory::%s::Packages" % (tree)]): - files.append(cfile.replace(tree+"/","",1)) - for cfile in compressnames("bindirectory::%s" % (tree), "Sources", AptCnf["bindirectory::%s::Sources" % (tree)]): - files.append(cfile.replace(tree+"/","",1)) - else: - print "ALERT: no tree/bindirectory for %s" % (tree) - - out.write("MD5Sum:\n") - print_md5_files(tree, files) - out.write("SHA1:\n") - print_sha1_files(tree, files) - out.write("SHA256:\n") - print_sha256_files(tree, files) + # For exact compatibility with old g-r, write out Description here instead + # of with the rest of the DB fields above + if getattr(suite, 'description') is not None: + out.write("Description: %s\n" % suite.description) + + for comp in components: + for dirpath, dirnames, filenames in os.walk(os.path.join(suite.archive.path, "dists", suite.suite_name, suite_suffix, comp), topdown=True): + if not re_gensubrelease.match(dirpath): + continue + + subfile = os.path.join(dirpath, "Release") + subrel = open(subfile + '.new', "w") + + for key, dbfield in subattribs: + if getattr(suite, dbfield) is not None: + subrel.write("%s: %s\n" % (key, getattr(suite, dbfield))) + + for key, dbfield in boolattrs: + if getattr(suite, dbfield, False): + subrel.write("%s: yes\n" % (key)) + + subrel.write("Component: %s%s\n" % (suite_suffix, comp)) + + # Urgh, but until we have all the suite/component/arch stuff in the DB, + # this'll have to do + arch = os.path.split(dirpath)[-1] + if arch.startswith('binary-'): + arch = arch[7:] + + subrel.write("Architecture: %s\n" % (arch)) + subrel.close() + + os.rename(subfile + '.new', subfile) + + # Now that we have done the groundwork, we want to get off and add the files with + # their checksums to the main Release file + oldcwd = os.getcwd() + + os.chdir(os.path.join(suite.archive.path, "dists", suite.suite_name, suite_suffix)) + + hashfuncs = { 'MD5Sum' : apt_pkg.md5sum, + 'SHA1' : apt_pkg.sha1sum, + 'SHA256' : apt_pkg.sha256sum } + + fileinfo = {} + + uncompnotseen = {} + + for dirpath, dirnames, filenames in os.walk(".", followlinks=True, topdown=True): + for entry in filenames: + # Skip things we don't want to include + if not re_includeinrelease.match(entry): + continue + + if dirpath == '.' and entry in ["Release", "Release.gpg", "InRelease"]: + continue + + filename = os.path.join(dirpath.lstrip('./'), entry) + fileinfo[filename] = {} + contents = open(filename, 'r').read() + + # If we find a file for which we have a compressed version and + # haven't yet seen the uncompressed one, store the possibility + # for future use + if entry.endswith(".gz") and entry[:-3] not in uncompnotseen.keys(): + uncompnotseen[filename[:-3]] = (gzip.GzipFile, filename) + elif entry.endswith(".bz2") and entry[:-4] not in uncompnotseen.keys(): + uncompnotseen[filename[:-4]] = (bz2.BZ2File, filename) + elif entry.endswith(".xz") and entry[:-3] not in uncompnotseen.keys(): + uncompnotseen[filename[:-3]] = (XzFile, filename) + + fileinfo[filename]['len'] = len(contents) + + for hf, func in hashfuncs.items(): + fileinfo[filename][hf] = func(contents) + + for filename, comp in uncompnotseen.items(): + # If we've already seen the uncompressed file, we don't + # need to do anything again + if filename in fileinfo.keys(): + continue + + # Skip uncompressed Contents files as they're huge, take ages to + # checksum and we checksum the compressed ones anyways + if os.path.basename(filename).startswith("Contents"): + continue + + fileinfo[filename] = {} + + # File handler is comp[0], filename of compressed file is comp[1] + contents = comp[0](comp[1], 'r').read() + + fileinfo[filename]['len'] = len(contents) + + for hf, func in hashfuncs.items(): + fileinfo[filename][hf] = func(contents) + + + for h in sorted(hashfuncs.keys()): + out.write('%s:\n' % h) + for filename in sorted(fileinfo.keys()): + out.write(" %s %8d %s\n" % (fileinfo[filename][h], fileinfo[filename]['len'], filename)) out.close() - if Cnf.has_key("Dinstall::SigningKeyring"): - keyring = "--secret-keyring \"%s\"" % Cnf["Dinstall::SigningKeyring"] - if Cnf.has_key("Dinstall::SigningPubKeyring"): - keyring += " --keyring \"%s\"" % Cnf["Dinstall::SigningPubKeyring"] - - arguments = "--no-options --batch --no-tty --armour" - if Cnf.has_key("Dinstall::SigningKeyIds"): - signkeyids = Cnf["Dinstall::SigningKeyIds"].split() + os.rename(outfile + '.new', outfile) + + sign_release_dir(suite, os.path.dirname(outfile)) + + os.chdir(oldcwd) + + return + + +def main (): + global Logger + + cnf = Config() + + for i in ["Help", "Suite", "Force", "Quiet"]: + if not cnf.has_key("Generate-Releases::Options::%s" % (i)): + cnf["Generate-Releases::Options::%s" % (i)] = "" + + Arguments = [('h',"help","Generate-Releases::Options::Help"), + ('a','archive','Generate-Releases::Options::Archive','HasArg'), + ('s',"suite","Generate-Releases::Options::Suite"), + ('f',"force","Generate-Releases::Options::Force"), + ('q',"quiet","Generate-Releases::Options::Quiet"), + ('o','option','','ArbItem')] + + suite_names = apt_pkg.parse_commandline(cnf.Cnf, Arguments, sys.argv) + Options = cnf.subtree("Generate-Releases::Options") + + if Options["Help"]: + usage() + + Logger = daklog.Logger('generate-releases') + pool = DakProcessPool() + + session = DBConn().session() + + if Options["Suite"]: + suites = [] + for s in suite_names: + suite = get_suite(s.lower(), session) + if suite: + suites.append(suite) else: - signkeyids = [""] - - dest = Cnf["Dir::Root"] + tree + "/Release.gpg" - if os.path.exists(dest): - os.unlink(dest) - - for keyid in signkeyids: - if keyid != "": defkeyid = "--default-key %s" % keyid - else: defkeyid = "" - os.system("gpg %s %s %s --detach-sign <%s >>%s" % - (keyring, defkeyid, arguments, - Cnf["Dir::Root"] + tree + "/Release", dest)) + print "cannot find suite %s" % s + Logger.log(['cannot find suite %s' % s]) + else: + query = session.query(Suite).filter(Suite.untouchable == False) + if 'Archive' in Options: + query = query.join(Suite.archive).filter(Archive.archive_name==Options['Archive']) + suites = query.all() + + broken=[] + + for s in suites: + # Setup a multiprocessing Pool. As many workers as we have CPU cores. + if s.untouchable and not Options["Force"]: + print "Skipping %s (untouchable)" % s.suite_name + continue + + if not Options["Quiet"]: + print "Processing %s" % s.suite_name + Logger.log(['Processing release file for Suite: %s' % (s.suite_name)]) + pool.apply_async(generate_helper, (s.suite_id, )) + + # No more work will be added to our pool, close it and then wait for all to finish + pool.close() + pool.join() + + retcode = pool.overall_status() + + if retcode > 0: + # TODO: CENTRAL FUNCTION FOR THIS / IMPROVE LOGGING + Logger.log(['Release file generation broken: %s' % (','.join([str(x[1]) for x in pool.results]))]) + + Logger.close() + + sys.exit(retcode) + +def generate_helper(suite_id): + ''' + This function is called in a new subprocess. + ''' + session = DBConn().session() + suite = Suite.get(suite_id, session) + + # We allow the process handler to catch and deal with any exceptions + rw = ReleaseWriter(suite) + rw.generate_release_files() + + return (PROC_STATUS_SUCCESS, 'Release file written for %s' % suite.suite_name) #######################################################################################