huangcm
2025-02-24 69ed55dec4b2116a19e4cca4393cbc014fce5fb2
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
// 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 *)&notification_count,
                    &event_notify);
   if (ret != EFI_SUCCESS) {
       efi_st_error("could not create event\n");
       return EFI_ST_FAILURE;
   }
   return EFI_ST_SUCCESS;
}
 
/*
 * Tear down unit test.
 *
 * Close the event created in setup.
 *
 * @return:    EFI_ST_SUCCESS for success
 */
static int teardown(void)
{
   efi_status_t ret;
 
   if (event_notify) {
       ret = boottime->close_event(event_notify);
       event_notify = NULL;
       if (ret != EFI_SUCCESS) {
           efi_st_error("could not close 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,
   .teardown = teardown,
};