VC下CSocket非阻塞方式下的连接框架

CSocket非阻塞方式(不用自己建立线程)

1.服务端
应用非阻塞方式需要自己声明一个类继承CSokcet类

// MySocket.h : header file
class CMySocket : public CSocket
{
// Attributes
public:
	CServerDlg* m_pServerDlg;	//主窗口指针
// Operations
public:
	CMySocket();
	virtual ~CMySocket();

// Overrides
public:
	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CMySocket)
	public:
	virtual void OnAccept(int nErrorCode);	//当准备连接时,系统调用
	virtual void OnReceive(int nErrorCode);	//当准备接受消息时,系统调用
	//}}AFX_VIRTUAL

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

// Implementation
protected:

};

// MySocket.cpp : implementation file

CMySocket::CMySocket(CServerDlg* pServerDlg)
{
	m_pServerDlg = pServerDlg;	//用于处理消息
}

CMySocket::~CMySocket()
{
}

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

/////////////////////////////////////////////////////////////////////////////
// CMySocket member functions

void CMySocket::OnAccept(int nErrorCode) 
{
	// TODO: Add your specialized code here and/or call the base class
	//处理连接请求
	m_pServerDlg->ServerAccept();
	
	CSocket::OnAccept(nErrorCode);
}

void CMySocket::OnReceive(int nErrorCode) 
{
	// TODO: Add your specialized code here and/or call the base class
	//处理接受消息请求
	m_pServerDlg->ServerReceive(this);	//传入本身Socket

	CSocket::OnReceive(nErrorCode);
}

////////////////////////////////////////////////////
// ServerDlg.h : header file

CMySocket* m_pServer;	//添加Socket指针
CPtrArray m_ConnectArray;	//用于保存连接指针的容器

////////////////////////////////////////////////////
// ServerDlg.cpp : implementation file

void CServerDlg::ServerInit()	//服务端初始化,供别的函数调用
{
	m_pServer = new CMySocket(this);	//传入窗口指针
	if(!m_pServer->Create(61111))	//指明端口
	{
		MessageBox("服务器初始化错误n程序退出");
		delete m_pServer;
		return;
	}
	if(!m_pServer->Listen())	//开始监听
	{
		MessageBox("服务器初始化错误n程序退出");
		delete m_pServer;
		return;
	}
}
void CServerDlg::ServerAccept()	//有客户端连接,系统调用
{
	CMySocket* connect = new CMySocket(this);	//建立连接用Socket

	if(!m_pServer->Accept(*connect))
	{
		MessageBox("客户端连接错误!");
		delete connect;
	}
	
	m_ConnectArray.Add(connect);	//把这个连接添加进容器
}
void CServerDlg::ServerReceive(CMySocket* connect)	//连接发来数据,准备接收
{
	char msg[100];
	connect->Receive(msg,100);	//接受数据
	MessageBox(CString(msg));
}
void CServerDlg::SocketSend(CMySocket* connect, CString msg)	//Socket发送数据
{
	char* str= msg.GetBuffer(msg.GetLength());
	connect->Send(str,msg.GetLength());	//发送数据
}
void CServerDlg::ServerSendToAllConnect(char* pChar)	//发送给所有客户端数据的函数
{
	for(int i=0; i<=m_ConnectArray.GetSize()-1; i++)	//遍历连接容器
	{
		((CMySocket*)m_ConnectArray[i])->Send(pChar,100);	//发送
	}
}
void CServerDlg::ServerClose()
{
	//发送踢出消息,使客户端自动断开连接
	//ServerSendToAllConnect("01");	//用户实现方式不同
	
	//清理服务器连接
	m_pServer->Close();
	delete m_pServer;

	//清理客户端连接
	for(int i=0; i<=m_ConnectArray.GetSize()-1; i++)
	{
		((CMySocket*)m_ConnectArray[i])->Close();
		delete (CMySocket*)m_ConnectArray[i];
	}
	m_ConnectArray.RemoveAll();
}

注意:客户端连接与断开时,必须根据实际情况,更新存放连接容器,否则会出现错误

2.客户端

//也使用了CMySocket,代码同服务端CMySokcet一样,只不过不用响应OnAccept事件

//////////////////////////////////////////////////
// ClientDlg.h : header file
CMySocket* m_pClient;	//本身连接用Socket

//////////////////////////////////////////////////
// ClientDlg.cpp : implementation file

bool CClientDlg::ClientInit()
{
	m_pClient = new CMySocket(this);
	if(!m_pClient->Create())
	{
		delete m_pClient;
		MessageBox("客户端初始化错误!n");
		return 0;
	}

	if(!WSAEINVAL==m_pClient->Connect("127.0.0.1",61111))	//连接服务器IP的某端口(端口须和上面一直)
	{
		delete m_pClient;
		MessageBox("客户端初始化错误!n");
		return 0;
	}

	return 1;
}
void CClientDlg::ClientReceive(CMySocket* Socket)	//客户端接受数据
{
	char ReceiveChar[100];
	Socket->Receive(ReceiveChar,100);
	CString ReceiveMsg(ReceiveChar);
	MessageBox(ReceiveMsg);
}
void CClientDlg::ClientSend(CString msg)	//客户端发送数据
{
	char* str= msg.GetBuffer(msg.GetLength());
	m_pClient->Send(str,msg.GetLength());	//发送数据
}
bool CClientDlg::ClientClose()		//客户端断开连接
{
	//应发送数据,告诉服务端,自己断开连接,更新服务端容器内的连接
	//m_Client->Send("02",100);//用户实现方式不同
	
	if(m_pClient)
	{
		m_pClient->Close();
		delete m_pClient;
		return 1;
	}
	return 0;
}

注意:客户端连接与断开时,必须根据实际情况,让服务端只是,自己连接与断开了,及时更新服务端的存放连接的容器

下面是我做的一个超级简单的聊天室的流程图,仅供参考