efi_loader: Test that active-DMA devices are removed

When exit-boot-services is called, active devices should be removed.

Update testapp to call this method and check that the USB controller is
removed as expected.

Signed-off-by: Simon Glass <sjg@chromium.org>
This commit is contained in:
Simon Glass
2025-01-06 07:47:55 -07:00
parent 6472fd0733
commit 65091e5d24
2 changed files with 20 additions and 0 deletions

View File

@@ -76,6 +76,15 @@ out:
return ret;
}
/* exit boot services so that this part of U-Boot can be tested */
con_out->output_string(con_out, u"Exiting boot services\n");
ret = boottime->exit_boot_services(handle, map_key);
if (ret) {
con_out->output_string(con_out, u"Failed exit-boot-services\n");
return ret;
}
/* now exit for real */
con_out->output_string(con_out, u"Exiting test app\n");
ret = boottime->exit(handle, ret, 0, NULL);

View File

@@ -23,6 +23,7 @@
#endif
#include <dm/device-internal.h>
#include <dm/lists.h>
#include <dm/uclass-internal.h>
#include <linux/libfdt.h>
#include <test/suites.h>
#include <test/ut.h>
@@ -1279,6 +1280,7 @@ static int bootflow_efi(struct unit_test_state *uts)
struct bootstd_priv *std;
struct udevice *bootstd;
const char **old_order;
struct udevice *usb;
int i;
ut_assertok(uclass_first_device_err(UCLASS_BOOTSTD, &bootstd));
@@ -1315,6 +1317,10 @@ static int bootflow_efi(struct unit_test_state *uts)
systab.fw_vendor = test_vendor;
/* the USB block-device should have beeen probed */
ut_assertok(uclass_find_device_by_seq(UCLASS_USB, 1, &usb));
ut_assert(device_active(usb));
ut_asserteq(1, run_command("bootflow boot", 0));
ut_assert_nextline(
"** Booting bootflow 'usb_mass_storage.lun0.bootdev.part_1' with efi");
@@ -1325,6 +1331,7 @@ static int bootflow_efi(struct unit_test_state *uts)
/* TODO: Why the \r ? */
ut_assert_nextline("U-Boot test app for EFI_LOADER\r");
ut_assert_nextline("Exiting boot services");
ut_assert_nextline("Exiting test app");
ut_assert_nextline("Boot failed (err=-14)");
@@ -1333,6 +1340,10 @@ static int bootflow_efi(struct unit_test_state *uts)
/* make sure the bootflow is still present */
ut_assertnonnull(std->cur_bootflow);
/* the USB block-device should have beeen removed */
ut_assertok(uclass_find_device_by_seq(UCLASS_USB, 1, &usb));
ut_assert(!device_active(usb));
/* check memory allocations are as expected */
if (!hdr)
return 0;