1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#include "SWrapper.h"
#include <filesystem>
#include <iostream>
#include <fstream>
#include "spdlog/spdlog.h"
namespace fs = std::filesystem;
bool debugMode = true;
SWrapper::SWrapper(const std::string& path) {
if(fs::exists(path)) {
fs::remove(path);
}
if(!SFileCreateArchive(path.c_str(), MPQ_CREATE_LISTFILE | MPQ_CREATE_ATTRIBUTES | MPQ_CREATE_ARCHIVE_V2, 16 * 1024, &this->hMpq)){
SPDLOG_ERROR("Failed to create archive {} with error code {}", path, GetLastError());
return;
}
}
bool SWrapper::CreateFile(const std::string& path, std::vector<char> data) {
if(debugMode){
SPDLOG_INFO("Creating debug file: debug/{}", path);
std::string dpath = "debug/" + path;
if(!fs::exists(fs::path(dpath).parent_path())){
fs::create_directories(fs::path(dpath).parent_path());
}
std::ofstream stream(dpath, std::ios::binary);
stream.write((char*)data.data(), data.size());
stream.close();
}
HANDLE hFile;
#ifdef _WIN32
SYSTEMTIME sysTime;
GetSystemTime(&sysTime);
FILETIME t;
SystemTimeToFileTime(&sysTime, &t);
ULONGLONG theTime = static_cast<uint64_t>(t.dwHighDateTime) << (sizeof(t.dwHighDateTime) * 8) | t.dwLowDateTime;
#else
time_t theTime;
time(&theTime);
#endif
char* raw = data.data();
size_t size = data.size();
if(size == 0){
SPDLOG_ERROR("File at path {} is empty", path);
return false;
}
if(size >> 32){
throw std::runtime_error("File at path " + path + " is too large with size " + std::to_string(size));
}
if(!SFileCreateFile(this->hMpq, path.c_str(), theTime, size, 0, MPQ_FILE_COMPRESS, &hFile)){
throw std::runtime_error("Failed to create file at path " + path + " with error " + std::to_string(GetLastError()));
}
if(!SFileWriteFile(hFile, (void*) raw, size, MPQ_COMPRESSION_ZLIB)){
throw std::runtime_error("Failed to write file at path " + path + " with error " + std::to_string(GetLastError()));
}
if(!SFileCloseFile(hFile)){
throw std::runtime_error("Failed to close file at path " + path + " with error " + std::to_string(GetLastError()));
}
return true;
}
void SWrapper::Close() {
if(this->hMpq == nullptr) {
SPDLOG_ERROR("Archive already closed");
return;
}
SFileCloseArchive(this->hMpq);
}
|
