The current localboot implementation assumes that a 'localcmd' environment variable is provided, with the instructions to follow. This may not be included, so provide a fallback in that case.
Add a test image and test as well. Signed-off-by: Simon Glass <s...@chromium.org> --- arch/sandbox/dts/test.dts | 8 +++++++ boot/Kconfig | 9 ++++++++ boot/pxe_utils.c | 33 ++++++++++++++++++++++++--- test/boot/bootflow.c | 47 +++++++++++++++++++++++++++++++++++++++ test/py/tests/test_ut.py | 17 ++++++++++++++ 5 files changed, 111 insertions(+), 3 deletions(-) diff --git a/arch/sandbox/dts/test.dts b/arch/sandbox/dts/test.dts index 36cfbf213e4..47e17070886 100644 --- a/arch/sandbox/dts/test.dts +++ b/arch/sandbox/dts/test.dts @@ -45,6 +45,7 @@ mmc6 = "/mmc6"; mmc7 = "/mmc7"; mmc8 = "/mmc8"; + mmc9 = "/mmc9"; pci0 = &pci0; pci1 = &pci1; pci2 = &pci2; @@ -1153,6 +1154,13 @@ filename = "mmc8.img"; }; + /* This is used for extlinux localboot */ + mmc9 { + status = "disabled"; + compatible = "sandbox,mmc"; + filename = "mmc9.img"; + }; + pch { compatible = "sandbox,pch"; }; diff --git a/boot/Kconfig b/boot/Kconfig index 705947cfa95..1356699021a 100644 --- a/boot/Kconfig +++ b/boot/Kconfig @@ -556,6 +556,15 @@ config BOOTMETH_EXTLINUX_PXE This provides a way to try out standard boot on an existing boot flow. +config BOOTMETH_EXTLINUX_LOCALBOOT + bool "Boot method for extlinux localboot" + depends on BOOTMETH_EXTLINUX + default y + help + Enables standard boot support for the extlinux 'localboot' feature. + This attempts to find a kernel and initrd on the disk and boot it, + in the case where there is no "localcmd" in the environment. + config BOOTMETH_EFILOADER bool "Bootdev support for EFI boot" depends on EFI_BINARY_EXEC diff --git a/boot/pxe_utils.c b/boot/pxe_utils.c index 8bebf4ec925..5d52a5965d6 100644 --- a/boot/pxe_utils.c +++ b/boot/pxe_utils.c @@ -640,6 +640,25 @@ static int label_run_boot(struct pxe_context *ctx, struct pxe_label *label, return 0; } +/** + * generate_localboot() - Try to come up with a localboot definition + * + * Adds a default kernel and initrd filename for use with localboot + * + * @label: Label to process + * Return 0 if OK, -ENOMEM if out of memory + */ +static int generate_localboot(struct pxe_label *label) +{ + label->kernel = strdup("/vmlinuz"); + label->kernel_label = strdup(label->kernel); + label->initrd = strdup("/initrd.img"); + if (!label->kernel || !label->kernel_label || !label->initrd) + return -ENOMEM; + + return 0; +} + /** * label_boot() - Boot according to the contents of a pxe_label * @@ -677,9 +696,17 @@ static int label_boot(struct pxe_context *ctx, struct pxe_label *label) label->attempted = 1; if (label->localboot) { - if (label->localboot_val >= 0) - label_localboot(label); - return 0; + if (label->localboot_val >= 0) { + ret = label_localboot(label); + + if (IS_ENABLED(CONFIG_BOOTMETH_EXTLINUX_LOCALBOOT) && + ret == -ENOENT) + ret = generate_localboot(label); + if (ret) + return ret; + } else { + return 0; + } } if (!label->kernel) { diff --git a/test/boot/bootflow.c b/test/boot/bootflow.c index 4d7d795cbe1..db1af0e5729 100644 --- a/test/boot/bootflow.c +++ b/test/boot/bootflow.c @@ -1510,3 +1510,50 @@ static int bootflow_scan_extlinux(struct unit_test_state *uts) return 0; } BOOTSTD_TEST(bootflow_scan_extlinux, UTF_DM | UTF_SCAN_FDT | UTF_CONSOLE); + +/* Check automatically generating a extlinux 'localboot' */ +static int bootflow_extlinux_localboot(struct unit_test_state *uts) +{ + const struct bootflow_img *img; + struct bootstd_priv *std; + const char **old_order; + struct bootflow *bflow; + + ut_assertok(prep_mmc_bootdev(uts, "mmc9", false, &old_order)); + + ut_assertok(run_command("bootflow scan", 0)); + ut_assert_console_end(); + + /* Restore the order used by the device tree */ + ut_assertok(bootstd_get_priv(&std)); + free(std->bootdev_order); + std->bootdev_order = old_order; + + /* boot the second bootflow */ + ut_asserteq(2, std->bootflows.count); + bflow = alist_getw(&std->bootflows, 1, struct bootflow); + std->cur_bootflow = bflow; + + /* read all the images, but don't actually boot */ + ut_assertok(bootflow_read_all(bflow)); + ut_assert_nextline("1:\tlocal"); + ut_assert_nextline("missing environment variable: localcmd"); + ut_assert_nextline("Retrieving file: /vmlinuz"); + ut_assert_nextline("Retrieving file: /initrd.img"); + + ut_assert_console_end(); + + ut_asserteq(3, bflow->images.count); + + /* check the two localboot images */ + img = alist_get(&bflow->images, 1, struct bootflow_img); + ut_asserteq(IH_TYPE_KERNEL, img->type); + ut_asserteq(0x1000000, img->addr); /* kernel_addr_r */ + + img = alist_get(&bflow->images, 2, struct bootflow_img); + ut_asserteq(IH_TYPE_RAMDISK, img->type); + ut_asserteq(0x2000000, img->addr); /* ramdisk_addr_r */ + + return 0; +} +BOOTSTD_TEST(bootflow_extlinux_localboot, UTF_DM | UTF_SCAN_FDT | UTF_CONSOLE); diff --git a/test/py/tests/test_ut.py b/test/py/tests/test_ut.py index e3b988efe23..b6b4717c834 100644 --- a/test/py/tests/test_ut.py +++ b/test/py/tests/test_ut.py @@ -575,6 +575,22 @@ def setup_efi_image(cons): u_boot_utils.run_and_log(cons, f'rm -rf {mnt}') u_boot_utils.run_and_log(cons, f'rm -f {fsfile}') + +def setup_localboot_image(cons): + """Create a 20MB disk image with a single FAT partition""" + mmc_dev = 9 + fname, mnt = setup_image(cons, mmc_dev, 0xc, second_part=True) + + script = '''DEFAULT local + +LABEL local + LOCALBOOT 0 +''' + vmlinux = 'vmlinuz' + initrd = 'initrd.img' + setup_extlinux_image(cons, mmc_dev, vmlinux, initrd, None, script) + + @pytest.mark.buildconfigspec('cmd_bootflow') @pytest.mark.buildconfigspec('sandbox') def test_ut_dm_init_bootstd(u_boot_console): @@ -586,6 +602,7 @@ def test_ut_dm_init_bootstd(u_boot_console): setup_cros_image(u_boot_console) setup_android_image(u_boot_console) setup_efi_image(u_boot_console) + setup_localboot_image(u_boot_console) # Restart so that the new mmc1.img is picked up u_boot_console.restart_uboot() -- 2.34.1