#!/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.
|
#
|
# SPDX-License-Identifier: GPL-2.0-only
|
#
|
|
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())
|
