基于SOCKET的完成端口示例

#include "InitSock.h" // 该类的作用只是用来初始化WS2_32.dll，即调用WSAStartup函数; CInitSock InitSock; #define BUFFER_SIZE 1024 // 关键字结构体; typedef struct _PER_HANDLE_DATA { SOCKET s; SOCKADDR_IN addr; } PER_HANDLE_DATA, *PPER_HANDLE_DATA; // 这个是用来表示发送或者接受的缓冲区，由于OVERLAPPED是第一个成员; // 所以在GetQueuedCompletionStatus返回的OVERLAPPED的指针中就是该结构体的指针; typedef struct _PER_IO_DATA { #define OP_READ 1 #define OP_WRITE 2 #define OP_ACCPET 3 OVERLAPPED ol; char buf[BUFFER_SIZE]; int iOpt; } PER_IO_DATA, *PPER_IO_DATA; // 工作线程; DWORD CALLBACK ServerThread( LPVOID lpParam ); int main() { // 创建完成端口; HANDLE hCompletion = CreateIoCompletionPort( INVALID_HANDLE_VALUE, 0, 0, 0 ); // 创建工作线程，这里指创建一个，可以创建多个，最好为CPU数目的两倍; CreateThread( NULL, 0, ServerThread, (LPVOID)hCompletion, 0, 0 ); // 传入端口号获取一个TCP套接字; SOCKET sListen = GetTCPSocket(4567); // 监听; listen(sListen, 5); // 这里循环接收套接字，并将新套接字与完成端口关联; while ( TRUE ) { // 接收套接字; SOCKADDR_IN saRemote; int iRemoteLen = sizeof(saRemote); SOCKET sNew = accept( sListen, (LPSOCKADDR)&saRemote, &iRemoteLen ); // 创建一个关键字结构体，并将套接字与完成关联; PPER_HANDLE_DATA pPerHandler = (PPER_HANDLE_DATA)GlobalAlloc(GPTR,sizeof(PER_HANDLE_DATA)); pPerHandler->s = sNew; memcpy( &pPerHandler->addr, &saRemote, iRemoteLen ); CreateIoCompletionPort( (HANDLE)pPerHandler->s, hCompletion, (ULONG_PTR)pPerHandler, 0 ); // 创建一个发送缓冲区，并发送异步接收请求; PPER_IO_DATA pPerIo = (PPER_IO_DATA)GlobalAlloc(GPTR, sizeof(PER_IO_DATA)); pPerIo->ol.hEvent = WSACreateEvent(); pPerIo->iOpt = OP_READ; WSABUF buf = { BUFFER_SIZE, pPerIo->buf }; DWORD dwRecv, dwFlags = 0; WSARecv( pPerHandler->s, &buf, 1, &dwRecv, &dwFlags, &pPerIo->ol, NULL ); } closesocket(sListen); return 0; } DWORD WINAPI ServerThread( LPVOID lpParam ) { HANDLE hCompletion = (HANDLE)lpParam; DWORD dwTrans; PPER_HANDLE_DATA pPerHandle; PPER_IO_DATA pPerIo; // 循环获取请求结果; while ( TRUE ) { // 获取请求结果; BOOL bOk = GetQueuedCompletionStatus( hCompletion, &dwTrans, (PULONG_PTR)&pPerHandle, (LPOVERLAPPED*)&pPerIo, WSA_INFINITE ); // 发生错误，关闭套接字; if ( !bOk ) { closesocket( pPerHandle->s ); GlobalFree( pPerHandle ); GlobalFree( pPerIo ); continue; } // 传送为0，说明对方关闭套接字; if ( dwTrans == 0 && (pPerIo->iOpt==OP_READ || pPerIo->iOpt==OP_WRITE) ) { closesocket( pPerHandle->s ); GlobalFree( pPerHandle ); GlobalFree( pPerIo ); continue; } // 根据套作类型，处理操作，这里只有可读操作; switch ( pPerIo->iOpt ) { case OP_READ: { // 读取成功; pPerIo->buf[dwTrans] = 0; printf( "接收到数据：%s\n", pPerIo->buf ); // 继续接收; WSABUF buf = { BUFFER_SIZE, pPerIo->buf }; DWORD dwRecv, dwFlags = 0; WSARecv( pPerHandle->s, &buf, 1, &dwRecv, &dwFlags, &pPerIo->ol, NULL ); } break; case OP_WRITE: break; case OP_ACCPET: break; } } return 0; }