开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 游戏 > 模拟服务器 > 查看源码
kroledbserver.cpp
资源名称：SwordOnline.rar　[点击查看]
上传用户：dzyhzl
上传日期：2019-04-29
资源大小：56270k
文件大小：9k
源码类别：

模拟服务器

开发平台：

C/C++

kroledbserver.cpp：源码内容
  1. #include "KRoleDBServer.h"
  2. #include "KRoleInfomation.h"
  3. #include "Windows.h"
  4. #include "KDBProcessThread.h"
  5. #include <crtdbg.h>
  6. #include "KException.h"
  7. #pragma warning(disable:4786)
  8. #include <objbase.h>
  9. #include <initguid.h>
  10. extern void * EnumSendMsg(unsigned long nId, size_t& nDataLen);
  11. //using namespace RoleDBServer
  12. RoleDBServer::KNetServer g_NetServer;
  13. size_t g_nMaxSendBuffer = 350;
  14. extern TRoleNetMsg  * pTestCmd;
  15. extern void SendToClient(unsigned long nId, TRoleNetMsg * pMsg, size_t nMsgLen);
  16. extern IServer * g_pNetServer;
  19. using OnlineGameLib::Win32::CPackager;
  20. using OnlineGameLib::Win32::CBuffer;
  22. namespace RoleDBServer
  23. {
  25. KClientUser::KClientUser()
  26. {
  27. m_nProcessLoadCount = m_nProcessWriteCount = 0;
  28. this->m_szClientName[0] = 0;
  29. this->m_nCurSendNo = 0;
  30. m_ulSendBufferCount = 0;
  31. m_ulRecvBufferCount = 0;
  32. m_CurrentProtocol = 0;
  33. }
  35. KClientUser::~KClientUser()
  36. {
  40. }
  43. int KNetServer::Init()
  44. {
  45. m_pNetServer = NULL;
  46. if (g_pNetServer)
  47. {
  48. g_pNetServer->QueryInterface( IID_IIOCPServer, reinterpret_cast< void ** >( &m_pNetServer ) );
  49. }
  51. if (!m_pNetServer)
  52. {
  53. throw KException("NetServer", "Cloning Net Server Error");
  54. }
  55. return 1;
  56. }
  58. void KNetServer::RegisterClient(unsigned long nId)
  59. {
  60. int t = ClearClientSend(nId);
  61. CCriticalSection::Owner locker(g_NetMsgListMutex);
  62. if (!m_ClientUserSet[nId])
  63. {
  64. KClientUser * pNewUser = new KClientUser;
  65. pNewUser->m_ID = nId;
  66. m_ClientUserSet[nId] = pNewUser;
  67. }
  68. }
  70. int KNetServer::ClearClientSend(unsigned long nId)
  71. {
  72. int i = 0;
  74. {
  75. CCriticalSection::Owner locker(g_NetMsgListMutex);
  76. list<TCmd *>::iterator I = g_NetServiceThreadCmdList.begin();
  78. while (I != g_NetServiceThreadCmdList.end())
  79. {
  80. if ((*I)->ulNetId == nId)
  81. {
  82. i++;
  83. list<TCmd *>::iterator II = I;
  84. I ++;
  85. delete []*II;
  86. g_NetServiceThreadCmdList.erase(II);
  87. }
  88. else
  89. I ++;
  90. }
  91. }
  93. {
  94. CCriticalSection::Owner locker(g_MainMsgListMutex);
  95. list<TCmd*>::iterator MainMsgI = g_MainThreadCmdList.begin();
  96. while(MainMsgI != g_MainThreadCmdList.end())
  97. {
  98. if ((*MainMsgI)->ulNetId == nId)
  99. {
  101. list<TCmd*>::iterator II1 = MainMsgI;
  102. MainMsgI ++;
  103. delete [](*II1);
  104. g_MainThreadCmdList.erase(II1);
  105. }
  106. else 
  107. MainMsgI ++;
  108. }
  109. }
  111. {
  112. CCriticalSection::Owner locker(g_GetRoleInfoMutex);
  113. list<TGetRoleInfoFromDB*>::iterator DBLoadI = g_DBLoadThreadCmdList.begin();
  114. while(DBLoadI != g_DBLoadThreadCmdList.end())
  115. {
  116. if ((*DBLoadI)->nNetId == nId)
  117. {
  118. list<TGetRoleInfoFromDB*>::iterator II2 = DBLoadI;
  119. DBLoadI ++;
  120. delete (*II2);
  121. g_DBLoadThreadCmdList.erase(II2);
  122. }
  123. else 
  124. DBLoadI ++;
  125. }
  126. }
  128. return i;
  129. }
  131. void KNetServer::DestoryClient(unsigned long nId)
  132. {
  133. CCriticalSection::Owner locker(g_NetMsgListMutex);
  134. int t = ClearClientSend(nId);
  135. KClientUser * pDelUse = m_ClientUserSet[nId];
  136. if (pDelUse)
  137. {
  138. delete pDelUse;
  139. m_ClientUserSet.erase(nId);
  140. }
  141. }
  143. //public
  144. int KNetServer::Receive()
  145. {
  146. KClientUserSet::iterator I;
  147. try{
  149. CCriticalSection::Owner locker(g_NetMsgListMutex);
  150. for ( I = m_ClientUserSet.begin(); I != m_ClientUserSet.end(); I ++ )
  151. {
  152. Receive(I->first);
  153. }
  155. }
  156. catch(...)
  157. {
  158. g_nMistakeCount ++;
  159. }
  160. return 1;
  161. }
  163. int KNetServer::Send()
  164. {
  165. KClientUserSet::iterator I;
  166. for ( I = m_ClientUserSet.begin(); I != m_ClientUserSet.end(); I ++)
  167. {
  168. Send(I->first);
  169. }
  170. return 1;
  171. }
  173. //private
  174. int KNetServer::Receive(unsigned long nId)
  175. {
  176. //获得数据
  177. size_t nDataLen = 0;
  178. const void * pMsg = NULL;
  179. pMsg = m_pNetServer->GetPackFromClient(nId, nDataLen);
  181. if (nDataLen == 0 || pMsg == NULL) return 0;
  182. TProcessData* pData = (TProcessData*)pMsg;
  183. return AppendData(nId, pMsg, nDataLen);
  184. }
  186. int KNetServer::AppendData(unsigned long nId, const void * pMsgBuffer, size_t nMsgLen)
  187. {
  188. KClientUser * pClientUser = m_ClientUserSet[nId];
  189. _ASSERT(pClientUser);
  191. pClientUser->m_ulRecvBufferCount += nMsgLen;
  192. g_dwRecvLen = pClientUser->m_ulRecvBufferCount;
  193. if (pClientUser->m_ulRecvBufferCount >= 0xFFFF1FFF)
  194. pClientUser->m_ulRecvBufferCount = 0;
  196. BYTE btProtoId = CPackager::Peek(pMsgBuffer);
  197. if (btProtoId > c2s_micropackbegin)
  198. {
  199. TProcessData * pProcessData = (TProcessData*)pMsgBuffer;
  200. if (pProcessData->nDataLen + sizeof(TProcessData) - 1 != nMsgLen)
  201. {
  202. _ASSERT(0);
  203. return 1;
  204. }
  205. return Service(pClientUser , (TProcessData *)pMsgBuffer, nId);
  206. }
  207. else
  208. {
  209. CBuffer* pData = pClientUser->m_RecvPackager.PackUp( pMsgBuffer, nMsgLen );
  210. if (pData)
  211. {
  212. TProcessData * pProcessData = (TProcessData*)pData->GetBuffer();
  213. if (pProcessData->nDataLen + sizeof(TProcessData) - 1 != pData->GetUsed())
  214. {
  215. _ASSERT(0);
  216. SAFE_RELEASE(pData);
  217. return 1;
  218. }
  219. Service(pClientUser , (TProcessData*)pData->GetBuffer(), nId);
  220. SAFE_RELEASE(pData);
  221. }
  222. }
  224. return 1;
  225. }
  227. int KNetServer::Service(KClientUser *pClientUser, TProcessData * pProcess, int nId)
  228. {
  229. if (!pProcess) return 0;
  230. TCmd * pNewCmd = (TCmd *) (new char [sizeof(TCmd) + pProcess->nDataLen - 1]);
  231. pNewCmd->ProcessData.nDataLen = pProcess->nDataLen;
  232. pNewCmd->ProcessData.nProtoId  = pProcess->nProtoId;
  233. pNewCmd->ProcessData.ulIdentity = pProcess->ulIdentity;
  234. pNewCmd->ulNetId = nId;
  235. memcpy(&pNewCmd->ProcessData.pDataBuffer[0], &pProcess->pDataBuffer[0], pProcess->nDataLen);
  237. CCriticalSection::Owner locker(g_MainMsgListMutex);
  238. g_MainThreadCmdList.push_back(pNewCmd);
  239. pClientUser->m_nProcessLoadCount ++;
  240. return 1;
  241. }
  243. int KNetServer::Send(unsigned long nId)
  244. {
  245. KClientUser * pClient = m_ClientUserSet[nId];
  246. CBuffer  * pSendBuffer = NULL;
  247. if (pClient)
  248. {
  249. if (pClient->m_CurrentProtocol)
  250. {
  251. pSendBuffer = pClient->m_SendPackager.GetNextPack( pClient->m_CurrentProtocol);
  252. if (pSendBuffer)
  253. {
  254. int nLen = pSendBuffer->GetUsed();
  255. }
  256. }
  258. if ((!pSendBuffer) || !pClient->m_CurrentProtocol) 
  259. {
  260. if (pClient->m_CurrentProtocol && pClient->m_CurrentProtocol < 32)
  261. pClient->m_SendPackager.DelData(pClient->m_CurrentProtocol);
  263. pClient->m_CurrentProtocol = 0;
  265. CCriticalSection::Owner locker(g_NetMsgListMutex);
  266. list<TCmd *>::iterator I = g_NetServiceThreadCmdList.begin();
  268. while (I != g_NetServiceThreadCmdList.end())
  269. {
  270. int nIdd = (*I)->ulNetId;
  271. if ((*I)->ulNetId == nId)
  272. {
  273. if ((*I)->ProcessData.nProtoId >= 32)
  274. {
  275. TProcessData * pProc = (TProcessData *)&(*I)->ProcessData;
  276. m_pNetServer->SendData(nId, &(*I)->ProcessData, (*I)->ProcessData.nDataLen + sizeof(TProcessData) - 1);
  277. pClient->m_ulSendBufferCount +=  (*I)->ProcessData.nDataLen + sizeof(TProcessData) - 1;
  278. g_dwSendLen = pClient->m_ulSendBufferCount;
  279. pClient->m_CurrentProtocol = 0;
  280. pSendBuffer = NULL;
  281. }
  282. else
  283. {
  284. pClient->m_CurrentProtocol = (*I)->ProcessData.nProtoId;
  285. TCmd * pCmd = *I;
  286. TProcessData * pTestData = (TProcessData *)&(*I)->ProcessData;
  287. pClient->m_SendPackager.AddData((*I)->ProcessData.nProtoId, (const char *)&(*I)->ProcessData, (*I)->ProcessData.nDataLen + sizeof(TProcessData) - 1, (*I)->ProcessData.ulIdentity );
  288. pSendBuffer = pClient->m_SendPackager.GetHeadPack(pClient->m_CurrentProtocol);
  289. }
  290. delete [](*I);
  291. g_NetServiceThreadCmdList.erase(I);
  292. pClient->m_nProcessWriteCount ++;
  293. break;
  294. }
  295. I ++;
  296. }
  297. }
  299. /* TRoleData * pRoleData = pBuffer;
  300. OnlineGameLib::Win32::CPackager m_Packager;
  301. TProcessData * pData = (TProcessData *)new char[pRoleData->dwDataLen + sizeof(TProcessData) -1 + 1];
  302. pData->nProtoId = c2s_roleserver_saveroleinfo;
  303. pData->pDataBuffer[0] = create_role;//or pData[0] = save_role    
  304. pData->nDataLen = pRoleData->dwDataLen + 1 ;
  305. pData->ulIdentity = id;
  306. memcpy(&pData->pDataBuffer[1], pRoleData, pRoleData->dwDataLen);
  308. m_Packager.AddData(pData->nProtoId, pData, pData->nDataLen + sizeof(TProcessData) -1, id);
  309. CBuffer * pBuffer = m_Packager.GetHeadPack(pData->nProtoId);
  311. while(pBuffer)
  312. {
  313. Send(pBuffer->GetBuffer());
  314. SAFE_RELEASE(pBuffer);
  315. pBuffer = m_Packager.GetNextPack(pData->nProtoId);
  316. }
  318. m_Packager.DelData(pData->nProtoId);
  319. */
  321. if (pSendBuffer)
  322. {
  323. m_pNetServer->SendData(nId, pSendBuffer->GetBuffer(), pSendBuffer->GetUsed());
  324. pClient->m_ulSendBufferCount += pSendBuffer->GetUsed();
  325. g_dwSendLen = pClient->m_ulSendBufferCount;
  327. if (pClient->m_ulSendBufferCount >= 0xFFFF1FFF)
  328. pClient->m_ulSendBufferCount = 0;
  329. SAFE_RELEASE(pSendBuffer);
  330. g_dwSendLen = pClient->m_ulSendBufferCount;
  331. }
  332. }
  333. return 1;
  334. }
  335. }