Open Menu / test / py / img / bls.py
Loading...
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
# SPDX-License-Identifier: GPL-2.0+
# Copyright 2026 Canonical Ltd

"""Create BLS test disk images"""

import gzip
import os

import utils
from fs_helper import DiskHelper, FsHelper
from img.common import mkdir_cond


def setup_bls_image(config, log, devnum, basename):
    """Create a 20MB BLS disk image with a single FAT partition

    Args:
        config (ArbitraryAttributeContainer): Configuration
        log (multiplexed_log.Logfile): Log to write to
        devnum (int): Device number to use, e.g. 5
        basename (str): Base name to use in the filename, e.g. 'mmc'
    """
    vmlinux = 'vmlinuz-6.8.0'
    initrd = 'initrd.img-6.8.0'
    dtb = 'sandbox.dtb'

    # BLS Type #1 entry format
    entry1 = f'''title Test Boot
version 6.8.0
linux /{vmlinux}
options root=/dev/mmcblk0p2 ro quiet
initrd /{initrd}
devicetree /{dtb}'''

    entry2 = f'''title Rescue Boot
version 6.8.0-rescue
linux /{vmlinux}
options root=/dev/mmcblk0p2 ro quiet single
initrd /{initrd}
devicetree /{dtb}'''

    fsh = FsHelper(config, 'vfat', 18, prefix=basename)
    fsh.setup()

    # Create loader/entries directory for BLS entries
    loader = os.path.join(fsh.srcdir, 'loader')
    mkdir_cond(loader)
    entries = os.path.join(loader, 'entries')
    mkdir_cond(entries)

    # Create two BLS entry files
    with open(os.path.join(entries, '6.8.0.conf'), 'w',
              encoding='ascii') as fd:
        print(entry1, file=fd)
    with open(os.path.join(entries, '6.8.0-rescue.conf'), 'w',
              encoding='ascii') as fd:
        print(entry2, file=fd)

    # Create compressed kernel image
    inf = os.path.join(config.persistent_data_dir, 'inf')
    with open(inf, 'wb') as fd:
        fd.write(gzip.compress(b'vmlinux'))
    mkimage = config.build_dir + '/tools/mkimage'
    utils.run_and_log_no_ubman(
        log, f'{mkimage} -f auto -d {inf} {os.path.join(fsh.srcdir, vmlinux)}')

    # Create initrd file
    with open(os.path.join(fsh.srcdir, initrd), 'w', encoding='ascii') as fd:
        print('initrd', file=fd)

    # Create device tree blob
    dtb_file = os.path.join(fsh.srcdir, dtb)
    utils.run_and_log_no_ubman(
        log, f'dtc -o {dtb_file}', stdin=b'/dts-v1/; / {};')

    fsh.mk_fs()

    # Create disk image with single bootable partition
    img = DiskHelper(config, devnum, basename)
    img.add_fs(fsh, DiskHelper.VFAT, bootable=True)
    img.create()
    fsh.cleanup()