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.5二、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();
}