1 2 /* 3 * Copyright (C) 2013 Jerry Hoemann <jerry.hoemann (at) hp.com> 4 * 5 * 6 * Application to allocate memory at EFI. Syntax of command 7 * mimics the EFI Boot Service "AllocatePages." 8 * 9 * See UEFI spec 2.3, Section 6.2. 10 * 11 * 12 13 14 15 16 FS1:\> memmap 17 Type Start End #pages Attributes 18 BS_Code 0000000000000000-0000000000000FFF 0000000000000001 000000000000000F 19 Available 0000000000001000-000000000008DFFF 000000000000008D 000000000000000F 20 Reserved 000000000008E000-000000000008FFFF 0000000000000002 000000000000000F 21 Available 0000000000090000-000000000009FFFF 0000000000000010 000000000000000F 22 Available 0000000000100000-000000000FFFFFFF 000000000000FF00 000000000000000F 23 BS_Code 0000000010000000-0000000010061FFF 0000000000000062 000000000000000F 24 Available 0000000010062000-000000005CDFFFFF 000000000004CD9E 000000000000000F 25 ACPI_NVS 000000005CE00000-000000005DDFFFFF 0000000000001000 000000000000000F 26 BS_Data 000000005DE00000-000000005DFFFFFF 0000000000000200 000000000000000F 27 Available 000000005E000000-000000005EF1CFFF 0000000000000F1D 000000000000000F 28 BS_Data 000000005EF1D000-00000000709FBFFF 0000000000011ADF 000000000000000F 29 Available 00000000709FC000-00000000710E3FFF 00000000000006E8 000000000000000F 30 LoaderCode 00000000710E4000-00000000711FEFFF 000000000000011B 000000000000000F 31 Available 00000000711FF000-0000000071901FFF 0000000000000703 000000000000000F 32 BS_Code 0000000071902000-00000000721FEFFF 00000000000008FD 000000000000000F 33 34 35 Example to allocat 5 pages type BootCode at address 20000000 (hex) 36 37 38 FS1:\> AllocPages.efi 2 3 5 20000000 39 AllocatePage: __AllocType__ __MemType__ __NumPages__ [__Addr__] 40 __AllocType__ {0,1,2} -- Any, MaxAddr, Addr 41 __MemType__ {0..13}, Reserved ==0, LCode==1, LData==2, BSCode==3, BSData==4, ... 42 __NumPages__ {0..F000000} 43 [__Addr__] 0... 3FFFFFFFFFFF 44 All numbers in hex no leading 0x 45 46 AllocatPage(2,3,5,20000000) 47 48 49 Example to allocat 5 pages type BootCode at address 30000000 (hex) 50 51 52 FS1:\> AllocPages.efi 2 3 5 30000000 53 AllocatePage: __AllocType__ __MemType__ __NumPages__ [__Addr__] 54 __AllocType__ {0,1,2} -- Any, MaxAddr, Addr 55 __MemType__ {0..13}, Reserved ==0, LCode==1, LData==2, BSCode==3, BSData==4, ... 56 __NumPages__ {0..F000000} 57 [__Addr__] 0... 3FFFFFFFFFFF 58 All numbers in hex no leading 0x 59 60 61 62 FS1:\> memmap 63 Type Start End #pages Attributes 64 BS_Code 0000000000000000-0000000000000FFF 0000000000000001 000000000000000F 65 Available 0000000000001000-000000000008DFFF 000000000000008D 000000000000000F 66 Reserved 000000000008E000-000000000008FFFF 0000000000000002 000000000000000F 67 Available 0000000000090000-000000000009FFFF 0000000000000010 000000000000000F 68 Available 0000000000100000-000000000FFFFFFF 000000000000FF00 000000000000000F 69 BS_Code 0000000010000000-0000000010061FFF 0000000000000062 000000000000000F 70 Available 0000000010062000-000000001FFFFFFF 000000000000FF9E 000000000000000F 71 BS_Code 0000000020000000-0000000020004FFF 0000000000000005 000000000000000F 72 Available 0000000020005000-000000002FFFFFFF 000000000000FFFB 000000000000000F 73 BS_Code 0000000030000000-0000000030004FFF 0000000000000005 000000000000000F 74 Available 0000000030005000-000000005CDFFFFF 000000000002CDFB 000000000000000F 75 ACPI_NVS 000000005CE00000-000000005DDFFFFF 0000000000001000 000000000000000F 76 BS_Data 000000005DE00000-000000005DFFFFFF 0000000000000200 000000000000000F 77 Available 000000005E000000-000000005EF1CFFF 0000000000000F1D 000000000000000F 78 BS_Data 000000005EF1D000-00000000709FBFFF 0000000000011ADF 000000000000000F 79 Available 00000000709FC000-00000000710E3FFF 00000000000006E8 000000000000000F 80 LoaderCode 00000000710E4000-00000000711FEFFF 000000000000011B 000000000000000F 81 Available 00000000711FF000-0000000071901FFF 0000000000000703 000000000000000F 82 BS_Code 0000000071902000-00000000721FEFFF 00000000000008FD 000000000000000F 83 84 85 86 87 88 */ 89 90 #include <efi.h> 91 #include <efilib.h> 92 #include <argify.h> 93 94 95 #define MAX_NUM_PAGES 0x000000000F000000 96 #define MAX_ADDR ((1ULL << 46) - 1) 97 98 99 #ifdef DEBUG 100 #undef DEBUG 101 #endif 102 #define DEBUG 0 103 104 105 106 EFI_STATUS 107 efi_main (EFI_HANDLE image, EFI_SYSTEM_TABLE *systab) 108 { 109 110 EFI_STATUS efi_status; 111 EFI_GUID LoadedImageProtocol = LOADED_IMAGE_PROTOCOL; 112 EFI_LOADED_IMAGE *info; 113 114 CHAR16 arglist[MAX_ARGS+1] = {0}; 115 CHAR16 *argv[MAX_ARGS]; 116 INTN argc = 0; 117 INTN err = 0; 118 119 INTN AllocType = -1; 120 INTN MemType = -1; 121 INTN NumPages = -1; 122 UINTN Addr = 0; 123 124 InitializeLib(image, systab); 125 126 efi_status = uefi_call_wrapper( BS->HandleProtocol, 3, image, 127 &LoadedImageProtocol, &info); 128 129 130 Print(L"AllocatePage: __AllocType__ __MemType__ __NumPages__ [__Addr__]\n"); 131 Print(L"__AllocType__ {0,1,2} -- Any, MaxAddr, Addr\n"); 132 Print(L"__MemType__ {0..13}, Reserved ==0, LCode==1, LData==2, BSCode==3, BSData==4, ...\n"); 133 Print(L"__NumPages__ {0..%x}\n", MAX_NUM_PAGES); 134 Print(L"[__Addr__] 0... %llx\n", MAX_ADDR); 135 Print(L"All numbers in hex no leading 0x\n"); 136 Print(L"\n"); 137 138 #if DEBUG 139 Print(L"%s\n", info->LoadOptions); 140 #endif 141 142 143 #if DEBUG 144 Print(L"Set up arglist\n"); 145 #endif 146 CopyMem(arglist, info->LoadOptions, info->LoadOptionsSize); 147 #if DEBUG 148 Print(L"arglist = <%s>\n", arglist); 149 #endif 150 151 #if DEBUG 152 Print(L"Now try argify\n"); 153 #endif 154 argc = argify(arglist, info->LoadOptionsSize, argv); 155 #if DEBUG 156 Print(L"argc = %d\n", argc); 157 #endif 158 159 #if DEBUG 160 for (c = 0; c < argc; c++ ) { 161 Print(L"argv[%d] = <%s>\n", c, argv[c]); 162 } 163 #endif 164 if ( (argc < 3) || (argc > 5) ) { 165 Print(L"Wrong argument count\n"); 166 return EFI_SUCCESS; 167 } 168 169 AllocType = xtoi(argv[1]); 170 MemType = xtoi(argv[2]); 171 NumPages = xtoi(argv[3]); 172 if ( argc == 5 ) Addr = xtoi(argv[4]); 173 174 if ( (AllocType < 0) || (AllocType > 2)) { 175 Print(L"Invalid AllocType\n"); 176 err++; 177 } 178 if ( (MemType < 0) || (MemType > 13) ) { 179 Print(L"Invalid MemType\n"); 180 err++; 181 } 182 if ( (NumPages < 0) || (NumPages > MAX_NUM_PAGES) ) { 183 Print(L"Inavlid NumPages\n"); 184 err++; 185 } 186 if ( Addr > MAX_ADDR ) { 187 Print(L"Inavlid Address\n"); 188 err++; 189 } 190 if ( err ) { 191 return EFI_INVALID_PARAMETER; 192 } 193 194 Print(L"AllocatPage(%d,%d,%d,%lx)\n", AllocType, MemType, NumPages, Addr); 195 196 efi_status = uefi_call_wrapper(BS->AllocatePages, 4, AllocType, MemType, NumPages, &Addr); 197 198 if ( EFI_ERROR(efi_status) ) { 199 Print(L"Allocate Pages Failed: %d\n", efi_status); 200 return efi_status; 201 } 202 203 return EFI_SUCCESS; 204 } 205