数据结构实习之Huffman编译码（四）

4,基于异步套接字的消息通信：

注：详细参看孙鑫老师的十六课视频：

(1)因为MFC自带的AfxSocketInit函数初始化支持的是1.1版本的套接字，不适合异步套接字，我们需要调用的是Winsock2版本的套接字，那么加载套接字库的过程只能使用WSAStartup了。在CChatApp的InitInstance初始化函数中添加：

WORD wVersionRequested;
	WSADATA wsaData;
	int err;

	wVersionRequested =MAKEWORD( 2, 2 );

	err = WSAStartup( wVersionRequested, &wsaData );//加载套接字库
	if ( err != 0 ) {
		return FALSE;
	}

	if ( LOBYTE( wsaData.wVersion ) != 2 ||
		HIBYTE( wsaData.wVersion ) != 2 ) {

		WSACleanup( );
		return FALSE;
	}

(2). StdAfx.h里添加#include <winsock2.h>，在setting里添加ws2_32.lib库文件。

给CChatApp类添加析构函数，在其中添加WSACleanup来终止对套接字库的使用。

给CChatDlg类添加成员变量SOCKET m_socket，并在构造函数中初始化为0

给CChatDlg类添加析构函数，添加:

if(m_socket) //判断socket是否有值

closesocket(m_socket);

(3).创建初始化函数InitSocket()，

BOOL CHuffmanDlg::InitSocket()

{
	m_socket=WSASocket(AF_INET,SOCK_DGRAM,0,NULL,0,0);
	if(INVALID_SOCKET==m_socket)
	{
		MessageBox("创建套接字失败！");
		return FALSE;
	}
	SOCKADDR_IN addrSock;
	addrSock.sin_addr.S_un.S_addr=htonl(INADDR_ANY);
	addrSock.sin_family=AF_INET;
	addrSock.sin_port=htons(6000);
	
	if (SOCKET_ERROR==bind(m_socket,(SOCKADDR*)&addrSock,sizeof(SOCKADDR)))
	{
		MessageBox("绑定失败！");
		return FALSE;
	}
	if (WSAAsyncSelect(m_socket,m_hWnd,WM_SOCK,FD_READ)==SOCKET_ERROR)
	{
		MessageBox("注册网络读取失败！");
		return FALSE;
	}

	return TRUE;
}
void CHuffmanDlg::OnSock(WPARAM wParam,LPARAM lParam)
{
	switch(LPARAM(lParam))
	{
	case FD_READ:
		WSABUF wsabuf;
		wsabuf.buf=new char[500];
		wsabuf.len=500;

		DWORD dwRead;
		DWORD dwFlag=0;
		SOCKADDR_IN addrFrom;
		int len=sizeof(SOCKADDR);
		CString codestr;//存储接受到得编码
		CString add;
		if(SOCKET_ERROR==WSARecvFrom(m_socket,&wsabuf,1,&dwRead,&dwFlag,
					(SOCKADDR*)&addrFrom,&len,NULL,NULL))
		{
			MessageBox("接受数据失败！");
			return;
		}
		codestr.Format("%s",wsabuf.buf);
		add.Format("%s",inet_ntoa(addrFrom.sin_addr));
		DecodeSockMes(add,codestr);
		//codelist.Format("%s",wsabuf[1].buf);
		//SetDlgItemText(IDC_EDIT_DECODEREC,codelist);
		break;
	}
}

(4) 消息响应函数的处理：

void CHuffmanDlg::OnButtonSend() 

{
	// TODO: Add your control notification handler code here
	DWORD dwIP;
	CString codeSend="";
	WSABUF wsabuf;

	DWORD dwSend;
	int len;
	((CIPAddressCtrl*)GetDlgItem(IDC_IPADDRESS1))->GetAddress(dwIP);

	SOCKADDR_IN addrTo;
	addrTo.sin_addr.S_un.S_addr=htonl(dwIP);
	addrTo.sin_family=AF_INET;
	addrTo.sin_port=htons(6000);

	GetDlgItemText(IDC_EDIT_CODE,codeSend);
	if (codeSend=="")
	{
		MessageBox("编码为空！");
		return;
	}
	CString s;//叶子节点数
	s.Format("%d",m_num);
	codeSend=s+codeSend;


	for (int i=0;i<m_num;i++)
	{
		codeSend+=m_plist.GetItemText(i,0);
		codeSend+=m_plist.GetItemText(i,3);
	}
	len=codeSend.GetLength();
	wsabuf.buf=codeSend.GetBuffer(len);
	wsabuf.len=len+1;

	OnClear();//清空所有内容
	if(SOCKET_ERROR==WSASendTo(m_socket,&wsabuf,1,&dwSend,0,
			(SOCKADDR*)&addrTo,sizeof(SOCKADDR),NULL,NULL))
	{
		MessageBox("发送数据失败！");
		return;
	}
}

(5)消息接收函数：

void CHuffmanDlg::OnSock(WPARAM wParam,LPARAM lParam)
{
	switch(LPARAM(lParam))
	{
	case FD_READ:
		WSABUF wsabuf;
		wsabuf.buf=new char[500];
		wsabuf.len=500;

		DWORD dwRead;
		DWORD dwFlag=0;
		SOCKADDR_IN addrFrom;
		int len=sizeof(SOCKADDR);
		CString codestr;//存储接受到得编码
		CString add;
		if(SOCKET_ERROR==WSARecvFrom(m_socket,&wsabuf,1,&dwRead,&dwFlag,
					(SOCKADDR*)&addrFrom,&len,NULL,NULL))
		{
			MessageBox("接受数据失败！");
			return;
		}
		codestr.Format("%s",wsabuf.buf);
		add.Format("%s",inet_ntoa(addrFrom.sin_addr));
		DecodeSockMes(add,codestr);
		//codelist.Format("%s",wsabuf[1].buf);
		//SetDlgItemText(IDC_EDIT_DECODEREC,codelist);
		break;
	}
}