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