blob: e32104412e14e0b1c1f97a032be74f7aa6ac09dd [file] [log] [blame]
Masahiro Yamada5a27c732015-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 Yamadab6ef3932016-05-19 15:51:58 +090020First, you must edit the Kconfig to add the menu entries for the configs
Joe Hershberger96464ba2015-05-19 13:21:17 -050021you are moving.
22
Masahiro Yamadab6ef3932016-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 Yamada5a27c732015-05-20 11:36:07 +090026
Masahiro Yamadab6ef3932016-05-19 15:51:58 +090027 $ tools/moveconfig.py CONFIG_CMD_USB CONFIG_SYS_TEXT_BASE
Masahiro Yamada5a27c732015-05-20 11:36:07 +090028
Masahiro Yamadab6ef3932016-05-19 15:51:58 +090029The tool walks through all the defconfig files and move the given CONFIGs.
Masahiro Yamada5a27c732015-05-20 11:36:07 +090030
31The log is also displayed on the terminal.
32
Masahiro Yamada1d085562016-05-19 15:52:02 +090033The log is printed for each defconfig as follows:
Masahiro Yamada5a27c732015-05-20 11:36:07 +090034
Masahiro Yamada1d085562016-05-19 15:52:02 +090035<defconfig_name>
36 <action1>
37 <action2>
38 <action3>
39 ...
Masahiro Yamada5a27c732015-05-20 11:36:07 +090040
Masahiro Yamada1d085562016-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 Yamada5a27c732015-05-20 11:36:07 +090044It looks like one of the followings:
45
46 - Move 'CONFIG_... '
47 This config option was moved to the defconfig
48
Masahiro Yamadacc008292016-05-19 15:51:56 +090049 - CONFIG_... is not defined in Kconfig. Do nothing.
50 The entry for this CONFIG was not found in Kconfig.
51 There are two common cases:
52 - You forgot to create an entry for the CONFIG before running
53 this tool, or made a typo in a CONFIG passed to this tool.
54 - The entry was hidden due to unmet 'depends on'.
55 This is correct behavior.
Masahiro Yamada5a27c732015-05-20 11:36:07 +090056
Masahiro Yamadacc008292016-05-19 15:51:56 +090057 - 'CONFIG_...' is the same as the define in Kconfig. Do nothing.
58 The define in the config header matched the one in Kconfig.
59 We do not need to touch it.
Masahiro Yamada5a27c732015-05-20 11:36:07 +090060
61 - Undefined. Do nothing.
62 This config option was not found in the config header.
63 Nothing to do.
64
Masahiro Yamada90ed6cb2016-05-19 15:51:53 +090065 - Compiler is missing. Do nothing.
66 The compiler specified for this architecture was not found
67 in your PATH environment.
68 (If -e option is passed, the tool exits immediately.)
69
70 - Failed to process.
Masahiro Yamada5a27c732015-05-20 11:36:07 +090071 An error occurred during processing this defconfig. Skipped.
72 (If -e option is passed, the tool exits immediately on error.)
73
74Finally, you will be asked, Clean up headers? [y/n]:
75
76If you say 'y' here, the unnecessary config defines are removed
77from the config headers (include/configs/*.h).
78It just uses the regex method, so you should not rely on it.
79Just in case, please do 'git diff' to see what happened.
80
81
Masahiro Yamadab6ef3932016-05-19 15:51:58 +090082How does it work?
83-----------------
Masahiro Yamada5a27c732015-05-20 11:36:07 +090084
85This tool runs configuration and builds include/autoconf.mk for every
86defconfig. The config options defined in Kconfig appear in the .config
87file (unless they are hidden because of unmet dependency.)
88On the other hand, the config options defined by board headers are seen
89in include/autoconf.mk. The tool looks for the specified options in both
Masahiro Yamadab6ef3932016-05-19 15:51:58 +090090of them to decide the appropriate action for the options. If the given
91config option is found in the .config, but its value does not match the
92one from the board header, the config option in the .config is replaced
93with the define in the board header. Then, the .config is synced by
94"make savedefconfig" and the defconfig is updated with it.
Masahiro Yamada5a27c732015-05-20 11:36:07 +090095
96For faster processing, this tool handles multi-threading. It creates
97separate build directories where the out-of-tree build is run. The
98temporary build directories are automatically created and deleted as
99needed. The number of threads are chosen based on the number of the CPU
100cores of your system although you can change it via -j (--jobs) option.
101
102
103Toolchains
104----------
105
106Appropriate toolchain are necessary to generate include/autoconf.mk
107for all the architectures supported by U-Boot. Most of them are available
108at the kernel.org site, some are not provided by kernel.org.
109
110The default per-arch CROSS_COMPILE used by this tool is specified by
111the list below, CROSS_COMPILE. You may wish to update the list to
112use your own. Instead of modifying the list directly, you can give
113them via environments.
114
115
116Available options
117-----------------
118
119 -c, --color
120 Surround each portion of the log with escape sequences to display it
121 in color on the terminal.
122
Joe Hershberger91040e82015-05-19 13:21:19 -0500123 -d, --defconfigs
124 Specify a file containing a list of defconfigs to move
125
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900126 -n, --dry-run
Masahiro Yamadab6ef3932016-05-19 15:51:58 +0900127 Perform a trial run that does not make any changes. It is useful to
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900128 see what is going to happen before one actually runs it.
129
130 -e, --exit-on-error
131 Exit immediately if Make exits with a non-zero status while processing
132 a defconfig file.
133
Joe Hershberger2144f882015-05-19 13:21:20 -0500134 -H, --headers-only
135 Only cleanup the headers; skip the defconfig processing
136
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900137 -j, --jobs
138 Specify the number of threads to run simultaneously. If not specified,
139 the number of threads is the same as the number of CPU cores.
140
Joe Hershberger95bf9c72015-05-19 13:21:24 -0500141 -v, --verbose
142 Show any build errors as boards are built
143
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900144To see the complete list of supported options, run
145
146 $ tools/moveconfig.py -h
147
148"""
149
150import fnmatch
151import multiprocessing
152import optparse
153import os
154import re
155import shutil
156import subprocess
157import sys
158import tempfile
159import time
160
161SHOW_GNU_MAKE = 'scripts/show-gnu-make'
162SLEEP_TIME=0.03
163
164# Here is the list of cross-tools I use.
165# Most of them are available at kernel.org
166# (https://www.kernel.org/pub/tools/crosstool/files/bin/), except the followings:
167# arc: https://github.com/foss-for-synopsys-dwc-arc-processors/toolchain/releases
168# blackfin: http://sourceforge.net/projects/adi-toolchain/files/
Bin Meng4440ece2015-09-25 01:22:39 -0700169# nds32: http://osdk.andestech.com/packages/nds32le-linux-glibc-v1.tgz
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900170# nios2: https://sourcery.mentor.com/GNUToolchain/subscription42545
171# sh: http://sourcery.mentor.com/public/gnu_toolchain/sh-linux-gnu
Bin Menge8aebc42016-02-21 21:18:02 -0800172#
173# openrisc kernel.org toolchain is out of date, download latest one from
174# http://opencores.org/or1k/OpenRISC_GNU_tool_chain#Prebuilt_versions
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900175CROSS_COMPILE = {
176 'arc': 'arc-linux-',
177 'aarch64': 'aarch64-linux-',
178 'arm': 'arm-unknown-linux-gnueabi-',
179 'avr32': 'avr32-linux-',
180 'blackfin': 'bfin-elf-',
181 'm68k': 'm68k-linux-',
182 'microblaze': 'microblaze-linux-',
183 'mips': 'mips-linux-',
184 'nds32': 'nds32le-linux-',
185 'nios2': 'nios2-linux-gnu-',
Bin Menge8aebc42016-02-21 21:18:02 -0800186 'openrisc': 'or1k-elf-',
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900187 'powerpc': 'powerpc-linux-',
188 'sh': 'sh-linux-gnu-',
189 'sparc': 'sparc-linux-',
190 'x86': 'i386-linux-'
191}
192
193STATE_IDLE = 0
194STATE_DEFCONFIG = 1
195STATE_AUTOCONF = 2
Joe Hershberger96464ba2015-05-19 13:21:17 -0500196STATE_SAVEDEFCONFIG = 3
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900197
198ACTION_MOVE = 0
Masahiro Yamadacc008292016-05-19 15:51:56 +0900199ACTION_NO_ENTRY = 1
200ACTION_NO_CHANGE = 2
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900201
202COLOR_BLACK = '0;30'
203COLOR_RED = '0;31'
204COLOR_GREEN = '0;32'
205COLOR_BROWN = '0;33'
206COLOR_BLUE = '0;34'
207COLOR_PURPLE = '0;35'
208COLOR_CYAN = '0;36'
209COLOR_LIGHT_GRAY = '0;37'
210COLOR_DARK_GRAY = '1;30'
211COLOR_LIGHT_RED = '1;31'
212COLOR_LIGHT_GREEN = '1;32'
213COLOR_YELLOW = '1;33'
214COLOR_LIGHT_BLUE = '1;34'
215COLOR_LIGHT_PURPLE = '1;35'
216COLOR_LIGHT_CYAN = '1;36'
217COLOR_WHITE = '1;37'
218
219### helper functions ###
220def get_devnull():
221 """Get the file object of '/dev/null' device."""
222 try:
223 devnull = subprocess.DEVNULL # py3k
224 except AttributeError:
225 devnull = open(os.devnull, 'wb')
226 return devnull
227
228def check_top_directory():
229 """Exit if we are not at the top of source directory."""
230 for f in ('README', 'Licenses'):
231 if not os.path.exists(f):
232 sys.exit('Please run at the top of source directory.')
233
Masahiro Yamadabd63e5b2016-05-19 15:51:54 +0900234def check_clean_directory():
235 """Exit if the source tree is not clean."""
236 for f in ('.config', 'include/config'):
237 if os.path.exists(f):
238 sys.exit("source tree is not clean, please run 'make mrproper'")
239
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900240def get_make_cmd():
241 """Get the command name of GNU Make.
242
243 U-Boot needs GNU Make for building, but the command name is not
244 necessarily "make". (for example, "gmake" on FreeBSD).
245 Returns the most appropriate command name on your system.
246 """
247 process = subprocess.Popen([SHOW_GNU_MAKE], stdout=subprocess.PIPE)
248 ret = process.communicate()
249 if process.returncode:
250 sys.exit('GNU Make not found')
251 return ret[0].rstrip()
252
253def color_text(color_enabled, color, string):
254 """Return colored string."""
255 if color_enabled:
Masahiro Yamada1d085562016-05-19 15:52:02 +0900256 # LF should not be surrounded by the escape sequence.
257 # Otherwise, additional whitespace or line-feed might be printed.
258 return '\n'.join([ '\033[' + color + 'm' + s + '\033[0m' if s else ''
259 for s in string.split('\n') ])
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900260 else:
261 return string
262
Masahiro Yamada90ed6cb2016-05-19 15:51:53 +0900263def update_cross_compile(color_enabled):
Robert P. J. Day1cc0a9f2016-05-04 04:47:31 -0400264 """Update per-arch CROSS_COMPILE via environment variables
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900265
266 The default CROSS_COMPILE values are available
267 in the CROSS_COMPILE list above.
268
Robert P. J. Day1cc0a9f2016-05-04 04:47:31 -0400269 You can override them via environment variables
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900270 CROSS_COMPILE_{ARCH}.
271
272 For example, if you want to override toolchain prefixes
273 for ARM and PowerPC, you can do as follows in your shell:
274
275 export CROSS_COMPILE_ARM=...
276 export CROSS_COMPILE_POWERPC=...
Masahiro Yamada90ed6cb2016-05-19 15:51:53 +0900277
278 Then, this function checks if specified compilers really exist in your
279 PATH environment.
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900280 """
281 archs = []
282
283 for arch in os.listdir('arch'):
284 if os.path.exists(os.path.join('arch', arch, 'Makefile')):
285 archs.append(arch)
286
287 # arm64 is a special case
288 archs.append('aarch64')
289
290 for arch in archs:
291 env = 'CROSS_COMPILE_' + arch.upper()
292 cross_compile = os.environ.get(env)
Masahiro Yamada90ed6cb2016-05-19 15:51:53 +0900293 if not cross_compile:
294 cross_compile = CROSS_COMPILE.get(arch, '')
295
296 for path in os.environ["PATH"].split(os.pathsep):
297 gcc_path = os.path.join(path, cross_compile + 'gcc')
298 if os.path.isfile(gcc_path) and os.access(gcc_path, os.X_OK):
299 break
300 else:
301 print >> sys.stderr, color_text(color_enabled, COLOR_YELLOW,
302 'warning: %sgcc: not found in PATH. %s architecture boards will be skipped'
303 % (cross_compile, arch))
304 cross_compile = None
305
306 CROSS_COMPILE[arch] = cross_compile
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900307
308def cleanup_one_header(header_path, patterns, dry_run):
309 """Clean regex-matched lines away from a file.
310
311 Arguments:
312 header_path: path to the cleaned file.
313 patterns: list of regex patterns. Any lines matching to these
314 patterns are deleted.
315 dry_run: make no changes, but still display log.
316 """
317 with open(header_path) as f:
318 lines = f.readlines()
319
320 matched = []
321 for i, line in enumerate(lines):
322 for pattern in patterns:
323 m = pattern.search(line)
324 if m:
325 print '%s: %s: %s' % (header_path, i + 1, line),
326 matched.append(i)
327 break
328
329 if dry_run or not matched:
330 return
331
332 with open(header_path, 'w') as f:
333 for i, line in enumerate(lines):
334 if not i in matched:
335 f.write(line)
336
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900337def cleanup_headers(configs, dry_run):
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900338 """Delete config defines from board headers.
339
340 Arguments:
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900341 configs: A list of CONFIGs to remove.
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900342 dry_run: make no changes, but still display log.
343 """
344 while True:
345 choice = raw_input('Clean up headers? [y/n]: ').lower()
346 print choice
347 if choice == 'y' or choice == 'n':
348 break
349
350 if choice == 'n':
351 return
352
353 patterns = []
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900354 for config in configs:
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900355 patterns.append(re.compile(r'#\s*define\s+%s\W' % config))
356 patterns.append(re.compile(r'#\s*undef\s+%s\W' % config))
357
Joe Hershberger60727f52015-05-19 13:21:21 -0500358 for dir in 'include', 'arch', 'board':
359 for (dirpath, dirnames, filenames) in os.walk(dir):
360 for filename in filenames:
361 if not fnmatch.fnmatch(filename, '*~'):
362 cleanup_one_header(os.path.join(dirpath, filename),
363 patterns, dry_run)
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900364
365### classes ###
Masahiro Yamadac5e60fd2016-05-19 15:51:55 +0900366class Progress:
367
368 """Progress Indicator"""
369
370 def __init__(self, total):
371 """Create a new progress indicator.
372
373 Arguments:
374 total: A number of defconfig files to process.
375 """
376 self.current = 0
377 self.total = total
378
379 def inc(self):
380 """Increment the number of processed defconfig files."""
381
382 self.current += 1
383
384 def show(self):
385 """Display the progress."""
386 print ' %d defconfigs out of %d\r' % (self.current, self.total),
387 sys.stdout.flush()
388
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900389class KconfigParser:
390
391 """A parser of .config and include/autoconf.mk."""
392
393 re_arch = re.compile(r'CONFIG_SYS_ARCH="(.*)"')
394 re_cpu = re.compile(r'CONFIG_SYS_CPU="(.*)"')
395
Masahiro Yamada522e8dc2016-05-19 15:52:01 +0900396 def __init__(self, configs, options, build_dir):
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900397 """Create a new parser.
398
399 Arguments:
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900400 configs: A list of CONFIGs to move.
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900401 options: option flags.
402 build_dir: Build directory.
403 """
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900404 self.configs = configs
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900405 self.options = options
Masahiro Yamada1f169922016-05-19 15:52:00 +0900406 self.dotconfig = os.path.join(build_dir, '.config')
407 self.autoconf = os.path.join(build_dir, 'include', 'autoconf.mk')
408 self.config_autoconf = os.path.join(build_dir, 'include', 'config',
409 'auto.conf')
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900410
411 def get_cross_compile(self):
412 """Parse .config file and return CROSS_COMPILE.
413
414 Returns:
415 A string storing the compiler prefix for the architecture.
Masahiro Yamada90ed6cb2016-05-19 15:51:53 +0900416 Return a NULL string for architectures that do not require
417 compiler prefix (Sandbox and native build is the case).
418 Return None if the specified compiler is missing in your PATH.
419 Caller should distinguish '' and None.
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900420 """
421 arch = ''
422 cpu = ''
Masahiro Yamada1f169922016-05-19 15:52:00 +0900423 for line in open(self.dotconfig):
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900424 m = self.re_arch.match(line)
425 if m:
426 arch = m.group(1)
427 continue
428 m = self.re_cpu.match(line)
429 if m:
430 cpu = m.group(1)
431
Masahiro Yamada90ed6cb2016-05-19 15:51:53 +0900432 if not arch:
433 return None
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900434
435 # fix-up for aarch64
436 if arch == 'arm' and cpu == 'armv8':
437 arch = 'aarch64'
438
Masahiro Yamada90ed6cb2016-05-19 15:51:53 +0900439 return CROSS_COMPILE.get(arch, None)
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900440
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900441 def parse_one_config(self, config, dotconfig_lines, autoconf_lines):
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900442 """Parse .config, defconfig, include/autoconf.mk for one config.
443
444 This function looks for the config options in the lines from
445 defconfig, .config, and include/autoconf.mk in order to decide
446 which action should be taken for this defconfig.
447
448 Arguments:
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900449 config: CONFIG name to parse.
Masahiro Yamadacc008292016-05-19 15:51:56 +0900450 dotconfig_lines: lines from the .config file.
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900451 autoconf_lines: lines from the include/autoconf.mk file.
452
453 Returns:
454 A tupple of the action for this defconfig and the line
455 matched for the config.
456 """
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900457 not_set = '# %s is not set' % config
458
Masahiro Yamadacc008292016-05-19 15:51:56 +0900459 for line in dotconfig_lines:
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900460 line = line.rstrip()
461 if line.startswith(config + '=') or line == not_set:
Masahiro Yamadacc008292016-05-19 15:51:56 +0900462 old_val = line
463 break
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900464 else:
Masahiro Yamadacc008292016-05-19 15:51:56 +0900465 return (ACTION_NO_ENTRY, config)
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900466
467 for line in autoconf_lines:
468 line = line.rstrip()
469 if line.startswith(config + '='):
Masahiro Yamadacc008292016-05-19 15:51:56 +0900470 new_val = line
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900471 break
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900472 else:
Masahiro Yamadacc008292016-05-19 15:51:56 +0900473 new_val = not_set
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900474
Masahiro Yamadacc008292016-05-19 15:51:56 +0900475 if old_val == new_val:
476 return (ACTION_NO_CHANGE, new_val)
477
478 # If this CONFIG is neither bool nor trisate
479 if old_val[-2:] != '=y' and old_val[-2:] != '=m' and old_val != not_set:
480 # tools/scripts/define2mk.sed changes '1' to 'y'.
481 # This is a problem if the CONFIG is int type.
482 # Check the type in Kconfig and handle it correctly.
483 if new_val[-2:] == '=y':
484 new_val = new_val[:-1] + '1'
485
486 return (ACTION_MOVE, new_val)
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900487
Masahiro Yamada1d085562016-05-19 15:52:02 +0900488 def update_dotconfig(self):
Masahiro Yamada6ff36d22016-05-19 15:51:50 +0900489 """Parse files for the config options and update the .config.
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900490
Masahiro Yamadacc008292016-05-19 15:51:56 +0900491 This function parses the generated .config and include/autoconf.mk
492 searching the target options.
Masahiro Yamada6ff36d22016-05-19 15:51:50 +0900493 Move the config option(s) to the .config as needed.
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900494
495 Arguments:
496 defconfig: defconfig name.
Masahiro Yamada522e8dc2016-05-19 15:52:01 +0900497
498 Returns:
Masahiro Yamada7fb0bac2016-05-19 15:52:04 +0900499 Return a tuple of (updated flag, log string).
500 The "updated flag" is True if the .config was updated, False
501 otherwise. The "log string" shows what happend to the .config.
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900502 """
503
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900504 results = []
Masahiro Yamada7fb0bac2016-05-19 15:52:04 +0900505 updated = False
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900506
Masahiro Yamada1f169922016-05-19 15:52:00 +0900507 with open(self.dotconfig) as f:
Masahiro Yamadacc008292016-05-19 15:51:56 +0900508 dotconfig_lines = f.readlines()
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900509
Masahiro Yamada1f169922016-05-19 15:52:00 +0900510 with open(self.autoconf) as f:
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900511 autoconf_lines = f.readlines()
512
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900513 for config in self.configs:
514 result = self.parse_one_config(config, dotconfig_lines,
Joe Hershberger96464ba2015-05-19 13:21:17 -0500515 autoconf_lines)
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900516 results.append(result)
517
518 log = ''
519
520 for (action, value) in results:
521 if action == ACTION_MOVE:
522 actlog = "Move '%s'" % value
523 log_color = COLOR_LIGHT_GREEN
Masahiro Yamadacc008292016-05-19 15:51:56 +0900524 elif action == ACTION_NO_ENTRY:
525 actlog = "%s is not defined in Kconfig. Do nothing." % value
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900526 log_color = COLOR_LIGHT_BLUE
Masahiro Yamadacc008292016-05-19 15:51:56 +0900527 elif action == ACTION_NO_CHANGE:
528 actlog = "'%s' is the same as the define in Kconfig. Do nothing." \
529 % value
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900530 log_color = COLOR_LIGHT_PURPLE
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900531 else:
532 sys.exit("Internal Error. This should not happen.")
533
Masahiro Yamada1d085562016-05-19 15:52:02 +0900534 log += color_text(self.options.color, log_color, actlog) + '\n'
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900535
Masahiro Yamada1f169922016-05-19 15:52:00 +0900536 with open(self.dotconfig, 'a') as f:
Masahiro Yamadae423d172016-05-19 15:51:49 +0900537 for (action, value) in results:
538 if action == ACTION_MOVE:
539 f.write(value + '\n')
Masahiro Yamada7fb0bac2016-05-19 15:52:04 +0900540 updated = True
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900541
Masahiro Yamada1f169922016-05-19 15:52:00 +0900542 os.remove(self.config_autoconf)
543 os.remove(self.autoconf)
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900544
Masahiro Yamada7fb0bac2016-05-19 15:52:04 +0900545 return (updated, log)
Masahiro Yamada522e8dc2016-05-19 15:52:01 +0900546
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900547class Slot:
548
549 """A slot to store a subprocess.
550
551 Each instance of this class handles one subprocess.
552 This class is useful to control multiple threads
553 for faster processing.
554 """
555
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900556 def __init__(self, configs, options, progress, devnull, make_cmd):
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900557 """Create a new process slot.
558
559 Arguments:
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900560 configs: A list of CONFIGs to move.
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900561 options: option flags.
Masahiro Yamadac5e60fd2016-05-19 15:51:55 +0900562 progress: A progress indicator.
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900563 devnull: A file object of '/dev/null'.
564 make_cmd: command name of GNU Make.
565 """
566 self.options = options
Masahiro Yamadac5e60fd2016-05-19 15:51:55 +0900567 self.progress = progress
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900568 self.build_dir = tempfile.mkdtemp()
569 self.devnull = devnull
570 self.make_cmd = (make_cmd, 'O=' + self.build_dir)
Masahiro Yamada522e8dc2016-05-19 15:52:01 +0900571 self.parser = KconfigParser(configs, options, self.build_dir)
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900572 self.state = STATE_IDLE
573 self.failed_boards = []
574
575 def __del__(self):
576 """Delete the working directory
577
578 This function makes sure the temporary directory is cleaned away
579 even if Python suddenly dies due to error. It should be done in here
580 because it is guranteed the destructor is always invoked when the
581 instance of the class gets unreferenced.
582
583 If the subprocess is still running, wait until it finishes.
584 """
585 if self.state != STATE_IDLE:
586 while self.ps.poll() == None:
587 pass
588 shutil.rmtree(self.build_dir)
589
Masahiro Yamadac5e60fd2016-05-19 15:51:55 +0900590 def add(self, defconfig):
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900591 """Assign a new subprocess for defconfig and add it to the slot.
592
593 If the slot is vacant, create a new subprocess for processing the
594 given defconfig and add it to the slot. Just returns False if
595 the slot is occupied (i.e. the current subprocess is still running).
596
597 Arguments:
598 defconfig: defconfig name.
599
600 Returns:
601 Return True on success or False on failure
602 """
603 if self.state != STATE_IDLE:
604 return False
605 cmd = list(self.make_cmd)
606 cmd.append(defconfig)
Joe Hershberger25400092015-05-19 13:21:23 -0500607 self.ps = subprocess.Popen(cmd, stdout=self.devnull,
608 stderr=subprocess.PIPE)
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900609 self.defconfig = defconfig
610 self.state = STATE_DEFCONFIG
Masahiro Yamada1d085562016-05-19 15:52:02 +0900611 self.log = ''
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900612 return True
613
614 def poll(self):
615 """Check the status of the subprocess and handle it as needed.
616
617 Returns True if the slot is vacant (i.e. in idle state).
618 If the configuration is successfully finished, assign a new
619 subprocess to build include/autoconf.mk.
620 If include/autoconf.mk is generated, invoke the parser to
Masahiro Yamada7fb0bac2016-05-19 15:52:04 +0900621 parse the .config and the include/autoconf.mk, moving
622 config options to the .config as needed.
623 If the .config was updated, run "make savedefconfig" to sync
624 it, update the original defconfig, and then set the slot back
625 to the idle state.
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900626
627 Returns:
628 Return True if the subprocess is terminated, False otherwise
629 """
630 if self.state == STATE_IDLE:
631 return True
632
633 if self.ps.poll() == None:
634 return False
635
636 if self.ps.poll() != 0:
Masahiro Yamada1d085562016-05-19 15:52:02 +0900637 self.log += color_text(self.options.color, COLOR_LIGHT_RED,
638 "Failed to process.\n")
Joe Hershberger95bf9c72015-05-19 13:21:24 -0500639 if self.options.verbose:
Masahiro Yamada1d085562016-05-19 15:52:02 +0900640 self.log += color_text(self.options.color, COLOR_LIGHT_CYAN,
641 self.ps.stderr.read())
Masahiro Yamada4efef992016-05-19 15:52:03 +0900642 self.finish(False)
Masahiro Yamadaff8725b2016-05-19 15:51:51 +0900643 return True
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900644
645 if self.state == STATE_AUTOCONF:
Masahiro Yamada7fb0bac2016-05-19 15:52:04 +0900646 (updated, log) = self.parser.update_dotconfig()
647 self.log += log
Joe Hershberger96464ba2015-05-19 13:21:17 -0500648
Masahiro Yamada7fb0bac2016-05-19 15:52:04 +0900649 if not updated:
650 self.finish(True)
651 return True
Joe Hershberger96464ba2015-05-19 13:21:17 -0500652 """Save off the defconfig in a consistent way"""
653 cmd = list(self.make_cmd)
654 cmd.append('savedefconfig')
655 self.ps = subprocess.Popen(cmd, stdout=self.devnull,
Joe Hershberger25400092015-05-19 13:21:23 -0500656 stderr=subprocess.PIPE)
Joe Hershberger96464ba2015-05-19 13:21:17 -0500657 self.state = STATE_SAVEDEFCONFIG
658 return False
659
660 if self.state == STATE_SAVEDEFCONFIG:
Masahiro Yamadae423d172016-05-19 15:51:49 +0900661 if not self.options.dry_run:
662 shutil.move(os.path.join(self.build_dir, 'defconfig'),
663 os.path.join('configs', self.defconfig))
Masahiro Yamada4efef992016-05-19 15:52:03 +0900664 self.finish(True)
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900665 return True
666
Joe Hershberger25400092015-05-19 13:21:23 -0500667 self.cross_compile = self.parser.get_cross_compile()
Masahiro Yamada90ed6cb2016-05-19 15:51:53 +0900668 if self.cross_compile is None:
Masahiro Yamada1d085562016-05-19 15:52:02 +0900669 self.log += color_text(self.options.color, COLOR_YELLOW,
670 "Compiler is missing. Do nothing.\n")
Masahiro Yamada4efef992016-05-19 15:52:03 +0900671 self.finish(False)
Masahiro Yamada90ed6cb2016-05-19 15:51:53 +0900672 return True
673
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900674 cmd = list(self.make_cmd)
Joe Hershberger25400092015-05-19 13:21:23 -0500675 if self.cross_compile:
676 cmd.append('CROSS_COMPILE=%s' % self.cross_compile)
Joe Hershberger7740f652015-05-19 13:21:18 -0500677 cmd.append('KCONFIG_IGNORE_DUPLICATES=1')
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900678 cmd.append('include/config/auto.conf')
Joe Hershberger25400092015-05-19 13:21:23 -0500679 self.ps = subprocess.Popen(cmd, stdout=self.devnull,
Joe Hershberger25400092015-05-19 13:21:23 -0500680 stderr=subprocess.PIPE)
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900681 self.state = STATE_AUTOCONF
682 return False
683
Masahiro Yamada4efef992016-05-19 15:52:03 +0900684 def finish(self, success):
685 """Display log along with progress and go to the idle state.
Masahiro Yamada1d085562016-05-19 15:52:02 +0900686
687 Arguments:
Masahiro Yamada4efef992016-05-19 15:52:03 +0900688 success: Should be True when the defconfig was processed
689 successfully, or False when it fails.
Masahiro Yamada1d085562016-05-19 15:52:02 +0900690 """
691 # output at least 30 characters to hide the "* defconfigs out of *".
692 log = self.defconfig.ljust(30) + '\n'
693
694 log += '\n'.join([ ' ' + s for s in self.log.split('\n') ])
695 # Some threads are running in parallel.
696 # Print log atomically to not mix up logs from different threads.
Masahiro Yamada4efef992016-05-19 15:52:03 +0900697 print >> (sys.stdout if success else sys.stderr), log
698
699 if not success:
700 if self.options.exit_on_error:
701 sys.exit("Exit on error.")
702 # If --exit-on-error flag is not set, skip this board and continue.
703 # Record the failed board.
704 self.failed_boards.append(self.defconfig)
705
Masahiro Yamada1d085562016-05-19 15:52:02 +0900706 self.progress.inc()
707 self.progress.show()
Masahiro Yamada4efef992016-05-19 15:52:03 +0900708 self.state = STATE_IDLE
Masahiro Yamada1d085562016-05-19 15:52:02 +0900709
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900710 def get_failed_boards(self):
711 """Returns a list of failed boards (defconfigs) in this slot.
712 """
713 return self.failed_boards
714
715class Slots:
716
717 """Controller of the array of subprocess slots."""
718
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900719 def __init__(self, configs, options, progress):
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900720 """Create a new slots controller.
721
722 Arguments:
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900723 configs: A list of CONFIGs to move.
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900724 options: option flags.
Masahiro Yamadac5e60fd2016-05-19 15:51:55 +0900725 progress: A progress indicator.
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900726 """
727 self.options = options
728 self.slots = []
729 devnull = get_devnull()
730 make_cmd = get_make_cmd()
731 for i in range(options.jobs):
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900732 self.slots.append(Slot(configs, options, progress, devnull,
Masahiro Yamadac5e60fd2016-05-19 15:51:55 +0900733 make_cmd))
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900734
Masahiro Yamadac5e60fd2016-05-19 15:51:55 +0900735 def add(self, defconfig):
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900736 """Add a new subprocess if a vacant slot is found.
737
738 Arguments:
739 defconfig: defconfig name to be put into.
740
741 Returns:
742 Return True on success or False on failure
743 """
744 for slot in self.slots:
Masahiro Yamadac5e60fd2016-05-19 15:51:55 +0900745 if slot.add(defconfig):
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900746 return True
747 return False
748
749 def available(self):
750 """Check if there is a vacant slot.
751
752 Returns:
753 Return True if at lease one vacant slot is found, False otherwise.
754 """
755 for slot in self.slots:
756 if slot.poll():
757 return True
758 return False
759
760 def empty(self):
761 """Check if all slots are vacant.
762
763 Returns:
764 Return True if all the slots are vacant, False otherwise.
765 """
766 ret = True
767 for slot in self.slots:
768 if not slot.poll():
769 ret = False
770 return ret
771
772 def show_failed_boards(self):
773 """Display all of the failed boards (defconfigs)."""
774 failed_boards = []
775
776 for slot in self.slots:
777 failed_boards += slot.get_failed_boards()
778
779 if len(failed_boards) > 0:
780 msg = [ "The following boards were not processed due to error:" ]
781 msg += failed_boards
782 for line in msg:
783 print >> sys.stderr, color_text(self.options.color,
784 COLOR_LIGHT_RED, line)
785
Joe Hershberger2559cd82015-05-19 13:21:22 -0500786 with open('moveconfig.failed', 'w') as f:
787 for board in failed_boards:
788 f.write(board + '\n')
789
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900790def move_config(configs, options):
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900791 """Move config options to defconfig files.
792
793 Arguments:
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900794 configs: A list of CONFIGs to move.
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900795 options: option flags
796 """
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900797 if len(configs) == 0:
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900798 print 'Nothing to do. exit.'
799 sys.exit(0)
800
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900801 print 'Move %s (jobs: %d)' % (', '.join(configs), options.jobs)
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900802
Joe Hershberger91040e82015-05-19 13:21:19 -0500803 if options.defconfigs:
804 defconfigs = [line.strip() for line in open(options.defconfigs)]
805 for i, defconfig in enumerate(defconfigs):
806 if not defconfig.endswith('_defconfig'):
807 defconfigs[i] = defconfig + '_defconfig'
808 if not os.path.exists(os.path.join('configs', defconfigs[i])):
809 sys.exit('%s - defconfig does not exist. Stopping.' %
810 defconfigs[i])
811 else:
812 # All the defconfig files to be processed
813 defconfigs = []
814 for (dirpath, dirnames, filenames) in os.walk('configs'):
815 dirpath = dirpath[len('configs') + 1:]
816 for filename in fnmatch.filter(filenames, '*_defconfig'):
817 defconfigs.append(os.path.join(dirpath, filename))
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900818
Masahiro Yamadac5e60fd2016-05-19 15:51:55 +0900819 progress = Progress(len(defconfigs))
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900820 slots = Slots(configs, options, progress)
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900821
822 # Main loop to process defconfig files:
823 # Add a new subprocess into a vacant slot.
824 # Sleep if there is no available slot.
Masahiro Yamadac5e60fd2016-05-19 15:51:55 +0900825 for defconfig in defconfigs:
826 while not slots.add(defconfig):
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900827 while not slots.available():
828 # No available slot: sleep for a while
829 time.sleep(SLEEP_TIME)
830
831 # wait until all the subprocesses finish
832 while not slots.empty():
833 time.sleep(SLEEP_TIME)
834
Joe Hershberger2e2ce6c2015-05-19 13:21:25 -0500835 print ''
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900836 slots.show_failed_boards()
837
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900838def main():
839 try:
840 cpu_count = multiprocessing.cpu_count()
841 except NotImplementedError:
842 cpu_count = 1
843
844 parser = optparse.OptionParser()
845 # Add options here
846 parser.add_option('-c', '--color', action='store_true', default=False,
847 help='display the log in color')
Joe Hershberger91040e82015-05-19 13:21:19 -0500848 parser.add_option('-d', '--defconfigs', type='string',
849 help='a file containing a list of defconfigs to move')
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900850 parser.add_option('-n', '--dry-run', action='store_true', default=False,
851 help='perform a trial run (show log with no changes)')
852 parser.add_option('-e', '--exit-on-error', action='store_true',
853 default=False,
854 help='exit immediately on any error')
Joe Hershberger2144f882015-05-19 13:21:20 -0500855 parser.add_option('-H', '--headers-only', dest='cleanup_headers_only',
856 action='store_true', default=False,
857 help='only cleanup the headers')
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900858 parser.add_option('-j', '--jobs', type='int', default=cpu_count,
859 help='the number of jobs to run simultaneously')
Joe Hershberger95bf9c72015-05-19 13:21:24 -0500860 parser.add_option('-v', '--verbose', action='store_true', default=False,
861 help='show any build errors as boards are built')
Masahiro Yamadab6ef3932016-05-19 15:51:58 +0900862 parser.usage += ' CONFIG ...'
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900863
Masahiro Yamadab6ef3932016-05-19 15:51:58 +0900864 (options, configs) = parser.parse_args()
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900865
Masahiro Yamadab6ef3932016-05-19 15:51:58 +0900866 if len(configs) == 0:
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900867 parser.print_usage()
868 sys.exit(1)
869
Masahiro Yamadab6ef3932016-05-19 15:51:58 +0900870 # prefix the option name with CONFIG_ if missing
871 configs = [ config if config.startswith('CONFIG_') else 'CONFIG_' + config
872 for config in configs ]
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900873
Joe Hershberger2144f882015-05-19 13:21:20 -0500874 check_top_directory()
875
Masahiro Yamadabd63e5b2016-05-19 15:51:54 +0900876 check_clean_directory()
877
Masahiro Yamada90ed6cb2016-05-19 15:51:53 +0900878 update_cross_compile(options.color)
Masahiro Yamada4b430c92016-05-19 15:51:52 +0900879
Joe Hershberger2144f882015-05-19 13:21:20 -0500880 if not options.cleanup_headers_only:
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900881 move_config(configs, options)
Joe Hershberger2144f882015-05-19 13:21:20 -0500882
Masahiro Yamadab134bc12016-05-19 15:51:57 +0900883 cleanup_headers(configs, options.dry_run)
Masahiro Yamada5a27c732015-05-20 11:36:07 +0900884
885if __name__ == '__main__':
886 main()