efi_init.c

Go to the documentation of this file.

/*
 * Copyright (C) 2008 Michael Brown <mbrown@fensystems.co.uk>.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of the
 * License, or any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301, USA.
 */

FILE_LICENCE ( GPL2_OR_LATER );

#include <string.h>
#include <errno.h>
#include <endian.h>
#include <ipxe/init.h>
#include <ipxe/rotate.h>
#include <ipxe/efi/efi.h>
#include <ipxe/efi/efi_driver.h>
#include <ipxe/efi/efi_path.h>
#include <ipxe/efi/efi_cmdline.h>
#include <ipxe/efi/Protocol/LoadedImage.h>

/** Image handle passed to entry point */
EFI_HANDLE efi_image_handle;

/** Loaded image protocol for this image */
EFI_LOADED_IMAGE_PROTOCOL *efi_loaded_image;

/** Device path for the loaded image's device handle */
EFI_DEVICE_PATH_PROTOCOL *efi_loaded_image_path;

/** System table passed to entry point
 *
 * We construct the symbol name efi_systab via the PLATFORM macro.
 * This ensures that the symbol is defined only in EFI builds, and so
 * prevents EFI code from being incorrectly linked in to a non-EFI
 * build.
 */
EFI_SYSTEM_TABLE * _C2 ( PLATFORM, _systab );

/** Internal task priority level */
EFI_TPL efi_internal_tpl = TPL_CALLBACK;

/** External task priority level */
EFI_TPL efi_external_tpl = TPL_APPLICATION;

/** EFI shutdown is in progress */
int efi_shutdown_in_progress;

/** Event used to signal shutdown */
static EFI_EVENT efi_shutdown_event;

/** Stack cookie */
unsigned long __stack_chk_guard;

/** Exit function
 *
 * Cached to minimise external dependencies when a stack check
 * failure is triggered.
 */
static EFI_EXIT efi_exit;

/* Forward declarations */
static EFI_STATUS EFIAPI efi_unload ( EFI_HANDLE image_handle );

/**
 * Shut down in preparation for booting an OS.
 *
 * This hook gets called at ExitBootServices time in order to make
 * sure that everything is properly shut down before the OS takes
 * over.
 */
static EFIAPI void efi_shutdown_hook ( EFI_EVENT event __unused,
                                       void *context __unused ) {

        /* This callback is invoked at TPL_NOTIFY in order to ensure
         * that we have an opportunity to shut down cleanly before
         * other shutdown hooks perform destructive operations such as
         * disabling the IOMMU.
         *
         * Modify the internal task priority level so that no code
         * attempts to raise from TPL_NOTIFY to TPL_CALLBACK (which
         * would trigger a fatal exception).
         */
        efi_internal_tpl = TPL_NOTIFY;

        /* Mark shutdown as being in progress, to indicate that large
         * parts of the system (e.g. timers) are no longer functional.
         */
        efi_shutdown_in_progress = 1;

        /* Shut down iPXE */
        shutdown_boot();
}

/**
 * Look up EFI configuration table
 *
 * @v guid              Configuration table GUID
 * @ret table           Configuration table, or NULL
 */
static void * efi_find_table ( EFI_GUID *guid ) {
        unsigned int i;

        for ( i = 0 ; i < efi_systab->NumberOfTableEntries ; i++ ) {
                if ( memcmp ( &efi_systab->ConfigurationTable[i].VendorGuid,
                              guid, sizeof ( *guid ) ) == 0 )
                        return efi_systab->ConfigurationTable[i].VendorTable;
        }

        return NULL;
}

/**
 * Construct a stack cookie value
 *
 * @v handle            Image handle
 * @ret cookie          Stack cookie
 */
__attribute__ (( noinline )) unsigned long
efi_stack_cookie ( EFI_HANDLE handle ) {
        unsigned long cookie = 0;
        unsigned int rotation = ( 8 * sizeof ( cookie ) / 4 );

        /* There is no viable source of entropy available at this
         * point.  Construct a value that is at least likely to vary
         * between platforms and invocations.
         */
        cookie ^= ( ( unsigned long ) handle );
        cookie = roll ( cookie, rotation );
        cookie ^= ( ( unsigned long ) &handle );
        cookie = roll ( cookie, rotation );
        cookie ^= profile_timestamp();
        cookie = roll ( cookie, rotation );
        cookie ^= build_id;

        /* Ensure that the value contains a NUL byte, to act as a
         * runaway string terminator.  Construct the NUL using a shift
         * rather than a mask, to avoid losing valuable entropy in the
         * lower-order bits.
         */
        cookie <<= 8;

        /* Ensure that the NUL byte is placed at the bottom of the
         * stack cookie, to avoid potential disclosure via an
         * unterminated string.
         */
        if ( __BYTE_ORDER == __BIG_ENDIAN )
                cookie >>= 8;

        return cookie;
}

/**
 * Initialise EFI environment
 *
 * @v image_handle      Image handle
 * @v systab            System table
 * @ret efirc           EFI return status code
 */
EFI_STATUS efi_init ( EFI_HANDLE image_handle,
                      EFI_SYSTEM_TABLE *systab ) {
        EFI_BOOT_SERVICES *bs;
        struct efi_protocol *prot;
        struct efi_config_table *tab;
        void *loaded_image;
        void *device_path;
        void *device_path_copy;
        size_t device_path_len;
        EFI_STATUS efirc;
        int rc;

        /* Store image handle and system table pointer for future use */
        efi_image_handle = image_handle;
        efi_systab = systab;

        /* Sanity checks */
        if ( ! systab ) {
                efirc = EFI_NOT_AVAILABLE_YET;
                goto err_sanity;
        }
        if ( ! systab->ConOut ) {
                efirc = EFI_NOT_AVAILABLE_YET;
                goto err_sanity;
        }
        if ( ! systab->BootServices ) {
                DBGC ( systab, "EFI provided no BootServices entry point\n" );
                efirc = EFI_NOT_AVAILABLE_YET;
                goto err_sanity;
        }
        if ( ! systab->RuntimeServices ) {
                DBGC ( systab, "EFI provided no RuntimeServices entry "
                       "point\n" );
                efirc = EFI_NOT_AVAILABLE_YET;
                goto err_sanity;
        }
        DBGC ( systab, "EFI handle %p systab %p\n", image_handle, systab );
        bs = systab->BootServices;

        /* Store abort function pointer */
        efi_exit = bs->Exit;

        /* Look up used protocols */
        for_each_table_entry ( prot, EFI_PROTOCOLS ) {
                if ( ( efirc = bs->LocateProtocol ( &prot->guid, NULL,
                                                    prot->protocol ) ) == 0 ) {
                        DBGC ( systab, "EFI protocol %s is at %p\n",
                               efi_guid_ntoa ( &prot->guid ),
                               *(prot->protocol) );
                } else {
                        DBGC ( systab, "EFI does not provide protocol %s\n",
                               efi_guid_ntoa ( &prot->guid ) );
                        /* Fail if protocol is required */
                        if ( prot->required )
                                goto err_missing_protocol;
                }
        }

        /* Look up used configuration tables */
        for_each_table_entry ( tab, EFI_CONFIG_TABLES ) {
                if ( ( *(tab->table) = efi_find_table ( &tab->guid ) ) ) {
                        DBGC ( systab, "EFI configuration table %s is at %p\n",
                               efi_guid_ntoa ( &tab->guid ), *(tab->table) );
                } else {
                        DBGC ( systab, "EFI does not provide configuration "
                               "table %s\n", efi_guid_ntoa ( &tab->guid ) );
                        if ( tab->required ) {
                                efirc = EFI_NOT_AVAILABLE_YET;
                                goto err_missing_table;
                        }
                }
        }

        /* Get loaded image protocol */
        if ( ( efirc = bs->OpenProtocol ( image_handle,
                                &efi_loaded_image_protocol_guid,
                                &loaded_image, image_handle, NULL,
                                EFI_OPEN_PROTOCOL_GET_PROTOCOL ) ) != 0 ) {
                rc = -EEFI ( efirc );
                DBGC ( systab, "EFI could not get loaded image protocol: %s",
                       strerror ( rc ) );
                goto err_no_loaded_image;
        }
        efi_loaded_image = loaded_image;
        DBGC ( systab, "EFI image base address %p\n",
               efi_loaded_image->ImageBase );

        /* Record command line */
        efi_cmdline = efi_loaded_image->LoadOptions;
        efi_cmdline_len = efi_loaded_image->LoadOptionsSize;

        /* Get loaded image's device handle's device path */
        if ( ( efirc = bs->OpenProtocol ( efi_loaded_image->DeviceHandle,
                                &efi_device_path_protocol_guid,
                                &device_path, image_handle, NULL,
                                EFI_OPEN_PROTOCOL_GET_PROTOCOL ) ) != 0 ) {
                rc = -EEFI ( efirc );
                DBGC ( systab, "EFI could not get loaded image's device path: "
                       "%s", strerror ( rc ) );
                goto err_no_device_path;
        }

        /* Make a copy of the loaded image's device handle's device
         * path, since the device handle itself may become invalidated
         * when we load our own drivers.
         */
        device_path_len = ( efi_path_len ( device_path ) +
                            sizeof ( EFI_DEVICE_PATH_PROTOCOL ) );
        if ( ( efirc = bs->AllocatePool ( EfiBootServicesData, device_path_len,
                                          &device_path_copy ) ) != 0 ) {
                rc = -EEFI ( efirc );
                goto err_alloc_device_path;
        }
        memcpy ( device_path_copy, device_path, device_path_len );
        efi_loaded_image_path = device_path_copy;
        DBGC ( systab, "EFI image device path %s\n",
               efi_devpath_text ( efi_loaded_image_path ) );

        /* EFI is perfectly capable of gracefully shutting down any
         * loaded devices if it decides to fall back to a legacy boot.
         * For no particularly comprehensible reason, it doesn't
         * bother doing so when ExitBootServices() is called.
         */
        if ( ( efirc = bs->CreateEvent ( EVT_SIGNAL_EXIT_BOOT_SERVICES,
                                         TPL_NOTIFY, efi_shutdown_hook,
                                         NULL, &efi_shutdown_event ) ) != 0 ) {
                rc = -EEFI ( efirc );
                DBGC ( systab, "EFI could not create ExitBootServices event: "
                       "%s\n", strerror ( rc ) );
                goto err_create_event;
        }

        /* Install driver binding protocol */
        if ( ( rc = efi_driver_install() ) != 0 ) {
                DBGC ( systab, "EFI could not install driver: %s\n",
                       strerror ( rc ) );
                efirc = EFIRC ( rc );
                goto err_driver_install;
        }

        /* Install image unload method */
        efi_loaded_image->Unload = efi_unload;

        return 0;

        efi_driver_uninstall();
 err_driver_install:
        bs->CloseEvent ( efi_shutdown_event );
 err_create_event:
        bs->FreePool ( efi_loaded_image_path );
 err_alloc_device_path:
 err_no_device_path:
 err_no_loaded_image:
 err_missing_table:
 err_missing_protocol:
 err_sanity:
        return efirc;
}

/**
 * Shut down EFI environment
 *
 * @v image_handle      Image handle
 */
static EFI_STATUS EFIAPI efi_unload ( EFI_HANDLE image_handle __unused ) {
        EFI_BOOT_SERVICES *bs = efi_systab->BootServices;
        EFI_SYSTEM_TABLE *systab = efi_systab;
        struct efi_saved_tpl tpl;

        DBGC ( systab, "EFI image unloading\n" );

        /* Raise TPL */
        efi_raise_tpl ( &tpl );

        /* Shut down */
        shutdown_exit();

        /* Disconnect any remaining devices */
        efi_driver_disconnect_all();

        /* Uninstall driver binding protocol */
        efi_driver_uninstall();

        /* Uninstall exit boot services event */
        bs->CloseEvent ( efi_shutdown_event );

        /* Free copy of loaded image's device handle's device path */
        bs->FreePool ( efi_loaded_image_path );

        DBGC ( systab, "EFI image unloaded\n" );

        /* Restore TPL */
        efi_restore_tpl ( &tpl );

        return 0;
}

/**
 * Abort on stack check failure
 *
 */
__attribute__ (( noreturn )) void __stack_chk_fail ( void ) {
        EFI_STATUS efirc;
        int rc;

        /* Report failure (when debugging) */
        DBGC ( efi_systab, "EFI stack check failed (cookie %#lx); aborting\n",
               __stack_chk_guard );

        /* Attempt to exit cleanly with an error status */
        if ( efi_exit ) {
                efirc = efi_exit ( efi_image_handle, EFI_COMPROMISED_DATA,
                                   0, NULL );
                rc = -EEFI ( efirc );
                DBGC ( efi_systab, "EFI stack check exit failed: %s\n",
                       strerror ( rc ) );
        }

        /* If the exit fails for any reason, lock the system */
        while ( 1 ) {}

}

/**
 * Raise task priority level to internal level
 *
 * @v tpl               Saved TPL
 */
void efi_raise_tpl ( struct efi_saved_tpl *tpl ) {
        EFI_BOOT_SERVICES *bs = efi_systab->BootServices;

        /* Record current external TPL */
        tpl->previous = efi_external_tpl;

        /* Raise TPL and record previous TPL as new external TPL */
        tpl->current = bs->RaiseTPL ( efi_internal_tpl );
        efi_external_tpl = tpl->current;
}

/**
 * Restore task priority level
 *
 * @v tpl               Saved TPL
 */
void efi_restore_tpl ( struct efi_saved_tpl *tpl ) {
        EFI_BOOT_SERVICES *bs = efi_systab->BootServices;

        /* Restore external TPL */
        efi_external_tpl = tpl->previous;

        /* Restore TPL */
        bs->RestoreTPL ( tpl->current );
}