rediff-patches.py

   1 #!/usr/bin/python3
   2
   3 # rediff-patches.py name.spec
   4
   5 import argparse
   6 import collections
   7 import logging
   8 import os
   9 import re
  10 import rpm
  11 import shutil
  12 import subprocess
  13 import sys
  14 import tempfile
  15
  16 RPMBUILD_ISPATCH = (1<<1)
  17
  18 def prepare_spec(r, patch_nr, before=False):
  19     tempspec = tempfile.NamedTemporaryFile()
  20     re_patch = re.compile(r'^%patch(?P<patch_number>\d+)\w*')
  21     for line in r.parsed.split('\n'):
  22         m = re_patch.match(line)
  23         if m:
  24             patch_number = int(m.group('patch_number'))
  25             if patch_nr == patch_number:
  26                 if before:
  27                     tempspec.write(b"exit 0\n# here was patch%d\n" % patch_nr)
  28                 else:
  29                     line = re.sub(r'#.*', "", line)
  30                     tempspec.write(b"%s\nexit 0\n" % line.encode('utf-8'))
  31                 continue
  32         tempspec.write(b"%s\n" % line.encode('utf-8'))
  33     tempspec.flush()
  34     return tempspec
  35
  36 def unpack(spec, appsourcedir, builddir):
  37     cmd = [ 'rpmbuild', '-bp',
  38            '--define',  '_builddir %s' % builddir,
  39            '--define', '_specdir %s' % appsourcedir,
  40            '--define', '_sourcedir %s' % appsourcedir,
  41            '--define', '_enable_debug_packages 0',
  42            '--define', '_default_patch_fuzz 2',
  43            spec ]
  44     logging.debug("running %s" % repr(cmd))
  45     try:
  46         res = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, check=True,
  47                               env={'LC_ALL': 'C.UTF-8'}, timeout=600)
  48     except subprocess.CalledProcessError as err:
  49         logging.error("unpacking exited with status code %d." % err.returncode)
  50         logging.error("STDOUT:")
  51         if err.stdout:
  52             for line in err.stdout.decode('utf-8').split("\n"):
  53                 logging.error(line)
  54         logging.error("STDERR:")
  55         if err.stderr:
  56             for line in err.stderr.decode('utf-8').split("\n"):
  57                 logging.error(line)
  58         raise
  59     else:
  60         logging.debug("unpacking exited with status code %d." % res.returncode)
  61         logging.debug("STDOUT/STDERR:")
  62         if res.stdout:
  63             for line in res.stdout.decode('utf-8').split("\n"):
  64                 logging.debug(line)
  65
  66
  67 def patch_comment_get(patch):
  68     patch_comment = ""
  69     with open(patch, 'rt') as f:
  70         for line in f:
  71             if line.startswith('diff ') or line.startswith('--- '):
  72                 break
  73             patch_comment += line
  74     return patch_comment
  75
  76 def diff(diffdir_org, diffdir, builddir, patch_comment, output):
  77     diffdir_org = os.path.basename(diffdir_org)
  78     diffdir = os.path.basename(diffdir)
  79
  80     with open(output, 'wt') as f:
  81         if patch_comment:
  82             f.write(patch_comment)
  83             f.flush()
  84         cmd = [ 'diff', '-urNp', '-x', '*.orig', diffdir_org, diffdir ]
  85         logging.debug("running %s" % repr(cmd))
  86         try:
  87             subprocess.check_call(cmd, cwd=builddir, stdout=f, stderr=sys.stderr,
  88                                   env={'LC_ALL': 'C.UTF-8'}, timeout=600)
  89         except subprocess.CalledProcessError as err:
  90             if err.returncode != 1:
  91                 raise
  92     logging.info("rediff generated as %s" % output)
  93
  94 def diffstat(patch):
  95     cmd = [ 'diffstat', patch ]
  96     logging.info("running diffstat for: %s" % patch)
  97     try:
  98         subprocess.check_call(cmd, stdout=sys.stdout, stderr=sys.stderr,
  99                               env={'LC_ALL': 'C.UTF-8'}, timeout=60)
 100     except subprocess.CalledProcessError as err:
 101         logging.error("running diffstat failed: %s" % err)
 102     except FileNotFoundError as err:
 103         logging.error("running diffstat failed: %s, install diffstat package?" % err)
 104
 105 def main():
 106     parser = parser = argparse.ArgumentParser(description='rediff patches to avoid fuzzy hunks')
 107     parser.add_argument('spec', type=str, help='spec file name')
 108     parser.add_argument('-p', '--patches', type=str, help='comma separated list of patch numbers to rediff')
 109     parser.add_argument('-s', '--skip-patches', type=str, help='comma separated list of patch numbers to skip rediff')
 110     parser.add_argument('-v', '--verbose', help='increase output verbosity', action='store_true')
 111     args = parser.parse_args()
 112
 113     logging.basicConfig(level=logging.INFO)
 114     rpm.setVerbosity(rpm.RPMLOG_ERR)
 115
 116     if args.verbose:
 117             logging.basicConfig(level=logging.DEBUG)
 118             rpm.setVerbosity(rpm.RPMLOG_DEBUG)
 119
 120     if args.patches:
 121         args.patches = [int(x) for x in args.patches.split(',')]
 122
 123     if args.skip_patches:
 124         args.skip_patches = [int(x) for x in args.skip_patches.split(',')]
 125
 126     specfile = args.spec
 127     appsourcedir = os.path.dirname(os.path.abspath(specfile))
 128
 129     tempdir = tempfile.TemporaryDirectory(dir="/dev/shm")
 130     topdir = tempdir.name
 131     builddir = os.path.join(topdir, 'BUILD')
 132
 133     rpm.addMacro("_builddir", builddir)
 134
 135     r = rpm.spec(specfile)
 136
 137     patches = {}
 138
 139     for (name, nr, flags) in r.sources:
 140         if flags & RPMBUILD_ISPATCH:
 141             patches[nr] = name
 142
 143     applied_patches = collections.OrderedDict()
 144     re_patch = re.compile(r'^%patch(?P<patch_number>\d+)\w*(?P<patch_args>.*)')
 145     for line in r.parsed.split('\n'):
 146         m = re_patch.match(line)
 147         if not m:
 148             continue
 149         patch_nr = int(m.group('patch_number'))
 150         patch_args = m.group('patch_args')
 151         applied_patches[patch_nr] = patch_args
 152
 153     appbuilddir = rpm.expandMacro("%{_builddir}/%{?buildsubdir}")
 154
 155     for patch_nr in applied_patches.keys():
 156         if args.patches and patch_nr not in args.patches:
 157             continue
 158         if args.skip_patches and patch_nr in args.skip_patches:
 159             continue
 160         patch_name = patches[patch_nr]
 161         logging.info("*** patch %d: %s" % (patch_nr, patch_name))
 162
 163         tempspec = prepare_spec(r, patch_nr, before=True)
 164         unpack(tempspec.name, appsourcedir, builddir)
 165         tempspec.close()
 166         os.rename(appbuilddir, appbuilddir + ".org")
 167
 168         tempspec = prepare_spec(r, patch_nr, before=False)
 169         unpack(tempspec.name, appsourcedir, builddir)
 170         tempspec.close()
 171
 172         patch_comment = patch_comment_get(patch_name)
 173         diff(appbuilddir + ".org",
 174              appbuilddir,
 175              builddir,
 176              patch_comment,
 177              os.path.join(topdir, os.path.join(appsourcedir, patch_name + ".rediff")))
 178
 179         diffstat(os.path.join(topdir, os.path.join(appsourcedir, patch_name)))
 180         diffstat(os.path.join(topdir, os.path.join(appsourcedir, patch_name + ".rediff")))
 181
 182         shutil.rmtree(builddir)
 183     tempdir.cleanup()
 184
 185 if __name__ == '__main__':
 186     main()