// NetClient.cpp: implementation of the CNetClient class.
|
//
|
//////////////////////////////////////////////////////////////////////
|
|
#include "stdafx.h"
|
#include "NetClient.h"
|
|
#ifdef _DEBUG
|
#undef THIS_FILE
|
static char THIS_FILE[]=__FILE__;
|
#define new DEBUG_NEW
|
#endif
|
|
//////////////////////////////////////////////////////////////////////
|
// Construction/Destruction
|
//////////////////////////////////////////////////////////////////////
|
|
CNetClient::CNetClient()
|
{
|
StartUp();
|
|
m_pN2M = NULL;
|
m_nServerSocketType = 0;
|
|
m_nPacketCount = 0;
|
m_pPacket = NULL;
|
m_nPacketIdx = 0;
|
InitializeCriticalSection(&m_csPacket);
|
InitializeCriticalSection(&m_csSending);
|
}
|
|
CNetClient::~CNetClient()
|
{
|
DeinitNetwork();
|
|
CleanUp();
|
|
if (m_pPacket)
|
delete [] m_pPacket;
|
m_pPacket = NULL;
|
m_nPacketCount = 0;
|
|
DeleteCriticalSection(&m_csPacket);
|
DeleteCriticalSection(&m_csSending);
|
}
|
|
BOOL CNetClient::SetServerInfo(int ST, short sNetworkCode, short sVersion, int nPacketCount)
|
{
|
m_nServerSocketType = ST; // ¿¬°áÇÒ ServerÀÇ Á¾·ù.
|
|
if (nPacketCount <= 0)
|
nPacketCount = 100;
|
|
if (m_pPacket)
|
delete [] m_pPacket;
|
m_pPacket = NULL;
|
m_pPacket = new CNetPacket[nPacketCount];
|
m_nPacketCount = nPacketCount;
|
|
for (int i = 0; i < m_nPacketCount; i++)
|
m_pPacket[i].SetVersion(sNetworkCode, sVersion);
|
|
m_sNetworkCode = sNetworkCode;
|
m_sVersion = sVersion;
|
|
return TRUE;
|
}
|
|
|
CNetPacket* CNetClient::GetPacket(int nMilSecond)
|
{
|
CNetPacket* pPacket;
|
EnterCriticalSection(&m_csPacket);
|
|
int nCount = 0;
|
do
|
{
|
pPacket = &m_pPacket[m_nPacketIdx];
|
m_nPacketIdx = ++m_nPacketIdx % m_nPacketCount;
|
if (!pPacket->IsUsing())
|
{
|
pPacket->LockPacket();
|
break;
|
}
|
Sleep(1);
|
if (++nCount > nMilSecond)
|
{
|
LeaveCriticalSection(&m_csPacket);
|
return NULL;
|
}
|
} while(TRUE);
|
pPacket->ResetValues();
|
|
LeaveCriticalSection(&m_csPacket);
|
|
return pPacket;
|
}
|
|
BOOL CNetClient::DisconnectServer()
|
{
|
if (m_SocketJoiner.GetSocketContext())
|
{
|
CloseSocket(m_SocketJoiner.GetSocketContext(), FALSE, FALSE);
|
m_SocketJoiner.Reset();
|
return TRUE;
|
}
|
|
return FALSE;
|
}
|
|
// Server SocketÀÌ Á¢¼ÓÀ» Çã¿ëÇßÀ» ¶§ È£ÃâÇÏ´Â ÇÔ¼ö.
|
int CNetClient::AcceptedSocket(pPerSocketContext pSocketCtx, ULONG ulIP, int nPort)
|
{
|
return TRUE;
|
}
|
|
// Client SocketÀÌ Á¢¼ÓµÇ¾úÀ» ¶§ È£ÃâÇÏ´Â ÇÔ¼ö
|
BOOL CNetClient::ConnectedSocket(pPerSocketContext pSocketCtx)
|
{
|
if (m_pN2M)
|
{
|
m_SocketJoiner.SetSocketContext(pSocketCtx);
|
m_pN2M->IOCPNet2P_Connected(m_nServerSocketType);
|
}
|
|
return TRUE;
|
}
|
|
BOOL CNetClient::RecvCompleted(pPerSocketContext pSocketCtx, DWORD dwBytes)
|
{
|
// ¹öÆÛ°¡ ºô ¶§±îÁö ó¸®.
|
WSABUF* pBuf = pSocketCtx->recvContext->GetLockedBuffer();
|
if (!pBuf)
|
{
|
NoticeMsg("Packet Lost :: Buf is not Available. Buf - %d", reinterpret_cast<__int64>(pBuf));
|
return TRUE;
|
}
|
|
int nRead = 0;
|
DWORD dwCompBytes = 0;
|
DWORD dwCurrentBytes = dwBytes + (MAX_BUFFER_SIZE - pBuf->len);
|
pBuf->buf = pSocketCtx->recvContext->GetBulkBuffer();
|
|
int nPacketCount = 0;
|
while (TRUE)
|
{
|
if (dwCurrentBytes < PACKET_HEADERSIZE) // ÃÖ¼Ò ÆÐŶ Å©±â.
|
{
|
// ÆÐŶÀ» ´Ù ¹ÞÁö ¸øÇÏ¸é ´õ ¹Þ´Â´Ù.
|
if (dwCompBytes > 0 && dwCurrentBytes > 0)
|
{
|
// NoticeMsg("ó¸®ÇÑ ºÎºÐ »èÁ¦ 1 Recv %d, Cur %d, Comp %d", dwBytes, dwCurrentBytes, dwCompBytes);
|
char* strTemp = new char[dwCurrentBytes + 1];
|
ZeroMemory(strTemp, dwCurrentBytes + 1);
|
memcpy(strTemp, pBuf->buf, dwCurrentBytes);
|
pBuf->buf = pSocketCtx->recvContext->GetBulkBuffer();
|
ZeroMemory(pBuf->buf, MAX_BUFFER_SIZE + 1);
|
memcpy(pBuf->buf, strTemp, dwCurrentBytes);
|
if (strTemp)
|
{
|
delete [] strTemp;
|
strTemp = NULL;
|
}
|
|
dwCompBytes = 0;
|
}
|
// NoticeMsg("Çì´õÆÐŶº¸´Ù ÀÛÀ½ Recv %d, Cur %d, Comp %d", dwBytes, dwCurrentBytes, dwCompBytes);
|
pBuf->buf = pSocketCtx->recvContext->GetBulkBuffer() + dwCurrentBytes;
|
pBuf->len = MAX_BUFFER_SIZE - dwCurrentBytes;
|
return FALSE;
|
}
|
|
CNetPacket* pPacket = GetPacket();
|
if (!pPacket)
|
{
|
NoticeMsg("RecvCompleted :: Packet is not Available. pPacket - %d", reinterpret_cast<__int64>(pPacket));
|
break;
|
}
|
|
nRead = pPacket->Read_PacketHeader(pBuf);
|
if (nRead == 0)
|
{
|
NoticeMsg("Packet Lost :: Read Packet Header. Read - %d", nRead);
|
pPacket->UnlockPacket();
|
return TRUE;
|
}
|
if (pPacket->GetVersion() != m_sVersion)
|
{
|
NoticeMsg("Packet Header :: Version Mismatch!! Set %d, Packet %d, Recv %d, Cur %d, Len %d, PktCount %d, Comp %d, Size %d"
|
, m_sVersion, pPacket->GetVersion(), dwBytes, dwCurrentBytes, pBuf->len, nPacketCount, dwCompBytes, pPacket->GetSize());
|
pPacket->UnlockPacket();
|
return TRUE;
|
}
|
if (pPacket->GetSize() <= 0) // ¿¹¿Üó¸® - ¹ÞÀº µ¥ÀÌÅÍ´Â Áö¿ö¹ö¸®°í ´Ù½Ã ¹Þ´Â´Ù.
|
{
|
NoticeMsg("Packet Lost :: Exception in Size - Size %d, Code %d, Module %d", pPacket->GetSize(), pPacket->GetPacketCode(), pPacket->GetModuleNo());
|
pPacket->UnlockPacket();
|
return TRUE;
|
}
|
|
if (dwCurrentBytes >= (DWORD)pPacket->GetSize()) // ±æÀÌ Ã¼Å©.
|
{
|
if (pPacket->GetSize() > PACKET_HEADERSIZE)
|
{
|
nRead = pPacket->Read_PacketBody(pBuf->buf + nRead);
|
if (nRead == -1)
|
{
|
NoticeMsg("Packet Lost :: Read Packet Body. Read - %d, Cur %d, Size %d", nRead, dwCurrentBytes, pPacket->GetSize());
|
pPacket->UnlockPacket();
|
return TRUE;
|
}
|
else if (nRead == -2)
|
{
|
NoticeMsg("Packet Lost :: Read Packet Body. Exception - %d, Cur %d, Size %d", nRead, dwCurrentBytes, pPacket->GetSize());
|
pPacket->UnlockPacket();
|
return TRUE;
|
}
|
}
|
|
// Context¿Í Module ¿¬°áÀ» À§ÇØ pSocketCtx¸¦ ³Ñ±ä´Ù.
|
if (m_pN2M)
|
m_pN2M->IOCPNet2P_Received(m_nServerSocketType, pPacket);
|
if (dwCurrentBytes == (DWORD)pPacket->GetSize())
|
{
|
pPacket->UnlockPacket();
|
return TRUE; // ¸ðµç ¹öÆÛ¸¦ ´Ù ó¸®Çß´Ù.
|
}
|
|
// ¹öÆÛ°¡ ³²¾ÆÀÖÀ¸¸é ´Ù½Ã ó¸®ÇÑ´Ù.
|
dwCompBytes += (DWORD)pPacket->GetSize();
|
dwCurrentBytes -= (DWORD)pPacket->GetSize();
|
pBuf->buf = pBuf->buf + pPacket->GetSize();
|
pPacket->UnlockPacket();
|
}
|
else
|
{
|
if (dwCompBytes > 0 && dwCurrentBytes > 0)
|
{
|
// NoticeMsg("ó¸®ÇÑ ºÎºÐ »èÁ¦ 2 Recv %d, Pak %d, Cur %d, Comp %d, PktSize %d", dwBytes, pPacket->GetSize(), dwCurrentBytes, dwCompBytes, pPacket->GetSize());
|
char* strTemp = new char[dwCurrentBytes + 1];
|
ZeroMemory(strTemp, dwCurrentBytes + 1);
|
memcpy(strTemp, pBuf->buf, dwCurrentBytes);
|
pBuf->buf = pSocketCtx->recvContext->GetBulkBuffer();
|
ZeroMemory(pBuf->buf, MAX_BUFFER_SIZE + 1);
|
memcpy(pBuf->buf, strTemp, dwCurrentBytes);
|
if (strTemp)
|
{
|
delete [] strTemp;
|
strTemp = NULL;
|
}
|
}
|
// NoticeMsg("¸ðÀÚ¶ó¼ ´Ù½Ã ¹Þ±â Recv %d, Pak %d, Cur %d, Comp %d", dwBytes, pPacket->GetSize(), dwCurrentBytes, dwCompBytes);
|
pBuf->buf = pSocketCtx->recvContext->GetBulkBuffer() + dwCurrentBytes;
|
pBuf->len = MAX_BUFFER_SIZE - dwCurrentBytes;
|
pPacket->UnlockPacket();
|
return FALSE;
|
}
|
}
|
|
return TRUE;
|
}
|
|
// SocketÀ» ÅëÇØ Send°¡ ¿Ï·áµÇ¾úÀ» ¶§ CIOCPController¿¡¼ È£ÃâÇÏ´Â ÇÔ¼ö. - ÈÄó¸®
|
BOOL CNetClient::SendCompleted(pPerSocketContext pSocketCtx, DWORD dwBytes)
|
{
|
return FALSE;
|
}
|
|
// SocketÀÌ ²÷¾îÁ³À½À» ¾Ë·ÁÁÖ±â À§ÇÑ ÇÔ¼ö
|
void CNetClient::SocketClosed(pPerSocketContext pSocketCtx)
|
{
|
if (m_pN2M)
|
{
|
m_pN2M->IOCPNet2P_Disconnected(m_nServerSocketType);
|
m_SocketJoiner.Reset();
|
}
|
}
|
|
//////////////////////////////////////////////////////////////////////////
|
// Packet Send
|
BOOL CNetClient::SendToServer(int nType, SHORT sResult, int nModuleNo)
|
{
|
return SendToServer(nType, sResult, nModuleNo, -1, NULL);
|
}
|
|
BOOL CNetClient::SendToServer(int nType, SHORT sResult, int nModuleNo, CNetPacket* pPacket)
|
{
|
return SendToServer(nType, sResult, nModuleNo, -1, pPacket);
|
}
|
|
BOOL CNetClient::SendToServer(int nType, SHORT sResult, int nModuleNo, SHORT sPacketIdx, CNetPacket* pPacket)
|
{
|
EnterCriticalSection(&m_csSending);
|
BOOL bRet = FALSE;
|
if (m_SocketJoiner.GetSocketContext())
|
{
|
WSABUF* pWsaBuf = NULL;
|
while (TRUE)
|
{
|
pWsaBuf = m_SocketJoiner.GetSendBuffer();
|
if (!pWsaBuf)
|
Sleep(0);
|
else
|
break;
|
}
|
|
// Buffer¸¦ ä¿î´Ù.
|
pPerSocketContext pCtx = m_SocketJoiner.GetSocketContext();
|
int nWriteSize = 0, i;
|
|
// Header ä¿ì±â.
|
// memcpy(pWsaBuf->buf + nWriteSize, &nSize, sizeof(int));
|
nWriteSize += sizeof(int);
|
memcpy(pWsaBuf->buf + nWriteSize, &m_sNetworkCode, sizeof(SHORT));
|
nWriteSize += sizeof(SHORT);
|
memcpy(pWsaBuf->buf + nWriteSize, &m_sVersion, sizeof(SHORT));
|
nWriteSize += sizeof(SHORT);
|
memcpy(pWsaBuf->buf + nWriteSize, &nType, sizeof(int));
|
nWriteSize += sizeof(int);
|
memcpy(pWsaBuf->buf + nWriteSize, &sPacketIdx, sizeof(SHORT));
|
nWriteSize += sizeof(SHORT);
|
memcpy(pWsaBuf->buf + nWriteSize, &sResult, sizeof(SHORT));
|
nWriteSize += sizeof(SHORT);
|
memcpy(pWsaBuf->buf + nWriteSize, &nModuleNo, sizeof(int));
|
nWriteSize += sizeof(int);
|
|
// Body ä¿ì±â
|
if (pPacket)
|
{
|
int nCount = pPacket->GetShortCount();
|
memcpy(pWsaBuf->buf + nWriteSize, &nCount, sizeof(int));
|
nWriteSize += sizeof(int);
|
for (i = 0; i < nCount; i++)
|
{
|
short sValue = pPacket->GetShort(i);
|
memcpy(pWsaBuf->buf + nWriteSize, &sValue, sizeof(short));
|
nWriteSize += sizeof(short);
|
}
|
|
nCount = pPacket->GetIntCount();
|
memcpy(pWsaBuf->buf + nWriteSize, &nCount, sizeof(int));
|
nWriteSize += sizeof(int);
|
for (i = 0; i < nCount; i++)
|
{
|
int nValue = pPacket->GetInt(i);
|
memcpy(pWsaBuf->buf + nWriteSize, &nValue, sizeof(int));
|
nWriteSize += sizeof(int);
|
}
|
|
nCount = pPacket->GetDoubleCount();
|
memcpy(pWsaBuf->buf + nWriteSize, &nCount, sizeof(int));
|
nWriteSize += sizeof(int);
|
for (i = 0; i < nCount; i++)
|
{
|
double dValue = pPacket->GetDouble(i);
|
memcpy(pWsaBuf->buf + nWriteSize, &dValue, sizeof(double));
|
nWriteSize += sizeof(double);
|
}
|
|
CString strString = _T("");
|
int nLength;
|
nCount = pPacket->GetStringCount();
|
memcpy(pWsaBuf->buf + nWriteSize, &nCount, sizeof(int));
|
nWriteSize += sizeof(int);
|
for (i = 0; i < nCount; i++)
|
{
|
strString = pPacket->GetString(i);
|
|
char * str_Data = NULL;
|
#ifdef _UNICODE
|
int length = WideCharToMultiByte(CP_ACP, 0, strString.GetBuffer(), -1, NULL, 0, NULL,NULL);
|
str_Data = new char[length];
|
WideCharToMultiByte(CP_ACP, 0, strString.GetBuffer(), -1, str_Data, length, NULL, NULL);
|
#else
|
int length = strString.GetLength();
|
str_Data = new char[length+1];
|
memcpy(str_Data, strString.GetBuffer(), sizeof(char)*length);
|
str_Data[length] = NULL;
|
#endif
|
if (str_Data==NULL)
|
{
|
continue;
|
}
|
|
nLength = strString.GetLength();
|
memcpy(pWsaBuf->buf + nWriteSize, &nLength, sizeof(int));
|
nWriteSize += sizeof(int);
|
|
if (nLength > 0)
|
{
|
memcpy(pWsaBuf->buf + nWriteSize, str_Data, nLength);
|
nWriteSize += nLength;
|
}
|
|
delete [] str_Data;
|
}
|
|
nLength = pPacket->GetBufferSize();
|
memcpy(pWsaBuf->buf + nWriteSize, &nLength, sizeof(int));
|
nWriteSize += sizeof(int);
|
if (nLength > 0)
|
{
|
memcpy(pWsaBuf->buf + nWriteSize, pPacket->GetBuffer(), nLength);
|
nWriteSize += nLength;
|
}
|
pPacket->UnlockPacket();
|
}
|
memcpy(pWsaBuf->buf, &nWriteSize, sizeof(int));
|
pWsaBuf->len = nWriteSize;
|
|
bRet = SendPost(m_SocketJoiner.GetSocketContext());
|
}
|
else
|
{
|
if (pPacket)
|
pPacket->UnlockPacket();
|
}
|
LeaveCriticalSection(&m_csSending);
|
return bRet;
|
}
|
