[svn] * Use ObjectMgr/AccountMgr functions rather than DB queries. Source mangos

--HG--
branch : trunk
This commit is contained in:
KingPin
2008-10-21 19:07:16 -05:00
parent 96d1c9f4d6
commit 389a9e3cea
4 changed files with 24 additions and 28 deletions
+4 -11
View File
@@ -4824,25 +4824,18 @@ bool ChatHandler::HandleBanInfoCommand(const char* args)
return false; return false;
} }
loginDatabase.escape_string(nameOrIP); accountid = objmgr.GetPlayerAccountIdByPlayerName (nameOrIP);
QueryResult *result = CharacterDatabase.PQuery("SELECT account FROM characters WHERE name = '%s'", nameOrIP.c_str()); if (!accountid)
if (!result)
{ {
PSendSysMessage(LANG_BANINFO_NOCHARACTER); PSendSysMessage(LANG_BANINFO_NOCHARACTER);
return true; return true;
} }
fields = result->Fetch();
accountid = fields[0].GetUInt32(); if (!accmgr.GetName (accountid,accountname))
delete result;
result = loginDatabase.PQuery("SELECT username FROM account WHERE id = '%u'", accountid);
if (!result)
{ {
PSendSysMessage(LANG_BANINFO_NOCHARACTER); PSendSysMessage(LANG_BANINFO_NOCHARACTER);
return true; return true;
} }
fields = result->Fetch();
accountname = fields[0].GetCppString();
delete result;
} }
else else
return false; return false;
+13
View File
@@ -1365,6 +1365,19 @@ uint32 ObjectMgr::GetPlayerAccountIdByGUID(const uint64 &guid) const
return 0; return 0;
} }
uint32 ObjectMgr::GetPlayerAccountIdByPlayerName(std::string name) const
{
QueryResult *result = CharacterDatabase.PQuery("SELECT account FROM characters WHERE name = '%s'", name.c_str());
if(result)
{
uint32 acc = (*result)[0].GetUInt32();
delete result;
return acc;
}
return 0;
}
void ObjectMgr::LoadAuctions() void ObjectMgr::LoadAuctions()
{ {
QueryResult *result = CharacterDatabase.Query("SELECT COUNT(*) FROM auctionhouse"); QueryResult *result = CharacterDatabase.Query("SELECT COUNT(*) FROM auctionhouse");
+1
View File
@@ -413,6 +413,7 @@ class ObjectMgr
bool GetPlayerNameByGUID(const uint64 &guid, std::string &name) const; bool GetPlayerNameByGUID(const uint64 &guid, std::string &name) const;
uint32 GetPlayerTeamByGUID(const uint64 &guid) const; uint32 GetPlayerTeamByGUID(const uint64 &guid) const;
uint32 GetPlayerAccountIdByGUID(const uint64 &guid) const; uint32 GetPlayerAccountIdByGUID(const uint64 &guid) const;
uint32 GetPlayerAccountIdByPlayerName(std::string name) const;
uint32 GetNearestTaxiNode( float x, float y, float z, uint32 mapid ); uint32 GetNearestTaxiNode( float x, float y, float z, uint32 mapid );
void GetTaxiPath( uint32 source, uint32 destination, uint32 &path, uint32 &cost); void GetTaxiPath( uint32 source, uint32 destination, uint32 &path, uint32 &cost);
+5 -16
View File
@@ -36,6 +36,7 @@
#include "SkillExtraItems.h" #include "SkillExtraItems.h"
#include "SkillDiscovery.h" #include "SkillDiscovery.h"
#include "World.h" #include "World.h"
#include "AccountMgr.h"
#include "ObjectMgr.h" #include "ObjectMgr.h"
#include "SpellMgr.h" #include "SpellMgr.h"
#include "Chat.h" #include "Chat.h"
@@ -2337,33 +2338,21 @@ bool World::RemoveBanAccount(std::string type, std::string nameOrIP)
uint32 account=0; uint32 account=0;
if(type == "account") if(type == "account")
{ {
//NO SQL injection as name is escaped if (!AccountMgr::normilizeString (nameOrIP))
loginDatabase.escape_string(nameOrIP);
QueryResult *resultAccounts = loginDatabase.PQuery("SELECT id FROM account WHERE username = '%s'",nameOrIP.c_str());
if(!resultAccounts)
return false; return false;
Field* fieldsAccount = resultAccounts->Fetch();
account = fieldsAccount->GetUInt32();
delete resultAccounts; account = accmgr.GetId (nameOrIP);
} }
else if(type == "character") else if(type == "character")
{ {
if(!normalizePlayerName(nameOrIP)) if(!normalizePlayerName(nameOrIP))
return false; return false;
//NO SQL injection as name is escaped account = objmgr.GetPlayerAccountIdByPlayerName (nameOrIP);
loginDatabase.escape_string(nameOrIP);
QueryResult *resultAccounts = CharacterDatabase.PQuery("SELECT account FROM characters WHERE name = '%s'",nameOrIP.c_str());
if(!resultAccounts)
return false;
Field* fieldsAccount = resultAccounts->Fetch();
account = fieldsAccount->GetUInt32();
delete resultAccounts;
} }
if(!account) if(!account)
return false; return false;
//NO SQL injection as account is uint32 //NO SQL injection as account is uint32
loginDatabase.PExecute("UPDATE account_banned SET active = '0' WHERE id = '%u'",account); loginDatabase.PExecute("UPDATE account_banned SET active = '0' WHERE id = '%u'",account);
} }