/* * Copyright: JessMA Open Source (ldcsaa@gmail.com) * * Author : Bruce Liang * Website : https://github.com/ldcsaa * Project : https://github.com/ldcsaa/HP-Socket * Blog : http://www.cnblogs.com/ldcsaa * Wiki : http://www.oschina.net/p/hp-socket * QQ Group : 44636872, 75375912 * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #pragma once #include "SocketHelper.h" #include "./common/GeneralHelper.h" class CTcpClient : public ITcpClient { public: virtual BOOL Start (LPCTSTR lpszRemoteAddress, USHORT usPort, BOOL bAsyncConnect = TRUE, LPCTSTR lpszBindAddress = nullptr, USHORT usLocalPort = 0); virtual BOOL Stop (); virtual BOOL Send (const BYTE* pBuffer, int iLength, int iOffset = 0); virtual BOOL SendSmallFile (LPCTSTR lpszFileName, const LPWSABUF pHead = nullptr, const LPWSABUF pTail = nullptr); virtual BOOL SendPackets (const WSABUF pBuffers[], int iCount) {return DoSendPackets(pBuffers, iCount);} virtual BOOL PauseReceive (BOOL bPause = TRUE); virtual BOOL Wait (DWORD dwMilliseconds = INFINITE) {return m_evWait.WaitFor(dwMilliseconds, WAIT_FOR_STOP_PREDICATE);} virtual BOOL HasStarted () {return m_enState == SS_STARTED || m_enState == SS_STARTING;} virtual EnServiceState GetState () {return m_enState;} virtual CONNID GetConnectionID () {return m_dwConnID;} virtual EnSocketError GetLastError () {return m_enLastError;} virtual LPCTSTR GetLastErrorDesc () {return ::GetSocketErrorDesc(m_enLastError);} virtual BOOL GetLocalAddress (TCHAR lpszAddress[], int& iAddressLen, USHORT& usPort); virtual BOOL GetRemoteHost (TCHAR lpszHost[], int& iHostLen, USHORT& usPort); virtual BOOL GetPendingDataLength (int& iPending) {iPending = m_lsSend.Length(); return HasStarted();} virtual BOOL IsPauseReceive (BOOL& bPaused) {bPaused = m_bPaused; return HasStarted();} virtual BOOL IsConnected () {return m_bConnected;} #ifdef _SSL_SUPPORT virtual BOOL SetupSSLContext (int iVerifyMode = SSL_VM_NONE, LPCTSTR lpszPemCertFile = nullptr, LPCTSTR lpszPemKeyFile = nullptr, LPCTSTR lpszKeyPassword = nullptr, LPCTSTR lpszCAPemCertFileOrPath = nullptr) {return FALSE;} virtual BOOL SetupSSLContextByMemory(int iVerifyMode = SSL_VM_NONE, LPCSTR lpszPemCert = nullptr, LPCSTR lpszPemKey = nullptr, LPCSTR lpszKeyPassword = nullptr, LPCSTR lpszCAPemCert = nullptr) {return FALSE;} virtual void CleanupSSLContext () {} virtual BOOL StartSSLHandShake () {return FALSE;} virtual void SetSSLAutoHandShake(BOOL bAutoHandShake) {} virtual BOOL IsSSLAutoHandShake () {return FALSE;} virtual void SetSSLCipherList (LPCTSTR lpszCipherList){} virtual LPCTSTR GetSSLCipherList() {return nullptr;} virtual BOOL GetSSLSessionInfo(EnSSLSessionInfo enInfo, LPVOID* lppInfo) {return FALSE;} protected: virtual BOOL StartSSLHandShakeNoCheck() {return FALSE;} #endif public: virtual BOOL IsSecure () {return FALSE;} virtual void SetReuseAddressPolicy (EnReuseAddressPolicy enReusePolicy){ENSURE_HAS_STOPPED(); m_enReusePolicy = enReusePolicy;} virtual void SetSocketBufferSize (DWORD dwSocketBufferSize) {ENSURE_HAS_STOPPED(); m_dwSocketBufferSize = dwSocketBufferSize;} virtual void SetKeepAliveTime (DWORD dwKeepAliveTime) {ENSURE_HAS_STOPPED(); m_dwKeepAliveTime = dwKeepAliveTime;} virtual void SetKeepAliveInterval (DWORD dwKeepAliveInterval) {ENSURE_HAS_STOPPED(); m_dwKeepAliveInterval = dwKeepAliveInterval;} virtual void SetFreeBufferPoolSize (DWORD dwFreeBufferPoolSize) {ENSURE_HAS_STOPPED(); m_dwFreeBufferPoolSize = dwFreeBufferPoolSize;} virtual void SetFreeBufferPoolHold (DWORD dwFreeBufferPoolHold) {ENSURE_HAS_STOPPED(); m_dwFreeBufferPoolHold = dwFreeBufferPoolHold;} virtual void SetNoDelay (BOOL bNoDelay) {ENSURE_HAS_STOPPED(); m_bNoDelay = bNoDelay;} virtual void SetExtra (PVOID pExtra) {m_pExtra = pExtra;} virtual EnReuseAddressPolicy GetReuseAddressPolicy () {return m_enReusePolicy;} virtual DWORD GetSocketBufferSize () {return m_dwSocketBufferSize;} virtual DWORD GetKeepAliveTime () {return m_dwKeepAliveTime;} virtual DWORD GetKeepAliveInterval () {return m_dwKeepAliveInterval;} virtual DWORD GetFreeBufferPoolSize () {return m_dwFreeBufferPoolSize;} virtual DWORD GetFreeBufferPoolHold () {return m_dwFreeBufferPoolHold;} virtual BOOL IsNoDelay () {return m_bNoDelay;} virtual PVOID GetExtra () {return m_pExtra;} protected: virtual EnHandleResult FirePrepareConnect(SOCKET socket) {return DoFirePrepareConnect(this, socket);} virtual EnHandleResult FireConnect() { EnHandleResult rs = DoFireConnect(this); if(rs != HR_ERROR) rs = FireHandShake(); return rs; } virtual EnHandleResult FireHandShake() {return DoFireHandShake(this);} virtual EnHandleResult FireSend(const BYTE* pData, int iLength) {return DoFireSend(this, pData, iLength);} virtual EnHandleResult FireReceive(const BYTE* pData, int iLength) {return DoFireReceive(this, pData, iLength);} virtual EnHandleResult FireReceive(int iLength) {return DoFireReceive(this, iLength);} virtual EnHandleResult FireClose(EnSocketOperation enOperation, int iErrorCode) {return DoFireClose(this, enOperation, iErrorCode);} virtual EnHandleResult DoFirePrepareConnect(ITcpClient* pSender, SOCKET socket) {return m_pListener->OnPrepareConnect(pSender, pSender->GetConnectionID(), socket);} virtual EnHandleResult DoFireConnect(ITcpClient* pSender) {return m_pListener->OnConnect(pSender, pSender->GetConnectionID());} virtual EnHandleResult DoFireHandShake(ITcpClient* pSender) {return m_pListener->OnHandShake(pSender, pSender->GetConnectionID());} virtual EnHandleResult DoFireSend(ITcpClient* pSender, const BYTE* pData, int iLength) {return m_pListener->OnSend(pSender, pSender->GetConnectionID(), pData, iLength);} virtual EnHandleResult DoFireReceive(ITcpClient* pSender, const BYTE* pData, int iLength) {return m_pListener->OnReceive(pSender, pSender->GetConnectionID(), pData, iLength);} virtual EnHandleResult DoFireReceive(ITcpClient* pSender, int iLength) {return m_pListener->OnReceive(pSender, pSender->GetConnectionID(), iLength);} virtual EnHandleResult DoFireClose(ITcpClient* pSender, EnSocketOperation enOperation, int iErrorCode) {return m_pListener->OnClose(pSender, pSender->GetConnectionID(), enOperation, iErrorCode);} void SetLastError(EnSocketError code, LPCSTR func, int ec); virtual BOOL CheckParams(); virtual void PrepareStart(); virtual void Reset(); virtual BOOL BeforeUnpause() {return TRUE;} virtual void OnWorkerThreadStart(THR_ID tid) {} virtual void OnWorkerThreadEnd(THR_ID tid) {} BOOL DoSendPackets(const WSABUF pBuffers[], int iCount); static BOOL DoSendPackets(CTcpClient* pClient, const WSABUF pBuffers[], int iCount) {return pClient->DoSendPackets(pBuffers, iCount);} protected: BOOL IsPaused () {return m_bPaused;} void SetReserved (PVOID pReserved) {m_pReserved = pReserved;} PVOID GetReserved () {return m_pReserved;} BOOL GetRemoteHost (LPCSTR* lpszHost, USHORT* pusPort = nullptr); private: void SetRemoteHost (LPCTSTR lpszHost, USHORT usPort); void SetConnected (BOOL bConnected = TRUE) {m_bConnected = bConnected; if(bConnected) m_enState = SS_STARTED;} BOOL CheckStarting(); BOOL CheckStoping(); BOOL CreateClientSocket(LPCTSTR lpszRemoteAddress, HP_SOCKADDR& addrRemote, USHORT usPort, LPCTSTR lpszBindAddress, HP_SOCKADDR& addrBind); BOOL BindClientSocket(const HP_SOCKADDR& addrBind, const HP_SOCKADDR& addrRemote, USHORT usLocalPort); BOOL ConnectToServer(const HP_SOCKADDR& addrRemote, BOOL bAsyncConnect); BOOL CreateWorkerThread(); BOOL ProcessNetworkEvent(SHORT events); BOOL ReadData(); BOOL SendData(); BOOL DoSendData(TItem* pItem, BOOL& bBlocked); int SendInternal(const WSABUF pBuffers[], int iCount); void WaitForWorkerThreadEnd(); BOOL HandleConnect (SHORT events); BOOL HandleClose (SHORT events); BOOL HandleRead (SHORT events); BOOL HandleWrite (SHORT events); UINT WINAPI WorkerThreadProc(LPVOID pv); public: CTcpClient(ITcpClientListener* pListener) : m_pListener (pListener) , m_lsSend (m_itPool) , m_soClient (INVALID_SOCKET) , m_nEvents (0) , m_dwConnID (0) , m_usPort (0) , m_bPaused (FALSE) , m_bConnected (FALSE) , m_enLastError (SE_OK) , m_enState (SS_STOPPED) , m_bNoDelay (FALSE) , m_pExtra (nullptr) , m_pReserved (nullptr) , m_enReusePolicy (RAP_ADDR_ONLY) , m_dwSocketBufferSize (DEFAULT_TCP_SOCKET_BUFFER_SIZE) , m_dwFreeBufferPoolSize(DEFAULT_CLIENT_FREE_BUFFER_POOL_SIZE) , m_dwFreeBufferPoolHold(DEFAULT_CLIENT_FREE_BUFFER_POOL_HOLD) , m_dwKeepAliveTime (DEFALUT_TCP_KEEPALIVE_TIME) , m_dwKeepAliveInterval (DEFALUT_TCP_KEEPALIVE_INTERVAL) { ASSERT(m_pListener); } virtual ~CTcpClient() { ENSURE_STOP(); } private: CSEM m_evWait; ITcpClientListener* m_pListener; TClientCloseContext m_ccContext; SOCKET m_soClient; SHORT m_nEvents; CONNID m_dwConnID; EnReuseAddressPolicy m_enReusePolicy; DWORD m_dwSocketBufferSize; DWORD m_dwFreeBufferPoolSize; DWORD m_dwFreeBufferPoolHold; DWORD m_dwKeepAliveTime; DWORD m_dwKeepAliveInterval; BOOL m_bNoDelay; EnSocketError m_enLastError; volatile BOOL m_bConnected; volatile EnServiceState m_enState; PVOID m_pExtra; PVOID m_pReserved; CBufferPtr m_rcBuffer; protected: CStringA m_strHost; USHORT m_usPort; CItemPool m_itPool; private: CSpinGuard m_csState; CCriSec m_csSend; TItemListExV m_lsSend; CEvt m_evSend; CEvt m_evRecv; CEvt m_evStop; volatile BOOL m_bPaused; CThread m_thWorker; };