MFC：异步套接字编程举例—网络聊天室程序

网络聊天室程序(MFC:利用win32的扩展函数)：
int WSAStartup(

  WORD wVersionRequested,
  LPWSADATA lpWSAData

);
一、加载套接字库:
//本程序要用到winsock2版本的函数，所以还要自己调用WSAStartup函数。将帮助文档中的例子代码复制
//放到CChatApp的InitInstance()函数中，并作相应修改,还需包含头文件#include <winsock2.h>和链接库
//文件ws2_32.lib(Alt+F7)：

WORD wVersionRequested;
WSADATA wsaData;
int err;
wVersionRequested = MAKEWORD( 2, 2 ); //请求Winsock2.2版本，也是最高版本
err = WSAStartup( wVersionRequested, &wsaData );
if ( err != 0 ) {

    return;

}

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

//    WSACleanup( ); 放到析构函数中

    return false; //false

}

二、在App中增加析构函数终止对套接字库的使用：
CChatApp::~CChatApp()
{

WSACleanup();//终止对套接字的使用

}
三、CChatDlg添加成员变量SOCKET m_socket,并初始化为0，加析构：
CChatDlg::~CChatDlg(){

if(m_socket)
closesocket(m_socket);

}
四、添加初始化套接字库的函数(1、创建套接字 2、绑定套接字 3、注册网络读取事件),并在OnInitDialog函数中调用：
BOOL CChatDlg::InitSocket()// 在OnInitDialog 函数中调用
{

m_socket=WSASocket(AF_INET,SOCK_DGRAM,0,NULL,0,0);//win32的扩展函数(WSA+)创建套接字
if(INVALID_SOCKET==m_socket)
{

MessageBox("创建套接字失败！");

return FALSE;

}

//绑定套接字
SOCKADDR_IN addrSock;
addrSock.sin_addr.S_un.S_addr=htonl(INADDR_ANY);//4字节 将主机字节顺序转换成网络字节顺序
//inet_addr("136.149.3.49");将点式ip地址转换成四字节的ulong类型
addrSock.sin_family=AF_INET;
addrSock.sin_port=htons(6000);//2字节 将主机字节顺序转换成网络字节顺序
if(SOCKET_ERROR==bind(m_socket,(SOCKADDR*)&addrSock,sizeof(SOCKADDR))) //绑定套接字
{

MessageBox("绑定失败！");

return FALSE;

}
//接收windows的基于消息的网络事件通知
//消息到来时系统通过WM_SOCK的消息响应函数来通知我们 
  if(SOCKET_ERROR==WSAAsyncSelect(m_socket,m_hWnd,UM_SOCK,FD_READ)) //我的另一篇用的是多线程编写
{

MessageBox("注册网络读取事件失败！");

return FALSE;

}
return TRUE;

}
五、自定义消息：
1、#define UM_SOCK WM_USER+1
2、消息响应函数原型的声明：
afx_msg void OnSock(WPARAM,LPARAM);//消息通过参数传递，所以要带参数
3、消息映射：
ON_MESSAGE(UM_SOCK,OnSock)
4、消息响应函数的实现：
void CChatDlg::OnSock(WPARAM wParam,LPARAM lParam) //在消息响应函数中接收消息

{
//首先判断是否网络读取事件发生了，然后调用接收函数接收数据
//wparam标识了socket，low(lparam)标识发生的网络事件，high（lparam）标识了任何错误代码
switch(LOWORD(lParam)){
case FD_READ: //FD_READ表明是网络读取事件发生了

WSABUF wsabuf;

wsabuf.buf=new char[200];

wsabuf.len=200;

DWORD dwRead;

DWORD dwFlag=0;

SOCKADDR_IN addrFrom;

int len=sizeof(SOCKADDR); 

CString str;

CString strTemp;

HOSTENT *pHost;

if(SOCKET_ERROR==WSARecvFrom(m_socket,&wsabuf,1,&dwRead,&dwFlag,(SOCKADDR*)&addrFrom,&len,NULL,NULL)) 

// 接收数据(非重叠套接字：后两个为NULL)

{

MessageBox("接收数据失败！");

return;

}

pHost=gethostbyaddr((char*)&addrFrom.sin_addr.S_un.S_addr,4,AF_INET);

//str.Format("%s说 :%s",inet_ntoa(addrFrom.sin_addr),wsabuf.buf);

str.Format("%s说 :%s",pHost->h_name,wsabuf.buf);

str+="\r\n";

GetDlgItemText(IDC_EDIT_RECV,strTemp);

str+=strTemp;

SetDlgItemText(IDC_EDIT_RECV,str); 

break;

}

}
六、发送端：发送按钮发送消息

void CChatDlg::OnBtnSend() 
{

DWORD dwIP;
CString strSend;
WSABUF wsabuf;
DWORD dwSend;
int len;
CString strHostName;
SOCKADDR_IN addrTo;
HOSTENT* pHost;
if(GetDlgItemText(IDC_EDIT_HOSTNAME,strHostName),strHostName=="")//如果没有主机名，就用IP地址
{

((CIPAddressCtrl*)GetDlgItem(IDC_IPADDRESS1))->GetAddress(dwIP);

addrTo.sin_addr.S_un.S_addr=htonl(dwIP);

}
else
{

pHost=gethostbyname(strHostName);//获取主机名

addrTo.sin_addr.S_un.S_addr=*((DWORD*)pHost->h_addr_list[0]);

}
addrTo.sin_family=AF_INET;
addrTo.sin_port=htons(6000); 
GetDlgItemText(IDC_EDIT_SEND,strSend);
len=strSend.GetLength();
wsabuf.buf=strSend.GetBuffer(len);
wsabuf.len=len+1;
SetDlgItemText(IDC_EDIT_SEND,"");
if(SOCKET_ERROR==WSASendTo(m_socket,&wsabuf,1,&dwSend,0,
(SOCKADDR*)&addrTo,sizeof(SOCKADDR),NULL,NULL)) //发送消息
{

MessageBox("发送数据失败！");

return;

}

} //the end;
注：struct hostent {

  char FAR* h_name;
  char FAR* FAR* h_aliases;
  short h_addrtype;
  short h_length;
  char FAR* FAR* h_addr_list;//存放IP地址，一台主机可能有多个IP地址

};