You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
1.1 KiB
38 lines
1.1 KiB
#include "bundler.h"
|
|
|
|
int __cdecl main(int argc, char** argv)
|
|
{
|
|
if (argc < 2)
|
|
{
|
|
std::printf("[!] invalid amount of parameters\n");
|
|
return -1;
|
|
}
|
|
|
|
std::vector<std::uint8_t> efi_module;
|
|
std::vector<std::uint8_t> bootmgfw;
|
|
|
|
impl::open_binary_file(argv[1], bootmgfw);
|
|
impl::open_binary_file(argv[2], efi_module);
|
|
|
|
if (efi_module.empty() || bootmgfw.empty())
|
|
{
|
|
std::printf("[!] unable to load efi module(s)...\n");
|
|
return -1;
|
|
}
|
|
|
|
efi_module.resize(NT_HEADER(efi_module.data())->OptionalHeader.SizeOfImage);
|
|
std::printf("bundling efi module, size -> 0x%x\n",
|
|
NT_HEADER(efi_module.data())->OptionalHeader.SizeOfImage);
|
|
|
|
bootmgfw.resize(NT_HEADER(bootmgfw.data())->OptionalHeader.SizeOfImage);
|
|
std::printf("bundling module into bootmgfw, size before patch -> 0x%x\n",
|
|
NT_HEADER(bootmgfw.data())->OptionalHeader.SizeOfImage);
|
|
|
|
bundler::bundle(bootmgfw, efi_module);
|
|
std::ofstream new_file("result.efi", std::ios::binary);
|
|
new_file.write((char*)bootmgfw.data(), bootmgfw.size());
|
|
new_file.close();
|
|
|
|
std::printf("bundled modules together....\n");
|
|
std::getchar();
|
|
} |