异步选择WSAAsyncSelect

http://blog.csdn.net/cloud95/article/details/7688786 


WSAAsyncSelect模型允许应用程序以Windows消息的方式接收网络事件通知。许多对性能要求不高的网络应用程序都采用WSAAsyncSelect模型,MFC的CSocket类也使用了它。

        WSAAsyncSelect自动把套接字设为非阻塞模式,并且为套接字绑定一个窗口句柄,当有网络事件发生时,便向这个窗口发送消息。

        intWSAAsyncSelect(

        SOCKET s,                  //需要设置的套接字句柄

        HWND  hWnd,     //指定一个窗口句柄, 套接字的通知消息将被发到此窗口中

       u_int    wMsg,     //网络事件到来的ID,可以在WM_USER以上数值中任意指定一个值

       long    IEvent     //指定哪些通知码需要发送

                                 //FD_READ可以读套接字

                                 //FD_WRITE 可以写套接字

                                //FD_ACCEPT 监听套接字有连接接入

                               //FD_CONNET  如果套接字连接对方主机,连接完成后会收到这个通知码

                              //FD_CLOSE 检测到套接字对应的连接被关闭

)

       回传过来的消息类型与注册的wMsg相同,wParam等于套接字句柄,lParam通过WSAGETSELECTEVENT转义后就是FD_READ,FD_WRITE,FD_ACCEPT,FD_CONNET,FD_CLOSE

        代码如下:

        新建一个基于MFC对话框的工程,工程名为test,然后在testDlg.h中定义      

[cpp]  view plain copy
  1. #include "winsock2.h"  
  2. #pragma comment(lib, "WS2_32")  
  3. #define WM_SOCKET (WM_USER+1)   
同时在该文件中加入一个自定义消息处理函数

[cpp]  view plain copy
  1. afx_msg LRESULT OnSocketMsg(WPARAM wParam,LPARAM lParam);  
在testDlg.cpp中定义

[cpp]  view plain copy
  1. ON_MESSAGE(WM_SOCKET,OnSocketMsg)  

在OnInitDialog中加入自己的初始化代码

[cpp]  view plain copy
  1. // TODO: 在此添加额外的初始化代码  
  2.   
  3.     WSADATA wsaData;  
  4.     WORD sockVersion = MAKEWORD(2, 2);  
  5.     if(WSAStartup(sockVersion, &wsaData) != 0)  
  6.     {  
  7.         return 0;  
  8.     }  
  9.     USHORT nPort=4567;  
  10.     SOCKET s=socket(AF_INET,SOCK_STREAM,0);  
  11.     sockaddr_in sin;  
  12.     sin.sin_family=AF_INET;  
  13.     sin.sin_port=ntohs(nPort);  
  14.     sin.sin_addr.S_un.S_addr=INADDR_ANY;  
  15.     if(bind(s,(sockaddr*)&sin,sizeof(sin))==SOCKET_ERROR)  
  16.     {  
  17.         DWORD error=GetLastError();  
  18.         return TRUE;  
  19.     }  
  20.   
  21.     WSAAsyncSelect(s,m_hWnd,WM_SOCKET,FD_ACCEPT|FD_CLOSE);  
  22.     listen(s,5);  

添加自定义消息函数的处理:

[cpp]  view plain copy
  1. LRESULT CtestDlg::OnSocketMsg(WPARAM wParam,LPARAM lParam)  
  2. {  
  3.     SOCKET s=wParam;  
  4.     CString strContent;  
  5.     if(WSAGETSELECTERROR(lParam))  
  6.     {  
  7.         closesocket(s);  
  8.         return false;  
  9.     }  
  10.     switch(WSAGETSELECTEVENT(lParam))  
  11.     {  
  12.     case FD_ACCEPT://检测到有套接字连上来  
  13.         {  
  14.             SOCKET client=accept(s,NULL,NULL);  
  15.             WSAAsyncSelect(client,m_hWnd,WM_SOCKET,FD_READ|FD_WRITE|FD_CLOSE);  
  16.             CString strContent;  
  17.             GetDlgItemText(IDC_EDIT1,strContent);  
  18.             if(!strContent.IsEmpty())  
  19.             {  
  20.                 strContent+="\r\n";  
  21.             }             
  22.             strContent+="有客户端连上来了";  
  23.             SetDlgItemText(IDC_EDIT1,strContent);//将内容显示到Edit控件  
  24.         }  
  25.         break;  
  26.     case FD_WRITE:  
  27.         {  
  28.         }  
  29.         break;  
  30.     case FD_READ:  
  31.         {  
  32.             char szText[1024]={0};  
  33.             if(recv(s,szText,1024,0)==-1)  
  34.                 closesocket(s);  
  35.             else  
  36.             {                 
  37.                 GetDlgItemText(IDC_EDIT1,strContent);  
  38.                 if(!strContent.IsEmpty())  
  39.                 {  
  40.                     strContent+="\r\n";  
  41.                 }  
  42.                 CString strTemp;  
  43.                 strTemp.Format("%s",szText);  
  44.                 strContent+=strTemp;  
  45.                 SetDlgItemText(IDC_EDIT1,strContent);//将内容显示到Edit控件中  
  46.             }  
  47.         }  
  48.         break;  
  49.     case FD_CLOSE:  
  50.         {  
  51.             closesocket(s);  
  52.             GetDlgItemText(IDC_EDIT1,strContent);  
  53.             if(!strContent.IsEmpty())  
  54.             {  
  55.                 strContent+="\r\n";  
  56.             }             
  57.             strContent+="有客户端断开了";  
  58.             SetDlgItemText(IDC_EDIT1,strContent);//将内容显示到Edit控件中  
  59.         }  
  60.         break;  
  61.     }  
  62.     return true;  
  63. }  

对WM_DESTROY消息处理,清理套接字库

[cpp]  view plain copy
  1. void CtestDlg::OnDestroy()  
  2. {  
  3.     CDialog::OnDestroy();  
  4.     WSACleanup();  
  5.     // TODO: 在此处添加消息处理程序代码  
  6. }  

你可能感兴趣的:(异步选择WSAAsyncSelect)