mirrors/UEFITool
1
0
Fork 0
mirror of https://github.com/LongSoft/UEFITool.git synced 2025-05-27 21:44:24 -04:00
Code Issues Projects Releases Packages Wiki Activity

Implement replace all and add -o (--output) path to UEFIReplace

Browse source
This commit is contained in:
vit9696 2018-05-19 18:35:38 +03:00
parent 63e40c058a
commit 8db52ea51a
4 changed files with 37 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

23
UEFIReplace/uefireplace_main.cpp
View file

@ -29,7 +29,7 @@ int main(int argc, char *argv[])
if (args.length() < 5) {
std::cout << "UEFIReplace 0.1.3 - UEFI image file replacement utility" << std::endl << std::endl <<
"Usage: UEFIReplace image_file guid section_type contents_file" << std::endl;
"Usage: UEFIReplace image_file guid section_type contents_file [-o output] [-all]" << std::endl;
return ERR_SUCCESS;
}
@ -37,10 +37,25 @@ int main(int argc, char *argv[])
QByteArray guid = QByteArray::fromRawData((const char*)&uuid.data1, sizeof(EFI_GUID));
bool converted;
UINT8 sectionType = (UINT8)args.at(3).toUShort(&converted, 16);
if (!converted)
if (!converted) {
result = ERR_INVALID_PARAMETER;
else
result = r.replace(args.at(1), guid, sectionType, args.at(4));
} else {
QString output = args.at(1) + ".patched";
bool replaceOnce = true;
for (int i = 5, sz = args.size(); i < sz; i++) {
if ((args.at(i) == "-o" || args.at(i) == "--output") && i + 1 < sz) {
output = args.at(i+1);
i++;
} else if (args.at(i) == "-all") {
replaceOnce = false;
} else {
result = ERR_INVALID_PARAMETER;
}
}
if (result == ERR_SUCCESS) {
result = r.replace(args.at(1), guid, sectionType, args.at(4), output, replaceOnce);
}
}
switch (result) {
case ERR_SUCCESS: