src/sd-boot/stub.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the GNU Lesser General Public License as published by
   5  * the Free Software Foundation; either version 2.1 of the License, or
   6  * (at your option) any later version.
   7  *
   8  * This program is distributed in the hope that it will be useful, but
   9  * WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  11  * Lesser General Public License for more details.
  12  *
  13  * Copyright (C) 2015 Kay Sievers <kay@vrfy.org>
  14  */
  15
  16 #include <efi.h>
  17 #include <efilib.h>
  18
  19 #include "util.h"
  20 #include "pefile.h"
  21 #include "linux.h"
  22
  23 /* magic string to find in the binary image */
  24 static const char __attribute__((used)) magic[] = "#### LoaderInfo: stub " VERSION " ####";
  25
  26 static const EFI_GUID global_guid = EFI_GLOBAL_VARIABLE;
  27
  28 EFI_STATUS efi_main(EFI_HANDLE image, EFI_SYSTEM_TABLE *sys_table) {
  29         EFI_LOADED_IMAGE *loaded_image;
  30         EFI_FILE *root_dir;
  31         CHAR16 *loaded_image_path;
  32         CHAR8 *b;
  33         UINTN size;
  34         BOOLEAN secure = FALSE;
  35         CHAR8 *sections[] = {
  36                 (UINT8 *)".cmdline",
  37                 (UINT8 *)".linux",
  38                 (UINT8 *)".initrd",
  39                 NULL
  40         };
  41         UINTN addrs[ELEMENTSOF(sections)-1] = {};
  42         UINTN offs[ELEMENTSOF(sections)-1] = {};
  43         UINTN szs[ELEMENTSOF(sections)-1] = {};
  44         CHAR8 *cmdline = NULL;
  45         UINTN cmdline_len;
  46         EFI_STATUS err;
  47
  48         InitializeLib(image, sys_table);
  49
  50         err = uefi_call_wrapper(BS->OpenProtocol, 6, image, &LoadedImageProtocol, (VOID **)&loaded_image,
  51                                 image, NULL, EFI_OPEN_PROTOCOL_GET_PROTOCOL);
  52         if (EFI_ERROR(err)) {
  53                 Print(L"Error getting a LoadedImageProtocol handle: %r ", err);
  54                 uefi_call_wrapper(BS->Stall, 1, 3 * 1000 * 1000);
  55                 return err;
  56         }
  57
  58         root_dir = LibOpenRoot(loaded_image->DeviceHandle);
  59         if (!root_dir) {
  60                 Print(L"Unable to open root directory: %r ", err);
  61                 uefi_call_wrapper(BS->Stall, 1, 3 * 1000 * 1000);
  62                 return EFI_LOAD_ERROR;
  63         }
  64
  65         loaded_image_path = DevicePathToStr(loaded_image->FilePath);
  66
  67         if (efivar_get_raw(&global_guid, L"SecureBoot", &b, &size) == EFI_SUCCESS) {
  68                 if (*b > 0)
  69                         secure = TRUE;
  70                 FreePool(b);
  71         }
  72
  73         err = pefile_locate_sections(root_dir, loaded_image_path, sections, addrs, offs, szs);
  74         if (EFI_ERROR(err)) {
  75                 Print(L"Unable to locate embedded .linux section: %r ", err);
  76                 uefi_call_wrapper(BS->Stall, 1, 3 * 1000 * 1000);
  77                 return err;
  78         }
  79
  80         if (szs[0] > 0)
  81                 cmdline = (CHAR8 *)(loaded_image->ImageBase + addrs[0]);
  82
  83         cmdline_len = szs[0];
  84
  85         /* if we are not in secure boot mode, accept a custom command line and replace the built-in one */
  86         if (!secure && loaded_image->LoadOptionsSize > 0) {
  87                 CHAR16 *options;
  88                 CHAR8 *line;
  89                 UINTN i;
  90
  91                 options = (CHAR16 *)loaded_image->LoadOptions;
  92                 cmdline_len = (loaded_image->LoadOptionsSize / sizeof(CHAR16)) * sizeof(CHAR8);
  93                 line = AllocatePool(cmdline_len);
  94                 for (i = 0; i < cmdline_len; i++)
  95                         line[i] = options[i];
  96                 cmdline = line;
  97         }
  98
  99         err = linux_exec(image, cmdline, cmdline_len,
 100                          (UINTN)loaded_image->ImageBase + addrs[1],
 101                          (UINTN)loaded_image->ImageBase + addrs[2], szs[2]);
 102
 103         Print(L"Execution of embedded linux image failed: %r\n", err);
 104         uefi_call_wrapper(BS->Stall, 1, 3 * 1000 * 1000);
 105         return err;
 106 }