mirror of
https://github.com/LongSoft/UEFITool.git
synced 2024-11-25 17:38:22 +08:00
143 lines
3.8 KiB
C++
143 lines
3.8 KiB
C++
/* guiddatabase.cpp
|
|
|
|
Copyright (c) 2017, LongSoft. 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,
|
|
WITHWARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
|
|
*/
|
|
|
|
#include "guiddatabase.h"
|
|
#include "ubytearray.h"
|
|
#include "ffs.h"
|
|
|
|
#include <fstream>
|
|
#include <string>
|
|
|
|
#if defined(U_ENABLE_GUID_DATABASE_SUPPORT)
|
|
#include <sstream>
|
|
#include <vector>
|
|
#include <cstdio>
|
|
|
|
static GuidDatabase gLocalGuidDatabase;
|
|
|
|
#ifdef QT_CORE_LIB
|
|
|
|
#include <QFile>
|
|
#include <QTextStream>
|
|
|
|
// This is required to be able to read Qt-embedded paths
|
|
|
|
static std::string readGuidDatabase(const UString &path) {
|
|
QFile guids(path);
|
|
if (guids.open(QFile::ReadOnly | QFile::Text))
|
|
return QTextStream(&guids).readAll().toStdString();
|
|
return std::string {};
|
|
}
|
|
|
|
#else
|
|
|
|
static std::string readGuidDatabase(const UString &path) {
|
|
std::ifstream guids(path.toLocal8Bit());
|
|
std::stringstream ret;
|
|
if (ret)
|
|
ret << guids.rdbuf();
|
|
return ret.str();
|
|
}
|
|
|
|
#endif
|
|
|
|
void initGuidDatabase(const UString & path, UINT32* numEntries)
|
|
{
|
|
gLocalGuidDatabase.clear();
|
|
|
|
std::stringstream file(readGuidDatabase(path));
|
|
|
|
while (!file.eof()) {
|
|
std::string line;
|
|
std::getline(file, line);
|
|
|
|
// Use sharp symbol as commentary
|
|
if (line.size() == 0 || line[0] == '#')
|
|
continue;
|
|
|
|
// GUID and name are comma-separated
|
|
std::vector<UString> lineParts;
|
|
std::string::size_type prev = 0, curr = 0;
|
|
while ((curr = line.find(',', curr)) != std::string::npos) {
|
|
std::string substring( line.substr(prev, curr-prev) );
|
|
lineParts.push_back(UString(substring.c_str()));
|
|
prev = ++curr;
|
|
}
|
|
lineParts.push_back(UString(line.substr(prev, curr-prev).c_str()));
|
|
|
|
if (lineParts.size() < 2)
|
|
continue;
|
|
|
|
EFI_GUID guid;
|
|
if (!ustringToGuid(lineParts[0], guid))
|
|
continue;
|
|
|
|
gLocalGuidDatabase[guid] = lineParts[1];
|
|
}
|
|
|
|
if (numEntries)
|
|
*numEntries = (UINT32)gLocalGuidDatabase.size();
|
|
}
|
|
|
|
UString guidDatabaseLookup(const EFI_GUID & guid)
|
|
{
|
|
return gLocalGuidDatabase[guid];
|
|
}
|
|
|
|
#else
|
|
void initGuidDatabase(const UString & path, UINT32* numEntries)
|
|
{
|
|
U_UNUSED_PARAMETER(path);
|
|
if (numEntries)
|
|
*numEntries = 0;
|
|
}
|
|
|
|
UString guidDatabaseLookup(const EFI_GUID & guid)
|
|
{
|
|
U_UNUSED_PARAMETER(guid);
|
|
return UString();
|
|
}
|
|
#endif
|
|
|
|
GuidDatabase guidDatabaseFromTreeRecursive(TreeModel * model, const UModelIndex index)
|
|
{
|
|
GuidDatabase db;
|
|
|
|
if (!index.isValid())
|
|
return db;
|
|
|
|
for (int i = 0; i < model->rowCount(index); i++) {
|
|
GuidDatabase tmpDb = guidDatabaseFromTreeRecursive(model, index.model()->index(i, index.column(), index));
|
|
|
|
db.insert(tmpDb.begin(), tmpDb.end());
|
|
}
|
|
|
|
if (model->type(index) == Types::File && !model->text(index).isEmpty())
|
|
db[readUnaligned((const EFI_GUID*)model->header(index).left(16).constData())] = model->text(index);
|
|
|
|
return db;
|
|
}
|
|
|
|
USTATUS guidDatabaseExportToFile(const UString & outPath, GuidDatabase & db)
|
|
{
|
|
std::ofstream outputFile(outPath.toLocal8Bit(), std::ios::out | std::ios::trunc);
|
|
if (!outputFile)
|
|
return U_FILE_OPEN;
|
|
for (GuidDatabase::iterator it = db.begin(); it != db.end(); it++) {
|
|
std::string guid(guidToUString (it->first, false).toLocal8Bit());
|
|
std::string name(it->second.toLocal8Bit());
|
|
outputFile << guid << ',' << name << '\n';
|
|
}
|
|
|
|
return U_SUCCESS;
|
|
}
|