IOCP初探

  这个是第二个版本了，这里我用了AcceptEx这个函数，使得接收连接也可以进行重叠操作了，我暂时开了2个工作线程，你可以看到，当GetQueuedCompletionStatus需要额外的线程来处理客户数据的时候，第二个工作线程就会被唤醒。

#include <stdio.h> #include <tchar.h> #include <process.h> #include <tchar.h> #include <iostream> #include <algorithm> #include <winsock2.h> #include <crtdbg.h> #include <cstring> #include <iomanip> #include <list> #pragma comment(lib, "ws2_32.lib")

// IOCP.cpp : 定义控制台应用程序的入口点。 // #include "stdafx.h" using namespace std; #define LOC_PORT 5678 #define LOC_ADDR "127.0.0.1" #define LOC_THREAD_COUNT 4 #define LOC_BUFFER_LEN 1024 class CSocketObject { public: CSocketObject(){WSAStartup(MAKEWORD(2,2), &m_wsaData);} virtual ~CSocketObject(){WSACleanup();} public: virtual void Run() = 0; protected: WSADATA m_wsaData; }; class CTcpServer : public CSocketObject { public: virtual void Run(); private: HANDLE m_hCompPort; SOCKET m_tSock; typedef BOOL (WINAPI *PFNACCEPTEX)(SOCKET, SOCKET, PVOID, DWORD, DWORD, DWORD, LPDWORD, LPOVERLAPPED); PFNACCEPTEX m_pfnAcceptEx; typedef struct { OVERLAPPED ovlp; SOCKET sock; sockaddr_in tAddr; WSABUF buf; char ucOpera; }EVENT_T; enum{OP_RECV, OP_SEND, OP_ACCEPT}; static unsigned int __stdcall WorkerThreadProc(void *pvArg); }; unsigned int CTcpServer::WorkerThreadProc(void *pvArg) { DWORD dwTrans; EVENT_T *ptEv, *ptMainEv; OVERLAPPED *povlp; CTcpServer *poTcpServer = (CTcpServer*)pvArg; DWORD udwID = GetCurrentThreadId(); sockaddr_in *ptRemoteAddr, *ptLocalAddr; int dwLocal, dwRemote; DWORD dwFlags; DWORD udwRecv; while (1) { GetQueuedCompletionStatus(poTcpServer->m_hCompPort, &dwTrans, (DWORD*)&ptEv, &povlp, INFINITE); printf("### this is worker (%u) ###", udwID); switch (ptEv->ucOpera) { case CTcpServer::OP_RECV: if (0 == dwTrans) { printf("client: %s leave/n", inet_ntoa(ptEv->tAddr.sin_addr)); closesocket(ptEv->sock); delete[] ptEv->buf.buf; delete ptEv; } else { ptEv->buf.buf[dwTrans] = 0; printf("message from %s: %s/n", inet_ntoa(ptEv->tAddr.sin_addr), ptEv->buf.buf); ZeroMemory(&ptEv->ovlp, sizeof(ptEv->ovlp)); DWORD dwRecvd, dwFlags = 0; WSARecv(ptEv->sock, &ptEv->buf, 1, &dwRecvd, &dwFlags, &ptEv->ovlp, NULL); } break; case CTcpServer::OP_SEND: break; case CTcpServer::OP_ACCEPT: ptMainEv = ptEv; ptEv = new EVENT_T; GetAcceptExSockaddrs(ptMainEv->buf.buf, 0, sizeof(SOCKADDR_IN) + 16, sizeof(SOCKADDR_IN) + 16, (sockaddr**)&ptLocalAddr, &dwLocal, (sockaddr**)&ptRemoteAddr, &dwRemote); memcpy(&ptEv->tAddr, ptRemoteAddr, dwRemote); ptEv->sock = ptMainEv->sock; printf("client %s come in/n", inet_ntoa(ptRemoteAddr->sin_addr)); send(ptEv->sock, "hello, client!", strlen("hello, client!"), 0); ptEv->buf.buf = new char[LOC_BUFFER_LEN]; ptEv->buf.len = LOC_BUFFER_LEN; ptEv->ucOpera = CTcpServer::OP_RECV; ZeroMemory(&ptEv->ovlp, sizeof(ptEv->ovlp)); dwFlags = 0; CreateIoCompletionPort((HANDLE)ptEv->sock, poTcpServer->m_hCompPort, (DWORD)ptEv, LOC_THREAD_COUNT); WSARecv(ptEv->sock, &ptEv->buf, 1, &udwRecv, &dwFlags, &ptEv->ovlp, NULL); ZeroMemory(&ptMainEv->ovlp, sizeof(ptMainEv->ovlp)); ptMainEv->sock = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, WSA_FLAG_OVERLAPPED); ptMainEv->ucOpera = CTcpServer::OP_ACCEPT; if (!AcceptEx(poTcpServer->m_tSock, ptMainEv->sock, ptMainEv->buf.buf, 0, sizeof(SOCKADDR_IN) + 16, sizeof(SOCKADDR_IN) + 16, &udwRecv, &ptMainEv->ovlp)) { if (WSA_IO_PENDING != WSAGetLastError()) { printf("AcceptEx error!/n"); } } break; default: continue; } } return 0; } void CTcpServer::Run() { m_tSock = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, WSA_FLAG_OVERLAPPED); SOCKADDR_IN tAddr; tAddr.sin_family = AF_INET; tAddr.sin_port = htons(LOC_PORT); tAddr.sin_addr.s_addr = inet_addr(LOC_ADDR); _ASSERT(bind(m_tSock,(SOCKADDR*)&tAddr, sizeof(tAddr)) == 0); m_hCompPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, 0, LOC_THREAD_COUNT); HANDLE m_hThreads[3]; m_hThreads[0] = (HANDLE)_beginthreadex(NULL, 0, WorkerThreadProc, this, 0, NULL); m_hThreads[1] = (HANDLE)_beginthreadex(NULL, 0, WorkerThreadProc, this, 0, NULL); EVENT_T *ptEv = new EVENT_T; ptEv->buf.buf = new char[LOC_BUFFER_LEN]; ptEv->buf.len = LOC_BUFFER_LEN; ptEv->ucOpera = CTcpServer::OP_ACCEPT; ptEv->sock = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, WSA_FLAG_OVERLAPPED); if (CreateIoCompletionPort((HANDLE)m_tSock, m_hCompPort, (DWORD)ptEv, LOC_THREAD_COUNT) == NULL) { printf("CreateIoCompletionPort error!/n"); delete ptEv; delete[] ptEv->buf.buf; return; } listen(m_tSock, 5); DWORD udwRecv; ZeroMemory(&ptEv->ovlp, sizeof(ptEv->ovlp)); if (!AcceptEx(m_tSock, ptEv->sock, ptEv->buf.buf, 0, sizeof(SOCKADDR_IN) + 16, sizeof(SOCKADDR_IN) + 16, &udwRecv, &ptEv->ovlp)) { if (ERROR_IO_PENDING != WSAGetLastError()) { printf("AcceptEx error!/n"); } } WaitForMultipleObjects(2, m_hThreads, true, INFINITE); closesocket(m_tSock); delete ptEv; delete[] ptEv->buf.buf; CloseHandle(m_hThreads[0]); CloseHandle(m_hThreads[1]); CloseHandle(m_hCompPort); } int _tmain(int argc, TCHAR *argv[]) { CSocketObject *poSockServer = new CTcpServer; poSockServer->Run(); delete poSockServer; return 0; }