treballant en resources.pack
This commit is contained in:
222
source/resource_pack.cpp
Normal file
222
source/resource_pack.cpp
Normal file
@@ -0,0 +1,222 @@
|
||||
#include "resource_pack.h"
|
||||
#include <fstream>
|
||||
#include <iostream>
|
||||
#include <filesystem>
|
||||
#include <algorithm>
|
||||
|
||||
const std::string ResourcePack::DEFAULT_ENCRYPT_KEY = "CCAE_RESOURCES_2024";
|
||||
|
||||
ResourcePack::ResourcePack() : loaded(false) {}
|
||||
|
||||
ResourcePack::~ResourcePack() {
|
||||
clear();
|
||||
}
|
||||
|
||||
uint32_t ResourcePack::calculateChecksum(const std::vector<uint8_t>& data) {
|
||||
uint32_t checksum = 0x12345678;
|
||||
for (size_t i = 0; i < data.size(); ++i) {
|
||||
checksum = ((checksum << 5) + checksum) + data[i];
|
||||
}
|
||||
return checksum;
|
||||
}
|
||||
|
||||
void ResourcePack::encryptData(std::vector<uint8_t>& data, const std::string& key) {
|
||||
if (key.empty()) return;
|
||||
|
||||
for (size_t i = 0; i < data.size(); ++i) {
|
||||
data[i] ^= key[i % key.length()];
|
||||
}
|
||||
}
|
||||
|
||||
void ResourcePack::decryptData(std::vector<uint8_t>& data, const std::string& key) {
|
||||
encryptData(data, key);
|
||||
}
|
||||
|
||||
bool ResourcePack::loadPack(const std::string& packFile) {
|
||||
std::ifstream file(packFile, std::ios::binary);
|
||||
if (!file) {
|
||||
std::cerr << "Error: Could not open pack file: " << packFile << std::endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
char header[4];
|
||||
file.read(header, 4);
|
||||
if (std::string(header, 4) != "CCAE") {
|
||||
std::cerr << "Error: Invalid pack file format" << std::endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
uint32_t version;
|
||||
file.read(reinterpret_cast<char*>(&version), sizeof(version));
|
||||
if (version != 1) {
|
||||
std::cerr << "Error: Unsupported pack version: " << version << std::endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
uint32_t resourceCount;
|
||||
file.read(reinterpret_cast<char*>(&resourceCount), sizeof(resourceCount));
|
||||
|
||||
resources.clear();
|
||||
resources.reserve(resourceCount);
|
||||
|
||||
for (uint32_t i = 0; i < resourceCount; ++i) {
|
||||
uint32_t filenameLength;
|
||||
file.read(reinterpret_cast<char*>(&filenameLength), sizeof(filenameLength));
|
||||
|
||||
std::string filename(filenameLength, '\0');
|
||||
file.read(&filename[0], filenameLength);
|
||||
|
||||
ResourceEntry entry;
|
||||
entry.filename = filename;
|
||||
file.read(reinterpret_cast<char*>(&entry.offset), sizeof(entry.offset));
|
||||
file.read(reinterpret_cast<char*>(&entry.size), sizeof(entry.size));
|
||||
file.read(reinterpret_cast<char*>(&entry.checksum), sizeof(entry.checksum));
|
||||
|
||||
resources[filename] = entry;
|
||||
}
|
||||
|
||||
uint64_t dataSize;
|
||||
file.read(reinterpret_cast<char*>(&dataSize), sizeof(dataSize));
|
||||
|
||||
data.resize(dataSize);
|
||||
file.read(reinterpret_cast<char*>(data.data()), dataSize);
|
||||
|
||||
decryptData(data, DEFAULT_ENCRYPT_KEY);
|
||||
|
||||
loaded = true;
|
||||
return true;
|
||||
}
|
||||
|
||||
bool ResourcePack::savePack(const std::string& packFile) {
|
||||
std::ofstream file(packFile, std::ios::binary);
|
||||
if (!file) {
|
||||
std::cerr << "Error: Could not create pack file: " << packFile << std::endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
file.write("CCAE", 4);
|
||||
|
||||
uint32_t version = 1;
|
||||
file.write(reinterpret_cast<const char*>(&version), sizeof(version));
|
||||
|
||||
uint32_t resourceCount = static_cast<uint32_t>(resources.size());
|
||||
file.write(reinterpret_cast<const char*>(&resourceCount), sizeof(resourceCount));
|
||||
|
||||
for (const auto& [filename, entry] : resources) {
|
||||
uint32_t filenameLength = static_cast<uint32_t>(filename.length());
|
||||
file.write(reinterpret_cast<const char*>(&filenameLength), sizeof(filenameLength));
|
||||
file.write(filename.c_str(), filenameLength);
|
||||
|
||||
file.write(reinterpret_cast<const char*>(&entry.offset), sizeof(entry.offset));
|
||||
file.write(reinterpret_cast<const char*>(&entry.size), sizeof(entry.size));
|
||||
file.write(reinterpret_cast<const char*>(&entry.checksum), sizeof(entry.checksum));
|
||||
}
|
||||
|
||||
std::vector<uint8_t> encryptedData = data;
|
||||
encryptData(encryptedData, DEFAULT_ENCRYPT_KEY);
|
||||
|
||||
uint64_t dataSize = encryptedData.size();
|
||||
file.write(reinterpret_cast<const char*>(&dataSize), sizeof(dataSize));
|
||||
file.write(reinterpret_cast<const char*>(encryptedData.data()), dataSize);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool ResourcePack::addFile(const std::string& filename, const std::string& filepath) {
|
||||
std::ifstream file(filepath, std::ios::binary | std::ios::ate);
|
||||
if (!file) {
|
||||
std::cerr << "Error: Could not open file: " << filepath << std::endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
std::streamsize fileSize = file.tellg();
|
||||
file.seekg(0, std::ios::beg);
|
||||
|
||||
std::vector<uint8_t> fileData(fileSize);
|
||||
if (!file.read(reinterpret_cast<char*>(fileData.data()), fileSize)) {
|
||||
std::cerr << "Error: Could not read file: " << filepath << std::endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
ResourceEntry entry;
|
||||
entry.filename = filename;
|
||||
entry.offset = data.size();
|
||||
entry.size = fileData.size();
|
||||
entry.checksum = calculateChecksum(fileData);
|
||||
|
||||
data.insert(data.end(), fileData.begin(), fileData.end());
|
||||
resources[filename] = entry;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool ResourcePack::addDirectory(const std::string& directory) {
|
||||
if (!std::filesystem::exists(directory)) {
|
||||
std::cerr << "Error: Directory does not exist: " << directory << std::endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
for (const auto& entry : std::filesystem::recursive_directory_iterator(directory)) {
|
||||
if (entry.is_regular_file()) {
|
||||
std::string filepath = entry.path().string();
|
||||
std::string filename = std::filesystem::relative(entry.path(), directory).string();
|
||||
|
||||
std::replace(filename.begin(), filename.end(), '\\', '/');
|
||||
|
||||
if (!addFile(filename, filepath)) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
std::vector<uint8_t> ResourcePack::getResource(const std::string& filename) {
|
||||
auto it = resources.find(filename);
|
||||
if (it == resources.end()) {
|
||||
std::cerr << "Error: Resource not found: " << filename << std::endl;
|
||||
return {};
|
||||
}
|
||||
|
||||
const ResourceEntry& entry = it->second;
|
||||
if (entry.offset + entry.size > data.size()) {
|
||||
std::cerr << "Error: Invalid resource data: " << filename << std::endl;
|
||||
return {};
|
||||
}
|
||||
|
||||
std::vector<uint8_t> result(data.begin() + entry.offset,
|
||||
data.begin() + entry.offset + entry.size);
|
||||
|
||||
uint32_t checksum = calculateChecksum(result);
|
||||
if (checksum != entry.checksum) {
|
||||
std::cerr << "Warning: Checksum mismatch for resource: " << filename << std::endl;
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
bool ResourcePack::hasResource(const std::string& filename) const {
|
||||
return resources.find(filename) != resources.end();
|
||||
}
|
||||
|
||||
void ResourcePack::clear() {
|
||||
resources.clear();
|
||||
data.clear();
|
||||
loaded = false;
|
||||
}
|
||||
|
||||
size_t ResourcePack::getResourceCount() const {
|
||||
return resources.size();
|
||||
}
|
||||
|
||||
std::vector<std::string> ResourcePack::getResourceList() const {
|
||||
std::vector<std::string> result;
|
||||
result.reserve(resources.size());
|
||||
|
||||
for (const auto& [filename, entry] : resources) {
|
||||
result.push_back(filename);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
Reference in New Issue
Block a user