listensocket.cpp

来自「一个采用C/S结构的信息发布接收管理程序」· C++ 代码 · 共 56 行

// ListenSocket.cpp : implementation file
//

#include "stdafx.h"
#include "MsgServer.h"
#include "ListenSocket.h"
#include "AcceptSocket.h"
#include "MsgServerDlg.h"

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

/////////////////////////////////////////////////////////////////////////////
// CListenSocket

CListenSocket::CListenSocket( CMsgServerDlg  *pMsgServerDlg)
{
	m_pMsgServerDlg  = pMsgServerDlg;
}

CListenSocket::~CListenSocket()
{
}


// 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

void CListenSocket::OnAccept(int nErrorCode) 
{
	CAcceptSocket *pSocket=new CAcceptSocket(m_pMsgServerDlg);
	CMsgServerDlg* pDlg=(CMsgServerDlg*)::AfxGetMainWnd();

	if(Accept(*pSocket))
	{
		m_pMsgServerDlg->m_pAcceptList.AddTail(pSocket);
	}
	else
	{
		delete pSocket;
	}	
	
	CSocket::OnAccept(nErrorCode);
}