IClientImpl.cpp 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607
  1. #include "StdAfx.h"
  2. #include "IClientImpl.h"
  3. #include <comdef.h>
  4. #include <atlbase.h>
  5. #include "ThreadPool.hpp"
  6. const int AF_IPV4 = 0;
  7. const int AF_IPV6 = 1;
  8. const int SOCK_TCP = SOCK_STREAM-1;
  9. const int SOCK_UDP = SOCK_DGRAM-1;
  10. //IClientImpl* IClientImpl::m_pTcpClient[TCPCLIENTNUM] = {NULL};
  11. IClientImpl::IClientImpl():m_nMode(AF_IPV4)
  12. ,m_nSockType(SOCK_TCP)
  13. {
  14. m_bSocket = FALSE;
  15. m_bStopbeat = FALSE;
  16. m_SocketClient.SetInterface(this);
  17. }
  18. IClientImpl::~IClientImpl()
  19. {
  20. //m_SocketClient.Terminate();
  21. DisConnectServer();
  22. }
  23. void IClientImpl::GetAddress(const SockAddrIn& addrIn, CString& rString) const
  24. {
  25. TCHAR szIPAddr[MAX_PATH] = { 0 };
  26. CSocketHandle::FormatIP(szIPAddr, MAX_PATH, addrIn);
  27. rString.Format(_T("%s : %d"), szIPAddr, static_cast<int>(static_cast<UINT>(ntohs(addrIn.GetPort()))) );
  28. }
  29. void IClientImpl::AppendText(LPCTSTR lpszFormat, ...)
  30. {
  31. // if ( !::IsWindow(m_ctlMsgList.GetSafeHwnd()) ) return;
  32. // TCHAR szBuffer[512];
  33. // HWND hWnd = m_ctlMsgList.GetSafeHwnd();
  34. // DWORD dwResult = 0;
  35. // if (SendMessageTimeout(hWnd, WM_GETTEXTLENGTH, 0, 0, SMTO_NORMAL, 500L, &dwResult) != 0)
  36. // {
  37. // int nLen = (int) dwResult;
  38. // if (SendMessageTimeout(hWnd, EM_SETSEL, nLen, nLen, SMTO_NORMAL, 500L, &dwResult) != 0)
  39. // {
  40. // size_t cb = 0;
  41. // va_list args;
  42. // va_start(args, lpszFormat);
  43. // ::StringCchVPrintfEx(szBuffer, 512, NULL, &cb, 0, lpszFormat, args);
  44. // va_end(args);
  45. // SendMessageTimeout(hWnd, EM_REPLACESEL, FALSE, reinterpret_cast<LPARAM>(szBuffer), SMTO_NORMAL, 500L, &dwResult);
  46. // }
  47. // }
  48. }
  49. bool IClientImpl::GetDestination(SockAddrIn& addrIn) const
  50. {
  51. CString strPort;
  52. int nFamily = (m_nMode == AF_IPV4) ? AF_INET : AF_INET6;
  53. return addrIn.CreateFrom(NULL, strPort, nFamily);
  54. }
  55. bool IClientImpl::SetupMCAST()
  56. {
  57. const TCHAR szIPv4MCAST[] = TEXT("239.121.1.2");
  58. const TCHAR szIPv6MCAST[] = TEXT("FF02:0:0:0:0:0:0:1"); // All Nodes local address
  59. bool result = false;
  60. if ( m_nSockType == SOCK_UDP )
  61. {
  62. if ( m_nMode == AF_IPV4 ) {
  63. result = m_SocketClient->AddMembership(szIPv4MCAST, NULL);
  64. } else {
  65. result = m_SocketClient->AddMembership(szIPv6MCAST, NULL);
  66. HRESULT hr = HRESULT_FROM_WIN32(GetLastError());
  67. hr = hr;
  68. }
  69. }
  70. return result;
  71. }
  72. ///////////////////////////////////////////////////////////////////////////////
  73. void IClientImpl::OnThreadBegin(CSocketHandle* pSH)
  74. {
  75. ASSERT( pSH == m_SocketClient );
  76. (pSH);
  77. CString strAddr;
  78. SockAddrIn sockAddr;
  79. m_SocketClient->GetSockName(sockAddr);
  80. GetAddress( sockAddr, strAddr );
  81. InitializeCriticalSection(&pSH->m_hClient2SrvSection);
  82. }
  83. void IClientImpl::OnThreadExit(CSocketHandle* pSH)
  84. {
  85. ASSERT( pSH == m_SocketClient );
  86. DeleteCriticalSection( &pSH->m_hClient2SrvSection );
  87. (pSH);
  88. }
  89. void IClientImpl::OnConnectionDropped(CSocketHandle* pSH)
  90. {
  91. ASSERT( pSH == m_SocketClient );
  92. (pSH);
  93. AppendText( _T("Connection lost with client.\r\n") );
  94. }
  95. void IClientImpl::OnConnectionError(CSocketHandle* pSH, DWORD dwError)
  96. {
  97. ASSERT( pSH == m_SocketClient );
  98. (pSH);
  99. _com_error err(dwError);
  100. AppendText( _T("Communication Error:\r\n%s\r\n"), err.ErrorMessage() );
  101. }
  102. void IClientImpl::OnDataReceived(CSocketHandle* pSH, const BYTE* pbData, DWORD dwCount, const SockAddrIn& addr)
  103. {
  104. ASSERT( pSH == m_SocketClient );
  105. (pSH);
  106. if( !m_SocketClient->IsOpen() ) return;
  107. // 处理接收回来的数据;
  108. ToprocessRecivebuf(pSH,pbData,dwCount);
  109. }
  110. void IClientImpl::ToprocessRecivebuf(IN CSocketHandle *pSockHandle, IN const BYTE* pReceivebuf, IN DWORD dwReceiveSize)
  111. {
  112. DWORD dwIndexOfProcessed = 0; // 当前接收的包被处理的长度;
  113. DWORD dwOnceProcessedLen = 0; // 循环一次处理了多少当前包;
  114. STProtocolheader *pstProtocolheader;
  115. while( dwIndexOfProcessed < dwReceiveSize)
  116. {
  117. pstProtocolheader = NULL;
  118. dwOnceProcessedLen = 0;
  119. if ( pSockHandle->m_npendingSize == 0) //----------------------------------------------// 1.第一次接收或完整组包后接收剩余包;
  120. {
  121. //pstProtocolheader = (STProtocolheader *)&((unsigned char *)pReceivebuf)[dwIndexOfProcessed];
  122. if( dwReceiveSize - dwIndexOfProcessed < sizeof(STProtocolheader) ) // 1.1.第一次接收或剩余数据不足一个包头;
  123. {
  124. // 如果第一次接包就没有收够包头,认为是非法包,扔掉,就是说已处理的长度dwOnceProcessedLen = 0;
  125. pSockHandle->m_npendingSize = dwReceiveSize - dwIndexOfProcessed;
  126. //memcpy(pSockHandle->m_szpendingbuf, pstProtocolheader, pSockHandle->m_npendingSize);
  127. memcpy(pSockHandle->m_szpendingbuf, &pReceivebuf[dwIndexOfProcessed], pSockHandle->m_npendingSize);
  128. dwOnceProcessedLen = dwReceiveSize - dwIndexOfProcessed;
  129. }
  130. else // 1.2.第一次接收或剩余数据大于一个包头;
  131. {
  132. pstProtocolheader = (STProtocolheader *)&((unsigned char *)pReceivebuf)[dwIndexOfProcessed];
  133. // dwOnceProcessedLen = pstProtocolheader->nDataLen;
  134. if( (int)pstProtocolheader->nDataLen > dwReceiveSize - dwIndexOfProcessed ) // 1.2.1.第一次接收或剩余数据不足一个完整的协议包;
  135. {
  136. //memcpy(pSockHandle->m_szpendingbuf, pstProtocolheader, dwReceiveSize - dwIndexOfProcessed);
  137. memcpy(pSockHandle->m_szpendingbuf, &pReceivebuf[dwIndexOfProcessed], dwReceiveSize - dwIndexOfProcessed);
  138. //如果第一次接包,pstProtocolheader->nDataLen大于当前包的总长,认为是非法包,扔掉(不处理这部分,不一定是非法包)
  139. //if( dwIndexOfProcessed == 0 )
  140. //{
  141. //组包错误,则扔掉当前包
  142. //LOG4C((LOG_NOTICE, "第一次接包,服务器pstProtocolheader->nDataLen大于当前包的总长,认为是非法包,扔掉\r\n"));
  143. //}
  144. pSockHandle->m_npendingSize = dwReceiveSize - dwIndexOfProcessed;
  145. dwOnceProcessedLen = dwReceiveSize - dwIndexOfProcessed; // 该次处理的数据长度;
  146. }
  147. else // ---------------------------------------------------------------------// 1.2.2.第一次接收或剩余数据大于一个完整的协议包;
  148. {
  149. //LOG4C((LOG_NOTICE, "正常包"));
  150. pSockHandle->m_npendingSize = 0;
  151. dwOnceProcessedLen = pstProtocolheader->nDataLen; // 该次处理的数据长度;
  152. }
  153. }
  154. }
  155. else //------------------------------------------------------------------------------// 2.第n+1次接收数据包;
  156. {
  157. if( pSockHandle->m_npendingSize < sizeof(STProtocolheader) ) // 2.1.pengingbuf数据小于包头
  158. {
  159. DWORD dwRestheader = sizeof(STProtocolheader) - pSockHandle->m_npendingSize;
  160. if( dwRestheader < dwReceiveSize ) // 2.1.1.当前包大小 > 剩下包头长度,可以组成一个完整包头;
  161. {
  162. // -1.先收完一个完整的包头;
  163. memcpy( &pSockHandle->m_szpendingbuf[pSockHandle->m_npendingSize], pReceivebuf, dwRestheader);
  164. // -2.处理剩余数据;
  165. pstProtocolheader = (STProtocolheader *)pSockHandle->m_szpendingbuf;
  166. dwOnceProcessedLen = pstProtocolheader->nDataLen - pSockHandle->m_npendingSize;
  167. if( dwOnceProcessedLen <= dwReceiveSize ) // 可以组成一个完整的协议包;
  168. {
  169. memcpy( &pSockHandle->m_szpendingbuf[sizeof(STProtocolheader)],
  170. &((char *)pReceivebuf)[dwRestheader],
  171. pstProtocolheader->nDataLen - sizeof(STProtocolheader));
  172. pSockHandle->m_npendingSize = 0;
  173. }
  174. else // 未能组成一个完整的协议包,断续接收等待;
  175. {
  176. //int nTemp = dwReceiveSize - dwRestheader; //除去头剩余部分的长度
  177. //if ( nTemp > 0 ) //刚好是Header的长度,不用拷贝内存,所以这里加了>0的判断
  178. //{
  179. //memcpy( &pSockHandle->m_szpendingbuf[sizeof(STProtocolheader)],&((char *)pReceivebuf)[dwRestheader],nTemp );
  180. //}
  181. memcpy( &pSockHandle->m_szpendingbuf[sizeof(STProtocolheader)],&((char *)pReceivebuf)[dwRestheader],dwReceiveSize - dwRestheader);
  182. pSockHandle->m_npendingSize += dwReceiveSize;
  183. }
  184. }
  185. else //------------------------------------------------------// 2.1.2.当前包大小 <= 剩下包头长度,未能或刚好组成一个完整的包头;
  186. {
  187. memcpy( &pSockHandle->m_szpendingbuf[pSockHandle->m_npendingSize], pReceivebuf, dwReceiveSize );
  188. pSockHandle->m_npendingSize += dwReceiveSize;
  189. dwOnceProcessedLen = dwReceiveSize;
  190. }
  191. }
  192. else // --------------------------------------------------------// 2.2.pengingbuf数据大于包头;
  193. {
  194. pstProtocolheader = (STProtocolheader *)pSockHandle->m_szpendingbuf;
  195. dwOnceProcessedLen = pstProtocolheader->nDataLen - pSockHandle->m_npendingSize;
  196. if ( dwOnceProcessedLen <= dwReceiveSize ) // 可以组成一个完整的协议包;
  197. {
  198. memcpy( &pSockHandle->m_szpendingbuf[pSockHandle->m_npendingSize], pReceivebuf, dwOnceProcessedLen );
  199. pSockHandle->m_npendingSize = 0;
  200. }
  201. else // 未能组成一个完整协议包;
  202. {
  203. memcpy( &pSockHandle->m_szpendingbuf[pSockHandle->m_npendingSize], pReceivebuf, dwReceiveSize );
  204. pSockHandle->m_npendingSize += dwReceiveSize;
  205. }
  206. }
  207. }
  208. if ( dwOnceProcessedLen == 0 )
  209. {
  210. // 没有收够包头,认为是非法包,扔掉
  211. LOG4C((LOG_NOTICE, "没有收够包头,认为是非法包,扔掉"));
  212. break;
  213. }
  214. if ( pSockHandle->m_npendingSize == 0 )
  215. {
  216. if ( pstProtocolheader->nDataLen > SOCKET_BUFFSIZE )
  217. {
  218. // 包长度超过限制,暂时不处理;
  219. LOG4C((LOG_NOTICE, "pstProtocolheader->nDataLen超过限制"));
  220. }
  221. if(-1 == OnIntegrityPacket(pSockHandle, pstProtocolheader))
  222. {
  223. LOG4C((LOG_NOTICE, "Error OnIntegrityPacket"));
  224. break;
  225. }
  226. }
  227. dwIndexOfProcessed += dwOnceProcessedLen;
  228. }
  229. }
  230. int IClientImpl::OnIntegrityPacket(IN CSocketHandle *pSockHandle, IN void *pIntegrityPacket)
  231. {
  232. STProtocolheader *pHeader = (STProtocolheader *)pIntegrityPacket;
  233. if( pHeader == NULL || pHeader->bof != PBOF)
  234. {
  235. LOG4C((LOG_NOTICE,"协议头标识错误"));
  236. return -1;
  237. }
  238. if( pHeader->nDataLen < 0 || pHeader->nDataLen > 65535 )
  239. {
  240. LOG4C((LOG_NOTICE,"协议长度越界错误"));
  241. return -1;
  242. }
  243. unsigned int tmp = VerifyIntegrityPacket(pHeader, pHeader->nDataLen);
  244. if(tmp != pHeader->nVerify)
  245. {
  246. LOG4C((LOG_NOTICE,"协议校验值错误:len=%d,%d != %d",pHeader->nDataLen,tmp,pHeader->nVerify));
  247. return -1;
  248. }
  249. switch (pHeader->nCmd)
  250. {
  251. case CMD_HEART: // 公共命令:心跳包
  252. {
  253. // 心跳包,不处理任务事务;
  254. //LOG4C((LOG_NOTICE,"接收到的心跳包"));
  255. }
  256. break;
  257. case CMD_TOCHAT: // 公共命令:文字聊天;
  258. {
  259. STChatbody *tChatbody = (STChatbody*)pIntegrityPacket;
  260. //TRACE("------------------");
  261. //AfxMessageBox(tChatbody->szChat);
  262. //LOG4C((LOG_NOTICE,"聊天内容:%s",tChatbody->szChat));
  263. }
  264. break;
  265. case CMD_DATABASEINFO:
  266. {
  267. // 1.接到服务器返回的数据库连接信息;
  268. // 2.连接数据库;
  269. // 同时,此处做为全局开关,只有获取到了数据库信息才断续其他操作;
  270. // 定义一个全局标识 g_bSuccess;才能启动重连线程的心跳机制等等;
  271. STDatabaseInfobody *pDatabaseInfobody = (STDatabaseInfobody*)pIntegrityPacket;
  272. LOG4C((LOG_NOTICE,"数据库源:%s;数据库源端口:%s;数据库用户:%s;数据库密码:%s;数据库名称:%s",
  273. pDatabaseInfobody->szDatabaseServer,
  274. pDatabaseInfobody->szDatabaseTCPPort,
  275. pDatabaseInfobody->szDatabaseAccount,
  276. pDatabaseInfobody->szDatabasePassword,
  277. pDatabaseInfobody->szDatabaseName));
  278. }
  279. break;
  280. default:
  281. return 0;
  282. }
  283. return 0;
  284. }
  285. //unsigned int IClientImpl::VerityIntegrityPacket(IN void *pIntegrityPacket,unsigned int nPacketSize)
  286. //{
  287. // unsigned int checksum = 0;
  288. // //if ( nPacketSize <= sizeof(STProtocolheader) )
  289. // //{
  290. // // return 0;
  291. // //}
  292. // unsigned char *pBody = &((unsigned char*)pIntegrityPacket)[sizeof(STProtocolheader)];
  293. //
  294. // if( pBody )
  295. // checksum = crc32( 0, pBody, nPacketSize-sizeof(STProtocolheader) );
  296. //
  297. // return checksum;
  298. //}
  299. BOOL IClientImpl::Initialize()
  300. {
  301. TCHAR szIPAddr[MAX_PATH] = { 0 };
  302. CSocketHandle::GetLocalAddress(szIPAddr, MAX_PATH, AF_INET);
  303. AppendText(_T("Local Address (IPv4): %s\r\n"), szIPAddr);
  304. CSocketHandle::GetLocalAddress(szIPAddr, MAX_PATH, AF_INET6);
  305. AppendText(_T("Local Address (IPv6): %s\r\n"), szIPAddr);
  306. return TRUE;
  307. }
  308. void IClientImpl::StartReConnectSrvThread()
  309. {
  310. // Jeff.启用重连服务端线程.-------------------
  311. m_hRunObject = CreateEvent( NULL, TRUE, FALSE, "ClientThreadRun" );
  312. if ( m_hRunObject == NULL )
  313. {
  314. LOG4C((LOG_NOTICE,"创建事件失败"));
  315. }
  316. m_hReConnectSrvThreadHandle = CreateThread(NULL,0,ReConnectSrvThread,this,0,NULL);
  317. if ( m_hReConnectSrvThreadHandle == NULL )
  318. {
  319. LOG4C((LOG_NOTICE,"创建线程失败"));
  320. }
  321. }
  322. BOOL IClientImpl::ConnectServer(LPCTSTR strAddr, LPCTSTR strPort)
  323. {
  324. int nFamily = (m_nMode == AF_IPV4) ? AF_INET : AF_INET6;
  325. if ( !m_SocketClient.StartClient(NULL, strAddr, strPort, nFamily, (m_nSockType+1) ) )
  326. {
  327. m_bSocket = FALSE;
  328. LOG4C((LOG_NOTICE, "连接失败 [%s, %s, %d, %d]",strAddr, strPort, nFamily, (m_nSockType+1)));
  329. return FALSE;
  330. }
  331. else
  332. {
  333. m_bSocket = TRUE;
  334. CSocketHandle* pSH = (CSocketHandle *)m_SocketClient;
  335. pSH->m_npendingSize = 0;
  336. memset(pSH->m_szpendingbuf, 0, SOCKET_BUFFSIZE);
  337. SetupMCAST();
  338. LOG4C((LOG_NOTICE, "连接成功 [%s, %s, %d, %d]",strAddr, strPort, nFamily, (m_nSockType+1)));
  339. return TRUE;
  340. }
  341. }
  342. void IClientImpl::DisConnectServer()
  343. {
  344. if(m_hRunObject)
  345. SetEvent(m_hRunObject);
  346. if( m_hReConnectSrvThreadHandle )
  347. {
  348. if (WaitForSingleObject(m_hReConnectSrvThreadHandle,INFINITE) != WAIT_FAILED)
  349. {
  350. CloseHandle(m_hReConnectSrvThreadHandle);
  351. m_hReConnectSrvThreadHandle = NULL;
  352. }
  353. }
  354. CloseHandle( m_hRunObject );
  355. m_hRunObject = NULL;
  356. m_SocketClient.Terminate();
  357. }
  358. void IClientImpl::SendMsg(void *pMsg,const int nLen)
  359. {
  360. if ( m_SocketClient.IsOpen() )
  361. {
  362. USES_CONVERSION;
  363. if (m_nSockType == SOCK_TCP)
  364. {
  365. m_SocketClient.Write((const LPBYTE)(pMsg), nLen, NULL);
  366. }
  367. else
  368. {
  369. SockAddrIn sockAddr;
  370. GetDestination(sockAddr);
  371. m_SocketClient.Write((const LPBYTE)(pMsg), nLen, sockAddr);
  372. }
  373. }
  374. else
  375. {
  376. AfxMessageBox(_T("Socket is not connected"));
  377. }
  378. }
  379. // 静态成员函数,提供全局访问的接口
  380. //IClientImpl* IClientImpl::GetInstancePtr( int iTCPIndex )
  381. //{
  382. // if( NULL == m_pTcpClient[iTCPIndex] )
  383. // {
  384. // m_pTcpClient[iTCPIndex] = new IClientImpl();
  385. // }
  386. //
  387. // return m_pTcpClient[iTCPIndex];
  388. //}
  389. DWORD WINAPI IClientImpl::ReConnectSrvThread(LPVOID pInstance)
  390. {
  391. LOG4C((LOG_NOTICE,"重连服务器线程"));
  392. IClientImpl *pClientImpl = (IClientImpl*)pInstance;
  393. #if 0
  394. STProtocolheader tProtocolheader;
  395. tProtocolheader.nCmd = CMD_HEART;
  396. tProtocolheader.nCmdType = 1;
  397. tProtocolheader.nDataLen = sizeof(STProtocolheader);
  398. //tProtocolheader.nVerify = crc32(0, reinterpret_cast<const unsigned char*>(&tProtocolheader), sizeof(STProtocolheader));
  399. tProtocolheader.nVerify = VerityIntegrityPacket(&tProtocolheader, sizeof(STProtocolheader));
  400. #else
  401. STChatbody tChatbody;
  402. tChatbody.tPHeader.nCmd = CMD_TOCHAT;
  403. tChatbody.tPHeader.nCmdType = 0;
  404. tChatbody.tPHeader.nDataLen = sizeof(STChatbody);
  405. memset(tChatbody.szChat,98,MAX_CHATLENGTH-3);
  406. tChatbody.szChat[MAX_CHATLENGTH-3] = 'C';
  407. tChatbody.szChat[MAX_CHATLENGTH-2] = 'D';
  408. tChatbody.szChat[MAX_CHATLENGTH-1] = '\0';
  409. //tChatbody.tPHeader.nVerify = VerifyIntegrityPacket(&tChatbody, sizeof(STChatbody));
  410. tChatbody.GetVerify();
  411. #endif
  412. do
  413. {
  414. #if 0 // 心跳包;
  415. if( !pClientImpl->m_bStopbeat && pClientImpl->m_bSocket )
  416. {
  417. //LOG4C((LOG_NOTICE,"发送心跳包"));
  418. USES_CONVERSION;
  419. if( pClientImpl->m_SocketClient->Write((const LPBYTE)&tProtocolheader,sizeof(STProtocolheader)) == -1)
  420. pClientImpl->m_bSocket = FALSE;
  421. }
  422. #else
  423. if( !pClientImpl->m_bStopbeat && pClientImpl->m_bSocket )
  424. {
  425. //LOG4C((LOG_NOTICE,"发送心跳包"));
  426. //USES_CONVERSION;
  427. if( pClientImpl->m_SocketClient->Write((const LPBYTE)&tChatbody,sizeof(STChatbody)) == -1)
  428. {
  429. pClientImpl->m_bSocket = FALSE;
  430. LOG4C((LOG_NOTICE,"发送心跳包失败"));
  431. }
  432. }
  433. #endif
  434. // 检测连接状态;
  435. if ( pClientImpl->m_bSocket == FALSE )
  436. {
  437. if ( pClientImpl->m_SocketClient->IsOpen() == TRUE )
  438. {
  439. pClientImpl->m_SocketClient.Terminate();
  440. }
  441. if ( FALSE == pClientImpl->ConnectServer(_T(g_szServerIP),_T(g_szCmdPort)))
  442. {
  443. LOG4C((LOG_NOTICE,"重连服务器失败"));
  444. pClientImpl->m_bSocket = FALSE;
  445. }
  446. else
  447. {
  448. LOG4C((LOG_NOTICE,"重连服务器成功"));
  449. pClientImpl->m_bSocket = TRUE;
  450. }
  451. }
  452. } while( WaitForSingleObject(pClientImpl->m_hRunObject,200L) == WAIT_TIMEOUT );
  453. return 0;
  454. }
  455. #if 1
  456. void IClientImpl::SendFile(LPCTSTR lpzFileName,LPCTSTR strPort)
  457. {
  458. CFile file;
  459. CFileException e;
  460. file.Open(lpzFileName, CFile::modeRead|CFile::shareDenyNone, &e);
  461. if (e.m_cause != 0)
  462. {
  463. LOG4C((LOG_NOTICE,"读取文件失败"));
  464. file.Close();
  465. return;
  466. }
  467. // 1 发送文件信息;
  468. LOG4C((LOG_NOTICE,"发送文件信息"));
  469. STFileInfobody tfInfo;
  470. GetFileName(lpzFileName,tfInfo.szFileName);
  471. tfInfo.nFileLength = file.GetLength();
  472. //tfInfo.tPHeader.nCmd = CMD_FILEINFO_TRANSFER;
  473. //tfInfo.tPHeader.nCmdType = 1;
  474. CFileStatus FileStatus ;
  475. if (file.GetStatus(FileStatus) != FALSE)
  476. {
  477. tfInfo.tCreateDate = FileStatus.m_ctime.GetTime();
  478. tfInfo.tModifyDate = FileStatus.m_mtime.GetTime();
  479. }
  480. //tfInfo.tPHeader.nDataLen = sizeof(STFileInfobody);// - sizeof(STProtocolheader);
  481. //tfInfo.tPHeader.nVerify = crc32(0,reinterpret_cast<const unsigned char*>(&tfInfo),sizeof(STFileInfobody));
  482. //tfInfo.tPHeader.nVerify = VerifyIntegrityPacket(&tfInfo,sizeof(STFileInfobody));
  483. tfInfo.GetVerify();
  484. LOG4C((LOG_NOTICE,"文件信息:协议标识:%d,协议命令:%d,协议类型:%d,协议长度:%d,协议校验:%d,文件大小:%d",
  485. tfInfo.tPHeader.bof,
  486. tfInfo.tPHeader.nCmd,
  487. tfInfo.tPHeader.nCmdType,
  488. tfInfo.tPHeader.nDataLen,
  489. tfInfo.tPHeader.nVerify,
  490. tfInfo.nFileLength));
  491. m_bStopbeat = TRUE;
  492. if( m_bSocket )
  493. {
  494. USES_CONVERSION;
  495. if ( m_SocketClient->Write((const LPBYTE)&tfInfo,sizeof(STFileInfobody)) == -1)
  496. {
  497. LOG4C((LOG_NOTICE,"发送文件信息失败"));
  498. m_bStopbeat = FALSE;
  499. return;
  500. }
  501. }
  502. //return ;
  503. Sleep(100); // 异步传输时这里的文件传输会出现发包乱序;
  504. // 2 发送文件数据;
  505. LOG4C((LOG_NOTICE,"发送文件数据"));
  506. STFileContextbody tFileContextbody;
  507. //tFileContextbody.tPHeader.nCmd = CMD_FILECONT_TRANSFER;
  508. //tFileContextbody.tPHeader.nCmdType = 0;
  509. //tFileContextbody.tPHeader.nDataLen = sizeof(STFileContextbody);
  510. do
  511. {
  512. USES_CONVERSION;
  513. //memset(tFileContextbody.szFileContext,0,MAX_FILETRANSFERLENGTH);
  514. tFileContextbody.nFileContextLen = file.Read(tFileContextbody.szFileContext,MAX_FILETRANSFERLENGTH);
  515. //tFileContextbody.tPHeader.nVerify = VerifyIntegrityPacket(&tFileContextbody,sizeof(STFileContextbody));
  516. tFileContextbody.GetVerify();
  517. LOG4C((LOG_NOTICE,"文件内容:协议标识:%d,协议命令:%d,协议类型:%d,协议长度:%d,协议校验:%d",
  518. tFileContextbody.tPHeader.bof,
  519. tFileContextbody.tPHeader.nCmd,
  520. tFileContextbody.tPHeader.nCmdType,
  521. tFileContextbody.tPHeader.nDataLen,
  522. tFileContextbody.tPHeader.nVerify));
  523. if ( m_SocketClient->Write((const LPBYTE)&tFileContextbody,sizeof(STFileContextbody)) == -1)
  524. {
  525. LOG4C((LOG_NOTICE,"文件内容传输出错"));
  526. break;
  527. }
  528. if ( tFileContextbody.nFileContextLen < MAX_FILETRANSFERLENGTH )
  529. {
  530. break;
  531. }
  532. tfInfo.nFileLength -= MAX_FILETRANSFERLENGTH;
  533. //Sleep(200);
  534. } while (tfInfo.nFileLength);
  535. LOG4C((LOG_NOTICE,"发送文件结束"));
  536. m_bStopbeat = FALSE;
  537. }
  538. #endif