mirror of
https://github.com/LongSoft/UEFITool.git
synced 2024-11-23 08:28:23 +08:00
79 lines
2.6 KiB
C++
79 lines
2.6 KiB
C++
|
/* uefireplace_main.cpp
|
||
|
|
|
||
|
|
Copyright (c) 2017, mxxxc. All rights reserved.
|
||
|
|
This program and the accompanying materials
|
||
|
|
are licensed and made available under the terms and conditions of the BSD License
|
||
|
|
which accompanies this distribution. The full text of the license may be found at
|
||
|
|
http://opensource.org/licenses/bsd-license.php
|
||
|
|
|
||
|
|
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
|
||
|
|
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
|
||
|
|
|
||
|
|
*/
|
||
|
|
#include <QCoreApplication>
|
||
|
|
#include <QString>
|
||
|
|
#include <QStringList>
|
||
|
|
#include <iostream>
|
||
|
|
#include "uefireplace.h"
|
||
|
|
|
||
|
|
int main(int argc, char *argv[])
|
||
|
|
{
|
||
|
|
QCoreApplication a(argc, argv);
|
||
|
|
a.setOrganizationName("CodeRush");
|
||
|
|
a.setOrganizationDomain("coderush.me");
|
||
|
|
a.setApplicationName("UEFIReplace");
|
||
|
|
|
||
|
|
UEFIReplace r;
|
||
|
|
UINT8 result = ERR_SUCCESS;
|
||
|
|
QStringList args = a.arguments();
|
||
|
|
|
||
|
|
if (args.length() < 5) {
|
||
|
|
std::cout << "UEFIReplace 0.3.9 - UEFI image file replacement utility" << std::endl << std::endl <<
|
||
|
|
"Usage: UEFIReplace image_file guid section_type contents_file" << std::endl;
|
||
|
|
return ERR_SUCCESS;
|
||
|
|
}
|
||
|
|
|
||
|
|
QUuid uuid = QUuid(args.at(2));
|
||
|
|
QByteArray guid = QByteArray::fromRawData((const char*)&uuid.data1, sizeof(EFI_GUID));
|
||
|
|
bool converted;
|
||
|
|
UINT8 sectionType = (UINT8)args.at(3).toUShort(&converted, 16);
|
||
|
|
if (!converted)
|
||
|
|
result = ERR_INVALID_PARAMETER;
|
||
|
|
else
|
||
|
|
result = r.replace(args.at(1), guid, sectionType, args.at(4));
|
||
|
|
|
||
|
|
switch (result) {
|
||
|
|
case ERR_SUCCESS:
|
||
|
|
std::cout << "File replaced" << std::endl;
|
||
|
|
break;
|
||
|
|
case ERR_INVALID_PARAMETER:
|
||
|
|
std::cout << "Function called with invalid parameter" << std::endl;
|
||
|
|
break;
|
||
|
|
case ERR_INVALID_FILE:
|
||
|
|
std::cout << "Invalid/corrupted file specified" << std::endl;
|
||
|
|
break;
|
||
|
|
case ERR_INVALID_SECTION:
|
||
|
|
std::cout << "Invalid/corrupted section specified" << std::endl;
|
||
|
|
break;
|
||
|
|
case ERR_NOTHING_TO_PATCH:
|
||
|
|
std::cout << "No replacements can be applied to input file" << std::endl;
|
||
|
|
break;
|
||
|
|
case ERR_NOT_IMPLEMENTED:
|
||
|
|
std::cout << "Can't replace body of this section type" << std::endl;
|
||
|
|
break;
|
||
|
|
case ERR_FILE_OPEN:
|
||
|
|
std::cout << "Input file not found" << std::endl;
|
||
|
|
break;
|
||
|
|
case ERR_FILE_READ:
|
||
|
|
std::cout << "Input file can't be read" << std::endl;
|
||
|
|
break;
|
||
|
|
case ERR_FILE_WRITE:
|
||
|
|
std::cout << "Output file can't be written" << std::endl;
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
std::cout << "Error " << result << std::endl;
|
||
|
|
}
|
||
|
|
|
||
|
|
return result;
|
||
|
|
}
|