MFC的CSocket的一个小Bug？

今天写的程序用到了MFC的CSocket类 。

首先在一个自己的线程中调用这个CSocket类对象的创建函数CSocket::Create()，这个线程用来执行ReSipRocate的协议栈。

然后当点击程序窗口的菜单时，程序的主线程（UI线程）调用销毁函数CSocket::Close()。当程序是Debug版本的时候，会报告一个断言错误“Debug Assertion Failed. File: sockcore.cpp. Line: 544”。（我用的是VC.net 2003。）

打开文件看到这个断言错误发生在CSocket的积累CAsyncSocket的处理中。我来回的分析我的CSocket对象的各种操作，确认各个操作均没有错误。好在能单步调试sockcore.cpp文件里的函数。可以看到断言的位置为（标为红色字体）：

void PASCAL CAsyncSocket::KillSocket(SOCKET hSocket, CAsyncSocket* pSocket)
{
 ASSERT(CAsyncSocket::LookupHandle(hSocket, FALSE) != NULL);

 _AFX_SOCK_THREAD_STATE* pState = _afxSockThreadState;

 CAsyncSocket::DetachHandle(hSocket, FALSE);
 if (pState->m_hSocketWindow != NULL)
 {
  ::PostMessage(pState->m_hSocketWindow, WM_SOCKET_DEAD,
   (WPARAM)hSocket, 0L);
  CAsyncSocket::AttachHandle(hSocket, pSocket, TRUE);
 }
}

于是找到LookupHandle()的定义，如下：

CAsyncSocket* PASCAL CAsyncSocket::LookupHandle(SOCKET hSocket, BOOL bDead)
{
 CAsyncSocket* pSocket;
 _AFX_SOCK_THREAD_STATE* pState = _afxSockThreadState;
 if (!bDead)
 {
  pSocket = (CAsyncSocket*)
   pState->m_pmapSocketHandle->GetValueAt((void*)hSocket);
  if (pSocket != NULL)
   return pSocket;
 }

。。。
 }

看出来问题出在pState->m_pmapSocketHandle->GetValueAt((void*)hSocket)，在根据SOCKET hSocket查找对应的CAsyncSocket对象指针时失败了。于是跟踪m_pmapSocketHandle这个结构的所有操作的地方。在程序运行的全过程中，只看到将hSocket放入到m_pmapSocketHandle中的操作（pState->m_pmapSocketHandle->SetAt((void*)hSocket, pSocket)）。没有看到从m_pmapSocketHandle中移除hSocket的操作（pState->m_pmapSocketHandle->RemoveKey((void*)hSocket)）。然而为什么使用这个hSocket来查找的时候会失败呢（pState->m_pmapSocketHandle->GetValueAt((void*)hSocket)）？

开始分析的时候觉得是不是程序哪里越界了，导致这个数据结构被破坏了？太可怕了，不愿意相信，还是再看看有没有别的可能。再看看 _AFX_SOCK_THREAD_STATE* pState = _afxSockThreadState这句语句。发现_afxSockThreadState不是一个全局变量，而是一个宏定义。展开来是：

#define _afxSockThreadState AfxGetModuleThreadState()

进一步跟进：

AFX_MODULE_THREAD_STATE* AFXAPI AfxGetModuleThreadState()
{
 return AfxGetModuleState()->m_thread.GetData();
}

再往下就发现这个这个pState是线程相关的，不同的线程拥有各自不同的pState。当调用CSocket::Create()创建socket时，SIP协议栈线程把这个socket记录到自己的pState->m_pmapSocketHandle中。等调用CSocket::Close()关闭这个socket时，UI线程在关闭前先去自己的pState->m_pmapSocketHandle中查找这个socket。（当然查不到），所以断言失败。

不知道这是一个Bug还是故意这么设计的。