#!/usr/bin/env python3 # ex:ts=4:sw=4:sts=4:et # -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- # Copyright (c) 2013 Wind River Systems, Inc. # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License version 2 as # published by the Free Software Foundation. # # 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 import os import sys import getopt import shutil import re import warnings import subprocess import argparse scripts_path = os.path.abspath(os.path.dirname(os.path.abspath(sys.argv[0]))) lib_path = scripts_path + '/lib' sys.path = sys.path + [lib_path] import scriptpath # Figure out where is the bitbake/lib/bb since we need bb.siggen and bb.process bitbakepath = scriptpath.add_bitbake_lib_path() if not bitbakepath: sys.stderr.write("Unable to find bitbake by searching parent directory of this script or PATH\n") sys.exit(1) scriptpath.add_oe_lib_path() import argparse_oe import bb.siggen import bb.process # Match the stamp's filename # group(1): PE_PV (may no PE) # group(2): PR # group(3): TASK # group(4): HASH stamp_re = re.compile("(?P<pv>.*)-(?P<pr>r\d+)\.(?P<task>do_\w+)\.(?P<hash>[^\.]*)") sigdata_re = re.compile(".*\.sigdata\..*") def gen_dict(stamps): """ Generate the dict from the stamps dir. The output dict format is: {fake_f: {pn: PN, pv: PV, pr: PR, task: TASK, path: PATH}} Where: fake_f: pv + task + hash path: the path to the stamp file """ # The member of the sub dict (A "path" will be appended below) sub_mem = ("pv", "pr", "task") d = {} for dirpath, _, files in os.walk(stamps): for f in files: # The "bitbake -S" would generate ".sigdata", but no "_setscene". fake_f = re.sub('_setscene.', '.', f) fake_f = re.sub('.sigdata', '', fake_f) subdict = {} tmp = stamp_re.match(fake_f) if tmp: for i in sub_mem: subdict[i] = tmp.group(i) if len(subdict) != 0: pn = os.path.basename(dirpath) subdict['pn'] = pn # The path will be used by os.stat() and bb.siggen subdict['path'] = dirpath + "/" + f fake_f = tmp.group('pv') + tmp.group('task') + tmp.group('hash') d[fake_f] = subdict return d # Re-construct the dict def recon_dict(dict_in): """ The output dict format is: {pn_task: {pv: PV, pr: PR, path: PATH}} """ dict_out = {} for k in dict_in.keys(): subdict = {} # The key pn_task = "%s_%s" % (dict_in.get(k).get('pn'), dict_in.get(k).get('task')) # If more than one stamps are found, use the latest one. if pn_task in dict_out: full_path_pre = dict_out.get(pn_task).get('path') full_path_cur = dict_in.get(k).get('path') if os.stat(full_path_pre).st_mtime > os.stat(full_path_cur).st_mtime: continue subdict['pv'] = dict_in.get(k).get('pv') subdict['pr'] = dict_in.get(k).get('pr') subdict['path'] = dict_in.get(k).get('path') dict_out[pn_task] = subdict return dict_out def split_pntask(s): """ Split the pn_task in to (pn, task) and return it """ tmp = re.match("(.*)_(do_.*)", s) return (tmp.group(1), tmp.group(2)) def print_added(d_new = None, d_old = None): """ Print the newly added tasks """ added = {} for k in list(d_new.keys()): if k not in d_old: # Add the new one to added dict, and remove it from # d_new, so the remaining ones are the changed ones added[k] = d_new.get(k) del(d_new[k]) if not added: return 0 # Format the output, the dict format is: # {pn: task1, task2 ...} added_format = {} counter = 0 for k in added.keys(): pn, task = split_pntask(k) if pn in added_format: # Append the value added_format[pn] = "%s %s" % (added_format.get(pn), task) else: added_format[pn] = task counter += 1 print("=== Newly added tasks: (%s tasks)" % counter) for k in added_format.keys(): print(" %s: %s" % (k, added_format.get(k))) return counter def print_vrchanged(d_new = None, d_old = None, vr = None): """ Print the pv or pr changed tasks. The arg "vr" is "pv" or "pr" """ pvchanged = {} counter = 0 for k in list(d_new.keys()): if d_new.get(k).get(vr) != d_old.get(k).get(vr): counter += 1 pn, task = split_pntask(k) if pn not in pvchanged: # Format the output, we only print pn (no task) since # all the tasks would be changed when pn or pr changed, # the dict format is: # {pn: pv/pr_old -> pv/pr_new} pvchanged[pn] = "%s -> %s" % (d_old.get(k).get(vr), d_new.get(k).get(vr)) del(d_new[k]) if not pvchanged: return 0 print("\n=== %s changed: (%s tasks)" % (vr.upper(), counter)) for k in pvchanged.keys(): print(" %s: %s" % (k, pvchanged.get(k))) return counter def print_depchanged(d_new = None, d_old = None, verbose = False): """ Print the dependency changes """ depchanged = {} counter = 0 for k in d_new.keys(): counter += 1 pn, task = split_pntask(k) if (verbose): full_path_old = d_old.get(k).get("path") full_path_new = d_new.get(k).get("path") # No counter since it is not ready here if sigdata_re.match(full_path_old) and sigdata_re.match(full_path_new): output = bb.siggen.compare_sigfiles(full_path_old, full_path_new) if output: print("\n=== The verbose changes of %s.%s:" % (pn, task)) print('\n'.join(output)) else: # Format the output, the format is: # {pn: task1, task2, ...} if pn in depchanged: depchanged[pn] = "%s %s" % (depchanged.get(pn), task) else: depchanged[pn] = task if len(depchanged) > 0: print("\n=== Dependencies changed: (%s tasks)" % counter) for k in depchanged.keys(): print(" %s: %s" % (k, depchanged[k])) return counter def main(): """ Print what will be done between the current and last builds: 1) Run "STAMPS_DIR=<path> bitbake -S recipe" to re-generate the stamps 2) Figure out what are newly added and changed, can't figure out what are removed since we can't know the previous stamps clearly, for example, if there are several builds, we can't know which stamps the last build has used exactly. 3) Use bb.siggen.compare_sigfiles to diff the old and new stamps """ parser = argparse_oe.ArgumentParser(usage = """%(prog)s [options] [package ...] print what will be done between the current and last builds, for example: $ bitbake core-image-sato # Edit the recipes $ bitbake-whatchanged core-image-sato The changes will be printed" Note: The amount of tasks is not accurate when the task is "do_build" since it usually depends on other tasks. The "nostamp" task is not included. """ ) parser.add_argument("recipe", help="recipe to check") parser.add_argument("-v", "--verbose", help = "print the verbose changes", action = "store_true") args = parser.parse_args() # Get the STAMPS_DIR print("Figuring out the STAMPS_DIR ...") cmdline = "bitbake -e | sed -ne 's/^STAMPS_DIR=\"\(.*\)\"/\\1/p'" try: stampsdir, err = bb.process.run(cmdline) except: raise if not stampsdir: print("ERROR: No STAMPS_DIR found for '%s'" % args.recipe, file=sys.stderr) return 2 stampsdir = stampsdir.rstrip("\n") if not os.path.isdir(stampsdir): print("ERROR: stamps directory \"%s\" not found!" % stampsdir, file=sys.stderr) return 2 # The new stamps dir new_stampsdir = stampsdir + ".bbs" if os.path.exists(new_stampsdir): print("ERROR: %s already exists!" % new_stampsdir, file=sys.stderr) return 2 try: # Generate the new stamps dir print("Generating the new stamps ... (need several minutes)") cmdline = "STAMPS_DIR=%s bitbake -S none %s" % (new_stampsdir, args.recipe) # FIXME # The "bitbake -S" may fail, not fatal error, the stamps will still # be generated, this might be a bug of "bitbake -S". try: bb.process.run(cmdline) except Exception as exc: print(exc) # The dict for the new and old stamps. old_dict = gen_dict(stampsdir) new_dict = gen_dict(new_stampsdir) # Remove the same one from both stamps. cnt_unchanged = 0 for k in list(new_dict.keys()): if k in old_dict: cnt_unchanged += 1 del(new_dict[k]) del(old_dict[k]) # Re-construct the dict to easily find out what is added or changed. # The dict format is: # {pn_task: {pv: PV, pr: PR, path: PATH}} new_recon = recon_dict(new_dict) old_recon = recon_dict(old_dict) del new_dict del old_dict # Figure out what are changed, the new_recon would be changed # by the print_xxx function. # Newly added cnt_added = print_added(new_recon, old_recon) # PV (including PE) and PR changed # Let the bb.siggen handle them if verbose cnt_rv = {} if not args.verbose: for i in ('pv', 'pr'): cnt_rv[i] = print_vrchanged(new_recon, old_recon, i) # Dependencies changed (use bitbake-diffsigs) cnt_dep = print_depchanged(new_recon, old_recon, args.verbose) total_changed = cnt_added + (cnt_rv.get('pv') or 0) + (cnt_rv.get('pr') or 0) + cnt_dep print("\n=== Summary: (%s changed, %s unchanged)" % (total_changed, cnt_unchanged)) if args.verbose: print("Newly added: %s\nDependencies changed: %s\n" % \ (cnt_added, cnt_dep)) else: print("Newly added: %s\nPV changed: %s\nPR changed: %s\nDependencies changed: %s\n" % \ (cnt_added, cnt_rv.get('pv') or 0, cnt_rv.get('pr') or 0, cnt_dep)) except: print("ERROR occurred!") raise finally: # Remove the newly generated stamps dir if os.path.exists(new_stampsdir): print("Removing the newly generated stamps dir ...") shutil.rmtree(new_stampsdir) if __name__ == "__main__": sys.exit(main())