WSAEventSelect模型客户端代码示例

配合上一篇WSAEventSelect模型的Server使用，测试效果用。做了些修改，修改为多线程实现IO业务处理。

// g_wsaEventSelect_Client.cpp : Defines the entry point for the console application. // #include "stdafx.h" #include <Winsock2.h> #include <malloc.h> #pragma comment(lib,"Ws2_32.lib") SOCKET g_socketServer = INVALID_SOCKET; WSAEVENT g_wsaEvent = WSA_INVALID_EVENT; // 初始化socket void InitSock() { WORD wVersionRequested; WSADATA wsaData; int err; wVersionRequested = MAKEWORD( 2, 2 ); err = WSAStartup( wVersionRequested, &wsaData ); if ( err != 0 ) { /* Tell the user that we could not find a usable */ /* WinSock DLL. */ return; } /* Confirm that the WinSock DLL supports 2.2. */ /* Note that if the DLL supports versions greater */ /* than 2.2 in addition to 2.2, it will still return */ /* 2.2 in wVersion since that is the version we */ /* requested. */ if ( LOBYTE( wsaData.wVersion ) != 2 || HIBYTE( wsaData.wVersion ) != 2 ) { /* Tell the user that we could not find a usable */ /* WinSock DLL. */ WSACleanup( ); return; } } void UnInitSock() { if (g_wsaEvent != WSA_INVALID_EVENT) { WSACloseEvent(g_wsaEvent); } if (g_socketServer != INVALID_SOCKET) { closesocket(g_socketServer); } WSACleanup(); } struct ST_THREAD_PARAM { SOCKET socket; WSAEVENT wsaEvent; }; DWORD WINAPI ServiceThread(LPVOID lpThreadParameter) { ST_THREAD_PARAM* pThread = (ST_THREAD_PARAM*)lpThreadParameter; SOCKET socketServer = pThread->socket; WSAEVENT wsaEvent = pThread->wsaEvent; printf("新线程%d起动/n",GetCurrentThreadId()); while(true) { int nRet=::WSAWaitForMultipleEvents(1,&wsaEvent,FALSE,10000,FALSE); if(nRet==WAIT_FAILED) // 失败 { printf("failed WSAWaitForMultipleEvents/n"); break; } else if(nRet==WSA_WAIT_TIMEOUT) // 超时 { printf(" WSA_WAIT_TIMEOUT ... /n"); continue; } else // 成功 -- 网络事件发生 { WSANETWORKEVENTS wsaNetEvent; ::WSAEnumNetworkEvents(socketServer,wsaEvent,&wsaNetEvent); if(wsaNetEvent.lNetworkEvents&FD_READ) { printf("FD_READ event occurs.../n"); char buffer[1024]; int nRet = recv(socketServer,buffer,1024,0); if (nRet>0) { buffer[nRet] = '/0'; printf("收到数据 %s/n",buffer); Sleep(3000); printf("发送数据..../n"); int nSend = send(socketServer,buffer,nRet,0); } else { UnInitSock(); printf("线程%d退出/n",GetCurrentThreadId()); return -1; } } else if(wsaNetEvent.lNetworkEvents&FD_WRITE) { printf("FD_WRITE event occurs.../n"); printf("发送数据..../n"); char szBuffer[] = "Hello world..."; int nLen = sizeof(szBuffer); nRet = send(socketServer,szBuffer,nLen,0); if (nRet == SOCKET_ERROR) { printf("send() error .../n"); UnInitSock(); printf("线程%d退出/n",GetCurrentThreadId()); return -1; } printf("nRet of send() is %d/n",nRet); } if(wsaNetEvent.lNetworkEvents&FD_CLOSE) { printf("FD_CLOSE event occurs.../n"); int nErrorCode = WSAGetLastError(); printf("Error code is %d/n",nErrorCode); if (nErrorCode == WSAECONNRESET) { printf("WSAECONNRESET error./n"); } else if (nErrorCode == WSAENETDOWN) { printf("WSAENETDOWN error./n"); } else if (nErrorCode == WSAENETRESET) { printf("WSAENETRESET error./n"); } UnInitSock(); printf("线程%d退出/n",GetCurrentThreadId()); return -1; } } } printf("线程%d退出/n",GetCurrentThreadId()); return 0; } int _tmain(int argc, _TCHAR* argv[]) { InitSock(); g_socketServer=socket(AF_INET,SOCK_STREAM,0); if (g_socketServer == INVALID_SOCKET) { UnInitSock(); return -1; } sockaddr_in sin; sin.sin_addr.S_un.S_addr=inet_addr("127.0.0.1"); sin.sin_family=AF_INET; sin.sin_port=htons(3456); if (connect(g_socketServer,(sockaddr*)&sin,sizeof(sockaddr)) == SOCKET_ERROR) { UnInitSock(); return -1; } g_wsaEvent=::WSACreateEvent(); ::WSAEventSelect(g_socketServer,g_wsaEvent,FD_READ|FD_WRITE|FD_CLOSE); ST_THREAD_PARAM* pThreadParam = new ST_THREAD_PARAM(); pThreadParam->socket = g_socketServer; pThreadParam->wsaEvent = g_wsaEvent; HANDLE hThread = ::CreateThread(NULL,0,ServiceThread,(LPVOID)pThreadParam,0,NULL); if (hThread == INVALID_HANDLE_VALUE) { printf("Failed to create new thread.../n"); UnInitSock(); return -1; } printf("WaitForSingleObject(hThread,INFINITE) waiting.../n"); WaitForSingleObject(hThread,INFINITE); printf("WaitForSingleObject(hThread,INFINITE) return .../n"); delete pThreadParam; //while(true) //{ // int nRet=::WSAWaitForMultipleEvents(1,&g_wsaEvent,FALSE,10000,FALSE); // if(nRet==WAIT_FAILED) // 失败 // { // printf("failed wait for single object/n"); // break; // } // else if(nRet==WSA_WAIT_TIMEOUT) // 超时 // { // printf(" WSA_WAIT_TIMEOUT ... /n"); // continue; // } // else // 成功 -- 网络事件发生 // { // WSANETWORKEVENTS wsaNetEvent; // ::WSAEnumNetworkEvents(g_socketServer,g_wsaEvent,&wsaNetEvent); // if(wsaNetEvent.lNetworkEvents&FD_READ) // { // printf("FD_READ event occurs.../n"); // char buffer[1024]; // int nRet = recv(g_socketServer,buffer,1024,0); // if (nRet>0) // { // buffer[nRet] = '/0'; // printf("收到数据 %s/n",buffer); // Sleep(3000); // printf("发送数据..../n"); // int nSend = send(g_socketServer,buffer,nRet,0); // } // else // { // UnInitSock(); // return -1; // } // // } // else if(wsaNetEvent.lNetworkEvents&FD_WRITE) // { // printf("FD_WRITE event occurs.../n"); // printf("发送数据..../n"); // char szBuffer[] = "Hello world..."; // int nLen = sizeof(szBuffer); // nRet = send(g_socketServer,szBuffer,nLen,0); // if (nRet == SOCKET_ERROR) // { // printf("send() error .../n"); // UnInitSock(); // return -1; // } // printf("nRet of send() is %d/n",nRet); // } // if(wsaNetEvent.lNetworkEvents&FD_CLOSE) // { // printf("FD_CLOSE event occurs.../n"); // int nErrorCode = WSAGetLastError(); // printf("Error code is %d/n",nErrorCode); // if (nErrorCode == WSAECONNRESET) // { // printf("WSAECONNRESET error./n"); // } // else if (nErrorCode == WSAENETDOWN) // { // printf("WSAENETDOWN error./n"); // } // else if (nErrorCode == WSAENETRESET) // { // printf("WSAENETRESET error./n"); // } // UnInitSock(); // return -1; // } // } //} return 0; } /* * Windows Sockets definitions of regular Berkeley error constants #define WSAEWOULDBLOCK (WSABASEERR+35) #define WSAEINPROGRESS (WSABASEERR+36) #define WSAEALREADY (WSABASEERR+37) #define WSAENOTSOCK (WSABASEERR+38) #define WSAEDESTADDRREQ (WSABASEERR+39) #define WSAEMSGSIZE (WSABASEERR+40) #define WSAEPROTOTYPE (WSABASEERR+41) #define WSAENOPROTOOPT (WSABASEERR+42) #define WSAEPROTONOSUPPORT (WSABASEERR+43) #define WSAESOCKTNOSUPPORT (WSABASEERR+44) #define WSAEOPNOTSUPP (WSABASEERR+45) #define WSAEPFNOSUPPORT (WSABASEERR+46) #define WSAEAFNOSUPPORT (WSABASEERR+47) #define WSAEADDRINUSE (WSABASEERR+48) #define WSAEADDRNOTAVAIL (WSABASEERR+49) #define WSAENETDOWN (WSABASEERR+50) #define WSAENETUNREACH (WSABASEERR+51) #define WSAENETRESET (WSABASEERR+52) #define WSAECONNABORTED (WSABASEERR+53) #define WSAECONNRESET (WSABASEERR+54) #define WSAENOBUFS (WSABASEERR+55) #define WSAEISCONN (WSABASEERR+56) #define WSAENOTCONN (WSABASEERR+57) #define WSAESHUTDOWN (WSABASEERR+58) #define WSAETOOMANYREFS (WSABASEERR+59) #define WSAETIMEDOUT (WSABASEERR+60) #define WSAECONNREFUSED (WSABASEERR+61) #define WSAELOOP (WSABASEERR+62) #define WSAENAMETOOLONG (WSABASEERR+63) #define WSAEHOSTDOWN (WSABASEERR+64) #define WSAEHOSTUNREACH (WSABASEERR+65) #define WSAENOTEMPTY (WSABASEERR+66) #define WSAEPROCLIM (WSABASEERR+67) #define WSAEUSERS (WSABASEERR+68) #define WSAEDQUOT (WSABASEERR+69) #define WSAESTALE (WSABASEERR+70) #define WSAEREMOTE (WSABASEERR+71) */