jh163888/UEFITool
1
0
Fork 0
mirror of https://github.com/LongSoft/UEFITool.git synced 2024-11-22 16:08:23 +08:00
Code Issues Projects Releases Wiki Activity

UEFIExtract 0.3.0

Browse Source 
- added filtering by FFS file GUID, only specified files can now be
unpacked
This commit is contained in:
Nikolaj Schlej 2014-08-15 13:24:03 +02:00
parent 754f9c5b13
commit 701717c554
5 changed files with 99 additions and 64 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
UEFIExtract/uefiextract.cpp
View File

@ -24,9 +24,9 @@ UEFIExtract::~UEFIExtract()
delete ffsEngine; delete ffsEngine;
} }
UINT8 UEFIExtract::extractAll(QString path) UINT8 UEFIExtract::init(const QString & path)
{ {
QFileInfo fileInfo = QFileInfo(path); fileInfo = QFileInfo(path);
if (!fileInfo.exists()) if (!fileInfo.exists())
return ERR_FILE_OPEN; return ERR_FILE_OPEN;
@ -44,8 +44,14 @@ UINT8 UEFIExtract::extractAll(QString path)
if (result) if (result)
return result; return result;
return ERR_SUCCESS;
}
UINT8 UEFIExtract::extract(QString guid)
{
QModelIndex rootIndex = ffsEngine->treeModel()->index(0, 0); QModelIndex rootIndex = ffsEngine->treeModel()->index(0, 0);
result = ffsEngine->dump(rootIndex, fileInfo.fileName().append(".dump"));
UINT8 result = ffsEngine->dump(rootIndex, fileInfo.fileName().append(".dump"), guid);
if (result) if (result)
return result; return result;

5
UEFIExtract/uefiextract.h
View File

@ -31,10 +31,13 @@ public:
explicit UEFIExtract(QObject *parent = 0); explicit UEFIExtract(QObject *parent = 0);
~UEFIExtract(); ~UEFIExtract();
UINT8 extractAll(QString path); UINT8 init(const QString & path);
UINT8 extract(QString guid = QString());
private: private:
FfsEngine* ffsEngine; FfsEngine* ffsEngine;
QFileInfo fileInfo;
}; };
#endif #endif

28
UEFIExtract/uefiextract_main.cpp
View File

@ -25,8 +25,11 @@ int main(int argc, char *argv[])
UEFIExtract w; UEFIExtract w;
UINT8 result = ERR_SUCCESS; UINT8 result = ERR_SUCCESS;
if (a.arguments().length() > 1) { if (a.arguments().length() > 1 ) {
result = w.extractAll(a.arguments().at(1)); w.init(a.arguments().at(1));
if (a.arguments().length() == 2) {
result = w.extract();
switch (result) { switch (result) {
case ERR_DIR_ALREADY_EXIST: case ERR_DIR_ALREADY_EXIST:
std::cout << "Dump directory already exist, please remove it" << std::endl; std::cout << "Dump directory already exist, please remove it" << std::endl;
@ -39,10 +42,27 @@ int main(int argc, char *argv[])
break; break;
} }
} }
else {
for (int i = 2; i < a.arguments().length(); i++) {
result = w.extract(a.arguments().at(i));
switch (result) {
case ERR_DIR_ALREADY_EXIST:
std::cout << "Dump directory already exist, please remove it" << std::endl;
break;
case ERR_DIR_CREATE:
std::cout << "Can't create directory" << std::endl;
break;
case ERR_FILE_OPEN:
std::cout << "Can't create file" << std::endl;
break;
}
}
}
}
else { else {
result = ERR_INVALID_PARAMETER; result = ERR_INVALID_PARAMETER;
std::cout << "UEFIExtract 0.2.2" << std::endl << std::endl << std::cout << "UEFIExtract 0.3.0" << std::endl << std::endl <<
"Usage: uefiextract imagefile\n" << std::endl; "Usage: uefiextract imagefile [FileGUID_1 FileGUID_2 ...]\n" << std::endl;
} }
return result; return result;

10
ffsengine.cpp
View File

@ -3432,12 +3432,17 @@ UINT32 FfsEngine::crc32(UINT32 initial, const UINT8* buffer, UINT32 length)
return(crc32 ^ 0xFFFFFFFF); return(crc32 ^ 0xFFFFFFFF);
} }
UINT8 FfsEngine::dump(const QModelIndex & index, const QString path) UINT8 FfsEngine::dump(const QModelIndex & index, const QString & path, const QString & guid)
{ {
if (!index.isValid()) if (!index.isValid())
return ERR_INVALID_PARAMETER; return ERR_INVALID_PARAMETER;
QDir dir; QDir dir;
if (guid.isEmpty() ||
guidToQString(*(EFI_GUID*)model->header(index).constData()) == guid ||
guidToQString(*(EFI_GUID*)model->header(model->findParentOfType(index, Types::File)).constData()) == guid) {
if (dir.cd(path)) if (dir.cd(path))
return ERR_DIR_ALREADY_EXIST; return ERR_DIR_ALREADY_EXIST;
@ -3471,12 +3476,13 @@ UINT8 FfsEngine::dump(const QModelIndex & index, const QString path)
return ERR_FILE_OPEN; return ERR_FILE_OPEN;
file.write(info.toLatin1()); file.write(info.toLatin1());
file.close(); file.close();
}
UINT8 result; UINT8 result;
for (int i = 0; i < model->rowCount(index); i++) { for (int i = 0; i < model->rowCount(index); i++) {
QModelIndex childIndex = index.child(i, 0); QModelIndex childIndex = index.child(i, 0);
QString childPath = tr("%1/%2 %3").arg(path).arg(i).arg(model->textString(childIndex).isEmpty() ? model->nameString(childIndex) : model->textString(childIndex)); QString childPath = tr("%1/%2 %3").arg(path).arg(i).arg(model->textString(childIndex).isEmpty() ? model->nameString(childIndex) : model->textString(childIndex));
result = dump(childIndex, childPath); result = dump(childIndex, childPath, guid);
if (result) if (result)
return result; return result;
} }

2
ffsengine.h
View File

@ -94,7 +94,7 @@ public:
UINT8 replace(const QModelIndex & index, const QByteArray & object, const UINT8 mode); UINT8 replace(const QModelIndex & index, const QByteArray & object, const UINT8 mode);
UINT8 remove(const QModelIndex & index); UINT8 remove(const QModelIndex & index);
UINT8 rebuild(const QModelIndex & index); UINT8 rebuild(const QModelIndex & index);
UINT8 dump(const QModelIndex & index, const QString path); UINT8 dump(const QModelIndex & index, const QString & path, const QString & filter = QString());
UINT8 patch(const QModelIndex & index, const QVector<PatchData> & patches); UINT8 patch(const QModelIndex & index, const QVector<PatchData> & patches);
// Search routines // Search routines