mirror of
https://github.com/DarkflameUniverse/DarkflameServer.git
synced 2024-11-26 07:27:18 +00:00
8cd7526dae
Add brick by brick conversion commands with 2 parameters to tell the program what to do with the data. Add zlib -> sd0 conversion. Files look good at a glance but should be tested in game to ensure stability. Tests to come.
89 lines
3.1 KiB
C++
89 lines
3.1 KiB
C++
#include <sstream>
|
|
#include <memory>
|
|
|
|
#include "BrickByBrickFix.h"
|
|
#include "Database.h"
|
|
#include "dLogger.h"
|
|
#include "Game.h"
|
|
|
|
//! Forward declarations
|
|
|
|
std::unique_ptr<sql::ResultSet> GetModelsFromDatabase();
|
|
void WriteSd0Magic(char* input);
|
|
|
|
uint32_t BrickByBrickFix::TruncateBrokenBrickByBrickXml() {
|
|
auto modelsToTruncate = GetModelsFromDatabase();
|
|
if (modelsToTruncate->next()) {
|
|
// Decompress with zlib and attempt to convert to xml. If that fails this model is broken.
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
uint32_t BrickByBrickFix::UpdateBrickByBrickModelsToSd0() {
|
|
uint32_t updatedModels = 0;
|
|
auto modelsToUpdate = GetModelsFromDatabase();
|
|
auto previousAutoCommitState = Database::GetAutoCommit();
|
|
Database::SetAutoCommit(false);
|
|
std::unique_ptr<sql::PreparedStatement> insertionStatement(Database::CreatePreppedStmt("UPDATE ugc SET lxfml = ? WHERE id = ?;"));
|
|
while (modelsToUpdate->next()) {
|
|
uint32_t modelId = modelsToUpdate->getInt(1);
|
|
auto oldLxfml = modelsToUpdate->getBlob(2);
|
|
// Check if the stored blob starts with zlib magic (0x78). If it does, convert it to sd0.
|
|
if (oldLxfml->get() == 0x78) {
|
|
Game::logger->Log("BrickByBrickFix", "Updating model %i", modelId);
|
|
|
|
// Get and save size of zlib compressed chunk.
|
|
oldLxfml->seekg(0, std::ios::end);
|
|
uint32_t oldLxfmlSize = static_cast<uint32_t>(oldLxfml->tellg());
|
|
oldLxfml->seekg(0);
|
|
|
|
// Allocate 9 extra bytes. 5 for sd0 magic, 4 for the only zlib compressed size.
|
|
uint32_t oldLxfmlSizeWithHeader = oldLxfmlSize + 9;
|
|
char* sd0ConvertedModel = static_cast<char*>(std::malloc(oldLxfmlSizeWithHeader));
|
|
|
|
WriteSd0Magic(sd0ConvertedModel);
|
|
// zlib compressed chunk size
|
|
sd0ConvertedModel[5] = static_cast<uint8_t>(oldLxfmlSize);
|
|
sd0ConvertedModel[6] = static_cast<uint8_t>(oldLxfmlSize >> 8U);
|
|
sd0ConvertedModel[7] = static_cast<uint8_t>(oldLxfmlSize >> 16U);
|
|
sd0ConvertedModel[8] = static_cast<uint8_t>(oldLxfmlSize >> 24U);
|
|
for (uint32_t i = 9; i < oldLxfmlSizeWithHeader; i++) {
|
|
sd0ConvertedModel[i] = oldLxfml->get();
|
|
}
|
|
|
|
std::string outputString(sd0ConvertedModel, oldLxfmlSizeWithHeader);
|
|
std::istringstream outputStringStream(outputString);
|
|
|
|
insertionStatement->setBlob(1, static_cast<std::istream*>(&outputStringStream));
|
|
insertionStatement->setInt(2, modelId);
|
|
try {
|
|
insertionStatement->executeUpdate();
|
|
Game::logger->Log("BrickByBrickFix", "Updated model %i", modelId);
|
|
updatedModels++;
|
|
} catch (sql::SQLException exception) {
|
|
Game::logger->Log(
|
|
"BrickByBrickFix",
|
|
"Failed to update model %i. This model should be inspected manually to see why."
|
|
"The database error is %s", modelId, exception.what());
|
|
}
|
|
free(sd0ConvertedModel);
|
|
}
|
|
}
|
|
Database::Commit();
|
|
Database::SetAutoCommit(previousAutoCommitState);
|
|
return updatedModels;
|
|
}
|
|
|
|
std::unique_ptr<sql::ResultSet> GetModelsFromDatabase() {
|
|
std::unique_ptr<sql::PreparedStatement> modelsRawDataQuery(Database::CreatePreppedStmt("SELECT id, lxfml FROM ugc;"));
|
|
return std::unique_ptr<sql::ResultSet>(modelsRawDataQuery->executeQuery());
|
|
}
|
|
|
|
void WriteSd0Magic(char* input) {
|
|
input[0] = 's';
|
|
input[1] = 'd';
|
|
input[2] = '0';
|
|
input[3] = 0x01;
|
|
input[4] = 0xFF;
|
|
}
|