blob: bae6804ddbfd19192cdb6ddaac2eeff19ab34709 [file] [log] [blame]
Jack Rosenthal543bb062020-06-05 15:30:09 -06001#!/usr/bin/env python3
2# Copyright 2020 The Chromium OS Authors. All rights reserved.
3# Use of this source code is governed by a BSD-style license that can be
4# found in the LICENSE file.
5
6# This file uses 2-space indentations.
7# pylint: disable=bad-indentation
8
9# This is contrib-quality code: not all functions/classes are
10# documented.
11# pylint: disable=missing-function-docstring
12# pylint: disable=missing-class-docstring
13# pylint: disable=class-missing-docstring
14
15# Classes make heavy-use of setattr to dynamically set the attributes
16# on an object. Disable this check which gets confused very
17# frequently.
18# pylint: disable=no-member
19
20"""Utility script to auto-convert a pre-unibuild board to unibuild."""
21
22import argparse
23import datetime
24import json
25import os
26import pathlib
27import re
28import shlex
29import subprocess
30import sys
31import tempfile
32# pylint: disable=import-error
33import yaml
34# pylint: enable=import-error
35
36
37make_defaults_search_and_destroy_re = re.compile(
38 r'(?:^\s*)*(?:^\s*#.*\s*)*^\s*USE="\s*\$\{?USE\}?\s*-unibuild\s*"\s*$',
39 re.MULTILINE)
40
41
42def log(message):
43 print('[{}] {}'.format(datetime.datetime.now(), message), file=sys.stderr)
44
45
46def prepend_all_lines(text, prepend):
47 return ''.join(
48 '{}{}\n'.format(prepend, line)
49 for line in text.splitlines())
50
51
52def gen_cros_copyright(line_comment='# '):
53 return prepend_all_lines(
54 """Copyright {} The Chromium OS Authors. All rights reserved.
Jack Rosenthalf72933c2020-06-24 13:37:50 -060055Use of this source code is governed by a BSD-style license that can be
Jack Rosenthal543bb062020-06-05 15:30:09 -060056found in the LICENSE file.""".format(datetime.datetime.now().strftime('%Y')),
57 line_comment)
58
59
60def yaml_str_representer(dumper, data):
61 style = None
62 tag = 'tag:yaml.org,2002:str'
63 if '\n' in data:
64 style = '|'
65 return dumper.represent_scalar(tag, data, style)
66
67
68yaml.add_representer(str, yaml_str_representer)
69
70
71def format_yaml(config):
72 conf_str = yaml.dump(config, indent=2, default_flow_style=False)
73 out = gen_cros_copyright()
74 out += """
75# This board only supports a single config, defined below, as it is a
76# migrated pre-unibuild device.
77device-config: &device_config\n"""
78 out += prepend_all_lines(conf_str, ' ')
79 out += """
80# Required dunder for chromeos-config to support a single device.
81chromeos:
82 devices:
83 - skus:
84 - config: *device_config\n"""
85 return out
86
87
88def generate_vpackage(depends):
89 return gen_cros_copyright() + """
90EAPI=7
91
Jack Rosenthal543bb062020-06-05 15:30:09 -060092DESCRIPTION="ChromeOS Unibuild Config virtual package"
93HOMEPAGE="https://chromium.googlesource.com/chromiumos/platform2/+/master/chromeos-config/README.md"
94
95LICENSE="BSD-Google"
96SLOT="0"
Jack Rosenthalbe2f1272020-06-10 09:31:09 -060097KEYWORDS="*"
Jack Rosenthal543bb062020-06-05 15:30:09 -060098
99DEPEND="%(depends)s"
100RDEPEND="${DEPEND}"
101""" % {
102 'depends':
103 (''.join('\n\t{}'.format(d) for d in depends) + '\n')
104 if len(depends) > 1 else
105 ''.join(depends)}
106
107
108def generate_bsp_ebuild(private=False):
109 return gen_cros_copyright() + """
110EAPI=7
111
112# cros_workon applies only to ebuild and files directory. Use the
113# canonical empty project.
114CROS_WORKON_PROJECT="chromiumos/infra/build/empty-project"
Jack Rosenthalbe2f1272020-06-10 09:31:09 -0600115CROS_WORKON_LOCALNAME="platform/empty-project"
Jack Rosenthal543bb062020-06-05 15:30:09 -0600116
117inherit cros-workon cros-unibuild
118
119DESCRIPTION="ChromeOS model configuration"
120HOMEPAGE="https://chromium.googlesource.com/chromiumos/platform2/+/master/chromeos-config/README.md"
121
122LICENSE="BSD-Google"
123SLOT="0"
124KEYWORDS="~*"
125
126src_install() {
127\tinstall%(maybe_private)s_model_files
128}
129""" % {'maybe_private': '_private' if private else ''}
130
131
132def generate_firmware_ebuild(board_name):
133 return gen_cros_copyright() + """
134# Change this version number when any change is made to model.yaml
135# in order to trigger an auto-revbump is required.
136# VERSION=REVBUMP-0.0.1
137
138EAPI=7
139CROS_WORKON_COMMIT=""
140CROS_WORKON_TREE=""
141CROS_WORKON_LOCALNAME="platform/firmware"
142CROS_WORKON_PROJECT="chromiumos/platform/firmware"
143CROS_BOARDS=( %(board_name)s )
144
145inherit cros-workon cros-firmware cros-unibuild
146
147DESCRIPTION="Chrome OS Firmware (%(board_name)s)"
148HOMEPAGE="http://src.chromium.org"
149LICENSE="BSD-Google"
150SLOT="0"
151KEYWORDS="~*"
152
153RDEPEND=""
154
155# Unified Builds firmware URL's are read from:
156# chromeos-base/chromeos-config-bsp-private/files/model.yaml
157# in this repository. Those config files output the SRC_URI's used by Portage.
158#
159# Update the model.yaml, then run this command from the
160# src/platform/dev/contrib directory:
161#
162# ./cros_update_firmware --board=%(board_name)s
163#
164# Verify the changes by running:
165# /build/%(board_name)s/usr/sbin/chromeos-firmwareupdate --manifest
166#
167# If this works then you can create a CL with your changes, which should include
168# the files:
169# chromeos-base/chromeos-config-bsp-private/files/model.yaml
170# chromeos-base/chromeos-firmware-%(board_name)s/Manifest
171# chromeos-base/chromeos-firmware-%(board_name)s/files/srcuris
172# chromeos-base/chromeos-firmware-%(board_name)s/chromeos-firmware-%(board_name)s-9999.ebuild
173cros-firmware_setup_source
174""" % {'board_name': board_name}
175
176
177def find_file(searchdir, name):
178 results = []
179 for root, _, files in os.walk(searchdir):
180 if name in files:
181 results.append(pathlib.Path(root) / name)
182 return results
183
184
185def find_one_file(searchdir, name):
186 results = find_file(searchdir, name)
187 assert len(results) == 1
188 return results.pop()
189
190
191def sh_getvar(script, varname):
192 script = script + ('\necho "${%s}"\n' % varname)
193 with tempfile.NamedTemporaryFile('w') as f:
194 f.write(script)
195 f.flush()
196 res = subprocess.run(['sh', f.name], stdout=subprocess.PIPE,
197 check=True, encoding='utf-8')
198 return res.stdout.strip() or None
199
200
Jack Rosenthalbe2f1272020-06-10 09:31:09 -0600201def write_file(fullpath, file_contents, make_ebuild_symlink=False):
Jack Rosenthal543bb062020-06-05 15:30:09 -0600202 os.makedirs(fullpath.parent, exist_ok=True)
203 log('Writing {}...'.format(fullpath))
204 with open(fullpath, 'w') as f:
205 f.write(file_contents)
Jack Rosenthalbe2f1272020-06-10 09:31:09 -0600206 if make_ebuild_symlink:
207 if not fullpath.name.endswith('.ebuild'):
208 raise ValueError(
209 'make_ebuild_symlink specified, but path does not look like an ebuild')
210 prefix, _, _ = fullpath.name.rpartition('.')
211 linkname = fullpath.parent / '{}-r1.ebuild'.format(prefix)
212 log('Creating symlink {} -> {}...'.format(linkname, fullpath))
213 os.symlink(fullpath.name, linkname)
Jack Rosenthal543bb062020-06-05 15:30:09 -0600214
215
216def generate_make_defaults(contents):
217 contents = make_defaults_search_and_destroy_re.sub('', contents)
218 contents += """
219# Enable chromeos-config.
220USE="${USE} unibuild"
221"""
222 return contents
223
224
225class CrosConfig:
226 def __init__(self, public_yaml_raw, private_yaml_raw):
227 with tempfile.NamedTemporaryFile(mode='w', delete=False) as merged_tempfile, \
228 tempfile.NamedTemporaryFile(mode='w') as public_yaml_tempfile, \
229 tempfile.NamedTemporaryFile(mode='w') as private_yaml_tempfile:
230 public_yaml_tempfile.write(public_yaml_raw)
231 public_yaml_tempfile.flush()
232
233 private_yaml_tempfile.write(private_yaml_raw)
234 private_yaml_tempfile.flush()
235
236 log('Merging and validating config schema...')
237 subprocess.run(['cros_config_schema', '-o', merged_tempfile.name,
238 '-m', public_yaml_tempfile.name,
239 private_yaml_tempfile.name], check=True)
240 self.merged_yaml = merged_tempfile.name
241
242 def run_host_command(self, *args):
243 return subprocess.run(['cros_config_host', '-c', self.merged_yaml]
244 + list(args),
245 check=True, encoding='utf-8',
246 stdout=subprocess.PIPE).stdout
247
248
249class BoardOverlays:
250 FIRMWARE_ATTRS = [
251 ('CROS_FIRMWARE_MAIN_IMAGE', 'bcs_main_ro'),
252 ('CROS_FIRMWARE_MAIN_RW_IMAGE', 'bcs_main_rw'),
253 ('CROS_FIRMWARE_EC_IMAGE', 'bcs_ec'),
254 ('CROS_FIRMWARE_PD_IMAGE', 'bcs_pd'),
255 ]
256
257 MAKE_DEFAULTS_ATTRS = [
258 ('EC_FIRMWARE', 'ec_firmwares'),
259 ('PD_FIRMWARE', 'pd_firmwares'),
260 ('EC_FIRMWARE_EXTRA', 'ec_firmware_extras'),
261 ('FPMCU_FIRMWARE', 'fpmcu_firmware'),
262 ('USE', 'use_flags'),
263 ]
264
265 def __init__(self, board_name, checkout, mosys_platform):
266 self.board_name = board_name
267 self.mosys_platform = mosys_platform
268 self.public_overlay = (checkout / 'src' / 'overlays'
269 / f'overlay-{board_name}')
270 log('Public overlay path: {}'.format(self.public_overlay))
271 self.private_overlay = (checkout / 'src' / 'private-overlays'
272 / f'overlay-{board_name}-private')
273 log('Private overlay path: {}'.format(self.private_overlay))
274
275 assert self.public_overlay.is_dir()
276 assert self.private_overlay.is_dir()
277
278 # Find the firmware ebuild
279 self.firmware_ebuild_path = find_one_file(
280 self.private_overlay, f'chromeos-firmware-{board_name}-9999.ebuild')
281 log('Firmware ebuild path: {}'.format(self.firmware_ebuild_path))
282
283 # Read the firmware attrs from it
284 for _, attr in self.FIRMWARE_ATTRS:
285 setattr(self, attr, None)
286
287 with open(self.firmware_ebuild_path) as f:
288 for line in f:
289 if '#' in line:
290 line, _, _ = line.partition('#')
291 line = line.strip()
292
293 for var, attr in self.FIRMWARE_ATTRS:
294 if line.startswith('{}='.format(var)):
295 _, _, value = line.partition('=')
296 value = value.replace('"', '').replace("'", '')
297 setattr(self, attr, value)
298
299 # Find make.defaults files
300 self.public_make_defaults_file = (
301 self.public_overlay / 'profiles' / 'base' / 'make.defaults')
302 self.private_make_defaults_file = (
303 self.private_overlay / 'profiles' / 'base' / 'make.defaults')
304
305 with open(self.public_make_defaults_file) as f:
306 self.public_make_defaults = f.read()
307 with open(self.private_make_defaults_file) as f:
308 self.private_make_defaults = f.read()
309
310 for var, attr in self.MAKE_DEFAULTS_ATTRS:
311 setattr(self, attr, set())
312 for script in (self.public_make_defaults, self.private_make_defaults):
313 value = sh_getvar(script, var)
314 if value:
315 for v in value.split():
316 getattr(self, attr).add(v)
317
318 if 'whiskers' in self.ec_firmware_extras:
319 self.ec_firmware_extras.remove('whiskers')
320 self.detachable_base_build_target = 'whiskers'
321 else:
322 self.detachable_base_build_target = None
323
324 self.ec_build_target = ' '.join(self.ec_firmwares) or None
325 self.ec_extras_build_target = sorted(list(self.ec_firmware_extras
326 | self.pd_firmwares)) or None
327
Jack Rosenthalbe2f1272020-06-10 09:31:09 -0600328 def write_file(self, overlay_flags, path, file_contents,
329 make_ebuild_symlink=False):
Jack Rosenthal543bb062020-06-05 15:30:09 -0600330 dirs = []
331 if overlay_flags & M_PUBLIC:
332 dirs += [self.public_overlay]
333 if overlay_flags & M_PRIVATE:
334 dirs += [self.private_overlay]
335 for d in dirs:
Jack Rosenthalbe2f1272020-06-10 09:31:09 -0600336 write_file(d / path, file_contents,
337 make_ebuild_symlink=make_ebuild_symlink)
Jack Rosenthal543bb062020-06-05 15:30:09 -0600338
339
340class Dut:
341 def __init__(self, hostname, checkout, port=22):
342 self.ssh_hostname = hostname
343
344 id_source = checkout / 'chromite' / 'ssh_keys' / 'testing_rsa'
345 with open(id_source, 'rb') as f:
346 id_contents = f.read()
347
348 with tempfile.NamedTemporaryFile(mode='wb', delete=False) as tmpfile:
349 tmpfile.write(id_contents)
350 self.ssh_identity = tmpfile.name
351
352 with tempfile.NamedTemporaryFile(delete=False) as tmpfile:
353 self.ssh_known_hosts_file = tmpfile.name
354
355 self.ssh_port = port
356
357 # Check connectivity.
358 log('Checking SSH connectivity to DUT...')
359 self.run_command(['/bin/true'])
360
361 # Linter is unaware that we set check=True in kwargs.
362 # pylint: disable=subprocess-run-check
363 def run_command(self, argv, *args, **kwargs):
364 kwargs.setdefault('check', True)
365 kwargs.setdefault('stdout', subprocess.PIPE)
366 kwargs.setdefault('encoding', 'utf-8')
367 quoted_argv = [shlex.quote(arg) for arg in argv]
368 return subprocess.run(['ssh',
369 '-p', '{}'.format(self.ssh_port),
370 '-i', self.ssh_identity,
371 '-o', 'UserKnownHostsFile={}'.format(
372 self.ssh_known_hosts_file),
373 '-o', 'StrictHostKeyChecking=no',
374 '-o', 'CheckHostIP=no',
375 '-o', 'ConnectTimeout=10',
376 'root@{}'.format(self.ssh_hostname)] + quoted_argv,
377 *args, **kwargs)
378 # pylint: enable=subprocess-run-check
379
380
381class DeviceConfig:
382 ATTRS = {
383 'brand_code': ['mosys', 'platform', 'brand'],
384 'model': ['mosys', 'platform', 'model'],
385 'lsb_release': ['cat', '/etc/lsb-release'],
386 'smbios_name': ['cat', '/sys/class/dmi/id/product_name'],
387 'fdt_compatible_raw': ['cat', '/proc/device-tree/compatible'],
388 'arc_build_props': ['cat', '/usr/share/arc/properties/build.prop'],
389 'mosys_psu_type': ['mosys', 'psu', 'type'],
390 'whitelabel_tag': ['vpd_get_value', 'whitelabel_tag'],
391 'customization_id': ['vpd_get_value', 'customization_id'],
392 'cras_config_dir': ['sh', '/etc/cras/get_device_config_dir'],
393 'internal_ucm_suffix': ['sh', '/etc/cras/get_internal_ucm_suffix'],
394 # disgusting, but whatever...
395 'powerd_raw':
396 ['python3', '-c',
397 'import os;'
398 'import json;'
399 'print(json.dumps('
400 '{f.replace("_", "-"): open("/usr/share/power_manager/board_specific/"+f).read().rstrip()'
401 ' for f in os.listdir("/usr/share/power_manager/board_specific")}))'],
402 }
403
404 @classmethod
405 def from_dut(cls, dut):
406 slf = cls()
407 for attr, cmd in cls.ATTRS.items():
408 try:
409 log('Running {!r} on DUT...'.format(cmd))
410 res = dut.run_command(cmd)
411 except subprocess.CalledProcessError:
412 setattr(slf, attr, None)
413 else:
414 setattr(slf, attr, res.stdout.strip())
415 return slf
416
417 def __str__(self):
418 return 'DeviceConfig({})'.format(
419 ', '.join('{}={!r}'.format(attr, getattr(self, attr))
420 for attr in self.ATTRS))
421
422 def lsb_val(self, name, default=None):
423 for item in self.lsb_release.splitlines():
424 k, _, v = item.partition('=')
425 if k == name:
426 return v
427 return default
428
429 def arc_build_prop(self, name, default=None):
430 for line in self.arc_build_props.splitlines():
431 if '#' in line:
432 line, _, _ = line.partition('#')
433 line = line.strip()
434 if line.startswith('{}='.format(name)):
435 _, _, val = line.partition('=')
436 return val
437 return default
438
439
440def genconf_first_api_level(_, overlay):
441 if overlay.board_name in ('atlas', 'nocturne'):
442 return '28'
443 return '25'
444
445
446def genconf_dt_compatible_match(device, overlay):
447 if not device.fdt_compatible_raw:
448 return None
449 compatible_strings = device.fdt_compatible_raw.strip('\x00').split('\x00')
450 compatible_strings.sort(key=lambda s: (s.startswith('google'),
451 'rev' not in s,
452 'sku' not in s,
453 overlay.board_name in s,
454 -len(s)))
455 return compatible_strings[-1]
456
457
458def genconf_psu_type(device, _):
459 if device.mosys_psu_type:
460 return device.mosys_psu_type
461 devicetype = device.lsb_val('DEVICETYPE')
462 if devicetype == 'CHROMEBOOK':
463 return 'battery'
464 if devicetype in ('CHROMEBIT', 'CHROMEBASE', 'CHROMEBOX'):
465 return 'AC_only'
466 return None
467
468
469def genconf_fp_board(_, overlay):
470 if overlay.fpmcu_firmware:
471 return ' '.join(overlay.fpmcu_firmware)
472 return None
473
474
475def genconf_fp_type(_, overlay):
476 if 'fp_on_power_button' in overlay.use_flags:
477 return 'on-power-button'
478 if overlay.fpmcu_firmware:
479 return 'stand-alone'
480 return None
481
482
483def genconf_fp_location(_, overlay):
484 if overlay.board_name == 'nocturne':
485 return 'power-button-top-left'
486 return None
487
488
489def genconf_signature_id(device, _):
490 if device.whitelabel_tag:
491 return device.whitelabel_tag.upper()
492 if device.customization_id:
493 return device.customization_id.upper().partition('-')[0]
494 return device.model
495
496
497def genconf_cras_config_dir(device, _):
498 prefix = '/etc/cras/'
499 if device.cras_config_dir and device.cras_config_dir.startswith(prefix):
500 return device.cras_config_dir[len(prefix):]
501 if device.cras_config_dir:
502 return '../../{}'.format(device.cras_config_dir)
503 return None
504
505
506def genconf_powerd_settings(device, overlay):
507 if not device.powerd_raw:
508 d = {}
509 else:
510 d = json.loads(device.powerd_raw)
511 if 'mosys_eventlog' in overlay.use_flags:
512 d['mosys-eventlog'] = '1'
513 return d
514
515
Jack Rosenthal36a4c242020-06-10 20:55:06 -0600516def genconf_whitelabel_tag(device, _):
517 # Devices with a Customization ID are not compatible with whitelabel
518 # tags.
519 if device.customization_id:
520 return None
521 return device.whitelabel_tag or None
522
523
Jack Rosenthal543bb062020-06-05 15:30:09 -0600524M_PUBLIC = (1 << 0)
525M_PRIVATE = (1 << 1)
526
527
528genconf_schema = {
529 'name': (M_PUBLIC | M_PRIVATE, lambda d, _: d.model),
530 'brand-code': (M_PUBLIC, lambda d, _: d.brand_code),
531 'arc': {
532 'build-properties': {
533 'device': (M_PRIVATE, lambda d, _:
534 d.arc_build_prop('ro.product.device')),
535 'marketing-name': (M_PRIVATE, lambda d, _:
536 d.arc_build_prop('ro.product.model')),
537 'oem': (M_PRIVATE,
538 lambda d, _: d.arc_build_prop('ro.product.brand')),
539 'first-api-level': (M_PRIVATE, genconf_first_api_level),
540 'metrics-tag': (M_PRIVATE,
541 lambda d, _: d.arc_build_prop('ro.product.board')),
542 'product': (M_PRIVATE, lambda d, _:
543 d.arc_build_prop('ro.product.name')),
544 },
545 },
546 'audio': {
547 'main': {
548 'cras-config-dir': (M_PUBLIC, genconf_cras_config_dir),
549 'ucm-suffix': (M_PUBLIC, lambda d, _: d.internal_ucm_suffix),
550 },
551 },
552 'fingerprint': {
553 'board': (M_PUBLIC, genconf_fp_board),
554 'fingerprint-sensor-type': (M_PUBLIC, genconf_fp_type),
555 'sensor-location': (M_PUBLIC, genconf_fp_location),
556 },
557 'firmware': {
558 'image-name': (M_PUBLIC, lambda d, _: d.model),
559 'name': (M_PRIVATE, lambda d, _: d.model),
560 'bcs-overlay': (M_PRIVATE, lambda _, b:
561 f'overlay-{b.board_name}-private'),
562 'ec-ro-image': (M_PRIVATE, lambda _, b: b.bcs_ec),
563 'pd-ro-image': (M_PRIVATE, lambda _, b: b.bcs_pd),
564 'main-ro-image': (M_PRIVATE, lambda _, b: b.bcs_main_ro),
565 'main-rw-image': (M_PRIVATE, lambda _, b: b.bcs_main_rw),
566 'build-targets': {
567 'base': (M_PUBLIC, lambda _, b: b.detachable_base_build_target),
568 'coreboot': (M_PUBLIC, lambda _, b: b.board_name),
569 'depthcharge': (M_PUBLIC, lambda _, b: b.board_name),
570 'ec': (M_PUBLIC, lambda _, b: b.ec_build_target),
571 'ec_extras': (M_PUBLIC, lambda _, b: b.ec_extras_build_target),
572 },
573 },
574 'firmware-signing': {
575 'key-id': (M_PRIVATE, lambda d, _: d.model.upper()),
576 'signature-id': (M_PRIVATE, genconf_signature_id),
577 },
578 'hardware-properties': {
579 'psu-type': (M_PUBLIC, genconf_psu_type),
580 },
581 'identity': {
582 'platform-name': (M_PUBLIC, lambda _, b: b.mosys_platform),
Jack Rosenthal9ec0ecd2020-06-17 18:13:56 -0600583 'smbios-name-match': (M_PUBLIC, lambda d, _: d.smbios_name),
584 'device-tree-compatible-match': (M_PUBLIC, genconf_dt_compatible_match),
585 'customization-id': (M_PUBLIC, lambda d, _: d.customization_id or None),
586 'whitelabel-tag': (M_PUBLIC, genconf_whitelabel_tag),
Jack Rosenthal543bb062020-06-05 15:30:09 -0600587 },
588 'power': (M_PUBLIC, genconf_powerd_settings),
589}
590
591
592def genconf(schema, device_conf, overlay_conf):
593
594 def qualifies_as_value(v):
595 return v is not None and v != {}
596
597 if isinstance(schema, dict):
598 pub, priv = {}, {}
599 for k, v in schema.items():
600 pub_r, priv_r = genconf(v, device_conf, overlay_conf)
601 if qualifies_as_value(pub_r):
602 pub[k] = pub_r
603 if qualifies_as_value(priv_r):
604 priv[k] = priv_r
605 return pub, priv
606
607 if isinstance(schema, tuple):
608 pub, priv = None, None
609 flags, func = schema
610 value = func(device_conf, overlay_conf)
611 if flags & M_PUBLIC:
612 pub = value
613 if flags & M_PRIVATE:
614 priv = value
615 return pub, priv
616
617
618def validate_gs_uri(uri):
619 log('Validating {}...'.format(uri))
620 subprocess.run(['gsutil', 'stat', uri], check=True, stdout=subprocess.DEVNULL)
621
622
623def parse_opts(argv):
624 parser = argparse.ArgumentParser()
625 parser.add_argument('--cros-checkout',
626 type=pathlib.Path,
627 default=pathlib.Path(os.getenv('HOME')) / 'trunk',
628 help='Location of the ChromeOS checkout')
629 parser.add_argument('--dut', '-d',
630 type=str,
631 required=True,
632 help='Hostname of DUT to use for querying and testing.')
633 parser.add_argument('--dut-ssh-port', type=int, default=22,
634 help='SSH port to use on the dut.')
635 parser.add_argument('--board', '-b',
636 type=str,
637 required=True,
638 help='Board name to convert.')
639 parser.add_argument('--mosys-platform', type=str, required=True)
640 parser.add_argument('--dry-run',
641 action='store_true',
642 default=False,
643 help='Dry run')
644 return parser.parse_args(argv)
645
646
647def main(argv):
648 opts = parse_opts(argv)
649
650 overlays = BoardOverlays(opts.board, opts.cros_checkout, opts.mosys_platform)
651 dut = Dut(opts.dut, opts.cros_checkout, port=opts.dut_ssh_port)
652
653 log('Loading configuration from DUT...')
654 dut_config = DeviceConfig.from_dut(dut)
655 log('Got configuration: {}'.format(dut_config))
656
657 assert dut_config.lsb_val('CHROMEOS_RELEASE_BOARD') == opts.board
658 assert dut_config.lsb_val('CHROMEOS_RELEASE_UNIBUILD', '0') != '1'
659
660 log('Generating chromeos-config values...')
661 public_config, private_config = genconf(genconf_schema, dut_config, overlays)
662
663 public_config_yaml = format_yaml(public_config)
664 private_config_yaml = format_yaml(private_config)
665 log('Got public config: \n{}'.format(public_config_yaml))
666 log('Got private config: \n{}'.format(private_config_yaml))
667
668 log('Generating ebuilds...')
669
670 public_vpackage = generate_vpackage(('chromeos-base/chromeos-config-bsp', ))
671 private_vpackage = generate_vpackage(
672 ('chromeos-base/chromeos-config-bsp',
673 'chromeos-base/chromeos-config-bsp-private'))
674 log('Got public vpackage: \n{}'.format(public_vpackage))
675 log('Got private vpackage: \n{}'.format(private_vpackage))
676
677 public_bsp_ebuild = generate_bsp_ebuild()
678 private_bsp_ebuild = generate_bsp_ebuild(private=True)
679 log('Got public bsp_ebuild: \n{}'.format(public_bsp_ebuild))
680 log('Got private bsp_ebuild: \n{}'.format(private_bsp_ebuild))
681
682 firmware_ebuild = generate_firmware_ebuild(opts.board)
683 log('Got firmware ebuild: \n{}'.format(firmware_ebuild))
684
685 public_make_defaults = generate_make_defaults(overlays.public_make_defaults)
686 log('Got public make defaults: \n{}'.format(public_make_defaults))
687 private_make_defaults = generate_make_defaults(overlays.private_make_defaults)
688 log('Got private make defaults: \n{}'.format(private_make_defaults))
689
690 cros_config = CrosConfig(public_config_yaml, private_config_yaml)
691 firmware_srcuris = cros_config.run_host_command('get-firmware-uris')
692 log('Got firmware URIs: {}'.format(firmware_srcuris))
693
694 log('Validating firmware srcuris...')
695 for uri in firmware_srcuris.split():
696 validate_gs_uri(uri)
697
698 firmware_srcuris_path = (overlays.firmware_ebuild_path.parent
699 / 'files' / 'srcuris')
700
701 if opts.dry_run:
702 return
703
704 overlays.write_file(
705 M_PUBLIC, 'chromeos-base/chromeos-config-bsp/files/model.yaml',
706 public_config_yaml)
707 overlays.write_file(
708 M_PRIVATE, 'chromeos-base/chromeos-config-bsp-private/files/model.yaml',
709 private_config_yaml)
710 overlays.write_file(
Jack Rosenthalbe2f1272020-06-10 09:31:09 -0600711 M_PUBLIC, 'virtual/chromeos-config-bsp/chromeos-config-bsp-2.ebuild',
712 public_vpackage, make_ebuild_symlink=True)
Jack Rosenthal543bb062020-06-05 15:30:09 -0600713 overlays.write_file(
Jack Rosenthalbe2f1272020-06-10 09:31:09 -0600714 M_PRIVATE, 'virtual/chromeos-config-bsp/chromeos-config-bsp-3.ebuild',
715 private_vpackage, make_ebuild_symlink=True)
Jack Rosenthal543bb062020-06-05 15:30:09 -0600716 overlays.write_file(
717 M_PUBLIC,
718 'chromeos-base/chromeos-config-bsp/chromeos-config-bsp-9999.ebuild',
719 public_bsp_ebuild)
720 overlays.write_file(
721 M_PRIVATE,
722 'chromeos-base/chromeos-config-bsp-private/chromeos-config-bsp-private-9999.ebuild',
723 private_bsp_ebuild)
724 write_file(overlays.firmware_ebuild_path, firmware_ebuild)
725 write_file(firmware_srcuris_path, firmware_srcuris)
726 write_file(overlays.public_make_defaults_file, public_make_defaults)
727 write_file(overlays.private_make_defaults_file, private_make_defaults)
728
729
730if __name__ == '__main__':
731 main(sys.argv[1:])