從CSDN的一個xd那裡看到這個程式碼，覺得對WSAEventSelect模型的多執行緒實現做的非常好，程式碼行文風格和程式碼質量都是非常不錯。我做了些小的改動學習之用。如果你也感興趣不妨一起留言討論討論。

// WSAEventSelect_MT.cpp : Defines the entry point for the console application. // #include "stdafx.h" #include <Winsock2.h> #include <malloc.h> #pragma comment(lib,"Ws2_32.lib") /* WSAEventSelect IO模型伺服器實現 ----------------------------------------- 1.多執行緒實現,以支援更多的客戶端連線 2.多執行緒共享同一個監聽套接字socketListen 3.IO業務邏輯模組單獨實現 4.各個執行緒各自管理自己的socket客戶連線 */ const int MAX_SOCKET_COUNT_PER_THREAD = WSA_MAXIMUM_WAIT_EVENTS - 1; // 定義每個執行緒最大管理的socket連線數量 // 初始化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; } } //套接字物件 -- 客戶連線套接字資訊 /* 一個套接字物件相關的資訊：套接字、該套接字相關聯的事件物件、套接字地址 */ typedef struct ST_SOCKET_OBJ { SOCKET s; //套接字控制代碼 HANDLE hEvent; //與套接字s相關聯的事件物件控制代碼 sockaddr_in addrRemote; //客戶端地址資訊 ST_SOCKET_OBJ *pNext; //指向一個SOCKET_OBJ }SOCKET_OBJ,*pSOCKET_OBJ; //執行緒物件 /* 執行緒引數：工作執行緒監視的socket物件列表以及每個套接字關聯的事件陣列 */ typedef struct ST_THREAD_OBJ { HANDLE hEvents[WSA_MAXIMUM_WAIT_EVENTS]; //記錄當前執行緒要等待的事件物件的控制代碼 int nSocketCount; //記錄當前執行緒處理的套接字數量 pSOCKET_OBJ pSockHeader; //當前執行緒處理的套接字物件列表的頭指標 pSOCKET_OBJ pSockTail; //當前執行緒處理的套接字物件列表的尾指標 CRITICAL_SECTION cs; //關鍵程式碼段變數,為的是同步對本結構的訪問 ST_THREAD_OBJ* pNext; //指向下一個THREAD_OBJ }THREAD_OBJ,*pTHREAD_OBJ; //申請一個套接字物件 - 為一個連線的socket申請一個ST_SOCKET_OBJ物件，初始化 pSOCKET_OBJ GetSocketObj(SOCKET s) { pSOCKET_OBJ pSocket= (pSOCKET_OBJ)malloc(sizeof(SOCKET_OBJ)); // new SOCKET_OBJ() if(pSocket!=NULL) { pSocket->hEvent=::WSACreateEvent(); // 建立事件 pSocket->s=s; // 初始化套接字socket pSocket->pNext=NULL; } return pSocket; } //釋放一個套接字物件 void FreeSocketObj(pSOCKET_OBJ pSocket) { ::CloseHandle(pSocket->hEvent); if(pSocket->s!=INVALID_SOCKET) { ::closesocket(pSocket->s); } free(pSocket); // delete pSocket; pSocket=NULL; } pTHREAD_OBJ g_pThreadList=NULL; //指向執行緒物件列表的表頭 -- 執行緒引數列表，和執行緒一一對應 CRITICAL_SECTION g_cs; //同步對全域性變數g_pThreadList的訪問 //申請一個執行緒物件,初始化它的成員,並將它新增到執行緒物件列表中 pTHREAD_OBJ GetThreadObj() { pTHREAD_OBJ pThread=(pTHREAD_OBJ)malloc(sizeof(THREAD_OBJ)); if(pThread!=NULL) { pThread->pNext=NULL; pThread->nSocketCount=0; pThread->pSockHeader=pThread->pSockTail=NULL; ::InitializeCriticalSection(&pThread->cs); //建立一個事件物件,用於指示該執行緒的控制代碼陣列hEvents[]需要重組 pThread->hEvents[0]=WSACreateEvent(); //將執行緒物件加入執行緒物件列表中 ::EnterCriticalSection(&g_cs); pThread->pNext=g_pThreadList; g_pThreadList=pThread; ::LeaveCriticalSection(&g_cs); } return pThread; } //釋放一個執行緒物件 void FreeThreadObj(pTHREAD_OBJ pThread) { ::EnterCriticalSection(&g_cs); pTHREAD_OBJ p=g_pThreadList; if(p==pThread) { g_pThreadList=pThread->pNext; } else { while(p!=NULL&&p->pNext!=pThread) //找到pThread的前一個節點 { p=p->pNext; } if(p!=NULL) { p->pNext=pThread->pNext; } } ::LeaveCriticalSection(&g_cs); //釋放資源 ::CloseHandle(pThread->hEvents[0]); // pThread->hEvents[0]為用來指示hEvents[]需要重組的事件 ::DeleteCriticalSection(&pThread->cs); ::free(pThread); } //重新建立執行緒物件的events陣列 即將pSocket中的event與pThread中的events關聯起來 void RebuildArray(pTHREAD_OBJ pThread) { ::EnterCriticalSection(&pThread->cs); pSOCKET_OBJ pSocket=pThread->pSockHeader; int n=1; //從下標為1開始 跳過pThread->hEvents[0] while(pSocket!=NULL) { pThread->hEvents[n++]=pSocket->hEvent; pSocket=pSocket->pNext; } ::LeaveCriticalSection(&pThread->cs); } LONG g_nTotalConnections; //總共連線數量 LONG g_nCurrentConnections; //當前連線數量 //向執行緒的套接字列表中插入一個套接字物件 BOOL InsertSocketObj(pTHREAD_OBJ pThread,pSOCKET_OBJ pSocket) { BOOL bRet=FALSE; ::EnterCriticalSection(&pThread->cs); if(pThread->nSocketCount<MAX_SOCKET_COUNT_PER_THREAD) // 每個執行緒最多管理MAX_SOCKET_COUNT_PER_THREAD個socket連線 { 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 RemoveSocketObj(pTHREAD_OBJ pThread,pSOCKET_OBJ pSocket) { ::EnterCriticalSection(&pThread->cs); //在套接字物件列表中查詢指定的套接字物件,找到後將它移除 pSOCKET_OBJ pTemp=pThread->pSockHeader; if(pTemp==pSocket) { if(pThread->pSockHeader==pThread->pSockTail) { pThread->pSockHeader=pThread->pSockTail=pTemp->pNext; } else { pThread->pSockHeader=pTemp->pNext; } } else { while(pTemp!=NULL&&pTemp->pNext!=pSocket) { pTemp=pTemp->pNext; } if(pTemp!=NULL) { if(pThread->pSockTail==pSocket) { pThread->pSockTail=pTemp; } pTemp->pNext=pSocket->pNext; } } pThread->nSocketCount--; ::LeaveCriticalSection(&pThread->cs); ::WSASetEvent(pThread->hEvents[0]); //指示執行緒重建控制代碼陣列 ::InterlockedDecrement(&g_nCurrentConnections); } pSOCKET_OBJ FindSocketObj(pTHREAD_OBJ pThread,int nIndex) { 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 wsaEvent; ::WSAEnumNetworkEvents(pSocket->s,pSocket->hEvent/*待復位控制代碼*/,&wsaEvent); // FD_READ事件 if(wsaEvent.lNetworkEvents&FD_READ) { if(wsaEvent.iErrorCode[FD_READ_BIT]==0) { char *buf=new char[1024]; __try { int nRec=recv(pSocket->s,buf,1024,0); if(nRec>0) { buf[nRec]='/0'; printf("收到資料:%s/n",buf); return true; } else { return false; } } __finally { delete[]buf; } } } // FD_CLOSE事件 else if(wsaEvent.lNetworkEvents&FD_CLOSE) { printf("一連線繼開/n"); RemoveSocketObj(pThread,pSocket); // 需要重建控制代碼陣列 //::WSASetEvent(pThread->hEvents[0]); //指示執行緒重建控制代碼陣列 FreeSocketObj(pSocket); return false; } return true; } //處理I/O的執行緒 DWORD WINAPI ServerThread(LPVOID lpParam) { pTHREAD_OBJ pThread=(pTHREAD_OBJ)lpParam; printf("新執行緒%d起動/n",GetCurrentThreadId()); while(true) { //等待網路事件 int nIndex=WSAWaitForMultipleEvents(pThread->nSocketCount+1,pThread->hEvents,false,WSA_INFINITE,false); nIndex=nIndex-WSA_WAIT_EVENT_0; //if(WSA_WAIT_FAILED==nIndex||WSA_WAIT_TIMEOUT==nIndex) //{ // 失敗或者超時 // continue; //} //else //{ // 成功受信 // if(0==nIndex) // pThread->hEvents[0]受信 // { // RebuildArray(pThread); // 重建控制代碼陣列 // if(pThread->nSocketCount==0) // { // 如果該執行緒連線的socket數量已經為0，說明所有的客戶都已經斷開連線則執行緒函式自動退出 // FreeThreadObj(pThread); // 執行緒退出時候回收執行緒引數分配的記憶體 // printf("執行緒%d退出/n",GetCurrentThreadId()); // return 0; // } // ::WSAResetEvent(pThread->hEvents[0]); // } // else // { // pSOCKET_OBJ pSocket=FindSocketObj(pThread,nIndex); // if(pSocket!=NULL) // { // if(!HandleIO(pThread,pSocket)) // { // RebuildArray(pThread); // 重建控制代碼陣列 // } // } // else // { // printf("unable to find socket object/n"); // } // } //} //檢視受信事件 for(int i=nIndex;i<pThread->nSocketCount+1;i++) { nIndex=WSAWaitForMultipleEvents(1,&pThread->hEvents[i],true,0,false); if(WSA_WAIT_FAILED==nIndex||WSA_WAIT_TIMEOUT==nIndex) { continue; } else { if(0==i) { RebuildArray(pThread); // 重建控制代碼陣列 if(pThread->nSocketCount==0) { FreeThreadObj(pThread); printf("執行緒%d退出/n",GetCurrentThreadId()); return 0; } ::WSAResetEvent(pThread->hEvents[0]); } else { pSOCKET_OBJ pSocket=FindSocketObj(pThread,i); if(pSocket!=NULL) { if(!HandleIO(pThread,pSocket)) { RebuildArray(pThread); // 重建控制代碼陣列 } } else { printf("unable to find socket object/n"); } } } } } // end while() } //將一個套接字物件安排給空閒的執行緒處理 //如果所有執行緒接受的socket已滿，則新建執行緒提供服務 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); } ::LeaveCriticalSection(&g_cs); //指示執行緒重建控制代碼陣列 ::WSASetEvent(pThread->hEvents[0]); } int _tmain(int argc, _TCHAR* argv[]) { InitSock(); SOCKET sListen=socket(AF_INET,SOCK_STREAM,0); sockaddr_in sin; sin.sin_addr.S_un.S_addr=INADDR_ANY; sin.sin_family=AF_INET; sin.sin_port=htons(3456); if(SOCKET_ERROR==bind(sListen,(sockaddr*)&sin,sizeof(sockaddr))) { printf("繫結失敗/n"); ::WSACleanup(); return 0; } listen(sListen,SOMAXCONN); ::WSAEVENT wsaEvent=::WSACreateEvent(); ::WSAEventSelect(sListen,wsaEvent,FD_ACCEPT); ::InitializeCriticalSection(&g_cs); while(true) { //int nRet=::WaitForSingleObject(wsaEvent,10000); int nRet=::WSAWaitForMultipleEvents(1,&wsaEvent,FALSE,10000,FALSE); if(nRet==WAIT_FAILED) // 失敗 { printf("failed wait for single object/n"); break; } else if(nRet==WSA_WAIT_TIMEOUT) // 超時 { printf(" TotalConnections:%d/n",g_nTotalConnections); printf(" TotalCurrentConnections:%d/n",g_nCurrentConnections); } else // 成功 -- FD_ACCEPT事件發生 { ::WSANETWORKEVENTS wsaNetEvent; ::WSAEnumNetworkEvents(sListen,wsaEvent,&wsaNetEvent); if(wsaNetEvent.lNetworkEvents&FD_ACCEPT) { if(wsaNetEvent.iErrorCode[FD_ACCEPT_BIT]==0) { while(true) { // 把等待連線隊列當中的使用者都接受連線建立請求，建立連線後派遣給相關的服務執行緒處理 sockaddr_in si; int nLen=sizeof(si); SOCKET client=accept(sListen,(sockaddr*)&si,&nLen); if(client==SOCKET_ERROR) break; // 沒有連線請求需要建立了 pSOCKET_OBJ pSocket=GetSocketObj(client); pSocket->addrRemote=si; ::WSAEventSelect(pSocket->s,pSocket->hEvent,FD_READ|FD_CLOSE); AssignToFreeThread(pSocket); printf("一新連線/n"); } } } //::ResetEvent(wsaEvent); //while(true) //{ // 把等待連線隊列當中的使用者都接受連線建立請求，建立連線後派遣給相關的服務執行緒處理 // sockaddr_in si; // int nLen=sizeof(si); // SOCKET client=accept(sListen,(sockaddr*)&si,&nLen); // if(client==SOCKET_ERROR) // break; // 沒有連線請求需要建立了 // pSOCKET_OBJ pSocket=GetSocketObj(client); // pSocket->addrRemote=si; // ::WSAEventSelect(pSocket->s,pSocket->hEvent,FD_READ|FD_CLOSE); // AssignToFreeThread(pSocket); // printf("一新連線/n"); //} } } ::DeleteCriticalSection(&g_cs); return 0; }