/3rdParty/include/nio/AsyncSocketExLayer.cpp
C++ | 604 lines | 473 code | 65 blank | 66 comment | 89 complexity | 6b972dd8fed60d7b96f96bd542c060b8 MD5 | raw file
Possible License(s): LGPL-3.0
- /*CAsyncSocketEx by Tim Kosse (Tim.Kosse@gmx.de)
- Version 1.1 (2002-11-01)
- --------------------------------------------------------
- Introduction:
- -------------
- CAsyncSocketEx is a replacement for the MFC class CAsyncSocket.
- This class was written because CAsyncSocket is not the fastest WinSock
- wrapper and it's very hard to add new functionality to CAsyncSocket
- derived classes. This class offers the same functionality as CAsyncSocket.
- Also, CAsyncSocketEx offers some enhancements which were not possible with
- CAsyncSocket without some tricks.
- How do I use it?
- ----------------
- Basically exactly like CAsyncSocket.
- To use CAsyncSocketEx, just replace all occurrences of CAsyncSocket in your
- code with CAsyncSocketEx, if you did not enhance CAsyncSocket yourself in
- any way, you won't have to change anything else in your code.
- Why is CAsyncSocketEx faster?
- -----------------------------
- CAsyncSocketEx is slightly faster when dispatching notification event messages.
- First have a look at the way CAsyncSocket works. For each thread that uses
- CAsyncSocket, a window is created. CAsyncSocket calls WSAAsyncSelect with
- the handle of that window. Until here, CAsyncSocketEx works the same way.
- But CAsyncSocket uses only one window message (WM_SOCKET_NOTIFY) for all
- sockets within one thread. When the window recieve WM_SOCKET_NOTIFY, wParam
- contains the socket handle and the window looks up an CAsyncSocket instance
- using a map. CAsyncSocketEx works differently. It's helper window uses a
- wide range of different window messages (WM_USER through 0xBFFF) and passes
- a different message to WSAAsyncSelect for each socket. When a message in
- the specified range is received, CAsyncSocketEx looks up the pointer to a
- CAsyncSocketEx instance in an Array using the index of message - WM_USER.
- As you can see, CAsyncSocketEx uses the helper window in a more efficient
- way, as it don't have to use the slow maps to lookup it's own instance.
- Still, speed increase is not very much, but it may be noticeable when using
- a lot of sockets at the same time.
- Please note that the changes do not affect the raw data throughput rate,
- CAsyncSocketEx only dispatches the notification messages faster.
- What else does CAsyncSocketEx offer?
- ------------------------------------
- CAsyncSocketEx offers a flexible layer system. One example is the proxy layer.
- Just create an instance of the proxy layer, configure it and add it to the layer
- chain of your CAsyncSocketEx instance. After that, you can connect through
- proxies.
- Benefit: You don't have to change much to use the layer system.
- Another layer that is currently in development is the SSL layer to establish
- SSL encrypted connections.
- License
- -------
- Feel free to use this class, as long as you don't claim that you wrote it
- and this copyright notice stays intact in the source files.
- If you use this class in commercial applications, please send a short message
- to tim.kosse@gmx.de
- */#include "stdafx.h"
- #include "AsyncSocketExLayer.h"
- #include "AsyncSocketEx.h"
- #ifdef _DEBUG
- #undef THIS_FILE
- static char THIS_FILE[]=__FILE__;
- #define new DEBUG_NEW
- #endif
- #define WM_SOCKETEX_NOTIFY (WM_USER+2)
- //////////////////////////////////////////////////////////////////////
- // Konstruktion/Destruktion
- //////////////////////////////////////////////////////////////////////
- CAsyncSocketExLayer::CAsyncSocketExLayer()
- {
- m_pOwnerSocket=0;
- m_pNextLayer=0;
- m_nLayerState=notsock;
- m_nCriticalError=0;
- }
- CAsyncSocketExLayer::~CAsyncSocketExLayer()
- {
- }
- CAsyncSocketExLayer *CAsyncSocketExLayer::AddLayer(CAsyncSocketExLayer *pLayer, CAsyncSocketEx *pOwnerSocket)
- {
- ASSERT(pLayer);
- ASSERT(pOwnerSocket);
- if (m_pNextLayer)
- {
- return m_pNextLayer->AddLayer(pLayer, pOwnerSocket);
- }
- else
- {
- ASSERT(m_pOwnerSocket==pOwnerSocket);
- pLayer->Init(this, m_pOwnerSocket);
- m_pNextLayer=pLayer;
- }
- return m_pNextLayer;
- }
- int CAsyncSocketExLayer::Receive(void* lpBuf, int nBufLen, int nFlags /*=0*/)
- {
- return ReceiveNext(lpBuf, nBufLen, nFlags);
- }
- int CAsyncSocketExLayer::Send(const void* lpBuf, int nBufLen, int nFlags /*=0*/)
- {
- return SendNext(lpBuf, nBufLen, nFlags);
- }
- void CAsyncSocketExLayer::OnReceive(int nErrorCode)
- {
- if (m_pPrevLayer)
- m_pPrevLayer->OnReceive(nErrorCode);
- else
- if (m_pOwnerSocket->m_lEvent&FD_READ)
- m_pOwnerSocket->onReceive(nErrorCode);
- }
- void CAsyncSocketExLayer::OnSend(int nErrorCode)
- {
- if (m_pPrevLayer)
- m_pPrevLayer->OnSend(nErrorCode);
- else
- if (m_pOwnerSocket->m_lEvent&FD_WRITE)
- m_pOwnerSocket->onSend(nErrorCode);
- }
- void CAsyncSocketExLayer::OnConnect(int nErrorCode)
- {
- if (m_pPrevLayer)
- m_pPrevLayer->OnConnect(nErrorCode);
- else
- if (m_pOwnerSocket->m_lEvent&FD_CONNECT)
- m_pOwnerSocket->onConnect(nErrorCode);
- }
- void CAsyncSocketExLayer::OnAccept(int nErrorCode)
- {
- if (m_pPrevLayer)
- m_pPrevLayer->OnAccept(nErrorCode);
- else
- if (m_pOwnerSocket->m_lEvent&FD_ACCEPT)
- m_pOwnerSocket->onAccept(nErrorCode);
- }
- void CAsyncSocketExLayer::OnClose(int nErrorCode)
- {
- if (m_pPrevLayer)
- m_pPrevLayer->OnClose(nErrorCode);
- else
- if (m_pOwnerSocket->m_lEvent&FD_CLOSE)
- m_pOwnerSocket->onClose(nErrorCode);
- }
- BOOL CAsyncSocketExLayer::TriggerEvent(long lEvent, int nErrorCode, BOOL bPassThrough /*=FALSE*/ )
- {
- ASSERT(m_pOwnerSocket);
- if (m_pOwnerSocket->m_SocketData.hSocket==INVALID_SOCKET)
- return FALSE;
- if ( lEvent&FD_CONNECT)
- {
- ASSERT(bPassThrough);
- if (!nErrorCode)
- ASSERT(bPassThrough && GetLayerState()==connected);
- else if (nErrorCode)
- {
- SetLayerState(aborted);
- m_nCriticalError=nErrorCode;
- }
- }
- else if ( lEvent&FD_CLOSE && !nErrorCode)
- {
- SetLayerState(closed);
- }
- else if ( lEvent&FD_CLOSE && nErrorCode)
- {
- SetLayerState(aborted);
- m_nCriticalError=nErrorCode;
- }
- ASSERT(m_pOwnerSocket->m_pLocalAsyncSocketExThreadData);
- ASSERT(m_pOwnerSocket->m_pLocalAsyncSocketExThreadData->m_pHelperWindow);
- ASSERT(m_pOwnerSocket->m_SocketData.nSocketIndex!=-1);
- t_LayerNotifyMsg *pMsg=new t_LayerNotifyMsg;
- pMsg->lEvent = ( lEvent % 0xffff ) + ( nErrorCode << 16);
- pMsg->pLayer=bPassThrough?m_pPrevLayer:this;
- BOOL res=PostMessage(m_pOwnerSocket->GetHelperWindowHandle(), WM_USER, (WPARAM)m_pOwnerSocket, (LPARAM)pMsg);
- if (!res)
- delete pMsg;
- return res;
- }
- void CAsyncSocketExLayer::Close()
- {
- CloseNext();
- }
- void CAsyncSocketExLayer::CloseNext()
- {
- SetLayerState(notsock);
- if (m_pNextLayer)
- m_pNextLayer->Close();
- }
- BOOL CAsyncSocketExLayer::Connect(LPCTSTR lpszHostAddress, UINT nHostPort)
- {
- return ConnectNext(lpszHostAddress, nHostPort);
- }
- BOOL CAsyncSocketExLayer::Connect( const SOCKADDR* lpSockAddr, int nSockAddrLen )
- {
- return ConnectNext(lpSockAddr, nSockAddrLen);
- }
- int CAsyncSocketExLayer::SendNext(const void *lpBuf, int nBufLen, int nFlags /*=0*/)
- {
- if (m_nCriticalError)
- {
- WSASetLastError(m_nCriticalError);
- return SOCKET_ERROR;
- }
- else if (GetLayerState()==notsock)
- {
- WSASetLastError(WSAENOTSOCK);
- return SOCKET_ERROR;
- }
- else if (GetLayerState()==unconnected || GetLayerState()==connecting || GetLayerState()==listening)
- {
- WSASetLastError(WSAENOTCONN);
- return SOCKET_ERROR;
- }
- if (!m_pNextLayer)
- {
- ASSERT(m_pOwnerSocket);
- return send(m_pOwnerSocket->GetSocketHandle(), (LPSTR)lpBuf, nBufLen, nFlags);
- }
- else
- return m_pNextLayer->Send(lpBuf, nBufLen, nFlags);
- }
- int CAsyncSocketExLayer::ReceiveNext(void *lpBuf, int nBufLen, int nFlags /*=0*/)
- {
- if (m_nCriticalError)
- {
- WSASetLastError(m_nCriticalError);
- return SOCKET_ERROR;
- }
- else if (GetLayerState()==notsock)
- {
- WSASetLastError(WSAENOTSOCK);
- return SOCKET_ERROR;
- }
- else if (GetLayerState()==unconnected || GetLayerState()==connecting || GetLayerState()==listening)
- {
- WSASetLastError(WSAENOTCONN);
- return SOCKET_ERROR;
- }
- if (!m_pNextLayer)
- {
- ASSERT(m_pOwnerSocket);
- return recv(m_pOwnerSocket->GetSocketHandle(), (LPSTR)lpBuf, nBufLen, nFlags);
- }
- else
- return m_pNextLayer->Receive(lpBuf, nBufLen, nFlags);
- }
- BOOL CAsyncSocketExLayer::ConnectNext(LPCTSTR lpszHostAddress, UINT nHostPort)
- {
- ASSERT(GetLayerState()==unconnected);
- ASSERT(m_pOwnerSocket);
- BOOL res;
- if (m_pNextLayer)
- res = m_pNextLayer->Connect(lpszHostAddress, nHostPort);
- else
- {
- USES_CONVERSION;
- ASSERT(lpszHostAddress != NULL);
- SOCKADDR_IN sockAddr;
- memset(&sockAddr,0,sizeof(sockAddr));
- LPSTR lpszAscii = T2A((LPTSTR)lpszHostAddress);
- sockAddr.sin_family = AF_INET;
- sockAddr.sin_addr.s_addr = inet_addr(lpszAscii);
- if (sockAddr.sin_addr.s_addr == INADDR_NONE)
- {
- LPHOSTENT lphost;
- lphost = gethostbyname(lpszAscii);
- if (lphost != NULL)
- sockAddr.sin_addr.s_addr = ((LPIN_ADDR)lphost->h_addr)->s_addr;
- else
- {
- WSASetLastError(WSAEINVAL);
- res = FALSE;
- }
- }
- sockAddr.sin_port = htons((u_short)nHostPort);
- res = ( SOCKET_ERROR!=connect(m_pOwnerSocket->GetSocketHandle(), (SOCKADDR*)&sockAddr, sizeof(sockAddr)) );
- }
- if (res || WSAGetLastError()==WSAEWOULDBLOCK)
- {
- SetLayerState(connecting);
- }
- return res;
- }
- BOOL CAsyncSocketExLayer::ConnectNext( const SOCKADDR* lpSockAddr, int nSockAddrLen )
- {
- ASSERT(GetLayerState()==unconnected);
- ASSERT(m_pOwnerSocket);
- BOOL res;
- if (m_pNextLayer)
- res=m_pNextLayer->Connect(lpSockAddr, nSockAddrLen);
- else
- res = (SOCKET_ERROR!=connect(m_pOwnerSocket->GetSocketHandle(), lpSockAddr, nSockAddrLen));
- if (res || WSAGetLastError()==WSAEWOULDBLOCK)
- SetLayerState(connecting);
- return res;
- }
- //Gets the address of the peer socket to which the socket is connected
- #ifdef _AFX
- BOOL CAsyncSocketExLayer::GetPeerName( CString& rPeerAddress, UINT& rPeerPort )
- {
- return GetPeerNameNext(rPeerAddress, rPeerPort);
- }
- BOOL CAsyncSocketExLayer::GetPeerNameNext( CString& rPeerAddress, UINT& rPeerPort )
- {
- if (m_pNextLayer)
- return m_pNextLayer->GetPeerName(rPeerAddress, rPeerPort);
- else
- {
- ASSERT(m_pOwnerSocket);
- SOCKADDR_IN sockAddr;
- memset(&sockAddr, 0, sizeof(sockAddr));
- int nSockAddrLen = sizeof(sockAddr);
- if (!getpeername(m_pOwnerSocket->GetSocketHandle(), (SOCKADDR*)&sockAddr, &nSockAddrLen))
- {
- rPeerPort = ntohs(sockAddr.sin_port);
- rPeerAddress = inet_ntoa(sockAddr.sin_addr);
- return TRUE;
- }
- else
- return FALSE;
- }
- }
- #endif //_AFX
- BOOL CAsyncSocketExLayer::GetPeerName( SOCKADDR* lpSockAddr, int* lpSockAddrLen )
- {
- return GetPeerNameNext(lpSockAddr, lpSockAddrLen);
- }
- BOOL CAsyncSocketExLayer::GetPeerNameNext( SOCKADDR* lpSockAddr, int* lpSockAddrLen )
- {
- if (m_pNextLayer)
- return m_pNextLayer->GetPeerName(lpSockAddr, lpSockAddrLen);
- else
- {
- ASSERT(m_pOwnerSocket);
- if ( !getpeername(m_pOwnerSocket->GetSocketHandle(), lpSockAddr, lpSockAddrLen) )
- return TRUE;
- else
- return FALSE;
- }
- }
- void CAsyncSocketExLayer::Init(CAsyncSocketExLayer *pPrevLayer, CAsyncSocketEx *pOwnerSocket)
- {
- ASSERT(pOwnerSocket);
- m_pPrevLayer=pPrevLayer;
- m_pOwnerSocket=pOwnerSocket;
- m_pNextLayer=0;
- }
- int CAsyncSocketExLayer::GetLayerState()
- {
- return m_nLayerState;
- }
- void CAsyncSocketExLayer::CallEvent(int nEvent, int nErrorCode)
- {
- if (m_nCriticalError)
- return;
- m_nCriticalError=nErrorCode;
- switch (nEvent)
- {
- case FD_READ:
- case FD_FORCEREAD:
- if (GetLayerState()==connected)
- {
- if (nErrorCode)
- SetLayerState(aborted);
- OnReceive(nErrorCode);
- }
- break;
- case FD_WRITE:
- if (GetLayerState()==connected)
- {
- if (nErrorCode)
- SetLayerState(aborted);
- OnSend(nErrorCode);
- }
- break;
- case FD_CONNECT:
- if (GetLayerState()==connecting)
- {
- if (!nErrorCode)
- SetLayerState(connected);
- else
- SetLayerState(aborted);
- OnConnect(nErrorCode);
- }
- break;
- case FD_ACCEPT:
- if (GetLayerState()==listening)
- {
- if (!nErrorCode)
- SetLayerState(connected);
- else
- SetLayerState(aborted);
- OnAccept(nErrorCode);
- }
- break;
- case FD_CLOSE:
- if (GetLayerState()==connected)
- {
- if (nErrorCode)
- SetLayerState(aborted);
- else
- SetLayerState(closed);
- OnClose(nErrorCode);
- }
- break;
- }
- }
- //Creates a socket
- BOOL CAsyncSocketExLayer::Create(UINT nSocketPort, int nSocketType,
- long lEvent, LPCTSTR lpszSocketAddress)
- {
- return CreateNext(nSocketPort, nSocketType, lEvent, lpszSocketAddress);
- }
- BOOL CAsyncSocketExLayer::CreateNext(UINT nSocketPort, int nSocketType, long lEvent, LPCTSTR lpszSocketAddress)
- {
- ASSERT(GetLayerState()==notsock);
- BOOL res=FALSE;
- if (m_pNextLayer)
- res=m_pNextLayer->Create(nSocketPort, nSocketType, lEvent, lpszSocketAddress);
- else
- {
- SOCKET hSocket=socket(AF_INET, nSocketType, 0);
- if (hSocket==INVALID_SOCKET)
- res=FALSE;
- m_pOwnerSocket->m_SocketData.hSocket=hSocket;
- m_pOwnerSocket->AttachHandle(hSocket);
- if (!m_pOwnerSocket->AsyncSelect(lEvent))
- {
- m_pOwnerSocket->Close();
- res=FALSE;
- }
- if (m_pOwnerSocket->m_pFirstLayer)
- {
- if (WSAAsyncSelect(m_pOwnerSocket->m_SocketData.hSocket, m_pOwnerSocket->GetHelperWindowHandle(), m_pOwnerSocket->m_SocketData.nSocketIndex+WM_SOCKETEX_NOTIFY, FD_READ | FD_WRITE | FD_OOB | FD_ACCEPT | FD_CONNECT | FD_CLOSE) )
- {
- m_pOwnerSocket->Close();
- res=FALSE;
- }
- }
- if (!m_pOwnerSocket->Bind(nSocketPort, lpszSocketAddress))
- {
- m_pOwnerSocket->Close();
- res=FALSE;
- }
- res=TRUE;
- }
- if (res)
- SetLayerState(unconnected);
- return res;
- }
- void CAsyncSocketExLayer::SetLayerState(int nLayerState)
- {
- ASSERT(m_pOwnerSocket);
- int nOldLayerState=GetLayerState();
- m_nLayerState=nLayerState;
- if (nOldLayerState!=nLayerState)
- DoLayerCallback(LAYERCALLBACK_STATECHANGE, GetLayerState(), nOldLayerState);
- }
- int CAsyncSocketExLayer::DoLayerCallback(int nType, int nParam1, int nParam2)
- {
- ASSERT(m_pOwnerSocket);
- int nError=WSAGetLastError();
- int res = m_pOwnerSocket->OnLayerCallback(this, nType, nParam1, nParam2);
- WSASetLastError(nError);
- return res;
- }
- BOOL CAsyncSocketExLayer::Listen( int nConnectionBacklog)
- {
- return ListenNext( nConnectionBacklog);
- }
- BOOL CAsyncSocketExLayer::ListenNext( int nConnectionBacklog)
- {
- ASSERT(GetLayerState()==unconnected);
- BOOL res;
- if (m_pNextLayer)
- res=m_pNextLayer->Listen(nConnectionBacklog);
- else
- res=listen(m_pOwnerSocket->GetSocketHandle(), nConnectionBacklog);
- if (res)
- {
- SetLayerState(listening);
- }
- return res;
- }
- BOOL CAsyncSocketExLayer::Accept( CAsyncSocketEx& rConnectedSocket, SOCKADDR* lpSockAddr /*=NULL*/, int* lpSockAddrLen /*=NULL*/ )
- {
- return AcceptNext(rConnectedSocket, lpSockAddr, lpSockAddrLen);
- }
- BOOL CAsyncSocketExLayer::AcceptNext( CAsyncSocketEx& rConnectedSocket, SOCKADDR* lpSockAddr /*=NULL*/, int* lpSockAddrLen /*=NULL*/ )
- {
- ASSERT(GetLayerState()==listening);
- BOOL res;
- if (m_pNextLayer)
- res=m_pNextLayer->Accept(rConnectedSocket, lpSockAddr, lpSockAddrLen);
- else
- {
- SOCKET hTemp = accept(m_pOwnerSocket->m_SocketData.hSocket, lpSockAddr, lpSockAddrLen);
- if (hTemp == INVALID_SOCKET)
- return FALSE;
- VERIFY(rConnectedSocket.InitAsyncSocketExInstance());
- rConnectedSocket.m_SocketData.hSocket=hTemp;
- rConnectedSocket.AttachHandle(hTemp);
- }
- return TRUE;
- }
- BOOL CAsyncSocketExLayer::ShutDown(int nHow /*=sends*/)
- {
- return ShutDownNext(nHow);
- }
- BOOL CAsyncSocketExLayer::ShutDownNext(int nHow /*=sends*/)
- {
- if (m_nCriticalError)
- {
- WSASetLastError(m_nCriticalError);
- return FALSE;
- }
- else if (GetLayerState()==notsock)
- {
- WSASetLastError(WSAENOTSOCK);
- return FALSE;
- }
- else if (GetLayerState()==unconnected || GetLayerState()==connecting || GetLayerState()==listening)
- {
- WSASetLastError(WSAENOTCONN);
- return FALSE;
- }
- if (!m_pNextLayer)
- {
- ASSERT(m_pOwnerSocket);
- return shutdown(m_pOwnerSocket->GetSocketHandle(), nHow);
- }
- else
- return m_pNextLayer->ShutDownNext(nHow);
- }