/* * Copyright (C) 2013 Jerry Hoemann * * * 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 #include #define MAX_NUM_PAGES 0x000000000F000000 #define MAX_ADDR ((1ULL << 46) - 1) #define MAX_ARGS 256 #define CHAR_SPACE L' ' #define DEBUG 0 INTN argify(CHAR16 *buf, UINTN len, CHAR16 **argv) { UINTN i=0, j=0; CHAR16 *p = buf; if (buf == 0) { argv[0] = NULL; return 0; } /* len represents the number of bytes, not the number of 16 bytes chars */ len = len >> 1; /* * Here we use CHAR_NULL as the terminator rather than the length * because it seems like the EFI shell return rather bogus values for it. * Apparently, we are guaranteed to find the '\0' character in the buffer * where the real input arguments stop, so we use it instead. */ for(;;) { while (buf[i] == CHAR_SPACE && buf[i] != CHAR_NULL && i < len) i++; if (buf[i] == CHAR_NULL || i == len) goto end; p = buf+i; i++; while (buf[i] != CHAR_SPACE && buf[i] != CHAR_NULL && i < len) i++; argv[j++] = p; if (buf[i] == CHAR_NULL) goto end; buf[i] = CHAR_NULL; if (i == len) goto end; i++; if (j == MAX_ARGS-1) { Print(L"too many arguments (%d) truncating\n", j); goto end; } } end: argv[j] = NULL; return j; } 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; }