listensocket.cpp

来自「这是VC学习的好资料,大家抽时间看看,一样画葫芦,画啊画,」· C++ 代码 · 共 86 行

// ListenSocket.cpp : implementation file
//

#include "stdafx.h"
#include "ListenSocket.h"

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

/////////////////////////////////////////////////////////////////////////////
// CListenSocket
//构造函数，用于初始化
CListenSocket::CListenSocket(CInterface *pTemp,CMyWaveIn *pIn,CUdpSocket *pUdp):
	m_sopClient (NULL),
	m_pIn(NULL),
	m_pUdp(NULL)
{
	m_sopClient = new CClientSocket(pTemp,pIn,pUdp);
	m_pInterface = pTemp;
	m_pIn = pIn;
	m_pUdp = pUdp;
}
//析构函数
CListenSocket::~CListenSocket()
{
	delete m_sopClient;
}


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

/////////////////////////////////////////////////////////////////////////////
// CListenSocket member functions
//重载OnAccept() 函数
void CListenSocket::OnAccept(int nErrorCode) 
{
	//客户端地址
	SOCKADDR add;
	//客户端地址长度
	int iLen;
	iLen = sizeof(SOCKADDR);
	//临时socket
	CSocket soTemp;

	//已经连接
	if (m_sopClient->m_bConnect )
	{
		//用临时socket接收连接请求
		soTemp.Accept (*this);
		//关闭
		soTemp.Close ();
		//返回
		return ;
	}

	//接受连接请求
	if (!Accept(*m_sopClient,&add,&iLen))
	{
		//返回
		return ;
	}

	//设置连接标记
	m_sopClient->m_bConnect = TRUE;

	//调用基类OnAccept()函数
	CSocket::OnAccept(nErrorCode);
}
//关闭套接字
void CListenSocket::CloseClient()
{
	//关闭套接字
	m_sopClient->Close ();
	//设置未连接标记
	m_sopClient->m_bConnect = FALSE;
}