12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788 |
- // SPDX-License-Identifier: GPL-2.0+
- /*
- * efi_selftest_exitbootservices
- *
- * Copyright (c) 2017 Heinrich Schuchardt <xypron.glpk@gmx.de>
- *
- * This unit test checks that the notification function of an
- * EVT_SIGNAL_EXIT_BOOT_SERVICES event is called exactly once.
- */
- #include <efi_selftest.h>
- static struct efi_boot_services *boottime;
- static struct efi_event *event_notify;
- static unsigned int notification_count;
- /*
- * Notification function, increments the notification count.
- *
- * @event notified event
- * @context pointer to the notification count
- */
- static void EFIAPI notify(struct efi_event *event, void *context)
- {
- unsigned int *count = context;
- ++*count;
- }
- /*
- * Setup unit test.
- *
- * Create an EVT_SIGNAL_EXIT_BOOT_SERVICES event.
- *
- * @handle: handle of the loaded image
- * @systable: system table
- * @return: EFI_ST_SUCCESS for success
- */
- static int setup(const efi_handle_t handle,
- const struct efi_system_table *systable)
- {
- efi_status_t ret;
- boottime = systable->boottime;
- notification_count = 0;
- ret = boottime->create_event(EVT_SIGNAL_EXIT_BOOT_SERVICES,
- TPL_CALLBACK, notify,
- (void *)¬ification_count,
- &event_notify);
- if (ret != EFI_SUCCESS) {
- efi_st_error("could not create event\n");
- return EFI_ST_FAILURE;
- }
- return EFI_ST_SUCCESS;
- }
- /*
- * Execute unit test.
- *
- * Check that the notification function of the EVT_SIGNAL_EXIT_BOOT_SERVICES
- * event has been called.
- *
- * Call ExitBootServices again and check that the notification function is
- * not called again.
- *
- * @return: EFI_ST_SUCCESS for success
- */
- static int execute(void)
- {
- if (notification_count != 1) {
- efi_st_error("ExitBootServices was not notified\n");
- return EFI_ST_FAILURE;
- }
- efi_st_exit_boot_services();
- if (notification_count != 1) {
- efi_st_error("ExitBootServices was notified twice\n");
- return EFI_ST_FAILURE;
- }
- return EFI_ST_SUCCESS;
- }
- EFI_UNIT_TEST(exitbootservices) = {
- .name = "ExitBootServices",
- .phase = EFI_SETUP_BEFORE_BOOTTIME_EXIT,
- .setup = setup,
- .execute = execute,
- };
|