4f8b339fac
This makes setup and teardown functions defined with
__attribute__((__constructor__) and __attribute__((__destructor__)) work
in normal circumstances in EFI binaries.
A couple of notes:
- it implements both the old-style .ctors/.dtors methods and the newer
style .init_array/.fini_array ELF constructor and destructor arrays,
processed in the order:
.init_array[]
.ctors[]
efi_main()
.dtors[]
.fini_array[]
- Destructors will only be called if efi_main() exits using "return";
any call to Exit() will still longjmp() past them.
- InitializeLib() has already been called before constructors run, so
they don't need to call it (and neither does anything else.) For
compatibility, it has been altered so calling it more than once is
safe.
- No attempt is made to handle any constructor or destructor with a
prototype other than "void func(void);", but note that InitializeLib
has been called, so LibImageHandle, ST, BS, and RT are set.
- The init_array/ctor/dtor/fini_array lists aren't the using the GNU
"CONSTRUCTOR" output section command, so they don't start with a size.
- The lists are individually sorted during the link stage via
SORT_BY_NAME() in the linker script.
- The default (empty) init_array/ctor/dtor/fini_array lists are padded
out to 8-byte alignment with ".p2align 3, 0", and each list always has
at least one ".long 0" at the end of it (even if it's completely
empty). As a result, they can have NULLs that need to be skipped.
The sections they're in are mergeable, so the NULLs don't have to be
exclusively at the end.
- The ia64 and mips64el arches have not been tested.
Signed-off-by: Peter Jones <pjones@redhat.com>
68 lines
1.6 KiB
C
68 lines
1.6 KiB
C
/*
|
|
* ctors.c
|
|
* Copyright 2019 Peter Jones <pjones@redhat.com>
|
|
*
|
|
*/
|
|
|
|
#include <efi.h>
|
|
#include <efilib.h>
|
|
|
|
/*
|
|
* Note that these aren't the using the GNU "CONSTRUCTOR" output section
|
|
* command, so they don't start with a size. Because of p2align and the
|
|
* end/END definitions, and the fact that they're mergeable, they can also
|
|
* have NULLs which aren't guaranteed to be at the end.
|
|
*/
|
|
extern UINTN _init_array, _init_array_end;
|
|
extern UINTN __CTOR_LIST__, __CTOR_END__;
|
|
extern UINTN _fini_array, _fini_array_end;
|
|
extern UINTN __DTOR_LIST__, __DTOR_END__;
|
|
|
|
typedef void (*funcp)(void);
|
|
|
|
static void ctors(void)
|
|
{
|
|
for (funcp *location = (void *)&_init_array; location < (funcp *)&_init_array_end; location++) {
|
|
funcp func = *location;
|
|
if (location != NULL)
|
|
func();
|
|
}
|
|
|
|
for (funcp *location = (void *)&__CTOR_LIST__; location < (funcp *)&__CTOR_END__; location++) {
|
|
funcp func = *location;
|
|
if (location != NULL)
|
|
func();
|
|
}
|
|
}
|
|
|
|
static void dtors(void)
|
|
{
|
|
for (funcp *location = (void *)&__DTOR_LIST__; location < (funcp *)&__DTOR_END__; location++) {
|
|
funcp func = *location;
|
|
if (location != NULL)
|
|
func();
|
|
}
|
|
|
|
for (funcp *location = (void *)&_fini_array; location < (funcp *)&_fini_array_end; location++) {
|
|
funcp func = *location;
|
|
if (location != NULL)
|
|
func();
|
|
}
|
|
}
|
|
|
|
extern EFI_STATUS efi_main(EFI_HANDLE image, EFI_SYSTEM_TABLE *systab);
|
|
|
|
EFI_STATUS _entry(EFI_HANDLE image, EFI_SYSTEM_TABLE *systab)
|
|
{
|
|
EFI_STATUS status;
|
|
InitializeLib(image, systab);
|
|
|
|
ctors();
|
|
status = efi_main(image, systab);
|
|
dtors();
|
|
|
|
return status;
|
|
}
|
|
|
|
// vim:fenc=utf-8:tw=75:noet
|