357 lines
12 KiB
C++
357 lines
12 KiB
C++
#include "Archive.h"
|
|
#include "Resource.h"
|
|
#include "File.h"
|
|
#include "spdlog/spdlog.h"
|
|
#include "Utils/StringHelper.h"
|
|
#include "Lib/StrHash64.h"
|
|
#include <filesystem>
|
|
|
|
namespace Ship {
|
|
Archive::Archive(const std::string& MainPath, bool enableWriting) : Archive(MainPath, "", enableWriting)
|
|
{
|
|
mainMPQ = nullptr;
|
|
}
|
|
|
|
Archive::Archive(const std::string& MainPath, const std::string& PatchesPath, bool enableWriting, bool genCRCMap) : MainPath(MainPath), PatchesPath(PatchesPath) {
|
|
mainMPQ = nullptr;
|
|
Load(enableWriting, genCRCMap);
|
|
}
|
|
|
|
Archive::~Archive() {
|
|
Unload();
|
|
}
|
|
|
|
bool Archive::IsMainMPQValid()
|
|
{
|
|
return mainMPQ != nullptr;
|
|
}
|
|
|
|
std::shared_ptr<Archive> Archive::CreateArchive(const std::string& archivePath, int fileCapacity)
|
|
{
|
|
Archive* archive = new Archive(archivePath, true);
|
|
|
|
TCHAR* t_filename = new TCHAR[archivePath.size() + 1];
|
|
t_filename[archivePath.size()] = 0;
|
|
std::copy(archivePath.begin(), archivePath.end(), t_filename);
|
|
|
|
bool success = SFileCreateArchive(t_filename, MPQ_CREATE_LISTFILE | MPQ_CREATE_ATTRIBUTES | MPQ_CREATE_ARCHIVE_V2, fileCapacity, &archive->mainMPQ);
|
|
int error = GetLastError();
|
|
|
|
if (success) {
|
|
archive->mpqHandles[archivePath] = archive->mainMPQ;
|
|
return std::make_shared<Archive>(*archive);
|
|
} else {
|
|
SPDLOG_ERROR("({}) We tried to create an archive, but it has fallen and cannot get up.");
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
std::shared_ptr<File> Archive::LoadFile(const std::string& filePath, bool includeParent, std::shared_ptr<File> FileToLoad) {
|
|
HANDLE fileHandle = NULL;
|
|
|
|
if (!SFileOpenFileEx(mainMPQ, filePath.c_str(), 0, &fileHandle)) {
|
|
SPDLOG_ERROR("({}) Failed to open file {} from mpq archive {}", GetLastError(), filePath.c_str(), MainPath.c_str());
|
|
std::unique_lock<std::mutex> Lock(FileToLoad->FileLoadMutex);
|
|
FileToLoad->bHasLoadError = true;
|
|
return nullptr;
|
|
}
|
|
|
|
DWORD dwFileSize = SFileGetFileSize(fileHandle, 0);
|
|
std::shared_ptr<char[]> fileData(new char[dwFileSize]);
|
|
DWORD dwBytes;
|
|
|
|
if (!SFileReadFile(fileHandle, fileData.get(), dwFileSize, &dwBytes, NULL)) {
|
|
SPDLOG_ERROR("({}) Failed to read file {} from mpq archive {}", GetLastError(), filePath.c_str(), MainPath.c_str());
|
|
if (!SFileCloseFile(fileHandle)) {
|
|
SPDLOG_ERROR("({}) Failed to close file {} from mpq after read failure in archive {}", GetLastError(), filePath.c_str(), MainPath.c_str());
|
|
}
|
|
std::unique_lock<std::mutex> Lock(FileToLoad->FileLoadMutex);
|
|
FileToLoad->bHasLoadError = true;
|
|
return nullptr;
|
|
}
|
|
|
|
if (!SFileCloseFile(fileHandle)) {
|
|
SPDLOG_ERROR("({}) Failed to close file {} from mpq archive {}", GetLastError(), filePath.c_str(), MainPath.c_str());
|
|
}
|
|
|
|
if (FileToLoad == nullptr) {
|
|
FileToLoad = std::make_shared<File>();
|
|
FileToLoad->path = filePath;
|
|
}
|
|
|
|
std::unique_lock<std::mutex> Lock(FileToLoad->FileLoadMutex);
|
|
FileToLoad->parent = includeParent ? shared_from_this() : nullptr;
|
|
FileToLoad->buffer = fileData;
|
|
FileToLoad->dwBufferSize = dwFileSize;
|
|
FileToLoad->bIsLoaded = true;
|
|
|
|
return FileToLoad;
|
|
}
|
|
|
|
std::shared_ptr<File> Archive::LoadPatchFile(const std::string& filePath, bool includeParent, std::shared_ptr<File> FileToLoad) {
|
|
HANDLE fileHandle = NULL;
|
|
HANDLE mpqHandle = NULL;
|
|
|
|
for(auto [path, handle] : mpqHandles) {
|
|
if (SFileOpenFileEx(mpqHandle, filePath.c_str(), 0, &fileHandle)) {
|
|
std::unique_lock Lock(FileToLoad->FileLoadMutex);
|
|
FileToLoad->bHasLoadError = false;
|
|
mpqHandle = handle;
|
|
}
|
|
}
|
|
|
|
if(!mpqHandle) {
|
|
FileToLoad->bHasLoadError = true;
|
|
return FileToLoad;
|
|
}
|
|
|
|
DWORD dwFileSize = SFileGetFileSize(fileHandle, 0);
|
|
std::shared_ptr<char[]> fileData(new char[dwFileSize]);
|
|
DWORD dwBytes;
|
|
|
|
if (!SFileReadFile(fileHandle, fileData.get(), dwFileSize, &dwBytes, NULL)) {
|
|
SPDLOG_ERROR("({}) Failed to read file {} from mpq archive {}", GetLastError(), filePath.c_str(), MainPath.c_str());
|
|
if (!SFileCloseFile(fileHandle)) {
|
|
SPDLOG_ERROR("({}) Failed to close file {} from mpq after read failure in archive {}", GetLastError(), filePath.c_str(), MainPath.c_str());
|
|
}
|
|
std::unique_lock<std::mutex> Lock(FileToLoad->FileLoadMutex);
|
|
FileToLoad->bHasLoadError = true;
|
|
return nullptr;
|
|
}
|
|
|
|
if (!SFileCloseFile(fileHandle)) {
|
|
SPDLOG_ERROR("({}) Failed to close file {} from mpq archive {}", GetLastError(), filePath.c_str(), MainPath.c_str());
|
|
}
|
|
|
|
if (FileToLoad == nullptr) {
|
|
FileToLoad = std::make_shared<File>();
|
|
FileToLoad->path = filePath;
|
|
}
|
|
|
|
std::unique_lock<std::mutex> Lock(FileToLoad->FileLoadMutex);
|
|
FileToLoad->parent = includeParent ? shared_from_this() : nullptr;
|
|
FileToLoad->buffer = fileData;
|
|
FileToLoad->dwBufferSize = dwFileSize;
|
|
FileToLoad->bIsLoaded = true;
|
|
|
|
return FileToLoad;
|
|
}
|
|
|
|
bool Archive::AddFile(const std::string& path, uintptr_t fileData, DWORD dwFileSize) {
|
|
HANDLE hFile;
|
|
|
|
SYSTEMTIME sysTime;
|
|
GetSystemTime(&sysTime);
|
|
FILETIME t;
|
|
SystemTimeToFileTime(&sysTime, &t);
|
|
ULONGLONG stupidHack = static_cast<uint64_t>(t.dwHighDateTime) << (sizeof(t.dwHighDateTime) * 8) | t.dwLowDateTime;
|
|
|
|
if (!SFileCreateFile(mainMPQ, path.c_str(), stupidHack, dwFileSize, 0, MPQ_FILE_COMPRESS, &hFile)) {
|
|
SPDLOG_ERROR("({}) Failed to create file of {} bytes {} in archive {}", GetLastError(), dwFileSize, path.c_str(), MainPath.c_str());
|
|
return false;
|
|
}
|
|
|
|
if (!SFileWriteFile(hFile, (void*)fileData, dwFileSize, MPQ_COMPRESSION_ZLIB)) {
|
|
SPDLOG_ERROR("({}) Failed to write {} bytes to {} in archive {}", GetLastError(), dwFileSize, path.c_str(), MainPath.c_str());
|
|
if (!SFileCloseFile(hFile)) {
|
|
SPDLOG_ERROR("({}) Failed to close file {} after write failure in archive {}", GetLastError(), path.c_str(), MainPath.c_str());
|
|
}
|
|
return false;
|
|
}
|
|
|
|
if (!SFileFinishFile(hFile)) {
|
|
SPDLOG_ERROR("({}) Failed to finish file {} in archive {}", GetLastError(), path.c_str(), MainPath.c_str());
|
|
if (!SFileCloseFile(hFile)) {
|
|
SPDLOG_ERROR("({}) Failed to close file {} after finish failure in archive {}", GetLastError(), path.c_str(), MainPath.c_str());
|
|
}
|
|
return false;
|
|
}
|
|
// SFileFinishFile already frees the handle, so no need to close it again.
|
|
|
|
addedFiles.push_back(path);
|
|
hashes[CRC64(path.c_str())] = path;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool Archive::RemoveFile(const std::string& path) {
|
|
// TODO: Notify the resource manager and child Files
|
|
|
|
if (!SFileRemoveFile(mainMPQ, path.c_str(), 0)) {
|
|
SPDLOG_ERROR("({}) Failed to remove file {} in archive {}", GetLastError(), path.c_str(), MainPath.c_str());
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool Archive::RenameFile(const std::string& oldPath, const std::string& newPath) {
|
|
// TODO: Notify the resource manager and child Files
|
|
|
|
if (!SFileRenameFile(mainMPQ, oldPath.c_str(), newPath.c_str())) {
|
|
SPDLOG_ERROR("({}) Failed to rename file {} to {} in archive {}", GetLastError(), oldPath.c_str(), newPath.c_str(), MainPath.c_str());
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
std::vector<SFILE_FIND_DATA> Archive::ListFiles(const std::string& searchMask) {
|
|
auto fileList = std::vector<SFILE_FIND_DATA>();
|
|
SFILE_FIND_DATA findContext;
|
|
HANDLE hFind;
|
|
|
|
|
|
hFind = SFileFindFirstFile(mainMPQ, searchMask.c_str(), &findContext, nullptr);
|
|
//if (hFind && GetLastError() != ERROR_NO_MORE_FILES) {
|
|
if (hFind != nullptr) {
|
|
fileList.push_back(findContext);
|
|
|
|
bool fileFound;
|
|
do {
|
|
fileFound = SFileFindNextFile(hFind, &findContext);
|
|
|
|
if (fileFound) {
|
|
fileList.push_back(findContext);
|
|
}
|
|
else if (!fileFound && GetLastError() != ERROR_NO_MORE_FILES)
|
|
//else if (!fileFound)
|
|
{
|
|
SPDLOG_ERROR("({}), Failed to search with mask {} in archive {}", GetLastError(), searchMask.c_str(), MainPath.c_str());
|
|
if (!SListFileFindClose(hFind)) {
|
|
SPDLOG_ERROR("({}) Failed to close file search {} after failure in archive {}", GetLastError(), searchMask.c_str(), MainPath.c_str());
|
|
}
|
|
return fileList;
|
|
}
|
|
} while (fileFound);
|
|
}
|
|
else if (GetLastError() != ERROR_NO_MORE_FILES) {
|
|
SPDLOG_ERROR("({}), Failed to search with mask {} in archive {}", GetLastError(), searchMask.c_str(), MainPath.c_str());
|
|
return fileList;
|
|
}
|
|
|
|
if (hFind != nullptr)
|
|
{
|
|
if (!SFileFindClose(hFind)) {
|
|
SPDLOG_ERROR("({}) Failed to close file search {} in archive {}", GetLastError(), searchMask.c_str(), MainPath.c_str());
|
|
}
|
|
}
|
|
|
|
return fileList;
|
|
}
|
|
|
|
bool Archive::HasFile(const std::string& filename) {
|
|
bool result = false;
|
|
auto start = std::chrono::steady_clock::now();
|
|
|
|
auto lst = ListFiles(filename);
|
|
|
|
for (const auto& item : lst) {
|
|
if (item.cFileName == filename) {
|
|
result = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
auto end = std::chrono::steady_clock::now();
|
|
auto diff = std::chrono::duration_cast<std::chrono::milliseconds>(end - start).count();
|
|
|
|
return result;
|
|
}
|
|
|
|
std::string Archive::HashToString(uint64_t hash) {
|
|
return hashes[hash];
|
|
}
|
|
|
|
bool Archive::Load(bool enableWriting, bool genCRCMap) {
|
|
return LoadMainMPQ(enableWriting, genCRCMap) && LoadPatchMPQs();
|
|
}
|
|
|
|
bool Archive::Unload()
|
|
{
|
|
bool success = true;
|
|
for (const auto& mpqHandle : mpqHandles) {
|
|
if (!SFileCloseArchive(mpqHandle.second)) {
|
|
SPDLOG_ERROR("({}) Failed to close mpq {}", GetLastError(), mpqHandle.first.c_str());
|
|
success = false;
|
|
}
|
|
}
|
|
|
|
mainMPQ = nullptr;
|
|
|
|
return success;
|
|
}
|
|
|
|
bool Archive::LoadPatchMPQs() {
|
|
// OTRTODO: We also want to periodically scan the patch directories for new MPQs. When new MPQs are found we will load the contents to fileCache and then copy over to gameResourceAddresses
|
|
if (PatchesPath.length() > 0) {
|
|
if (std::filesystem::is_directory(PatchesPath)) {
|
|
for (const auto& p : std::filesystem::recursive_directory_iterator(PatchesPath)) {
|
|
if (StringHelper::IEquals(p.path().extension().string(), ".otr") || StringHelper::IEquals(p.path().extension().string(), ".mpq")) {
|
|
SPDLOG_ERROR("Reading {} mpq patch", p.path().string().c_str());
|
|
if (!LoadPatchMPQ(p.path().string())) {
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool Archive::LoadMainMPQ(bool enableWriting, bool genCRCMap) {
|
|
HANDLE mpqHandle = NULL;
|
|
std::string fullPath = std::filesystem::absolute(MainPath).string();
|
|
|
|
std::wstring wFileName = std::filesystem::absolute(MainPath).wstring();
|
|
|
|
if (!SFileOpenArchive(wFileName.c_str(), 0, enableWriting ? 0 : MPQ_OPEN_READ_ONLY, &mpqHandle)) {
|
|
SPDLOG_ERROR("({}) Failed to open main mpq file {}.", GetLastError(), fullPath.c_str());
|
|
return false;
|
|
}
|
|
|
|
mpqHandles[fullPath] = mpqHandle;
|
|
mainMPQ = mpqHandle;
|
|
|
|
if (genCRCMap) {
|
|
auto listFile = LoadFile("(listfile)", false);
|
|
|
|
std::vector<std::string> lines = StringHelper::Split(std::string(listFile->buffer.get(), listFile->dwBufferSize), "\n");
|
|
|
|
for (size_t i = 0; i < lines.size(); i++) {
|
|
std::string line = StringHelper::Strip(lines[i], "\r");
|
|
//uint64_t hash = StringHelper::StrToL(lines[i], 16);
|
|
|
|
uint64_t hash = CRC64(line.c_str());
|
|
hashes[hash] = line;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool Archive::LoadPatchMPQ(const std::string& path) {
|
|
HANDLE patchHandle = NULL;
|
|
std::string fullPath = std::filesystem::absolute(path).string();
|
|
if (mpqHandles.contains(fullPath)) {
|
|
return true;
|
|
}
|
|
|
|
std::wstring wPath = std::filesystem::absolute(path).wstring();
|
|
|
|
if (!SFileOpenArchive(wPath.c_str(), 0, MPQ_OPEN_READ_ONLY, &patchHandle)) {
|
|
SPDLOG_ERROR("({}) Failed to open patch mpq file {} while applying to {}.", GetLastError(), path.c_str(), MainPath.c_str());
|
|
return false;
|
|
}
|
|
|
|
if (!SFileOpenPatchArchive(mainMPQ, wPath.c_str(), "", 0)) {
|
|
SPDLOG_ERROR("({}) Failed to apply patch mpq file {} to main mpq {}.", GetLastError(), path.c_str(), MainPath.c_str());
|
|
return false;
|
|
}
|
|
|
|
mpqHandles[fullPath] = patchHandle;
|
|
|
|
return true;
|
|
}
|
|
} |