mangos-mirror/server
0
0
Fork 0
mirror of https://github.com/mangosfour/server.git synced 2025-12-13 04:37:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[sql/updates/2008_10_23_01_mangos_command.sql sql/updates/2008_10_23_02_mangos_mangos_string.sql] Better 'pdump' commands args checking and related error output.

Browse source
This commit is contained in:
VladimirMangos 2008-10-23 06:07:03 +04:00
parent b10349a737
commit 0c9f783ddc
9 changed files with 203 additions and 70 deletions
Download patch file Download diff file Expand all files Collapse all files

2
sql/mangos.sql
View file

@ -342,6 +342,8 @@ INSERT INTO `command` VALUES
('npc textemote',3,'Syntax: .npc textemote #emoteid\r\n\r\nMake the selected creature to do textemote with an emote of id #emoteid.'), ('npc textemote',3,'Syntax: .npc textemote #emoteid\r\n\r\nMake the selected creature to do textemote with an emote of id #emoteid.'),
('npc whisper',1,'Syntax: .npc whisper #playerguid #text\r\nMake the selected npc whisper #text to #playerguid.'), ('npc whisper',1,'Syntax: .npc whisper #playerguid #text\r\nMake the selected npc whisper #text to #playerguid.'),
('password',0,'Syntax: .password $old_password $new_password $new_password\r\n\r\nChange your account password.'), ('password',0,'Syntax: .password $old_password $new_password $new_password\r\n\r\nChange your account password.'),
('pdump write',3,'Syntax is: pdump write $filename $playerNameOrGUID\r\nWrite character dump with name/guid $playerNameOrGUID to file $filename.'),
('pdump load',3,'Syntax is: pdump load $filename $account [$newname] [$newguid]\r\nLoad character dump from dump file into character list of $account with saved or $newname, with saved (or first free) or $newguid guid.'),
('pinfo',2,'Syntax: .pinfo [$player_name] [rep]\r\n\r\nOutput account information for selected player or player find by $player_name. If \"rep\" parameter provided show reputation information for player.'), ('pinfo',2,'Syntax: .pinfo [$player_name] [rep]\r\n\r\nOutput account information for selected player or player find by $player_name. If \"rep\" parameter provided show reputation information for player.'),
('plimit',3,'Syntax: .plimit [#num|-1|-2|-3|reset|player|moderator|gamemaster|administrator]\r\n\r\nWithout arg show current player amount and security level limitations for login to server, with arg set player linit ($num > 0) or securiti limitation ($num < 0 or security leme name. With `reset` sets player limit to the one in the config file'), ('plimit',3,'Syntax: .plimit [#num|-1|-2|-3|reset|player|moderator|gamemaster|administrator]\r\n\r\nWithout arg show current player amount and security level limitations for login to server, with arg set player linit ($num > 0) or securiti limitation ($num < 0 or security leme name. With `reset` sets player limit to the one in the config file'),
('quest add',3,'Syntax: .quest add #quest_id\r\n\r\nAdd to character quest log quest #quest_id. Quest started from item can\'t be added by this command but correct .additem call provided in command output.'), ('quest add',3,'Syntax: .quest add #quest_id\r\n\r\nAdd to character quest log quest #quest_id. Quest started from item can\'t be added by this command but correct .additem call provided in command output.'),

5
sql/updates/2008_10_23_01_mangos_command.sql Normal file
View file

@ -0,0 +1,5 @@
DELETE FROM command WHERE name IN ('pdump write','pdump load');
INSERT INTO command VALUES
('pdump write',3,'Syntax is: pdump write $filename $playerNameOrGUID\r\nWrite character dump with name/guid $playerNameOrGUID to file $filename.'),
('pdump load',3,'Syntax is: pdump load $filename $account [$newname] [$newguid]\r\nLoad character dump from dump file into character list of $account with saved or $newname, with saved (or first free) or $newguid guid.');

9
sql/updates/2008_10_23_02_mangos_mangos_string.sql Normal file
View file

@ -0,0 +1,9 @@
DELETE FROM mangos_string WHERE entry IN (1112,1113,1114,1115,1116,1117);
INSERT INTO mangos_string VALUES
(1112,'Failed to open file: %s',NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL),
(1113,'Account %s (%u) have max amount allowed characters (client limit)',NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL),
(1114,'Dump file have broken data!',NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL),
(1115,'Invalid character name!',NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL),
(1116,'Invalid character guid!',NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL),
(1117,'Character guid %u in use!',NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL);

4
sql/updates/Makefile.am
View file

@ -101,6 +101,8 @@ pkgdata_DATA = \
2008_10_21_03_mangos_command.sql \ 2008_10_21_03_mangos_command.sql \
2008_10_22_01_mangos_quest_template.sql \ 2008_10_22_01_mangos_quest_template.sql \
2008_10_22_02_mangos_mangos_string.sql \ 2008_10_22_02_mangos_mangos_string.sql \
2008_10_23_01_mangos_command.sql \
2008_10_23_02_mangos_mangos_string.sql \
README README
## Additional files to include when running 'make dist' ## Additional files to include when running 'make dist'
@ -183,4 +185,6 @@ EXTRA_DIST = \
2008_10_21_03_mangos_command.sql \ 2008_10_21_03_mangos_command.sql \
2008_10_22_01_mangos_quest_template.sql \ 2008_10_22_01_mangos_quest_template.sql \
2008_10_22_02_mangos_mangos_string.sql \ 2008_10_22_02_mangos_mangos_string.sql \
2008_10_23_01_mangos_command.sql \
2008_10_23_02_mangos_mangos_string.sql \
README README

1
src/game/AccountMgr.h
View file

@ -48,7 +48,6 @@ class AccountMgr
bool CheckPassword(uint32 accid, std::string passwd); bool CheckPassword(uint32 accid, std::string passwd);
uint32 GetId(std::string username); uint32 GetId(std::string username);
uint32 GetIdByGUID(const uint64 &guid) const;
uint32 GetSecurity(uint32 acc_id); uint32 GetSecurity(uint32 acc_id);
bool GetName(uint32 acc_id, std::string &name); bool GetName(uint32 acc_id, std::string &name);

8
src/game/Language.h
View file

@ -676,7 +676,13 @@ enum MangosStrings
LANG_SKILL_LIST_CONSOLE = 1109, LANG_SKILL_LIST_CONSOLE = 1109,
LANG_CREATURE_LIST_CONSOLE = 1110, LANG_CREATURE_LIST_CONSOLE = 1110,
LANG_GO_LIST_CONSOLE = 1111, LANG_GO_LIST_CONSOLE = 1111,
// Room for more level 3 1112-1199 not used LANG_FILE_OPEN_FAIL = 1112,
LANG_ACCOUNT_CHARACTER_LIST_FULL = 1113,
LANG_DUMP_BROKEN = 1114,
LANG_INVALID_CHARACTER_NAME = 1115,
LANG_INVALID_CHARACTER_GUID = 1116,
LANG_CHARACTER_GUID_IN_USE = 1117,
// Room for more level 3 1118-1199 not used
// FREE IDS 1200-9999 // FREE IDS 1200-9999

89
src/game/Level3.cpp
View file

@ -5346,6 +5346,7 @@ bool ChatHandler::HandleLoadPDumpCommand(const char *args)
SetSentErrorMessage(true); SetSentErrorMessage(true);
return false; return false;
} }
}
if(!accmgr.GetName(account_id,account_name)) if(!accmgr.GetName(account_id,account_name))
{ {
@ -5353,17 +5354,74 @@ bool ChatHandler::HandleLoadPDumpCommand(const char *args)
SetSentErrorMessage(true); SetSentErrorMessage(true);
return false; return false;
} }
char* guid_str = NULL;
char* name_str = strtok(NULL, " ");
std::string name;
if(name_str)
{
name = name_str;
// normalize the name if specified and check if it exists
if(!normalizePlayerName(name))
{
PSendSysMessage(LANG_INVALID_CHARACTER_NAME);
SetSentErrorMessage(true);
return false;
} }
char * name = strtok(NULL, " "); if(!ObjectMgr::IsValidName(name,true))
char * guid_str = name ? strtok(NULL, " ") : NULL; {
PSendSysMessage(LANG_INVALID_CHARACTER_NAME);
SetSentErrorMessage(true);
return false;
}
uint32 guid = guid_str ? atoi(guid_str) : 0; guid_str = strtok(NULL, " ");
}
if(PlayerDumpReader().LoadDump(file, account_id, name ? name : "", guid)) uint32 guid = 0;
if(guid_str)
{
guid = atoi(guid_str);
if(!guid)
{
PSendSysMessage(LANG_INVALID_CHARACTER_GUID);
SetSentErrorMessage(true);
return false;
}
if(objmgr.GetPlayerAccountIdByGUID(guid))
{
PSendSysMessage(LANG_CHARACTER_GUID_IN_USE,guid);
SetSentErrorMessage(true);
return false;
}
}
switch(PlayerDumpReader().LoadDump(file, account_id, name, guid))
{
case DUMP_SUCCESS:
PSendSysMessage(LANG_COMMAND_IMPORT_SUCCESS); PSendSysMessage(LANG_COMMAND_IMPORT_SUCCESS);
else break;
case DUMP_FILE_OPEN_ERROR:
PSendSysMessage(LANG_FILE_OPEN_FAIL,file);
SetSentErrorMessage(true);
return false;
case DUMP_FILE_BROKEN:
PSendSysMessage(LANG_DUMP_BROKEN,file);
SetSentErrorMessage(true);
return false;
case DUMP_TOO_MANY_CHARS:
PSendSysMessage(LANG_ACCOUNT_CHARACTER_LIST_FULL,account_name,account_id);
SetSentErrorMessage(true);
return false;
default:
PSendSysMessage(LANG_COMMAND_IMPORT_FAILED); PSendSysMessage(LANG_COMMAND_IMPORT_FAILED);
SetSentErrorMessage(true);
return false;
}
return true; return true;
} }
@ -5407,10 +5465,27 @@ bool ChatHandler::HandleWritePDumpCommand(const char *args)
if(!guid) if(!guid)
guid = atoi(p2); guid = atoi(p2);
if (PlayerDumpWriter().WriteDump(file, guid)) if(!objmgr.GetPlayerAccountIdByGUID(guid))
{
PSendSysMessage(LANG_PLAYER_NOT_FOUND);
SetSentErrorMessage(true);
return false;
}
switch(PlayerDumpWriter().WriteDump(file, guid))
{
case DUMP_SUCCESS:
PSendSysMessage(LANG_COMMAND_EXPORT_SUCCESS); PSendSysMessage(LANG_COMMAND_EXPORT_SUCCESS);
else break;
case DUMP_FILE_OPEN_ERROR:
PSendSysMessage(LANG_FILE_OPEN_FAIL,file);
SetSentErrorMessage(true);
return false;
default:
PSendSysMessage(LANG_COMMAND_EXPORT_FAILED); PSendSysMessage(LANG_COMMAND_EXPORT_FAILED);
SetSentErrorMessage(true);
return false;
}
return true; return true;
} }

124
src/game/PlayerDump.cpp
View file

@ -24,7 +24,7 @@
#include "ObjectMgr.h" #include "ObjectMgr.h"
// Character Dump tables // Character Dump tables
#define DUMP_TABLE_COUNT 20 #define DUMP_TABLE_COUNT 19
struct DumpTable struct DumpTable
{ {
@ -256,11 +256,8 @@ void StoreGUID(QueryResult *result,uint32 data,uint32 field, std::set<uint32>& g
} }
// Writing - High-level functions // Writing - High-level functions
bool PlayerDumpWriter::DumpTable(std::string& dump, uint32 guid, char const*tableFrom, char const*tableTo, DumpTableType type) void PlayerDumpWriter::DumpTable(std::string& dump, uint32 guid, char const*tableFrom, char const*tableTo, DumpTableType type)
{ {
if (!tableFrom || !tableTo)
return false;
GUIDs const* guids = NULL; GUIDs const* guids = NULL;
char const* fieldname = NULL; char const* fieldname = NULL;
@ -278,7 +275,7 @@ bool PlayerDumpWriter::DumpTable(std::string& dump, uint32 guid, char const*tabl
// for guid set stop if set is empty // for guid set stop if set is empty
if(guids && guids->empty()) if(guids && guids->empty())
return true; // nothing to do return; // nothing to do
// setup for guids case start position // setup for guids case start position
GUIDs::const_iterator guids_itr; GUIDs::const_iterator guids_itr;
@ -296,7 +293,7 @@ bool PlayerDumpWriter::DumpTable(std::string& dump, uint32 guid, char const*tabl
QueryResult *result = CharacterDatabase.PQuery("SELECT * FROM %s WHERE %s", tableFrom, wherestr.c_str()); QueryResult *result = CharacterDatabase.PQuery("SELECT * FROM %s WHERE %s", tableFrom, wherestr.c_str());
if(!result) if(!result)
return false; return;
do do
{ {
@ -326,8 +323,6 @@ bool PlayerDumpWriter::DumpTable(std::string& dump, uint32 guid, char const*tabl
delete result; delete result;
} }
while(guids && guids_itr != guids->end()); // not set case iterate single time, set case iterate for all guids while(guids && guids_itr != guids->end()); // not set case iterate single time, set case iterate for all guids
return true;
} }
std::string PlayerDumpWriter::GetDump(uint32 guid) std::string PlayerDumpWriter::GetDump(uint32 guid)
@ -336,28 +331,29 @@ std::string PlayerDumpWriter::GetDump(uint32 guid)
for(int i = 0; i < DUMP_TABLE_COUNT; i++) for(int i = 0; i < DUMP_TABLE_COUNT; i++)
DumpTable(dump, guid, dumpTables[i].name, dumpTables[i].name, dumpTables[i].type); DumpTable(dump, guid, dumpTables[i].name, dumpTables[i].name, dumpTables[i].type);
// TODO: Add instance/group/gifts.. // TODO: Add instance/group..
// TODO: Add a dump level option to skip some non-important tables // TODO: Add a dump level option to skip some non-important tables
return dump; return dump;
} }
bool PlayerDumpWriter::WriteDump(std::string file, uint32 guid) DumpReturn PlayerDumpWriter::WriteDump(std::string file, uint32 guid)
{ {
FILE *fout = fopen(file.c_str(), "w"); FILE *fout = fopen(file.c_str(), "w");
if (!fout) { sLog.outError("Failed to open file!\r\n"); return false; } if (!fout)
return DUMP_FILE_OPEN_ERROR;
std::string dump = GetDump(guid); std::string dump = GetDump(guid);
fprintf(fout,"%s\n",dump.c_str()); fprintf(fout,"%s\n",dump.c_str());
fclose(fout); fclose(fout);
return true; return DUMP_SUCCESS;
} }
// Reading - High-level functions // Reading - High-level functions
#define ROLLBACK {CharacterDatabase.RollbackTransaction(); fclose(fin); return false;} #define ROLLBACK(DR) {CharacterDatabase.RollbackTransaction(); fclose(fin); return (DR);}
bool PlayerDumpReader::LoadDump(std::string file, uint32 account, std::string name, uint32 guid) DumpReturn PlayerDumpReader::LoadDump(std::string file, uint32 account, std::string name, uint32 guid)
{ {
// check character count // check character count
{ {
@ -370,12 +366,13 @@ bool PlayerDumpReader::LoadDump(std::string file, uint32 account, std::string na
delete result; delete result;
if (charcount >= 10) if (charcount >= 10)
return false; return DUMP_TOO_MANY_CHARS;
} }
} }
FILE *fin = fopen(file.c_str(), "r"); FILE *fin = fopen(file.c_str(), "r");
if(!fin) if(!fin)
return false; return DUMP_FILE_OPEN_ERROR;
QueryResult * result = NULL; QueryResult * result = NULL;
char newguid[20], chraccount[20], newpetid[20], currpetid[20], lastpetid[20]; char newguid[20], chraccount[20], newpetid[20], currpetid[20], lastpetid[20];
@ -431,8 +428,7 @@ bool PlayerDumpReader::LoadDump(std::string file, uint32 account, std::string na
if(!fgets(buf, 32000, fin)) if(!fgets(buf, 32000, fin))
{ {
if(feof(fin)) break; if(feof(fin)) break;
sLog.outError("LoadPlayerDump: File read error!"); ROLLBACK(DUMP_FILE_BROKEN);
ROLLBACK;
} }
std::string line; line.assign(buf); std::string line; line.assign(buf);
@ -446,7 +442,7 @@ bool PlayerDumpReader::LoadDump(std::string file, uint32 account, std::string na
if(tn.empty()) if(tn.empty())
{ {
sLog.outError("LoadPlayerDump: Can't extract table name from line: '%s'!", line.c_str()); sLog.outError("LoadPlayerDump: Can't extract table name from line: '%s'!", line.c_str());
ROLLBACK; ROLLBACK(DUMP_FILE_BROKEN);
} }
DumpTableType type; DumpTableType type;
@ -463,27 +459,33 @@ bool PlayerDumpReader::LoadDump(std::string file, uint32 account, std::string na
if (i == DUMP_TABLE_COUNT) if (i == DUMP_TABLE_COUNT)
{ {
sLog.outError("LoadPlayerDump: Unknown table: '%s'!", tn.c_str()); sLog.outError("LoadPlayerDump: Unknown table: '%s'!", tn.c_str());
ROLLBACK; ROLLBACK(DUMP_FILE_BROKEN);
} }
// change the data to server values // change the data to server values
switch(type) switch(type)
{ {
case DTT_CHAR_TABLE: case DTT_CHAR_TABLE:
if(!changenth(line, 1, newguid)) ROLLBACK; if(!changenth(line, 1, newguid))
ROLLBACK(DUMP_FILE_BROKEN);
break; break;
case DTT_CHARACTER: // character t. case DTT_CHARACTER: // character t.
{ {
if(!changenth(line, 1, newguid)) ROLLBACK; if(!changenth(line, 1, newguid))
ROLLBACK(DUMP_FILE_BROKEN);
// guid, data field:guid, items // guid, data field:guid, items
if(!changenth(line, 2, chraccount)) ROLLBACK; if(!changenth(line, 2, chraccount))
ROLLBACK(DUMP_FILE_BROKEN);
std::string vals = getnth(line, 3); std::string vals = getnth(line, 3);
if(!changetoknth(vals, OBJECT_FIELD_GUID+1, newguid)) ROLLBACK; if(!changetoknth(vals, OBJECT_FIELD_GUID+1, newguid))
ROLLBACK(DUMP_FILE_BROKEN);
for(uint16 field = PLAYER_FIELD_INV_SLOT_HEAD; field < PLAYER_FARSIGHT; field++) for(uint16 field = PLAYER_FIELD_INV_SLOT_HEAD; field < PLAYER_FARSIGHT; field++)
if(!changetokGuid(vals, field+1, items, objmgr.m_hiItemGuid, true)) ROLLBACK; if(!changetokGuid(vals, field+1, items, objmgr.m_hiItemGuid, true))
if(!changenth(line, 3, vals.c_str())) ROLLBACK; ROLLBACK(DUMP_FILE_BROKEN);
if(!changenth(line, 3, vals.c_str()))
ROLLBACK(DUMP_FILE_BROKEN);
if (name == "") if (name == "")
{ {
// check if the original name already exists // check if the original name already exists
@ -495,38 +497,50 @@ bool PlayerDumpReader::LoadDump(std::string file, uint32 account, std::string na
{ {
delete result; delete result;
// rename on login: `at_login` field 30 in raw field list // rename on login: `at_login` field 30 in raw field list
if(!changenth(line, 30, "1")) ROLLBACK; if(!changenth(line, 30, "1"))
ROLLBACK(DUMP_FILE_BROKEN);
} }
} }
else if(!changenth(line, 4, name.c_str())) ROLLBACK; else if(!changenth(line, 4, name.c_str()))
ROLLBACK(DUMP_FILE_BROKEN);
break; break;
} }
case DTT_INVENTORY: // character_inventory t. case DTT_INVENTORY: // character_inventory t.
{ {
if(!changenth(line, 1, newguid)) ROLLBACK; if(!changenth(line, 1, newguid))
ROLLBACK(DUMP_FILE_BROKEN);
// bag, item // bag, item
if(!changeGuid(line, 2, items, objmgr.m_hiItemGuid, true)) ROLLBACK; if(!changeGuid(line, 2, items, objmgr.m_hiItemGuid, true))
if(!changeGuid(line, 4, items, objmgr.m_hiItemGuid)) ROLLBACK; ROLLBACK(DUMP_FILE_BROKEN);
if(!changeGuid(line, 4, items, objmgr.m_hiItemGuid))
ROLLBACK(DUMP_FILE_BROKEN);
break; break;
} }
case DTT_ITEM: // item_instance t. case DTT_ITEM: // item_instance t.
{ {
// item, owner, data field:item, owner guid // item, owner, data field:item, owner guid
if(!changeGuid(line, 1, items, objmgr.m_hiItemGuid)) ROLLBACK; if(!changeGuid(line, 1, items, objmgr.m_hiItemGuid))
if(!changenth(line, 2, newguid)) ROLLBACK; ROLLBACK(DUMP_FILE_BROKEN);
if(!changenth(line, 2, newguid))
ROLLBACK(DUMP_FILE_BROKEN);
std::string vals = getnth(line,3); std::string vals = getnth(line,3);
if(!changetokGuid(vals, OBJECT_FIELD_GUID+1, items, objmgr.m_hiItemGuid)) ROLLBACK; if(!changetokGuid(vals, OBJECT_FIELD_GUID+1, items, objmgr.m_hiItemGuid))
if(!changetoknth(vals, ITEM_FIELD_OWNER+1, newguid)) ROLLBACK; ROLLBACK(DUMP_FILE_BROKEN);
if(!changenth(line, 3, vals.c_str())) ROLLBACK; if(!changetoknth(vals, ITEM_FIELD_OWNER+1, newguid))
ROLLBACK(DUMP_FILE_BROKEN);
if(!changenth(line, 3, vals.c_str()))
ROLLBACK(DUMP_FILE_BROKEN);
break; break;
} }
case DTT_ITEM_GIFT: // character_gift case DTT_ITEM_GIFT: // character_gift
{ {
// guid,item_guid, // guid,item_guid,
if(!changenth(line, 1, newguid)) ROLLBACK; if(!changenth(line, 1, newguid))
if(!changeGuid(line, 2, items, objmgr.m_hiItemGuid)) ROLLBACK; ROLLBACK(DUMP_FILE_BROKEN);
if(!changeGuid(line, 2, items, objmgr.m_hiItemGuid))
ROLLBACK(DUMP_FILE_BROKEN);
break; break;
} }
case DTT_PET: // character_pet t case DTT_PET: // character_pet t
@ -548,8 +562,10 @@ bool PlayerDumpReader::LoadDump(std::string file, uint32 account, std::string na
} }
// item, entry, owner, ... // item, entry, owner, ...
if(!changenth(line, 1, newpetid)) ROLLBACK; if(!changenth(line, 1, newpetid))
if(!changenth(line, 3, newguid)) ROLLBACK; ROLLBACK(DUMP_FILE_BROKEN);
if(!changenth(line, 3, newguid))
ROLLBACK(DUMP_FILE_BROKEN);
break; break;
} }
@ -559,27 +575,34 @@ bool PlayerDumpReader::LoadDump(std::string file, uint32 account, std::string na
// lookup currpetid and match to new inserted pet id // lookup currpetid and match to new inserted pet id
std::map<uint32, uint32> :: const_iterator petids_iter = petids.find(atoi(currpetid)); std::map<uint32, uint32> :: const_iterator petids_iter = petids.find(atoi(currpetid));
if(petids_iter == petids.end()) ROLLBACK; // couldn't find new inserted id if(petids_iter == petids.end()) // couldn't find new inserted id
ROLLBACK(DUMP_FILE_BROKEN);
snprintf(newpetid, 20, "%d", petids_iter->second); snprintf(newpetid, 20, "%d", petids_iter->second);
if(!changenth(line, 1, newpetid)) ROLLBACK; if(!changenth(line, 1, newpetid))
ROLLBACK(DUMP_FILE_BROKEN);
break; break;
} }
case DTT_MAIL: // mail case DTT_MAIL: // mail
{ {
// id,messageType,stationery,sender,receiver // id,messageType,stationery,sender,receiver
if(!changeGuid(line, 1, mails, objmgr.m_mailid)) ROLLBACK; if(!changeGuid(line, 1, mails, objmgr.m_mailid))
if(!changenth(line, 5, newguid)) ROLLBACK; ROLLBACK(DUMP_FILE_BROKEN);
if(!changenth(line, 5, newguid))
ROLLBACK(DUMP_FILE_BROKEN);
break; break;
} }
case DTT_MAIL_ITEM: // mail_items case DTT_MAIL_ITEM: // mail_items
{ {
// mail_id,item_guid,item_template,receiver // mail_id,item_guid,item_template,receiver
if(!changeGuid(line, 1, mails, objmgr.m_mailid)) ROLLBACK; if(!changeGuid(line, 1, mails, objmgr.m_mailid))
if(!changeGuid(line, 2, items, objmgr.m_hiItemGuid)) ROLLBACK; ROLLBACK(DUMP_FILE_BROKEN);
if(!changenth(line, 4, newguid)) ROLLBACK; if(!changeGuid(line, 2, items, objmgr.m_hiItemGuid))
ROLLBACK(DUMP_FILE_BROKEN);
if(!changenth(line, 4, newguid))
ROLLBACK(DUMP_FILE_BROKEN);
break; break;
} }
default: default:
@ -587,7 +610,8 @@ bool PlayerDumpReader::LoadDump(std::string file, uint32 account, std::string na
break; break;
} }
if(!CharacterDatabase.Execute(line.c_str())) ROLLBACK; if(!CharacterDatabase.Execute(line.c_str()))
ROLLBACK(DUMP_FILE_BROKEN);
} }
CharacterDatabase.CommitTransaction(); CharacterDatabase.CommitTransaction();
@ -600,5 +624,5 @@ bool PlayerDumpReader::LoadDump(std::string file, uint32 account, std::string na
fclose(fin); fclose(fin);
return true; return DUMP_SUCCESS;
} }

15
src/game/PlayerDump.h
View file

@ -71,6 +71,15 @@ enum DumpTableType
DTT_ITEM_TEXT, // <- item_text // item_text DTT_ITEM_TEXT, // <- item_text // item_text
}; };
enum DumpReturn
{
DUMP_SUCCESS,
DUMP_FILE_OPEN_ERROR,
DUMP_TOO_MANY_CHARS,
DUMP_UNEXPECTED_END,
DUMP_FILE_BROKEN,
};
class PlayerDump class PlayerDump
{ {
protected: protected:
@ -83,11 +92,11 @@ class PlayerDumpWriter : public PlayerDump
PlayerDumpWriter() {} PlayerDumpWriter() {}
std::string GetDump(uint32 guid); std::string GetDump(uint32 guid);
bool WriteDump(std::string file, uint32 guid); DumpReturn WriteDump(std::string file, uint32 guid);
private: private:
typedef std::set<uint32> GUIDs; typedef std::set<uint32> GUIDs;
bool DumpTable(std::string& dump, uint32 guid, char const*tableFrom, char const*tableTo, DumpTableType type); void DumpTable(std::string& dump, uint32 guid, char const*tableFrom, char const*tableTo, DumpTableType type);
std::string GenerateWhereStr(char const* field, GUIDs const& guids, GUIDs::const_iterator& itr); std::string GenerateWhereStr(char const* field, GUIDs const& guids, GUIDs::const_iterator& itr);
std::string GenerateWhereStr(char const* field, uint32 guid); std::string GenerateWhereStr(char const* field, uint32 guid);
@ -102,7 +111,7 @@ class PlayerDumpReader : public PlayerDump
public: public:
PlayerDumpReader() {} PlayerDumpReader() {}
bool LoadDump(std::string file, uint32 account, std::string name, uint32 guid); DumpReturn LoadDump(std::string file, uint32 account, std::string name, uint32 guid);
}; };
#endif #endif