WInsock

使用CSoket多次了，但对于它的block模式的理解并不是很深入。昨天使用csoket的udp多发测试(server接到数据后，需要通过某种方式将数据发送到client,使用tcp方式比较可靠，我一直这样用的，但是比较费时，需要逐一发送)，发现了问题：

  1）create(),sendto(),receivefrom()....

  2）其中，发送方一直定时发送数据无问题；

  3）而接收方，通过一个单独的接收线程实现( 注意：csocket不能跨线程使用！主线程中socket create()后，detach()并将sock作为lpvoid传入接收线程 )。代码如下：

  //处理接收数据
UINT CSockSvr::DealSvrRevData(LPVOID lParam)
{

  ......
 DWORD dwError;
 TCHAR cBuff[1000];
 CString sIP;
 UINT uPort;
 for(;!pDlg->m_bExit;)
 {
  ::memset( cBuff,0,sizeof(cBuff) );

  // 如果没有接到数据，一直等待。。。。
  // 阻塞模式的弊端：：：在退出时候，通过CancelBlockingCall
   int iRst=SockSvr.ReceiveFrom( cBuff,sizeof(cBuff),sIP,uPort,0 );
    if( iRst!=SOCKET_ERROR )
  {
   CString sTemp=cBuff;
   TRACE1( _T("/r/n Rev Data: %s/r/n"),sTemp );
  }
  else
  {
   dwError=GetLastError();
   TRACE1( _T("/r/n Rev Data Error code: %d/r/n"),dwError );
  }
  ::Sleep(200);
 }
 
 return 0;
}

  问题：如果发送方没有数据，SockSvr.ReceiveFrom()会一致处于等待状态，导致整个处理接收线程的停止，如果用户需要退出/结束程序，无法正确释放资源（无法关闭在线程函数重打开的socket--不能跨线程操作socket，无法关闭线程)，造成系统memory leak...

  针对这种问题，微软的解决办法是：http://support.microsoft.com/default.aspx?scid=kb;zh-cn;138692 ，经过测试，在socket进行connect()的时候不好用。后来，结合微软的办法，我通过自定义的定时器，实现了block超时间后自动退出的功能〉

  #if !defined(AFX_TIMEOUTSOCK_H__19897A81_4EAF_4005_91FD_DC3047725139__INCLUDED_)
#define AFX_TIMEOUTSOCK_H__19897A81_4EAF_4005_91FD_DC3047725139__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000
// TimeOutSock.h : header file
//

/*
CSocket 操作，如接收(Receive)、发送(Send) 和连接(Connect) 均是阻塞操作，
 即要等到操作成功执行完毕或套接字上出现错误后，对这些函数的调用才有返回结果。
某些情况下，操作可能永远不能成功完成，这将导致程序无限循环等待操作完成。
 一种解决方法是通过编程限制完成操作使用的时间。本文将讨论这种方法。
*/
/////////////////////////////////////////////////////////////////////////////
// CTimeOutSock command target

class CTimeOutSock : public CSocket
{
// Attributes
public:

// Operations
public:
 CTimeOutSock();
 virtual ~CTimeOutSock();

// Overrides
public:
 // ClassWizard generated virtual function overrides
 //{{AFX_VIRTUAL(CTimeOutSock)
 public:
 virtual BOOL OnMessagePending();
 //}}AFX_VIRTUAL

 // Generated message map functions
 //{{AFX_MSG(CTimeOutSock)
  // NOTE - the ClassWizard will add and remove member functions here.
 //}}AFX_MSG

/*
//定时器设置block超时 不是很好用。。。徐卫话 2006.4.19
CSocket 操作，如接收(Receive)、发送(Send) 和连接(Connect) 均是阻塞操作，
 即要等到操作成功执行完毕或套接字上出现错误后，对这些函数的调用才有返回结果。
某些情况下，操作可能永远不能成功完成，这将导致程序无限循环等待操作完成。
 一种解决方法是通过编程限制完成操作使用的时间。本文将讨论这种方法。
*/
// 自己计算时间的办法 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();
}

 

  经过改进后， 对处理接收线成的函数进行了部分改进：

  SockSvr.SetTimeOut(500);  int iRst=SockSvr.ReceiveFrom( cBuff,sizeofcBuff),sIP,uPort,0 );  SockSvr.KillTimeOut();

 当block超过定时后，socket自动退出block，防止接收线程停止。问题终于解决了！