/3rdParty/include/nio/AsyncSocketEx.cpp
C++ | 1041 lines | 821 code | 123 blank | 97 comment | 118 complexity | 68dd2541b2edd40600add232adce49a9 MD5 | raw file
Possible License(s): LGPL-3.0
- /*CAsyncSocketEx by Tim Kosse (Tim.Kosse@gmx.de)
- Version 1.2 (2003-03-28)
- --------------------------------------------------------
- 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 "AsyncSocketEx.h"
- #include "wtypes.h"
- #include "oleauto.h"
- #include "atlconv.h"
- #include "utility/utilCommonAPI.h"
- #ifndef NOLAYERS
- #include "AsyncSocketExLayer.h"
- #endif //NOLAYERS
- #ifdef _DEBUG
- #undef THIS_FILE
- static char THIS_FILE[]=__FILE__;
- #endif
- #ifndef CCRITICALSECTIONWRAPPERINCLUDED
- class CCriticalSectionWrapper
- {
- public:
- CCriticalSectionWrapper()
- {
- InitializeCriticalSection(&m_criticalSection);
- }
- ~CCriticalSectionWrapper()
- {
- DeleteCriticalSection(&m_criticalSection);
- }
- void Lock()
- {
- EnterCriticalSection(&m_criticalSection);
- }
- void Unlock()
- {
- LeaveCriticalSection(&m_criticalSection);
- }
- protected:
- CRITICAL_SECTION m_criticalSection;
- };
- #define CCRITICALSECTIONWRAPPERINCLUDED
- #endif
- CCriticalSectionWrapper CAsyncSocketEx::m_sGlobalCriticalSection;
- CAsyncSocketEx::t_AsyncSocketExThreadDataList *CAsyncSocketEx::m_spAsyncSocketExThreadDataList=0;
- #ifndef _AFX
- #ifndef VERIFY
- #define VERIFY(x) (void(x))
- #endif //VERIFY
- #ifndef ASSERT
- #define ASSERT(x)
- #endif //ASSERT
- #endif //_AFX
- /////////////////////////////
- //Helper Window class
- #define WM_SOCKETEX_NOTIFY (WM_USER+2)
- #define MAX_SOCKETS (0xBFFF-WM_SOCKETEX_NOTIFY+1)
- class CAsyncSocketExHelperWindow
- {
- public:
- CAsyncSocketExHelperWindow()
- {
- //Initialize data
- m_pAsyncSocketExWindowData=new t_AsyncSocketExWindowData[512]; //Reserve space for 512 active sockets
- memset(m_pAsyncSocketExWindowData, 0, 512*sizeof(t_AsyncSocketExWindowData));
- m_nWindowDataSize=512;
- m_nSocketCount=0;
- m_nWindowDataPos=0;
- //Create window
- WNDCLASSEX wndclass;
- wndclass.cbSize=sizeof wndclass;
- wndclass.style=0;
- wndclass.lpfnWndProc=WindowProc;
- wndclass.cbClsExtra=0;
- wndclass.cbWndExtra=0;
- wndclass.hInstance=GetModuleHandle(0);
- wndclass.hIcon=0;
- wndclass.hCursor=0;
- wndclass.hbrBackground=0;
- wndclass.lpszMenuName=0;
- wndclass.lpszClassName=_T("CAsyncSocketEx Helper Window");
- wndclass.hIconSm=0;
- RegisterClassEx(&wndclass);
- m_hWnd=CreateWindow(_T("CAsyncSocketEx Helper Window"), _T("CAsyncSocketEx Helper Window"), 0, 0, 0, 0, 0, 0, 0, 0, GetModuleHandle(0));
- ASSERT(m_hWnd);
- SetWindowLong(m_hWnd, GWL_USERDATA, (LONG)this);
- };
- virtual ~CAsyncSocketExHelperWindow()
- {
- //Clean up socket storage
- delete [] m_pAsyncSocketExWindowData;
- m_pAsyncSocketExWindowData=0;
- m_nWindowDataSize=0;
- m_nSocketCount=0;
- //Destroy window
- if (m_hWnd)
- {
- DestroyWindow(m_hWnd);
- m_hWnd=0;
- }
- }
- //Adds a socket to the list of attached sockets
- BOOL AddSocket(CAsyncSocketEx *pSocket, int &nSocketIndex)
- {
- ASSERT(pSocket);
- if (!m_nWindowDataSize)
- {
- ASSERT(!m_nSocketCount);
- m_nWindowDataSize=512;
- m_pAsyncSocketExWindowData=new t_AsyncSocketExWindowData[512]; //Reserve space for 512 active sockets
- memset(m_pAsyncSocketExWindowData, 0, 512*sizeof(t_AsyncSocketExWindowData));
- }
- if (nSocketIndex!=-1)
- {
- ASSERT(m_pAsyncSocketExWindowData);
- ASSERT(m_nWindowDataSize>nSocketIndex);
- ASSERT(m_pAsyncSocketExWindowData[nSocketIndex].m_pSocket==pSocket);
- ASSERT(m_nSocketCount);
- return TRUE;
- }
- //Increase socket storage if too small
- if (m_nSocketCount>=(m_nWindowDataSize-10))
- {
- int nOldWindowDataSize=m_nWindowDataSize;
- ASSERT(m_nWindowDataSize<MAX_SOCKETS);
- m_nWindowDataSize+=512;
- if (m_nWindowDataSize>MAX_SOCKETS)
- m_nWindowDataSize=MAX_SOCKETS;
- t_AsyncSocketExWindowData *tmp=m_pAsyncSocketExWindowData;
- m_pAsyncSocketExWindowData = new t_AsyncSocketExWindowData[m_nWindowDataSize];
- memcpy(m_pAsyncSocketExWindowData, tmp, nOldWindowDataSize * sizeof(t_AsyncSocketExWindowData));
- memset(m_pAsyncSocketExWindowData+nOldWindowDataSize, 0, (m_nWindowDataSize-nOldWindowDataSize)*sizeof(t_AsyncSocketExWindowData));
- delete [] tmp;
- }
- //Search for free slot
- for (int i=m_nWindowDataPos;i<(m_nWindowDataSize+m_nWindowDataPos);i++)
- {
- if (!m_pAsyncSocketExWindowData[i%m_nWindowDataSize].m_pSocket)
- {
- m_pAsyncSocketExWindowData[i%m_nWindowDataSize].m_pSocket=pSocket;
- nSocketIndex=i%m_nWindowDataSize;
- m_nWindowDataPos=(i+1)%m_nWindowDataSize;
- m_nSocketCount++;
- return TRUE;
- }
- }
- //No slot found, maybe there are too much sockets!
- return FALSE;
- }
- //Removes a socket from the socket storage
- BOOL RemoveSocket(CAsyncSocketEx *pSocket, int &nSocketIndex)
- {
- ASSERT(pSocket);
- if (nSocketIndex==-1)
- return TRUE;
- ASSERT(m_pAsyncSocketExWindowData);
- ASSERT(m_nWindowDataSize>0);
- ASSERT(m_nSocketCount>0);
- ASSERT(m_pAsyncSocketExWindowData[nSocketIndex].m_pSocket==pSocket);
- m_pAsyncSocketExWindowData[nSocketIndex].m_pSocket=0;
- nSocketIndex=-1;
- m_nSocketCount--;
- return TRUE;
- }
- //Processes event notifications sent by the sockets or the layers
- static LRESULT CALLBACK WindowProc( HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam )
- {
- if (message>=WM_SOCKETEX_NOTIFY)
- {
- //Verify parameters
- ASSERT(hWnd);
- CAsyncSocketExHelperWindow *pWnd=(CAsyncSocketExHelperWindow *)GetWindowLong(hWnd, GWL_USERDATA);
- ASSERT(pWnd);
- if (message<static_cast<UINT>(WM_SOCKETEX_NOTIFY+pWnd->m_nWindowDataSize)) //Index is within socket storage
- {
- //Lookup socket and verify if it's valid
- CAsyncSocketEx *pSocket=pWnd->m_pAsyncSocketExWindowData[message-WM_SOCKETEX_NOTIFY].m_pSocket;
- SOCKET hSocket=wParam;
- if (!pSocket)
- return 0;
- if (hSocket==INVALID_SOCKET)
- return 0;
- if (pSocket->m_SocketData.hSocket!=hSocket)
- return 0;
- int nEvent=lParam&0xFFFF;
- int nErrorCode=lParam>>16;
- //Dispatch notification
- #ifndef NOLAYERS
- if (!pSocket->m_pFirstLayer)
- {
- #endif //NOLAYERS
- //Dispatch to CAsyncSocketEx instance
- switch (nEvent)
- {
- case FD_READ:
- {
- DWORD nBytes;
- if (!pSocket->IOCtl(FIONREAD, &nBytes))
- nErrorCode = WSAGetLastError();
- if (nBytes != 0 || nErrorCode != 0)
- pSocket->onReceive(nErrorCode);
- }
- break;
- case FD_FORCEREAD: //Forceread does not check if there's data waiting
- pSocket->onReceive(nErrorCode);
- break;
- case FD_WRITE:
- pSocket->onSend(nErrorCode);
- break;
- case FD_CONNECT:
- pSocket->onConnect(nErrorCode);
- break;
- case FD_ACCEPT:
- pSocket->onAccept(nErrorCode);
- break;
- case FD_CLOSE:
- pSocket->onClose(nErrorCode);
- break;
- }
- }
- #ifndef NOLAYERS
- else //Dispatch notification to the lowest layer
- {
- if (nEvent==FD_READ)
- {
- DWORD nBytes;
- if (!pSocket->IOCtl(FIONREAD, &nBytes))
- nErrorCode = WSAGetLastError();
- if (nBytes != 0 || nErrorCode != 0)
- pSocket->m_pLastLayer->CallEvent(nEvent, nErrorCode);
- }
- else
- pSocket->m_pLastLayer->CallEvent(nEvent, nErrorCode);
- }
- }
- #endif //NOLAYERS
- return 0;
- }
- #ifndef NOLAYERS
- else if (message==WM_USER) //Notification event sent by a layer
- {
- //Verify parameters, lookup socket and notification message
- if (!wParam)
- return 0;
- CAsyncSocketEx *pSocket=(CAsyncSocketEx *)wParam;
- CAsyncSocketExLayer::t_LayerNotifyMsg *pMsg=(CAsyncSocketExLayer::t_LayerNotifyMsg *)lParam;
- if (pSocket->m_SocketData.hSocket==INVALID_SOCKET)
- {
- delete pMsg;
- return 0;
- }
- int nEvent=pMsg->lEvent&0xFFFF;
- int nErrorCode=pMsg->lEvent>>16;
- //Dispatch to layer
- if (pMsg->pLayer)
- pMsg->pLayer->CallEvent(nEvent, nErrorCode);
- else
- {
- //Dispatch to socket class
- switch (nEvent)
- {
- case FD_READ:
- if (pSocket->m_lEvent&FD_READ)
- pSocket->onReceive(nErrorCode);
- break;
- case FD_FORCEREAD:
- if (pSocket->m_lEvent&FD_FORCEREAD)
- pSocket->onReceive(nErrorCode);
- break;
- case FD_WRITE:
- if (pSocket->m_lEvent&FD_WRITE)
- pSocket->onSend(nErrorCode);
- break;
- case FD_CONNECT:
- if (pSocket->m_lEvent&FD_CONNECT)
- pSocket->onConnect(nErrorCode);
- break;
- case FD_ACCEPT:
- if (pSocket->m_lEvent&FD_ACCEPT)
- pSocket->onAccept(nErrorCode);
- break;
- case FD_CLOSE:
- if (pSocket->m_lEvent&FD_CLOSE)
- pSocket->onClose(nErrorCode);
- break;
- }
- }
- delete pMsg;
- return 0;
- }
- #endif //NOLAYERS
- else if (message == WM_USER+1)
- {
- //WSAAsyncGetHostByName reply
- //Verify parameters
- ASSERT(hWnd);
- CAsyncSocketExHelperWindow *pWnd=(CAsyncSocketExHelperWindow *)GetWindowLong(hWnd, GWL_USERDATA);
- ASSERT(pWnd);
- CAsyncSocketEx *pSocket = nullptr;
- int i = 0;
- for (i=0; i<pWnd->m_nWindowDataSize; i++)
- {
- pSocket = pWnd->m_pAsyncSocketExWindowData[i].m_pSocket;
- if (pSocket && pSocket->m_hAsyncGetHostByNameHandle &&
- pSocket->m_hAsyncGetHostByNameHandle == (HANDLE)wParam)
- break;
- }
- if (i == pWnd->m_nWindowDataSize)
- return 0;
- int nErrorCode = lParam >> 16;
- int len = lParam % 0xFFFF;
- if (nErrorCode)
- {
- pSocket->onConnect(nErrorCode);
- return 0;
- }
- SOCKADDR_IN sockAddr;
- memset(&sockAddr,0,sizeof(sockAddr));
- sockAddr.sin_family=AF_INET;
- sockAddr.sin_addr.s_addr = ((LPIN_ADDR)((LPHOSTENT)pSocket->m_pAsyncGetHostByNameBuffer)->h_addr)->s_addr;
- sockAddr.sin_port = htons(pSocket->m_nAsyncGetHostByNamePort);
- BOOL res = pSocket->Connect((SOCKADDR*)&sockAddr, sizeof(sockAddr));
- delete [] pSocket->m_pAsyncGetHostByNameBuffer;
- pSocket->m_pAsyncGetHostByNameBuffer=0;
- pSocket->m_hAsyncGetHostByNameHandle=0;
- if (!res)
- if (GetLastError()!=WSAEWOULDBLOCK)
- pSocket->onConnect(GetLastError());
- return 0;
- }
- return DefWindowProc(hWnd, message, wParam, lParam);
- }
- HWND CAsyncSocketExHelperWindow::GetHwnd()
- {
- return m_hWnd;
- }
- private:
- HWND m_hWnd;
- struct t_AsyncSocketExWindowData
- {
- CAsyncSocketEx *m_pSocket;
- } *m_pAsyncSocketExWindowData;
- int m_nWindowDataSize;
- int m_nWindowDataPos;
- int m_nSocketCount;
- };
- //////////////////////////////////////////////////////////////////////
- // Konstruktion/Destruktion
- //////////////////////////////////////////////////////////////////////
- CAsyncSocketEx::CAsyncSocketEx()
- {
- m_SocketData.hSocket=INVALID_SOCKET;
- m_SocketData.nSocketIndex=-1;
- m_pLocalAsyncSocketExThreadData=0;
- #ifndef NOLAYERS
- m_pFirstLayer=0;
- m_pLastLayer=0;
- #endif //NOLAYERS
- m_pAsyncGetHostByNameBuffer = NULL;
- m_hAsyncGetHostByNameHandle = NULL;
- }
- CAsyncSocketEx::~CAsyncSocketEx()
- {
- Close();
- FreeAsyncSocketExInstance();
- }
- BOOL CAsyncSocketEx::Create( UINT nSocketPort /*=0*/, int nSocketType /*=SOCK_STREAM*/, long lEvent /*=FD_READ | FD_WRITE | FD_OOB | FD_ACCEPT | FD_CONNECT | FD_CLOSE*/, LPCTSTR lpszSocketAddress /*=NULL*/ )
- {
- ASSERT(GetSocketHandle()==INVALID_SOCKET);
- //Close the socket, although this should not happen
- if (GetSocketHandle() != INVALID_SOCKET)
- {
- WSASetLastError(WSAEALREADY);
- return FALSE;
- }
- BOOL res=InitAsyncSocketExInstance();
- ASSERT(res);
- if (!res)
- {
- WSASetLastError(WSANOTINITIALISED);
- return FALSE;
- }
- #ifndef NOLAYERS
- if (m_pFirstLayer)
- return m_pFirstLayer->Create(nSocketPort, nSocketType, lEvent, lpszSocketAddress);
- else
- #endif //NOLAYERS
- {
- SOCKET hSocket=socket(AF_INET, nSocketType, 0);
- if (hSocket==INVALID_SOCKET)
- return FALSE;
- m_SocketData.hSocket=hSocket;
- AttachHandle(hSocket);
- if (!AsyncSelect(lEvent))
- {
- Close();
- return FALSE;
- }
- #ifndef NOLAYERS
- if (m_pFirstLayer)
- {
- if (WSAAsyncSelect(m_SocketData.hSocket, GetHelperWindowHandle()
- , m_SocketData.nSocketIndex+WM_SOCKETEX_NOTIFY
- , FD_READ | FD_WRITE | FD_OOB | FD_ACCEPT | FD_CONNECT | FD_CLOSE) )
- {
- Close();
- return FALSE;
- }
- }
- #endif //NOLAYERS
- if (!Bind(nSocketPort, lpszSocketAddress))
- {
- Close();
- return FALSE;
- }
- return TRUE;
- }
- }
- void CAsyncSocketEx::onReceive(int nErrorCode)
- {
- }
- void CAsyncSocketEx::onSend(int nErrorCode)
- {
- }
- void CAsyncSocketEx::onConnect(int nErrorCode)
- {
- }
- void CAsyncSocketEx::onAccept(int nErrorCode)
- {
- }
- void CAsyncSocketEx::onClose(int nErrorCode)
- {
- }
- BOOL CAsyncSocketEx::Bind(UINT nSocketPort, LPCTSTR lpszSocketAddress)
- {
- USES_CONVERSION;
- SOCKADDR_IN sockAddr;
- memset(&sockAddr,0,sizeof(sockAddr));
- LPSTR lpszAscii = T2A((LPTSTR)lpszSocketAddress);
- sockAddr.sin_family = AF_INET;
- if (lpszAscii == NULL)
- sockAddr.sin_addr.s_addr = htonl(INADDR_ANY);
- else
- {
- DWORD lResult = inet_addr(lpszAscii);
- if (lResult == INADDR_NONE)
- {
- WSASetLastError(WSAEINVAL);
- return FALSE;
- }
- sockAddr.sin_addr.s_addr = lResult;
- }
- sockAddr.sin_port = htons((u_short)nSocketPort);
- return Bind((SOCKADDR*)&sockAddr, sizeof(sockAddr));
- }
- BOOL CAsyncSocketEx::Bind(const SOCKADDR* lpSockAddr, int nSockAddrLen)
- {
- if (!bind(m_SocketData.hSocket, lpSockAddr, nSockAddrLen))
- return TRUE;
- else
- return FALSE;
- }
- void CAsyncSocketEx::AttachHandle(SOCKET hSocket)
- {
- ASSERT(m_pLocalAsyncSocketExThreadData);
- VERIFY(m_pLocalAsyncSocketExThreadData->m_pHelperWindow->AddSocket(this, m_SocketData.nSocketIndex));
- }
- void CAsyncSocketEx::DetachHandle(SOCKET hSocket)
- {
- ASSERT(m_pLocalAsyncSocketExThreadData);
- if (!m_pLocalAsyncSocketExThreadData)
- return;
- ASSERT(m_pLocalAsyncSocketExThreadData->m_pHelperWindow);
- if (!m_pLocalAsyncSocketExThreadData->m_pHelperWindow)
- return;
- VERIFY(m_pLocalAsyncSocketExThreadData->m_pHelperWindow->RemoveSocket(this, m_SocketData.nSocketIndex));
- }
- void CAsyncSocketEx::Close()
- {
- #ifndef NOLAYERS
- if (m_pFirstLayer)
- m_pFirstLayer->Close();
- #endif //NOLAYERS
- if (m_SocketData.hSocket != INVALID_SOCKET)
- {
- if(closesocket(m_SocketData.hSocket)==SOCKET_ERROR)
- {
- APP_LOG(LOG_ERROR,TRUE,_T("close socket error"));
- }
- DetachHandle(m_SocketData.hSocket);
- m_SocketData.hSocket=INVALID_SOCKET;
- }
- #ifndef NOLAYERS
- RemoveAllLayers();
- #endif //NOLAYERS
- delete [] m_pAsyncGetHostByNameBuffer;
- m_pAsyncGetHostByNameBuffer = NULL;
- if (m_hAsyncGetHostByNameHandle)
- WSACancelAsyncRequest(m_hAsyncGetHostByNameHandle);
- m_hAsyncGetHostByNameHandle = NULL;
- }
- BOOL CAsyncSocketEx::InitAsyncSocketExInstance()
- {
- //Check if already initialized
- if (m_pLocalAsyncSocketExThreadData)
- return TRUE;
- DWORD id=GetCurrentThreadId();
- m_sGlobalCriticalSection.Lock();
- //WSADATA wsa_data;
- //int r = WSAStartup(MAKEWORD(2,2), &wsa_data);
- //Get thread specific data
- if (m_spAsyncSocketExThreadDataList)
- {
- t_AsyncSocketExThreadDataList *pList=m_spAsyncSocketExThreadDataList;
- while (pList)
- {
- ASSERT(pList->pThreadData);
- ASSERT(pList->pThreadData->nInstanceCount>0);
- if (pList->pThreadData->nThreadId==id)
- {
- m_pLocalAsyncSocketExThreadData=pList->pThreadData;
- m_pLocalAsyncSocketExThreadData->nInstanceCount++;
- break;
- }
- pList=pList->pNext;
- }
- //Current thread yet has no sockets
- if (!pList)
- {
- //Initialize data for current thread
- pList=new t_AsyncSocketExThreadDataList;
- pList->pNext=m_spAsyncSocketExThreadDataList;
- m_spAsyncSocketExThreadDataList=pList;
- m_pLocalAsyncSocketExThreadData=new t_AsyncSocketExThreadData;
- m_pLocalAsyncSocketExThreadData->nInstanceCount=1;
- m_pLocalAsyncSocketExThreadData->nThreadId=id;
- m_pLocalAsyncSocketExThreadData->m_pHelperWindow=new CAsyncSocketExHelperWindow;
- m_spAsyncSocketExThreadDataList->pThreadData=m_pLocalAsyncSocketExThreadData;
- }
- }
- else
- { //No thread has instances of CAsyncSocketEx; Initialize data
- m_spAsyncSocketExThreadDataList=new t_AsyncSocketExThreadDataList;
- m_spAsyncSocketExThreadDataList->pNext=0;
- m_pLocalAsyncSocketExThreadData=new t_AsyncSocketExThreadData;
- m_pLocalAsyncSocketExThreadData->nInstanceCount=1;
- m_pLocalAsyncSocketExThreadData->nThreadId=id;
- m_pLocalAsyncSocketExThreadData->m_pHelperWindow=new CAsyncSocketExHelperWindow;
- m_spAsyncSocketExThreadDataList->pThreadData=m_pLocalAsyncSocketExThreadData;
- }
- m_sGlobalCriticalSection.Unlock();
- return TRUE;
- }
- void CAsyncSocketEx::FreeAsyncSocketExInstance()
- {
- //Check if already freed
- if (!m_pLocalAsyncSocketExThreadData)
- return;
- DWORD id=m_pLocalAsyncSocketExThreadData->nThreadId;
- m_sGlobalCriticalSection.Lock();
- ASSERT(m_spAsyncSocketExThreadDataList);
- t_AsyncSocketExThreadDataList *pList=m_spAsyncSocketExThreadDataList;
- t_AsyncSocketExThreadDataList *pPrev=0;
- //Serach for data for current thread and decrease instance count
- while (pList)
- {
- ASSERT(pList->pThreadData);
- ASSERT(pList->pThreadData->nInstanceCount>0);
- if (pList->pThreadData->nThreadId==id)
- {
- ASSERT(m_pLocalAsyncSocketExThreadData==pList->pThreadData);
- m_pLocalAsyncSocketExThreadData->nInstanceCount--;
- //Freeing last instance?
- //If so, destroy helper window
- if (!m_pLocalAsyncSocketExThreadData->nInstanceCount)
- {
- delete m_pLocalAsyncSocketExThreadData->m_pHelperWindow;
- delete m_pLocalAsyncSocketExThreadData;
- if (pPrev)
- pPrev->pNext=pList->pNext;
- else
- m_spAsyncSocketExThreadDataList=pList->pNext;
- delete pList;
- break;
- }
- break;
- }
- pPrev=pList;
- pList=pList->pNext;
- ASSERT(pList);
- }
- m_sGlobalCriticalSection.Unlock();
- }
- int CAsyncSocketEx::Receive(void* lpBuf, int nBufLen, int nFlags /*=0*/)
- {
- #ifndef NOLAYERS
- if (m_pFirstLayer)
- return m_pFirstLayer->Receive(lpBuf, nBufLen, nFlags);
- else
- #endif //NOLAYERS
- return recv(m_SocketData.hSocket, (LPSTR)lpBuf, nBufLen, nFlags);
- }
- int CAsyncSocketEx::Send(const void* lpBuf, int nBufLen, int nFlags /*=0*/)
- {
- #ifndef NOLAYERS
- if (m_pFirstLayer)
- return m_pFirstLayer->Send(lpBuf, nBufLen, nFlags);
- else
- #endif //NOLAYERS
- return send(m_SocketData.hSocket, (LPSTR)lpBuf, nBufLen, nFlags);
- }
- BOOL CAsyncSocketEx::Connect(LPCTSTR lpszHostAddress, UINT nHostPort)
- {
- #ifndef NOLAYERS
- if (m_pFirstLayer)
- return m_pFirstLayer->Connect(lpszHostAddress, nHostPort);
- else
- #endif //NOLAYERS
- {
- 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)
- {
- m_pAsyncGetHostByNameBuffer=new char[MAXGETHOSTSTRUCT];
- m_nAsyncGetHostByNamePort=nHostPort;
- m_hAsyncGetHostByNameHandle=WSAAsyncGetHostByName(GetHelperWindowHandle(), WM_USER+1, lpszAscii, m_pAsyncGetHostByNameBuffer, MAXGETHOSTSTRUCT);
- if (!m_hAsyncGetHostByNameHandle)
- return FALSE;
- WSASetLastError(WSAEWOULDBLOCK);
- return TRUE;
- }
- sockAddr.sin_port = htons((u_short)nHostPort);
- return CAsyncSocketEx::Connect((SOCKADDR*)&sockAddr, sizeof(sockAddr));
- }
- }
- BOOL CAsyncSocketEx::Connect( const SOCKADDR* lpSockAddr, int nSockAddrLen )
- {
- #ifndef NOLAYERS
- if (m_pFirstLayer)
- return m_pFirstLayer->Connect(lpSockAddr, nSockAddrLen);
- else
- #endif //NOLAYERS
- return SOCKET_ERROR!=connect(m_SocketData.hSocket, lpSockAddr, nSockAddrLen);
- }
- #ifdef _AFX
- BOOL CAsyncSocketEx::GetPeerName( CString& rPeerAddress, UINT& rPeerPort )
- {
- #ifndef NOLAYERS
- if (m_pFirstLayer)
- return m_pFirstLayer->GetPeerName(rPeerAddress, rPeerPort);
- #endif NOLAYERS
- SOCKADDR_IN sockAddr;
- memset(&sockAddr, 0, sizeof(sockAddr));
- int nSockAddrLen = sizeof(sockAddr);
- BOOL bResult = GetPeerName((SOCKADDR*)&sockAddr, &nSockAddrLen);
- if (bResult)
- {
- rPeerPort = ntohs(sockAddr.sin_port);
- rPeerAddress = inet_ntoa(sockAddr.sin_addr);
- }
- return bResult;
- }
- #endif //AFX
- BOOL CAsyncSocketEx::GetPeerName( SOCKADDR* lpSockAddr, int* lpSockAddrLen )
- {
- #ifndef NOLAYERS
- if (m_pFirstLayer)
- return m_pFirstLayer->GetPeerName(lpSockAddr, lpSockAddrLen);
- #endif //NOLAYERS
- if ( !getpeername(m_SocketData.hSocket, lpSockAddr, lpSockAddrLen) )
- return TRUE;
- else
- return FALSE;
- }
- #ifdef _AFX
- BOOL CAsyncSocketEx::GetSockName(CString& rSocketAddress, UINT& rSocketPort)
- {
- SOCKADDR_IN sockAddr;
- memset(&sockAddr, 0, sizeof(sockAddr));
- int nSockAddrLen = sizeof(sockAddr);
- BOOL bResult = GetSockName((SOCKADDR*)&sockAddr, &nSockAddrLen);
- if (bResult)
- {
- rSocketPort = ntohs(sockAddr.sin_port);
- rSocketAddress = inet_ntoa(sockAddr.sin_addr);
- }
- return bResult;
- }
- #endif
- BOOL CAsyncSocketEx::GetSockName( SOCKADDR* lpSockAddr, int* lpSockAddrLen )
- {
- if ( !getsockname(m_SocketData.hSocket, lpSockAddr, lpSockAddrLen) )
- return TRUE;
- else
- return FALSE;
- }
- BOOL CAsyncSocketEx::ShutDown( int nHow /*=sends*/ )
- {
- #ifndef NOLAYERS
- if (m_pFirstLayer)
- {
- return m_pFirstLayer->ShutDown();
- }
- else
- #endif
- {
- if ( !shutdown(m_SocketData.hSocket, nHow) )
- return TRUE;
- else
- return FALSE;
- }
- }
- SOCKET CAsyncSocketEx::Detach( )
- {
- SOCKET socket=m_SocketData.hSocket;
- DetachHandle(socket);
- m_SocketData.hSocket=INVALID_SOCKET;
- return socket;
- }
- BOOL CAsyncSocketEx::Attach( SOCKET hSocket, long lEvent /*= FD_READ | FD_WRITE | FD_OOB | FD_ACCEPT | FD_CONNECT | FD_CLOSE*/ )
- {
- if (hSocket==INVALID_SOCKET || !hSocket)
- return FALSE;
- VERIFY(InitAsyncSocketExInstance());
- m_SocketData.hSocket=hSocket;
- AttachHandle(hSocket);
- return AsyncSelect(lEvent);
- }
- BOOL CAsyncSocketEx::AsyncSelect( long lEvent /*= FD_READ | FD_WRITE | FD_OOB | FD_ACCEPT | FD_CONNECT | FD_CLOSE*/ )
- {
- ASSERT(m_pLocalAsyncSocketExThreadData);
- m_lEvent=lEvent;
- #ifndef NOLAYERS
- if (!m_pFirstLayer)
- #endif //NOLAYERS
- {
- if ( !WSAAsyncSelect(m_SocketData.hSocket, GetHelperWindowHandle(), m_SocketData.nSocketIndex+WM_SOCKETEX_NOTIFY, lEvent) )
- return TRUE;
- else
- return FALSE;
- }
- return TRUE;
- }
- BOOL CAsyncSocketEx::Listen( int nConnectionBacklog /*=5*/ )
- {
- #ifndef NOLAYERS
- if (m_pFirstLayer)
- return m_pFirstLayer->Listen(nConnectionBacklog);
- #endif //NOLAYERS
- if (!listen(m_SocketData.hSocket, nConnectionBacklog))
- return TRUE;
- else
- return FALSE;
- }
- BOOL CAsyncSocketEx::Accept( CAsyncSocketEx& rConnectedSocket, SOCKADDR* lpSockAddr /*=NULL*/, int* lpSockAddrLen /*=NULL*/ )
- {
- ASSERT(rConnectedSocket.m_SocketData.hSocket == INVALID_SOCKET);
- #ifndef NOLAYERS
- if (m_pFirstLayer)
- {
- return m_pFirstLayer->Accept(rConnectedSocket, lpSockAddr, lpSockAddrLen);
- }
- else
- #endif //NOLAYERS
- {
- SOCKET hTemp = accept(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 CAsyncSocketEx::IOCtl( long lCommand, DWORD* lpArgument )
- {
- return ioctlsocket(m_SocketData.hSocket, lCommand, lpArgument) != SOCKET_ERROR;
- }
- int CAsyncSocketEx::GetLastError()
- {
- return WSAGetLastError();
- }
- BOOL CAsyncSocketEx::TriggerEvent(long lEvent)
- {
- if (m_SocketData.hSocket==INVALID_SOCKET)
- return FALSE;
- ASSERT(m_pLocalAsyncSocketExThreadData);
- ASSERT(m_pLocalAsyncSocketExThreadData->m_pHelperWindow);
- ASSERT(m_SocketData.nSocketIndex!=-1);
- #ifndef NOLAYERS
- if (m_pFirstLayer)
- {
- CAsyncSocketExLayer::t_LayerNotifyMsg *pMsg=new CAsyncSocketExLayer::t_LayerNotifyMsg;
- pMsg->lEvent=lEvent%0xFFFF;
- pMsg->pLayer=0;
- BOOL res=PostMessage(GetHelperWindowHandle(), WM_USER, (WPARAM)this, (LPARAM)pMsg);
- if (!res)
- delete pMsg;
- return res;
- }
- else
- #endif //NOLAYERS
- return PostMessage(GetHelperWindowHandle(), m_SocketData.nSocketIndex+WM_SOCKETEX_NOTIFY, m_SocketData.hSocket, lEvent%0xFFFF);
- }
- SOCKET CAsyncSocketEx::GetSocketHandle()
- {
- return m_SocketData.hSocket;
- }
- HWND CAsyncSocketEx::GetHelperWindowHandle()
- {
- if (!m_pLocalAsyncSocketExThreadData)
- return 0;
- if (!m_pLocalAsyncSocketExThreadData->m_pHelperWindow)
- return 0;
- return m_pLocalAsyncSocketExThreadData->m_pHelperWindow->GetHwnd();
- }
- #ifndef NOLAYERS
- BOOL CAsyncSocketEx::AddLayer(CAsyncSocketExLayer *pLayer)
- {
- ASSERT(pLayer);
- if (m_SocketData.hSocket!=INVALID_SOCKET)
- return FALSE;
- if (m_pFirstLayer)
- {
- ASSERT(m_pLastLayer);
- m_pLastLayer=m_pLastLayer->AddLayer(pLayer, this);
- return m_pLastLayer?TRUE:FALSE;
- }
- else
- {
- ASSERT(!m_pLastLayer);
- pLayer->Init(0, this);
- m_pFirstLayer=pLayer;
- m_pLastLayer=m_pFirstLayer;
- }
- return TRUE;
- }
- void CAsyncSocketEx::RemoveAllLayers()
- {
- m_pFirstLayer=0;
- m_pLastLayer=0;
- }
- int CAsyncSocketEx::OnLayerCallback(const CAsyncSocketExLayer *pLayer, int nType, int nParam1, int nParam2)
- {
- ASSERT(pLayer);
- return 1;
- }
- #endif //NOLAYERS
- BOOL CAsyncSocketEx::GetSockOpt(int nOptionName, void* lpOptionValue, int* lpOptionLen, int nLevel /*=SOL_SOCKET*/)
- {
- return (SOCKET_ERROR != getsockopt(m_SocketData.hSocket, nLevel, nOptionName, (LPSTR)lpOptionValue, lpOptionLen));
- }
- BOOL CAsyncSocketEx::SetSockOpt(int nOptionName, const void* lpOptionValue, int nOptionLen, int nLevel /*=SOL_SOCKET*/)
- {
- return (SOCKET_ERROR != setsockopt(m_SocketData.hSocket, nLevel, nOptionName, (LPSTR)lpOptionValue, nOptionLen));
- }