天天看點

TCP流式套接字的異步事件WSAAsyncSelect程式設計

WSAAsyncSelect( )是Winsock提供的一個适合于Windows程式設計使用的函數,它允許在一個套接口上當發生特定的網絡事件時,給Windows網絡應用程式(視窗或對話框)發送一個消息(事件通知)。

WSAAsyncSelect函數原型如下:

int WSAAsyncSelect(

    SOCKET s,  //辨別一個需要事件通知的套接口描述符

    HWND hWnd,  //辨別一個在網絡事件發生時要想收到消息的視窗或對話框的句柄

    u_int wMsg,  //在網絡事件發生時要接收的消息,該消息會投遞到由hWnd句柄指定的視窗或對話框

    long lEvent  //位屏蔽碼,用于指明應用程式感興趣的網絡事件集合

);

     若應用程式對一個套接口s調用了WSAAsyncSelect( )函數,那麼套接口s的模式會自動從阻塞模式變成非阻塞模式。如果應用程式同時對多個網絡事件感興趣,那麼隻需對各種類型的網絡事件執行按位或(OR)的運算即可。進行一次WSAAsyncSelect( )調用,将使為同一個套接口啟動的所有以前的WSAAsyncSelect( )調用廢棄。如果要取消所有的通知,也就是指出Windows Sockets的實作不再在套接口上發送任何和網絡事件相關的消息,則把lEvent字段置為0,然後調用WSAAsyncSelect( )。當某一套接口s上發生了一個已命名的網絡事件時,應用程式視窗hWnd會接收到消息wMsg。應用程式視窗例程的wParam參數辨別了網絡事件發生的套接口。lParam參數的低位字指明了發生的網絡事件,高位字則含有一個錯誤代碼,該錯誤代碼可以是Winsock2.h中定義的任何錯誤。

以下為測試WSAAsyncSelect()函數的程式,一個伺服器端兩個用戶端

下面是伺服器端程式:

001

002

003

004

005

006

007

008

009

010

011

012

013

014

015

016

017

018

019

020

021

022

023

024

025

026

027

028

029

030

031

032

033

034

035

036

037

038

039

040

041

042

043

044

045

046

047

048

049

050

051

052

053

054

055

056

057

058

059

060

061

062

063

064

065

066

067

068

069

070

071

072

073

074

075

076

077

078

079

080

081

082

083

084

085

086

087

088

089

090

091

092

093

094

095

096

097

098

099

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

#define WM_SOCKET WM_USER+101 

#include <WINSOCK2.H> 

#pragma comment(lib,"WS2_32") 

//----------------視窗過程函數的聲明------------- 

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

//----------------WinMain()函數------------------ 

int WINAPI WinMain( HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nShowCmd ) 

    WNDCLASS wc; 

    wc.style=CS_HREDRAW|CS_VREDRAW; 

    wc.lpfnWndProc=WindowProc; 

    wc.cbClsExtra=0; 

    wc.cbWndExtra=0; 

    wc.hInstance=hInstance; 

    wc.hIcon=LoadIcon(NULL,IDI_EXCLAMATION); 

    wc.hCursor=LoadCursor(NULL,IDC_ARROW); 

    wc.hbrBackground=(HBRUSH)GetStockObject(BLACK_BRUSH); 

    wc.lpszMenuName=NULL; 

    wc.lpszClassName="Test"; 

    //---注冊視窗類---- 

    RegisterClass(&wc); 

    //---建立視窗---- 

    HWND hwnd=CreateWindow("Test","視窗标題",WS_SYSMENU,300,0,600,400,NULL,NULL,hInstance,NULL); 

    if (hwnd==NULL) 

    { 

        MessageBox(hwnd,"建立視窗出錯","标題欄提升",MB_OK); 

        return 1; 

    } 

    //---顯示視窗---- 

    ShowWindow(hwnd,SW_SHOWNORMAL); 

    UpdateWindow(hwnd); 

    //---socket----- 

    WSADATA wsaData; 

    WORD wVersionRequested=MAKEWORD(2,2); 

    if (WSAStartup(wVersionRequested,&wsaData)!=0) 

    { 

        MessageBox(NULL,"WSAStartup() Failed","調用失敗",0); 

        return 1; 

    } 

    SOCKET s=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); 

    if (s==INVALID_SOCKET) 

    { 

        MessageBox(NULL,"socket() Failed","調用失敗",0); 

        return 1; 

    }     

    sockaddr_in sin; 

    sin.sin_family=AF_INET; 

    sin.sin_port=htons(1111); 

    sin.sin_addr.S_un.S_addr=inet_addr("192.168.31.1"); 

    if (bind(s,(sockaddr*)&sin,sizeof(sin))==SOCKET_ERROR) 

    { 

        MessageBox(NULL,"bind() Failed","調用失敗",0); 

        return 1; 

    } 

    if (listen(s,3)==SOCKET_ERROR) 

    { 

        MessageBox(NULL,"listen() Failed","調用失敗",0); 

        return 1; 

    } 

    else 

        MessageBox(hwnd,"進入監聽狀态!","标題欄提示",MB_OK); 

    WSAAsyncSelect(s,hwnd,WM_SOCKET,FD_ACCEPT|FD_CLOSE); 

    //---消息循環---- 

    MSG msg; 

    while (GetMessage(&msg,0,0,0)) 

    { 

        TranslateMessage(&msg); 

        DispatchMessage(&msg); 

    } 

    closesocket(s); 

    WSACleanup(); 

    return msg.wParam; 

//-------------------視窗過程---------------------- 

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

    switch(uMsg) 

    { 

    case WM_SOCKET: 

        { 

            SOCKET ss=wParam;   //wParam參數标志了網絡事件發生的套接口 

            if (WSAGETSELECTERROR(lParam)) 

            { 

                closesocket(ss); 

                return 0; 

            } 

            switch (WSAGETSELECTEVENT(lParam)) 

            { 

            case FD_ACCEPT:   //-----①連接配接請求到來 

                { 

                     sockaddr_in Cadd; 

                     int Cadd_len=sizeof(Cadd); 

                    SOCKET sNew=accept(ss,(sockaddr*)&Cadd,&Cadd_len); 

                    if (ss==INVALID_SOCKET)                     

                        MessageBox(hwnd,"調用accept()失敗!","标題欄提示",MB_OK);                     

                    WSAAsyncSelect(sNew,hwnd,WM_SOCKET,FD_READ|FD_CLOSE); 

                }break; 

            case FD_READ:   //-----②資料發送來 

                { 

                    char cbuf[256]; 

                    memset(cbuf,0,256); 

                    int cRecv=recv(ss,cbuf,256,0); 

                    if ((cRecv==SOCKET_ERROR&& WSAGetLastError() == WSAECONNRESET)|| cRecv==0) 

                    { 

                        MessageBox(hwnd,"調用recv()失敗!","标題欄提示",MB_OK); 

                        closesocket(ss); 

                    } 

                    else if (cRecv>0) 

                    { 

                        MessageBox(hwnd,cbuf,"收到的資訊",MB_OK);     

                        char Sbuf[]="Hello client!I am server"; 

                        int isend=send(ss,Sbuf,sizeof(Sbuf),0); 

                        if (isend==SOCKET_ERROR || isend<=0) 

                        { 

                            MessageBox(hwnd,"發送消息失敗!","标題欄提示",MB_OK);                             

                        } 

                        else 

                            MessageBox(hwnd,"已經發資訊到用戶端!","标題欄提示",MB_OK); 

                    } 

                }break; 

            case FD_CLOSE:    //----③關閉連接配接 

                { 

                    closesocket(ss); 

                } 

                break; 

            } 

        } 

        break; 

    case WM_CLOSE: 

        if (IDYES==MessageBox(hwnd,"是否确定退出?","message",MB_YESNO)) 

            DestroyWindow(hwnd); 

        break; 

    case WM_DESTROY: 

        PostQuitMessage(0); 

        break; 

    default: 

        return DefWindowProc(hwnd,uMsg,wParam,lParam); 

    } 

    return 0; 

用戶端與上兩次的都一樣,這裡不再給出。詳見這裡 【TCP流式套接字的阻塞模式程式設計】http://blog.csdn.net/akof1314/archive/2010/05/13/5588265.aspx1

先啟動伺服器端,再啟動1号用戶端,接着啟動2号用戶端,就會看到如圖所示:

TCP流式套接字的異步事件WSAAsyncSelect程式設計

繼續閱讀