123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595 |
- #include "StdAfx.h"
- #include "IClientImpl.h"
- #include <comdef.h>
- #include <atlbase.h>
- #include "ThreadPool.hpp"
- const int AF_IPV4 = 0;
- const int AF_IPV6 = 1;
- const int SOCK_TCP = SOCK_STREAM-1;
- const int SOCK_UDP = SOCK_DGRAM-1;
- IClientImpl::IClientImpl():m_nMode(AF_IPV4)
- ,m_nSockType(SOCK_TCP)
- {
- m_bSocket = FALSE;
- m_bStopbeat = FALSE;
- m_SocketClient.SetInterface(this);
- }
- IClientImpl::~IClientImpl()
- {
-
- DisConnectServer();
- }
- void IClientImpl::GetAddress(const SockAddrIn& addrIn, CString& rString) const
- {
- TCHAR szIPAddr[MAX_PATH] = { 0 };
- CSocketHandle::FormatIP(szIPAddr, MAX_PATH, addrIn);
- rString.Format(_T("%s : %d"), szIPAddr, static_cast<int>(static_cast<UINT>(ntohs(addrIn.GetPort()))) );
- }
- void IClientImpl::AppendText(LPCTSTR lpszFormat, ...)
- {
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- }
- bool IClientImpl::GetDestination(SockAddrIn& addrIn) const
- {
- CString strPort;
- int nFamily = (m_nMode == AF_IPV4) ? AF_INET : AF_INET6;
- return addrIn.CreateFrom(NULL, strPort, nFamily);
- }
- bool IClientImpl::SetupMCAST()
- {
- const TCHAR szIPv4MCAST[] = TEXT("239.121.1.2");
- const TCHAR szIPv6MCAST[] = TEXT("FF02:0:0:0:0:0:0:1");
- bool result = false;
- if ( m_nSockType == SOCK_UDP )
- {
- if ( m_nMode == AF_IPV4 ) {
- result = m_SocketClient->AddMembership(szIPv4MCAST, NULL);
- } else {
- result = m_SocketClient->AddMembership(szIPv6MCAST, NULL);
- HRESULT hr = HRESULT_FROM_WIN32(GetLastError());
- hr = hr;
- }
- }
- return result;
- }
- void IClientImpl::OnThreadBegin(CSocketHandle* pSH)
- {
- ASSERT( pSH == m_SocketClient );
- (pSH);
- CString strAddr;
- SockAddrIn sockAddr;
- m_SocketClient->GetSockName(sockAddr);
- GetAddress( sockAddr, strAddr );
- InitializeCriticalSection(&pSH->m_hClient2SrvSection);
- }
- void IClientImpl::OnThreadExit(CSocketHandle* pSH)
- {
- ASSERT( pSH == m_SocketClient );
- DeleteCriticalSection( &pSH->m_hClient2SrvSection );
- (pSH);
- }
- void IClientImpl::OnConnectionDropped(CSocketHandle* pSH)
- {
- ASSERT( pSH == m_SocketClient );
- (pSH);
- AppendText( _T("Connection lost with client.\r\n") );
- }
- void IClientImpl::OnConnectionError(CSocketHandle* pSH, DWORD dwError)
- {
- ASSERT( pSH == m_SocketClient );
- (pSH);
- _com_error err(dwError);
- AppendText( _T("Communication Error:\r\n%s\r\n"), err.ErrorMessage() );
- }
- void IClientImpl::OnDataReceived(CSocketHandle* pSH, const BYTE* pbData, DWORD dwCount, const SockAddrIn& addr)
- {
- ASSERT( pSH == m_SocketClient );
- (pSH);
- if( !m_SocketClient->IsOpen() ) return;
-
- ToprocessRecivebuf(pSH,pbData,dwCount);
- }
- void IClientImpl::ToprocessRecivebuf(IN CSocketHandle *pSockHandle, IN const BYTE* pReceivebuf, IN DWORD dwReceiveSize)
- {
- DWORD dwIndexOfProcessed = 0;
- DWORD dwOnceProcessedLen = 0;
- STProtocolheader *pstProtocolheader;
- while( dwIndexOfProcessed < dwReceiveSize)
- {
- pstProtocolheader = NULL;
- dwOnceProcessedLen = 0;
- if ( pSockHandle->m_npendingSize == 0)
- {
- pstProtocolheader = (STProtocolheader *)&((unsigned char *)pReceivebuf)[dwIndexOfProcessed];
- if( dwReceiveSize - dwIndexOfProcessed < sizeof(STProtocolheader) )
- {
-
- pSockHandle->m_npendingSize = dwReceiveSize - dwIndexOfProcessed;
- memcpy(pSockHandle->m_szpendingbuf, pstProtocolheader, pSockHandle->m_npendingSize);
- }
- else
- {
-
- if( (int)pstProtocolheader->nDataLen > dwReceiveSize - dwIndexOfProcessed )
- {
- memcpy(pSockHandle->m_szpendingbuf, pstProtocolheader, dwReceiveSize - dwIndexOfProcessed);
-
- if( dwIndexOfProcessed == 0 )
- {
-
-
- }
- pSockHandle->m_npendingSize = dwReceiveSize - dwIndexOfProcessed;
- dwOnceProcessedLen = dwReceiveSize - dwIndexOfProcessed;
- }
- else
- {
-
- pSockHandle->m_npendingSize = 0;
- dwOnceProcessedLen = pstProtocolheader->nDataLen;
- }
- }
- }
- else
- {
- pstProtocolheader = (STProtocolheader *)pSockHandle->m_szpendingbuf;
- if( pSockHandle->m_npendingSize < sizeof(STProtocolheader) )
- {
- DWORD dwRestheader = sizeof(STProtocolheader) - pSockHandle->m_npendingSize;
- if( dwRestheader < dwReceiveSize )
- {
-
- memcpy( &pSockHandle->m_szpendingbuf[pSockHandle->m_npendingSize], pReceivebuf, dwRestheader);
-
- dwOnceProcessedLen = pstProtocolheader->nDataLen - pSockHandle->m_npendingSize;
- if( dwOnceProcessedLen <= dwReceiveSize )
- {
- memcpy( &pSockHandle->m_szpendingbuf[sizeof(STProtocolheader)],
- &((char *)pReceivebuf)[dwRestheader],
- pstProtocolheader->nDataLen - sizeof(STProtocolheader));
- }
- else
- {
-
-
-
-
-
- memcpy( &pSockHandle->m_szpendingbuf[sizeof(STProtocolheader)],&((char *)pReceivebuf)[dwRestheader],dwReceiveSize - dwRestheader);
- pSockHandle->m_npendingSize += dwReceiveSize;
- }
- }
- else
- {
- memcpy( &pSockHandle->m_szpendingbuf[pSockHandle->m_npendingSize], pReceivebuf, dwReceiveSize );
- pSockHandle->m_npendingSize += dwReceiveSize;
- dwOnceProcessedLen = dwReceiveSize;
- }
- }
- else
- {
- dwOnceProcessedLen = pstProtocolheader->nDataLen - pSockHandle->m_npendingSize;
- if ( dwOnceProcessedLen <= dwReceiveSize )
- {
- memcpy( &pSockHandle->m_szpendingbuf[pSockHandle->m_npendingSize], pReceivebuf, dwOnceProcessedLen );
- pSockHandle->m_npendingSize = 0;
- }
- else
- {
- memcpy( &pSockHandle->m_szpendingbuf[pSockHandle->m_npendingSize], pReceivebuf, dwReceiveSize );
- pSockHandle->m_npendingSize += dwReceiveSize;
- }
- }
- }
- if ( dwOnceProcessedLen == 0 )
- {
-
-
- break;
- }
- if ( pSockHandle->m_npendingSize == 0 )
- {
- if ( pstProtocolheader->nDataLen > SOCKET_BUFFSIZE )
- {
-
-
- }
- if(-1 == OnIntegrityPacket(pSockHandle, pstProtocolheader))
- {
- LOG4C((LOG_NOTICE, "Error OnIntegrityPacket"));
- break;
- }
- }
- dwIndexOfProcessed += dwOnceProcessedLen;
- }
- }
- int IClientImpl::OnIntegrityPacket(IN CSocketHandle *pSockHandle, IN void *pIntegrityPacket)
- {
- STProtocolheader *pHeader = (STProtocolheader *)pIntegrityPacket;
- if( pHeader == NULL || pHeader->bof != PBOF)
- {
- LOG4C((LOG_NOTICE,"协议头标识错误"));
- return -1;
- }
- if( pHeader->nDataLen < 0 || pHeader->nDataLen > 65535 )
- {
- LOG4C((LOG_NOTICE,"协议长度越界错误"));
- return -1;
- }
- unsigned int tmp = VerifyIntegrityPacket(pHeader, pHeader->nDataLen);
- if(tmp != pHeader->nVerify)
- {
- LOG4C((LOG_NOTICE,"协议校验值错误:len=%d,%d != %d",pHeader->nDataLen,tmp,pHeader->nVerify));
- return -1;
- }
- switch (pHeader->nCmd)
- {
- case CMD_HEART:
- {
-
-
- }
- break;
- case CMD_TOCHAT:
- {
- STChatbody *tChatbody = (STChatbody*)pIntegrityPacket;
-
-
-
- }
- break;
- case CMD_DATABASEINFO:
- {
-
-
-
-
- STDatabaseInfobody *pDatabaseInfobody = (STDatabaseInfobody*)pIntegrityPacket;
-
- LOG4C((LOG_NOTICE,"数据库源:%s;数据库源端口:%s;数据库用户:%s;数据库密码:%s;数据库名称:%s",
- pDatabaseInfobody->szDatabaseServer,
- pDatabaseInfobody->szDatabaseTCPPort,
- pDatabaseInfobody->szDatabaseAccount,
- pDatabaseInfobody->szDatabasePassword,
- pDatabaseInfobody->szDatabaseName));
- }
- break;
- default:
- return 0;
- }
- return 0;
- }
- BOOL IClientImpl::Initialize()
- {
- TCHAR szIPAddr[MAX_PATH] = { 0 };
- CSocketHandle::GetLocalAddress(szIPAddr, MAX_PATH, AF_INET);
- AppendText(_T("Local Address (IPv4): %s\r\n"), szIPAddr);
- CSocketHandle::GetLocalAddress(szIPAddr, MAX_PATH, AF_INET6);
- AppendText(_T("Local Address (IPv6): %s\r\n"), szIPAddr);
- return TRUE;
- }
- void IClientImpl::StartReConnectSrvThread()
- {
-
- m_hRunObject = CreateEvent( NULL, TRUE, FALSE, "ClientThreadRun" );
- if ( m_hRunObject == NULL )
- {
- LOG4C((LOG_NOTICE,"创建事件失败"));
- }
- m_hReConnectSrvThreadHandle = CreateThread(NULL,0,ReConnectSrvThread,this,0,NULL);
- if ( m_hReConnectSrvThreadHandle == NULL )
- {
- LOG4C((LOG_NOTICE,"创建线程失败"));
- }
- }
- BOOL IClientImpl::ConnectServer(LPCTSTR strAddr, LPCTSTR strPort)
- {
- int nFamily = (m_nMode == AF_IPV4) ? AF_INET : AF_INET6;
- if ( !m_SocketClient.StartClient(NULL, strAddr, strPort, nFamily, (m_nSockType+1) ) )
- {
- m_bSocket = FALSE;
- LOG4C((LOG_NOTICE, "连接失败 [%s, %s, %d, %d]",strAddr, strPort, nFamily, (m_nSockType+1)));
- return FALSE;
- }
- else
- {
- m_bSocket = TRUE;
- CSocketHandle* pSH = (CSocketHandle *)m_SocketClient;
- pSH->m_npendingSize = 0;
- memset(pSH->m_szpendingbuf, 0, SOCKET_BUFFSIZE);
- SetupMCAST();
- LOG4C((LOG_NOTICE, "连接成功 [%s, %s, %d, %d]",strAddr, strPort, nFamily, (m_nSockType+1)));
- return TRUE;
- }
- }
- void IClientImpl::DisConnectServer()
- {
- if(m_hRunObject)
- SetEvent(m_hRunObject);
- if( m_hReConnectSrvThreadHandle )
- {
- if (WaitForSingleObject(m_hReConnectSrvThreadHandle,INFINITE) != WAIT_FAILED)
- {
- CloseHandle(m_hReConnectSrvThreadHandle);
- m_hReConnectSrvThreadHandle = NULL;
- }
- }
- CloseHandle( m_hRunObject );
- m_hRunObject = NULL;
- m_SocketClient.Terminate();
- }
- void IClientImpl::SendMsg(void *pMsg,const int nLen)
- {
- if ( m_SocketClient.IsOpen() )
- {
- USES_CONVERSION;
- if (m_nSockType == SOCK_TCP)
- {
- m_SocketClient.Write((const LPBYTE)(pMsg), nLen, NULL);
- }
- else
- {
- SockAddrIn sockAddr;
- GetDestination(sockAddr);
- m_SocketClient.Write((const LPBYTE)(pMsg), nLen, sockAddr);
- }
- }
- else
- {
- AfxMessageBox(_T("Socket is not connected"));
- }
- }
- DWORD WINAPI IClientImpl::ReConnectSrvThread(LPVOID pInstance)
- {
- LOG4C((LOG_NOTICE,"重连服务器线程"));
- IClientImpl *pClientImpl = (IClientImpl*)pInstance;
- #if 0
- STProtocolheader tProtocolheader;
- tProtocolheader.nCmd = CMD_HEART;
- tProtocolheader.nCmdType = 1;
- tProtocolheader.nDataLen = sizeof(STProtocolheader);
-
- tProtocolheader.nVerify = VerityIntegrityPacket(&tProtocolheader, sizeof(STProtocolheader));
- #else
- STChatbody tChatbody;
- tChatbody.tPHeader.nCmd = CMD_TOCHAT;
- tChatbody.tPHeader.nCmdType = 0;
- tChatbody.tPHeader.nDataLen = sizeof(STChatbody);
- memset(tChatbody.szChat,99,MAX_CHATLENGTH);
- tChatbody.szChat[MAX_CHATLENGTH-1] = '\0';
-
- tChatbody.GetVerify();
- #endif
- do
- {
- #if 0
- if( !pClientImpl->m_bStopbeat && pClientImpl->m_bSocket )
- {
-
- USES_CONVERSION;
- if( pClientImpl->m_SocketClient->Write((const LPBYTE)&tProtocolheader,sizeof(STProtocolheader)) == -1)
- pClientImpl->m_bSocket = FALSE;
- }
- #else
- if( !pClientImpl->m_bStopbeat && pClientImpl->m_bSocket )
- {
-
-
- if( pClientImpl->m_SocketClient->Write((const LPBYTE)&tChatbody,sizeof(STChatbody)) == -1)
- pClientImpl->m_bSocket = FALSE;
- }
- #endif
-
- if ( pClientImpl->m_bSocket == FALSE )
- {
- if ( pClientImpl->m_SocketClient->IsOpen() == TRUE )
- {
- pClientImpl->m_SocketClient.Terminate();
- }
- if ( FALSE == pClientImpl->ConnectServer(_T(g_szServerIP),_T(g_szCmdPort)))
- {
- LOG4C((LOG_NOTICE,"重连服务器失败"));
- pClientImpl->m_bSocket = FALSE;
- }
- else
- {
- LOG4C((LOG_NOTICE,"重连服务器成功"));
- pClientImpl->m_bSocket = TRUE;
- }
- }
- } while( WaitForSingleObject(pClientImpl->m_hRunObject,200L) == WAIT_TIMEOUT );
- return 0;
- }
- #if 1
- void IClientImpl::SendFile(LPCTSTR lpzFileName,LPCTSTR strPort)
- {
- CFile file;
- CFileException e;
- file.Open(lpzFileName, CFile::modeRead|CFile::shareDenyNone, &e);
- if (e.m_cause != 0)
- {
- LOG4C((LOG_NOTICE,"读取文件失败"));
- file.Close();
- return;
- }
-
- LOG4C((LOG_NOTICE,"发送文件信息"));
- STFileInfobody tfInfo;
- GetFileName(lpzFileName,tfInfo.szFileName);
- tfInfo.nFileLength = file.GetLength();
-
-
-
- CFileStatus FileStatus ;
- if (file.GetStatus(FileStatus) != FALSE)
- {
- tfInfo.tCreateDate = FileStatus.m_ctime.GetTime();
- tfInfo.tModifyDate = FileStatus.m_mtime.GetTime();
- }
-
-
-
- tfInfo.GetVerify();
- LOG4C((LOG_NOTICE,"文件信息:协议标识:%d,协议命令:%d,协议类型:%d,协议长度:%d,协议校验:%d,文件大小:%d",
- tfInfo.tPHeader.bof,
- tfInfo.tPHeader.nCmd,
- tfInfo.tPHeader.nCmdType,
- tfInfo.tPHeader.nDataLen,
- tfInfo.tPHeader.nVerify,
- tfInfo.nFileLength));
- m_bStopbeat = TRUE;
- if( m_bSocket )
- {
- USES_CONVERSION;
- if ( m_SocketClient->Write((const LPBYTE)&tfInfo,sizeof(STFileInfobody)) == -1)
- {
- LOG4C((LOG_NOTICE,"发送文件信息失败"));
- m_bStopbeat = FALSE;
- return;
- }
- }
-
- Sleep(100);
-
- LOG4C((LOG_NOTICE,"发送文件数据"));
- STFileContextbody tFileContextbody;
-
-
-
-
- do
- {
- USES_CONVERSION;
-
- tFileContextbody.nFileContextLen = file.Read(tFileContextbody.szFileContext,MAX_FILETRANSFERLENGTH);
-
- tFileContextbody.GetVerify();
- LOG4C((LOG_NOTICE,"文件内容:协议标识:%d,协议命令:%d,协议类型:%d,协议长度:%d,协议校验:%d",
- tFileContextbody.tPHeader.bof,
- tFileContextbody.tPHeader.nCmd,
- tFileContextbody.tPHeader.nCmdType,
- tFileContextbody.tPHeader.nDataLen,
- tFileContextbody.tPHeader.nVerify));
- if ( m_SocketClient->Write((const LPBYTE)&tFileContextbody,sizeof(STFileContextbody)) == -1)
- {
- LOG4C((LOG_NOTICE,"文件内容传输出错"));
- break;
- }
-
- if ( tFileContextbody.nFileContextLen < MAX_FILETRANSFERLENGTH )
- {
- break;
- }
- tfInfo.nFileLength -= MAX_FILETRANSFERLENGTH;
-
- } while (tfInfo.nFileLength);
- LOG4C((LOG_NOTICE,"发送文件结束"));
- m_bStopbeat = FALSE;
- }
- #endif
|