/
white.sheikh
/
mangoszero-server
Обзор
Документация
Войти
/
white.sheikh
/
mangoszero-server
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
src/game/Tools/PlayerDump.cpp
865 строк
27 KB
Antz
refactor emails and remove item_text in line with other cores
03 фев 2023, 02:01
Не верифицирован
03 фев 2023, 02:01
2ef8c52
Код
Авторство
О чём код?
/** * MaNGOS is a full featured server for World of Warcraft, supporting * the following clients: 1.12.x, 2.4.3, 3.3.5a, 4.3.4a and 5.4.8 * * Copyright (C) 2005-2023 MaNGOS <https://getmangos.eu> * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA * * World of Warcraft, and all World of Warcraft or Warcraft art, images, * and lore are copyrighted by Blizzard Entertainment, Inc. */ #include "Common.h" #include "PlayerDump.h" #include "Database/DatabaseEnv.h" #include "SQLStorages.h" #include "UpdateFields.h" #include "ObjectMgr.h" #include "AccountMgr.h" // Character Dump tables struct DumpTable { char const* name; DumpTableType type; // helpers bool isValid() const { return name != NULL; } }; static DumpTable dumpTables[] = { { "characters", DTT_CHARACTER }, // -> guid, must be first for name check { "character_action", DTT_CHAR_TABLE }, { "character_aura", DTT_CHAR_TABLE }, { "character_homebind", DTT_CHAR_TABLE }, { "character_honor_cp", DTT_CHAR_TABLE }, { "character_inventory", DTT_INVENTORY }, // -> item guids { "character_queststatus", DTT_CHAR_TABLE }, { "character_pet", DTT_PET }, // -> pet number { "character_reputation", DTT_CHAR_TABLE }, { "character_skills", DTT_CHAR_TABLE }, { "character_spell", DTT_CHAR_TABLE }, { "character_spell_cooldown", DTT_CHAR_TABLE }, { "character_ticket", DTT_CHAR_TABLE }, { "mail", DTT_MAIL }, // -> mail guids { "mail_items", DTT_MAIL_ITEM }, // -> item guids <- mail guids { "pet_aura", DTT_PET_TABLE }, // <- pet number { "pet_spell", DTT_PET_TABLE }, // <- pet number { "pet_spell_cooldown", DTT_PET_TABLE }, // <- pet number { "character_gifts", DTT_ITEM_GIFT }, // <- item guids { "item_instance", DTT_ITEM }, // <- item guids { "item_loot", DTT_ITEM_LOOT }, // <- item guids { NULL, DTT_CHAR_TABLE }, // end marker }; // Low level functions static bool findtoknth(std::string& str, int n, std::string::size_type& s, std::string::size_type& e) { int i; s = e = 0; std::string::size_type size = str.size(); for (i = 1; s < size && i < n; ++s) if (str[s] == ' ') { ++i; } if (i < n) { return false; } e = str.find(' ', s); return e != std::string::npos; } std::string gettoknth(std::string& str, int n) { std::string::size_type s = 0, e = 0; if (!findtoknth(str, n, s, e)) { return ""; } return str.substr(s, e - s); } bool findnth(std::string& str, int n, std::string::size_type& s, std::string::size_type& e) { s = str.find("VALUES ('") + 9; if (s == std::string::npos) { return false; } do { e = str.find("'", s); if (e == std::string::npos) { return false; } } while (str[e - 1] == '\\'); for (int i = 1; i < n; ++i) { do { s = e + 4; e = str.find("'", s); if (e == std::string::npos) { return false; } } while (str[e - 1] == '\\'); } return true; } std::string gettablename(std::string& str) { std::string::size_type s = 13; std::string::size_type e = str.find('`', s); if (e == std::string::npos) { return ""; } return str.substr(s, e - s); } bool changenth(std::string& str, int n, const char* with, bool insert = false, bool nonzero = false) { std::string::size_type s, e; if (!findnth(str, n, s, e)) { return false; } if (nonzero && str.substr(s, e - s) == "0") { return true; // not an error } if (!insert) { str.replace(s, e - s, with); } else { str.insert(s, with); } return true; } std::string getnth(std::string& str, int n) { std::string::size_type s, e; if (!findnth(str, n, s, e)) { return ""; } return str.substr(s, e - s); } bool changetoknth(std::string& str, int n, const char* with, bool insert = false, bool nonzero = false) { std::string::size_type s = 0, e = 0; if (!findtoknth(str, n, s, e)) { return false; } if (nonzero && str.substr(s, e - s) == "0") { return true; // not an error } if (!insert) { str.replace(s, e - s, with); } else { str.insert(s, with); } return true; } uint32 registerNewGuid(uint32 oldGuid, std::map<uint32, uint32>& guidMap, uint32 hiGuid) { std::map<uint32, uint32>::const_iterator itr = guidMap.find(oldGuid); if (itr != guidMap.end()) { return itr->second; } uint32 newguid = hiGuid + guidMap.size(); guidMap[oldGuid] = newguid; return newguid; } bool changeGuid(std::string& str, int n, std::map<uint32, uint32>& guidMap, uint32 hiGuid, bool nonzero = false) { char chritem[20]; uint32 oldGuid = atoi(getnth(str, n).c_str()); if (nonzero && oldGuid == 0) { return true; // not an error } uint32 newGuid = registerNewGuid(oldGuid, guidMap, hiGuid); snprintf(chritem, 20, "%u", newGuid); return changenth(str, n, chritem, false, nonzero); } bool changetokGuid(std::string& str, int n, std::map<uint32, uint32>& guidMap, uint32 hiGuid, bool nonzero = false) { char chritem[20]; uint32 oldGuid = atoi(gettoknth(str, n).c_str()); if (nonzero && oldGuid == 0) { return true; // not an error } uint32 newGuid = registerNewGuid(oldGuid, guidMap, hiGuid); snprintf(chritem, 20, "%u", newGuid); return changetoknth(str, n, chritem, false, nonzero); } std::string CreateDumpString(char const* tableName, char const* tableColumnNamesAsChars, QueryResult* result) { if (!tableName || !result) { return ""; } std::ostringstream ss; ss << "INSERT INTO `" << tableName << "` ("<< tableColumnNamesAsChars <<") VALUES ("; Field* fields = result->Fetch(); for (uint32 i = 0; i < result->GetFieldCount(); ++i) { if (i != 0) { ss << ", "; } if (fields[i].IsNULL()) { ss << "NULL"; } else { std::string s = fields[i].GetCppString(); CharacterDatabase.escape_string(s); ss << "'" << s << "'"; } } ss << ");"; return ss.str(); } std::string PlayerDumpWriter::GenerateWhereStr(char const* field, uint32 guid) { std::ostringstream wherestr; wherestr << field << " = '" << guid << "'"; return wherestr.str(); } std::string PlayerDumpWriter::GenerateWhereStr(char const* field, GUIDs const& guids, GUIDs::const_iterator& itr) { std::ostringstream wherestr; wherestr << field << " IN ('"; for (; itr != guids.end(); ++itr) { wherestr << *itr; if (wherestr.str().size() > MAX_QUERY_LEN - 50) // near to max query { ++itr; break; } GUIDs::const_iterator itr2 = itr; if (++itr2 != guids.end()) { wherestr << "','"; } } wherestr << "')"; return wherestr.str(); } void StoreGUID(QueryResult* result, uint32 field, std::set<uint32>& guids) { Field* fields = result->Fetch(); uint32 guid = fields[field].GetUInt32(); if (guid) { guids.insert(guid); } } void StoreGUID(QueryResult* result, uint32 data, uint32 field, std::set<uint32>& guids) { Field* fields = result->Fetch(); std::string dataStr = fields[data].GetCppString(); uint32 guid = atoi(gettoknth(dataStr, field).c_str()); if (guid) { guids.insert(guid); } } // Writing - High-level functions void PlayerDumpWriter::DumpTableContent(std::string& dump, uint32 guid, char const* tableFrom, char const* tableTo, DumpTableType type) { GUIDs const* guids = NULL; char const* fieldname; switch (type) { case DTT_ITEM: fieldname = "guid"; guids = &items; break; case DTT_ITEM_GIFT: fieldname = "item_guid"; guids = &items; break; case DTT_ITEM_LOOT: fieldname = "guid"; guids = &items; break; case DTT_PET: fieldname = "owner"; break; case DTT_PET_TABLE: fieldname = "guid"; guids = &pets; break; case DTT_MAIL: fieldname = "receiver"; break; case DTT_MAIL_ITEM: fieldname = "mail_id"; guids = &mails; break; default: fieldname = "guid"; break; } // for guid set stop if set is empty if (guids && guids->empty()) { return; // nothing to do } // setup for guids case start position GUIDs::const_iterator guids_itr; if (guids) { guids_itr = guids->begin(); } do { std::string wherestr; if (guids) // set case, get next guids string { wherestr = GenerateWhereStr(fieldname, *guids, guids_itr); } else // not set case, get single guid string { wherestr = GenerateWhereStr(fieldname, guid); } //fetch table columns std::string tableColumnNamesStr = ""; QueryNamedResult* resNames = CharacterDatabase.PQueryNamed("SELECT * FROM `%s` LIMIT 1", tableFrom); if (!resNames) { return; } // There will be a result since if not teh code does not hit lines before... so no check needed QueryFieldNames const& namesMap = resNames->GetFieldNames(); for (QueryFieldNames::const_iterator itr = namesMap.begin(); itr != namesMap.end(); ++itr) { tableColumnNamesStr += "`" + *itr +"`,"; } // remove last character of tableColumnNamesStr = "," tableColumnNamesStr.pop_back(); namesMap.empty(); // fetch results of the table QueryResult* result = CharacterDatabase.PQuery("SELECT %s FROM `%s` WHERE %s", tableColumnNamesStr.c_str(), tableFrom, wherestr.c_str()); if (!result) { return; } do { // collect guids switch (type) { case DTT_INVENTORY: StoreGUID(result, 3, items); break; // item guid collection case DTT_PET: StoreGUID(result, 0, pets); break; // pet petnumber collection (character_pet.id) case DTT_MAIL: StoreGUID(result, 0, mails); // mail id collection (mail.id) case DTT_MAIL_ITEM: StoreGUID(result, 1, items); break; // item guid collection (mail_items.item_guid) default: break; } dump += CreateDumpString(tableTo, tableColumnNamesStr.c_str(), result); dump += "\n"; } while (result->NextRow()); delete result; } while (guids && guids_itr != guids->end()); // not set case iterate single time, set case iterate for all guids } std::string PlayerDumpWriter::GetDump(uint32 guid) { std::string dump; dump += "IMPORTANT NOTE: This sql queries not created for apply directly, use '.pdump load' command in console or client chat instead.\n"; dump += "IMPORTANT NOTE: NOT APPLY ITS DIRECTLY to character DB or you will DAMAGE and CORRUPT character DB\n"; // revision check guard // Check the revision of character DB which will be the first line of the whole version/structure set QueryResult* result = CharacterDatabase.Query("SELECT `version`, `structure`, `description`, `comment` FROM `db_version` ORDER BY `version` DESC, `structure` DESC, `content` ASC LIMIT 1"); if (result) { Field* fields = result->Fetch(); dump += "DUMPED_WITH:"+std::to_string(fields[0].GetInt16()) + "." + std::to_string(fields[1].GetInt16()) + ".X " +" CHAR. DB VERSION ( " + fields[2].GetCppString() + " / " + fields[3].GetCppString() + ")\n\n" ; delete result; } else { sLog.outError("Character DB not have 'db_version' table"); } for (DumpTable* itr = &dumpTables[0]; itr->isValid(); ++itr) { DumpTableContent(dump, guid, itr->name, itr->name, itr->type); } // TODO: Add instance/group.. // TODO: Add a dump level option to skip some non-important tables return dump; } DumpReturn PlayerDumpWriter::WriteDump(const std::string& file, uint32 guid) { FILE* fout = fopen(file.c_str(), "w"); if (!fout) { return DUMP_FILE_OPEN_ERROR; } std::string dump = GetDump(guid); fprintf(fout, "%s\n", dump.c_str()); fclose(fout); return DUMP_SUCCESS; } // Reading - High-level functions #define ROLLBACK(DR) {CharacterDatabase.RollbackTransaction(); fclose(fin); return (DR);} DumpReturn PlayerDumpReader::LoadDump(const std::string& file, uint32 account, std::string name, uint32 guid) { // check character count uint32 charcount = sAccountMgr.GetCharactersCount(account); if (charcount >= 10) { return DUMP_TOO_MANY_CHARS; } FILE* fin = fopen(file.c_str(), "r"); if (!fin) { return DUMP_FILE_OPEN_ERROR; } QueryResult* result; char newguid[20], chraccount[20], newpetid[20], currpetid[20], lastpetid[20]; // make sure the same guid doesn't already exist and is safe to use bool incHighest = true; if (guid != 0 && guid < sObjectMgr.m_CharGuids.GetNextAfterMaxUsed()) { result = CharacterDatabase.PQuery("SELECT * FROM `characters` WHERE `guid` = '%u'", guid); if (result) { guid = sObjectMgr.m_CharGuids.GetNextAfterMaxUsed(); delete result; } else { incHighest = false; } } else { guid = sObjectMgr.m_CharGuids.GetNextAfterMaxUsed(); } // normalize the name if specified and check if it exists if (!normalizePlayerName(name)) { name.clear(); } if (ObjectMgr::CheckPlayerName(name, true) == CHAR_NAME_SUCCESS) { CharacterDatabase.escape_string(name); // for safe, we use name only for sql quearies anyway result = CharacterDatabase.PQuery("SELECT * FROM `characters` WHERE `name` = '%s'", name.c_str()); if (result) { name.clear(); // use the one from the dump delete result; } } else { name.clear(); } // name encoded or empty snprintf(newguid, 20, "%u", guid); snprintf(chraccount, 20, "%u", account); snprintf(newpetid, 20, "%u", sObjectMgr.GeneratePetNumber()); snprintf(lastpetid, 20, "%s", ""); std::map<uint32, uint32> items; std::map<uint32, uint32> mails; std::map<uint32, uint32> itemTexts; char buf[32000] = ""; typedef std::map<uint32, uint32> PetIds; // old->new petid relation typedef PetIds::value_type PetIdsPair; PetIds petids; CharacterDatabase.BeginTransaction(); while (!feof(fin)) { if (!fgets(buf, 32000, fin)) { if (feof(fin)) { break; } ROLLBACK(DUMP_FILE_BROKEN); } std::string line; line.assign(buf); // skip empty strings size_t nw_pos = line.find_first_not_of(" \t\n\r\7"); if (nw_pos == std::string::npos) { continue; } // skip NOTE if (line.substr(nw_pos, 15) == "IMPORTANT NOTE:") { continue; } // Check db version corresp. std::string dbVersionLinePrefix = line.substr(nw_pos, 12); if (dbVersionLinePrefix =="DUMPED_WITH:") { QueryResult* result = CharacterDatabase.Query("SELECT `version`, `structure`, `description`, `comment` FROM `db_version` ORDER BY `version` DESC, `structure` DESC, `content` ASC LIMIT 1"); if (result) { Field* fields = result->Fetch(); // Only version / structure is needed std::string dbversion = std::to_string(fields[0].GetInt16()) + "." + std::to_string(fields[1].GetInt16())+".X"; size_t dbversionLen = dbversion.size(); std::string dbversionInDumpFile = line.substr(nw_pos+12, dbversionLen); delete result; if (dbversionInDumpFile != dbversion) { sLog.outError("LoadPlayerDump: Cannot load player dump - file version is %s, DB needs %s", dbversionInDumpFile.c_str(), dbversion.c_str()); ROLLBACK(DUMP_DB_VERSION_MISMATCH); } else { continue; } } } // determine table name and load type std::string tn = gettablename(line); if (tn.empty()) { sLog.outError("LoadPlayerDump: Can't extract table name from line: '%s'!", line.c_str()); ROLLBACK(DUMP_FILE_BROKEN); } DumpTableType type = DTT_CHARACTER; // Fixed: Using uninitialized memory 'type' DumpTable* dTable = &dumpTables[0]; for (; dTable->isValid(); ++dTable) { if (tn == dTable->name) { type = dTable->type; break; } } if (!dTable->isValid()) { sLog.outError("LoadPlayerDump: Unknown table: '%s'!", tn.c_str()); ROLLBACK(DUMP_FILE_BROKEN); } // change the data to server values switch (type) { case DTT_CHAR_TABLE: if (!changenth(line, 1, newguid)) // character_*.guid update { ROLLBACK(DUMP_FILE_BROKEN); } break; case DTT_CHARACTER: { if (!changenth(line, 1, newguid)) // characters.guid update { ROLLBACK(DUMP_FILE_BROKEN); } if (!changenth(line, 2, chraccount)) // characters.account update { ROLLBACK(DUMP_FILE_BROKEN); } if (name.empty()) { // check if the original name already exists name = getnth(line, 3); // characters.name CharacterDatabase.escape_string(name); result = CharacterDatabase.PQuery("SELECT * FROM `characters` WHERE `name` = '%s'", name.c_str()); if (result) { delete result; if (!changenth(line, 35, "1")) // characters.at_login set to "rename on login" { ROLLBACK(DUMP_FILE_BROKEN); } } } else { if (!changenth(line, 3, name.c_str())) // characters.name update { ROLLBACK(DUMP_FILE_BROKEN); } } break; } case DTT_INVENTORY: { if (!changenth(line, 1, newguid)) // character_inventory.guid update { ROLLBACK(DUMP_FILE_BROKEN); } if (!changeGuid(line, 2, items, sObjectMgr.m_ItemGuids.GetNextAfterMaxUsed(), true)) { ROLLBACK(DUMP_FILE_BROKEN); // character_inventory.bag update } if (!changeGuid(line, 4, items, sObjectMgr.m_ItemGuids.GetNextAfterMaxUsed())) { ROLLBACK(DUMP_FILE_BROKEN); // character_inventory.item update } break; } case DTT_ITEM: { // item, owner, data field:item, owner guid if (!changeGuid(line, 1, items, sObjectMgr.m_ItemGuids.GetNextAfterMaxUsed())) { ROLLBACK(DUMP_FILE_BROKEN); // item_instance.guid update } if (!changenth(line, 2, newguid)) // item_instance.owner_guid update { ROLLBACK(DUMP_FILE_BROKEN); } std::string vals = getnth(line, 3); // item_instance.data get if (!changetokGuid(vals, OBJECT_FIELD_GUID + 1, items, sObjectMgr.m_ItemGuids.GetNextAfterMaxUsed())) { ROLLBACK(DUMP_FILE_BROKEN); // item_instance.data.OBJECT_FIELD_GUID update } if (!changetoknth(vals, ITEM_FIELD_OWNER + 1, newguid)) { ROLLBACK(DUMP_FILE_BROKEN); // item_instance.data.ITEM_FIELD_OWNER update } if (!changenth(line, 3, vals.c_str())) // item_instance.data update { ROLLBACK(DUMP_FILE_BROKEN); } break; } case DTT_ITEM_GIFT: { if (!changenth(line, 1, newguid)) // character_gifts.guid update { ROLLBACK(DUMP_FILE_BROKEN); } if (!changeGuid(line, 2, items, sObjectMgr.m_ItemGuids.GetNextAfterMaxUsed())) { ROLLBACK(DUMP_FILE_BROKEN); // character_gifts.item_guid update } break; } case DTT_ITEM_LOOT: { // item, owner if (!changeGuid(line, 1, items, sObjectMgr.m_ItemGuids.GetNextAfterMaxUsed())) { ROLLBACK(DUMP_FILE_BROKEN); // item_loot.guid update } if (!changenth(line, 2, newguid)) // item_Loot.owner_guid update { ROLLBACK(DUMP_FILE_BROKEN); } break; } case DTT_PET: { // store a map of old pet id to new inserted pet id for use by type 5 tables snprintf(currpetid, 20, "%s", getnth(line, 1).c_str()); if (strlen(lastpetid) == 0) { snprintf(lastpetid, 20, "%s", currpetid); } if (strcmp(lastpetid, currpetid) != 0) { snprintf(newpetid, 20, "%d", sObjectMgr.GeneratePetNumber()); snprintf(lastpetid, 20, "%s", currpetid); } std::map<uint32, uint32> :: const_iterator petids_iter = petids.find(atoi(currpetid)); if (petids_iter == petids.end()) { petids.insert(PetIdsPair(atoi(currpetid), atoi(newpetid))); } if (!changenth(line, 1, newpetid)) // character_pet.id update { ROLLBACK(DUMP_FILE_BROKEN); } if (!changenth(line, 3, newguid)) // character_pet.owner update { ROLLBACK(DUMP_FILE_BROKEN); } break; } case DTT_PET_TABLE: // pet_aura, pet_spell, pet_spell_cooldown { snprintf(currpetid, 20, "%s", getnth(line, 1).c_str()); // lookup currpetid and match to new inserted pet id std::map<uint32, uint32> :: const_iterator petids_iter = petids.find(atoi(currpetid)); if (petids_iter == petids.end()) // couldn't find new inserted id { ROLLBACK(DUMP_FILE_BROKEN); } snprintf(newpetid, 20, "%d", petids_iter->second); if (!changenth(line, 1, newpetid)) // pet_*.guid -> petid in fact { ROLLBACK(DUMP_FILE_BROKEN); } break; } case DTT_MAIL: // mail { if (!changeGuid(line, 1, mails, sObjectMgr.m_MailIds.GetNextAfterMaxUsed())) { ROLLBACK(DUMP_FILE_BROKEN); // mail.id update } if (!changenth(line, 6, newguid)) // mail.receiver update { ROLLBACK(DUMP_FILE_BROKEN); } break; } case DTT_MAIL_ITEM: // mail_items { if (!changeGuid(line, 1, mails, sObjectMgr.m_MailIds.GetNextAfterMaxUsed())) { ROLLBACK(DUMP_FILE_BROKEN); // mail_items.id } if (!changeGuid(line, 2, items, sObjectMgr.m_ItemGuids.GetNextAfterMaxUsed())) { ROLLBACK(DUMP_FILE_BROKEN); // mail_items.item_guid } if (!changenth(line, 4, newguid)) // mail_items.receiver { ROLLBACK(DUMP_FILE_BROKEN); } break; } default: sLog.outError("Unknown dump table type: %u", type); break; } if (!CharacterDatabase.Execute(line.c_str())) { ROLLBACK(DUMP_FILE_BROKEN); } } CharacterDatabase.CommitTransaction(); // FIXME: current code with post-updating guids not safe for future per-map threads sObjectMgr.m_ItemGuids.Set(sObjectMgr.m_ItemGuids.GetNextAfterMaxUsed() + items.size()); sObjectMgr.m_MailIds.Set(sObjectMgr.m_MailIds.GetNextAfterMaxUsed() + mails.size()); if (incHighest) { sObjectMgr.m_CharGuids.Set(sObjectMgr.m_CharGuids.GetNextAfterMaxUsed() + 1); } fclose(fin); return DUMP_SUCCESS; }