MFC网络编程聊天室之服务器

样式

开启服务器和关闭服务器按钮 下面是一个list control

需要先创建CListenSocket类和CListenSocket类 基类都为CSocket 后面细说

给控件初始化

//修改按钮的状态
GetDlgItem(IDC_BTN_START)->EnableWindow(TRUE);      //开启服务器按钮可以点击
GetDlgItem(IDC_BTN_CLOSE)->EnableWindow(FALSE);     //关闭服务器按钮不可以点击
m_list.SetExtendedStyle(LVS_EX_FULLROWSELECT | LVS_EX_GRIDLINES);  //整行选中 画网格线拓展
m_list.InsertColumn(0, L"IP地址", LVCFMT_LEFT, 200);
m_list.InsertColumn(1, L"端口", LVCFMT_LEFT, 100);

开启服务器

void CTCPChatServerDlg::OnBnClickedBtnStart()
{
    //CListenSocket类对象 这里用来创建套接字  监听 
	m_pListenSocket = new CListenSocket;  
    
	//创建套接字
	if (FALSE == m_pListenSocket->Create(PORT, SOCK_STREAM))  //#define PORT	2022
	{
		CString strMsg;
		strMsg.Format(L"创建套接字失败.错误代号:%d", GetLastError());
		MessageBox(strMsg);
		return;
	}

	//将套接字设置为监听模式
	if (FALSE == m_pListenSocket->Listen())
	{
		CString strMsg;
		strMsg.Format(L"监听失败.错误代号:%d", GetLastError());
		MessageBox(strMsg);
		return;
	}

	//修改按钮的状态
	GetDlgItem(IDC_BTN_START)->EnableWindow(FALSE);
	GetDlgItem(IDC_BTN_CLOSE)->EnableWindow(TRUE);
}

CListenSocket类

接受并建立连接请请求 类向导添加OnAccept虚函数

void CListenSocket::OnAccept(int nErrorCode)
{
	//接受并建立连接请求
    //这个对象就是客户端连接上服务器后 服务器这边的对象 用来接收客户端发送的消息和转发消息 
    //与客户端对象一一对应
	CClientSocket *pSocket = new CClientSocket;
	Accept(*pSocket);`  

	//往链表里面塞 
    //g_ClientSocketList用来存储着所有客户端的套接字地址  用来转发消息给其他客户端
	g_ClientSocketList.push_back(pSocket);

	CString strIPAddress;
	UINT uPort;

	pSocket->GetPeerName(strIPAddress, uPort);  //得到连接窗口的客户端ip地址和端口号

	//获取主窗口 显示连接客户端的信息
	CTCPChatServerDlg* pMainDlg = (CTCPChatServerDlg*)AfxGetMainWnd();
	pMainDlg->InsertUser(strIPAddress, uPort, pSocket);
	
	CSocket::OnAccept(nErrorCode);
}

InsertUser(strIPAddress, uPort, pSocket)

//插入用户
void CTCPChatServerDlg::InsertUser(CString strIPAddress, UINT uPort, void* pSocket)
{
	//获取总行数
	int nCount = m_list.GetItemCount(); // m_list为list control的变量

	m_list.InsertItem(nCount, strIPAddress);

	CString strPort;
	strPort.Format(L"%d", uPort);
   	m_list.SetItemText(nCount, 1, strPort);   
    m_list.SetItemData(nCount, (DWORD)pSocket);  //插入隐藏数据
}  

CClientSocket类

接收客户端发来的消息同时转发消息 类向导添加OnReceive虚函数

void CClientSocket::OnReceive(int nErrorCode)
{
	//接受数据
	TCHAR szRecvMsg[1025];
	//清零
	ZeroMemory(szRecvMsg, sizeof(szRecvMsg));

	int nRealRecv = Receive(szRecvMsg, sizeof(szRecvMsg));

	//获取客户端的IP地址和端口号
	CString strIPAddress;
	UINT uPort;
	GetPeerName(strIPAddress, uPort);

	CTime time = CTime::GetCurrentTime();

	CString strSendMsg;//重新发送给其他客户端的文本
	strSendMsg.Format(L"%s:%d\t%s\r\n\t%s\r\n", strIPAddress, uPort, time.Format(L"%Y-%m-%d %H:%M:%S"), szRecvMsg);

	//转发给别人，或者所有人
	std::list<CClientSocket*>::iterator it;

	for (it = g_ClientSocketList.begin(); it != g_ClientSocketList.end(); it++)
	{
		//取出套接字指针
		CClientSocket* pSocket = *it;
		if (pSocket != this)  //不发给自己
			pSocket->Send(strSendMsg, strSendMsg.GetLength() * 2);
	}

	CSocket::OnReceive(nErrorCode);
}

处理客户端断开连接 类向导添加OnClose虚函数

void CClientSocket::OnClose(int nErrorCode)
{
	//获取客户端的IP地址和端口号
	CString strMyIPAddress;
	UINT uMyPort;
	GetPeerName(strMyIPAddress, uMyPort);
	 
	//获取主窗口
	CTCPChatServerDlg* pMainDlg = (CTCPChatServerDlg*)AfxGetMainWnd();
	//先从界面上删除
	int nCount = pMainDlg->m_list.GetItemCount();
	for (int i = 0; i < nCount;i++)
	{
		//获取IP地址和端口号
		CString strIPAddress = pMainDlg->m_list.GetItemText(i, 0);
		UINT uPort = _wtoi(pMainDlg->m_list.GetItemText(i, 1));
		CClientSocket* pSocket = (CClientSocket*)pMainDlg->m_list.GetItemData(i);

		if (strMyIPAddress == strIPAddress && uMyPort == uPort && pSocket == this)
		{
			pMainDlg->m_list.DeleteItem(i);
			break;
		}
	}


	//从链表中删除
	std::list<CClientSocket*>::iterator it;

	for (it = g_ClientSocketList.begin(); it != g_ClientSocketList.end(); it++)
	{
		if (this == *it)
		{
			g_ClientSocketList.erase(it);
			delete this;
			return;
		}
	}



	CSocket::OnClose(nErrorCode);
}

关闭服务器

void CTCPChatServerDlg::OnBnClickedBtnClose()
{
	if (m_pListenSocket)
	{
		m_pListenSocket->Close();

		delete m_pListenSocket;
		m_pListenSocket = NULL;

		//修改按钮的状态
		GetDlgItem(IDC_BTN_START)->EnableWindow(TRUE);
		GetDlgItem(IDC_BTN_CLOSE)->EnableWindow(FALSE);
	}
}