WIN网络编程-WSAAsyncSelect

///////////////////////////////////////////////
// WSAAsyncSelect.cpp文件

//#define WSAGETSELECTERROR(lParam) HIWORD(lParam)//高字为出错代码

//#define WSAGETSELECTEVENT(lParam) LOWORD(lParam)//低字为通知码

//允许应用程序以WINDOWS消息的形式接收网络事件通知

//int WSAAsyncSelect(

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

//HWND hWnd,//指定一个窗口句柄

//u_int wMsg,//网络事件到来时接收到的消息ID,可以在WM_USER以上的

//            数值中任意选择一个用做ID

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

//)

//lEvent具体组合

//FD_READ接收到对方发过来的数据包

//FD_WRITE数据缓冲区满后再次变空时，WinSock接口通过该通知码通知应用程序。

//FD_ACCEPT监听中套接字检测到有连接进入

//FD_CONNECT如果用套接字去连接对方的主机，当连接动作完成后会接收到

//         这个通知码

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

 

#include "../common/initsock.h"
#include <stdio.h>

#define WM_SOCKET WM_USER +101  // 自定义消息
CInitSock theSock;

LRESULT CALLBACK WindowProc(HWND hWnd, UINT uMsg, WPARAM wParam,LPARAM lParam);
int main()
{
 char szClassName[] ="MainWClass"; 
 WNDCLASSEX wndclass;
 // 用描述主窗口的参数填充WNDCLASSEX结构
 wndclass.cbSize =sizeof(wndclass); 
 wndclass.style =CS_HREDRAW|CS_VREDRAW; 
 wndclass.lpfnWndProc =WindowProc; 
 wndclass.cbClsExtra =0;  
 wndclass.cbWndExtra =0;  
 wndclass.hInstance =NULL;  
 wndclass.hIcon = ::LoadIcon(NULL,IDI_APPLICATION); 
 wndclass.hCursor = ::LoadCursor(NULL,IDC_ARROW);  
 wndclass.hbrBackground =(HBRUSH)::GetStockObject(WHITE_BRUSH); 
 wndclass.lpszMenuName =NULL;  
 wndclass.lpszClassName = szClassName ;
 wndclass.hIconSm = NULL; 
 ::RegisterClassEx(&wndclass);
 // 创建主窗口
 HWND hWnd = ::CreateWindowEx(
  0,      
  szClassName,   
  "", 
  WS_OVERLAPPEDWINDOW, 
  CW_USEDEFAULT, 
  CW_USEDEFAULT,  
  CW_USEDEFAULT, 
  CW_USEDEFAULT,   
  NULL,    
  NULL,  
  NULL, 
  NULL);  
 if(hWnd == NULL)
 {
  ::MessageBox(NULL, "创建窗口出错！","error", MB_OK);
  return -1;
 }

 USHORT nPort = 4567; //此服务器监听的端口号

 // 创建监听套节字
 SOCKET sListen = ::socket(AF_INET, SOCK_STREAM,IPPROTO_TCP); 
 sockaddr_in sin;
 sin.sin_family = AF_INET;
 sin.sin_port = htons(nPort);
 sin.sin_addr.S_un.S_addr = INADDR_ANY;
 // 绑定套节字到本地机器
 if(::bind(sListen,(sockaddr*)&sin, sizeof(sin)) ==SOCKET_ERROR)
 {
  printf(" Failed bind()/n");
  return -1;
 }

 // 将套接字设为窗口通知消息类型。
 ::WSAAsyncSelect(sListen, hWnd, WM_SOCKET,FD_ACCEPT|FD_CLOSE);

 // 进入监听模式
 ::listen(sListen, 5);

 // 从消息队列中取出消息
 MSG msg; 
 while(::GetMessage(&msg, NULL, 0,0))
 {
  // 转化键盘消息
  ::TranslateMessage(&msg);
  // 将消息发送到相应的窗口函数
  ::DispatchMessage(&msg);
 }
 // 当GetMessage返回0时程序结束
 return msg.wParam;
}

LRESULT CALLBACK WindowProc(HWND hWnd, UINT uMsg, WPARAM wParam,LPARAM lParam)
{
 switch (uMsg)
 { 
 case WM_SOCKET:
  {
   //取得有事件发生的套节字句柄
   SOCKET s =wParam;
   //查看是否出错
   if(WSAGETSELECTERROR(lParam))
   {
    ::closesocket(s);
    return0;
   }
   //处理发生的事件
   switch(WSAGETSELECTEVENT(lParam))
   {
   caseFD_ACCEPT:  //监听中的套接字检测到有连接进入
    {
     SOCKETclient = ::accept(s, NULL, NULL);
     ::WSAAsyncSelect(client,hWnd, WM_SOCKET, FD_READ|FD_WRITE|FD_CLOSE);
    }
    break;
   caseFD_WRITE:
    {
    }
    break;
   caseFD_READ:
    {
     charszText[1024] = { 0 };
     if(::recv(s,szText, 1024, 0) == -1)
      ::closesocket(s);
     else
      printf("接收数据：%s",szText);
    }
    break;
   caseFD_CLOSE:
    {
     ::closesocket(s);
    }
    break;
   }
  }
  return 0;
 case WM_DESTROY:
  ::PostQuitMessage(0) ;
  return 0 ;
 }

 // 将我们不处理的消息交给系统做默认处理
 return ::DefWindowProc(hWnd, uMsg, wParam,lParam);
}