基于WSAEventSelect的多线程服务器C++程序

#include "InitSock.h" #include<stdio.h> CInitSock initSock; PTHREAD_OBJ g_pThreadList;//指向线程对象列表表头 CRITICAL_SECTION g_cs;//同步对此全局变量的访问 LONG g_nTotalConnections;//总共连接的数量 LONG g_nCurrentConnections;//当前连接的数量 PTHREAD_OBJ GetThreadObj();//申请一个线程对象，初始化他的成员，并将它添加到线程对象的列表中 PSOCKET_OBJ GetSocketObj(SOCKET s);//申请一个套接字对象，以连成一个表 void FreeThreadObj(PTHREAD_OBJ pThread);//释放一个线程对象，并将它从线程对象列表中移除 void RebuildArray(PTHREAD_OBJ pThread);//重建线程对象的events数组 BOOL InsertSocketObj(PTHREAD_OBJ pThread,PSOCKET_OBJ pSocket);//像一个线程的套接字列表中插入一个套接字 void AssignToFreeThread(PSOCKET_OBJ pSocket);//将一个套接字对象安排给空闲的线程处理 void RemoveSocketObj(PTHREAD_OBJ pThread,PSOCKET_OBJ pSocket);//从给定线程的套接字对象列表中移除一个套接字对象 DWORD WINAPI ServerThread(LPVOID lpParam);//处理I/O线程 PSOCKET_OBJ FindSocketObj(PTHREAD_OBJ pThread,int nIndex);//nIndex从1开始 BOOL HandleIO(PTHREAD_OBJ pThread,PSOCKET_OBJ pSocket); PSOCKET_OBJ GetSocketObj(SOCKET s)//申请一个套接字对象，以连成一个表 { PSOCKET_OBJ pSocket=(PSOCKET_OBJ)::GlobalAlloc(GPTR,sizeof(SOCKET_OBJ)); if(pSocket!=NULL) { pSocket->s=s; pSocket->event=::WSACreateEvent(); } return pSocket; } void FreeSocketObj(PSOCKET_OBJ pSocket) //释放一个套接字对象 { ::CloseHandle(pSocket->event); if(pSocket->s!=INVALID_SOCKET) {::closesocket(pSocket->s);} ::GlobalFree(pSocket); } PTHREAD_OBJ GetThreadObj()//申请一个线程对象，初始化他的成员，并将它添加到线程对象的列表中 { PTHREAD_OBJ pThread=(PTHREAD_OBJ)::GlobalAlloc(GPTR,sizeof(THREAD_OBJ)); if(pThread!=NULL) { ::InitializeCriticalSection(&pThread->cs); //创建一个事件对象，用于指示该线程的句柄数组需要重建 pThread->events[0]=::WSACreateEvent(); //将新申请的线程对象添加到列表中 ::EnterCriticalSection(&g_cs); pThread->pNext=g_pThreadList; g_pThreadList=pThread; ::LeaveCriticalSection(&g_cs); } return pThread; } void FreeThreadObj(PTHREAD_OBJ pThread)//释放一个线程对象，并将它从线程对象列表中移除 { //在线程对象列表中查找pThread所指向的对象，如果找到就从中移除 ::EnterCriticalSection(&g_cs); PTHREAD_OBJ p=g_pThreadList; if(p==pThread)//是第一个？ { g_pThreadList=p->pNext; } else { while(p!=NULL&&p->pNext!=pThread) p=p->pNext; if(p!=NULL) { //此时，p是pThread的前一个，即p->pNext==pThread p->pNext=pThread; } } ::LeaveCriticalSection(&g_cs); //释放资源 ::CloseHandle(pThread->events[0]); ::DeleteCriticalSection(&pThread->cs); ::GlobalFree(pThread); } void RebuildArray(PTHREAD_OBJ pThread)//重建线程对象的events数组 { ::EnterCriticalSection(&pThread->cs); PSOCKET_OBJ pSocket=pThread->pSockHeader; int n=1;//从第一个开始写，第0个用于标示重建了 while(pSocket!=NULL) { pThread->events[n++]=pSocket->event; pSocket=pSocket->pNext; } ::LeaveCriticalSection(&pThread->cs); } BOOL InsertSocketObj(PTHREAD_OBJ pThread,PSOCKET_OBJ pSocket)//像一个线程的套接字列表中插入一个套接字 { BOOL bRet=FALSE; ::EnterCriticalSection(&pThread->cs); if(pThread->nSocketCount<WSA_MAXIMUM_WAIT_EVENTS-1) { if(pThread->pSockHeader==NULL) pThread->pSockHeader=pThread->pSockTail=pSocket; else { pThread->pSockTail->pNext=pSocket; pThread->pSockTail=pSocket; } pThread->nSocketCount++; bRet=TRUE; } ::LeaveCriticalSection(&pThread->cs); //插入成功，说明成功处理了客户的连接请求 if(bRet) { ::InterlockedIncrement(&g_nTotalConnections); ::InterlockedIncrement(&g_nCurrentConnections); } return bRet; } void AssignToFreeThread(PSOCKET_OBJ pSocket)//将一个套接字对象安排给空闲的线程处理 { pSocket->pNext=NULL; ::EnterCriticalSection(&g_cs); PTHREAD_OBJ pThread=g_pThreadList; //试图插入到现存线程； while(pThread!=NULL) { if(InsertSocketObj(pThread,pSocket)) break; pThread=pThread->pNext; } //没有空闲线程，为这个套接字创建新的线程 if(pThread==NULL) { pThread=GetThreadObj(); InsertSocketObj(pThread,pSocket); ::CreateThread(NULL,0,ServerThread,pThread,0,NULL);//MFC中要换函数 } ::LeaveCriticalSection(&g_cs); //指示线程重建句柄数组 ::WSASetEvent(pThread->events[0]); } void RemoveSocketObj(PTHREAD_OBJ pThread,PSOCKET_OBJ pSocket)//从给定线程的套接字对象列表中移除一个套接字对象 { ::EnterCriticalSection(&pThread->cs); //在套接字对象列表中查找指定的套接字对象，找到之后将其移除 PSOCKET_OBJ pTest=pThread->pSockHeader; if(pTest==pSocket) { if(pThread->pSockHeader==pThread->pSockTail) pThread->pSockTail=pThread->pSockHeader=pTest->pNext; else pThread->pSockHeader=pTest->pNext; } else { while(pTest!=NULL&&pTest->pNext!=pSocket) pTest=pTest->pNext; if(pTest!=NULL) { if(pThread->pSockTail==pSocket) pThread->pSockTail=pTest; pTest->pNext=pSocket->pNext; } } pThread->nSocketCount--; ::LeaveCriticalSection(&pThread->cs); ::WSASetEvent(pThread->events[0]);//指示线程重建句柄数组 ::InterlockedIncrement(&g_nCurrentConnections);//说明一个连接中断 } DWORD WINAPI ServerThread(LPVOID lpParam)//处理I/O线程 { //取得本线程对象的指针 PTHREAD_OBJ pThread=(PTHREAD_OBJ)lpParam; while(TRUE) { //等待网络事件 int nIndex=::WSAWaitForMultipleEvents( pThread->nSocketCount+1,pThread->events,FALSE,WSA_INFINITE,FALSE); nIndex=nIndex-WSA_WAIT_EVENT_0; //查看守信事件对象 for(int i=nIndex;i<pThread->nSocketCount+1;i++) { nIndex=::WSAWaitForMultipleEvents(1,&pThread->events[i],TRUE,1000,FALSE); if(nIndex==WSA_WAIT_FAILED||nIndex==WSA_WAIT_TIMEOUT) continue; else { if(i==0) //events[0]受信，重建数组 { RebuildArray(pThread); //如果没有客户I/O要处理，则本线程退出 if(pThread->nSocketCount==0) { FreeThreadObj(pThread); return 0; } ::WSAResetEvent(pThread->events[0]); } else //处理网络事件 { //查找对应的套接字对象指针，调用HandleIO处理网络事件 PSOCKET_OBJ pSocket=(PSOCKET_OBJ)FindSocketObj(pThread,i); if(pSocket!=NULL) { if(!HandleIO(pThread,pSocket)) RebuildArray(pThread); } else printf("Unable to find socket object\n"); } } } } return 0; } PSOCKET_OBJ FindSocketObj(PTHREAD_OBJ pThread,int nIndex)//nIndex从1开始 { //在套接字列表中查找 PSOCKET_OBJ pSocket=pThread->pSockHeader; while(--nIndex) { if(pSocket==NULL) return NULL; pSocket=pSocket->pNext; } return pSocket; } BOOL HandleIO(PTHREAD_OBJ pThread,PSOCKET_OBJ pSocket) { //获取具体发生的网络事件 WSANETWORKEVENTS event; ::WSAEnumNetworkEvents(pSocket->s,pSocket->event,&event); do { if(event.lNetworkEvents &FD_READ) //套接字可读 { if(event.iErrorCode[FD_READ_BIT]==0) { char szText[256]; int nRecv=::recv(pSocket->s,szText,strlen(szText),0); if(nRecv>0) { szText[nRecv]='\0'; printf("接收到数据：%s\n",szText); } } else break; } else if(event.lNetworkEvents &FD_CLOSE)//套接字关闭 break; else if(event.lNetworkEvents &FD_WRITE)//套接字可写 { if(event.iErrorCode[FD_READ_BIT]==0) { } else break; } return TRUE; } while(FALSE); //套接字关闭，或者有错误发生，程序都会转到这来执行 RemoveSocketObj(pThread,pSocket); FreeSocketObj(pSocket); return FALSE; } int main() { 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; } ::listen(sListen,200); //创建事件对象，并关联到监听的套接字上 WSAEVENT event=::WSACreateEvent(); ::WSAEventSelect(sListen,event,FD_ACCEPT|FD_CLOSE); ::InitializeCriticalSection(&g_c