Category

Development Platform
Home > SourceCode/Document > Game Program > Game Server Simulator > View Code
NetRelay.cpp
Package: SwordOnline.rar　[view]
Upload User: dzyhzl
Upload Date: 2019-04-29
Package Size: 56270k
Code Size: 3k
Category:

Game Server Simulator

Development Platform:

C/C++

NetRelay.cpp：Code Content
  1. // NetRelay.cpp: implementation of the CNetRelay class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  5. #include "stdafx.h"
  6. #include "S3Relay.h"
  7. #include "NetRelay.h"
  9. #include "../../Multiserver/Heaven/Interface/IServer.h"
  13. //////////////////////////////////////////////////////////////////////
  14. // Construction/Destruction
  15. //////////////////////////////////////////////////////////////////////
  17. CNetRelay::CNetRelay()
  18. : m_hDataExist(NULL)
  19. {
  20. }
  22. CNetRelay::~CNetRelay()
  23. {
  24. assert(!m_hDataExist);
  26. Clearup();
  27. }
  32. DWORD CNetRelay::Main(LPVOID lpParam)
  33. {
  34. for (;;)
  35. {
  36. const HANDLE hdls[] = {m_hStop, m_hDataExist};
  37. DWORD wait = ::WaitForMultipleObjects(sizeof(hdls)/sizeof(hdls[0]), hdls, FALSE, INFINITE);
  38. if (wait != WAIT_OBJECT_0 + 1) // !(data exist)
  39. break;
  40. Breathe();
  41. }
  43. return 0;
  44. }
  47. void CNetRelay::Breathe()
  48. {
  49. Lock(); // <<<
  50. if (m_queuePckg.empty())
  51. {
  52. assert(::WaitForSingleObject(m_hDataExist, 0) == WAIT_TIMEOUT);
  53. ::ResetEvent(m_hDataExist);
  55. Unlock(); // >>>
  56. return;
  57. }
  58. else
  59. {
  60. PACKAGE pckg = m_queuePckg.front();
  61. m_queuePckg.pop_front();
  63. if (m_queuePckg.empty())
  64. ::ResetEvent(m_hDataExist);
  67. Unlock(); // >>>
  70. //send ...
  72. assert(pckg.pData && pckg.size > 0);
  74. g_pNetServer->SendPackage(pckg.idFrom, pckg.ipTo, pckg.portTo, pckg.pData, pckg.size);
  76. g_pMemory->Free(pckg.pData);
  79. in_addr ia;
  80. ia.s_addr = pckg.ipTo;
  81. gTrace("%s -- send size: %d", inet_ntoa(ia), pckg.size);
  82. }
  83. }
  85. void CNetRelay::Clearup()
  86. {
  87. for (PACKAGEQUEUE::iterator it = m_queuePckg.begin(); it != m_queuePckg.end(); it++)
  88. g_pMemory->Free((*it).pData);
  90. m_queuePckg.clear();
  91. }
  94. int CNetRelay::RelayPackage(unsigned long idFrom, DWORD ipTo, unsigned short portTo, void* pData, unsigned size)
  95. {
  96. if (pData == NULL || size < sizeof(RELAY_COMMAND))
  97. return FALSE;
  99. RELAY_COMMAND* pRelayCmd = (RELAY_COMMAND*)pData;
  100. assert(pRelayCmd->ProtocolType == c2s_relay);
  102. if (pRelayCmd->wLength < sizeof(RELAY_COMMAND) - sizeof(pRelayCmd->ProtocolType))
  103. return FALSE;
  106. PACKAGE pckg;
  107. pckg.idFrom = idFrom;
  108. pckg.ipTo = ipTo;
  109. pckg.portTo = portTo;
  110. pckg.pData = pData;
  111. pckg.size = size;
  114. Lock();
  116. m_queuePckg.push_back(pckg);
  118. if (m_hDataExist)
  119. ::SetEvent(m_hDataExist);
  121. Unlock();
  123. return TRUE;
  124. }
  126. BOOL CNetRelay::Stop()
  127. {
  128. BOOL ret = KThread::Stop();
  129. if (!ret)
  130. return FALSE;
  132. if (m_hDataExist)
  133. {
  134. ::CloseHandle(m_hDataExist);
  135. m_hDataExist = NULL;
  136. }
  138. return TRUE;
  139. }
  141. HANDLE CNetRelay::Start()
  142. {
  143. if (!m_hDataExist)
  144. m_hDataExist = ::CreateEvent(NULL, TRUE, !m_queuePckg.empty(), NULL);
  145. if (!m_hDataExist)
  146. return NULL;
  148. HANDLE hProc = KThread::Start();
  149. if (!hProc)
  150. {
  151. ::CloseHandle(m_hDataExist);
  152. m_hDataExist = NULL;
  153. return NULL;
  154. }
  156. return hProc;
  157. }