205 lines
6.2 KiB
C
205 lines
6.2 KiB
C
|
|
/*
|
|
* Copyright (C) 2013 Jerry Hoemann <jerry.hoemann@hp.com>
|
|
*
|
|
*
|
|
* Application to allocate memory at EFI. Syntax of command
|
|
* mimics the EFI Boot Service "AllocatePages."
|
|
*
|
|
* See UEFI spec 2.3, Section 6.2.
|
|
*
|
|
*
|
|
|
|
|
|
|
|
|
|
FS1:\> memmap
|
|
Type Start End #pages Attributes
|
|
BS_Code 0000000000000000-0000000000000FFF 0000000000000001 000000000000000F
|
|
Available 0000000000001000-000000000008DFFF 000000000000008D 000000000000000F
|
|
Reserved 000000000008E000-000000000008FFFF 0000000000000002 000000000000000F
|
|
Available 0000000000090000-000000000009FFFF 0000000000000010 000000000000000F
|
|
Available 0000000000100000-000000000FFFFFFF 000000000000FF00 000000000000000F
|
|
BS_Code 0000000010000000-0000000010061FFF 0000000000000062 000000000000000F
|
|
Available 0000000010062000-000000005CDFFFFF 000000000004CD9E 000000000000000F
|
|
ACPI_NVS 000000005CE00000-000000005DDFFFFF 0000000000001000 000000000000000F
|
|
BS_Data 000000005DE00000-000000005DFFFFFF 0000000000000200 000000000000000F
|
|
Available 000000005E000000-000000005EF1CFFF 0000000000000F1D 000000000000000F
|
|
BS_Data 000000005EF1D000-00000000709FBFFF 0000000000011ADF 000000000000000F
|
|
Available 00000000709FC000-00000000710E3FFF 00000000000006E8 000000000000000F
|
|
LoaderCode 00000000710E4000-00000000711FEFFF 000000000000011B 000000000000000F
|
|
Available 00000000711FF000-0000000071901FFF 0000000000000703 000000000000000F
|
|
BS_Code 0000000071902000-00000000721FEFFF 00000000000008FD 000000000000000F
|
|
|
|
|
|
Example to allocat 5 pages type BootCode at address 20000000 (hex)
|
|
|
|
|
|
FS1:\> AllocPages.efi 2 3 5 20000000
|
|
AllocatePage: __AllocType__ __MemType__ __NumPages__ [__Addr__]
|
|
__AllocType__ {0,1,2} -- Any, MaxAddr, Addr
|
|
__MemType__ {0..13}, Reserved ==0, LCode==1, LData==2, BSCode==3, BSData==4, ...
|
|
__NumPages__ {0..F000000}
|
|
[__Addr__] 0... 3FFFFFFFFFFF
|
|
All numbers in hex no leading 0x
|
|
|
|
AllocatPage(2,3,5,20000000)
|
|
|
|
|
|
Example to allocat 5 pages type BootCode at address 30000000 (hex)
|
|
|
|
|
|
FS1:\> AllocPages.efi 2 3 5 30000000
|
|
AllocatePage: __AllocType__ __MemType__ __NumPages__ [__Addr__]
|
|
__AllocType__ {0,1,2} -- Any, MaxAddr, Addr
|
|
__MemType__ {0..13}, Reserved ==0, LCode==1, LData==2, BSCode==3, BSData==4, ...
|
|
__NumPages__ {0..F000000}
|
|
[__Addr__] 0... 3FFFFFFFFFFF
|
|
All numbers in hex no leading 0x
|
|
|
|
|
|
|
|
FS1:\> memmap
|
|
Type Start End #pages Attributes
|
|
BS_Code 0000000000000000-0000000000000FFF 0000000000000001 000000000000000F
|
|
Available 0000000000001000-000000000008DFFF 000000000000008D 000000000000000F
|
|
Reserved 000000000008E000-000000000008FFFF 0000000000000002 000000000000000F
|
|
Available 0000000000090000-000000000009FFFF 0000000000000010 000000000000000F
|
|
Available 0000000000100000-000000000FFFFFFF 000000000000FF00 000000000000000F
|
|
BS_Code 0000000010000000-0000000010061FFF 0000000000000062 000000000000000F
|
|
Available 0000000010062000-000000001FFFFFFF 000000000000FF9E 000000000000000F
|
|
BS_Code 0000000020000000-0000000020004FFF 0000000000000005 000000000000000F
|
|
Available 0000000020005000-000000002FFFFFFF 000000000000FFFB 000000000000000F
|
|
BS_Code 0000000030000000-0000000030004FFF 0000000000000005 000000000000000F
|
|
Available 0000000030005000-000000005CDFFFFF 000000000002CDFB 000000000000000F
|
|
ACPI_NVS 000000005CE00000-000000005DDFFFFF 0000000000001000 000000000000000F
|
|
BS_Data 000000005DE00000-000000005DFFFFFF 0000000000000200 000000000000000F
|
|
Available 000000005E000000-000000005EF1CFFF 0000000000000F1D 000000000000000F
|
|
BS_Data 000000005EF1D000-00000000709FBFFF 0000000000011ADF 000000000000000F
|
|
Available 00000000709FC000-00000000710E3FFF 00000000000006E8 000000000000000F
|
|
LoaderCode 00000000710E4000-00000000711FEFFF 000000000000011B 000000000000000F
|
|
Available 00000000711FF000-0000000071901FFF 0000000000000703 000000000000000F
|
|
BS_Code 0000000071902000-00000000721FEFFF 00000000000008FD 000000000000000F
|
|
|
|
|
|
|
|
|
|
|
|
*/
|
|
|
|
#include <efi.h>
|
|
#include <efilib.h>
|
|
#include <argify.h>
|
|
|
|
|
|
#define MAX_NUM_PAGES 0x000000000F000000
|
|
#define MAX_ADDR ((1ULL << 46) - 1)
|
|
|
|
|
|
#ifdef DEBUG
|
|
#undef DEBUG
|
|
#endif
|
|
#define DEBUG 0
|
|
|
|
|
|
|
|
EFI_STATUS
|
|
efi_main (EFI_HANDLE image, EFI_SYSTEM_TABLE *systab)
|
|
{
|
|
|
|
EFI_STATUS efi_status;
|
|
EFI_GUID LoadedImageProtocol = LOADED_IMAGE_PROTOCOL;
|
|
EFI_LOADED_IMAGE *info;
|
|
|
|
CHAR16 arglist[MAX_ARGS+1] = {0};
|
|
CHAR16 *argv[MAX_ARGS];
|
|
INTN argc = 0;
|
|
INTN err = 0;
|
|
|
|
INTN AllocType = -1;
|
|
INTN MemType = -1;
|
|
INTN NumPages = -1;
|
|
UINTN Addr = 0;
|
|
|
|
InitializeLib(image, systab);
|
|
|
|
efi_status = uefi_call_wrapper( BS->HandleProtocol, 3, image,
|
|
&LoadedImageProtocol, &info);
|
|
|
|
|
|
Print(L"AllocatePage: __AllocType__ __MemType__ __NumPages__ [__Addr__]\n");
|
|
Print(L"__AllocType__ {0,1,2} -- Any, MaxAddr, Addr\n");
|
|
Print(L"__MemType__ {0..13}, Reserved ==0, LCode==1, LData==2, BSCode==3, BSData==4, ...\n");
|
|
Print(L"__NumPages__ {0..%x}\n", MAX_NUM_PAGES);
|
|
Print(L"[__Addr__] 0... %llx\n", MAX_ADDR);
|
|
Print(L"All numbers in hex no leading 0x\n");
|
|
Print(L"\n");
|
|
|
|
#if DEBUG
|
|
Print(L"%s\n", info->LoadOptions);
|
|
#endif
|
|
|
|
|
|
#if DEBUG
|
|
Print(L"Set up arglist\n");
|
|
#endif
|
|
CopyMem(arglist, info->LoadOptions, info->LoadOptionsSize);
|
|
#if DEBUG
|
|
Print(L"arglist = <%s>\n", arglist);
|
|
#endif
|
|
|
|
#if DEBUG
|
|
Print(L"Now try argify\n");
|
|
#endif
|
|
argc = argify(arglist, info->LoadOptionsSize, argv);
|
|
#if DEBUG
|
|
Print(L"argc = %d\n", argc);
|
|
#endif
|
|
|
|
#if DEBUG
|
|
for (c = 0; c < argc; c++ ) {
|
|
Print(L"argv[%d] = <%s>\n", c, argv[c]);
|
|
}
|
|
#endif
|
|
if ( (argc < 3) || (argc > 5) ) {
|
|
Print(L"Wrong argument count\n");
|
|
return EFI_SUCCESS;
|
|
}
|
|
|
|
AllocType = xtoi(argv[1]);
|
|
MemType = xtoi(argv[2]);
|
|
NumPages = xtoi(argv[3]);
|
|
if ( argc == 5 ) Addr = xtoi(argv[4]);
|
|
|
|
if ( (AllocType < 0) || (AllocType > 2)) {
|
|
Print(L"Invalid AllocType\n");
|
|
err++;
|
|
}
|
|
if ( (MemType < 0) || (MemType > 13) ) {
|
|
Print(L"Invalid MemType\n");
|
|
err++;
|
|
}
|
|
if ( (NumPages < 0) || (NumPages > MAX_NUM_PAGES) ) {
|
|
Print(L"Inavlid NumPages\n");
|
|
err++;
|
|
}
|
|
if ( Addr > MAX_ADDR ) {
|
|
Print(L"Inavlid Address\n");
|
|
err++;
|
|
}
|
|
if ( err ) {
|
|
return EFI_INVALID_PARAMETER;
|
|
}
|
|
|
|
Print(L"AllocatPage(%d,%d,%d,%lx)\n", AllocType, MemType, NumPages, Addr);
|
|
|
|
efi_status = uefi_call_wrapper(BS->AllocatePages, 4, AllocType, MemType, NumPages, &Addr);
|
|
|
|
if ( EFI_ERROR(efi_status) ) {
|
|
Print(L"Allocate Pages Failed: %d\n", efi_status);
|
|
return efi_status;
|
|
}
|
|
|
|
return EFI_SUCCESS;
|
|
}
|