int GetAllFreeCardCount(TADOQuery *query)
{
query->SQL->Clear();
query->SQL->Add("SELECT count(*) as cnt FROM wow_card where is_card_send = 0");
if(!OpenSQL(query))
return 0;
int total = query->FieldByName("cnt")->AsInteger;
return total;
}
BOOL CBCGPODBCGridCtrl::OpenTable (LPCTSTR lpszTable)
{
ASSERT (lpszTable != NULL);
CString strSQL = _T("SELECT * FROM ");
strSQL += lpszTable;
return OpenSQL (strSQL);
}
bool GetExpireTime(TADOQuery *query, String username, vector<String> *outputMsg, TDateTime *result)
{
outputMsg->clear();
query->SQL->Text = FormatStr("SELECT end_time FROM WOW_USER WHERE ACCID = '%s'", username);
if(!OpenSQL(query))
{
outputMsg->push_back(GBText("错误(15)"));
return false;
}
query->First();
TDateTime endTime = query->FieldByName("end_time")->AsDateTime;
if(result)
{
*result = endTime;
}
return true;
}
bool GetCard(TADOQuery *query, String cardType, String *outputCardNo, vector<String> *outputMsg)
{
query->SQL->Text = FormatStr("UPDATE top (1) wow_card SET is_card_send = 1 , send_time = '%s' output INSERTED.card_no, INSERTED.card_type WHERE card_type = '%s' and is_card_send = 0",
DateTimeToStr(Now()), cardType);
if(!OpenSQL(query))
{
outputMsg->push_back(GBText("错误(8)"));
return false;
}
if(!query->RecordCount)
return false;
query->First();
if(outputCardNo)
{
*outputCardNo = query->FieldByName("card_no")->AsString;
}
return true;
}
int AuthClientUser::ClientProcessThread(SingleThread * self)
{
CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
int timeout = SOCKET_TIMEOUT;
setsockopt(m_ClientSocket, SOL_SOCKET, SO_RCVTIMEO,(char*)&timeout, sizeof(timeout));
vector<BYTE> output;
//1. Auth
if(!this->SocketReadBuff(sizeof(tagCClientAuth), &output))
{
SocketReadError("tagCClientAuth SocketReadBuff Error");
return -1;
}
tagCClientAuth *clietAuth = (tagCClientAuth *)&output[0];
clietAuth->UserName[sizeof(clietAuth->UserName)-1] = '\0';
m_UserName = clietAuth->UserName;
m_UserName = m_UserName.LowerCase();
m_UserName = m_UserName.Trim();
string input_username = clietAuth->UserName;
SendLogMessage(FormatStr(GBText("用户[%s]开始验证..."), m_UserName));
m_Build = clietAuth->Build;
BigNumber A, s, B, K, M2;
A.SetBinary(clietAuth->A, sizeof(clietAuth->A));
m_IsChargeProcess = clietAuth->IsChargeProcess;
if(!m_SRP6Server.CalcKey(input_username, input_username, A, &s, &B, &K))
{
SendLogMessage("错误(1)");
SocketReadError("CalcKey Error");
return -1;
}
tagSClientAuth sendAuth;
memcpy(sendAuth.s, s.AsByteArray(), sizeof(sendAuth.s));
memcpy(sendAuth.B, B.AsByteArray(), sizeof(sendAuth.B));
if(!this->SendAuthMessage((BYTE *)&sendAuth, sizeof(sendAuth)))
{
SocketReadError("tagSClientAuth SocketSend Error");
return -1;
}
//2. Auth Check
if(!this->SocketReadBuff(sizeof(tagCAuthCheck), &output))
{
SocketReadError("tagCAuthCheck SocketReadBuff Error");
return -1;
}
tagCAuthCheck *clientAuthCheck = (tagCAuthCheck *)&output[0];
if(!m_SRP6Server.CheckClientAuth(input_username, clientAuthCheck->M, &M2))
{
SendLogMessage("错误(2)");
SocketReadError("Auth Check Error");
return -1;
}
tagSAuthCheck sendAuthCheck;
memcpy(sendAuthCheck.M2, M2.AsByteArray(), sizeof(sendAuthCheck.M2));
if(!this->SendAuthMessage((BYTE *)&sendAuthCheck, sizeof(sendAuthCheck)))
{
SocketReadError("tagSAuthCheck SocketSend Error");
return -1;
}
m_AuthCrypt.Init(&K, WOWWGUserServerEncryptionKey, WOWWGUserServerDecryptionKey);
vector<String> outputMsg;
if(!CheckUserCanLogin(&outputMsg))
{
for(DWORD i=0; i<outputMsg.size(); i++)
{
SendLogMessage(outputMsg[i]);
}
SocketReadError(FormatStr("[RefuseLogin]:%s", m_RefuseLoginReason));
return -1;
}
for(DWORD i=0; i<outputMsg.size(); i++)
{
SendLogMessage(outputMsg[i]);
}
// SendLogMessage(FormatStr(GBText("用户[%s]验证成功!"), m_UserName));
TDateTime expTime;
if(m_IsChargeProcess)
{
//充值
BYTE len = 0;
if(!this->SocketReadBYTE(&len))
{
SendLogMessage("错误(17)");
SocketReadError("Charge CardNo Read Len Error");
return -1;
}
if(!this->SocketReadBuff(len, &output))
{
SendLogMessage("错误(14)");
SocketReadError("Charge CardNo Read Error");
return -1;
}
AnsiString cardNo = AnsiString((char *)&output[0], len);
cardNo = cardNo.Trim();
cardNo = cardNo.UpperCase();
outputMsg.clear();
if(!Charge(m_ADOQuery, m_UserName, cardNo, &outputMsg))
{
for(DWORD i=0; i<outputMsg.size(); i++)
{
SendLogMessage(outputMsg[i]);
}
SocketReadError("Charge Fail!");
return -1;
}
for(DWORD i=0; i<outputMsg.size(); i++)
{
SendLogMessage(outputMsg[i]);
}
outputMsg.clear();
if(!GetExpireTime(m_ADOQuery, m_UserName, &outputMsg, &expTime))
{
for(DWORD i=0; i<outputMsg.size(); i++)
{
SendLogMessage(outputMsg[i]);
}
SocketReadError("Charge GetExpireTime Fail!");
return -1;
}
for(DWORD i=0; i<outputMsg.size(); i++)
{
SendLogMessage(outputMsg[i]);
}
SendLogMessage(GBText("充值成功!"));
}
else
{
//验证
tagWOWAddrInfo addrInfo;
if(!FindAddrInfo(m_Build, &addrInfo))
{
SendLogMessage("错误(3)");
GetLog()->Error("No Build Msg. Build = %d", m_Build);
SocketReadError("We Don't Have This Build!");
return -1;
}
outputMsg.clear();
if(!GetExpireTime(m_ADOQuery, m_UserName, &outputMsg, &expTime))
{
for(DWORD i=0; i<outputMsg.size(); i++)
{
SendLogMessage(outputMsg[i]);
}
SocketReadError("GetExpireTime Fail!");
return -1;
}
for(DWORD i=0; i<outputMsg.size(); i++)
{
SendLogMessage(outputMsg[i]);
}
tagClientAddrInfo info;
info.Address = addrInfo.Addr;
info.Offset = addrInfo.Offset;
info.ExpireTime = expTime;
if(!this->SendAuthMessage((BYTE *)&info, sizeof(tagClientAddrInfo)))
{
SocketReadError("tagClientAddrInfo SocketSend Error");
return -1;
}
m_ADOQuery->SQL->Text = FormatStr("SELECT * FROM wow_msg where msg_key = '%s'", L"login");
if(OpenSQL(m_ADOQuery))
{
if(m_ADOQuery->RecordCount)
{
m_ADOQuery->First();
String text = m_ADOQuery->FieldByName("msg")->AsString;
SendLogMessage(text);
}
}
SendLogMessage(GBText("验证成功!"));
}
SendLogMessage(GBText(FormatStr("你的帐号[%s]到期时间为:%s", m_UserName, DateTimeToStr(expTime))));
OnLoginSuccess();
Sleep(1000);
this->SetIsClosed(1);
return -1;
}
bool AuthClientUser::CheckUserCanLogin(vector<String> *outputMsg)
{
m_ADOConnection = new TADOConnection(NULL);
m_ADOQuery = new TADOQuery(NULL);
m_ADOQuery->Connection = m_ADOConnection;
int tryCnt = 0;
String m_ConnectString;
do
{
try
{
if(tryCnt > 5)
break;
tryCnt++;
m_ADOConnection->LoginPrompt = false;
m_ADOConnection->KeepConnection = true;
m_ADOConnection->ConnectionTimeout = SQL_TIMEOUT;
m_ADOConnection->ConnectionString= FormatStr("Provider=SQLOLEDB.1;Persist Security Info=True;User ID=%s;Password=%s;Initial Catalog=wowwg;Data Source=%s,%d",
GetAuthServerListener()->GetDatabaseUserID(), GetAuthServerListener()->GetDatabaseUserPsw(), GetAuthServerListener()->GetDatabaseIP(), GetAuthServerListener()->GetDatabasePort());
m_ADOConnection->Connected=true;
Sleep(5);
}
catch (Exception &e)
{
m_RefuseLoginReason = FormatStr("SQL Error:%s", e.Message);
}
}
while(m_ADOConnection->Connected == false);
if(m_ADOConnection->Connected == false)
{
outputMsg->push_back(GBText("错误(6)"));
return false;
}
m_ADOQuery->SQL->Text = FormatStr("SELECT * FROM WOW_USER WHERE ACCID = '%s'", m_UserName);
if(!OpenSQL(m_ADOQuery))
{
outputMsg->push_back(GBText("错误(7)"));
return false;
}
if(m_ADOQuery->RecordCount == 0)
{
if(!OnUserFirstLogin(outputMsg))
{
return false;
}
}
else
{
//验证用户是否到期
if(!m_IsChargeProcess)
{
m_ADOQuery->First();
if(Now() > m_ADOQuery->FieldByName("end_time")->AsDateTime)
{
outputMsg->push_back(GBText("你的时间已到期, 请注册!"));
SendRegMessage();
return false;
}
}
}
return true;
}
bool AuthClientUser::OnUserFirstLogin(vector<String> *outputMsg)
{
//1. 取一张免费卡
vector<tagCardInfo> freeCardList;
for(int i=0; i<GetAuthServerListener()->GetCardInfoCount(); i++)
{
tagCardInfo *info = GetAuthServerListener()->GetCardInfoByIndex(i);
if(!info)
continue;
if(info->CardMoney != 0)
continue;
freeCardList.push_back(*info);
}
sort(freeCardList.begin(), freeCardList.end(),greater<tagCardInfo>());
String cardType, cardNo;
for(DWORD i=0; i<freeCardList.size(); i++)
{
if(GetCard(m_ADOQuery, freeCardList[i].CardType, &cardNo, outputMsg))
{
cardType = freeCardList[i].CardType;
break;
}
}
if(cardNo == "")
{
if(m_IsChargeProcess == 0)
{
//免费卡已经发放完毕
outputMsg->push_back(GBText("免费卡已经发放完毕, 请注册!"));
SendRegMessage();
return false;
}
else
{
outputMsg->push_back(GBText("对不起, 免费卡已经发放完毕!"));
}
}
//2. 创建用户
String nowStr = DateTimeToStr(Now());
m_ADOQuery->SQL->Text = FormatStr("INSERT wow_user ("
"accid, last_login_time, login_count, start_time, end_time, "
"total_time, money_total_time, money, login_ip, create_time, "
"create_ip) VALUES("
"'%s', '%s', %d, '%s', '%s',"
"%d, %d, %d, '%s', '%s',"
"'%s'"
")",
m_UserName, nowStr, 0, nowStr, nowStr,
0, 0, 0, m_ClientIP, nowStr,
m_ClientIP
);
if(!ExecSQL(m_ADOQuery))
{
outputMsg->push_back(GBText("错误(9)"));
return false;
}
m_ADOQuery->SQL->Text = FormatStr("SELECT * FROM wow_msg where msg_key = '%s'", L"first_login");
if(!OpenSQL(m_ADOQuery))
{
outputMsg->push_back(GBText("错误(10)"));
return false;
}
if(m_ADOQuery->RecordCount)
{
m_ADOQuery->First();
String text = m_ADOQuery->FieldByName("msg")->AsString;
outputMsg->push_back(text);
}
//3. 充值
if(cardNo != "")
{
if(!Charge(m_ADOQuery, m_UserName, cardNo, outputMsg))
{
return false;
}
}
return true;
}
bool Charge(TADOQuery *query, String username, String card_no, vector<String> *outputMsg)
{
query->SQL->Clear();
query->SQL->Text = FormatStr("SELECT * FROM wow_card WHERE card_no = '%s'", card_no);
if(!OpenSQL(query))
{
outputMsg->push_back(GBText("错误(11)"));
return false;
}
if(query->RecordCount == 0)
{
outputMsg->push_back(GBText("充值卡卡号错误"));
GetLog()->Warn("Charge Fail! Can't Find Card %s", card_no);
return false;
}
query->First();
String cardType = query->FieldByName("card_type")->AsString;
query->SQL->Text = FormatStr("SELECT * FROM used_wow_card WHERE card_no = '%s'", card_no);
if(!OpenSQL(query))
{
outputMsg->push_back(GBText("错误(16)"));
return false;
}
if(query->RecordCount != 0)
{
outputMsg->push_back(GBText("充值卡已经被用过"));
GetLog()->Warn("Charge Fail! Card Used %s", card_no);
return false;
}
query->SQL->Text = FormatStr("INSERT used_wow_card (card_no, accid, card_type, use_time) VALUES('%s', '%s', '%s', '%s')",
card_no, username, cardType, DateTimeToStr(Now()));
if(!ExecSQL(query))
{
outputMsg->push_back(GBText("错误(12)"));
return false;
}
int cardDay = GetDaysByCardType(cardType);
int cardMoney = GetMoneyByCardType(cardType);
int money_time = cardDay;
if(cardMoney == 0)
{
money_time = 0;
}
TDateTime expTime;
if(!GetExpireTime(query, username, outputMsg, &expTime))
{
return false;
}
if (expTime < Now())
{
expTime = Now();
}
query->SQL->Text = FormatStr("UPDATE wow_user SET end_time = CAST('%s' AS DateTime) + %d, total_time = total_time + %d, "
"money_total_time = money_total_time + %d, money = money + %d where accid = '%s'",
DateTimeToStr(expTime), cardDay, cardDay, money_time, cardMoney, username);
if(!ExecSQL(query))
{
outputMsg->push_back(GBText("错误(13)"));
return false;
}
outputMsg->push_back(FormatStr(GBText("已冲入充值卡:%s"), GetCardNameByCardType(cardType)));
return true;
}
//*****************************************************************************
void CBCGPDBGridCtrl::Sort (int nColumn, BOOL bAscending, BOOL bAdd)
{
if (m_bIsSorting || !CanSortByColumn (nColumn))
{
return;
}
if (!m_bDbSort || m_strSQL.IsEmpty ())
{
CBCGPGridCtrl::Sort (nColumn, bAscending, bAdd);
return;
}
SetCurSel (NULL);
m_CachedItems.CleanUpCache ();
CString strSQLOrign = m_strSQL;
CString strColumn = GetColumnName (nColumn);
if (strColumn.Find (_T(' ')) >= 0)
{
strColumn = _T('\'') + strColumn + _T('\'');
}
CString strSQL;
strSQL.Format (_T("%s ORDER BY %s %s"),
m_strSQL, strColumn,
bAscending ? _T(" ASC") : _T(" DESC"));
if (bAdd)
{
for (POSITION pos = m_Columns.m_mapSortColumn.GetStartPosition (); pos != NULL; )
{
int nListColumn, nState;
m_Columns.m_mapSortColumn.GetNextAssoc (pos, nListColumn, nState);
if (nState != 0 && nListColumn != nColumn)
{
CString strListColumn = GetColumnName (nListColumn);
if (strListColumn.Find (_T(' ')) >= 0)
{
strListColumn = _T('\'') + strListColumn + _T('\'');
}
CString strOrder;
strOrder.Format (_T(", %s %s"),
strListColumn,
nState > 0 ? _T(" ASC") : _T(" DESC"));
strSQL += strOrder;
}
}
}
CWaitCursor wait;
m_bRebuildTerminalItems = TRUE;
m_bIsSorting = TRUE;
if (OpenSQL (strSQL))
{
m_Columns.SetSortColumn (nColumn, bAscending, bAdd);
}
RedrawWindow (m_rectHeader);
m_bIsSorting = FALSE;
m_strSQL = strSQLOrign;
}