CSocket同步操作阻塞时设置超时的解决方案

    本文讲解CSocket同步操作阻塞时设置超时的解决方案。
    最近参加了中兴公司的通信软件设计大赛，开发一个模拟手机和基站信令通信的软件，遇到CSocket发送阻塞的问题，这里有一个简单的解决方案供大家参考。
    CSocket继承自CAsyncSocket，他们的不同是前者是同步套接字，后者是异步套接字，操作都是异步的。Socket中的Receive、Send 和Connect是阻塞操作，我们知道阻塞操作的特点是要么运行成功退出，要么出现错误退出，而且有的时候如果不成功，则该函数会一直阻塞，整个程序会一直等待操作的完成。
    本文提出的解决方法是通过编程限制完成操作使用的时间，设置定时，设定一定的超时时间，如果阻塞操作时间过程，则启动该超时机制。虽然操作是阻塞操作，但是我们可以处理操作中到达的消息。如果通过使用 SetTimer 设置定时器，那么可以查找 WM_TIMER 消息，并在收到该消息时终止操作。
    这种方法比较常见，也比较容易想到，网上也有很多相关的思想。设置定时的方法有::SetTimer函数，具体参加msdn。处理消息过程是CSocket中的CSocket::OnMessagePending 函数，退出阻塞的函数是CSocket中的CSocket::CancelBlockingCall 函数。我们在使用的时候往往是定义自己的CClientSocket类继承自CSocket类，这样这些函数都可以获得了。

    本人通过网上的资料得知，这种方法目前仅适用于 Visual C++ 的 1.52、1.52b、2.1 和 2.2 版本。
    相关函数如下：

BOOL SetTimeOut(UINT uTimeOut)

     调用此函数之后仅接着调用 CSocket 函数（如 Receive、Send 和 Accept）。uTimeOut 参数是以毫秒为单位指定的。之后，进行定时器的设置。如果设置定时器失败，那么函数返回 FALSE。有关详细情况，请参阅 SetTimer 函数的 Windows API 文档。

BOOL KillTimeOut()

     在完成阻塞操作后，必须调用此函数。此函数删除用 SetTimeOut 设置的定时器。如果调用 KillTimer 失败，则返回 FALSE。有关详细情况，请参阅 KillTimer 函数的 Windows API 文档。

BOOL OnMessagePending()

     这是一个虚拟回调函数，在等待操作完成时由 CSocket 类进行调用。此函数给您提供处理传入消息的机会。此实施过程检查用 SetTimeOut 调用函数设置的定时器的 WM_TIMER 消息。如果收到消息，则调用 CancelBlockingCall 函数。有关 OnMessagePending 和 CancelBlockingCall 函数详细的信息，请参阅 MFC 文档。请注意：调用 CancelBlockingCall 函数 将导致操作失败，而且 GetLastError 函数返回 WSAEINTR（表示操作中断）。

    相关实现如下：
头文件

class CClientSocket : public CSocket
{
   //friend CClientThread;
// Attributes
public:
   //CMCSServerDlg* m_dlgServer;
   //CWinThread* m_pThread;

private:
   int m_nTimerID;
// Overrides
public:
  BOOL KillTimeOut();
  BOOL SetTimeOut(UINT uTimeOut);
   // ClassWizard generated virtual function overrides
   //{{AFX_VIRTUAL(CClientSocket)
   public:
   virtual void OnReceive( int nErrorCode);
   virtual void OnClose( int nErrorCode);
   virtual BOOL OnMessagePending();
   //}}AFX_VIRTUAL

//设置超时
BOOL CClientSocket::SetTimeOut(UINT uTimeOut)
{
  m_nTimerID = SetTimer(NULL,0,uTimeOut,NULL);
   return m_nTimerID;
}

//取消设置超时
BOOL CClientSocket::KillTimeOut()
{
   return KillTimer(NULL,m_nTimerID);
}

//用于CSocket函数阻塞时，如果超时，则退出该阻塞函数
BOOL CClientSocket::OnMessagePending()
{
  MSG msg;
   if(::PeekMessage(&msg, NULL, WM_TIMER, WM_TIMER, PM_NOREMOVE)) {
     if (msg.wParam == (UINT) m_nTimerID) {
      ::PeekMessage(&msg, NULL, WM_TIMER, WM_TIMER, PM_REMOVE);
      CancelBlockingCall();
       return FALSE;
    };
  };

   return CSocket::OnMessagePending();
}

    使用这种机制的代码如下：

if(!sockServer.SetTimeOut(10000))
{
     // 错误处理，不能建立定时
}
if(!sockServer.Accept(sockAccept))
{
       int nError = GetLastError();
       if(nError==WSAEINTR)
             //重传处理等等
        else
             //错误处理
}
i f(!sockServer.KillTimeOut())
{
    //不能取消定时，错误处理
}

网上的另一种实现：

// 自己计算时间的办法 OK
public:
         BOOL SetTimeOut(UINT uTimeOut=1000);
         BOOL KillTimeOut();
private:
    LONGLONG m_llDtStart;
    UINT    m_uTimeOut;
};

/////////////////////////////////////////////////////////////////////////////

//{{AFX_INSERT_LOCATION}}
// Microsoft Visual C++ will insert additional declarations immediately before the previous line.

#endif // !defined(AFX_TIMEOUTSOCK_H__19897A81_4EAF_4005_91FD_DC3047725139__INCLUDED_)

     // TimeOutSock.cpp : implementation file
//

#include "stdafx.h"
#include "NetBroad.h"
#include "TimeOutSock.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CTimeOutSock

CTimeOutSock::CTimeOutSock()
{

}

CTimeOutSock::~CTimeOutSock()
{
}


// Do not edit the following lines, which are needed by ClassWizard.
# if 0
BEGIN_MESSAGE_MAP(CTimeOutSock, CSocket)
//{{AFX_MSG_MAP(CTimeOutSock)
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
#endif // 0

/////////////////////////////////////////////////////////////////////////////
// CTimeOutSock member functions

//设置超时
BOOL CTimeOutSock::SetTimeOut(UINT uTimeOut)
{    
//get start cnt
LARGE_INTEGER llCnt;
::QueryPerformanceCounter(&llCnt);
m_llDtStart=llCnt.QuadPart;
m_uTimeOut=uTimeOut;

         return TRUE;
}

//删除超时参数
BOOL CTimeOutSock::KillTimeOut()
{
m_llDtStart=0; //表明取消计时
return TRUE;
}

//检查是否超时间
BOOL CTimeOutSock::OnMessagePending()
{
// TODO: Add your specialized code here and/or call the base class
/*
MSG msg;
    if(::PeekMessage(&msg, NULL, WM_TIMER, WM_TIMER, PM_NOREMOVE))
    {
     if (msg.wParam == (UINT) m_nTimerID)
     {
        // Remove the message and call CancelBlockingCall.
        ::PeekMessage(&msg, NULL, WM_TIMER, WM_TIMER, PM_REMOVE);
        CancelBlockingCall();
        return FALSE;    // No need for idle time processing.
     };
    };
*/
if( m_llDtStart )
{
    LARGE_INTEGER lldtEnd;
    ::QueryPerformanceCounter(&lldtEnd);    
    LARGE_INTEGER llFrq;
    ::QueryPerformanceFrequency(&llFrq);
     double dbDealy=( double)(lldtEnd.QuadPart-m_llDtStart)*1000/llFrq.QuadPart;
     if( dbDealy>m_uTimeOut )
    {
     CancelBlockingCall();
     return FALSE;     // No need for idle time processing.
    }
}

return CSocket::OnMessagePending();
}