1 | |
---|
2 | /* |
---|
3 | * Copyright (C) 2013 Jerry Hoemann <jerry.hoemann@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 | } |
---|