IClientImpl.cpp 18 KB

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