--- /dev/null
+#!/usr/bin/env python
+
+###########################################################
+# generates partial package updates list
+
+# idea and basic implementation by Anthony, some changes by Andreas
+# parts are stolen from ziyi
+#
+# Copyright (C) 2004-5 Anthony Towns <aj@azure.humbug.org.au>
+# Copyright (C) 2004-5 Andreas Barth <aba@not.so.argh.org>
+
+# 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
+
+
+# < elmo> bah, don't bother me with annoying facts
+# < elmo> I was on a roll
+
+
+################################################################################
+
+import sys, os, tempfile
+import apt_pkg
+import utils
+
+################################################################################
+
+projectB = None;
+Cnf = None;
+Logger = None;
+Options = None;
+
+################################################################################
+
+def usage (exit_code=0):
+ print """Usage: tiffani [OPTIONS] [suites]
+Write out ed-style diffs to Packages/Source lists
+
+ -h, --help show this help and exit
+ -c give the canonical path of the file
+ -p name for the patch (defaults to current time)
+ -n take no action
+ """
+ sys.exit(exit_code);
+
+
+def tryunlink(file):
+ try:
+ os.unlink(file)
+ except OSError:
+ print "warning: removing of %s denied" % (file)
+
+def smartstat(file):
+ for ext in ["", ".gz", ".bz2"]:
+ if os.path.isfile(file + ext):
+ return (ext, os.stat(file + ext))
+ return (None, None)
+
+def smartlink(f, t):
+ if os.path.isfile(f):
+ os.link(f,t)
+ elif os.path.isfile("%s.gz" % (f)):
+ os.system("gzip -d < %s.gz > %s" % (f, t))
+ elif os.path.isfile("%s.bz2" % (f)):
+ os.system("bzip2 -d < %s.bz2 > %s" % (f, t))
+ else:
+ print "missing: %s" % (f)
+ raise IOError, f
+
+def smartopen(file):
+ if os.path.isfile(file):
+ f = open(file, "r")
+ elif os.path.isfile("%s.gz" % file):
+ f = create_temp_file(os.popen("zcat %s.gz" % file, "r"))
+ elif os.path.isfile("%s.bz2" % file):
+ f = create_temp_file(os.popen("bzcat %s.bz2" % file, "r"))
+ else:
+ f = None
+ return f
+
+def pipe_file(f, t):
+ f.seek(0)
+ while 1:
+ l = f.read()
+ if not l: break
+ t.write(l)
+ t.close()
+
+class Updates:
+ def __init__(self, readpath = None, max = 14):
+ self.can_path = None
+ self.history = {}
+ self.history_order = []
+ self.max = max
+ self.readpath = readpath
+ self.filesizesha1 = None
+
+ if readpath:
+ try:
+ f = open(readpath + "/Index")
+ x = f.readline()
+
+ def read_hashs(ind, f, self, x=x):
+ while 1:
+ x = f.readline()
+ if not x or x[0] != " ": break
+ l = x.split()
+ if not self.history.has_key(l[2]):
+ self.history[l[2]] = [None,None]
+ self.history_order.append(l[2])
+ self.history[l[2]][ind] = (l[0], int(l[1]))
+ return x
+
+ while x:
+ l = x.split()
+
+ if len(l) == 0:
+ x = f.readline()
+ continue
+
+ if l[0] == "SHA1-History:":
+ x = read_hashs(0,f,self)
+ continue
+
+ if l[0] == "SHA1-Patches:":
+ x = read_hashs(1,f,self)
+ continue
+
+ if l[0] == "Canonical-Name:" or l[0]=="Canonical-Path:":
+ self.can_path = l[1]
+
+ if l[0] == "SHA1-Current:" and len(l) == 3:
+ self.filesizesha1 = (l[1], int(l[2]))
+
+ x = f.readline()
+
+ except IOError:
+ 0
+
+ def dump(self, out=sys.stdout):
+ if self.can_path:
+ out.write("Canonical-Path: %s\n" % (self.can_path))
+
+ if self.filesizesha1:
+ out.write("SHA1-Current: %s %7d\n" % (self.filesizesha1))
+
+ hs = self.history
+ l = self.history_order[:]
+
+ cnt = len(l)
+ if cnt > self.max:
+ for h in l[:cnt-self.max]:
+ tryunlink("%s/%s.gz" % (self.readpath, h))
+ del hs[h]
+ l = l[cnt-self.max:]
+ self.history_order = l[:]
+
+ out.write("SHA1-History:\n")
+ for h in l:
+ out.write(" %s %7d %s\n" % (hs[h][0][0], hs[h][0][1], h))
+ out.write("SHA1-Patches:\n")
+ for h in l:
+ out.write(" %s %7d %s\n" % (hs[h][1][0], hs[h][1][1], h))
+
+def create_temp_file(r):
+ f = tempfile.TemporaryFile()
+ while 1:
+ x = r.readline()
+ if not x: break
+ f.write(x)
+ r.close()
+ del x,r
+ f.flush()
+ f.seek(0)
+ return f
+
+def sizesha1(f):
+ size = os.fstat(f.fileno())[6]
+ f.seek(0)
+ sha1sum = apt_pkg.sha1sum(f)
+ return (sha1sum, size)
+
+def genchanges(Options, outdir, oldfile, origfile, maxdiffs = 14):
+ if Options.has_key("NoAct"):
+ print "not doing anything"
+ return
+
+ patchname = Options["PatchName"]
+
+ # origfile = /path/to/Packages
+ # oldfile = ./Packages
+ # newfile = ./Packages.tmp
+ # difffile = outdir/patchname
+ # index => outdir/Index
+
+ # (outdir, oldfile, origfile) = argv
+
+ newfile = oldfile + ".new"
+ difffile = "%s/%s" % (outdir, patchname)
+
+ upd = Updates(outdir, int(maxdiffs))
+ (oldext, oldstat) = smartstat(oldfile)
+ (origext, origstat) = smartstat(origfile)
+ if not origstat:
+ print "%s doesn't exist" % (origfile)
+ return
+ if not oldstat:
+ print "initial run"
+ os.link(origfile + origext, oldfile + origext)
+ return
+
+ if oldstat[1:3] == origstat[1:3]:
+ print "hardlink unbroken, assuming unchanged"
+ return
+
+ oldf = smartopen(oldfile)
+ oldsizesha1 = sizesha1(oldf)
+
+ # should probably early exit if either of these checks fail
+ # alternatively (optionally?) could just trim the patch history
+
+ if upd.filesizesha1:
+ if upd.filesizesha1 != oldsizesha1:
+ print "old file seems to have changed! %s %s => %s %s" % (upd.filesizesha1 + oldsizesha1)
+
+ # XXX this should be usable now
+ #
+ #for d in upd.history.keys():
+ # df = smartopen("%s/%s" % (outdir,d))
+ # act_sha1size = sizesha1(df)
+ # df.close()
+ # exp_sha1size = upd.history[d][1]
+ # if act_sha1size != exp_sha1size:
+ # print "patch file %s seems to have changed! %s %s => %s %s" % \
+ # (d,) + exp_sha1size + act_sha1size
+
+ if Options.has_key("CanonicalPath"): upd.can_path=Options["CanonicalPath"]
+
+ if os.path.exists(newfile): os.unlink(newfile)
+ smartlink(origfile, newfile)
+ newf = open(newfile, "r")
+ newsizesha1 = sizesha1(newf)
+ newf.close()
+
+ if newsizesha1 == oldsizesha1:
+ os.unlink(newfile)
+ oldf.close()
+ print "file unchanged, not generating diff"
+ else:
+ if not os.path.isdir(outdir): os.mkdir(outdir)
+ print "generating diff"
+ w = os.popen("diff --ed - %s | gzip -c -9 > %s.gz" %
+ (newfile, difffile), "w")
+ pipe_file(oldf, w)
+ oldf.close()
+
+ difff = smartopen(difffile)
+ difsizesha1 = sizesha1(difff)
+ difff.close()
+
+ upd.history[patchname] = (oldsizesha1, difsizesha1)
+ upd.history_order.append(patchname)
+
+ upd.filesizesha1 = newsizesha1
+
+ os.unlink(oldfile + oldext)
+ os.link(origfile + origext, oldfile + origext)
+ os.unlink(newfile)
+
+ f = open(outdir + "/Index", "w")
+ upd.dump(f)
+ f.close()
+
+
+def main():
+ global Cnf, Options, Logger
+
+ os.umask(0002);
+
+ Cnf = utils.get_conf();
+ Arguments = [ ('h', "help", "Tiffani::Options::Help"),
+ ('c', None, "Tiffani::Options::CanonicalPath", "hasArg"),
+ ('p', "patchname", "Tiffani::Options::PatchName", "hasArg"),
+ ('r', "rootdir", "Tiffani::Options::RootDir", "hasArg"),
+ ('d', "tmpdir", "Tiffani::Options::TempDir", "hasArg"),
+ ('m', "maxdiffs", "Tiffani::Options::MaxDiffs", "hasArg"),
+ ('n', "n-act", "Tiffani::Options::NoAct"),
+ ];
+ suites = apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv);
+ Options = Cnf.SubTree("Tiffani::Options");
+ if Options.has_key("Help"): usage();
+
+ maxdiffs = Options.get("MaxDiffs::Default", "14")
+ maxpackages = Options.get("MaxDiffs::Packages", maxdiffs)
+ maxcontents = Options.get("MaxDiffs::Contents", maxdiffs)
+ maxsources = Options.get("MaxDiffs::Sources", maxdiffs)
+
+ if not Options.has_key("PatchName"):
+ format = "%Y-%m-%d-%H%M.%S"
+ i,o = os.popen2("date +%s" % (format))
+ i.close()
+ Options["PatchName"] = o.readline()[:-1]
+ o.close()
+
+ AptCnf = apt_pkg.newConfiguration()
+ apt_pkg.ReadConfigFileISC(AptCnf,utils.which_apt_conf_file())
+
+ if Options.has_key("RootDir"): Cnf["Dir::Root"] = Options["RootDir"]
+
+ if not suites:
+ suites = Cnf.SubTree("Suite").List()
+
+ for suite in suites:
+ if suite == "Experimental": continue
+
+ print "Processing: " + suite
+ SuiteBlock = Cnf.SubTree("Suite::" + suite)
+
+ if SuiteBlock.has_key("Untouchable"):
+ print "Skipping: " + suite + " (untouchable)"
+ continue
+
+ suite = suite.lower()
+
+ architectures = SuiteBlock.ValueList("Architectures")
+
+ 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)):
+ sections = AptCnf["tree::%s::Sections" % (tree)].split()
+ elif AptCnf.has_key("bindirectory::%s" % (tree)):
+ sections = AptCnf["bindirectory::%s::Sections" % (tree)].split()
+ else:
+ aptcnf_filename = os.path.basename(utils.which_apt_conf_file());
+ print "ALERT: suite %s not in %s, nor untouchable!" % (suite, aptcnf_filename);
+ continue
+
+ for architecture in architectures:
+ if architecture == "all":
+ continue
+
+ if architecture != "source":
+ # Process Contents
+ file = "%s/Contents-%s" % (Cnf["Dir::Root"] + tree,
+ architecture)
+ storename = "%s/%s_contents_%s" % (Options["TempDir"], suite, architecture)
+ print "running contents for %s %s : " % (suite, architecture),
+ genchanges(Options, file + ".diff", storename, file, \
+ Cnf.get("Suite::%s::Tiffani::MaxDiffs::Contents" % (suite), maxcontents))
+
+ # use sections instead of components since katie.conf
+ # treats "foo/bar main" as suite "foo", suitesuffix "bar" and
+ # component "bar/main". suck.
+
+ for component in sections:
+ if architecture == "source":
+ longarch = architecture
+ packages = "Sources"
+ maxsuite = maxsources
+ else:
+ longarch = "binary-%s"% (architecture)
+ packages = "Packages"
+ maxsuite = maxpackages
+
+ file = "%s/%s/%s/%s" % (Cnf["Dir::Root"] + tree,
+ component, longarch, packages)
+ storename = "%s/%s_%s_%s" % (Options["TempDir"], suite, component, architecture)
+ print "running for %s %s %s : " % (suite, component, architecture),
+ genchanges(Options, file + ".diff", storename, file, \
+ Cnf.get("Suite::%s::Tiffani::MaxDiffs::%s" % (suite, packages), maxsuite))
+
+################################################################################
+
+if __name__ == '__main__':
+ main()