LEGO_Universe/DarkflameServer
1
0
Fork 0
mirror of https://github.com/DarkflameUniverse/DarkflameServer synced 2024-08-30 18:43:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Calculate FDB checksum in chunks

Browse Source
This commit is contained in:
Jett 2021-12-12 15:14:04 +00:00
parent 49c1cb3aff
commit 830cf22beb
1 changed files with 29 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
dWorldServer/WorldServer.cpp
View File

@ -89,6 +89,7 @@ struct tempSessionInfo {
std::map<std::string, tempSessionInfo> m_PendingUsers; std::map<std::string, tempSessionInfo> m_PendingUsers;
int instanceID = 0; int instanceID = 0;
int g_CloneID = 0; int g_CloneID = 0;
std::string fdbChecksum = "";
int main(int argc, char** argv) { int main(int argc, char** argv) {
Diagnostics::SetProcessName("World"); Diagnostics::SetProcessName("World");
@ -234,6 +235,30 @@ int main(int argc, char** argv) {
Game::physicsWorld = &dpWorld::Instance(); //just in case some old code references it Game::physicsWorld = &dpWorld::Instance(); //just in case some old code references it
dZoneManager::Instance()->Initialize(LWOZONEID(zoneID, instanceID, cloneID)); dZoneManager::Instance()->Initialize(LWOZONEID(zoneID, instanceID, cloneID));
g_CloneID = cloneID; g_CloneID = cloneID;
// pre calculate the FDB checksum
if (bool(std::stoi(Game::config->GetValue("check_fdb")))) {
std::ifstream fileStream;
fileStream.open ("res/CDServer.fdb", std::ios::binary | std::ios::in);
const int bufferSize = 1024;
MD5* md5 = new MD5();
while (!fileStream.eof()) {
char * fileStreamBuffer = new char[bufferSize];
fileStream.read(fileStreamBuffer, bufferSize);
std::streamsize size = ((fileStream) ? bufferSize : fileStream.gcount());
md5->update(fileStreamBuffer, size);
}
const char* nullTerminateBuffer = "\0";
md5->update(nullTerminateBuffer, 1); // null terminate the data
md5->finalize();
fdbChecksum = md5->hexdigest();
delete md5;
Game::logger->Log("WorldServer", "FDB Checksum calculated as: %s\n", fdbChecksum.c_str());
}
} }
while (true) { while (true) {
@ -838,25 +863,11 @@ void HandlePacket(Packet* packet) {
case MSG_WORLD_CLIENT_VALIDATION: { case MSG_WORLD_CLIENT_VALIDATION: {
std::string username = PacketUtils::ReadString(0x08, packet, true); std::string username = PacketUtils::ReadString(0x08, packet, true);
std::string sessionKey = PacketUtils::ReadString(74, packet, true); std::string sessionKey = PacketUtils::ReadString(74, packet, true);
std::string fdbChecksum = PacketUtils::ReadString(packet->length - 33, packet, false); std::string theirFdbChecksum = PacketUtils::ReadString(packet->length - 33, packet, false);
if (bool(std::stoi(Game::config->GetValue("check_fdb")))) { if (bool(std::stoi(Game::config->GetValue("check_fdb"))) && fdbChecksum != "") { // if fdbChecksum is empty, likely means we are a character server.
std::ifstream fileStream; Game::logger->Log("WorldServer", "Got client checksum %s and we have server checksum %s. \n", theirFdbChecksum.c_str(), fdbChecksum.c_str());
fileStream.open ("res/CDServer.fdb", std::ios::binary | std::ios::in); if (theirFdbChecksum != fdbChecksum) {
fileStream.seekg (0, std::ios::end);
uint64_t fileStreamLength = fileStream.tellg();
fileStream.seekg (0, std::ios::beg);
char * fileStreamData = new char[fileStreamLength + 1];
fileStream.read(fileStreamData, fileStreamLength);
*(fileStreamData + (fileStreamLength + 1)) = 0x00; // null terminate the string
MD5 md5 = MD5(fileStreamData, fileStreamLength + 1);
std::string ourFdbChecksum = md5.hexdigest();
Game::logger->Log("WorldServer", "Got client checksum %s and we have server checksum %s. \n", fdbChecksum.c_str(), ourFdbChecksum.c_str());
if (fdbChecksum != ourFdbChecksum) {
Game::logger->Log("WorldServer", "Client checksum does not match server checksum.\n"); Game::logger->Log("WorldServer", "Client checksum does not match server checksum.\n");
Game::server->Disconnect(packet->systemAddress, SERVER_DISCON_KICK); Game::server->Disconnect(packet->systemAddress, SERVER_DISCON_KICK);
return; return;