blob: 95ef352f05feb2dab09998642d6209c4b5c19fa8 [file] [log] [blame]
Masahiro Yamadab6160812015-05-20 11:36:07 +09001#!/usr/bin/env python2
2#
3# Author: Masahiro Yamada <yamada.masahiro@socionext.com>
4#
5# SPDX-License-Identifier: GPL-2.0+
6#
7
8"""
9Move config options from headers to defconfig files.
10
11Since Kconfig was introduced to U-Boot, we have worked on moving
12config options from headers to Kconfig (defconfig).
13
14This tool intends to help this tremendous work.
15
16
17Usage
18-----
19
Masahiro Yamadab903c4e2016-05-19 15:51:58 +090020First, you must edit the Kconfig to add the menu entries for the configs
Joe Hershberger166edec2015-05-19 13:21:17 -050021you are moving.
22
Masahiro Yamadab903c4e2016-05-19 15:51:58 +090023And then run this tool giving CONFIG names you want to move.
24For example, if you want to move CONFIG_CMD_USB and CONFIG_SYS_TEXT_BASE,
25simply type as follows:
Masahiro Yamadab6160812015-05-20 11:36:07 +090026
Masahiro Yamadab903c4e2016-05-19 15:51:58 +090027 $ tools/moveconfig.py CONFIG_CMD_USB CONFIG_SYS_TEXT_BASE
Masahiro Yamadab6160812015-05-20 11:36:07 +090028
Masahiro Yamadab903c4e2016-05-19 15:51:58 +090029The tool walks through all the defconfig files and move the given CONFIGs.
Masahiro Yamadab6160812015-05-20 11:36:07 +090030
31The log is also displayed on the terminal.
32
Masahiro Yamada465b7c02016-05-19 15:52:02 +090033The log is printed for each defconfig as follows:
Masahiro Yamadab6160812015-05-20 11:36:07 +090034
Masahiro Yamada465b7c02016-05-19 15:52:02 +090035<defconfig_name>
36 <action1>
37 <action2>
38 <action3>
39 ...
Masahiro Yamadab6160812015-05-20 11:36:07 +090040
Masahiro Yamada465b7c02016-05-19 15:52:02 +090041<defconfig_name> is the name of the defconfig.
42
43<action*> shows what the tool did for that defconfig.
Masahiro Yamada7facf882016-08-21 16:12:36 +090044It looks like one of the following:
Masahiro Yamadab6160812015-05-20 11:36:07 +090045
46 - Move 'CONFIG_... '
47 This config option was moved to the defconfig
48
Masahiro Yamada5643d6e2016-05-19 15:51:56 +090049 - CONFIG_... is not defined in Kconfig. Do nothing.
Masahiro Yamada35204d92016-08-22 22:18:21 +090050 The entry for this CONFIG was not found in Kconfig. The option is not
51 defined in the config header, either. So, this case can be just skipped.
52
53 - CONFIG_... is not defined in Kconfig (suspicious). Do nothing.
54 This option is defined in the config header, but its entry was not found
55 in Kconfig.
Masahiro Yamada5643d6e2016-05-19 15:51:56 +090056 There are two common cases:
57 - You forgot to create an entry for the CONFIG before running
58 this tool, or made a typo in a CONFIG passed to this tool.
59 - The entry was hidden due to unmet 'depends on'.
Masahiro Yamada35204d92016-08-22 22:18:21 +090060 The tool does not know if the result is reasonable, so please check it
61 manually.
Masahiro Yamadab6160812015-05-20 11:36:07 +090062
Masahiro Yamada5643d6e2016-05-19 15:51:56 +090063 - 'CONFIG_...' is the same as the define in Kconfig. Do nothing.
64 The define in the config header matched the one in Kconfig.
65 We do not need to touch it.
Masahiro Yamadab6160812015-05-20 11:36:07 +090066
Masahiro Yamadac4d76eb2016-05-19 15:51:53 +090067 - Compiler is missing. Do nothing.
68 The compiler specified for this architecture was not found
69 in your PATH environment.
70 (If -e option is passed, the tool exits immediately.)
71
72 - Failed to process.
Masahiro Yamadab6160812015-05-20 11:36:07 +090073 An error occurred during processing this defconfig. Skipped.
74 (If -e option is passed, the tool exits immediately on error.)
75
76Finally, you will be asked, Clean up headers? [y/n]:
77
78If you say 'y' here, the unnecessary config defines are removed
79from the config headers (include/configs/*.h).
80It just uses the regex method, so you should not rely on it.
81Just in case, please do 'git diff' to see what happened.
82
83
Masahiro Yamadab903c4e2016-05-19 15:51:58 +090084How does it work?
85-----------------
Masahiro Yamadab6160812015-05-20 11:36:07 +090086
87This tool runs configuration and builds include/autoconf.mk for every
88defconfig. The config options defined in Kconfig appear in the .config
89file (unless they are hidden because of unmet dependency.)
90On the other hand, the config options defined by board headers are seen
91in include/autoconf.mk. The tool looks for the specified options in both
Masahiro Yamadab903c4e2016-05-19 15:51:58 +090092of them to decide the appropriate action for the options. If the given
93config option is found in the .config, but its value does not match the
94one from the board header, the config option in the .config is replaced
95with the define in the board header. Then, the .config is synced by
96"make savedefconfig" and the defconfig is updated with it.
Masahiro Yamadab6160812015-05-20 11:36:07 +090097
98For faster processing, this tool handles multi-threading. It creates
99separate build directories where the out-of-tree build is run. The
100temporary build directories are automatically created and deleted as
101needed. The number of threads are chosen based on the number of the CPU
102cores of your system although you can change it via -j (--jobs) option.
103
104
105Toolchains
106----------
107
108Appropriate toolchain are necessary to generate include/autoconf.mk
109for all the architectures supported by U-Boot. Most of them are available
110at the kernel.org site, some are not provided by kernel.org.
111
112The default per-arch CROSS_COMPILE used by this tool is specified by
113the list below, CROSS_COMPILE. You may wish to update the list to
114use your own. Instead of modifying the list directly, you can give
115them via environments.
116
117
118Available options
119-----------------
120
121 -c, --color
122 Surround each portion of the log with escape sequences to display it
123 in color on the terminal.
124
Simon Glass8bf41c22016-09-12 23:18:21 -0600125 -C, --commit
126 Create a git commit with the changes when the operation is complete. A
127 standard commit message is used which may need to be edited.
128
Joe Hershbergerc6e043a2015-05-19 13:21:19 -0500129 -d, --defconfigs
Masahiro Yamada3984d6e2016-10-19 14:39:54 +0900130 Specify a file containing a list of defconfigs to move. The defconfig
131 files can be given with shell-style wildcards.
Joe Hershbergerc6e043a2015-05-19 13:21:19 -0500132
Masahiro Yamadab6160812015-05-20 11:36:07 +0900133 -n, --dry-run
Masahiro Yamadab903c4e2016-05-19 15:51:58 +0900134 Perform a trial run that does not make any changes. It is useful to
Masahiro Yamadab6160812015-05-20 11:36:07 +0900135 see what is going to happen before one actually runs it.
136
137 -e, --exit-on-error
138 Exit immediately if Make exits with a non-zero status while processing
139 a defconfig file.
140
Masahiro Yamada83c17672016-05-19 15:52:08 +0900141 -s, --force-sync
142 Do "make savedefconfig" forcibly for all the defconfig files.
143 If not specified, "make savedefconfig" only occurs for cases
144 where at least one CONFIG was moved.
145
Masahiro Yamada6d139172016-08-22 22:18:22 +0900146 -S, --spl
147 Look for moved config options in spl/include/autoconf.mk instead of
148 include/autoconf.mk. This is useful for moving options for SPL build
149 because SPL related options (mostly prefixed with CONFIG_SPL_) are
150 sometimes blocked by CONFIG_SPL_BUILD ifdef conditionals.
151
Joe Hershberger23475932015-05-19 13:21:20 -0500152 -H, --headers-only
153 Only cleanup the headers; skip the defconfig processing
154
Masahiro Yamadab6160812015-05-20 11:36:07 +0900155 -j, --jobs
156 Specify the number of threads to run simultaneously. If not specified,
157 the number of threads is the same as the number of CPU cores.
158
Joe Hershbergerb1a570f2016-06-10 14:53:32 -0500159 -r, --git-ref
160 Specify the git ref to clone for building the autoconf.mk. If unspecified
161 use the CWD. This is useful for when changes to the Kconfig affect the
162 default values and you want to capture the state of the defconfig from
163 before that change was in effect. If in doubt, specify a ref pre-Kconfig
164 changes (use HEAD if Kconfig changes are not committed). Worst case it will
165 take a bit longer to run, but will always do the right thing.
166
Joe Hershberger808b63f2015-05-19 13:21:24 -0500167 -v, --verbose
168 Show any build errors as boards are built
169
Simon Glass13e05a02016-09-12 23:18:20 -0600170 -y, --yes
171 Instead of prompting, automatically go ahead with all operations. This
172 includes cleaning up headers and CONFIG_SYS_EXTRA_OPTIONS.
173
Masahiro Yamadab6160812015-05-20 11:36:07 +0900174To see the complete list of supported options, run
175
176 $ tools/moveconfig.py -h
177
178"""
179
Masahiro Yamadaea8f5342016-07-25 19:15:24 +0900180import copy
Masahiro Yamada573b3902016-07-25 19:15:25 +0900181import difflib
Masahiro Yamada0f6beda2016-05-19 15:52:07 +0900182import filecmp
Masahiro Yamadab6160812015-05-20 11:36:07 +0900183import fnmatch
Masahiro Yamada3984d6e2016-10-19 14:39:54 +0900184import glob
Masahiro Yamadab6160812015-05-20 11:36:07 +0900185import multiprocessing
186import optparse
187import os
188import re
189import shutil
190import subprocess
191import sys
192import tempfile
193import time
194
195SHOW_GNU_MAKE = 'scripts/show-gnu-make'
196SLEEP_TIME=0.03
197
198# Here is the list of cross-tools I use.
199# Most of them are available at kernel.org
Masahiro Yamada7facf882016-08-21 16:12:36 +0900200# (https://www.kernel.org/pub/tools/crosstool/files/bin/), except the following:
Masahiro Yamadab6160812015-05-20 11:36:07 +0900201# arc: https://github.com/foss-for-synopsys-dwc-arc-processors/toolchain/releases
Bin Meng3bb02f62015-09-25 01:22:39 -0700202# nds32: http://osdk.andestech.com/packages/nds32le-linux-glibc-v1.tgz
Masahiro Yamadab6160812015-05-20 11:36:07 +0900203# nios2: https://sourcery.mentor.com/GNUToolchain/subscription42545
204# sh: http://sourcery.mentor.com/public/gnu_toolchain/sh-linux-gnu
205CROSS_COMPILE = {
206 'arc': 'arc-linux-',
207 'aarch64': 'aarch64-linux-',
208 'arm': 'arm-unknown-linux-gnueabi-',
209 'avr32': 'avr32-linux-',
Masahiro Yamadab6160812015-05-20 11:36:07 +0900210 'm68k': 'm68k-linux-',
211 'microblaze': 'microblaze-linux-',
212 'mips': 'mips-linux-',
213 'nds32': 'nds32le-linux-',
214 'nios2': 'nios2-linux-gnu-',
Masahiro Yamadab6160812015-05-20 11:36:07 +0900215 'powerpc': 'powerpc-linux-',
216 'sh': 'sh-linux-gnu-',
Masahiro Yamadad1d9d602016-08-21 16:03:08 +0900217 'x86': 'i386-linux-',
218 'xtensa': 'xtensa-linux-'
Masahiro Yamadab6160812015-05-20 11:36:07 +0900219}
220
221STATE_IDLE = 0
222STATE_DEFCONFIG = 1
223STATE_AUTOCONF = 2
Joe Hershberger166edec2015-05-19 13:21:17 -0500224STATE_SAVEDEFCONFIG = 3
Masahiro Yamadab6160812015-05-20 11:36:07 +0900225
226ACTION_MOVE = 0
Masahiro Yamada5643d6e2016-05-19 15:51:56 +0900227ACTION_NO_ENTRY = 1
Masahiro Yamada35204d92016-08-22 22:18:21 +0900228ACTION_NO_ENTRY_WARN = 2
229ACTION_NO_CHANGE = 3
Masahiro Yamadab6160812015-05-20 11:36:07 +0900230
231COLOR_BLACK = '0;30'
232COLOR_RED = '0;31'
233COLOR_GREEN = '0;32'
234COLOR_BROWN = '0;33'
235COLOR_BLUE = '0;34'
236COLOR_PURPLE = '0;35'
237COLOR_CYAN = '0;36'
238COLOR_LIGHT_GRAY = '0;37'
239COLOR_DARK_GRAY = '1;30'
240COLOR_LIGHT_RED = '1;31'
241COLOR_LIGHT_GREEN = '1;32'
242COLOR_YELLOW = '1;33'
243COLOR_LIGHT_BLUE = '1;34'
244COLOR_LIGHT_PURPLE = '1;35'
245COLOR_LIGHT_CYAN = '1;36'
246COLOR_WHITE = '1;37'
247
248### helper functions ###
249def get_devnull():
250 """Get the file object of '/dev/null' device."""
251 try:
252 devnull = subprocess.DEVNULL # py3k
253 except AttributeError:
254 devnull = open(os.devnull, 'wb')
255 return devnull
256
257def check_top_directory():
258 """Exit if we are not at the top of source directory."""
259 for f in ('README', 'Licenses'):
260 if not os.path.exists(f):
261 sys.exit('Please run at the top of source directory.')
262
Masahiro Yamada990e6772016-05-19 15:51:54 +0900263def check_clean_directory():
264 """Exit if the source tree is not clean."""
265 for f in ('.config', 'include/config'):
266 if os.path.exists(f):
267 sys.exit("source tree is not clean, please run 'make mrproper'")
268
Masahiro Yamadab6160812015-05-20 11:36:07 +0900269def get_make_cmd():
270 """Get the command name of GNU Make.
271
272 U-Boot needs GNU Make for building, but the command name is not
273 necessarily "make". (for example, "gmake" on FreeBSD).
274 Returns the most appropriate command name on your system.
275 """
276 process = subprocess.Popen([SHOW_GNU_MAKE], stdout=subprocess.PIPE)
277 ret = process.communicate()
278 if process.returncode:
279 sys.exit('GNU Make not found')
280 return ret[0].rstrip()
281
Masahiro Yamada3984d6e2016-10-19 14:39:54 +0900282def get_matched_defconfigs(defconfigs_file):
283 """Get all the defconfig files that match the patterns in a file."""
284 defconfigs = []
285 for i, line in enumerate(open(defconfigs_file)):
286 line = line.strip()
287 if not line:
288 continue # skip blank lines silently
289 pattern = os.path.join('configs', line)
290 matched = glob.glob(pattern) + glob.glob(pattern + '_defconfig')
291 if not matched:
292 print >> sys.stderr, "warning: %s:%d: no defconfig matched '%s'" % \
293 (defconfigs_file, i + 1, line)
294
295 defconfigs += matched
296
297 # use set() to drop multiple matching
298 return [ defconfig[len('configs') + 1:] for defconfig in set(defconfigs) ]
299
Masahiro Yamada58175e32016-07-25 19:15:28 +0900300def get_all_defconfigs():
301 """Get all the defconfig files under the configs/ directory."""
302 defconfigs = []
303 for (dirpath, dirnames, filenames) in os.walk('configs'):
304 dirpath = dirpath[len('configs') + 1:]
305 for filename in fnmatch.filter(filenames, '*_defconfig'):
306 defconfigs.append(os.path.join(dirpath, filename))
307
308 return defconfigs
309
Masahiro Yamadab6160812015-05-20 11:36:07 +0900310def color_text(color_enabled, color, string):
311 """Return colored string."""
312 if color_enabled:
Masahiro Yamada465b7c02016-05-19 15:52:02 +0900313 # LF should not be surrounded by the escape sequence.
314 # Otherwise, additional whitespace or line-feed might be printed.
315 return '\n'.join([ '\033[' + color + 'm' + s + '\033[0m' if s else ''
316 for s in string.split('\n') ])
Masahiro Yamadab6160812015-05-20 11:36:07 +0900317 else:
318 return string
319
Masahiro Yamadaa1a4b092016-07-25 19:15:26 +0900320def show_diff(a, b, file_path, color_enabled):
Masahiro Yamada573b3902016-07-25 19:15:25 +0900321 """Show unidified diff.
322
323 Arguments:
324 a: A list of lines (before)
325 b: A list of lines (after)
326 file_path: Path to the file
Masahiro Yamadaa1a4b092016-07-25 19:15:26 +0900327 color_enabled: Display the diff in color
Masahiro Yamada573b3902016-07-25 19:15:25 +0900328 """
329
330 diff = difflib.unified_diff(a, b,
331 fromfile=os.path.join('a', file_path),
332 tofile=os.path.join('b', file_path))
333
334 for line in diff:
Masahiro Yamadaa1a4b092016-07-25 19:15:26 +0900335 if line[0] == '-' and line[1] != '-':
336 print color_text(color_enabled, COLOR_RED, line),
337 elif line[0] == '+' and line[1] != '+':
338 print color_text(color_enabled, COLOR_GREEN, line),
339 else:
340 print line,
Masahiro Yamada573b3902016-07-25 19:15:25 +0900341
Masahiro Yamadac4d76eb2016-05-19 15:51:53 +0900342def update_cross_compile(color_enabled):
Robert P. J. Day8c60f922016-05-04 04:47:31 -0400343 """Update per-arch CROSS_COMPILE via environment variables
Masahiro Yamadab6160812015-05-20 11:36:07 +0900344
345 The default CROSS_COMPILE values are available
346 in the CROSS_COMPILE list above.
347
Robert P. J. Day8c60f922016-05-04 04:47:31 -0400348 You can override them via environment variables
Masahiro Yamadab6160812015-05-20 11:36:07 +0900349 CROSS_COMPILE_{ARCH}.
350
351 For example, if you want to override toolchain prefixes
352 for ARM and PowerPC, you can do as follows in your shell:
353
354 export CROSS_COMPILE_ARM=...
355 export CROSS_COMPILE_POWERPC=...
Masahiro Yamadac4d76eb2016-05-19 15:51:53 +0900356
357 Then, this function checks if specified compilers really exist in your
358 PATH environment.
Masahiro Yamadab6160812015-05-20 11:36:07 +0900359 """
360 archs = []
361
362 for arch in os.listdir('arch'):
363 if os.path.exists(os.path.join('arch', arch, 'Makefile')):
364 archs.append(arch)
365
366 # arm64 is a special case
367 archs.append('aarch64')
368
369 for arch in archs:
370 env = 'CROSS_COMPILE_' + arch.upper()
371 cross_compile = os.environ.get(env)
Masahiro Yamadac4d76eb2016-05-19 15:51:53 +0900372 if not cross_compile:
373 cross_compile = CROSS_COMPILE.get(arch, '')
374
375 for path in os.environ["PATH"].split(os.pathsep):
376 gcc_path = os.path.join(path, cross_compile + 'gcc')
377 if os.path.isfile(gcc_path) and os.access(gcc_path, os.X_OK):
378 break
379 else:
380 print >> sys.stderr, color_text(color_enabled, COLOR_YELLOW,
381 'warning: %sgcc: not found in PATH. %s architecture boards will be skipped'
382 % (cross_compile, arch))
383 cross_compile = None
384
385 CROSS_COMPILE[arch] = cross_compile
Masahiro Yamadab6160812015-05-20 11:36:07 +0900386
Masahiro Yamadaea8f5342016-07-25 19:15:24 +0900387def extend_matched_lines(lines, matched, pre_patterns, post_patterns, extend_pre,
388 extend_post):
389 """Extend matched lines if desired patterns are found before/after already
390 matched lines.
391
392 Arguments:
393 lines: A list of lines handled.
394 matched: A list of line numbers that have been already matched.
395 (will be updated by this function)
396 pre_patterns: A list of regular expression that should be matched as
397 preamble.
398 post_patterns: A list of regular expression that should be matched as
399 postamble.
400 extend_pre: Add the line number of matched preamble to the matched list.
401 extend_post: Add the line number of matched postamble to the matched list.
402 """
403 extended_matched = []
404
405 j = matched[0]
406
407 for i in matched:
408 if i == 0 or i < j:
409 continue
410 j = i
411 while j in matched:
412 j += 1
413 if j >= len(lines):
414 break
415
416 for p in pre_patterns:
417 if p.search(lines[i - 1]):
418 break
419 else:
420 # not matched
421 continue
422
423 for p in post_patterns:
424 if p.search(lines[j]):
425 break
426 else:
427 # not matched
428 continue
429
430 if extend_pre:
431 extended_matched.append(i - 1)
432 if extend_post:
433 extended_matched.append(j)
434
435 matched += extended_matched
436 matched.sort()
437
Chris Packham85e15c52017-05-02 21:30:46 +1200438def confirm(options, prompt):
439 if not options.yes:
440 while True:
441 choice = raw_input('{} [y/n]: '.format(prompt))
442 choice = choice.lower()
443 print choice
444 if choice == 'y' or choice == 'n':
445 break
446
447 if choice == 'n':
448 return False
449
450 return True
451
Masahiro Yamadaa1a4b092016-07-25 19:15:26 +0900452def cleanup_one_header(header_path, patterns, options):
Masahiro Yamadab6160812015-05-20 11:36:07 +0900453 """Clean regex-matched lines away from a file.
454
455 Arguments:
456 header_path: path to the cleaned file.
457 patterns: list of regex patterns. Any lines matching to these
458 patterns are deleted.
Masahiro Yamadaa1a4b092016-07-25 19:15:26 +0900459 options: option flags.
Masahiro Yamadab6160812015-05-20 11:36:07 +0900460 """
461 with open(header_path) as f:
462 lines = f.readlines()
463
464 matched = []
465 for i, line in enumerate(lines):
Masahiro Yamada6d798ba2016-07-25 19:15:27 +0900466 if i - 1 in matched and lines[i - 1][-2:] == '\\\n':
467 matched.append(i)
468 continue
Masahiro Yamadab6160812015-05-20 11:36:07 +0900469 for pattern in patterns:
Masahiro Yamadaea8f5342016-07-25 19:15:24 +0900470 if pattern.search(line):
Masahiro Yamadab6160812015-05-20 11:36:07 +0900471 matched.append(i)
472 break
473
Masahiro Yamadaea8f5342016-07-25 19:15:24 +0900474 if not matched:
475 return
476
477 # remove empty #ifdef ... #endif, successive blank lines
478 pattern_if = re.compile(r'#\s*if(def|ndef)?\W') # #if, #ifdef, #ifndef
479 pattern_elif = re.compile(r'#\s*el(if|se)\W') # #elif, #else
480 pattern_endif = re.compile(r'#\s*endif\W') # #endif
481 pattern_blank = re.compile(r'^\s*$') # empty line
482
483 while True:
484 old_matched = copy.copy(matched)
485 extend_matched_lines(lines, matched, [pattern_if],
486 [pattern_endif], True, True)
487 extend_matched_lines(lines, matched, [pattern_elif],
488 [pattern_elif, pattern_endif], True, False)
489 extend_matched_lines(lines, matched, [pattern_if, pattern_elif],
490 [pattern_blank], False, True)
491 extend_matched_lines(lines, matched, [pattern_blank],
492 [pattern_elif, pattern_endif], True, False)
493 extend_matched_lines(lines, matched, [pattern_blank],
494 [pattern_blank], True, False)
495 if matched == old_matched:
496 break
497
Masahiro Yamada573b3902016-07-25 19:15:25 +0900498 tolines = copy.copy(lines)
499
500 for i in reversed(matched):
501 tolines.pop(i)
502
Masahiro Yamadaa1a4b092016-07-25 19:15:26 +0900503 show_diff(lines, tolines, header_path, options.color)
Masahiro Yamadaea8f5342016-07-25 19:15:24 +0900504
Masahiro Yamadaa1a4b092016-07-25 19:15:26 +0900505 if options.dry_run:
Masahiro Yamadab6160812015-05-20 11:36:07 +0900506 return
507
508 with open(header_path, 'w') as f:
Masahiro Yamada573b3902016-07-25 19:15:25 +0900509 for line in tolines:
510 f.write(line)
Masahiro Yamadab6160812015-05-20 11:36:07 +0900511
Masahiro Yamadaa1a4b092016-07-25 19:15:26 +0900512def cleanup_headers(configs, options):
Masahiro Yamadab6160812015-05-20 11:36:07 +0900513 """Delete config defines from board headers.
514
515 Arguments:
Masahiro Yamadab80a6672016-05-19 15:51:57 +0900516 configs: A list of CONFIGs to remove.
Masahiro Yamadaa1a4b092016-07-25 19:15:26 +0900517 options: option flags.
Masahiro Yamadab6160812015-05-20 11:36:07 +0900518 """
Chris Packham85e15c52017-05-02 21:30:46 +1200519 if not confirm(options, 'Clean up headers?'):
520 return
Masahiro Yamadab6160812015-05-20 11:36:07 +0900521
522 patterns = []
Masahiro Yamadab80a6672016-05-19 15:51:57 +0900523 for config in configs:
Masahiro Yamadab6160812015-05-20 11:36:07 +0900524 patterns.append(re.compile(r'#\s*define\s+%s\W' % config))
525 patterns.append(re.compile(r'#\s*undef\s+%s\W' % config))
526
Joe Hershbergerb78ad422015-05-19 13:21:21 -0500527 for dir in 'include', 'arch', 'board':
528 for (dirpath, dirnames, filenames) in os.walk(dir):
Masahiro Yamada28a6d352016-07-25 19:15:22 +0900529 if dirpath == os.path.join('include', 'generated'):
530 continue
Joe Hershbergerb78ad422015-05-19 13:21:21 -0500531 for filename in filenames:
532 if not fnmatch.fnmatch(filename, '*~'):
533 cleanup_one_header(os.path.join(dirpath, filename),
Masahiro Yamadaa1a4b092016-07-25 19:15:26 +0900534 patterns, options)
Masahiro Yamadab6160812015-05-20 11:36:07 +0900535
Masahiro Yamadadce28de2016-07-25 19:15:29 +0900536def cleanup_one_extra_option(defconfig_path, configs, options):
537 """Delete config defines in CONFIG_SYS_EXTRA_OPTIONS in one defconfig file.
538
539 Arguments:
540 defconfig_path: path to the cleaned defconfig file.
541 configs: A list of CONFIGs to remove.
542 options: option flags.
543 """
544
545 start = 'CONFIG_SYS_EXTRA_OPTIONS="'
546 end = '"\n'
547
548 with open(defconfig_path) as f:
549 lines = f.readlines()
550
551 for i, line in enumerate(lines):
552 if line.startswith(start) and line.endswith(end):
553 break
554 else:
555 # CONFIG_SYS_EXTRA_OPTIONS was not found in this defconfig
556 return
557
558 old_tokens = line[len(start):-len(end)].split(',')
559 new_tokens = []
560
561 for token in old_tokens:
562 pos = token.find('=')
563 if not (token[:pos] if pos >= 0 else token) in configs:
564 new_tokens.append(token)
565
566 if new_tokens == old_tokens:
567 return
568
569 tolines = copy.copy(lines)
570
571 if new_tokens:
572 tolines[i] = start + ','.join(new_tokens) + end
573 else:
574 tolines.pop(i)
575
576 show_diff(lines, tolines, defconfig_path, options.color)
577
578 if options.dry_run:
579 return
580
581 with open(defconfig_path, 'w') as f:
582 for line in tolines:
583 f.write(line)
584
585def cleanup_extra_options(configs, options):
586 """Delete config defines in CONFIG_SYS_EXTRA_OPTIONS in defconfig files.
587
588 Arguments:
589 configs: A list of CONFIGs to remove.
590 options: option flags.
591 """
Chris Packham85e15c52017-05-02 21:30:46 +1200592 if not confirm(options, 'Clean up CONFIG_SYS_EXTRA_OPTIONS?'):
593 return
Masahiro Yamadadce28de2016-07-25 19:15:29 +0900594
595 configs = [ config[len('CONFIG_'):] for config in configs ]
596
597 defconfigs = get_all_defconfigs()
598
599 for defconfig in defconfigs:
600 cleanup_one_extra_option(os.path.join('configs', defconfig), configs,
601 options)
602
Chris Packham9d5274f2017-05-02 21:30:47 +1200603def cleanup_whitelist(configs, options):
604 """Delete config whitelist entries
605
606 Arguments:
607 configs: A list of CONFIGs to remove.
608 options: option flags.
609 """
610 if not confirm(options, 'Clean up whitelist entries?'):
611 return
612
613 with open(os.path.join('scripts', 'config_whitelist.txt')) as f:
614 lines = f.readlines()
615
616 lines = [x for x in lines if x.strip() not in configs]
617
618 with open(os.path.join('scripts', 'config_whitelist.txt'), 'w') as f:
619 f.write(''.join(lines))
620
Chris Packham0e6deff2017-05-02 21:30:48 +1200621def find_matching(patterns, line):
622 for pat in patterns:
623 if pat.search(line):
624 return True
625 return False
626
627def cleanup_readme(configs, options):
628 """Delete config description in README
629
630 Arguments:
631 configs: A list of CONFIGs to remove.
632 options: option flags.
633 """
634 if not confirm(options, 'Clean up README?'):
635 return
636
637 patterns = []
638 for config in configs:
639 patterns.append(re.compile(r'^\s+%s' % config))
640
641 with open('README') as f:
642 lines = f.readlines()
643
644 found = False
645 newlines = []
646 for line in lines:
647 if not found:
648 found = find_matching(patterns, line)
649 if found:
650 continue
651
652 if found and re.search(r'^\s+CONFIG', line):
653 found = False
654
655 if not found:
656 newlines.append(line)
657
658 with open('README', 'w') as f:
659 f.write(''.join(newlines))
660
Chris Packham9d5274f2017-05-02 21:30:47 +1200661
Masahiro Yamadab6160812015-05-20 11:36:07 +0900662### classes ###
Masahiro Yamadacefaa582016-05-19 15:51:55 +0900663class Progress:
664
665 """Progress Indicator"""
666
667 def __init__(self, total):
668 """Create a new progress indicator.
669
670 Arguments:
671 total: A number of defconfig files to process.
672 """
673 self.current = 0
674 self.total = total
675
676 def inc(self):
677 """Increment the number of processed defconfig files."""
678
679 self.current += 1
680
681 def show(self):
682 """Display the progress."""
683 print ' %d defconfigs out of %d\r' % (self.current, self.total),
684 sys.stdout.flush()
685
Masahiro Yamadab6160812015-05-20 11:36:07 +0900686class KconfigParser:
687
688 """A parser of .config and include/autoconf.mk."""
689
690 re_arch = re.compile(r'CONFIG_SYS_ARCH="(.*)"')
691 re_cpu = re.compile(r'CONFIG_SYS_CPU="(.*)"')
692
Masahiro Yamada69e2bbc2016-05-19 15:52:01 +0900693 def __init__(self, configs, options, build_dir):
Masahiro Yamadab6160812015-05-20 11:36:07 +0900694 """Create a new parser.
695
696 Arguments:
Masahiro Yamadab80a6672016-05-19 15:51:57 +0900697 configs: A list of CONFIGs to move.
Masahiro Yamadab6160812015-05-20 11:36:07 +0900698 options: option flags.
699 build_dir: Build directory.
700 """
Masahiro Yamadab80a6672016-05-19 15:51:57 +0900701 self.configs = configs
Masahiro Yamadab6160812015-05-20 11:36:07 +0900702 self.options = options
Masahiro Yamada5393b612016-05-19 15:52:00 +0900703 self.dotconfig = os.path.join(build_dir, '.config')
704 self.autoconf = os.path.join(build_dir, 'include', 'autoconf.mk')
Masahiro Yamada6d139172016-08-22 22:18:22 +0900705 self.spl_autoconf = os.path.join(build_dir, 'spl', 'include',
706 'autoconf.mk')
Masahiro Yamada5393b612016-05-19 15:52:00 +0900707 self.config_autoconf = os.path.join(build_dir, 'include', 'config',
708 'auto.conf')
Masahiro Yamada07f98522016-05-19 15:52:06 +0900709 self.defconfig = os.path.join(build_dir, 'defconfig')
Masahiro Yamadab6160812015-05-20 11:36:07 +0900710
711 def get_cross_compile(self):
712 """Parse .config file and return CROSS_COMPILE.
713
714 Returns:
715 A string storing the compiler prefix for the architecture.
Masahiro Yamadac4d76eb2016-05-19 15:51:53 +0900716 Return a NULL string for architectures that do not require
717 compiler prefix (Sandbox and native build is the case).
718 Return None if the specified compiler is missing in your PATH.
719 Caller should distinguish '' and None.
Masahiro Yamadab6160812015-05-20 11:36:07 +0900720 """
721 arch = ''
722 cpu = ''
Masahiro Yamada5393b612016-05-19 15:52:00 +0900723 for line in open(self.dotconfig):
Masahiro Yamadab6160812015-05-20 11:36:07 +0900724 m = self.re_arch.match(line)
725 if m:
726 arch = m.group(1)
727 continue
728 m = self.re_cpu.match(line)
729 if m:
730 cpu = m.group(1)
731
Masahiro Yamadac4d76eb2016-05-19 15:51:53 +0900732 if not arch:
733 return None
Masahiro Yamadab6160812015-05-20 11:36:07 +0900734
735 # fix-up for aarch64
736 if arch == 'arm' and cpu == 'armv8':
737 arch = 'aarch64'
738
Masahiro Yamadac4d76eb2016-05-19 15:51:53 +0900739 return CROSS_COMPILE.get(arch, None)
Masahiro Yamadab6160812015-05-20 11:36:07 +0900740
Masahiro Yamadab80a6672016-05-19 15:51:57 +0900741 def parse_one_config(self, config, dotconfig_lines, autoconf_lines):
Masahiro Yamadab6160812015-05-20 11:36:07 +0900742 """Parse .config, defconfig, include/autoconf.mk for one config.
743
744 This function looks for the config options in the lines from
745 defconfig, .config, and include/autoconf.mk in order to decide
746 which action should be taken for this defconfig.
747
748 Arguments:
Masahiro Yamadab80a6672016-05-19 15:51:57 +0900749 config: CONFIG name to parse.
Masahiro Yamada5643d6e2016-05-19 15:51:56 +0900750 dotconfig_lines: lines from the .config file.
Masahiro Yamadab6160812015-05-20 11:36:07 +0900751 autoconf_lines: lines from the include/autoconf.mk file.
752
753 Returns:
754 A tupple of the action for this defconfig and the line
755 matched for the config.
756 """
Masahiro Yamadab6160812015-05-20 11:36:07 +0900757 not_set = '# %s is not set' % config
758
Masahiro Yamadab6160812015-05-20 11:36:07 +0900759 for line in autoconf_lines:
760 line = line.rstrip()
761 if line.startswith(config + '='):
Masahiro Yamada5643d6e2016-05-19 15:51:56 +0900762 new_val = line
Masahiro Yamadab6160812015-05-20 11:36:07 +0900763 break
Masahiro Yamadab6160812015-05-20 11:36:07 +0900764 else:
Masahiro Yamada5643d6e2016-05-19 15:51:56 +0900765 new_val = not_set
766
Masahiro Yamada35204d92016-08-22 22:18:21 +0900767 for line in dotconfig_lines:
768 line = line.rstrip()
769 if line.startswith(config + '=') or line == not_set:
770 old_val = line
771 break
772 else:
773 if new_val == not_set:
774 return (ACTION_NO_ENTRY, config)
775 else:
776 return (ACTION_NO_ENTRY_WARN, config)
777
Masahiro Yamada5643d6e2016-05-19 15:51:56 +0900778 # If this CONFIG is neither bool nor trisate
779 if old_val[-2:] != '=y' and old_val[-2:] != '=m' and old_val != not_set:
780 # tools/scripts/define2mk.sed changes '1' to 'y'.
781 # This is a problem if the CONFIG is int type.
782 # Check the type in Kconfig and handle it correctly.
783 if new_val[-2:] == '=y':
784 new_val = new_val[:-1] + '1'
785
Masahiro Yamadab48387f2016-06-15 14:33:50 +0900786 return (ACTION_NO_CHANGE if old_val == new_val else ACTION_MOVE,
787 new_val)
Masahiro Yamadab6160812015-05-20 11:36:07 +0900788
Masahiro Yamada465b7c02016-05-19 15:52:02 +0900789 def update_dotconfig(self):
Masahiro Yamada7c0d9d22016-05-19 15:51:50 +0900790 """Parse files for the config options and update the .config.
Masahiro Yamadab6160812015-05-20 11:36:07 +0900791
Masahiro Yamada5643d6e2016-05-19 15:51:56 +0900792 This function parses the generated .config and include/autoconf.mk
793 searching the target options.
Masahiro Yamada7c0d9d22016-05-19 15:51:50 +0900794 Move the config option(s) to the .config as needed.
Masahiro Yamadab6160812015-05-20 11:36:07 +0900795
796 Arguments:
797 defconfig: defconfig name.
Masahiro Yamada69e2bbc2016-05-19 15:52:01 +0900798
799 Returns:
Masahiro Yamada263d1372016-05-19 15:52:04 +0900800 Return a tuple of (updated flag, log string).
801 The "updated flag" is True if the .config was updated, False
802 otherwise. The "log string" shows what happend to the .config.
Masahiro Yamadab6160812015-05-20 11:36:07 +0900803 """
804
Masahiro Yamadab6160812015-05-20 11:36:07 +0900805 results = []
Masahiro Yamada263d1372016-05-19 15:52:04 +0900806 updated = False
Masahiro Yamada35204d92016-08-22 22:18:21 +0900807 suspicious = False
Masahiro Yamada6d139172016-08-22 22:18:22 +0900808 rm_files = [self.config_autoconf, self.autoconf]
809
810 if self.options.spl:
811 if os.path.exists(self.spl_autoconf):
812 autoconf_path = self.spl_autoconf
813 rm_files.append(self.spl_autoconf)
814 else:
815 for f in rm_files:
816 os.remove(f)
817 return (updated, suspicious,
818 color_text(self.options.color, COLOR_BROWN,
819 "SPL is not enabled. Skipped.") + '\n')
820 else:
821 autoconf_path = self.autoconf
Masahiro Yamadab6160812015-05-20 11:36:07 +0900822
Masahiro Yamada5393b612016-05-19 15:52:00 +0900823 with open(self.dotconfig) as f:
Masahiro Yamada5643d6e2016-05-19 15:51:56 +0900824 dotconfig_lines = f.readlines()
Masahiro Yamadab6160812015-05-20 11:36:07 +0900825
Masahiro Yamada6d139172016-08-22 22:18:22 +0900826 with open(autoconf_path) as f:
Masahiro Yamadab6160812015-05-20 11:36:07 +0900827 autoconf_lines = f.readlines()
828
Masahiro Yamadab80a6672016-05-19 15:51:57 +0900829 for config in self.configs:
830 result = self.parse_one_config(config, dotconfig_lines,
Joe Hershberger166edec2015-05-19 13:21:17 -0500831 autoconf_lines)
Masahiro Yamadab6160812015-05-20 11:36:07 +0900832 results.append(result)
833
834 log = ''
835
836 for (action, value) in results:
837 if action == ACTION_MOVE:
838 actlog = "Move '%s'" % value
839 log_color = COLOR_LIGHT_GREEN
Masahiro Yamada5643d6e2016-05-19 15:51:56 +0900840 elif action == ACTION_NO_ENTRY:
841 actlog = "%s is not defined in Kconfig. Do nothing." % value
Masahiro Yamadab6160812015-05-20 11:36:07 +0900842 log_color = COLOR_LIGHT_BLUE
Masahiro Yamada35204d92016-08-22 22:18:21 +0900843 elif action == ACTION_NO_ENTRY_WARN:
844 actlog = "%s is not defined in Kconfig (suspicious). Do nothing." % value
845 log_color = COLOR_YELLOW
846 suspicious = True
Masahiro Yamada5643d6e2016-05-19 15:51:56 +0900847 elif action == ACTION_NO_CHANGE:
848 actlog = "'%s' is the same as the define in Kconfig. Do nothing." \
849 % value
Masahiro Yamadab6160812015-05-20 11:36:07 +0900850 log_color = COLOR_LIGHT_PURPLE
Masahiro Yamada6d139172016-08-22 22:18:22 +0900851 elif action == ACTION_SPL_NOT_EXIST:
852 actlog = "SPL is not enabled for this defconfig. Skip."
853 log_color = COLOR_PURPLE
Masahiro Yamadab6160812015-05-20 11:36:07 +0900854 else:
855 sys.exit("Internal Error. This should not happen.")
856
Masahiro Yamada465b7c02016-05-19 15:52:02 +0900857 log += color_text(self.options.color, log_color, actlog) + '\n'
Masahiro Yamadab6160812015-05-20 11:36:07 +0900858
Masahiro Yamada5393b612016-05-19 15:52:00 +0900859 with open(self.dotconfig, 'a') as f:
Masahiro Yamada953d93b2016-05-19 15:51:49 +0900860 for (action, value) in results:
861 if action == ACTION_MOVE:
862 f.write(value + '\n')
Masahiro Yamada263d1372016-05-19 15:52:04 +0900863 updated = True
Masahiro Yamadab6160812015-05-20 11:36:07 +0900864
Masahiro Yamada07f98522016-05-19 15:52:06 +0900865 self.results = results
Masahiro Yamada6d139172016-08-22 22:18:22 +0900866 for f in rm_files:
867 os.remove(f)
Masahiro Yamadab6160812015-05-20 11:36:07 +0900868
Masahiro Yamada35204d92016-08-22 22:18:21 +0900869 return (updated, suspicious, log)
Masahiro Yamada69e2bbc2016-05-19 15:52:01 +0900870
Masahiro Yamada07f98522016-05-19 15:52:06 +0900871 def check_defconfig(self):
872 """Check the defconfig after savedefconfig
873
874 Returns:
875 Return additional log if moved CONFIGs were removed again by
876 'make savedefconfig'.
877 """
878
879 log = ''
880
881 with open(self.defconfig) as f:
882 defconfig_lines = f.readlines()
883
884 for (action, value) in self.results:
885 if action != ACTION_MOVE:
886 continue
887 if not value + '\n' in defconfig_lines:
888 log += color_text(self.options.color, COLOR_YELLOW,
889 "'%s' was removed by savedefconfig.\n" %
890 value)
891
892 return log
893
Masahiro Yamadab6160812015-05-20 11:36:07 +0900894class Slot:
895
896 """A slot to store a subprocess.
897
898 Each instance of this class handles one subprocess.
899 This class is useful to control multiple threads
900 for faster processing.
901 """
902
Joe Hershbergerb1a570f2016-06-10 14:53:32 -0500903 def __init__(self, configs, options, progress, devnull, make_cmd, reference_src_dir):
Masahiro Yamadab6160812015-05-20 11:36:07 +0900904 """Create a new process slot.
905
906 Arguments:
Masahiro Yamadab80a6672016-05-19 15:51:57 +0900907 configs: A list of CONFIGs to move.
Masahiro Yamadab6160812015-05-20 11:36:07 +0900908 options: option flags.
Masahiro Yamadacefaa582016-05-19 15:51:55 +0900909 progress: A progress indicator.
Masahiro Yamadab6160812015-05-20 11:36:07 +0900910 devnull: A file object of '/dev/null'.
911 make_cmd: command name of GNU Make.
Joe Hershbergerb1a570f2016-06-10 14:53:32 -0500912 reference_src_dir: Determine the true starting config state from this
913 source tree.
Masahiro Yamadab6160812015-05-20 11:36:07 +0900914 """
915 self.options = options
Masahiro Yamadacefaa582016-05-19 15:51:55 +0900916 self.progress = progress
Masahiro Yamadab6160812015-05-20 11:36:07 +0900917 self.build_dir = tempfile.mkdtemp()
918 self.devnull = devnull
919 self.make_cmd = (make_cmd, 'O=' + self.build_dir)
Joe Hershbergerb1a570f2016-06-10 14:53:32 -0500920 self.reference_src_dir = reference_src_dir
Masahiro Yamada69e2bbc2016-05-19 15:52:01 +0900921 self.parser = KconfigParser(configs, options, self.build_dir)
Masahiro Yamadab6160812015-05-20 11:36:07 +0900922 self.state = STATE_IDLE
Masahiro Yamada1271b672016-08-22 22:18:20 +0900923 self.failed_boards = set()
924 self.suspicious_boards = set()
Masahiro Yamadab6160812015-05-20 11:36:07 +0900925
926 def __del__(self):
927 """Delete the working directory
928
929 This function makes sure the temporary directory is cleaned away
930 even if Python suddenly dies due to error. It should be done in here
Joe Hershberger640de872016-06-10 14:53:29 -0500931 because it is guaranteed the destructor is always invoked when the
Masahiro Yamadab6160812015-05-20 11:36:07 +0900932 instance of the class gets unreferenced.
933
934 If the subprocess is still running, wait until it finishes.
935 """
936 if self.state != STATE_IDLE:
937 while self.ps.poll() == None:
938 pass
939 shutil.rmtree(self.build_dir)
940
Masahiro Yamadacefaa582016-05-19 15:51:55 +0900941 def add(self, defconfig):
Masahiro Yamadab6160812015-05-20 11:36:07 +0900942 """Assign a new subprocess for defconfig and add it to the slot.
943
944 If the slot is vacant, create a new subprocess for processing the
945 given defconfig and add it to the slot. Just returns False if
946 the slot is occupied (i.e. the current subprocess is still running).
947
948 Arguments:
949 defconfig: defconfig name.
950
951 Returns:
952 Return True on success or False on failure
953 """
954 if self.state != STATE_IDLE:
955 return False
Masahiro Yamadacb256cb2016-06-08 11:47:37 +0900956
Masahiro Yamadab6160812015-05-20 11:36:07 +0900957 self.defconfig = defconfig
Masahiro Yamada465b7c02016-05-19 15:52:02 +0900958 self.log = ''
Masahiro Yamada8f5256a2016-06-15 14:33:52 +0900959 self.current_src_dir = self.reference_src_dir
Masahiro Yamadacb256cb2016-06-08 11:47:37 +0900960 self.do_defconfig()
Masahiro Yamadab6160812015-05-20 11:36:07 +0900961 return True
962
963 def poll(self):
964 """Check the status of the subprocess and handle it as needed.
965
966 Returns True if the slot is vacant (i.e. in idle state).
967 If the configuration is successfully finished, assign a new
968 subprocess to build include/autoconf.mk.
969 If include/autoconf.mk is generated, invoke the parser to
Masahiro Yamada263d1372016-05-19 15:52:04 +0900970 parse the .config and the include/autoconf.mk, moving
971 config options to the .config as needed.
972 If the .config was updated, run "make savedefconfig" to sync
973 it, update the original defconfig, and then set the slot back
974 to the idle state.
Masahiro Yamadab6160812015-05-20 11:36:07 +0900975
976 Returns:
977 Return True if the subprocess is terminated, False otherwise
978 """
979 if self.state == STATE_IDLE:
980 return True
981
982 if self.ps.poll() == None:
983 return False
984
985 if self.ps.poll() != 0:
Masahiro Yamadacb256cb2016-06-08 11:47:37 +0900986 self.handle_error()
987 elif self.state == STATE_DEFCONFIG:
Masahiro Yamada8f5256a2016-06-15 14:33:52 +0900988 if self.reference_src_dir and not self.current_src_dir:
Joe Hershbergerb1a570f2016-06-10 14:53:32 -0500989 self.do_savedefconfig()
990 else:
991 self.do_autoconf()
Masahiro Yamadacb256cb2016-06-08 11:47:37 +0900992 elif self.state == STATE_AUTOCONF:
Masahiro Yamada8f5256a2016-06-15 14:33:52 +0900993 if self.current_src_dir:
994 self.current_src_dir = None
Joe Hershbergerb1a570f2016-06-10 14:53:32 -0500995 self.do_defconfig()
996 else:
997 self.do_savedefconfig()
Masahiro Yamadacb256cb2016-06-08 11:47:37 +0900998 elif self.state == STATE_SAVEDEFCONFIG:
999 self.update_defconfig()
1000 else:
1001 sys.exit("Internal Error. This should not happen.")
Masahiro Yamadab6160812015-05-20 11:36:07 +09001002
Masahiro Yamadacb256cb2016-06-08 11:47:37 +09001003 return True if self.state == STATE_IDLE else False
Joe Hershberger166edec2015-05-19 13:21:17 -05001004
Masahiro Yamadacb256cb2016-06-08 11:47:37 +09001005 def handle_error(self):
1006 """Handle error cases."""
Masahiro Yamada83c17672016-05-19 15:52:08 +09001007
Masahiro Yamadacb256cb2016-06-08 11:47:37 +09001008 self.log += color_text(self.options.color, COLOR_LIGHT_RED,
1009 "Failed to process.\n")
1010 if self.options.verbose:
1011 self.log += color_text(self.options.color, COLOR_LIGHT_CYAN,
1012 self.ps.stderr.read())
1013 self.finish(False)
Joe Hershberger166edec2015-05-19 13:21:17 -05001014
Masahiro Yamadacb256cb2016-06-08 11:47:37 +09001015 def do_defconfig(self):
1016 """Run 'make <board>_defconfig' to create the .config file."""
Masahiro Yamada0f6beda2016-05-19 15:52:07 +09001017
Masahiro Yamadacb256cb2016-06-08 11:47:37 +09001018 cmd = list(self.make_cmd)
1019 cmd.append(self.defconfig)
1020 self.ps = subprocess.Popen(cmd, stdout=self.devnull,
Masahiro Yamada8f5256a2016-06-15 14:33:52 +09001021 stderr=subprocess.PIPE,
1022 cwd=self.current_src_dir)
Masahiro Yamadacb256cb2016-06-08 11:47:37 +09001023 self.state = STATE_DEFCONFIG
Masahiro Yamada0f6beda2016-05-19 15:52:07 +09001024
Masahiro Yamadacb256cb2016-06-08 11:47:37 +09001025 def do_autoconf(self):
1026 """Run 'make include/config/auto.conf'."""
Masahiro Yamadab6160812015-05-20 11:36:07 +09001027
Joe Hershberger11b02702015-05-19 13:21:23 -05001028 self.cross_compile = self.parser.get_cross_compile()
Masahiro Yamadac4d76eb2016-05-19 15:51:53 +09001029 if self.cross_compile is None:
Masahiro Yamada465b7c02016-05-19 15:52:02 +09001030 self.log += color_text(self.options.color, COLOR_YELLOW,
1031 "Compiler is missing. Do nothing.\n")
Masahiro Yamada274a5ee2016-05-19 15:52:03 +09001032 self.finish(False)
Masahiro Yamadacb256cb2016-06-08 11:47:37 +09001033 return
Masahiro Yamadac4d76eb2016-05-19 15:51:53 +09001034
Masahiro Yamadab6160812015-05-20 11:36:07 +09001035 cmd = list(self.make_cmd)
Joe Hershberger11b02702015-05-19 13:21:23 -05001036 if self.cross_compile:
1037 cmd.append('CROSS_COMPILE=%s' % self.cross_compile)
Joe Hershberger765442b2015-05-19 13:21:18 -05001038 cmd.append('KCONFIG_IGNORE_DUPLICATES=1')
Masahiro Yamadab6160812015-05-20 11:36:07 +09001039 cmd.append('include/config/auto.conf')
Joe Hershberger11b02702015-05-19 13:21:23 -05001040 self.ps = subprocess.Popen(cmd, stdout=self.devnull,
Masahiro Yamada8f5256a2016-06-15 14:33:52 +09001041 stderr=subprocess.PIPE,
1042 cwd=self.current_src_dir)
Masahiro Yamadab6160812015-05-20 11:36:07 +09001043 self.state = STATE_AUTOCONF
Masahiro Yamadacb256cb2016-06-08 11:47:37 +09001044
1045 def do_savedefconfig(self):
1046 """Update the .config and run 'make savedefconfig'."""
1047
Masahiro Yamada35204d92016-08-22 22:18:21 +09001048 (updated, suspicious, log) = self.parser.update_dotconfig()
1049 if suspicious:
1050 self.suspicious_boards.add(self.defconfig)
Masahiro Yamadacb256cb2016-06-08 11:47:37 +09001051 self.log += log
1052
1053 if not self.options.force_sync and not updated:
1054 self.finish(True)
1055 return
1056 if updated:
1057 self.log += color_text(self.options.color, COLOR_LIGHT_GREEN,
1058 "Syncing by savedefconfig...\n")
1059 else:
1060 self.log += "Syncing by savedefconfig (forced by option)...\n"
1061
1062 cmd = list(self.make_cmd)
1063 cmd.append('savedefconfig')
1064 self.ps = subprocess.Popen(cmd, stdout=self.devnull,
1065 stderr=subprocess.PIPE)
1066 self.state = STATE_SAVEDEFCONFIG
1067
1068 def update_defconfig(self):
1069 """Update the input defconfig and go back to the idle state."""
1070
Masahiro Yamada3c9bfea2016-06-15 14:33:54 +09001071 log = self.parser.check_defconfig()
1072 if log:
Masahiro Yamada1271b672016-08-22 22:18:20 +09001073 self.suspicious_boards.add(self.defconfig)
Masahiro Yamada3c9bfea2016-06-15 14:33:54 +09001074 self.log += log
Masahiro Yamadacb256cb2016-06-08 11:47:37 +09001075 orig_defconfig = os.path.join('configs', self.defconfig)
1076 new_defconfig = os.path.join(self.build_dir, 'defconfig')
1077 updated = not filecmp.cmp(orig_defconfig, new_defconfig)
1078
1079 if updated:
Joe Hershberger93f1c2d2016-06-10 14:53:30 -05001080 self.log += color_text(self.options.color, COLOR_LIGHT_BLUE,
Masahiro Yamadacb256cb2016-06-08 11:47:37 +09001081 "defconfig was updated.\n")
1082
1083 if not self.options.dry_run and updated:
1084 shutil.move(new_defconfig, orig_defconfig)
1085 self.finish(True)
Masahiro Yamadab6160812015-05-20 11:36:07 +09001086
Masahiro Yamada274a5ee2016-05-19 15:52:03 +09001087 def finish(self, success):
1088 """Display log along with progress and go to the idle state.
Masahiro Yamada465b7c02016-05-19 15:52:02 +09001089
1090 Arguments:
Masahiro Yamada274a5ee2016-05-19 15:52:03 +09001091 success: Should be True when the defconfig was processed
1092 successfully, or False when it fails.
Masahiro Yamada465b7c02016-05-19 15:52:02 +09001093 """
1094 # output at least 30 characters to hide the "* defconfigs out of *".
1095 log = self.defconfig.ljust(30) + '\n'
1096
1097 log += '\n'.join([ ' ' + s for s in self.log.split('\n') ])
1098 # Some threads are running in parallel.
1099 # Print log atomically to not mix up logs from different threads.
Masahiro Yamada274a5ee2016-05-19 15:52:03 +09001100 print >> (sys.stdout if success else sys.stderr), log
1101
1102 if not success:
1103 if self.options.exit_on_error:
1104 sys.exit("Exit on error.")
1105 # If --exit-on-error flag is not set, skip this board and continue.
1106 # Record the failed board.
Masahiro Yamada1271b672016-08-22 22:18:20 +09001107 self.failed_boards.add(self.defconfig)
Masahiro Yamada274a5ee2016-05-19 15:52:03 +09001108
Masahiro Yamada465b7c02016-05-19 15:52:02 +09001109 self.progress.inc()
1110 self.progress.show()
Masahiro Yamada274a5ee2016-05-19 15:52:03 +09001111 self.state = STATE_IDLE
Masahiro Yamada465b7c02016-05-19 15:52:02 +09001112
Masahiro Yamadab6160812015-05-20 11:36:07 +09001113 def get_failed_boards(self):
Masahiro Yamada1271b672016-08-22 22:18:20 +09001114 """Returns a set of failed boards (defconfigs) in this slot.
Masahiro Yamadab6160812015-05-20 11:36:07 +09001115 """
1116 return self.failed_boards
1117
Masahiro Yamada3c9bfea2016-06-15 14:33:54 +09001118 def get_suspicious_boards(self):
Masahiro Yamada1271b672016-08-22 22:18:20 +09001119 """Returns a set of boards (defconfigs) with possible misconversion.
Masahiro Yamada3c9bfea2016-06-15 14:33:54 +09001120 """
Masahiro Yamada35204d92016-08-22 22:18:21 +09001121 return self.suspicious_boards - self.failed_boards
Masahiro Yamada3c9bfea2016-06-15 14:33:54 +09001122
Masahiro Yamadab6160812015-05-20 11:36:07 +09001123class Slots:
1124
1125 """Controller of the array of subprocess slots."""
1126
Joe Hershbergerb1a570f2016-06-10 14:53:32 -05001127 def __init__(self, configs, options, progress, reference_src_dir):
Masahiro Yamadab6160812015-05-20 11:36:07 +09001128 """Create a new slots controller.
1129
1130 Arguments:
Masahiro Yamadab80a6672016-05-19 15:51:57 +09001131 configs: A list of CONFIGs to move.
Masahiro Yamadab6160812015-05-20 11:36:07 +09001132 options: option flags.
Masahiro Yamadacefaa582016-05-19 15:51:55 +09001133 progress: A progress indicator.
Joe Hershbergerb1a570f2016-06-10 14:53:32 -05001134 reference_src_dir: Determine the true starting config state from this
1135 source tree.
Masahiro Yamadab6160812015-05-20 11:36:07 +09001136 """
1137 self.options = options
1138 self.slots = []
1139 devnull = get_devnull()
1140 make_cmd = get_make_cmd()
1141 for i in range(options.jobs):
Masahiro Yamadab80a6672016-05-19 15:51:57 +09001142 self.slots.append(Slot(configs, options, progress, devnull,
Joe Hershbergerb1a570f2016-06-10 14:53:32 -05001143 make_cmd, reference_src_dir))
Masahiro Yamadab6160812015-05-20 11:36:07 +09001144
Masahiro Yamadacefaa582016-05-19 15:51:55 +09001145 def add(self, defconfig):
Masahiro Yamadab6160812015-05-20 11:36:07 +09001146 """Add a new subprocess if a vacant slot is found.
1147
1148 Arguments:
1149 defconfig: defconfig name to be put into.
1150
1151 Returns:
1152 Return True on success or False on failure
1153 """
1154 for slot in self.slots:
Masahiro Yamadacefaa582016-05-19 15:51:55 +09001155 if slot.add(defconfig):
Masahiro Yamadab6160812015-05-20 11:36:07 +09001156 return True
1157 return False
1158
1159 def available(self):
1160 """Check if there is a vacant slot.
1161
1162 Returns:
1163 Return True if at lease one vacant slot is found, False otherwise.
1164 """
1165 for slot in self.slots:
1166 if slot.poll():
1167 return True
1168 return False
1169
1170 def empty(self):
1171 """Check if all slots are vacant.
1172
1173 Returns:
1174 Return True if all the slots are vacant, False otherwise.
1175 """
1176 ret = True
1177 for slot in self.slots:
1178 if not slot.poll():
1179 ret = False
1180 return ret
1181
1182 def show_failed_boards(self):
1183 """Display all of the failed boards (defconfigs)."""
Masahiro Yamada1271b672016-08-22 22:18:20 +09001184 boards = set()
Masahiro Yamada0153f032016-06-15 14:33:53 +09001185 output_file = 'moveconfig.failed'
Masahiro Yamadab6160812015-05-20 11:36:07 +09001186
1187 for slot in self.slots:
Masahiro Yamada1271b672016-08-22 22:18:20 +09001188 boards |= slot.get_failed_boards()
Masahiro Yamadab6160812015-05-20 11:36:07 +09001189
Masahiro Yamada0153f032016-06-15 14:33:53 +09001190 if boards:
1191 boards = '\n'.join(boards) + '\n'
1192 msg = "The following boards were not processed due to error:\n"
1193 msg += boards
1194 msg += "(the list has been saved in %s)\n" % output_file
1195 print >> sys.stderr, color_text(self.options.color, COLOR_LIGHT_RED,
1196 msg)
Masahiro Yamadab6160812015-05-20 11:36:07 +09001197
Masahiro Yamada0153f032016-06-15 14:33:53 +09001198 with open(output_file, 'w') as f:
1199 f.write(boards)
Joe Hershbergerdade12e2015-05-19 13:21:22 -05001200
Masahiro Yamada3c9bfea2016-06-15 14:33:54 +09001201 def show_suspicious_boards(self):
1202 """Display all boards (defconfigs) with possible misconversion."""
Masahiro Yamada1271b672016-08-22 22:18:20 +09001203 boards = set()
Masahiro Yamada3c9bfea2016-06-15 14:33:54 +09001204 output_file = 'moveconfig.suspicious'
1205
1206 for slot in self.slots:
Masahiro Yamada1271b672016-08-22 22:18:20 +09001207 boards |= slot.get_suspicious_boards()
Masahiro Yamada3c9bfea2016-06-15 14:33:54 +09001208
1209 if boards:
1210 boards = '\n'.join(boards) + '\n'
1211 msg = "The following boards might have been converted incorrectly.\n"
1212 msg += "It is highly recommended to check them manually:\n"
1213 msg += boards
1214 msg += "(the list has been saved in %s)\n" % output_file
1215 print >> sys.stderr, color_text(self.options.color, COLOR_YELLOW,
1216 msg)
1217
1218 with open(output_file, 'w') as f:
1219 f.write(boards)
1220
Masahiro Yamada2e74fee2016-06-15 14:33:51 +09001221class ReferenceSource:
1222
1223 """Reference source against which original configs should be parsed."""
1224
1225 def __init__(self, commit):
1226 """Create a reference source directory based on a specified commit.
1227
1228 Arguments:
1229 commit: commit to git-clone
1230 """
1231 self.src_dir = tempfile.mkdtemp()
1232 print "Cloning git repo to a separate work directory..."
1233 subprocess.check_output(['git', 'clone', os.getcwd(), '.'],
1234 cwd=self.src_dir)
1235 print "Checkout '%s' to build the original autoconf.mk." % \
1236 subprocess.check_output(['git', 'rev-parse', '--short', commit]).strip()
1237 subprocess.check_output(['git', 'checkout', commit],
1238 stderr=subprocess.STDOUT, cwd=self.src_dir)
Joe Hershbergerb1a570f2016-06-10 14:53:32 -05001239
1240 def __del__(self):
Masahiro Yamada2e74fee2016-06-15 14:33:51 +09001241 """Delete the reference source directory
Joe Hershbergerb1a570f2016-06-10 14:53:32 -05001242
1243 This function makes sure the temporary directory is cleaned away
1244 even if Python suddenly dies due to error. It should be done in here
1245 because it is guaranteed the destructor is always invoked when the
1246 instance of the class gets unreferenced.
1247 """
Masahiro Yamada2e74fee2016-06-15 14:33:51 +09001248 shutil.rmtree(self.src_dir)
Joe Hershbergerb1a570f2016-06-10 14:53:32 -05001249
Masahiro Yamada2e74fee2016-06-15 14:33:51 +09001250 def get_dir(self):
1251 """Return the absolute path to the reference source directory."""
1252
1253 return self.src_dir
Joe Hershbergerb1a570f2016-06-10 14:53:32 -05001254
Masahiro Yamadab80a6672016-05-19 15:51:57 +09001255def move_config(configs, options):
Masahiro Yamadab6160812015-05-20 11:36:07 +09001256 """Move config options to defconfig files.
1257
1258 Arguments:
Masahiro Yamadab80a6672016-05-19 15:51:57 +09001259 configs: A list of CONFIGs to move.
Masahiro Yamadab6160812015-05-20 11:36:07 +09001260 options: option flags
1261 """
Masahiro Yamadab80a6672016-05-19 15:51:57 +09001262 if len(configs) == 0:
Masahiro Yamada9566abd2016-05-19 15:52:09 +09001263 if options.force_sync:
1264 print 'No CONFIG is specified. You are probably syncing defconfigs.',
1265 else:
1266 print 'Neither CONFIG nor --force-sync is specified. Nothing will happen.',
1267 else:
1268 print 'Move ' + ', '.join(configs),
1269 print '(jobs: %d)\n' % options.jobs
Masahiro Yamadab6160812015-05-20 11:36:07 +09001270
Joe Hershbergerb1a570f2016-06-10 14:53:32 -05001271 if options.git_ref:
Masahiro Yamada2e74fee2016-06-15 14:33:51 +09001272 reference_src = ReferenceSource(options.git_ref)
1273 reference_src_dir = reference_src.get_dir()
1274 else:
Masahiro Yamada8f5256a2016-06-15 14:33:52 +09001275 reference_src_dir = None
Joe Hershbergerb1a570f2016-06-10 14:53:32 -05001276
Joe Hershbergerc6e043a2015-05-19 13:21:19 -05001277 if options.defconfigs:
Masahiro Yamada3984d6e2016-10-19 14:39:54 +09001278 defconfigs = get_matched_defconfigs(options.defconfigs)
Joe Hershbergerc6e043a2015-05-19 13:21:19 -05001279 else:
Masahiro Yamada58175e32016-07-25 19:15:28 +09001280 defconfigs = get_all_defconfigs()
Masahiro Yamadab6160812015-05-20 11:36:07 +09001281
Masahiro Yamadacefaa582016-05-19 15:51:55 +09001282 progress = Progress(len(defconfigs))
Joe Hershbergerb1a570f2016-06-10 14:53:32 -05001283 slots = Slots(configs, options, progress, reference_src_dir)
Masahiro Yamadab6160812015-05-20 11:36:07 +09001284
1285 # Main loop to process defconfig files:
1286 # Add a new subprocess into a vacant slot.
1287 # Sleep if there is no available slot.
Masahiro Yamadacefaa582016-05-19 15:51:55 +09001288 for defconfig in defconfigs:
1289 while not slots.add(defconfig):
Masahiro Yamadab6160812015-05-20 11:36:07 +09001290 while not slots.available():
1291 # No available slot: sleep for a while
1292 time.sleep(SLEEP_TIME)
1293
1294 # wait until all the subprocesses finish
1295 while not slots.empty():
1296 time.sleep(SLEEP_TIME)
1297
Joe Hershberger3fa1ab72015-05-19 13:21:25 -05001298 print ''
Masahiro Yamadab6160812015-05-20 11:36:07 +09001299 slots.show_failed_boards()
Masahiro Yamada3c9bfea2016-06-15 14:33:54 +09001300 slots.show_suspicious_boards()
Masahiro Yamadab6160812015-05-20 11:36:07 +09001301
Masahiro Yamadab6160812015-05-20 11:36:07 +09001302def main():
1303 try:
1304 cpu_count = multiprocessing.cpu_count()
1305 except NotImplementedError:
1306 cpu_count = 1
1307
1308 parser = optparse.OptionParser()
1309 # Add options here
1310 parser.add_option('-c', '--color', action='store_true', default=False,
1311 help='display the log in color')
Simon Glass8bf41c22016-09-12 23:18:21 -06001312 parser.add_option('-C', '--commit', action='store_true', default=False,
1313 help='Create a git commit for the operation')
Joe Hershbergerc6e043a2015-05-19 13:21:19 -05001314 parser.add_option('-d', '--defconfigs', type='string',
1315 help='a file containing a list of defconfigs to move')
Masahiro Yamadab6160812015-05-20 11:36:07 +09001316 parser.add_option('-n', '--dry-run', action='store_true', default=False,
1317 help='perform a trial run (show log with no changes)')
1318 parser.add_option('-e', '--exit-on-error', action='store_true',
1319 default=False,
1320 help='exit immediately on any error')
Masahiro Yamada83c17672016-05-19 15:52:08 +09001321 parser.add_option('-s', '--force-sync', action='store_true', default=False,
1322 help='force sync by savedefconfig')
Masahiro Yamada6d139172016-08-22 22:18:22 +09001323 parser.add_option('-S', '--spl', action='store_true', default=False,
1324 help='parse config options defined for SPL build')
Joe Hershberger23475932015-05-19 13:21:20 -05001325 parser.add_option('-H', '--headers-only', dest='cleanup_headers_only',
1326 action='store_true', default=False,
1327 help='only cleanup the headers')
Masahiro Yamadab6160812015-05-20 11:36:07 +09001328 parser.add_option('-j', '--jobs', type='int', default=cpu_count,
1329 help='the number of jobs to run simultaneously')
Joe Hershbergerb1a570f2016-06-10 14:53:32 -05001330 parser.add_option('-r', '--git-ref', type='string',
1331 help='the git ref to clone for building the autoconf.mk')
Simon Glass13e05a02016-09-12 23:18:20 -06001332 parser.add_option('-y', '--yes', action='store_true', default=False,
1333 help="respond 'yes' to any prompts")
Joe Hershberger808b63f2015-05-19 13:21:24 -05001334 parser.add_option('-v', '--verbose', action='store_true', default=False,
1335 help='show any build errors as boards are built')
Masahiro Yamadab903c4e2016-05-19 15:51:58 +09001336 parser.usage += ' CONFIG ...'
Masahiro Yamadab6160812015-05-20 11:36:07 +09001337
Masahiro Yamadab903c4e2016-05-19 15:51:58 +09001338 (options, configs) = parser.parse_args()
Masahiro Yamadab6160812015-05-20 11:36:07 +09001339
Masahiro Yamada9566abd2016-05-19 15:52:09 +09001340 if len(configs) == 0 and not options.force_sync:
Masahiro Yamadab6160812015-05-20 11:36:07 +09001341 parser.print_usage()
1342 sys.exit(1)
1343
Masahiro Yamadab903c4e2016-05-19 15:51:58 +09001344 # prefix the option name with CONFIG_ if missing
1345 configs = [ config if config.startswith('CONFIG_') else 'CONFIG_' + config
1346 for config in configs ]
Masahiro Yamadab6160812015-05-20 11:36:07 +09001347
Joe Hershberger23475932015-05-19 13:21:20 -05001348 check_top_directory()
1349
1350 if not options.cleanup_headers_only:
Masahiro Yamadad0a9d2a2016-07-25 19:15:23 +09001351 check_clean_directory()
1352 update_cross_compile(options.color)
Masahiro Yamadab80a6672016-05-19 15:51:57 +09001353 move_config(configs, options)
Joe Hershberger23475932015-05-19 13:21:20 -05001354
Masahiro Yamada9566abd2016-05-19 15:52:09 +09001355 if configs:
Masahiro Yamadaa1a4b092016-07-25 19:15:26 +09001356 cleanup_headers(configs, options)
Masahiro Yamadadce28de2016-07-25 19:15:29 +09001357 cleanup_extra_options(configs, options)
Chris Packham9d5274f2017-05-02 21:30:47 +12001358 cleanup_whitelist(configs, options)
Chris Packham0e6deff2017-05-02 21:30:48 +12001359 cleanup_readme(configs, options)
Masahiro Yamadab6160812015-05-20 11:36:07 +09001360
Simon Glass8bf41c22016-09-12 23:18:21 -06001361 if options.commit:
1362 subprocess.call(['git', 'add', '-u'])
1363 if configs:
1364 msg = 'Convert %s %sto Kconfig' % (configs[0],
1365 'et al ' if len(configs) > 1 else '')
1366 msg += ('\n\nThis converts the following to Kconfig:\n %s\n' %
1367 '\n '.join(configs))
1368 else:
1369 msg = 'configs: Resync with savedefconfig'
1370 msg += '\n\nRsync all defconfig files using moveconfig.py'
1371 subprocess.call(['git', 'commit', '-s', '-m', msg])
1372
Masahiro Yamadab6160812015-05-20 11:36:07 +09001373if __name__ == '__main__':
1374 main()