MFC SOCKET通信 WSAAsyncSelect 基于消息的异步套接字

方法如下::

1.版本协商 WSAStartup

WSADATA wsaData;
	//版本协商
	BOOL ret = WSAStartup(MAKEWORD(2,2), &wsaData);
	if(ret != 0)
	{
		addMsg("初始化网络协议失败!");
		return FALSE;
	}
	addMsg("初始化网络协议成功");
	addMsg("服务器开始创建Socket");

2.创建套接字 socket

//创建服务器端套接字
	ServerSock = socket(PF_INET, SOCK_STREAM, 0);
	if(ServerSock == INVALID_SOCKET)
	{
		addMsg("创建套接字失败!");
		closesocket(ServerSock);
		WSACleanup();
		return FALSE;
	}
	addMsg("创建套接字成功");

3.绑定套接字 bind

addMsg("服务器开始绑定");
	//绑定到本地一个端口上
	sockaddr_in localaddr;
	localaddr.sin_family = AF_INET;
	//获取IP地址
	BYTE nf1,nf2,nf3,nf4;
	CString strIP;
	((CIPAddressCtrl*)GetDlgItem(IDC_IPADDRESS_Server))->GetAddress(nf1,nf2,nf3,nf4);
	strIP.Format("%d.%d.%d.%d",nf1,nf2,nf3,nf4);//这里的nf得到的值是IP值了
	localaddr.sin_addr.s_addr = inet_addr(strIP);   //IP地址	
	//获取端口号
	CString strPort;
	((CEdit*)GetDlgItem(IDC_EDIT_ServerPort))->GetWindowText(strPort);
	int i_Port; 
	i_Port=_ttoi(strPort);
	localaddr.sin_port = htons(i_Port);
	

	if(bind(ServerSock ,(struct sockaddr*)&localaddr,sizeof(sockaddr))==SOCKET_ERROR)
	{
		CString str;
		str.Format("绑定IP地址%s端口%s失败!",strIP,strPort);
		addMsg(str);
		closesocket(ServerSock);
		WSACleanup();
		return FALSE;
	}
	CString str;
	str.Format("绑定%s %s成功!",strIP,strPort);
	addMsg(str);;

4.头文件中增加自定义消息

#define NETWORK_EVENT WM_USER+1000//定义网络事件

5.网络事件连接消息 WSAAsyncSelect

addMsg("服务器开始注册网络事件");
	//将SeverSock设置为异步非阻塞模式,并为它注册各种网络异步事件,其中m_hWnd
	//为应用程序的主对话框或主窗口的句柄
	if(WSAAsyncSelect(ServerSock, m_hWnd, NETWORK_EVENT, FD_ACCEPT | FD_CLOSE | FD_READ | FD_WRITE) == SOCKET_ERROR)
	{
		addMsg("注册网络异步事件失败!");
		WSACleanup();
		return FALSE;
	}
	addMsg("注册网络事件成功");

6.监听等待连接

addMsg("开始监听");
	if (listen(ServerSock, 5)==SOCKET_ERROR)
	{
		addMsg("服务器监听失败");
	}
	addMsg("服务器监听成功");

6.添加网络事件响应函数 afx_msg 函数原型 ON_MESSAGE 消息映射 具体实现函数

  LRESULT	 OnNetEvent(WPARAM wParam, LPARAM lParam); //异步事件回调函数

ON_MESSAGE(NETWORK_EVENT,OnNetEvent)

//定义网络异步事件的回调函数
LRESULT  CommunicationsSetting::OnNetEvent(WPARAM wParam, LPARAM lParam)
{
	//调用Winsock API函数,得到网络事件类型
	int iEvent = WSAGETSELECTEVENT(lParam);
	//调用Winsock API函数,得到发生此事件的客户端套接字
	SOCKET CurSock= (SOCKET)wParam;
	switch(iEvent)
	{
	case FD_ACCEPT: //客户端连接请求事件
		OnAccept(CurSock);
		break;
	case FD_CLOSE: //客户端断开事件:
		OnClose(CurSock);
		break;
	case FD_READ: //网络数据包到达事件
		OnReceive(CurSock);
		break;
	case FD_WRITE: //发送网络数据事件
		OnSend(CurSock);
		break;
	default: break;
	}
	return 0;
}

7.Send与Recv函数

void CommunicationsSetting::OnReceive(SOCKET CurSock)
{
	//读出网络缓冲区中的数据包
	addMsg("服务器发来消息");
	if (CurSock==ClientSock[0])
	{
		char recvbuf[12]={0};
		if(recv(ClientSock[0],(char *)&recvbuf,sizeof(recvbuf),0) == SOCKET_ERROR)
		{
			addMsg("接收数据发生错误。");
			return ;
		}
		//将接收到的信息写入TXt文件
		CString strRev;
		strRev.Format("%s",recvbuf);
		addMsg(strRev);
		addMsg("接收数据成功");
		//马上返回预定义字符串
		OnBnClickedBtnSend();

	}
	if (CurSock==ClientSock[1])
	{
		addMsg("2号工站消息到来");
	}
	if (CurSock==ClientSock[2])
	{
		addMsg("3号工站消息到来");
		if(recv(ClientSock[2],(char *)&msg,sizeof(msg),0) == SOCKET_ERROR)
		{
			addMsg("3号工站接收数据发生错误。");
			return ;
		}
		addMsg("3号工站接收消息成功");
		
	}

}
void CommunicationsSetting::addMsg( CString msg )
{
	SYSTEMTIME st; 
	CString strTime;
	GetLocalTime(&st); 
	strTime.Format("%.2d:%.2d:%.2d  ",st.wHour,st.wMinute,st.wSecond);
	
	int count = ((CListBox*)GetDlgItem(IDC_LIST_Message))->GetCount();
	if (count==2000)
	{
		((CListBox*)GetDlgItem(IDC_LIST_Message))->DeleteString(0);
		count--;
	}
	((CListBox*)GetDlgItem(IDC_LIST_Message))->InsertString(count,strTime+msg);
	((CListBox*)GetDlgItem(IDC_LIST_Message))->SetCurSel(count);
}
void CommunicationsSetting::OnBnClickedBtnSend()
{
	// TODO: 在此添加控件通知处理程序代码

	addMsg("准备向客户端发送消息");

	UpdateData(TRUE);
	strcpy_s(msg.msg,m_EDIT_SendStr);
	msg.i = 0;
	if (send(ClientSock[0],m_EDIT_SendStr,m_EDIT_SendStr.GetLength(),0)==SOCKET_ERROR)
	{
		addMsg("发送消息失败");
		return;
	}

addMsg("向客户端发送"+m_EDIT_SendStr+"消息成功");

8.终止套接字库使用 WSACleanup

void CommunicationsSetting::OnDestroy()
{
	//当程序退出时,把SOCKET清空
	WSACleanup();
	CDialogEx::OnDestroy();
	// TODO: 在此处添加消息处理程序代码
}

你可能感兴趣的:(MFC SOCKET通信 WSAAsyncSelect 基于消息的异步套接字)