include/efi_selftest.h

   1 /* SPDX-License-Identifier: GPL-2.0+ */
   2 /*
   3  *  EFI application loader
   4  *
   5  *  Copyright (c) 2017 Heinrich Schuchardt <xypron.glpk@gmx.de>
   6  */
   7
   8 #ifndef _EFI_SELFTEST_H
   9 #define _EFI_SELFTEST_H
  10
  11 #include <common.h>
  12 #include <efi.h>
  13 #include <efi_api.h>
  14 #include <efi_loader.h>
  15 #include <linker_lists.h>
  16
  17 #define EFI_ST_SUCCESS 0
  18 #define EFI_ST_FAILURE 1
  19 #define EFI_ST_SUCCESS_STR L"SUCCESS"
  20
  21 /**
  22  * efi_st_printf() - print a message
  23  *
  24  * @...:        format string followed by fields to print
  25  */
  26 #define efi_st_printf(...) \
  27         (efi_st_printc(-1, __VA_ARGS__))
  28
  29 /**
  30  * efi_st_error() - prints an error message
  31  *
  32  * @...:        format string followed by fields to print
  33  */
  34 #define efi_st_error(...) \
  35         (efi_st_printc(EFI_LIGHTRED, "%s(%u):\nERROR: ", __FILE__, __LINE__), \
  36         efi_st_printc(EFI_LIGHTRED, __VA_ARGS__))
  37
  38 /**
  39  * efi_st_todo() - prints a TODO message
  40  *
  41  * @...:        format string followed by fields to print
  42  */
  43 #define efi_st_todo(...) \
  44         (efi_st_printc(EFI_YELLOW, "%s(%u):\nTODO: ", __FILE__, __LINE__), \
  45         efi_st_printc(EFI_YELLOW, __VA_ARGS__)) \
  46
  47 /**
  48  * enum efi_test_phase - phase when test will be executed
  49  *
  50  * A test may be setup and executed at boottime,
  51  * it may be setup at boottime and executed at runtime,
  52  * or it may be setup and executed at runtime.
  53  */
  54 enum efi_test_phase {
  55         /**
  56          * @EFI_EXECUTE_BEFORE_BOOTTIME_EXIT: - execute before ExitBootServices
  57          *
  58          * Setup, execute, and teardown are executed before ExitBootServices().
  59          */
  60         EFI_EXECUTE_BEFORE_BOOTTIME_EXIT = 1,
  61         /**
  62          * @EFI_SETUP_BEFORE_BOOTTIME_EXIT: - setup before ExitBootServices
  63          *
  64          * Setup is executed before ExitBootServices() while execute, and
  65          * teardown are executed after ExitBootServices().
  66          */
  67         EFI_SETUP_BEFORE_BOOTTIME_EXIT,
  68         /**
  69          * @EFI_SETTING_VIRTUAL_ADDRESS_MAP - calls SetVirtualAddressMap()
  70          * Execute calls SetVirtualAddressMap().
  71          */
  72         EFI_SETTING_VIRTUAL_ADDRESS_MAP,
  73 };
  74
  75 extern struct efi_simple_text_output_protocol *con_out;
  76 extern struct efi_simple_text_input_protocol *con_in;
  77
  78 /**
  79  * efi_st_exit_boot_services() - exit the boot services
  80  *
  81  * * The size of the memory map is determined.
  82  * * Pool memory is allocated to copy the memory map.
  83  * * The memory map is copied and the map key is obtained.
  84  * * The map key is used to exit the boot services.
  85  */
  86 void efi_st_exit_boot_services(void);
  87
  88 /**
  89  * efi_st_printc() - print a colored message
  90  *
  91  * @color:      color, see constants in efi_api.h, use -1 for no color
  92  * @fmt:        printf style format string
  93  * @...:        arguments to be printed
  94  */
  95 void efi_st_printc(int color, const char *fmt, ...)
  96                  __attribute__ ((format (__printf__, 2, 3)));
  97
  98 /**
  99  * efi_st_translate_char() - translate a Unicode character to a string
 100  *
 101  * @code:       Unicode character
 102  * Return:      string
 103  */
 104 u16 *efi_st_translate_char(u16 code);
 105
 106 /**
 107  * efi_st_translate_code() - translate a scan code to a human readable string
 108  *
 109  * This function translates the scan code returned by the simple text input
 110  * protocol to a human readable string, e.g. 0x04 is translated to L"Left".
 111  *
 112  * @code:       scan code
 113  * Return:      Unicode string
 114  */
 115 u16 *efi_st_translate_code(u16 code);
 116
 117 /**
 118  * efi_st_strcmp_16_8() - compare an u16 string to a char string
 119  *
 120  * This function compares each u16 value to the char value at the same
 121  * position. This function is only useful for ANSI strings.
 122  *
 123  * @buf1:       u16 string
 124  * @buf2:       char string
 125  * Return:      0 if both buffers contain equivalent strings
 126  */
 127 int efi_st_strcmp_16_8(const u16 *buf1, const char *buf2);
 128
 129 /**
 130  * efi_st_get_key() - reads an Unicode character from the input device
 131  *
 132  * Return:      Unicode character
 133  */
 134 u16 efi_st_get_key(void);
 135
 136 /**
 137  * struct efi_unit_test - EFI unit test
 138  *
 139  * The &struct efi_unit_test structure provides a interface to an EFI unit test.
 140  *
 141  * @name:       name of the unit test used in the user interface
 142  * @phase:      specifies when setup and execute are executed
 143  * @setup:      set up function of the unit test
 144  * @execute:    execute function of the unit test
 145  * @teardown:   tear down function of the unit test
 146  * @on_request: flag indicating that the test shall only be executed on request
 147  */
 148 struct efi_unit_test {
 149         const char *name;
 150         const enum efi_test_phase phase;
 151         int (*setup)(const efi_handle_t handle,
 152                      const struct efi_system_table *systable);
 153         int (*execute)(void);
 154         int (*teardown)(void);
 155         bool on_request;
 156 };
 157
 158 /**
 159  * EFI_UNIT_TEST() - macro to declare a new EFI unit test
 160  *
 161  * The macro EFI_UNIT_TEST() declares an EFI unit test using the &struct
 162  * efi_unit_test structure. The test is added to a linker generated list which
 163  * is evaluated by the 'bootefi selftest' command.
 164  *
 165  * @__name:     string identifying the unit test in the linker generated list
 166  */
 167 #define EFI_UNIT_TEST(__name)                                           \
 168         ll_entry_declare(struct efi_unit_test, __name, efi_unit_test)
 169
 170 #endif /* _EFI_SELFTEST_H */