dak/check_archive.py

   1 #!/usr/bin/env python
   2
   3 """ Various different sanity checks
   4
   5 @contact: Debian FTP Master <ftpmaster@debian.org>
   6 @copyright: (C) 2000, 2001, 2002, 2003, 2004, 2006  James Troup <james@nocrew.org>
   7 @license: GNU General Public License version 2 or later
   8 """
   9
  10 # This program is free software; you can redistribute it and/or modify
  11 # it under the terms of the GNU General Public License as published by
  12 # the Free Software Foundation; either version 2 of the License, or
  13 # (at your option) any later version.
  14
  15 # This program is distributed in the hope that it will be useful,
  16 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  17 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18 # GNU General Public License for more details.
  19
  20 # You should have received a copy of the GNU General Public License
  21 # along with this program; if not, write to the Free Software
  22 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  23
  24 ################################################################################
  25
  26 #   And, lo, a great and menacing voice rose from the depths, and with
  27 #   great wrath and vehemence it's voice boomed across the
  28 #   land... ``hehehehehehe... that *tickles*''
  29 #                                                       -- aj on IRC
  30
  31 ################################################################################
  32
  33 import commands
  34 import os
  35 import stat
  36 import sys
  37 import time
  38 import apt_pkg
  39 import apt_inst
  40
  41 from daklib.dbconn import *
  42 from daklib import utils
  43 from daklib.config import Config
  44
  45 ################################################################################
  46
  47 db_files = {}                  #: Cache of filenames as known by the database
  48 waste = 0.0                    #: How many bytes are "wasted" by files not referenced in database
  49 excluded = {}                  #: List of files which are excluded from files check
  50 current_file = None
  51 future_files = {}
  52 current_time = time.time()     #: now()
  53
  54 ################################################################################
  55
  56 def usage(exit_code=0):
  57     print """Usage: dak check-archive MODE
  58 Run various sanity checks of the archive and/or database.
  59
  60   -h, --help                show this help and exit.
  61
  62 The following MODEs are available:
  63
  64   checksums          - validate the checksums stored in the database
  65   files              - check files in the database against what's in the archive
  66   dsc-syntax         - validate the syntax of .dsc files in the archive
  67   missing-overrides  - check for missing overrides
  68   source-in-one-dir  - ensure the source for each package is in one directory
  69   timestamps         - check for future timestamps in .deb's
  70   files-in-dsc       - ensure each .dsc references appropriate Files
  71   validate-indices   - ensure files mentioned in Packages & Sources exist
  72   files-not-symlinks - check files in the database aren't symlinks
  73   validate-builddeps - validate build-dependencies of .dsc files in the archive
  74 """
  75     sys.exit(exit_code)
  76
  77 ################################################################################
  78
  79 def process_dir (unused, dirname, filenames):
  80     """
  81     Process a directory and output every files name which is not listed already
  82     in the C{filenames} or global C{excluded} dictionaries.
  83
  84     @type dirname: string
  85     @param dirname: the directory to look at
  86
  87     @type filenames: dict
  88     @param filenames: Known filenames to ignore
  89     """
  90     global waste, db_files, excluded
  91
  92     if dirname.find('/disks-') != -1 or dirname.find('upgrade-') != -1:
  93         return
  94     # hack; can't handle .changes files
  95     if dirname.find('proposed-updates') != -1:
  96         return
  97     for name in filenames:
  98         filename = os.path.abspath(os.path.join(dirname,name))
  99         if os.path.isfile(filename) and not os.path.islink(filename) and not db_files.has_key(filename) and not excluded.has_key(filename):
 100             waste += os.stat(filename)[stat.ST_SIZE]
 101             print "%s" % (filename)
 102
 103 ################################################################################
 104
 105 def check_files():
 106     """
 107     Prepare the dictionary of existing filenames, then walk through the archive
 108     pool/ directory to compare it.
 109     """
 110     global db_files
 111
 112     cnf = Config()
 113
 114     print "Building list of database files..."
 115     q = DBConn().session().query(PoolFile).join(Location).order_by('path', 'location')
 116
 117     print "Missing files:"
 118     db_files.clear()
 119
 120     for f in q.all():
 121         filename = os.path.abspath(os.path.join(f.location.path, f.filename))
 122         db_files[filename] = ""
 123         if os.access(filename, os.R_OK) == 0:
 124             if f.last_used:
 125                 print "(last used: %s) %s" % (f.last_used, filename)
 126             else:
 127                 print "%s" % (filename)
 128
 129
 130     filename = os.path.join(cnf["Dir::Override"], 'override.unreferenced')
 131     if os.path.exists(filename):
 132         f = utils.open_file(filename)
 133         for filename in f.readlines():
 134             filename = filename[:-1]
 135             excluded[filename] = ""
 136
 137     print "Existent files not in db:"
 138
 139     os.path.walk(os.path.join(cnf["Dir::Root"], 'pool/'), process_dir, None)
 140
 141     print
 142     print "%s wasted..." % (utils.size_type(waste))
 143
 144 ################################################################################
 145
 146 def check_dscs():
 147     """
 148     Parse every .dsc file in the archive and check for it's validity.
 149     """
 150
 151     cnf = Config()
 152
 153     count = 0
 154     suite = 'unstable'
 155
 156     for component in cnf.SubTree("Component").List():
 157         component = component.lower()
 158         list_filename = '%s%s_%s_source.list' % (cnf["Dir::Lists"], suite, component)
 159         list_file = utils.open_file(list_filename)
 160
 161         for line in list_file.readlines():
 162             f = line[:-1]
 163             try:
 164                 utils.parse_changes(f, signing_rules=1, dsc_file=1)
 165             except InvalidDscError, line:
 166                 utils.warn("syntax error in .dsc file '%s', line %s." % (f, line))
 167                 count += 1
 168             except ChangesUnicodeError:
 169                 utils.warn("found invalid changes file, not properly utf-8 encoded")
 170                 count += 1
 171
 172     if count:
 173         utils.warn("Found %s invalid .dsc files." % (count))
 174
 175 ################################################################################
 176
 177 def check_override():
 178     """
 179     Check for missing overrides in stable and unstable.
 180     """
 181     session = DBConn().session()
 182
 183     for suite_name in [ "stable", "unstable" ]:
 184         print suite_name
 185         print "-" * len(suite_name)
 186         print
 187         suite = get_suite(suite_name)
 188         q = session.execute("""
 189 SELECT DISTINCT b.package FROM binaries b, bin_associations ba
 190  WHERE b.id = ba.bin AND ba.suite = :suiteid AND NOT EXISTS
 191        (SELECT 1 FROM override o WHERE o.suite = :suiteid AND o.package = b.package)"""
 192                           % {'suiteid': suite.suite_id})
 193
 194         for j in q.fetchall():
 195             print j[0]
 196
 197         q = session.execute("""
 198 SELECT DISTINCT s.source FROM source s, src_associations sa
 199   WHERE s.id = sa.source AND sa.suite = :suiteid AND NOT EXISTS
 200        (SELECT 1 FROM override o WHERE o.suite = :suiteid and o.package = s.source)"""
 201                           % {'suiteid': suite.suite_id})
 202         for j in q.fetchall():
 203             print j[0]
 204
 205 ################################################################################
 206
 207
 208 def check_source_in_one_dir():
 209     """
 210     Ensure that the source files for any given package is all in one
 211     directory so that 'apt-get source' works...
 212     """
 213
 214     # Not the most enterprising method, but hey...
 215     broken_count = 0
 216
 217     session = DBConn().session()
 218
 219     q = session.query(DBSource)
 220     for s in q.all():
 221         first_path = ""
 222         first_filename = ""
 223         broken = False
 224
 225         qf = session.query(PoolFile).join(Location).join(DSCFile).filter_by(source_id=s.source_id)
 226         for f in qf.all():
 227             # 0: path
 228             # 1: filename
 229             filename = os.path.join(f.location.path, f.filename)
 230             path = os.path.dirname(filename)
 231
 232             if first_path == "":
 233                 first_path = path
 234                 first_filename = filename
 235             elif first_path != path:
 236                 symlink = path + '/' + os.path.basename(first_filename)
 237                 if not os.path.exists(symlink):
 238                     broken = True
 239                     print "WOAH, we got a live one here... %s [%s] {%s}" % (filename, s.source_id, symlink)
 240         if broken:
 241             broken_count += 1
 242
 243     print "Found %d source packages where the source is not all in one directory." % (broken_count)
 244
 245 ################################################################################
 246 def check_checksums():
 247     """
 248     Validate all files
 249     """
 250     print "Getting file information from database..."
 251     q = DBConn().session().query(PoolFile)
 252
 253     print "Checking file checksums & sizes..."
 254     for f in q:
 255         filename = os.path.abspath(os.path.join(f.location.path, f.filename))
 256
 257         try:
 258             fi = utils.open_file(filename)
 259         except:
 260             utils.warn("can't open '%s'." % (filename))
 261             continue
 262
 263         size = os.stat(filename)[stat.ST_SIZE]
 264         if size != f.filesize:
 265             utils.warn("**WARNING** size mismatch for '%s' ('%s' [current] vs. '%s' [db])." % (filename, size, f.filesize))
 266
 267         md5sum = apt_pkg.md5sum(fi)
 268         if md5sum != f.md5sum:
 269             utils.warn("**WARNING** md5sum mismatch for '%s' ('%s' [current] vs. '%s' [db])." % (filename, md5sum, f.md5sum))
 270
 271         fi.seek(0)
 272         sha1sum = apt_pkg.sha1sum(fi)
 273         if sha1sum != f.sha1sum:
 274             utils.warn("**WARNING** sha1sum mismatch for '%s' ('%s' [current] vs. '%s' [db])." % (filename, sha1sum, f.sha1sum))
 275
 276         fi.seek(0)
 277         sha256sum = apt_pkg.sha256sum(fi)
 278         if sha256sum != f.sha256sum:
 279             utils.warn("**WARNING** sha256sum mismatch for '%s' ('%s' [current] vs. '%s' [db])." % (filename, sha256sum, f.sha256sum))
 280
 281     print "Done."
 282
 283 ################################################################################
 284 #
 285
 286 def Ent(Kind,Name,Link,Mode,UID,GID,Size,MTime,Major,Minor):
 287     global future_files
 288
 289     if MTime > current_time:
 290         future_files[current_file] = MTime
 291         print "%s: %s '%s','%s',%u,%u,%u,%u,%u,%u,%u" % (current_file, Kind,Name,Link,Mode,UID,GID,Size, MTime, Major, Minor)
 292
 293 def check_timestamps():
 294     """
 295     Check all files for timestamps in the future; common from hardware
 296     (e.g. alpha) which have far-future dates as their default dates.
 297     """
 298
 299     global current_file
 300
 301     q = DBConn().session().query(PoolFile).filter(PoolFile.filename.like('.deb$'))
 302
 303     db_files.clear()
 304     count = 0
 305
 306     for pf in q.all():
 307         filename = os.path.abspath(os.path.join(pf.location.path, pf.filename))
 308         if os.access(filename, os.R_OK):
 309             f = utils.open_file(filename)
 310             current_file = filename
 311             sys.stderr.write("Processing %s.\n" % (filename))
 312             apt_inst.debExtract(f, Ent, "control.tar.gz")
 313             f.seek(0)
 314             apt_inst.debExtract(f, Ent, "data.tar.gz")
 315             count += 1
 316
 317     print "Checked %d files (out of %d)." % (count, len(db_files.keys()))
 318
 319 ################################################################################
 320
 321 def check_files_in_dsc():
 322     """
 323     Ensure each .dsc lists appropriate files in its Files field (according
 324     to the format announced in its Format field).
 325     """
 326     count = 0
 327
 328     print "Building list of database files..."
 329     q = DBConn().session().query(PoolFile).filter(PoolFile.filename.like('.dsc$'))
 330
 331     if q.count() > 0:
 332         print "Checking %d files..." % len(ql)
 333     else:
 334         print "No files to check."
 335
 336     for pf in q.all():
 337         filename = os.path.abspath(os.path.join(pf.location.path + pf.filename))
 338
 339         try:
 340             # NB: don't enforce .dsc syntax
 341             dsc = utils.parse_changes(filename, dsc_file=1)
 342         except:
 343             utils.fubar("error parsing .dsc file '%s'." % (filename))
 344
 345         reasons = utils.check_dsc_files(filename, dsc)
 346         for r in reasons:
 347             utils.warn(r)
 348
 349         if len(reasons) > 0:
 350             count += 1
 351
 352     if count:
 353         utils.warn("Found %s invalid .dsc files." % (count))
 354
 355
 356 ################################################################################
 357
 358 def validate_sources(suite, component):
 359     """
 360     Ensure files mentioned in Sources exist
 361     """
 362     filename = "%s/dists/%s/%s/source/Sources.gz" % (Cnf["Dir::Root"], suite, component)
 363     print "Processing %s..." % (filename)
 364     # apt_pkg.ParseTagFile needs a real file handle and can't handle a GzipFile instance...
 365     (fd, temp_filename) = utils.temp_filename()
 366     (result, output) = commands.getstatusoutput("gunzip -c %s > %s" % (filename, temp_filename))
 367     if (result != 0):
 368         sys.stderr.write("Gunzip invocation failed!\n%s\n" % (output))
 369         sys.exit(result)
 370     sources = utils.open_file(temp_filename)
 371     Sources = apt_pkg.ParseTagFile(sources)
 372     while Sources.Step():
 373         source = Sources.Section.Find('Package')
 374         directory = Sources.Section.Find('Directory')
 375         files = Sources.Section.Find('Files')
 376         for i in files.split('\n'):
 377             (md5, size, name) = i.split()
 378             filename = "%s/%s/%s" % (Cnf["Dir::Root"], directory, name)
 379             if not os.path.exists(filename):
 380                 if directory.find("potato") == -1:
 381                     print "W: %s missing." % (filename)
 382                 else:
 383                     pool_location = utils.poolify (source, component)
 384                     pool_filename = "%s/%s/%s" % (Cnf["Dir::Pool"], pool_location, name)
 385                     if not os.path.exists(pool_filename):
 386                         print "E: %s missing (%s)." % (filename, pool_filename)
 387                     else:
 388                         # Create symlink
 389                         pool_filename = os.path.normpath(pool_filename)
 390                         filename = os.path.normpath(filename)
 391                         src = utils.clean_symlink(pool_filename, filename, Cnf["Dir::Root"])
 392                         print "Symlinking: %s -> %s" % (filename, src)
 393                         #os.symlink(src, filename)
 394     sources.close()
 395     os.unlink(temp_filename)
 396
 397 ########################################
 398
 399 def validate_packages(suite, component, architecture):
 400     """
 401     Ensure files mentioned in Packages exist
 402     """
 403     filename = "%s/dists/%s/%s/binary-%s/Packages.gz" \
 404                % (Cnf["Dir::Root"], suite, component, architecture)
 405     print "Processing %s..." % (filename)
 406     # apt_pkg.ParseTagFile needs a real file handle and can't handle a GzipFile instance...
 407     (fd, temp_filename) = utils.temp_filename()
 408     (result, output) = commands.getstatusoutput("gunzip -c %s > %s" % (filename, temp_filename))
 409     if (result != 0):
 410         sys.stderr.write("Gunzip invocation failed!\n%s\n" % (output))
 411         sys.exit(result)
 412     packages = utils.open_file(temp_filename)
 413     Packages = apt_pkg.ParseTagFile(packages)
 414     while Packages.Step():
 415         filename = "%s/%s" % (Cnf["Dir::Root"], Packages.Section.Find('Filename'))
 416         if not os.path.exists(filename):
 417             print "W: %s missing." % (filename)
 418     packages.close()
 419     os.unlink(temp_filename)
 420
 421 ########################################
 422
 423 def check_indices_files_exist():
 424     """
 425     Ensure files mentioned in Packages & Sources exist
 426     """
 427     for suite in [ "stable", "testing", "unstable" ]:
 428         for component in Cnf.ValueList("Suite::%s::Components" % (suite)):
 429             architectures = get_suite_architectures(suite)
 430             for arch in [ i.arch_string.lower() for i in architectures ]:
 431                 if arch == "source":
 432                     validate_sources(suite, component)
 433                 elif arch == "all":
 434                     continue
 435                 else:
 436                     validate_packages(suite, component, arch)
 437
 438 ################################################################################
 439
 440 def check_files_not_symlinks():
 441     """
 442     Check files in the database aren't symlinks
 443     """
 444     print "Building list of database files... ",
 445     before = time.time()
 446     q = DBConn().session().query(PoolFile).filter(PoolFile.filename.like('.dsc$'))
 447
 448     for pf in q.all():
 449         filename = os.path.abspath(os.path.join(pf.location.path, pf.filename))
 450         if os.access(filename, os.R_OK) == 0:
 451             utils.warn("%s: doesn't exist." % (filename))
 452         else:
 453             if os.path.islink(filename):
 454                 utils.warn("%s: is a symlink." % (filename))
 455
 456 ################################################################################
 457
 458 def chk_bd_process_dir (unused, dirname, filenames):
 459     for name in filenames:
 460         if not name.endswith(".dsc"):
 461             continue
 462         filename = os.path.abspath(dirname+'/'+name)
 463         dsc = utils.parse_changes(filename, dsc_file=1)
 464         for field_name in [ "build-depends", "build-depends-indep" ]:
 465             field = dsc.get(field_name)
 466             if field:
 467                 try:
 468                     apt_pkg.ParseSrcDepends(field)
 469                 except:
 470                     print "E: [%s] %s: %s" % (filename, field_name, field)
 471                     pass
 472
 473 ################################################################################
 474
 475 def check_build_depends():
 476     """ Validate build-dependencies of .dsc files in the archive """
 477     cnf = Config()
 478     os.path.walk(cnf["Dir::Root"], chk_bd_process_dir, None)
 479
 480 ################################################################################
 481
 482 def main ():
 483     global db_files, waste, excluded
 484
 485     cnf = Config()
 486
 487     Arguments = [('h',"help","Check-Archive::Options::Help")]
 488     for i in [ "help" ]:
 489         if not cnf.has_key("Check-Archive::Options::%s" % (i)):
 490             cnf["Check-Archive::Options::%s" % (i)] = ""
 491
 492     args = apt_pkg.ParseCommandLine(cnf.Cnf, Arguments, sys.argv)
 493
 494     Options = cnf.SubTree("Check-Archive::Options")
 495     if Options["Help"]:
 496         usage()
 497
 498     if len(args) < 1:
 499         utils.warn("dak check-archive requires at least one argument")
 500         usage(1)
 501     elif len(args) > 1:
 502         utils.warn("dak check-archive accepts only one argument")
 503         usage(1)
 504     mode = args[0].lower()
 505
 506     # Initialize DB
 507     DBConn()
 508
 509     if mode == "checksums":
 510         check_checksums()
 511     elif mode == "files":
 512         check_files()
 513     elif mode == "dsc-syntax":
 514         check_dscs()
 515     elif mode == "missing-overrides":
 516         check_override()
 517     elif mode == "source-in-one-dir":
 518         check_source_in_one_dir()
 519     elif mode == "timestamps":
 520         check_timestamps()
 521     elif mode == "files-in-dsc":
 522         check_files_in_dsc()
 523     elif mode == "validate-indices":
 524         check_indices_files_exist()
 525     elif mode == "files-not-symlinks":
 526         check_files_not_symlinks()
 527     elif mode == "validate-builddeps":
 528         check_build_depends()
 529     else:
 530         utils.warn("unknown mode '%s'" % (mode))
 531         usage(1)
 532
 533 ################################################################################
 534
 535 if __name__ == '__main__':
 536     main()