WSAEventSelect 模型的实现

来源：互联网发布：唯品会快递配送网络编辑：程序博客网时间：2024/04/25 15:19

WSAEventSelect 模型跟WSAAsync模型类似，他们都是由网络事件通知的，不同的是

WSAAsyncSelect是通过窗口的消息来获得通知，而WSAEventSelect是通过事件对象受信来得到通知的。

1. WSAEventSelect首先把socket，事件对象，和一组由FD_READ，FD_WRITE,FD_ACCEPT,FD_CONNECT,FD_CLOSE等等组成的网络事件给绑定在一起。

2.然后调用WSAWaitForMultibleEvents来等待这些网络事件的发生。如果等待完成：

3.如果等待完成，那么就调用WSAEnumNetworkEvents来获得具体是什么网络事件。

下面是服务器服务主线程的代码：

　bool WSAEventSelectServer::StartServer(int nType,sockaddr_in addr){SOCKET socket_listen=socket(AF_INET,nType,0);if (socket_listen==INVALID_SOCKET){cout<<"服务器创建监听 socket失败"<<endl;return false;}else{cout<<"服务器创建监听 socket成功"<<endl;} int iRet=bind(socket_listen,(sockaddr*)&addr,sizeof(sockaddr)); if (iRet!=0) {cout<<"服务器 bind 失败"<<endl;return false; } else {cout<<"服务器 bind 成功"<<endl; }iRet=listen(socket_listen,SOMAXCONN);if (iRet!=0){cout<<"服务器 listen 失败"<<endl;return false;}else{cout<<"服务器 listen 成功"<<endl;}u_long value=1;iRet=ioctlsocket(socket_listen,FIONBIO ,&value);if (iRet==0){cout<<"设置服务器非阻塞模式成功"<<endl;} else{cout<<"设置服务器非阻塞模式失败"<<endl;return false;}//创建一个跟socket_listen相关的WSAEventWSAEVENT listen_event=WSACreateEvent();WSAEventSelect(socket_listen,listen_event,FD_CLOSE|FD_ACCEPT);//初始化全局关键代码段InitializeCriticalSection(&g_cs);while (true){DWORD dwRet=WSAWaitForMultipleEvents(1,&listen_event,TRUE,5*1000,FALSE); if(dwRet==WSA_WAIT_TIMEOUT){continue;}else if (dwRet==WSA_WAIT_FAILED){cout<<"WSAWaitForMultipleEvent失败，退出主循环"<<endl;break;} else{//WSANETWORKEVENTS networkevents;ZeroMemory(&networkevents,sizeof(WSANETWORKEVENTS)); //查看是什么网络事件WSAEnumNetworkEvents(socket_listen,listen_event,&networkevents);if (networkevents.lNetworkEvents & FD_ACCEPT){if (networkevents.iErrorCode[FD_ACCEPT_BIT]==0){//如果是 FD_ACCEPT并且没有错误，那么处理连接sockaddr_in addr;int len=sizeof(sockaddr_in);SOCKET socket_client=accept(socket_listen,(sockaddr *)&addr,&len);if (socket_client==INVALID_SOCKET){cout<<"accept返回INVALID_SOCKET，退出主循环"<<endl;break;}//将新的socket放入到eventselect 模型，等待网络事件的发生。WSAEVENT event_client=WSACreateEvent();WSAEventSelect(socket_client,event_client,FD_READ|FD_WRITE|FD_CLOSE);//初始化SOCKEET_OBJSOCKET_OBJ * lpSocketObj=CreateSocketObj();lpSocketObj->socket=socket_client;lpSocketObj->socket_event=event_client;lpSocketObj->remote_addr=addr;lpSocketObj->pNext=NULL;//将这个SOCKETOBJ分配的一个线程AddStockObjToThreadObj(lpSocketObj);}}}}//删除全局关键代码段DeleteCriticalSection(&g_cs);return true;}

因为每个WSAWaitForMultipleEvents只能够最多等待WSA_MAXMUM_WAIT_OBJECTS个事件对象，这个值被定义为64，所以最多等待64个对象。要等待更多的客户端，就需要创建更多的线程，所有这里实现了个线程池，如果一个线程

已经等待了64个，那么就新建一个线程去等待去管理这些套接字。

　这里实现了一个线程池，通过两个结构体：

　//链表SOCKET 数据结构typedef struct _SOCKET_OBJ{SOCKET socket;WSAEVENT socket_event;sockaddr_in remote_addr;_SOCKET_OBJ * pNext;}SOCKET_OBJ;//线程 THREAD_OBJ 数据结构typedef struct _THREAD_OBJ{DWORD dwEventNum;//线程要等待的事件对象数组WSAEVENT events[WSA_MAXIMUM_WAIT_EVENTS];SOCKET_OBJ * socket_begin;SOCKET_OBJ * socket_end;//同步对这个数据结构的访问CRITICAL_SECTION cs;_THREAD_OBJ * pNext;}THREAD_OBJ;

　来实现线程对套接字的管理，有个全局的对象：

　//THREAD _OBJ 全局列表
THREAD_OBJ * g_thread_list=NULL;

　用来保存所有的线程结构链表，而所有的SOCKET_OBJ结构体又被包含每个THREAD_OBJ结构体中，通过这两个list来实现线程池的。

线程工作函数如下:

//工作线程函数,传入的参数是THREAD_OJB *void WorkThreadFuc(void *param){cout<<"进入工作线程函数"<<endl;THREAD_OBJ * lpthreadobj=(THREAD_OBJ *)param;while(true){int iRet=WSAWaitForMultipleEvents(lpthreadobj->dwEventNum+1,lpthreadobj->events,false,1000,FALSE);//找到是从第几个事件对象开始授信iRet=iRet-WSA_WAIT_EVENT_0;//从开始的地方依次查看是否授信for (int i=iRet;i<lpthreadobj->dwEventNum+1;i++){iRet=WSAWaitForMultipleEvents(1,&lpthreadobj->events[i],true,0,FALSE);if (iRet==WSA_WAIT_FAILED || iRet==WSA_WAIT_TIMEOUT){continue;}else{//如果是需要重置数组if (i==0){//重置数组RebuildEventArray(lpthreadobj);}else{//查找跟现在的事件对象相对应的socketobj,这个查找方法跟RebuildArray有很大的关系 SOCKET_OBJ * lpsocketobj=FindSocketObjByEnvet(lpthreadobj,lpthreadobj->events[i]);if (lpsocketobj){//处理HandleIOFuc(lpthreadobj,lpsocketobj);} else{cout<<"没有找到事件对象所对应的socket"<<endl;} }}}}}

各相关辅助函数如下：

//申请一个SOCKET_OBJ的空间，并且返回指针SOCKET_OBJ * CreateSocketObj(){SOCKET_OBJ * lpRet;lpRet= (SOCKET_OBJ *) malloc(sizeof(SOCKET_OBJ));return lpRet;}//释放一个SOCKET_OBJ的空间void FreeSocketObj(SOCKET_OBJ * lpsocketobj){if (lpsocketobj->socket!=INVALID_SOCKET){closesocket(lpsocketobj->socket);}if (lpsocketobj->socket_event){CloseHandle(lpsocketobj->socket_event);}free((void *)lpsocketobj);}//申请一个THREAD_OBJ的空间，返回指针THREAD_OBJ * CreateThreadObj(){THREAD_OBJ * lpRet;lpRet=(THREAD_OBJ *)malloc(sizeof(THREAD_OBJ));//初始化if (lpRet!=NULL){InitializeCriticalSection(&lpRet->cs);lpRet->dwEventNum=0;lpRet->socket_begin=lpRet->socket_end=NULL;//将创建的thareadobj插入到全局g_thread_list的前部 EnterCriticalSection(&g_cs);lpRet->pNext=g_thread_list;g_thread_list=lpRet;LeaveCriticalSection(&g_cs);//创建一个事件对象，用于指示句柄数组需要重建lpRet->events[0]=WSACreateEvent();}return lpRet;}//重置数组void RebuildEventArray(THREAD_OBJ * lpthreadobj){if (lpthreadobj==NULL){return ;}EnterCriticalSection(&lpthreadobj->cs);lpthreadobj->dwEventNum=0;SOCKET_OBJ * lpsocketobj=lpthreadobj->socket_begin;while(lpsocketobj!=NULL){lpthreadobj->dwEventNum++;lpthreadobj->events[lpthreadobj->dwEventNum]=lpsocketobj->socket_event;//循环到下一个lpsocketobj=lpsocketobj->pNext;}//重置WSAResetEvent(lpthreadobj->events[0]);LeaveCriticalSection(&lpthreadobj->cs);}//将一个套接字OBJ插入一个线程OBJ,如果成功，那么就返回true ,否则返回falsebool InsertSocketObjToThreadObj(THREAD_OBJ * lpthreadobj,SOCKET_OBJ *lpsocketobj){ bool bRet=false;EnterCriticalSection(&(lpthreadobj->cs)); //如果当前threadobj 还没有满if (lpthreadobj->dwEventNum<WSA_MAXIMUM_WAIT_EVENTS-1){//如果是第一个SOCKET_OBJif (lpthreadobj->socket_begin==NULL){lpthreadobj->socket_begin=lpthreadobj->socket_end=lpsocketobj;} else{//否则，添加SOCKET_OBJ到尾部lpthreadobj->socket_end->pNext=lpsocketobj;lpthreadobj->socket_end=lpsocketobj;}//增加SOCKET_OBJ计数lpthreadobj->dwEventNum++;//RebuildEventArray(lpthreadobj);bRet=true;}else{bRet=false;}LeaveCriticalSection(&(lpthreadobj->cs));return bRet;}//查找事件对象所对应的SOCKET_OBJSOCKET_OBJ * FindSocketObjByEnvet(THREAD_OBJ * lpthreadobj,WSAEVENT event){SOCKET_OBJ * lpsocketobj=lpthreadobj->socket_begin;while(lpsocketobj){if (lpsocketobj->socket_event==event){return lpsocketobj;}else{lpsocketobj=lpsocketobj->pNext;}}}//从ThreadObj中移除SocketObjvoid RemoveSocketObjFromThreadObj(THREAD_OBJ * lpthreadobj,SOCKET_OBJ * lpsocketobj){if (lpsocketobj==NULL || lpthreadobj==NULL){return ;}EnterCriticalSection(&lpthreadobj->cs);SOCKET_OBJ * lpformer=lpthreadobj->socket_begin;//如果要删除的是头节点if (lpformer==lpsocketobj){//如果头尾相等，那么同时指向头的下一个节点if (lpthreadobj->socket_begin==lpthreadobj->socket_end){lpthreadobj->socket_begin=lpthreadobj->socket_end=lpformer->pNext;} else{//否则，头指向下一个lpthreadobj->socket_begin=lpformer->pNext;}}else{while(lpformer!=NULL && lpformer->pNext!=lpsocketobj){lpformer=lpformer->pNext;}if (lpformer!=NULL){//意味着 lpformer->pNext==lpsocketobjif (lpsocketobj==lpthreadobj->socket_end){//如果要移除的是尾节点lpthreadobj->socket_end=lpformer;}lpformer->pNext=lpsocketobj->pNext;}}lpthreadobj->dwEventNum--;LeaveCriticalSection(&lpthreadobj->cs);//重建数组WSASetEvent(lpthreadobj->events[0]);}//处理IO事件的函数bool HandleIOFuc(THREAD_OBJ *lpthreadobj,SOCKET_OBJ * lpsocketobj){WSANETWORKEVENTS events;WSAEnumNetworkEvents(lpsocketobj->socket,lpsocketobj->socket_event,&events);do {if (events.lNetworkEvents & FD_READ){if (events.iErrorCode[FD_READ_BIT]==0){char buffer[MAX_PATH]={0};if (recv(lpsocketobj->socket,buffer,MAX_PATH,0)){cout<<"收到消息："<<buffer<<endl;}}else{break;}}else if (events.lNetworkEvents & FD_CLOSE){cout<<"客户端连接断开"<<endl;break;}else if (events.lNetworkEvents & FD_WRITE){if (events.iErrorCode[FD_WRITE_BIT]==0){cout<<"客户端可写"<<endl;}else{break;}}//如果不出错，或者套接字不被关闭，那么直接返回truereturn true;} while (0);//如果出错，或者套接字被关闭，那么就要移除套接字了 RemoveSocketObjFromThreadObj(lpthreadobj,lpsocketobj);//释放socketobj结构体FreeSocketObj(lpsocketobj);}//将一个StockOjb添加到一个ThreadObj中bool AddStockObjToThreadObj(SOCKET_OBJ * lpSocketObj){lpSocketObj->pNext=NULL;//关键代码段保护对g_thread_list的访问EnterCriticalSection(&g_cs);THREAD_OBJ *lpthreadobj=g_thread_list; while(lpthreadobj){//如果能够在当前thrare_obj插入成功，那么就跳出循环 if (InsertSocketObjToThreadObj(lpthreadobj,lpSocketObj)) {break; }//否则查找下一个threadobjlpthreadobj=lpthreadobj->pNext;}//如果最终都没有在存在的threadobj中插入socketobj成功，那么新建一个threadobj以及新线程if (lpthreadobj==NULL){ lpthreadobj=CreateThreadObj();InsertSocketObjToThreadObj(lpthreadobj,lpSocketObj);//先调用，否则线程中不能够等待到数组重建的事件授信RebuildEventArray(lpthreadobj);//新建线程_beginthread(WorkThreadFuc,NULL,lpthreadobj);}//离开关键代码段LeaveCriticalSection(&g_cs);//插入一个socketobj后，需要更新事件句柄数组WSASetEvent(lpthreadobj->events[0]);return true;}