浅析CAsyncSocket

    首先CAsyncSocket采用的WSAAsynSelect模型，WSAAsynSelect是一种异步I/O模型，通过该模型，应用程序可以接收以Windows消息为基础的网络事件通知。而我们这里讲的就是CSocketWnd，它是从CWnd继承的。

class CSocketWnd : public CWnd { public: CSocketWnd(); protected: LRESULT OnSocketNotify(WPARAM wParam, LPARAM lParam); LRESULT OnSocketDead(WPARAM wParam, LPARAM lParam); DECLARE_MESSAGE_MAP() };

从源码中看出它只处理两个消息，一个是WM_SOCKET_NOTIFY,另一个是WM_SOCKET_DEAD。

 

我原以为没创建一个Socket就会创建一个CSocketWnd,其实不是，是所有的Socket共享一个CSocketWnd,其实想想也是，要是每创建一个Socket就创建一个CSocketWnd那开销也大了点。

 

既然是共享的，那它放在哪里呢?

全局中有这样一个结构体：_AFX_SOCK_THREAD_STATE，它保存一个那个共享的窗口句柄：m_hSocketWindow。

当没有创建Socket时，它是Null的。

除了m_hSocketWindow结构体_AFX_SOCK_THREAD_STATE还有几个重要的变量：

1. mCEmbeddedButActsLikePtr<CMapPtrToPtr> m_pmapSocketHandle：记录了CAysncSocket*与没死的SOCKET的映射关系

2. CEmbeddedButActsLikePtr<CMapPtrToPtr> m_pmapDeadSockets：记录了SOCKET与死了的次数的映射关系

3. CEmbeddedButActsLikePtr<CPtrList> m_plistSocketNotifications：记录的所有的SOCKET事件的通知消息

 

每当调用Attach时，Attach调用AttachHandle,在AttachHandle把一个CAysncSocket*和SOCKET加入m_pmapSocketHandle中，

当调用Socket时也会调用AttachHandle，与之对于是Detach.

 

当一个非死亡事件发生时，会向m_hSocketWindow发送WM_SOCKET_NOTIFY，其消息处理函数如下

LRESULT CSocketWnd::OnSocketNotify(WPARAM wParam, LPARAM lParam) { CSocket::AuxQueueAdd(WM_SOCKET_NOTIFY, wParam, lParam); CSocket::ProcessAuxQueue(); return 0L; }

 

CSocket::AuxQueueAdd(WM_SOCKET_NOTIFY, wParam, lParam);源码如下：

void PASCAL CSocket::AuxQueueAdd(UINT message, WPARAM wParam, LPARAM lParam) { _AFX_SOCK_THREAD_STATE* pState = _afxSockThreadState; MSG* pMsg = new MSG; pMsg->message = message; pMsg->wParam = wParam; pMsg->lParam = lParam; pState->m_plistSocketNotifications->AddTail(pMsg); }

接到消息以后把它放入：m_plistSocketNotification中

 

CSocket::ProcessAuxQueue(WM_SOCKET_NOTIFY, wParam, lParam)；源码如下：

int PASCAL CSocket::ProcessAuxQueue() { _AFX_SOCK_THREAD_STATE* pState = _afxSockThreadState; if (pState->m_plistSocketNotifications->IsEmpty()) return 0; int nCount = 0; while(!pState->m_plistSocketNotifications->IsEmpty()) { nCount++; MSG* pMsg = (MSG*)pState->m_plistSocketNotifications->RemoveHead(); ASSERT(pMsg != NULL); //如果是是WM_SOCKET_NOTIFY调用CAsyncSocket::DoCallBack if (pMsg->message == WM_SOCKET_NOTIFY) { CAsyncSocket::DoCallBack(pMsg->wParam, pMsg->lParam); } //如果是死亡事件执行以下代码 else { ASSERT(CAsyncSocket::LookupHandle((SOCKET)pMsg->wParam, TRUE) != NULL); CAsyncSocket::DetachHandle((SOCKET)pMsg->wParam, TRUE); } delete pMsg; } return nCount; }

从中考研看出最终调用的是：CAsyncSocket::DoCallBack，源码如下：

void PASCAL CAsyncSocket::DoCallBack(WPARAM wParam, LPARAM lParam) { if (wParam == 0 && lParam == 0) return; // Has the socket be closed - lookup in dead handle list CAsyncSocket* pSocket = CAsyncSocket::LookupHandle((SOCKET)wParam, TRUE); // If yes ignore message if (pSocket != NULL) return; pSocket = CAsyncSocket::LookupHandle((SOCKET)wParam, FALSE); if (pSocket == NULL) { // Must be in the middle of an Accept call pSocket = CAsyncSocket::LookupHandle(INVALID_SOCKET, FALSE); ENSURE(pSocket != NULL); if(pSocket == NULL) return; pSocket->m_hSocket = (SOCKET)wParam; CAsyncSocket::DetachHandle(INVALID_SOCKET, FALSE); CAsyncSocket::AttachHandle(pSocket->m_hSocket, pSocket, FALSE); } int nErrorCode = WSAGETSELECTERROR(lParam); switch (WSAGETSELECTEVENT(lParam)) { case FD_READ: { fd_set fds; int nReady; timeval timeout; timeout.tv_sec = 0; timeout.tv_usec = 0; FD_ZERO(&fds); FD_SET(pSocket->m_hSocket, &fds); nReady = select(0, &fds, NULL, NULL, &timeout); if (nReady == SOCKET_ERROR) nErrorCode = WSAGetLastError(); if ((nReady == 1) || (nErrorCode != 0)) pSocket->OnReceive(nErrorCode); } break; case FD_WRITE: pSocket->OnSend(nErrorCode); break; case FD_OOB: pSocket->OnOutOfBandData(nErrorCode); break; case FD_ACCEPT: pSocket->OnAccept(nErrorCode); break; case FD_CONNECT: pSocket->OnConnect(nErrorCode); break; case FD_CLOSE: pSocket->OnClose(nErrorCode); break; } }

 

消息WM_SOCKET_DEAD处理函数与WM_SOCKET_NOTIFY是一样的

 

从中考研清楚的看出CAsyncSocket的事件处理过程。

以上一定会有一些错误，请各位路过的大侠指正，不胜感激。