mirror of
https://github.com/AsahiLinux/u-boot
synced 2024-11-17 18:28:55 +00:00
941671a19c
It is useful to be able to search for CONFIG options that match a regex, such as this, which lists boards which define SPL_FIT_GENERATOR and anything not starting with ROCKCHIP: ./tools/moveconfig.py -f SPL_FIT_GENERATOR ~ROCKCHIP.* Add support for this. Signed-off-by: Simon Glass <sjg@chromium.org>
1846 lines
62 KiB
Python
Executable file
1846 lines
62 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# SPDX-License-Identifier: GPL-2.0+
|
|
#
|
|
# Author: Masahiro Yamada <yamada.masahiro@socionext.com>
|
|
#
|
|
|
|
"""
|
|
Move config options from headers to defconfig files.
|
|
|
|
See doc/develop/moveconfig.rst for documentation.
|
|
"""
|
|
|
|
from argparse import ArgumentParser
|
|
import asteval
|
|
import collections
|
|
from contextlib import ExitStack
|
|
import copy
|
|
import difflib
|
|
import doctest
|
|
import filecmp
|
|
import fnmatch
|
|
import glob
|
|
import multiprocessing
|
|
import os
|
|
import queue
|
|
import re
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import threading
|
|
import time
|
|
import unittest
|
|
|
|
from buildman import bsettings
|
|
from buildman import kconfiglib
|
|
from buildman import toolchain
|
|
|
|
SHOW_GNU_MAKE = 'scripts/show-gnu-make'
|
|
SLEEP_TIME=0.03
|
|
|
|
STATE_IDLE = 0
|
|
STATE_DEFCONFIG = 1
|
|
STATE_AUTOCONF = 2
|
|
STATE_SAVEDEFCONFIG = 3
|
|
|
|
ACTION_MOVE = 0
|
|
ACTION_NO_ENTRY = 1
|
|
ACTION_NO_ENTRY_WARN = 2
|
|
ACTION_NO_CHANGE = 3
|
|
|
|
COLOR_BLACK = '0;30'
|
|
COLOR_RED = '0;31'
|
|
COLOR_GREEN = '0;32'
|
|
COLOR_BROWN = '0;33'
|
|
COLOR_BLUE = '0;34'
|
|
COLOR_PURPLE = '0;35'
|
|
COLOR_CYAN = '0;36'
|
|
COLOR_LIGHT_GRAY = '0;37'
|
|
COLOR_DARK_GRAY = '1;30'
|
|
COLOR_LIGHT_RED = '1;31'
|
|
COLOR_LIGHT_GREEN = '1;32'
|
|
COLOR_YELLOW = '1;33'
|
|
COLOR_LIGHT_BLUE = '1;34'
|
|
COLOR_LIGHT_PURPLE = '1;35'
|
|
COLOR_LIGHT_CYAN = '1;36'
|
|
COLOR_WHITE = '1;37'
|
|
|
|
AUTO_CONF_PATH = 'include/config/auto.conf'
|
|
CONFIG_DATABASE = 'moveconfig.db'
|
|
|
|
CONFIG_LEN = len('CONFIG_')
|
|
|
|
SIZES = {
|
|
'SZ_1': 0x00000001, 'SZ_2': 0x00000002,
|
|
'SZ_4': 0x00000004, 'SZ_8': 0x00000008,
|
|
'SZ_16': 0x00000010, 'SZ_32': 0x00000020,
|
|
'SZ_64': 0x00000040, 'SZ_128': 0x00000080,
|
|
'SZ_256': 0x00000100, 'SZ_512': 0x00000200,
|
|
'SZ_1K': 0x00000400, 'SZ_2K': 0x00000800,
|
|
'SZ_4K': 0x00001000, 'SZ_8K': 0x00002000,
|
|
'SZ_16K': 0x00004000, 'SZ_32K': 0x00008000,
|
|
'SZ_64K': 0x00010000, 'SZ_128K': 0x00020000,
|
|
'SZ_256K': 0x00040000, 'SZ_512K': 0x00080000,
|
|
'SZ_1M': 0x00100000, 'SZ_2M': 0x00200000,
|
|
'SZ_4M': 0x00400000, 'SZ_8M': 0x00800000,
|
|
'SZ_16M': 0x01000000, 'SZ_32M': 0x02000000,
|
|
'SZ_64M': 0x04000000, 'SZ_128M': 0x08000000,
|
|
'SZ_256M': 0x10000000, 'SZ_512M': 0x20000000,
|
|
'SZ_1G': 0x40000000, 'SZ_2G': 0x80000000,
|
|
'SZ_4G': 0x100000000
|
|
}
|
|
|
|
RE_REMOVE_DEFCONFIG = re.compile(r'(.*)_defconfig')
|
|
|
|
### helper functions ###
|
|
def remove_defconfig(defc):
|
|
"""Drop the _defconfig suffix on a string
|
|
|
|
Args:
|
|
defc (str): String to convert
|
|
|
|
Returns:
|
|
str: string with the '_defconfig' suffix removed
|
|
"""
|
|
return RE_REMOVE_DEFCONFIG.match(defc)[1]
|
|
|
|
def check_top_directory():
|
|
"""Exit if we are not at the top of source directory."""
|
|
for fname in 'README', 'Licenses':
|
|
if not os.path.exists(fname):
|
|
sys.exit('Please run at the top of source directory.')
|
|
|
|
def check_clean_directory():
|
|
"""Exit if the source tree is not clean."""
|
|
for fname in '.config', 'include/config':
|
|
if os.path.exists(fname):
|
|
sys.exit("source tree is not clean, please run 'make mrproper'")
|
|
|
|
def get_make_cmd():
|
|
"""Get the command name of GNU Make.
|
|
|
|
U-Boot needs GNU Make for building, but the command name is not
|
|
necessarily "make". (for example, "gmake" on FreeBSD).
|
|
Returns the most appropriate command name on your system.
|
|
"""
|
|
with subprocess.Popen([SHOW_GNU_MAKE], stdout=subprocess.PIPE) as proc:
|
|
ret = proc.communicate()
|
|
if proc.returncode:
|
|
sys.exit('GNU Make not found')
|
|
return ret[0].rstrip()
|
|
|
|
def get_matched_defconfig(line):
|
|
"""Get the defconfig files that match a pattern
|
|
|
|
Args:
|
|
line (str): Path or filename to match, e.g. 'configs/snow_defconfig' or
|
|
'k2*_defconfig'. If no directory is provided, 'configs/' is
|
|
prepended
|
|
|
|
Returns:
|
|
list of str: a list of matching defconfig files
|
|
"""
|
|
dirname = os.path.dirname(line)
|
|
if dirname:
|
|
pattern = line
|
|
else:
|
|
pattern = os.path.join('configs', line)
|
|
return glob.glob(pattern) + glob.glob(pattern + '_defconfig')
|
|
|
|
def get_matched_defconfigs(defconfigs_file):
|
|
"""Get all the defconfig files that match the patterns in a file.
|
|
|
|
Args:
|
|
defconfigs_file (str): File containing a list of defconfigs to process,
|
|
or '-' to read the list from stdin
|
|
|
|
Returns:
|
|
list of str: A list of paths to defconfig files, with no duplicates
|
|
"""
|
|
defconfigs = []
|
|
with ExitStack() as stack:
|
|
if defconfigs_file == '-':
|
|
inf = sys.stdin
|
|
defconfigs_file = 'stdin'
|
|
else:
|
|
inf = stack.enter_context(open(defconfigs_file, encoding='utf-8'))
|
|
for i, line in enumerate(inf):
|
|
line = line.strip()
|
|
if not line:
|
|
continue # skip blank lines silently
|
|
if ' ' in line:
|
|
line = line.split(' ')[0] # handle 'git log' input
|
|
matched = get_matched_defconfig(line)
|
|
if not matched:
|
|
print(f"warning: {defconfigs_file}:{i + 1}: no defconfig matched '{line}'",
|
|
file=sys.stderr)
|
|
|
|
defconfigs += matched
|
|
|
|
# use set() to drop multiple matching
|
|
return [defconfig[len('configs') + 1:] for defconfig in set(defconfigs)]
|
|
|
|
def get_all_defconfigs():
|
|
"""Get all the defconfig files under the configs/ directory.
|
|
|
|
Returns:
|
|
list of str: List of paths to defconfig files
|
|
"""
|
|
defconfigs = []
|
|
for (dirpath, _, filenames) in os.walk('configs'):
|
|
dirpath = dirpath[len('configs') + 1:]
|
|
for filename in fnmatch.filter(filenames, '*_defconfig'):
|
|
defconfigs.append(os.path.join(dirpath, filename))
|
|
|
|
return defconfigs
|
|
|
|
def color_text(color_enabled, color, string):
|
|
"""Return colored string."""
|
|
if color_enabled:
|
|
# LF should not be surrounded by the escape sequence.
|
|
# Otherwise, additional whitespace or line-feed might be printed.
|
|
return '\n'.join([ '\033[' + color + 'm' + s + '\033[0m' if s else ''
|
|
for s in string.split('\n') ])
|
|
return string
|
|
|
|
def show_diff(alines, blines, file_path, color_enabled):
|
|
"""Show unidified diff.
|
|
|
|
Args:
|
|
alines (list of str): A list of lines (before)
|
|
blines (list of str): A list of lines (after)
|
|
file_path (str): Path to the file
|
|
color_enabled (bool): Display the diff in color
|
|
"""
|
|
diff = difflib.unified_diff(alines, blines,
|
|
fromfile=os.path.join('a', file_path),
|
|
tofile=os.path.join('b', file_path))
|
|
|
|
for line in diff:
|
|
if line.startswith('-') and not line.startswith('--'):
|
|
print(color_text(color_enabled, COLOR_RED, line))
|
|
elif line.startswith('+') and not line.startswith('++'):
|
|
print(color_text(color_enabled, COLOR_GREEN, line))
|
|
else:
|
|
print(line)
|
|
|
|
def extend_matched_lines(lines, matched, pre_patterns, post_patterns,
|
|
extend_pre, extend_post):
|
|
"""Extend matched lines if desired patterns are found before/after already
|
|
matched lines.
|
|
|
|
Args:
|
|
lines (list of str): list of lines handled.
|
|
matched (list of int): list of line numbers that have been already
|
|
matched (will be updated by this function)
|
|
pre_patterns (list of re.Pattern): list of regular expression that should
|
|
be matched as preamble
|
|
post_patterns (list of re.Pattern): list of regular expression that should
|
|
be matched as postamble
|
|
extend_pre (bool): Add the line number of matched preamble to the matched
|
|
list
|
|
extend_post (bool): Add the line number of matched postamble to the
|
|
matched list
|
|
"""
|
|
extended_matched = []
|
|
|
|
j = matched[0]
|
|
|
|
for i in matched:
|
|
if i == 0 or i < j:
|
|
continue
|
|
j = i
|
|
while j in matched:
|
|
j += 1
|
|
if j >= len(lines):
|
|
break
|
|
|
|
for pat in pre_patterns:
|
|
if pat.search(lines[i - 1]):
|
|
break
|
|
else:
|
|
# not matched
|
|
continue
|
|
|
|
for pat in post_patterns:
|
|
if pat.search(lines[j]):
|
|
break
|
|
else:
|
|
# not matched
|
|
continue
|
|
|
|
if extend_pre:
|
|
extended_matched.append(i - 1)
|
|
if extend_post:
|
|
extended_matched.append(j)
|
|
|
|
matched += extended_matched
|
|
matched.sort()
|
|
|
|
def confirm(args, prompt):
|
|
"""Ask the user to confirm something
|
|
|
|
Args:
|
|
args (Namespace ): program arguments
|
|
|
|
Returns:
|
|
bool: True to confirm, False to cancel/stop
|
|
"""
|
|
if not args.yes:
|
|
while True:
|
|
choice = input(f'{prompt} [y/n]: ')
|
|
choice = choice.lower()
|
|
print(choice)
|
|
if choice in ('y', 'n'):
|
|
break
|
|
|
|
if choice == 'n':
|
|
return False
|
|
|
|
return True
|
|
|
|
def write_file(fname, data):
|
|
"""Write data to a file
|
|
|
|
Args:
|
|
fname (str): Filename to write to
|
|
data (list of str): Lines to write (with or without trailing newline);
|
|
or str to write
|
|
"""
|
|
with open(fname, 'w', encoding='utf-8') as out:
|
|
if isinstance(data, list):
|
|
for line in data:
|
|
print(line.rstrip('\n'), file=out)
|
|
else:
|
|
out.write(data)
|
|
|
|
def read_file(fname, as_lines=True, skip_unicode=False):
|
|
"""Read a file and return the contents
|
|
|
|
Args:
|
|
fname (str): Filename to read from
|
|
as_lines: Return file contents as a list of lines
|
|
skip_unicode (bool): True to report unicode errors and continue
|
|
|
|
Returns:
|
|
iter of str: List of ;ines from the file with newline removed; str if
|
|
as_lines is False with newlines intact; or None if a unicode error
|
|
occurred
|
|
|
|
Raises:
|
|
UnicodeDecodeError: Unicode error occurred when reading
|
|
"""
|
|
with open(fname, encoding='utf-8') as inf:
|
|
try:
|
|
if as_lines:
|
|
return [line.rstrip('\n') for line in inf.readlines()]
|
|
else:
|
|
return inf.read()
|
|
except UnicodeDecodeError as e:
|
|
if not skip_unicode:
|
|
raises
|
|
print("Failed on file %s': %s" % (fname, e))
|
|
return None
|
|
|
|
def cleanup_empty_blocks(header_path, args):
|
|
"""Clean up empty conditional blocks
|
|
|
|
Args:
|
|
header_path (str): path to the cleaned file.
|
|
args (Namespace): program arguments
|
|
"""
|
|
pattern = re.compile(r'^\s*#\s*if.*$\n^\s*#\s*endif.*$\n*', flags=re.M)
|
|
data = read_file(header_path, as_lines=False, skip_unicode=True)
|
|
if data is None:
|
|
return
|
|
|
|
new_data = pattern.sub('\n', data)
|
|
|
|
show_diff(data.splitlines(True), new_data.splitlines(True), header_path,
|
|
args.color)
|
|
|
|
if args.dry_run:
|
|
return
|
|
|
|
if new_data != data:
|
|
write_file(header_path, new_data)
|
|
|
|
def cleanup_one_header(header_path, patterns, args):
|
|
"""Clean regex-matched lines away from a file.
|
|
|
|
Args:
|
|
header_path: path to the cleaned file.
|
|
patterns: list of regex patterns. Any lines matching to these
|
|
patterns are deleted.
|
|
args (Namespace): program arguments
|
|
"""
|
|
lines = read_file(header_path, skip_unicode=True)
|
|
if lines is None:
|
|
return
|
|
|
|
matched = []
|
|
for i, line in enumerate(lines):
|
|
if i - 1 in matched and lines[i - 1].endswith('\\'):
|
|
matched.append(i)
|
|
continue
|
|
for pattern in patterns:
|
|
if pattern.search(line):
|
|
matched.append(i)
|
|
break
|
|
|
|
if not matched:
|
|
return
|
|
|
|
# remove empty #ifdef ... #endif, successive blank lines
|
|
pattern_if = re.compile(r'#\s*if(def|ndef)?\b') # #if, #ifdef, #ifndef
|
|
pattern_elif = re.compile(r'#\s*el(if|se)\b') # #elif, #else
|
|
pattern_endif = re.compile(r'#\s*endif\b') # #endif
|
|
pattern_blank = re.compile(r'^\s*$') # empty line
|
|
|
|
while True:
|
|
old_matched = copy.copy(matched)
|
|
extend_matched_lines(lines, matched, [pattern_if],
|
|
[pattern_endif], True, True)
|
|
extend_matched_lines(lines, matched, [pattern_elif],
|
|
[pattern_elif, pattern_endif], True, False)
|
|
extend_matched_lines(lines, matched, [pattern_if, pattern_elif],
|
|
[pattern_blank], False, True)
|
|
extend_matched_lines(lines, matched, [pattern_blank],
|
|
[pattern_elif, pattern_endif], True, False)
|
|
extend_matched_lines(lines, matched, [pattern_blank],
|
|
[pattern_blank], True, False)
|
|
if matched == old_matched:
|
|
break
|
|
|
|
tolines = copy.copy(lines)
|
|
|
|
for i in reversed(matched):
|
|
tolines.pop(i)
|
|
|
|
show_diff(lines, tolines, header_path, args.color)
|
|
|
|
if args.dry_run:
|
|
return
|
|
|
|
write_file(header_path, tolines)
|
|
|
|
def cleanup_headers(configs, args):
|
|
"""Delete config defines from board headers.
|
|
|
|
Args:
|
|
configs: A list of CONFIGs to remove.
|
|
args (Namespace): program arguments
|
|
"""
|
|
if not confirm(args, 'Clean up headers?'):
|
|
return
|
|
|
|
patterns = []
|
|
for config in configs:
|
|
patterns.append(re.compile(r'#\s*define\s+%s\b' % config))
|
|
patterns.append(re.compile(r'#\s*undef\s+%s\b' % config))
|
|
|
|
for dir in 'include', 'arch', 'board':
|
|
for (dirpath, dirnames, filenames) in os.walk(dir):
|
|
if dirpath == os.path.join('include', 'generated'):
|
|
continue
|
|
for filename in filenames:
|
|
if not filename.endswith(('~', '.dts', '.dtsi', '.bin',
|
|
'.elf','.aml','.dat')):
|
|
header_path = os.path.join(dirpath, filename)
|
|
# This file contains UTF-16 data and no CONFIG symbols
|
|
if header_path == 'include/video_font_data.h':
|
|
continue
|
|
cleanup_one_header(header_path, patterns, args)
|
|
cleanup_empty_blocks(header_path, args)
|
|
|
|
def cleanup_one_extra_option(defconfig_path, configs, args):
|
|
"""Delete config defines in CONFIG_SYS_EXTRA_OPTIONS in one defconfig file.
|
|
|
|
Args:
|
|
defconfig_path: path to the cleaned defconfig file.
|
|
configs: A list of CONFIGs to remove.
|
|
args (Namespace): program arguments
|
|
"""
|
|
|
|
start = 'CONFIG_SYS_EXTRA_OPTIONS="'
|
|
end = '"'
|
|
|
|
lines = read_file(defconfig_path)
|
|
|
|
for i, line in enumerate(lines):
|
|
if line.startswith(start) and line.endswith(end):
|
|
break
|
|
else:
|
|
# CONFIG_SYS_EXTRA_OPTIONS was not found in this defconfig
|
|
return
|
|
|
|
old_tokens = line[len(start):-len(end)].split(',')
|
|
new_tokens = []
|
|
|
|
for token in old_tokens:
|
|
pos = token.find('=')
|
|
if not (token[:pos] if pos >= 0 else token) in configs:
|
|
new_tokens.append(token)
|
|
|
|
if new_tokens == old_tokens:
|
|
return
|
|
|
|
tolines = copy.copy(lines)
|
|
|
|
if new_tokens:
|
|
tolines[i] = start + ','.join(new_tokens) + end
|
|
else:
|
|
tolines.pop(i)
|
|
|
|
show_diff(lines, tolines, defconfig_path, args.color)
|
|
|
|
if args.dry_run:
|
|
return
|
|
|
|
write_file(defconfig_path, tolines)
|
|
|
|
def cleanup_extra_options(configs, args):
|
|
"""Delete config defines in CONFIG_SYS_EXTRA_OPTIONS in defconfig files.
|
|
|
|
Args:
|
|
configs: A list of CONFIGs to remove.
|
|
args (Namespace): program arguments
|
|
"""
|
|
if not confirm(args, 'Clean up CONFIG_SYS_EXTRA_OPTIONS?'):
|
|
return
|
|
|
|
configs = [ config[len('CONFIG_'):] for config in configs ]
|
|
|
|
defconfigs = get_all_defconfigs()
|
|
|
|
for defconfig in defconfigs:
|
|
cleanup_one_extra_option(os.path.join('configs', defconfig), configs,
|
|
args)
|
|
|
|
def cleanup_whitelist(configs, args):
|
|
"""Delete config whitelist entries
|
|
|
|
Args:
|
|
configs: A list of CONFIGs to remove.
|
|
args (Namespace): program arguments
|
|
"""
|
|
if not confirm(args, 'Clean up whitelist entries?'):
|
|
return
|
|
|
|
lines = read_file(os.path.join('scripts', 'config_whitelist.txt'))
|
|
|
|
lines = [x for x in lines if x.strip() not in configs]
|
|
|
|
write_file(os.path.join('scripts', 'config_whitelist.txt'), lines)
|
|
|
|
def find_matching(patterns, line):
|
|
for pat in patterns:
|
|
if pat.search(line):
|
|
return True
|
|
return False
|
|
|
|
def cleanup_readme(configs, args):
|
|
"""Delete config description in README
|
|
|
|
Args:
|
|
configs: A list of CONFIGs to remove.
|
|
args (Namespace): program arguments
|
|
"""
|
|
if not confirm(args, 'Clean up README?'):
|
|
return
|
|
|
|
patterns = []
|
|
for config in configs:
|
|
patterns.append(re.compile(r'^\s+%s' % config))
|
|
|
|
lines = read_file('README')
|
|
|
|
found = False
|
|
newlines = []
|
|
for line in lines:
|
|
if not found:
|
|
found = find_matching(patterns, line)
|
|
if found:
|
|
continue
|
|
|
|
if found and re.search(r'^\s+CONFIG', line):
|
|
found = False
|
|
|
|
if not found:
|
|
newlines.append(line)
|
|
|
|
write_file('README', newlines)
|
|
|
|
def try_expand(line):
|
|
"""If value looks like an expression, try expanding it
|
|
Otherwise just return the existing value
|
|
"""
|
|
if line.find('=') == -1:
|
|
return line
|
|
|
|
try:
|
|
aeval = asteval.Interpreter( usersyms=SIZES, minimal=True )
|
|
cfg, val = re.split("=", line)
|
|
val= val.strip('\"')
|
|
if re.search(r'[*+-/]|<<|SZ_+|\(([^\)]+)\)', val):
|
|
newval = hex(aeval(val))
|
|
print('\tExpanded expression %s to %s' % (val, newval))
|
|
return cfg+'='+newval
|
|
except:
|
|
print('\tFailed to expand expression in %s' % line)
|
|
|
|
return line
|
|
|
|
|
|
### classes ###
|
|
class Progress:
|
|
|
|
"""Progress Indicator"""
|
|
|
|
def __init__(self, total):
|
|
"""Create a new progress indicator.
|
|
|
|
Args:
|
|
total: A number of defconfig files to process.
|
|
"""
|
|
self.current = 0
|
|
self.total = total
|
|
|
|
def inc(self):
|
|
"""Increment the number of processed defconfig files."""
|
|
|
|
self.current += 1
|
|
|
|
def show(self):
|
|
"""Display the progress."""
|
|
print(' %d defconfigs out of %d\r' % (self.current, self.total), end=' ')
|
|
sys.stdout.flush()
|
|
|
|
|
|
class KconfigScanner:
|
|
"""Kconfig scanner."""
|
|
|
|
def __init__(self):
|
|
"""Scan all the Kconfig files and create a Config object."""
|
|
# Define environment variables referenced from Kconfig
|
|
os.environ['srctree'] = os.getcwd()
|
|
os.environ['UBOOTVERSION'] = 'dummy'
|
|
os.environ['KCONFIG_OBJDIR'] = ''
|
|
self.conf = kconfiglib.Kconfig()
|
|
|
|
|
|
class KconfigParser:
|
|
|
|
"""A parser of .config and include/autoconf.mk."""
|
|
|
|
re_arch = re.compile(r'CONFIG_SYS_ARCH="(.*)"')
|
|
re_cpu = re.compile(r'CONFIG_SYS_CPU="(.*)"')
|
|
|
|
def __init__(self, configs, args, build_dir):
|
|
"""Create a new parser.
|
|
|
|
Args:
|
|
configs: A list of CONFIGs to move.
|
|
args (Namespace): program arguments
|
|
build_dir: Build directory.
|
|
"""
|
|
self.configs = configs
|
|
self.args = args
|
|
self.dotconfig = os.path.join(build_dir, '.config')
|
|
self.autoconf = os.path.join(build_dir, 'include', 'autoconf.mk')
|
|
self.spl_autoconf = os.path.join(build_dir, 'spl', 'include',
|
|
'autoconf.mk')
|
|
self.config_autoconf = os.path.join(build_dir, AUTO_CONF_PATH)
|
|
self.defconfig = os.path.join(build_dir, 'defconfig')
|
|
|
|
def get_arch(self):
|
|
"""Parse .config file and return the architecture.
|
|
|
|
Returns:
|
|
Architecture name (e.g. 'arm').
|
|
"""
|
|
arch = ''
|
|
cpu = ''
|
|
for line in read_file(self.dotconfig):
|
|
m = self.re_arch.match(line)
|
|
if m:
|
|
arch = m.group(1)
|
|
continue
|
|
m = self.re_cpu.match(line)
|
|
if m:
|
|
cpu = m.group(1)
|
|
|
|
if not arch:
|
|
return None
|
|
|
|
# fix-up for aarch64
|
|
if arch == 'arm' and cpu == 'armv8':
|
|
arch = 'aarch64'
|
|
|
|
return arch
|
|
|
|
def parse_one_config(self, config, dotconfig_lines, autoconf_lines):
|
|
"""Parse .config, defconfig, include/autoconf.mk for one config.
|
|
|
|
This function looks for the config options in the lines from
|
|
defconfig, .config, and include/autoconf.mk in order to decide
|
|
which action should be taken for this defconfig.
|
|
|
|
Args:
|
|
config: CONFIG name to parse.
|
|
dotconfig_lines: lines from the .config file.
|
|
autoconf_lines: lines from the include/autoconf.mk file.
|
|
|
|
Returns:
|
|
A tupple of the action for this defconfig and the line
|
|
matched for the config.
|
|
"""
|
|
not_set = '# %s is not set' % config
|
|
|
|
for line in autoconf_lines:
|
|
line = line.rstrip()
|
|
if line.startswith(config + '='):
|
|
new_val = line
|
|
break
|
|
else:
|
|
new_val = not_set
|
|
|
|
new_val = try_expand(new_val)
|
|
|
|
for line in dotconfig_lines:
|
|
line = line.rstrip()
|
|
if line.startswith(config + '=') or line == not_set:
|
|
old_val = line
|
|
break
|
|
else:
|
|
if new_val == not_set:
|
|
return (ACTION_NO_ENTRY, config)
|
|
else:
|
|
return (ACTION_NO_ENTRY_WARN, config)
|
|
|
|
# If this CONFIG is neither bool nor trisate
|
|
if old_val[-2:] != '=y' and old_val[-2:] != '=m' and old_val != not_set:
|
|
# tools/scripts/define2mk.sed changes '1' to 'y'.
|
|
# This is a problem if the CONFIG is int type.
|
|
# Check the type in Kconfig and handle it correctly.
|
|
if new_val[-2:] == '=y':
|
|
new_val = new_val[:-1] + '1'
|
|
|
|
return (ACTION_NO_CHANGE if old_val == new_val else ACTION_MOVE,
|
|
new_val)
|
|
|
|
def update_dotconfig(self):
|
|
"""Parse files for the config options and update the .config.
|
|
|
|
This function parses the generated .config and include/autoconf.mk
|
|
searching the target options.
|
|
Move the config option(s) to the .config as needed.
|
|
|
|
Args:
|
|
defconfig: defconfig name.
|
|
|
|
Returns:
|
|
Return a tuple of (updated flag, log string).
|
|
The "updated flag" is True if the .config was updated, False
|
|
otherwise. The "log string" shows what happend to the .config.
|
|
"""
|
|
|
|
results = []
|
|
updated = False
|
|
suspicious = False
|
|
rm_files = [self.config_autoconf, self.autoconf]
|
|
|
|
if self.args.spl:
|
|
if os.path.exists(self.spl_autoconf):
|
|
autoconf_path = self.spl_autoconf
|
|
rm_files.append(self.spl_autoconf)
|
|
else:
|
|
for f in rm_files:
|
|
os.remove(f)
|
|
return (updated, suspicious,
|
|
color_text(self.args.color, COLOR_BROWN,
|
|
"SPL is not enabled. Skipped.") + '\n')
|
|
else:
|
|
autoconf_path = self.autoconf
|
|
|
|
dotconfig_lines = read_file(self.dotconfig)
|
|
|
|
autoconf_lines = read_file(autoconf_path)
|
|
|
|
for config in self.configs:
|
|
result = self.parse_one_config(config, dotconfig_lines,
|
|
autoconf_lines)
|
|
results.append(result)
|
|
|
|
log = ''
|
|
|
|
for (action, value) in results:
|
|
if action == ACTION_MOVE:
|
|
actlog = "Move '%s'" % value
|
|
log_color = COLOR_LIGHT_GREEN
|
|
elif action == ACTION_NO_ENTRY:
|
|
actlog = '%s is not defined in Kconfig. Do nothing.' % value
|
|
log_color = COLOR_LIGHT_BLUE
|
|
elif action == ACTION_NO_ENTRY_WARN:
|
|
actlog = '%s is not defined in Kconfig (suspicious). Do nothing.' % value
|
|
log_color = COLOR_YELLOW
|
|
suspicious = True
|
|
elif action == ACTION_NO_CHANGE:
|
|
actlog = "'%s' is the same as the define in Kconfig. Do nothing." \
|
|
% value
|
|
log_color = COLOR_LIGHT_PURPLE
|
|
elif action == ACTION_SPL_NOT_EXIST:
|
|
actlog = 'SPL is not enabled for this defconfig. Skip.'
|
|
log_color = COLOR_PURPLE
|
|
else:
|
|
sys.exit('Internal Error. This should not happen.')
|
|
|
|
log += color_text(self.args.color, log_color, actlog) + '\n'
|
|
|
|
with open(self.dotconfig, 'a', encoding='utf-8') as out:
|
|
for (action, value) in results:
|
|
if action == ACTION_MOVE:
|
|
out.write(value + '\n')
|
|
updated = True
|
|
|
|
self.results = results
|
|
for f in rm_files:
|
|
os.remove(f)
|
|
|
|
return (updated, suspicious, log)
|
|
|
|
def check_defconfig(self):
|
|
"""Check the defconfig after savedefconfig
|
|
|
|
Returns:
|
|
Return additional log if moved CONFIGs were removed again by
|
|
'make savedefconfig'.
|
|
"""
|
|
|
|
log = ''
|
|
|
|
defconfig_lines = read_file(self.defconfig)
|
|
|
|
for (action, value) in self.results:
|
|
if action != ACTION_MOVE:
|
|
continue
|
|
if not value in defconfig_lines:
|
|
log += color_text(self.args.color, COLOR_YELLOW,
|
|
"'%s' was removed by savedefconfig.\n" %
|
|
value)
|
|
|
|
return log
|
|
|
|
|
|
class DatabaseThread(threading.Thread):
|
|
"""This thread processes results from Slot threads.
|
|
|
|
It collects the data in the master config directary. There is only one
|
|
result thread, and this helps to serialise the build output.
|
|
"""
|
|
def __init__(self, config_db, db_queue):
|
|
"""Set up a new result thread
|
|
|
|
Args:
|
|
builder: Builder which will be sent each result
|
|
"""
|
|
threading.Thread.__init__(self)
|
|
self.config_db = config_db
|
|
self.db_queue= db_queue
|
|
|
|
def run(self):
|
|
"""Called to start up the result thread.
|
|
|
|
We collect the next result job and pass it on to the build.
|
|
"""
|
|
while True:
|
|
defconfig, configs = self.db_queue.get()
|
|
self.config_db[defconfig] = configs
|
|
self.db_queue.task_done()
|
|
|
|
|
|
class Slot:
|
|
|
|
"""A slot to store a subprocess.
|
|
|
|
Each instance of this class handles one subprocess.
|
|
This class is useful to control multiple threads
|
|
for faster processing.
|
|
"""
|
|
|
|
def __init__(self, toolchains, configs, args, progress, devnull,
|
|
make_cmd, reference_src_dir, db_queue):
|
|
"""Create a new process slot.
|
|
|
|
Args:
|
|
toolchains: Toolchains object containing toolchains.
|
|
configs: A list of CONFIGs to move.
|
|
args: Program arguments
|
|
progress: A progress indicator.
|
|
devnull: A file object of '/dev/null'.
|
|
make_cmd: command name of GNU Make.
|
|
reference_src_dir: Determine the true starting config state from this
|
|
source tree.
|
|
db_queue: output queue to write config info for the database
|
|
"""
|
|
self.toolchains = toolchains
|
|
self.args = args
|
|
self.progress = progress
|
|
self.build_dir = tempfile.mkdtemp()
|
|
self.devnull = devnull
|
|
self.make_cmd = (make_cmd, 'O=' + self.build_dir)
|
|
self.reference_src_dir = reference_src_dir
|
|
self.db_queue = db_queue
|
|
self.parser = KconfigParser(configs, args, self.build_dir)
|
|
self.state = STATE_IDLE
|
|
self.failed_boards = set()
|
|
self.suspicious_boards = set()
|
|
|
|
def __del__(self):
|
|
"""Delete the working directory
|
|
|
|
This function makes sure the temporary directory is cleaned away
|
|
even if Python suddenly dies due to error. It should be done in here
|
|
because it is guaranteed the destructor is always invoked when the
|
|
instance of the class gets unreferenced.
|
|
|
|
If the subprocess is still running, wait until it finishes.
|
|
"""
|
|
if self.state != STATE_IDLE:
|
|
while self.ps.poll() == None:
|
|
pass
|
|
shutil.rmtree(self.build_dir)
|
|
|
|
def add(self, defconfig):
|
|
"""Assign a new subprocess for defconfig and add it to the slot.
|
|
|
|
If the slot is vacant, create a new subprocess for processing the
|
|
given defconfig and add it to the slot. Just returns False if
|
|
the slot is occupied (i.e. the current subprocess is still running).
|
|
|
|
Args:
|
|
defconfig: defconfig name.
|
|
|
|
Returns:
|
|
Return True on success or False on failure
|
|
"""
|
|
if self.state != STATE_IDLE:
|
|
return False
|
|
|
|
self.defconfig = defconfig
|
|
self.log = ''
|
|
self.current_src_dir = self.reference_src_dir
|
|
self.do_defconfig()
|
|
return True
|
|
|
|
def poll(self):
|
|
"""Check the status of the subprocess and handle it as needed.
|
|
|
|
Returns True if the slot is vacant (i.e. in idle state).
|
|
If the configuration is successfully finished, assign a new
|
|
subprocess to build include/autoconf.mk.
|
|
If include/autoconf.mk is generated, invoke the parser to
|
|
parse the .config and the include/autoconf.mk, moving
|
|
config options to the .config as needed.
|
|
If the .config was updated, run "make savedefconfig" to sync
|
|
it, update the original defconfig, and then set the slot back
|
|
to the idle state.
|
|
|
|
Returns:
|
|
Return True if the subprocess is terminated, False otherwise
|
|
"""
|
|
if self.state == STATE_IDLE:
|
|
return True
|
|
|
|
if self.ps.poll() == None:
|
|
return False
|
|
|
|
if self.ps.poll() != 0:
|
|
self.handle_error()
|
|
elif self.state == STATE_DEFCONFIG:
|
|
if self.reference_src_dir and not self.current_src_dir:
|
|
self.do_savedefconfig()
|
|
else:
|
|
self.do_autoconf()
|
|
elif self.state == STATE_AUTOCONF:
|
|
if self.current_src_dir:
|
|
self.current_src_dir = None
|
|
self.do_defconfig()
|
|
elif self.args.build_db:
|
|
self.do_build_db()
|
|
else:
|
|
self.do_savedefconfig()
|
|
elif self.state == STATE_SAVEDEFCONFIG:
|
|
self.update_defconfig()
|
|
else:
|
|
sys.exit('Internal Error. This should not happen.')
|
|
|
|
return True if self.state == STATE_IDLE else False
|
|
|
|
def handle_error(self):
|
|
"""Handle error cases."""
|
|
|
|
self.log += color_text(self.args.color, COLOR_LIGHT_RED,
|
|
'Failed to process.\n')
|
|
if self.args.verbose:
|
|
self.log += color_text(self.args.color, COLOR_LIGHT_CYAN,
|
|
self.ps.stderr.read().decode())
|
|
self.finish(False)
|
|
|
|
def do_defconfig(self):
|
|
"""Run 'make <board>_defconfig' to create the .config file."""
|
|
|
|
cmd = list(self.make_cmd)
|
|
cmd.append(self.defconfig)
|
|
self.ps = subprocess.Popen(cmd, stdout=self.devnull,
|
|
stderr=subprocess.PIPE,
|
|
cwd=self.current_src_dir)
|
|
self.state = STATE_DEFCONFIG
|
|
|
|
def do_autoconf(self):
|
|
"""Run 'make AUTO_CONF_PATH'."""
|
|
|
|
arch = self.parser.get_arch()
|
|
try:
|
|
toolchain = self.toolchains.Select(arch)
|
|
except ValueError:
|
|
self.log += color_text(self.args.color, COLOR_YELLOW,
|
|
"Tool chain for '%s' is missing. Do nothing.\n" % arch)
|
|
self.finish(False)
|
|
return
|
|
env = toolchain.MakeEnvironment(False)
|
|
|
|
cmd = list(self.make_cmd)
|
|
cmd.append('KCONFIG_IGNORE_DUPLICATES=1')
|
|
cmd.append(AUTO_CONF_PATH)
|
|
self.ps = subprocess.Popen(cmd, stdout=self.devnull, env=env,
|
|
stderr=subprocess.PIPE,
|
|
cwd=self.current_src_dir)
|
|
self.state = STATE_AUTOCONF
|
|
|
|
def do_build_db(self):
|
|
"""Add the board to the database"""
|
|
configs = {}
|
|
for line in read_file(os.path.join(self.build_dir, AUTO_CONF_PATH)):
|
|
if line.startswith('CONFIG'):
|
|
config, value = line.split('=', 1)
|
|
configs[config] = value.rstrip()
|
|
self.db_queue.put([self.defconfig, configs])
|
|
self.finish(True)
|
|
|
|
def do_savedefconfig(self):
|
|
"""Update the .config and run 'make savedefconfig'."""
|
|
|
|
(updated, suspicious, log) = self.parser.update_dotconfig()
|
|
if suspicious:
|
|
self.suspicious_boards.add(self.defconfig)
|
|
self.log += log
|
|
|
|
if not self.args.force_sync and not updated:
|
|
self.finish(True)
|
|
return
|
|
if updated:
|
|
self.log += color_text(self.args.color, COLOR_LIGHT_GREEN,
|
|
'Syncing by savedefconfig...\n')
|
|
else:
|
|
self.log += 'Syncing by savedefconfig (forced by option)...\n'
|
|
|
|
cmd = list(self.make_cmd)
|
|
cmd.append('savedefconfig')
|
|
self.ps = subprocess.Popen(cmd, stdout=self.devnull,
|
|
stderr=subprocess.PIPE)
|
|
self.state = STATE_SAVEDEFCONFIG
|
|
|
|
def update_defconfig(self):
|
|
"""Update the input defconfig and go back to the idle state."""
|
|
|
|
log = self.parser.check_defconfig()
|
|
if log:
|
|
self.suspicious_boards.add(self.defconfig)
|
|
self.log += log
|
|
orig_defconfig = os.path.join('configs', self.defconfig)
|
|
new_defconfig = os.path.join(self.build_dir, 'defconfig')
|
|
updated = not filecmp.cmp(orig_defconfig, new_defconfig)
|
|
|
|
if updated:
|
|
self.log += color_text(self.args.color, COLOR_LIGHT_BLUE,
|
|
'defconfig was updated.\n')
|
|
|
|
if not self.args.dry_run and updated:
|
|
shutil.move(new_defconfig, orig_defconfig)
|
|
self.finish(True)
|
|
|
|
def finish(self, success):
|
|
"""Display log along with progress and go to the idle state.
|
|
|
|
Args:
|
|
success: Should be True when the defconfig was processed
|
|
successfully, or False when it fails.
|
|
"""
|
|
# output at least 30 characters to hide the "* defconfigs out of *".
|
|
log = self.defconfig.ljust(30) + '\n'
|
|
|
|
log += '\n'.join([ ' ' + s for s in self.log.split('\n') ])
|
|
# Some threads are running in parallel.
|
|
# Print log atomically to not mix up logs from different threads.
|
|
print(log, file=(sys.stdout if success else sys.stderr))
|
|
|
|
if not success:
|
|
if self.args.exit_on_error:
|
|
sys.exit('Exit on error.')
|
|
# If --exit-on-error flag is not set, skip this board and continue.
|
|
# Record the failed board.
|
|
self.failed_boards.add(self.defconfig)
|
|
|
|
self.progress.inc()
|
|
self.progress.show()
|
|
self.state = STATE_IDLE
|
|
|
|
def get_failed_boards(self):
|
|
"""Returns a set of failed boards (defconfigs) in this slot.
|
|
"""
|
|
return self.failed_boards
|
|
|
|
def get_suspicious_boards(self):
|
|
"""Returns a set of boards (defconfigs) with possible misconversion.
|
|
"""
|
|
return self.suspicious_boards - self.failed_boards
|
|
|
|
class Slots:
|
|
|
|
"""Controller of the array of subprocess slots."""
|
|
|
|
def __init__(self, toolchains, configs, args, progress,
|
|
reference_src_dir, db_queue):
|
|
"""Create a new slots controller.
|
|
|
|
Args:
|
|
toolchains: Toolchains object containing toolchains.
|
|
configs: A list of CONFIGs to move.
|
|
args: Program arguments
|
|
progress: A progress indicator.
|
|
reference_src_dir: Determine the true starting config state from this
|
|
source tree.
|
|
db_queue: output queue to write config info for the database
|
|
"""
|
|
self.args = args
|
|
self.slots = []
|
|
devnull = subprocess.DEVNULL
|
|
make_cmd = get_make_cmd()
|
|
for i in range(args.jobs):
|
|
self.slots.append(Slot(toolchains, configs, args, progress,
|
|
devnull, make_cmd, reference_src_dir,
|
|
db_queue))
|
|
|
|
def add(self, defconfig):
|
|
"""Add a new subprocess if a vacant slot is found.
|
|
|
|
Args:
|
|
defconfig: defconfig name to be put into.
|
|
|
|
Returns:
|
|
Return True on success or False on failure
|
|
"""
|
|
for slot in self.slots:
|
|
if slot.add(defconfig):
|
|
return True
|
|
return False
|
|
|
|
def available(self):
|
|
"""Check if there is a vacant slot.
|
|
|
|
Returns:
|
|
Return True if at lease one vacant slot is found, False otherwise.
|
|
"""
|
|
for slot in self.slots:
|
|
if slot.poll():
|
|
return True
|
|
return False
|
|
|
|
def empty(self):
|
|
"""Check if all slots are vacant.
|
|
|
|
Returns:
|
|
Return True if all the slots are vacant, False otherwise.
|
|
"""
|
|
ret = True
|
|
for slot in self.slots:
|
|
if not slot.poll():
|
|
ret = False
|
|
return ret
|
|
|
|
def show_failed_boards(self):
|
|
"""Display all of the failed boards (defconfigs)."""
|
|
boards = set()
|
|
output_file = 'moveconfig.failed'
|
|
|
|
for slot in self.slots:
|
|
boards |= slot.get_failed_boards()
|
|
|
|
if boards:
|
|
boards = '\n'.join(boards) + '\n'
|
|
msg = 'The following boards were not processed due to error:\n'
|
|
msg += boards
|
|
msg += '(the list has been saved in %s)\n' % output_file
|
|
print(color_text(self.args.color, COLOR_LIGHT_RED,
|
|
msg), file=sys.stderr)
|
|
|
|
write_file(output_file, boards)
|
|
|
|
def show_suspicious_boards(self):
|
|
"""Display all boards (defconfigs) with possible misconversion."""
|
|
boards = set()
|
|
output_file = 'moveconfig.suspicious'
|
|
|
|
for slot in self.slots:
|
|
boards |= slot.get_suspicious_boards()
|
|
|
|
if boards:
|
|
boards = '\n'.join(boards) + '\n'
|
|
msg = 'The following boards might have been converted incorrectly.\n'
|
|
msg += 'It is highly recommended to check them manually:\n'
|
|
msg += boards
|
|
msg += '(the list has been saved in %s)\n' % output_file
|
|
print(color_text(self.args.color, COLOR_YELLOW,
|
|
msg), file=sys.stderr)
|
|
|
|
write_file(output_file, boards)
|
|
|
|
class ReferenceSource:
|
|
|
|
"""Reference source against which original configs should be parsed."""
|
|
|
|
def __init__(self, commit):
|
|
"""Create a reference source directory based on a specified commit.
|
|
|
|
Args:
|
|
commit: commit to git-clone
|
|
"""
|
|
self.src_dir = tempfile.mkdtemp()
|
|
print('Cloning git repo to a separate work directory...')
|
|
subprocess.check_output(['git', 'clone', os.getcwd(), '.'],
|
|
cwd=self.src_dir)
|
|
print("Checkout '%s' to build the original autoconf.mk." % \
|
|
subprocess.check_output(['git', 'rev-parse', '--short', commit]).strip())
|
|
subprocess.check_output(['git', 'checkout', commit],
|
|
stderr=subprocess.STDOUT, cwd=self.src_dir)
|
|
|
|
def __del__(self):
|
|
"""Delete the reference source directory
|
|
|
|
This function makes sure the temporary directory is cleaned away
|
|
even if Python suddenly dies due to error. It should be done in here
|
|
because it is guaranteed the destructor is always invoked when the
|
|
instance of the class gets unreferenced.
|
|
"""
|
|
shutil.rmtree(self.src_dir)
|
|
|
|
def get_dir(self):
|
|
"""Return the absolute path to the reference source directory."""
|
|
|
|
return self.src_dir
|
|
|
|
def move_config(toolchains, configs, args, db_queue):
|
|
"""Move config options to defconfig files.
|
|
|
|
Args:
|
|
configs: A list of CONFIGs to move.
|
|
args: Program arguments
|
|
"""
|
|
if len(configs) == 0:
|
|
if args.force_sync:
|
|
print('No CONFIG is specified. You are probably syncing defconfigs.', end=' ')
|
|
elif args.build_db:
|
|
print('Building %s database' % CONFIG_DATABASE)
|
|
else:
|
|
print('Neither CONFIG nor --force-sync is specified. Nothing will happen.', end=' ')
|
|
else:
|
|
print('Move ' + ', '.join(configs), end=' ')
|
|
print('(jobs: %d)\n' % args.jobs)
|
|
|
|
if args.git_ref:
|
|
reference_src = ReferenceSource(args.git_ref)
|
|
reference_src_dir = reference_src.get_dir()
|
|
else:
|
|
reference_src_dir = None
|
|
|
|
if args.defconfigs:
|
|
defconfigs = get_matched_defconfigs(args.defconfigs)
|
|
else:
|
|
defconfigs = get_all_defconfigs()
|
|
|
|
progress = Progress(len(defconfigs))
|
|
slots = Slots(toolchains, configs, args, progress, reference_src_dir,
|
|
db_queue)
|
|
|
|
# Main loop to process defconfig files:
|
|
# Add a new subprocess into a vacant slot.
|
|
# Sleep if there is no available slot.
|
|
for defconfig in defconfigs:
|
|
while not slots.add(defconfig):
|
|
while not slots.available():
|
|
# No available slot: sleep for a while
|
|
time.sleep(SLEEP_TIME)
|
|
|
|
# wait until all the subprocesses finish
|
|
while not slots.empty():
|
|
time.sleep(SLEEP_TIME)
|
|
|
|
print('')
|
|
slots.show_failed_boards()
|
|
slots.show_suspicious_boards()
|
|
|
|
def find_kconfig_rules(kconf, config, imply_config):
|
|
"""Check whether a config has a 'select' or 'imply' keyword
|
|
|
|
Args:
|
|
kconf: Kconfiglib.Kconfig object
|
|
config: Name of config to check (without CONFIG_ prefix)
|
|
imply_config: Implying config (without CONFIG_ prefix) which may or
|
|
may not have an 'imply' for 'config')
|
|
|
|
Returns:
|
|
Symbol object for 'config' if found, else None
|
|
"""
|
|
sym = kconf.syms.get(imply_config)
|
|
if sym:
|
|
for sel, cond in (sym.selects + sym.implies):
|
|
if sel.name == config:
|
|
return sym
|
|
return None
|
|
|
|
def check_imply_rule(kconf, config, imply_config):
|
|
"""Check if we can add an 'imply' option
|
|
|
|
This finds imply_config in the Kconfig and looks to see if it is possible
|
|
to add an 'imply' for 'config' to that part of the Kconfig.
|
|
|
|
Args:
|
|
kconf: Kconfiglib.Kconfig object
|
|
config: Name of config to check (without CONFIG_ prefix)
|
|
imply_config: Implying config (without CONFIG_ prefix) which may or
|
|
may not have an 'imply' for 'config')
|
|
|
|
Returns:
|
|
tuple:
|
|
filename of Kconfig file containing imply_config, or None if none
|
|
line number within the Kconfig file, or 0 if none
|
|
message indicating the result
|
|
"""
|
|
sym = kconf.syms.get(imply_config)
|
|
if not sym:
|
|
return 'cannot find sym'
|
|
nodes = sym.nodes
|
|
if len(nodes) != 1:
|
|
return '%d locations' % len(nodes)
|
|
node = nodes[0]
|
|
fname, linenum = node.filename, node.linenr
|
|
cwd = os.getcwd()
|
|
if cwd and fname.startswith(cwd):
|
|
fname = fname[len(cwd) + 1:]
|
|
file_line = ' at %s:%d' % (fname, linenum)
|
|
data = read_file(fname)
|
|
if data[linenum - 1] != 'config %s' % imply_config:
|
|
return None, 0, 'bad sym format %s%s' % (data[linenum], file_line)
|
|
return fname, linenum, 'adding%s' % file_line
|
|
|
|
def add_imply_rule(config, fname, linenum):
|
|
"""Add a new 'imply' option to a Kconfig
|
|
|
|
Args:
|
|
config: config option to add an imply for (without CONFIG_ prefix)
|
|
fname: Kconfig filename to update
|
|
linenum: Line number to place the 'imply' before
|
|
|
|
Returns:
|
|
Message indicating the result
|
|
"""
|
|
file_line = ' at %s:%d' % (fname, linenum)
|
|
data = read_file(fname)
|
|
linenum -= 1
|
|
|
|
for offset, line in enumerate(data[linenum:]):
|
|
if line.strip().startswith('help') or not line:
|
|
data.insert(linenum + offset, '\timply %s' % config)
|
|
write_file(fname, data)
|
|
return 'added%s' % file_line
|
|
|
|
return 'could not insert%s'
|
|
|
|
(IMPLY_MIN_2, IMPLY_TARGET, IMPLY_CMD, IMPLY_NON_ARCH_BOARD) = (
|
|
1, 2, 4, 8)
|
|
|
|
IMPLY_FLAGS = {
|
|
'min2': [IMPLY_MIN_2, 'Show options which imply >2 boards (normally >5)'],
|
|
'target': [IMPLY_TARGET, 'Allow CONFIG_TARGET_... options to imply'],
|
|
'cmd': [IMPLY_CMD, 'Allow CONFIG_CMD_... to imply'],
|
|
'non-arch-board': [
|
|
IMPLY_NON_ARCH_BOARD,
|
|
'Allow Kconfig options outside arch/ and /board/ to imply'],
|
|
}
|
|
|
|
|
|
def read_database():
|
|
"""Read in the config database
|
|
|
|
Returns:
|
|
tuple:
|
|
set of all config options seen (each a str)
|
|
set of all defconfigs seen (each a str)
|
|
dict of configs for each defconfig:
|
|
key: defconfig name, e.g. "MPC8548CDS_legacy_defconfig"
|
|
value: dict:
|
|
key: CONFIG option
|
|
value: Value of option
|
|
dict of defconfigs for each config:
|
|
key: CONFIG option
|
|
value: set of boards using that option
|
|
|
|
"""
|
|
configs = {}
|
|
|
|
# key is defconfig name, value is dict of (CONFIG_xxx, value)
|
|
config_db = {}
|
|
|
|
# Set of all config options we have seen
|
|
all_configs = set()
|
|
|
|
# Set of all defconfigs we have seen
|
|
all_defconfigs = set()
|
|
|
|
defconfig_db = collections.defaultdict(set)
|
|
for line in read_file(CONFIG_DATABASE):
|
|
line = line.rstrip()
|
|
if not line: # Separator between defconfigs
|
|
config_db[defconfig] = configs
|
|
all_defconfigs.add(defconfig)
|
|
configs = {}
|
|
elif line[0] == ' ': # CONFIG line
|
|
config, value = line.strip().split('=', 1)
|
|
configs[config] = value
|
|
defconfig_db[config].add(defconfig)
|
|
all_configs.add(config)
|
|
else: # New defconfig
|
|
defconfig = line
|
|
|
|
return all_configs, all_defconfigs, config_db, defconfig_db
|
|
|
|
|
|
def do_imply_config(config_list, add_imply, imply_flags, skip_added,
|
|
check_kconfig=True, find_superset=False):
|
|
"""Find CONFIG options which imply those in the list
|
|
|
|
Some CONFIG options can be implied by others and this can help to reduce
|
|
the size of the defconfig files. For example, CONFIG_X86 implies
|
|
CONFIG_CMD_IRQ, so we can put 'imply CMD_IRQ' under 'config X86' and
|
|
all x86 boards will have that option, avoiding adding CONFIG_CMD_IRQ to
|
|
each of the x86 defconfig files.
|
|
|
|
This function uses the moveconfig database to find such options. It
|
|
displays a list of things that could possibly imply those in the list.
|
|
The algorithm ignores any that start with CONFIG_TARGET since these
|
|
typically refer to only a few defconfigs (often one). It also does not
|
|
display a config with less than 5 defconfigs.
|
|
|
|
The algorithm works using sets. For each target config in config_list:
|
|
- Get the set 'defconfigs' which use that target config
|
|
- For each config (from a list of all configs):
|
|
- Get the set 'imply_defconfig' of defconfigs which use that config
|
|
-
|
|
- If imply_defconfigs contains anything not in defconfigs then
|
|
this config does not imply the target config
|
|
|
|
Params:
|
|
config_list: List of CONFIG options to check (each a string)
|
|
add_imply: Automatically add an 'imply' for each config.
|
|
imply_flags: Flags which control which implying configs are allowed
|
|
(IMPLY_...)
|
|
skip_added: Don't show options which already have an imply added.
|
|
check_kconfig: Check if implied symbols already have an 'imply' or
|
|
'select' for the target config, and show this information if so.
|
|
find_superset: True to look for configs which are a superset of those
|
|
already found. So for example if CONFIG_EXYNOS5 implies an option,
|
|
but CONFIG_EXYNOS covers a larger set of defconfigs and also
|
|
implies that option, this will drop the former in favour of the
|
|
latter. In practice this option has not proved very used.
|
|
|
|
Note the terminoloy:
|
|
config - a CONFIG_XXX options (a string, e.g. 'CONFIG_CMD_EEPROM')
|
|
defconfig - a defconfig file (a string, e.g. 'configs/snow_defconfig')
|
|
"""
|
|
kconf = KconfigScanner().conf if check_kconfig else None
|
|
if add_imply and add_imply != 'all':
|
|
add_imply = add_imply.split(',')
|
|
|
|
all_configs, all_defconfigs, config_db, defconfig_db = read_database()
|
|
|
|
# Work through each target config option in turn, independently
|
|
for config in config_list:
|
|
defconfigs = defconfig_db.get(config)
|
|
if not defconfigs:
|
|
print('%s not found in any defconfig' % config)
|
|
continue
|
|
|
|
# Get the set of defconfigs without this one (since a config cannot
|
|
# imply itself)
|
|
non_defconfigs = all_defconfigs - defconfigs
|
|
num_defconfigs = len(defconfigs)
|
|
print('%s found in %d/%d defconfigs' % (config, num_defconfigs,
|
|
len(all_configs)))
|
|
|
|
# This will hold the results: key=config, value=defconfigs containing it
|
|
imply_configs = {}
|
|
rest_configs = all_configs - set([config])
|
|
|
|
# Look at every possible config, except the target one
|
|
for imply_config in rest_configs:
|
|
if 'ERRATUM' in imply_config:
|
|
continue
|
|
if not imply_flags & IMPLY_CMD:
|
|
if 'CONFIG_CMD' in imply_config:
|
|
continue
|
|
if not imply_flags & IMPLY_TARGET:
|
|
if 'CONFIG_TARGET' in imply_config:
|
|
continue
|
|
|
|
# Find set of defconfigs that have this config
|
|
imply_defconfig = defconfig_db[imply_config]
|
|
|
|
# Get the intersection of this with defconfigs containing the
|
|
# target config
|
|
common_defconfigs = imply_defconfig & defconfigs
|
|
|
|
# Get the set of defconfigs containing this config which DO NOT
|
|
# also contain the taret config. If this set is non-empty it means
|
|
# that this config affects other defconfigs as well as (possibly)
|
|
# the ones affected by the target config. This means it implies
|
|
# things we don't want to imply.
|
|
not_common_defconfigs = imply_defconfig & non_defconfigs
|
|
if not_common_defconfigs:
|
|
continue
|
|
|
|
# If there are common defconfigs, imply_config may be useful
|
|
if common_defconfigs:
|
|
skip = False
|
|
if find_superset:
|
|
for prev in list(imply_configs.keys()):
|
|
prev_count = len(imply_configs[prev])
|
|
count = len(common_defconfigs)
|
|
if (prev_count > count and
|
|
(imply_configs[prev] & common_defconfigs ==
|
|
common_defconfigs)):
|
|
# skip imply_config because prev is a superset
|
|
skip = True
|
|
break
|
|
elif count > prev_count:
|
|
# delete prev because imply_config is a superset
|
|
del imply_configs[prev]
|
|
if not skip:
|
|
imply_configs[imply_config] = common_defconfigs
|
|
|
|
# Now we have a dict imply_configs of configs which imply each config
|
|
# The value of each dict item is the set of defconfigs containing that
|
|
# config. Rank them so that we print the configs that imply the largest
|
|
# number of defconfigs first.
|
|
ranked_iconfigs = sorted(imply_configs,
|
|
key=lambda k: len(imply_configs[k]), reverse=True)
|
|
kconfig_info = ''
|
|
cwd = os.getcwd()
|
|
add_list = collections.defaultdict(list)
|
|
for iconfig in ranked_iconfigs:
|
|
num_common = len(imply_configs[iconfig])
|
|
|
|
# Don't bother if there are less than 5 defconfigs affected.
|
|
if num_common < (2 if imply_flags & IMPLY_MIN_2 else 5):
|
|
continue
|
|
missing = defconfigs - imply_configs[iconfig]
|
|
missing_str = ', '.join(missing) if missing else 'all'
|
|
missing_str = ''
|
|
show = True
|
|
if kconf:
|
|
sym = find_kconfig_rules(kconf, config[CONFIG_LEN:],
|
|
iconfig[CONFIG_LEN:])
|
|
kconfig_info = ''
|
|
if sym:
|
|
nodes = sym.nodes
|
|
if len(nodes) == 1:
|
|
fname, linenum = nodes[0].filename, nodes[0].linenr
|
|
if cwd and fname.startswith(cwd):
|
|
fname = fname[len(cwd) + 1:]
|
|
kconfig_info = '%s:%d' % (fname, linenum)
|
|
if skip_added:
|
|
show = False
|
|
else:
|
|
sym = kconf.syms.get(iconfig[CONFIG_LEN:])
|
|
fname = ''
|
|
if sym:
|
|
nodes = sym.nodes
|
|
if len(nodes) == 1:
|
|
fname, linenum = nodes[0].filename, nodes[0].linenr
|
|
if cwd and fname.startswith(cwd):
|
|
fname = fname[len(cwd) + 1:]
|
|
in_arch_board = not sym or (fname.startswith('arch') or
|
|
fname.startswith('board'))
|
|
if (not in_arch_board and
|
|
not imply_flags & IMPLY_NON_ARCH_BOARD):
|
|
continue
|
|
|
|
if add_imply and (add_imply == 'all' or
|
|
iconfig in add_imply):
|
|
fname, linenum, kconfig_info = (check_imply_rule(kconf,
|
|
config[CONFIG_LEN:], iconfig[CONFIG_LEN:]))
|
|
if fname:
|
|
add_list[fname].append(linenum)
|
|
|
|
if show and kconfig_info != 'skip':
|
|
print('%5d : %-30s%-25s %s' % (num_common, iconfig.ljust(30),
|
|
kconfig_info, missing_str))
|
|
|
|
# Having collected a list of things to add, now we add them. We process
|
|
# each file from the largest line number to the smallest so that
|
|
# earlier additions do not affect our line numbers. E.g. if we added an
|
|
# imply at line 20 it would change the position of each line after
|
|
# that.
|
|
for fname, linenums in add_list.items():
|
|
for linenum in sorted(linenums, reverse=True):
|
|
add_imply_rule(config[CONFIG_LEN:], fname, linenum)
|
|
|
|
def defconfig_matches(configs, re_match):
|
|
"""Check if any CONFIG option matches a regex
|
|
|
|
The match must be complete, i.e. from the start to end of the CONFIG option.
|
|
|
|
Args:
|
|
configs (dict): Dict of CONFIG options:
|
|
key: CONFIG option
|
|
value: Value of option
|
|
re_match (re.Pattern): Match to check
|
|
|
|
Returns:
|
|
bool: True if any CONFIG matches the regex
|
|
"""
|
|
for cfg in configs:
|
|
m_cfg = re_match.match(cfg)
|
|
if m_cfg and m_cfg.span()[1] == len(cfg):
|
|
return True
|
|
return False
|
|
|
|
def do_find_config(config_list):
|
|
"""Find boards with a given combination of CONFIGs
|
|
|
|
Params:
|
|
config_list: List of CONFIG options to check (each a regex consisting
|
|
of a config option, with or without a CONFIG_ prefix. If an option
|
|
is preceded by a tilde (~) then it must be false, otherwise it must
|
|
be true)
|
|
"""
|
|
all_configs, all_defconfigs, config_db, defconfig_db = read_database()
|
|
|
|
# Get the whitelist
|
|
adhoc_configs = set(read_file('scripts/config_whitelist.txt'))
|
|
|
|
# Start with all defconfigs
|
|
out = all_defconfigs
|
|
|
|
# Work through each config in turn
|
|
adhoc = []
|
|
for item in config_list:
|
|
# Get the real config name and whether we want this config or not
|
|
cfg = item
|
|
want = True
|
|
if cfg[0] == '~':
|
|
want = False
|
|
cfg = cfg[1:]
|
|
|
|
if cfg in adhoc_configs:
|
|
adhoc.append(cfg)
|
|
continue
|
|
|
|
# Search everything that is still in the running. If it has a config
|
|
# that we want, or doesn't have one that we don't, add it into the
|
|
# running for the next stage
|
|
in_list = out
|
|
out = set()
|
|
re_match = re.compile(cfg)
|
|
for defc in in_list:
|
|
has_cfg = defconfig_matches(config_db[defc], re_match)
|
|
if has_cfg == want:
|
|
out.add(defc)
|
|
if adhoc:
|
|
print(f"Error: Not in Kconfig: %s" % ' '.join(adhoc))
|
|
else:
|
|
print(f'{len(out)} matches')
|
|
print(' '.join([remove_defconfig(item) for item in out]))
|
|
|
|
|
|
def prefix_config(cfg):
|
|
"""Prefix a config with CONFIG_ if needed
|
|
|
|
This handles ~ operator, which indicates that the CONFIG should be disabled
|
|
|
|
>>> prefix_config('FRED')
|
|
'CONFIG_FRED'
|
|
>>> prefix_config('CONFIG_FRED')
|
|
'CONFIG_FRED'
|
|
>>> prefix_config('~FRED')
|
|
'~CONFIG_FRED'
|
|
>>> prefix_config('~CONFIG_FRED')
|
|
'~CONFIG_FRED'
|
|
>>> prefix_config('A123')
|
|
'CONFIG_A123'
|
|
"""
|
|
op = ''
|
|
if cfg[0] == '~':
|
|
op = cfg[0]
|
|
cfg = cfg[1:]
|
|
if not cfg.startswith('CONFIG_'):
|
|
cfg = 'CONFIG_' + cfg
|
|
return op + cfg
|
|
|
|
|
|
def main():
|
|
try:
|
|
cpu_count = multiprocessing.cpu_count()
|
|
except NotImplementedError:
|
|
cpu_count = 1
|
|
|
|
epilog = '''Move config options from headers to defconfig files. See
|
|
doc/develop/moveconfig.rst for documentation.'''
|
|
|
|
parser = ArgumentParser(epilog=epilog)
|
|
# Add arguments here
|
|
parser.add_argument('-a', '--add-imply', type=str, default='',
|
|
help='comma-separated list of CONFIG options to add '
|
|
"an 'imply' statement to for the CONFIG in -i")
|
|
parser.add_argument('-A', '--skip-added', action='store_true', default=False,
|
|
help="don't show options which are already marked as "
|
|
'implying others')
|
|
parser.add_argument('-b', '--build-db', action='store_true', default=False,
|
|
help='build a CONFIG database')
|
|
parser.add_argument('-c', '--color', action='store_true', default=False,
|
|
help='display the log in color')
|
|
parser.add_argument('-C', '--commit', action='store_true', default=False,
|
|
help='Create a git commit for the operation')
|
|
parser.add_argument('-d', '--defconfigs', type=str,
|
|
help='a file containing a list of defconfigs to move, '
|
|
"one per line (for example 'snow_defconfig') "
|
|
"or '-' to read from stdin")
|
|
parser.add_argument('-e', '--exit-on-error', action='store_true',
|
|
default=False,
|
|
help='exit immediately on any error')
|
|
parser.add_argument('-f', '--find', action='store_true', default=False,
|
|
help='Find boards with a given config combination')
|
|
parser.add_argument('-H', '--headers-only', dest='cleanup_headers_only',
|
|
action='store_true', default=False,
|
|
help='only cleanup the headers')
|
|
parser.add_argument('-i', '--imply', action='store_true', default=False,
|
|
help='find options which imply others')
|
|
parser.add_argument('-I', '--imply-flags', type=str, default='',
|
|
help="control the -i option ('help' for help")
|
|
parser.add_argument('-j', '--jobs', type=int, default=cpu_count,
|
|
help='the number of jobs to run simultaneously')
|
|
parser.add_argument('-n', '--dry-run', action='store_true', default=False,
|
|
help='perform a trial run (show log with no changes)')
|
|
parser.add_argument('-r', '--git-ref', type=str,
|
|
help='the git ref to clone for building the autoconf.mk')
|
|
parser.add_argument('-s', '--force-sync', action='store_true', default=False,
|
|
help='force sync by savedefconfig')
|
|
parser.add_argument('-S', '--spl', action='store_true', default=False,
|
|
help='parse config options defined for SPL build')
|
|
parser.add_argument('-t', '--test', action='store_true', default=False,
|
|
help='run unit tests')
|
|
parser.add_argument('-y', '--yes', action='store_true', default=False,
|
|
help="respond 'yes' to any prompts")
|
|
parser.add_argument('-v', '--verbose', action='store_true', default=False,
|
|
help='show any build errors as boards are built')
|
|
parser.add_argument('configs', nargs='*')
|
|
|
|
args = parser.parse_args()
|
|
configs = args.configs
|
|
|
|
if args.test:
|
|
sys.argv = [sys.argv[0]]
|
|
fail, count = doctest.testmod()
|
|
if fail:
|
|
return 1
|
|
unittest.main()
|
|
|
|
if not any((len(configs), args.force_sync, args.build_db, args.imply,
|
|
args.find)):
|
|
parser.print_usage()
|
|
sys.exit(1)
|
|
|
|
# prefix the option name with CONFIG_ if missing
|
|
configs = [prefix_config(cfg) for cfg in configs]
|
|
|
|
check_top_directory()
|
|
|
|
if args.imply:
|
|
imply_flags = 0
|
|
if args.imply_flags == 'all':
|
|
imply_flags = -1
|
|
|
|
elif args.imply_flags:
|
|
for flag in args.imply_flags.split(','):
|
|
bad = flag not in IMPLY_FLAGS
|
|
if bad:
|
|
print("Invalid flag '%s'" % flag)
|
|
if flag == 'help' or bad:
|
|
print("Imply flags: (separate with ',')")
|
|
for name, info in IMPLY_FLAGS.items():
|
|
print(' %-15s: %s' % (name, info[1]))
|
|
parser.print_usage()
|
|
sys.exit(1)
|
|
imply_flags |= IMPLY_FLAGS[flag][0]
|
|
|
|
do_imply_config(configs, args.add_imply, imply_flags, args.skip_added)
|
|
return
|
|
|
|
if args.find:
|
|
do_find_config(configs)
|
|
return
|
|
|
|
config_db = {}
|
|
db_queue = queue.Queue()
|
|
t = DatabaseThread(config_db, db_queue)
|
|
t.setDaemon(True)
|
|
t.start()
|
|
|
|
if not args.cleanup_headers_only:
|
|
check_clean_directory()
|
|
bsettings.Setup('')
|
|
toolchains = toolchain.Toolchains()
|
|
toolchains.GetSettings()
|
|
toolchains.Scan(verbose=False)
|
|
move_config(toolchains, configs, args, db_queue)
|
|
db_queue.join()
|
|
|
|
if configs:
|
|
cleanup_headers(configs, args)
|
|
cleanup_extra_options(configs, args)
|
|
cleanup_whitelist(configs, args)
|
|
cleanup_readme(configs, args)
|
|
|
|
if args.commit:
|
|
subprocess.call(['git', 'add', '-u'])
|
|
if configs:
|
|
msg = 'Convert %s %sto Kconfig' % (configs[0],
|
|
'et al ' if len(configs) > 1 else '')
|
|
msg += ('\n\nThis converts the following to Kconfig:\n %s\n' %
|
|
'\n '.join(configs))
|
|
else:
|
|
msg = 'configs: Resync with savedefconfig'
|
|
msg += '\n\nRsync all defconfig files using moveconfig.py'
|
|
subprocess.call(['git', 'commit', '-s', '-m', msg])
|
|
|
|
if args.build_db:
|
|
with open(CONFIG_DATABASE, 'w', encoding='utf-8') as fd:
|
|
for defconfig, configs in config_db.items():
|
|
fd.write('%s\n' % defconfig)
|
|
for config in sorted(configs.keys()):
|
|
fd.write(' %s=%s\n' % (config, configs[config]))
|
|
fd.write('\n')
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|