Windows网络编程：WSAAsynSelect的基本使用

Windows网络编程：WSAAsynSelect的基本使用

WSAAsynSelect的基本使用，下面是一个简单的例子

#include  < windows.h >
#include  < stdio.h >

#pragma comment(lib,  " ws2_32.lib " )

#define  LISTEN_PORT  5080
#define  UWM_SOCKET WM_USER+1

TCHAR szClassName[]  =  TEXT( " WSAAsynSelectClass " );
TCHAR szWndName[]  =  TEXT( " WSAAsynSelectWnd " );

LRESULT CALLBACK MainWndProc(HWND hwnd, UINT uMsg, WPARAM wParam,LPARAM lParam);

int  WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine,  int  nShowCmd)
{
    WNDCLASS wndClass  =  { 0 };
    HWND hMainWnd;

    MSG msg;
    WSADATA wsaData;
     int  ret;
    SOCKET listen_sock;
    TCHAR szErrorMsg[ 256 ];
    SOCKADDR_IN listen_addr;

    ret  =  WSAStartup(MAKEWORD( 2 , 2 ),  & wsaData);
     if ( 0   !=  ret)
    {
        wsprintf(szErrorMsg, TEXT( " WSAStartup failed. Error:%d.\n " ), WSAGetLastError());
        MessageBox(NULL, szErrorMsg, szWndName, MB_OK);
         return  ret;
    }

    listen_sock  =  socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
     if (INVALID_SOCKET  ==  listen_sock)
    {
        wsprintf(szErrorMsg, TEXT( " socket failed, Error:%d.\n " ), WSAGetLastError());
        MessageBox(NULL, szErrorMsg, szWndName, MB_OK);
         goto  rettag1;
    }

    listen_addr.sin_family  =  AF_INET;
    listen_addr.sin_addr.s_addr  =  htonl(INADDR_ANY);
    listen_addr.sin_port  =  htons(LISTEN_PORT);
    ret  =  bind(listen_sock, (SOCKADDR * ) & listen_addr,  sizeof (listen_addr));
     if (SOCKET_ERROR  ==  ret)
    {
        wsprintf(szErrorMsg, TEXT( " bind failed. Error:%d.\n " ), WSAGetLastError());
        MessageBox(NULL, szErrorMsg, szWndName, MB_OK);
         goto  rettag2;
    }

    wndClass.style  =  CS_HREDRAW  |  CS_VREDRAW;
    wndClass.cbClsExtra  =   0 ;
    wndClass.cbWndExtra  =   0 ;
    wndClass.hbrBackground  =  GetStockObject(WHITE_BRUSH);
    wndClass.hCursor  =  LoadCursor(NULL, IDC_ARROW);
    wndClass.hIcon  =  LoadIcon(NULL, IDI_APPLICATION);
    wndClass.hInstance  =  hInstance;
    wndClass.lpfnWndProc  =  MainWndProc;
    wndClass.lpszClassName  =  szClassName;
    wndClass.lpszMenuName  =  NULL;

     if ( ! RegisterClass( & wndClass))
    {
        wsprintf(szErrorMsg, TEXT( " RegisterClass failed. " ));
        MessageBox(NULL, szErrorMsg, szWndName, MB_OK);
         goto  rettag2;
    }

    hMainWnd  =  CreateWindow(szClassName, szWndName, WS_OVERLAPPEDWINDOW  |  WS_VSCROLL, 
                            CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
                            NULL, NULL, hInstance, NULL);
     if (NULL  ==  hMainWnd)
    {
        wsprintf(szErrorMsg, TEXT( " CreateWindow failed. " ));
        MessageBox(NULL, szErrorMsg, szWndName, MB_OK);
        UnregisterClass(szClassName, hInstance);
         goto  rettag2;
    }

    ShowWindow(hMainWnd, nShowCmd);
    UpdateWindow(hMainWnd);

    ret  =  WSAAsyncSelect(listen_sock, hMainWnd, UWM_SOCKET, FD_ACCEPT | FD_CLOSE);
     if (SOCKET_ERROR  ==  ret)
    {
        wsprintf(szErrorMsg, TEXT( " WSAAsyncSelect failed. Error:%d.\n " ), WSAGetLastError());
        MessageBox(NULL, szErrorMsg, szWndName, MB_OK);        
         goto  rettag2;
    }

    ret  =  listen(listen_sock,  5 );
     if (SOCKET_ERROR  ==  ret)
    {
        wsprintf(szErrorMsg, TEXT( " listen failed. Error:%d.\n " ), WSAGetLastError());
        MessageBox(NULL, szErrorMsg, szWndName, MB_OK);        
         goto  rettag2;
    }

     while (GetMessage( & msg, NULL,  0 ,  0 ))
    {
        TranslateMessage( & msg);
        DispatchMessage( & msg);
    }


rettag2:
    closesocket(listen_sock);
rettag1:
    WSACleanup();

     return   0 ;
}

LRESULT CALLBACK MainWndProc(HWND hwnd, UINT uMsg, WPARAM wParam,LPARAM lParam)
{
    TCHAR szErrorMsg[ 256 ];
     char  szMsg[ 256 ];
    SOCKET client_sock  =  INVALID_SOCKET;
    SOCKADDR_IN client_addr;
     int  client_addr_len  =   sizeof (client_addr);
     int  ret;
    HDC hdc;
     static   int  row  =   0 ;
    TEXTMETRIC textMetric;
     int  textHeight  =   5 ;

    hdc  =  GetDC(hwnd);
    GetTextMetrics(hdc,  & textMetric);
    textHeight  =  textMetric.tmHeight;

     switch (uMsg)
    {
    
     case  WM_DESTROY:
        ReleaseDC(hwnd, hdc);
        PostQuitMessage( 0 );
         break ;
     case  UWM_SOCKET:
         if (WSAGETSELECTERROR(lParam))
        {
            wsprintf(szErrorMsg, TEXT( " WSAGETSELECTERROR failed. Error:%d.\n " ), WSAGETSELECTERROR(lParam));
            MessageBox(NULL, szErrorMsg, szWndName, MB_OK);        
             break ;
        }

         switch (WSAGETSELECTEVENT(lParam))
        {
         case  FD_ACCEPT:
            client_sock  =  accept(wParam, (SOCKADDR * ) & client_addr,  & client_addr_len);
             if (INVALID_SOCKET  ==  client_sock)
            {
                wsprintf(szErrorMsg, TEXT( " accept failed. Error:%d.\n " ), WSAGetLastError());
                MessageBox(NULL, szErrorMsg, szWndName, MB_OK);        
                 break ;
            }

            sprintf_s(szMsg,  256 ,  " accept client:%s:%d. " , inet_ntoa(client_addr.sin_addr), client_addr.sin_port);
     //         MessageBoxA(NULL, szMsg, "Information", MB_OK);
            TextOutA(hdc,  0 , row, szMsg, strlen(szMsg));
            row  +=  textHeight;

            ret  =  WSAAsyncSelect(client_sock, hwnd, UWM_SOCKET, FD_READ | FD_WRITE | FD_CLOSE);
             if (SOCKET_ERROR  ==  ret)
            {
                wsprintf(szErrorMsg, TEXT( " WSAAsyncSelect failed. Error:%d.\n " ), WSAGetLastError());
                MessageBox(NULL, szErrorMsg, szWndName, MB_OK);    
            }
             break ;
         case  FD_READ:
            ret  =  recv(wParam, szMsg,  256 ,  0 );
             if (SOCKET_ERROR  ==  ret)
            {
                wsprintf(szErrorMsg, TEXT( " recv failed. Error:%d.\n " ), WSAGetLastError());
                MessageBox(NULL, szErrorMsg, szWndName, MB_OK);        
                 break ;
            }
             else   if ( 0   ==  ret)
            {
                sprintf_s(szMsg,  256 ,  " client closed.\n " );
                row  +=  textHeight;
                TextOutA(hdc,  0 , row, szMsg, strlen(szMsg));
                row  +=  textHeight;
                 break ;
            }
             else
            {
                TextOutA(hdc,  0 , row, szMsg, strlen(szMsg));
                row  +=  textHeight;
            }
             break ;
         case  FD_WRITE:
             break ;
         case  FD_CLOSE:
            closesocket(wParam);
             break ;
        }
         break ;

     default :
         return  DefWindowProc(hwnd, uMsg, wParam, lParam);
    }
     return  TRUE;
}