source: bootcd/isolinux/syslinux-6.03/gnu-efi/gnu-efi-3.0/apps/AllocPages.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;
}