Home | History | Annotate | Line # | Download | only in efiboot
efiboot.c revision 1.7 
      1 /*	$NetBSD: efiboot.c,v 1.7 2018/04/11 10:32:09 nonaka Exp $	*/
      2 
      3 /*-
      4  * Copyright (c) 2016 Kimihiro Nonaka <nonaka (at) netbsd.org>
      5  * All rights reserved.
      6  *
      7  * Redistribution and use in source and binary forms, with or without
      8  * modification, are permitted provided that the following conditions
      9  * are met:
     10  * 1. Redistributions of source code must retain the above copyright
     11  *    notice, this list of conditions and the following disclaimer.
     12  * 2. Redistributions in binary form must reproduce the above copyright
     13  *    notice, this list of conditions and the following disclaimer in the
     14  *    documentation and/or other materials provided with the distribution.
     15  *
     16  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     26  * SUCH DAMAGE.
     27  */
     28 
     29 #include "efiboot.h"
     30 
     31 #include "biosdisk_ll.h"
     32 #include "bootinfo.h"
     33 #include "devopen.h"
     34 
     35 EFI_HANDLE IH;
     36 EFI_DEVICE_PATH *efi_bootdp;
     37 enum efi_boot_device_type efi_bootdp_type = BOOT_DEVICE_TYPE_HD;
     38 EFI_LOADED_IMAGE *efi_li;
     39 uintptr_t efi_main_sp;
     40 physaddr_t efi_loadaddr, efi_kernel_start;
     41 u_long efi_kernel_size;
     42 bool efi_cleanuped;
     43 
     44 static EFI_PHYSICAL_ADDRESS heap_start = EFI_ALLOCATE_MAX_ADDRESS;
     45 static UINTN heap_size = 1 * 1024 * 1024;			/* 1MB */
     46 static struct btinfo_efi btinfo_efi;
     47 
     48 static void efi_heap_init(void);
     49 
     50 EFI_STATUS EFIAPI efi_main(EFI_HANDLE, EFI_SYSTEM_TABLE *);
     51 EFI_STATUS EFIAPI
     52 efi_main(EFI_HANDLE imageHandle, EFI_SYSTEM_TABLE *systemTable)
     53 {
     54 	EFI_STATUS status;
     55 	EFI_DEVICE_PATH *dp0, *dp;
     56 	extern char twiddle_toggle;
     57 
     58 	IH = imageHandle;
     59 	InitializeLib(IH, systemTable);
     60 
     61 	efi_main_sp = (uintptr_t)&status;
     62 	twiddle_toggle = 1;	/* no twiddling until we're ready */
     63 
     64 	cninit();
     65 	efi_heap_init();
     66 	efi_md_init();
     67 
     68 	status = uefi_call_wrapper(BS->HandleProtocol, 3, IH,
     69 	    &LoadedImageProtocol, (void **)&efi_li);
     70 	if (EFI_ERROR(status))
     71 		panic("HandleProtocol(LoadedImageProtocol): %" PRIxMAX,
     72 		    (uintmax_t)status);
     73 	status = uefi_call_wrapper(BS->HandleProtocol, 3, efi_li->DeviceHandle,
     74 	    &DevicePathProtocol, (void **)&dp0);
     75 	if (EFI_ERROR(status))
     76 		panic("HandleProtocol(DevicePathProtocol): %" PRIxMAX,
     77 		    (uintmax_t)status);
     78 	efi_bootdp = dp0;
     79 	for (dp = dp0; !IsDevicePathEnd(dp); dp = NextDevicePathNode(dp)) {
     80 		if (DevicePathType(dp) == MEDIA_DEVICE_PATH &&
     81 		    DevicePathSubType(dp) == MEDIA_CDROM_DP) {
     82 			efi_bootdp_type = BOOT_DEVICE_TYPE_CD;
     83 			break;
     84 		}
     85 		if (DevicePathType(dp) == MESSAGING_DEVICE_PATH &&
     86 		    DevicePathSubType(dp) == MSG_MAC_ADDR_DP) {
     87 			efi_bootdp_type = BOOT_DEVICE_TYPE_NET;
     88 			break;
     89 		}
     90 	}
     91 
     92 	efi_disk_probe();
     93 	efi_pxe_probe();
     94 	efi_net_probe();
     95 
     96 	status = uefi_call_wrapper(BS->SetWatchdogTimer, 4, 0, 0, 0, NULL);
     97 	if (EFI_ERROR(status))
     98 		panic("SetWatchdogTimer: %" PRIxMAX, (uintmax_t)status);
     99 
    100 	boot();
    101 
    102 	return EFI_SUCCESS;
    103 }
    104 
    105 void
    106 efi_cleanup(void)
    107 {
    108 	EFI_STATUS status;
    109 	EFI_MEMORY_DESCRIPTOR *desc;
    110 	UINTN NoEntries, MapKey, DescriptorSize;
    111 	UINT32 DescriptorVersion;
    112 	struct btinfo_efimemmap *bim;
    113 	size_t allocsz;
    114 
    115 	clearit();
    116 
    117 	memset(&btinfo_efi, 0, sizeof(btinfo_efi));
    118 	btinfo_efi.systblpa = (intptr_t)ST;
    119 #ifdef	__i386__	/* bootia32.efi */
    120 	btinfo_efi.flags |= BI_EFI_32BIT;
    121 #endif
    122 	BI_ADD(&btinfo_efi, BTINFO_EFI, sizeof(btinfo_efi));
    123 
    124 	NoEntries = 0;
    125 	desc = efi_memory_get_map(&NoEntries, &MapKey, &DescriptorSize,
    126 	    &DescriptorVersion, true);
    127 	status = uefi_call_wrapper(BS->ExitBootServices, 2, IH, MapKey);
    128 	if (EFI_ERROR(status)) {
    129 		FreePool(desc);
    130 		desc = efi_memory_get_map(&NoEntries, &MapKey, &DescriptorSize,
    131 		    &DescriptorVersion, true);
    132 		status = uefi_call_wrapper(BS->ExitBootServices, 2, IH, MapKey);
    133 		if (EFI_ERROR(status))
    134 			panic("ExitBootServices failed");
    135 	}
    136 	efi_cleanuped = true;
    137 
    138 	allocsz = sizeof(struct btinfo_efimemmap) - 1
    139 	    + NoEntries * DescriptorSize;
    140 	bim = alloc(allocsz);
    141 	bim->num = NoEntries;
    142 	bim->version = DescriptorVersion;
    143 	bim->size = DescriptorSize;
    144 	memcpy(bim->memmap, desc, NoEntries * DescriptorSize);
    145 	BI_ADD(bim, BTINFO_EFIMEMMAP, allocsz);
    146 }
    147 
    148 static void
    149 efi_heap_init(void)
    150 {
    151 	EFI_STATUS status;
    152 	u_int sz = EFI_SIZE_TO_PAGES(heap_size);
    153 
    154 	status = uefi_call_wrapper(BS->AllocatePages, 4, AllocateMaxAddress,
    155 	    EfiLoaderData, sz, &heap_start);
    156 	if (EFI_ERROR(status))
    157 		Panic(L"%a: AllocatePages() failed: %d page(s): %r",
    158 		    __func__, sz, status);
    159 	setheap((void *)(UINTN)heap_start,
    160 	    (void *)(UINTN)(heap_start + heap_size));
    161 }
    162