/* $NetBSD: efiboot.c,v 1.10.2.3 2023/05/13 11:45:53 martin Exp $ */ /*- * Copyright (c) 2016 Kimihiro Nonaka * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF * SUCH DAMAGE. */ #include "efiboot.h" #include "bootinfo.h" #include "devopen.h" EFI_HANDLE IH; EFI_DEVICE_PATH *efi_bootdp; enum efi_boot_device_type efi_bootdp_type = BOOT_DEVICE_TYPE_HD; EFI_LOADED_IMAGE *efi_li; uintptr_t efi_main_sp; physaddr_t efi_loadaddr, efi_kernel_start, efi_load_start; physaddr_t efi_kernel_reloc = 0; enum efi_reloc_type efi_reloc_type = RELOC_DEFAULT; u_long efi_kernel_size; bool efi_cleanuped; struct btinfo_efimemmap *btinfo_efimemmap = NULL; static EFI_PHYSICAL_ADDRESS heap_start = EFI_ALLOCATE_MAX_ADDRESS; static UINTN heap_size = 1 * 1024 * 1024; /* 1MB */ static struct btinfo_efi btinfo_efi; static void efi_heap_init(void); EFI_STATUS EFIAPI efi_main(EFI_HANDLE, EFI_SYSTEM_TABLE *); EFI_STATUS EFIAPI efi_main(EFI_HANDLE imageHandle, EFI_SYSTEM_TABLE *systemTable) { EFI_STATUS status; EFI_DEVICE_PATH *dp0, *dp; extern char twiddle_toggle; IH = imageHandle; InitializeLib(IH, systemTable); efi_main_sp = (uintptr_t)&status; twiddle_toggle = 1; /* no twiddling until we're ready */ cninit(); efi_heap_init(); efi_md_init(); status = uefi_call_wrapper(BS->HandleProtocol, 3, IH, &LoadedImageProtocol, (void **)&efi_li); if (EFI_ERROR(status)) panic("HandleProtocol(LoadedImageProtocol): %" PRIxMAX, (uintmax_t)status); status = uefi_call_wrapper(BS->HandleProtocol, 3, efi_li->DeviceHandle, &DevicePathProtocol, (void **)&dp0); if (EFI_ERROR(status)) panic("HandleProtocol(DevicePathProtocol): %" PRIxMAX, (uintmax_t)status); efi_bootdp = dp0; for (dp = dp0; !IsDevicePathEnd(dp); dp = NextDevicePathNode(dp)) { if (DevicePathType(dp) == MEDIA_DEVICE_PATH && DevicePathSubType(dp) == MEDIA_CDROM_DP) { efi_bootdp_type = BOOT_DEVICE_TYPE_CD; break; } if (DevicePathType(dp) == MESSAGING_DEVICE_PATH && DevicePathSubType(dp) == MSG_MAC_ADDR_DP) { efi_bootdp_type = BOOT_DEVICE_TYPE_NET; break; } } efi_memory_probe(); efi_disk_probe(); efi_pxe_probe(); efi_net_probe(); status = uefi_call_wrapper(BS->SetWatchdogTimer, 4, 0, 0, 0, NULL); if (EFI_ERROR(status)) panic("SetWatchdogTimer: %" PRIxMAX, (uintmax_t)status); boot(); return EFI_SUCCESS; } void efi_cleanup(void) { EFI_STATUS status; EFI_MEMORY_DESCRIPTOR *desc; UINTN NoEntries, MapKey, DescriptorSize; UINT32 DescriptorVersion; size_t allocsz; memset(&btinfo_efi, 0, sizeof(btinfo_efi)); btinfo_efi.systblpa = (intptr_t)ST; #ifdef __i386__ /* bootia32.efi */ btinfo_efi.flags |= BI_EFI_32BIT; #endif BI_ADD(&btinfo_efi, BTINFO_EFI, sizeof(btinfo_efi)); NoEntries = 0; desc = efi_memory_get_map(&NoEntries, &MapKey, &DescriptorSize, &DescriptorVersion, true); status = uefi_call_wrapper(BS->ExitBootServices, 2, IH, MapKey); if (EFI_ERROR(status)) { FreePool(desc); desc = efi_memory_get_map(&NoEntries, &MapKey, &DescriptorSize, &DescriptorVersion, true); status = uefi_call_wrapper(BS->ExitBootServices, 2, IH, MapKey); if (EFI_ERROR(status)) panic("ExitBootServices failed"); } efi_cleanuped = true; efi_memory_compact_map(desc, &NoEntries, DescriptorSize); allocsz = sizeof(struct btinfo_efimemmap) - 1 + NoEntries * DescriptorSize; btinfo_efimemmap = alloc(allocsz); btinfo_efimemmap->num = NoEntries; btinfo_efimemmap->version = DescriptorVersion; btinfo_efimemmap->size = DescriptorSize; memcpy(btinfo_efimemmap->memmap, desc, NoEntries * DescriptorSize); BI_ADD(btinfo_efimemmap, BTINFO_EFIMEMMAP, allocsz); } static void efi_heap_init(void) { EFI_STATUS status; u_int sz = EFI_SIZE_TO_PAGES(heap_size); status = uefi_call_wrapper(BS->AllocatePages, 4, AllocateMaxAddress, EfiLoaderData, sz, &heap_start); if (EFI_ERROR(status)) Panic(L"%a: AllocatePages() failed: %d page(s): %r", __func__, sz, status); setheap((void *)(UINTN)heap_start, (void *)(UINTN)(heap_start + heap_size)); }