forked from metin2/server
713 lines
16 KiB
C++
713 lines
16 KiB
C++
#include "stdafx.h"
|
|
#include <sstream>
|
|
#include <common/length.h>
|
|
|
|
#include "db.h"
|
|
|
|
#include "config.h"
|
|
#include "desc_client.h"
|
|
#include "desc_manager.h"
|
|
#include "char.h"
|
|
#include "char_manager.h"
|
|
#include "item.h"
|
|
#include "item_manager.h"
|
|
#include "p2p.h"
|
|
#include "matrix_card.h"
|
|
#include "log.h"
|
|
#include "login_data.h"
|
|
#include "locale_service.h"
|
|
#include "spam.h"
|
|
|
|
extern std::string g_stBlockDate;
|
|
|
|
DBManager::DBManager() : m_bIsConnect(false)
|
|
{
|
|
}
|
|
|
|
DBManager::~DBManager()
|
|
{
|
|
}
|
|
|
|
bool DBManager::Connect(const char * host, const int port, const char * user, const char * pwd, const char * db)
|
|
{
|
|
if (m_sql.Setup(host, user, pwd, db, g_stLocale.c_str(), false, port))
|
|
m_bIsConnect = true;
|
|
|
|
if (!m_sql_direct.Setup(host, user, pwd, db, g_stLocale.c_str(), true, port))
|
|
SPDLOG_ERROR("cannot open direct sql connection to host {}", host);
|
|
|
|
return m_bIsConnect;
|
|
}
|
|
|
|
void DBManager::Query(const char * c_pszFormat, ...)
|
|
{
|
|
char szQuery[4096];
|
|
va_list args;
|
|
|
|
va_start(args, c_pszFormat);
|
|
vsnprintf(szQuery, sizeof(szQuery), c_pszFormat, args);
|
|
va_end(args);
|
|
|
|
m_sql.AsyncQuery(szQuery);
|
|
}
|
|
|
|
SQLMsg * DBManager::DirectQuery(const char * c_pszFormat, ...)
|
|
{
|
|
char szQuery[4096];
|
|
va_list args;
|
|
|
|
va_start(args, c_pszFormat);
|
|
vsnprintf(szQuery, sizeof(szQuery), c_pszFormat, args);
|
|
va_end(args);
|
|
|
|
return m_sql_direct.DirectQuery(szQuery);
|
|
}
|
|
|
|
bool DBManager::IsConnected()
|
|
{
|
|
return m_bIsConnect;
|
|
}
|
|
|
|
void DBManager::ReturnQuery(int iType, DWORD dwIdent, void * pvData, const char * c_pszFormat, ...)
|
|
{
|
|
char szQuery[4096];
|
|
va_list args;
|
|
|
|
va_start(args, c_pszFormat);
|
|
vsnprintf(szQuery, sizeof(szQuery), c_pszFormat, args);
|
|
va_end(args);
|
|
|
|
CReturnQueryInfo * p = M2_NEW CReturnQueryInfo;
|
|
|
|
p->iQueryType = QUERY_TYPE_RETURN;
|
|
p->iType = iType;
|
|
p->dwIdent = dwIdent;
|
|
p->pvData = pvData;
|
|
|
|
m_sql.ReturnQuery(szQuery, p);
|
|
}
|
|
|
|
SQLMsg * DBManager::PopResult()
|
|
{
|
|
SQLMsg * p;
|
|
|
|
if (m_sql.PopResult(&p))
|
|
return p;
|
|
|
|
return NULL;
|
|
}
|
|
|
|
void DBManager::Process()
|
|
{
|
|
SQLMsg* pMsg = NULL;
|
|
|
|
while ((pMsg = PopResult()))
|
|
{
|
|
if( NULL != pMsg->pvUserData )
|
|
{
|
|
switch( reinterpret_cast<CQueryInfo*>(pMsg->pvUserData)->iQueryType )
|
|
{
|
|
case QUERY_TYPE_RETURN:
|
|
AnalyzeReturnQuery(pMsg);
|
|
break;
|
|
|
|
case QUERY_TYPE_FUNCTION:
|
|
{
|
|
CFuncQueryInfo* qi = reinterpret_cast<CFuncQueryInfo*>( pMsg->pvUserData );
|
|
qi->f(pMsg);
|
|
M2_DELETE(qi);
|
|
}
|
|
break;
|
|
|
|
case QUERY_TYPE_AFTER_FUNCTION:
|
|
{
|
|
CFuncAfterQueryInfo* qi = reinterpret_cast<CFuncAfterQueryInfo*>( pMsg->pvUserData );
|
|
qi->f();
|
|
M2_DELETE(qi);
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
delete pMsg;
|
|
}
|
|
}
|
|
|
|
CLoginData * DBManager::GetLoginData(DWORD dwKey)
|
|
{
|
|
std::map<DWORD, CLoginData *>::iterator it = m_map_pkLoginData.find(dwKey);
|
|
|
|
if (it == m_map_pkLoginData.end())
|
|
return NULL;
|
|
|
|
return it->second;
|
|
}
|
|
|
|
void DBManager::InsertLoginData(CLoginData * pkLD)
|
|
{
|
|
m_map_pkLoginData.insert(std::make_pair(pkLD->GetKey(), pkLD));
|
|
}
|
|
|
|
void DBManager::DeleteLoginData(CLoginData * pkLD)
|
|
{
|
|
std::map<DWORD, CLoginData *>::iterator it = m_map_pkLoginData.find(pkLD->GetKey());
|
|
|
|
if (it == m_map_pkLoginData.end())
|
|
return;
|
|
|
|
SPDLOG_DEBUG("DeleteLoginData {} {}", pkLD->GetLogin(), (void*) pkLD);
|
|
|
|
M2_DELETE(it->second);
|
|
m_map_pkLoginData.erase(it);
|
|
}
|
|
|
|
void DBManager::SendLoginPing(const char * c_pszLogin)
|
|
{
|
|
TPacketGGLoginPing ptog;
|
|
|
|
ptog.bHeader = HEADER_GG_LOGIN_PING;
|
|
strlcpy(ptog.szLogin, c_pszLogin, sizeof(ptog.szLogin));
|
|
|
|
if (!g_pkAuthMasterDesc) // If I am master, broadcast to others
|
|
{
|
|
P2P_MANAGER::instance().Send(&ptog, sizeof(TPacketGGLoginPing));
|
|
}
|
|
else // If I am slave send login ping to master
|
|
{
|
|
g_pkAuthMasterDesc->Packet(&ptog, sizeof(TPacketGGLoginPing));
|
|
}
|
|
}
|
|
|
|
void DBManager::SendAuthLogin(LPDESC d)
|
|
{
|
|
const TAccountTable & r = d->GetAccountTable();
|
|
|
|
CLoginData * pkLD = GetLoginData(d->GetLoginKey());
|
|
|
|
if (!pkLD)
|
|
return;
|
|
|
|
TPacketGDAuthLogin ptod;
|
|
ptod.dwID = r.id;
|
|
|
|
trim_and_lower(r.login, ptod.szLogin, sizeof(ptod.szLogin));
|
|
strlcpy(ptod.szSocialID, r.social_id, sizeof(ptod.szSocialID));
|
|
ptod.dwLoginKey = d->GetLoginKey();
|
|
|
|
memcpy(ptod.iPremiumTimes, pkLD->GetPremiumPtr(), sizeof(ptod.iPremiumTimes));
|
|
memcpy(&ptod.adwClientKey, pkLD->GetClientKey(), sizeof(DWORD) * 4);
|
|
|
|
db_clientdesc->DBPacket(HEADER_GD_AUTH_LOGIN, d->GetHandle(), &ptod, sizeof(TPacketGDAuthLogin));
|
|
SPDLOG_DEBUG("SendAuthLogin {} key {}", ptod.szLogin, ptod.dwID);
|
|
|
|
SendLoginPing(r.login);
|
|
}
|
|
|
|
void DBManager::LoginPrepare(LPDESC d, DWORD * pdwClientKey, int * paiPremiumTimes)
|
|
{
|
|
const TAccountTable & r = d->GetAccountTable();
|
|
|
|
CLoginData * pkLD = M2_NEW CLoginData;
|
|
|
|
pkLD->SetKey(d->GetLoginKey());
|
|
pkLD->SetLogin(r.login);
|
|
pkLD->SetIP(d->GetHostName());
|
|
pkLD->SetClientKey(pdwClientKey);
|
|
|
|
if (paiPremiumTimes)
|
|
pkLD->SetPremium(paiPremiumTimes);
|
|
|
|
InsertLoginData(pkLD);
|
|
|
|
if (*d->GetMatrixCode())
|
|
{
|
|
unsigned int rows = 0, cols = 0;
|
|
MatrixCardRndCoordinate(rows, cols);
|
|
|
|
d->SetMatrixCardRowsAndColumns(rows, cols);
|
|
|
|
TPacketGCMatrixCard pm;
|
|
|
|
pm.bHeader = HEADER_GC_MATRIX_CARD;
|
|
pm.dwRows = rows;
|
|
pm.dwCols = cols;
|
|
|
|
d->Packet(&pm, sizeof(TPacketGCMatrixCard));
|
|
|
|
SPDLOG_DEBUG("MATRIX_QUERY: {} {}{} {}{} {}{} {}{} {}",
|
|
r.login,
|
|
MATRIX_CARD_ROW(rows, 0) + 'A',
|
|
MATRIX_CARD_COL(cols, 0) + 1,
|
|
MATRIX_CARD_ROW(rows, 1) + 'A',
|
|
MATRIX_CARD_COL(cols, 1) + 1,
|
|
MATRIX_CARD_ROW(rows, 2) + 'A',
|
|
MATRIX_CARD_COL(cols, 2) + 1,
|
|
MATRIX_CARD_ROW(rows, 3) + 'A',
|
|
MATRIX_CARD_COL(cols, 3) + 1,
|
|
d->GetMatrixCode());
|
|
}
|
|
else
|
|
{
|
|
SendAuthLogin(d);
|
|
}
|
|
}
|
|
|
|
void DBManager::AnalyzeReturnQuery(SQLMsg * pMsg)
|
|
{
|
|
CReturnQueryInfo * qi = (CReturnQueryInfo *) pMsg->pvUserData;
|
|
|
|
switch (qi->iType)
|
|
{
|
|
case QID_AUTH_LOGIN:
|
|
{
|
|
TPacketCGLogin3 * pinfo = (TPacketCGLogin3 *) qi->pvData;
|
|
LPDESC d = DESC_MANAGER::instance().FindByLoginKey(qi->dwIdent);
|
|
|
|
if (!d)
|
|
{
|
|
M2_DELETE(pinfo);
|
|
break;
|
|
}
|
|
// Change Location
|
|
d->SetLogin(pinfo->login);
|
|
|
|
SPDLOG_DEBUG("QID_AUTH_LOGIN: START {} {}", qi->dwIdent, (void*) get_pointer(d));
|
|
|
|
if (pMsg->Get()->uiNumRows == 0)
|
|
{
|
|
SPDLOG_DEBUG(" NOID");
|
|
LoginFailure(d, "NOID");
|
|
M2_DELETE(pinfo);
|
|
}
|
|
else
|
|
{
|
|
MYSQL_ROW row = mysql_fetch_row(pMsg->Get()->pSQLResult);
|
|
int col = 0;
|
|
|
|
// PASSWORD('%s'), password, securitycode, social_id, id, status
|
|
char szEncrytPassword[45 + 1];
|
|
char szPassword[45 + 1];
|
|
char szMatrixCode[MATRIX_CODE_MAX_LEN + 1];
|
|
char szSocialID[SOCIAL_ID_MAX_LEN + 1];
|
|
char szStatus[ACCOUNT_STATUS_MAX_LEN + 1];
|
|
DWORD dwID = 0;
|
|
|
|
if (!row[col])
|
|
{
|
|
SPDLOG_ERROR("error column {}", col);
|
|
M2_DELETE(pinfo);
|
|
break;
|
|
}
|
|
|
|
strlcpy(szEncrytPassword, row[col++], sizeof(szEncrytPassword));
|
|
|
|
if (!row[col])
|
|
{
|
|
SPDLOG_ERROR("error column {}", col);
|
|
M2_DELETE(pinfo);
|
|
break;
|
|
}
|
|
|
|
strlcpy(szPassword, row[col++], sizeof(szPassword));
|
|
|
|
if (!row[col])
|
|
{
|
|
*szMatrixCode = '\0';
|
|
col++;
|
|
}
|
|
else
|
|
{
|
|
strlcpy(szMatrixCode, row[col++], sizeof(szMatrixCode));
|
|
}
|
|
|
|
if (!row[col])
|
|
{
|
|
SPDLOG_ERROR("error column {}", col);
|
|
M2_DELETE(pinfo);
|
|
break;
|
|
}
|
|
|
|
strlcpy(szSocialID, row[col++], sizeof(szSocialID));
|
|
|
|
if (!row[col])
|
|
{
|
|
SPDLOG_ERROR("error column {}", col);
|
|
M2_DELETE(pinfo);
|
|
break;
|
|
}
|
|
|
|
str_to_number(dwID, row[col++]);
|
|
|
|
if (!row[col])
|
|
{
|
|
SPDLOG_ERROR("error column {}", col);
|
|
M2_DELETE(pinfo);
|
|
break;
|
|
}
|
|
|
|
strlcpy(szStatus, row[col++], sizeof(szStatus));
|
|
|
|
BYTE bNotAvail = 0;
|
|
str_to_number(bNotAvail, row[col++]);
|
|
|
|
int aiPremiumTimes[PREMIUM_MAX_NUM];
|
|
memset(&aiPremiumTimes, 0, sizeof(aiPremiumTimes));
|
|
|
|
char szCreateDate[256] = "00000000";
|
|
|
|
if (!g_iUseLocale)
|
|
{
|
|
str_to_number(aiPremiumTimes[PREMIUM_EXP], row[col++]);
|
|
str_to_number(aiPremiumTimes[PREMIUM_ITEM], row[col++]);
|
|
str_to_number(aiPremiumTimes[PREMIUM_SAFEBOX], row[col++]);
|
|
str_to_number(aiPremiumTimes[PREMIUM_AUTOLOOT], row[col++]);
|
|
str_to_number(aiPremiumTimes[PREMIUM_FISH_MIND], row[col++]);
|
|
str_to_number(aiPremiumTimes[PREMIUM_MARRIAGE_FAST], row[col++]);
|
|
str_to_number(aiPremiumTimes[PREMIUM_GOLD], row[col++]);
|
|
}
|
|
else
|
|
{
|
|
str_to_number(aiPremiumTimes[PREMIUM_EXP], row[col++]);
|
|
str_to_number(aiPremiumTimes[PREMIUM_ITEM], row[col++]);
|
|
str_to_number(aiPremiumTimes[PREMIUM_SAFEBOX], row[col++]);
|
|
str_to_number(aiPremiumTimes[PREMIUM_AUTOLOOT], row[col++]);
|
|
str_to_number(aiPremiumTimes[PREMIUM_FISH_MIND], row[col++]);
|
|
str_to_number(aiPremiumTimes[PREMIUM_MARRIAGE_FAST], row[col++]);
|
|
str_to_number(aiPremiumTimes[PREMIUM_GOLD], row[col++]);
|
|
|
|
if (LC_IsEurope() || test_server)
|
|
{
|
|
int retValue = 0;
|
|
str_to_number(retValue, row[col]);
|
|
|
|
time_t create_time = retValue;
|
|
struct tm * tm1;
|
|
tm1 = localtime(&create_time);
|
|
strftime(szCreateDate, 255, "%Y%m%d", tm1);
|
|
|
|
SPDLOG_DEBUG("Create_Time {} {}", retValue, szCreateDate);
|
|
SPDLOG_DEBUG("Block Time {} ", strncmp(szCreateDate, g_stBlockDate.c_str(), 8));
|
|
}
|
|
}
|
|
|
|
int nPasswordDiff = strcmp(szEncrytPassword, szPassword);
|
|
|
|
if (true == LC_IsBrazil())
|
|
{
|
|
nPasswordDiff = 0; // 브라질 버전에서는 비밀번호 체크를 하지 않는다.
|
|
}
|
|
|
|
if (nPasswordDiff)
|
|
{
|
|
LoginFailure(d, "WRONGPWD");
|
|
SPDLOG_DEBUG(" WRONGPWD");
|
|
M2_DELETE(pinfo);
|
|
}
|
|
else if (bNotAvail)
|
|
{
|
|
LoginFailure(d, "NOTAVAIL");
|
|
SPDLOG_DEBUG(" NOTAVAIL");
|
|
M2_DELETE(pinfo);
|
|
}
|
|
else if (DESC_MANAGER::instance().FindByLoginName(pinfo->login))
|
|
{
|
|
LoginFailure(d, "ALREADY");
|
|
SPDLOG_DEBUG(" ALREADY");
|
|
M2_DELETE(pinfo);
|
|
}
|
|
else if (strcmp(szStatus, "OK"))
|
|
{
|
|
LoginFailure(d, szStatus);
|
|
SPDLOG_DEBUG(" STATUS: {}", szStatus);
|
|
M2_DELETE(pinfo);
|
|
}
|
|
else
|
|
{
|
|
if (LC_IsEurope())
|
|
{
|
|
//stBlockData >= 0 == 날짜가 BlockDate 보다 미래
|
|
if (strncmp(szCreateDate, g_stBlockDate.c_str(), 8) >= 0)
|
|
{
|
|
LoginFailure(d, "BLKLOGIN");
|
|
SPDLOG_DEBUG(" BLKLOGIN");
|
|
M2_DELETE(pinfo);
|
|
break;
|
|
}
|
|
|
|
char szQuery[1024];
|
|
snprintf(szQuery, sizeof(szQuery), "UPDATE account SET last_play=NOW() WHERE id=%u", dwID);
|
|
std::unique_ptr<SQLMsg> msg( DBManager::instance().DirectQuery(szQuery) );
|
|
}
|
|
|
|
TAccountTable & r = d->GetAccountTable();
|
|
|
|
r.id = dwID;
|
|
trim_and_lower(pinfo->login, r.login, sizeof(r.login));
|
|
strlcpy(r.passwd, pinfo->passwd, sizeof(r.passwd));
|
|
strlcpy(r.social_id, szSocialID, sizeof(r.social_id));
|
|
DESC_MANAGER::instance().ConnectAccount(r.login, d);
|
|
|
|
d->SetMatrixCode(szMatrixCode);
|
|
|
|
SPDLOG_DEBUG("QID_AUTH_LOGIN: SUCCESS {}", pinfo->login);
|
|
|
|
LoginPrepare(d, pinfo->adwClientKey, aiPremiumTimes);
|
|
M2_DELETE(pinfo);
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
|
|
case QID_SAFEBOX_SIZE:
|
|
{
|
|
LPCHARACTER ch = CHARACTER_MANAGER::instance().FindByPID(qi->dwIdent);
|
|
|
|
if (ch)
|
|
{
|
|
if (pMsg->Get()->uiNumRows > 0)
|
|
{
|
|
MYSQL_ROW row = mysql_fetch_row(pMsg->Get()->pSQLResult);
|
|
int size = 0;
|
|
str_to_number(size, row[0]);
|
|
ch->SetSafeboxSize(SAFEBOX_PAGE_SIZE * size);
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
|
|
case QID_LOTTO:
|
|
{
|
|
LPCHARACTER ch = CHARACTER_MANAGER::instance().FindByPID(qi->dwIdent);
|
|
DWORD * pdw = (DWORD *) qi->pvData;
|
|
|
|
if (ch)
|
|
{
|
|
if (pMsg->Get()->uiAffectedRows == 0 || pMsg->Get()->uiAffectedRows == (uint32_t)-1)
|
|
{
|
|
SPDLOG_DEBUG("GIVE LOTTO FAIL TO pid {}", ch->GetPlayerID());
|
|
}
|
|
else
|
|
{
|
|
LPITEM pkItem = ch->AutoGiveItem(pdw[0], pdw[1]);
|
|
|
|
if (pkItem)
|
|
{
|
|
SPDLOG_DEBUG("GIVE LOTTO SUCCESS TO {} (pid {})", ch->GetName(), qi->dwIdent);
|
|
//ch->ChatPacket(CHAT_TYPE_INFO, LC_TEXT("%s received"), pkItem->GetName());
|
|
|
|
pkItem->SetSocket(0, pMsg->Get()->uiInsertID);
|
|
pkItem->SetSocket(1, pdw[2]);
|
|
}
|
|
else
|
|
SPDLOG_DEBUG("GIVE LOTTO FAIL2 TO pid {}", ch->GetPlayerID());
|
|
}
|
|
}
|
|
|
|
M2_DELETE_ARRAY(pdw);
|
|
}
|
|
break;
|
|
|
|
// BLOCK_CHAT
|
|
case QID_BLOCK_CHAT_LIST:
|
|
{
|
|
LPCHARACTER ch = CHARACTER_MANAGER::instance().FindByPID(qi->dwIdent);
|
|
|
|
if (ch == NULL)
|
|
break;
|
|
if (pMsg->Get()->uiNumRows)
|
|
{
|
|
MYSQL_ROW row;
|
|
while ((row = mysql_fetch_row(pMsg->Get()->pSQLResult)))
|
|
{
|
|
ch->ChatPacket(CHAT_TYPE_INFO, "%s %s sec", row[0], row[1]);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
ch->ChatPacket(CHAT_TYPE_INFO, "No one currently blocked.");
|
|
}
|
|
}
|
|
break;
|
|
// END_OF_BLOCK_CHAT
|
|
|
|
default:
|
|
SPDLOG_ERROR("FATAL ERROR!!! Unhandled return query id {}", qi->iType);
|
|
break;
|
|
}
|
|
|
|
M2_DELETE(qi);
|
|
}
|
|
|
|
void DBManager::SendMoneyLog(BYTE type, DWORD vnum, int gold)
|
|
{
|
|
if (!gold)
|
|
return;
|
|
TPacketMoneyLog p;
|
|
p.type = type;
|
|
p.vnum = vnum;
|
|
p.gold = gold;
|
|
db_clientdesc->DBPacket(HEADER_GD_MONEY_LOG, 0, &p, sizeof(p));
|
|
}
|
|
|
|
size_t DBManager::EscapeString(char* dst, size_t dstSize, const char *src, size_t srcSize)
|
|
{
|
|
return m_sql_direct.EscapeString(dst, dstSize, src, srcSize);
|
|
}
|
|
|
|
//
|
|
// Common SQL
|
|
//
|
|
AccountDB::AccountDB() :
|
|
m_IsConnect(false)
|
|
{
|
|
}
|
|
|
|
bool AccountDB::IsConnected()
|
|
{
|
|
return m_IsConnect;
|
|
}
|
|
|
|
bool AccountDB::Connect(const char * host, const int port, const char * user, const char * pwd, const char * db)
|
|
{
|
|
m_IsConnect = m_sql_direct.Setup(host, user, pwd, db, nullptr, true, port);
|
|
|
|
if (false == m_IsConnect)
|
|
{
|
|
SPDLOG_ERROR("cannot open direct sql connection to host: {} user: {} db: {}", host, user, db);
|
|
return false;
|
|
}
|
|
|
|
return m_IsConnect;
|
|
}
|
|
|
|
bool AccountDB::ConnectAsync(const char * host, const int port, const char * user, const char * pwd, const char * db, const char * locale)
|
|
{
|
|
m_sql.Setup(host, user, pwd, db, locale, false, port);
|
|
return true;
|
|
}
|
|
|
|
void AccountDB::SetLocale(const std::string & stLocale)
|
|
{
|
|
m_sql_direct.SetLocale(stLocale);
|
|
m_sql_direct.QueryLocaleSet();
|
|
}
|
|
|
|
SQLMsg* AccountDB::DirectQuery(const char * query)
|
|
{
|
|
return m_sql_direct.DirectQuery(query);
|
|
}
|
|
|
|
void AccountDB::AsyncQuery(const char* query)
|
|
{
|
|
m_sql.AsyncQuery(query);
|
|
}
|
|
|
|
void AccountDB::ReturnQuery(int iType, DWORD dwIdent, void * pvData, const char * c_pszFormat, ...)
|
|
{
|
|
char szQuery[4096];
|
|
va_list args;
|
|
|
|
va_start(args, c_pszFormat);
|
|
vsnprintf(szQuery, sizeof(szQuery), c_pszFormat, args);
|
|
va_end(args);
|
|
|
|
CReturnQueryInfo * p = M2_NEW CReturnQueryInfo;
|
|
|
|
p->iQueryType = QUERY_TYPE_RETURN;
|
|
p->iType = iType;
|
|
p->dwIdent = dwIdent;
|
|
p->pvData = pvData;
|
|
|
|
m_sql.ReturnQuery(szQuery, p);
|
|
}
|
|
|
|
SQLMsg * AccountDB::PopResult()
|
|
{
|
|
SQLMsg * p;
|
|
|
|
if (m_sql.PopResult(&p))
|
|
return p;
|
|
|
|
return NULL;
|
|
}
|
|
|
|
void AccountDB::Process()
|
|
{
|
|
SQLMsg* pMsg = NULL;
|
|
|
|
while ((pMsg = PopResult()))
|
|
{
|
|
CQueryInfo* qi = (CQueryInfo *) pMsg->pvUserData;
|
|
|
|
switch (qi->iQueryType)
|
|
{
|
|
case QUERY_TYPE_RETURN:
|
|
AnalyzeReturnQuery(pMsg);
|
|
break;
|
|
}
|
|
}
|
|
|
|
delete pMsg;
|
|
}
|
|
|
|
extern unsigned int g_uiSpamReloadCycle;
|
|
|
|
enum EAccountQID
|
|
{
|
|
QID_SPAM_DB,
|
|
};
|
|
|
|
// 10분마다 리로드
|
|
static LPEVENT s_pkReloadSpamEvent = NULL;
|
|
|
|
EVENTINFO(reload_spam_event_info)
|
|
{
|
|
// used to send command
|
|
DWORD empty;
|
|
};
|
|
|
|
EVENTFUNC(reload_spam_event)
|
|
{
|
|
AccountDB::instance().ReturnQuery(QID_SPAM_DB, 0, NULL, "SELECT word, score FROM spam_db WHERE type='SPAM'");
|
|
return PASSES_PER_SEC(g_uiSpamReloadCycle);
|
|
}
|
|
|
|
void LoadSpamDB()
|
|
{
|
|
AccountDB::instance().ReturnQuery(QID_SPAM_DB, 0, NULL, "SELECT word, score FROM spam_db WHERE type='SPAM'");
|
|
|
|
if (NULL == s_pkReloadSpamEvent)
|
|
{
|
|
reload_spam_event_info* info = AllocEventInfo<reload_spam_event_info>();
|
|
s_pkReloadSpamEvent = event_create(reload_spam_event, info, PASSES_PER_SEC(g_uiSpamReloadCycle));
|
|
}
|
|
}
|
|
|
|
void CancelReloadSpamEvent() {
|
|
s_pkReloadSpamEvent = NULL;
|
|
}
|
|
|
|
void AccountDB::AnalyzeReturnQuery(SQLMsg * pMsg)
|
|
{
|
|
CReturnQueryInfo * qi = (CReturnQueryInfo *) pMsg->pvUserData;
|
|
|
|
switch (qi->iType)
|
|
{
|
|
case QID_SPAM_DB:
|
|
{
|
|
if (pMsg->Get()->uiNumRows > 0)
|
|
{
|
|
MYSQL_ROW row;
|
|
|
|
SpamManager::instance().Clear();
|
|
|
|
while ((row = mysql_fetch_row(pMsg->Get()->pSQLResult)))
|
|
SpamManager::instance().Insert(row[0], atoi(row[1]));
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
|
|
M2_DELETE(qi);
|
|
}
|