efi_selftests: fix controllers repeated selftesting

Running the controller selftest more than one times fails with

=> setenv efi_selftest 'controllers' && bootefi selftest
Testing EFI API implementation
Selected test: 'controllers'
Setting up 'controllers'
Setting up 'controllers' succeeded
Executing 'controllers'
Executing 'controllers' succeeded
Summary: 0 failures

=> bootefi selftest
Testing EFI API implementation
Selected test: 'controllers'
Setting up 'controllers'
lib/efi_selftest/efi_selftest_controllers.c(280):
ERROR: InstallProtocolInterface failed
lib/efi_selftest/efi_selftest.c(89):
ERROR: Setting up 'controllers' failed
Summary: 1 failures

There are multiple reason for this.  We don't uninstall the binding
interface from the controller handle and we don't reset the handle
pointers either.  So let's uninstall all the protocols properly and
reset the handles to NULL on setup().

While at it add a forgotten check when uninstalling protocols from the
handle_controller and make sure the number of child controllers is 0

Signed-off-by: Ilias Apalodimas <ilias.apalodimas@linaro.org>
Reviewed-by: Heinrich Schuchardt <heinrich.schuchardt@canonical.com>
Signed-off-by: Heinrich Schuchardt <heinrich.schuchardt@canonical.com>

diff --git a/lib/efi_selftest/efi_selftest_controllers.c b/lib/efi_selftest/efi_selftest_controllers.c
index d2bbd1c4f65b30624f032515ef062d2076355854..63e674bedc0c663fc256f76c68a5cdf96993177b 100644 (file)
--- a/lib/efi_selftest/efi_selftest_controllers.c
+++ b/lib/efi_selftest/efi_selftest_controllers.c
@@ -271,6 +271,8 @@ static int setup(const efi_handle_t img_handle,
        efi_status_t ret;
 
        boottime = systable->boottime;
+       handle_controller =  NULL;
+       handle_driver = NULL;
 
        /* Create controller handle */
        ret = boottime->install_protocol_interface(
@@ -402,14 +404,36 @@ static int execute(void)
        /* Check number of child controllers */
        ret = count_child_controllers(handle_controller, &guid_controller,
                                      &count);
-       if (ret == EFI_SUCCESS)
+       if (ret == EFI_SUCCESS || count) {
                efi_st_error("Uninstall failed\n");
+               return EFI_ST_FAILURE;
+       }
+
+
        return EFI_ST_SUCCESS;
 }
 
+  /*
+   * Tear down unit test.
+   *
+   */
+static int teardown(void)
+{
+       efi_status_t ret;
+       /* Uninstall binding protocol */
+       ret = boottime->uninstall_protocol_interface(handle_driver,
+                                                    &guid_driver_binding_protocol,
+                                                    &binding_interface);
+       if (ret != EFI_SUCCESS)
+               efi_st_error("Failed to uninstall protocols\n");
+
+       return ret;
+}
+
 EFI_UNIT_TEST(controllers) = {
        .name = "controllers",
        .phase = EFI_EXECUTE_BEFORE_BOOTTIME_EXIT,
        .setup = setup,
        .execute = execute,
+       .teardown = teardown,
 };