Category

Development Platform
Home > SourceCode/Document > Game Program > Game Server Simulator > View Code
ServerSockMsg.cpp
Package: LegendOfMir2_Server020418.zip　[view]
Upload User: szopptop
Upload Date: 2013-04-23
Package Size: 1047k
Code Size: 4k
Category:

Game Server Simulator

Development Platform:

Visual C++

ServerSockMsg.cpp：Code Content
  1. #include "stdafx.h"
  3. void SendExToServer(char *pszPacket);
  5. extern SOCKET g_ssock;
  6. extern SOCKET g_csock;
  8. extern HWND g_hStatusBar;
  10. extern HANDLE g_hIOCP;
  12. CWHList<CSessionInfo*> g_xSessionList;
  14. void UpdateStatusBar(BOOL fGrow)
  15. {
  16. static LONG nNumOfCurrSession = 0;
  18. TCHAR szText[20];
  20. (fGrow ? InterlockedIncrement(&nNumOfCurrSession) : InterlockedDecrement(&nNumOfCurrSession));
  22. wsprintf(szText, _TEXT("%d Sessions"), nNumOfCurrSession);
  24. SendMessage(g_hStatusBar, SB_SETTEXT, MAKEWORD(3, 0), (LPARAM)szText);
  25. }
  27. //UINT WINAPI AcceptThread(LPVOID lpParameter)
  28. DWORD WINAPI AcceptThread(LPVOID lpParameter)
  29. {
  30. int nLen = sizeof(SOCKADDR_IN);
  31. char szMsg[64];
  33. SOCKET Accept;
  34. SOCKADDR_IN Address;
  36. while (TRUE)
  37. {
  38. Accept = accept(g_ssock, (struct sockaddr FAR *)&Address, &nLen);
  40. if (g_fTerminated)
  41. return 0;
  43. CSessionInfo* pNewUserInfo = (CSessionInfo*)GlobalAlloc(GPTR, sizeof(CSessionInfo));
  45. if (pNewUserInfo)
  46. {
  47. pNewUserInfo->sock = Accept;
  49. CreateIoCompletionPort((HANDLE)pNewUserInfo->sock, g_hIOCP, (DWORD)pNewUserInfo, 0);
  51. if (g_xSessionList.AddNewNode(pNewUserInfo))
  52. {
  53. int zero = 0;
  55. setsockopt(pNewUserInfo->sock, SOL_SOCKET, SO_SNDBUF, (char *)&zero, sizeof(zero) );
  57. // ORZ:
  58. pNewUserInfo->Recv();
  60. UpdateStatusBar(TRUE);
  62. szMsg[0] = '%';
  63. szMsg[1] = 'O';
  65. char *pszPos = ValToAnsiStr((int)Accept, &szMsg[2]);
  67. *pszPos++ = '/';
  69. pszPos = ValToAnsiStr((int)Address.sin_addr.s_net, pszPos);
  70. *pszPos++ = '.';
  71. pszPos = ValToAnsiStr((int)Address.sin_addr.s_host, pszPos);
  72. *pszPos++ = '.';
  73. pszPos = ValToAnsiStr((int)Address.sin_addr.s_lh, pszPos);
  74. *pszPos++ = '.';
  75. pszPos = ValToAnsiStr((int)Address.sin_addr.s_impno, pszPos);
  77. *pszPos++ = '$';
  78. *pszPos = '';
  80. SendExToServer(szMsg);
  81. }
  82. }
  83. }
  85. return 0;
  86. }
  88. //void CloseSession(CSessionInfo* pSessionInfo)
  89. void CloseSession(int s)
  90. {
  91. char szMsg[32];
  93. // Send close msg to login server
  94. //wsprintf(szMsg, _TEXT("%%X%d$"), s);
  95. szMsg[0] = '%';
  96. szMsg[1] = 'X';
  98. char *pszPos = ValToAnsiStr(s, &szMsg[2]);
  100. *pszPos++ = '$';
  101. *pszPos = '';
  103. SendExToServer(szMsg); 
  105. closesocket(s);
  107. UpdateStatusBar(FALSE);
  108. }
  110. DWORD WINAPI ServerWorkerThread(LPVOID CompletionPortID)
  111. {
  112. DWORD dwBytesTransferred = 0;
  114. CSessionInfo* pSessionInfo = NULL;
  115. _LPTCOMPLETIONPORT lpPerIoData = NULL;
  117. char szPacket[DATA_BUFSIZE * 2];
  118. char szMsg[32];
  119. char *pszPos;
  121. while (TRUE)
  122. {
  123. if ( GetQueuedCompletionStatus( 
  124. (HANDLE)CompletionPortID, 
  125. &dwBytesTransferred, 
  126. (LPDWORD)&pSessionInfo, 
  127. (LPOVERLAPPED *)&lpPerIoData, 
  128. INFINITE) == 0)
  129. {
  130. if (g_fTerminated)
  131. return 0;
  133. if (pSessionInfo)
  134. {
  135. szMsg[0] = '%';
  136. szMsg[1] = 'X';
  138. char *pszPos = ValToAnsiStr((int)pSessionInfo->sock, &szMsg[2]);
  140. *pszPos++ = '$';
  141. *pszPos = '';
  143. SendExToServer(szMsg); 
  145. g_xSessionList.RemoveNodeByData(pSessionInfo);
  147. closesocket(pSessionInfo->sock);
  148. pSessionInfo->sock = INVALID_SOCKET;
  150. UpdateStatusBar(FALSE);
  152. GlobalFree(pSessionInfo);
  153. }
  155. continue;
  156. }
  158. if (g_fTerminated)
  159. return 0;
  161. if (dwBytesTransferred == 0)
  162. {
  163. szMsg[0] = '%';
  164. szMsg[1] = 'X';
  166. char *pszPos = ValToAnsiStr((int)pSessionInfo->sock, &szMsg[2]);
  168. *pszPos++ = '$';
  169. *pszPos = '';
  171. SendExToServer(szMsg); 
  173. g_xSessionList.RemoveNodeByData(pSessionInfo);
  175. closesocket(pSessionInfo->sock);
  176. pSessionInfo->sock = INVALID_SOCKET;
  178. UpdateStatusBar(FALSE);
  180. GlobalFree(pSessionInfo);
  182. continue;
  183. }
  186. // ORZ:
  187. pSessionInfo->bufLen += dwBytesTransferred;
  189. while ( pSessionInfo->HasCompletionPacket() )
  190. {
  191. szPacket[0] = '%';
  192. szPacket[1] = 'A';
  193. pszPos = ValToAnsiStr( (int) pSessionInfo->sock, &szPacket[2] );
  194. *pszPos++ = '/';
  195. pszPos = pSessionInfo->ExtractPacket( pszPos );
  196. *pszPos++ = '$';
  197. *pszPos = '';
  199. SendExToServer( szPacket );
  200. }
  202. // ORZ:
  203. if ( pSessionInfo->Recv() == SOCKET_ERROR && WSAGetLastError() != ERROR_IO_PENDING )
  204. {
  205. InsertLogMsg(_TEXT("WSARecv() failed"));
  207. CloseSession(pSessionInfo->sock);
  208. continue;
  209. }
  210. }
  212. return 0;
  213. }