LXR uboot/lib/efi_selftest/efi_selftest_miniapp

   1// SPDX-License-Identifier: GPL-2.0+
   2/*
   3 * efi_selftest_miniapp_exit
   4 *
   5 * Copyright (c) 2018 Heinrich Schuchardt
   6 *
   7 * This EFI application is run by the StartImage selftest.
   8 * It uses the Exit boot service to return.
   9 */
  10
  11#include <common.h>
  12#include <efi_selftest.h>
  13
  14static efi_guid_t loaded_image_protocol_guid = EFI_LOADED_IMAGE_PROTOCOL_GUID;
  15
  16/**
  17 * check_loaded_image_protocol() - check image_base/image_size
  18 *
  19 * Try to open the loaded image protocol. Check that this function is located
  20 * between image_base and image_base + image_size.
  21 *
  22 * @image_handle:       handle of the loaded image
  23 * @systable:           system table
  24 * @return:             status code
  25 */
  26static efi_status_t EFIAPI check_loaded_image_protocol
  27                (efi_handle_t image_handle, struct efi_system_table *systable)
  28{
  29        struct efi_simple_text_output_protocol *cout = systable->con_out;
  30        struct efi_boot_services *boottime = systable->boottime;
  31        struct efi_loaded_image *loaded_image_protocol;
  32        efi_status_t ret;
  33
  34        /*
  35         * Open the loaded image protocol.
  36         */
  37        ret = boottime->open_protocol
  38                                (image_handle, &loaded_image_protocol_guid,
  39                                 (void **)&loaded_image_protocol, NULL,
  40                                  NULL, EFI_OPEN_PROTOCOL_GET_PROTOCOL);
  41        if (ret != EFI_SUCCESS) {
  42                cout->output_string(cout,
  43                                    L"Could not open loaded image protocol");
  44                return ret;
  45        }
  46        if ((void *)check_loaded_image_protocol <
  47            loaded_image_protocol->image_base ||
  48            (void *)check_loaded_image_protocol >=
  49            loaded_image_protocol->image_base +
  50            loaded_image_protocol->image_size) {
  51                cout->output_string(cout,
  52                                    L"Incorrect image_base or image_size\n");
  53                return EFI_NOT_FOUND;
  54        }
  55        return EFI_SUCCESS;
  56}
  57
  58/**
  59 * Entry point of the EFI application.
  60 *
  61 * @handle:     handle of the loaded image
  62 * @systable:   system table
  63 * @return:     status code
  64 */
  65efi_status_t EFIAPI efi_main(efi_handle_t handle,
  66                             struct efi_system_table *systable)
  67{
  68        struct efi_simple_text_output_protocol *con_out = systable->con_out;
  69        efi_status_t ret;
  70        u16 text[] = EFI_ST_SUCCESS_STR;
  71
  72        con_out->output_string(con_out, L"EFI application calling Exit\n");
  73
  74        if (check_loaded_image_protocol(handle, systable) != EFI_SUCCESS) {
  75                con_out->output_string(con_out,
  76                                       L"Loaded image protocol missing\n");
  77                ret = EFI_NOT_FOUND;
  78                goto out;
  79        }
  80
  81        /* This return value is expected by the calling test */
  82        ret = EFI_UNSUPPORTED;
  83out:
  84        systable->boottime->exit(handle, ret, sizeof(text), text);
  85
  86        /*
  87         * This statement should not be reached.
  88         * To enable testing use a different return value.
  89         */
  90        return EFI_SUCCESS;
  91}
  92