1 /*- 2 * Copyright (c) 2013 The FreeBSD Foundation 3 * All rights reserved. 4 * 5 * This software was developed by Benno Rice under sponsorship from 6 * the FreeBSD Foundation. 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 AUTHOR 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 AUTHOR 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 <sys/cdefs.h> 30 __FBSDID("$FreeBSD: head/sys/boot/efi/loader/copy.c 293724 2016-01-12 02:17:39Z smh $"); 31 32 #include <sys/param.h> 33 34 #include <stand.h> 35 #include <bootstrap.h> 36 37 #include <efi.h> 38 #include <efilib.h> 39 40 #include "loader_efi.h" 41 42 #ifndef EFI_STAGING_SIZE 43 #define EFI_STAGING_SIZE 96 44 #endif 45 46 #define STAGE_PAGES EFI_SIZE_TO_PAGES((EFI_STAGING_SIZE) * 1024 * 1024) 47 48 EFI_PHYSICAL_ADDRESS staging, staging_end; 49 int stage_offset_set = 0; 50 ssize_t stage_offset; 51 52 int 53 efi_copy_init(void) 54 { 55 EFI_STATUS status; 56 size_t pages = STAGE_PAGES; 57 58 status = BS->AllocatePages(AllocateAnyPages, EfiLoaderData, 59 pages, &staging); 60 if (EFI_ERROR(status)) { 61 printf("failed to allocate %uMB staging area: %llu\n", 62 EFI_STAGING_SIZE, status); 63 printf("retrying with smaller %uMB allocation\n", 64 EFI_STAGING_SIZE/2); 65 pages /= 2; 66 status = BS->AllocatePages(AllocateAnyPages, EfiLoaderData, 67 pages, &staging); 68 } 69 if (EFI_ERROR(status)) { 70 printf("failed to allocate staging area: %llu\n", 71 status); 72 return (status); 73 } 74 staging_end = staging + pages * EFI_PAGE_SIZE; 75 76 #if defined(__aarch64__) 77 /* 78 * Round the kernel load address to a 2MiB value. This is needed 79 * because the kernel builds a page table based on where it has 80 * been loaded in physical address space. As the kernel will use 81 * either a 1MiB or 2MiB page for this we need to make sure it 82 * is correctly aligned for both cases. 83 */ 84 staging = roundup2(staging, 2 * 1024 * 1024); 85 #endif 86 87 return (0); 88 } 89 90 void * 91 efi_translate(vm_offset_t ptr) 92 { 93 94 return ((void *)(ptr + stage_offset)); 95 } 96 97 ssize_t 98 efi_copyin(const void *src, vm_offset_t dest, const size_t len) 99 { 100 101 if (!stage_offset_set) { 102 stage_offset = (vm_offset_t)staging - dest; 103 stage_offset_set = 1; 104 } 105 106 /* XXX: Callers do not check for failure. */ 107 if (dest + stage_offset + len > staging_end) { 108 errno = ENOMEM; 109 return (-1); 110 } 111 bcopy(src, (void *)(dest + stage_offset), len); 112 return (len); 113 } 114 115 ssize_t 116 efi_copyout(const vm_offset_t src, void *dest, const size_t len) 117 { 118 119 /* XXX: Callers do not check for failure. */ 120 if (src + stage_offset + len > staging_end) { 121 errno = ENOMEM; 122 return (-1); 123 } 124 bcopy((void *)(src + stage_offset), dest, len); 125 return (len); 126 } 127 128 129 ssize_t 130 efi_readin(const int fd, vm_offset_t dest, const size_t len) 131 { 132 133 if (dest + stage_offset + len > staging_end) { 134 errno = ENOMEM; 135 return (-1); 136 } 137 return (read(fd, (void *)(dest + stage_offset), len)); 138 } 139 140 void 141 efi_copy_finish(void) 142 { 143 uint64_t *src, *dst, *last; 144 145 src = (uint64_t *)staging; 146 dst = (uint64_t *)(staging - stage_offset); 147 last = (uint64_t *)staging_end; 148 149 while (src < last) 150 *dst++ = *src++; 151 } 152