IOCP二

简介: // IOCP2.cpp : Defines the entry point for the console application.

// IOCP2.cpp : Defines the entry point for the console application. // #include "stdafx.h" #include <WinSock2.h> #include <MSWSock.h> #include <Windows.h> #include <process.h> #pragma comment(lib, "WS2_32.lib") #define MAX_BUFFER 256 #define MAX_TIMEOUT 1000 #define MAX_SOCKET 1024 #define MAX_THREAD 64 #define MAX_ACCEPT 5 typedef enum _OPERATION_INFO_ { OP_NULL, OP_ACCEPT, OP_READ, OP_WRITE }OPERATIONINFO; typedef struct _PER_HANDLE_DATA_ { public: _PER_HANDLE_DATA_() { clean(); } ~_PER_HANDLE_DATA_() { clean(); } protected: void clean() { sock = INVALID_SOCKET; memset(&addr, 0, sizeof(addr)); addr.sin_addr.S_un.S_addr = INADDR_ANY; addr.sin_port = htons(0); addr.sin_family = AF_INET; } public: SOCKET sock; SOCKADDR_IN addr; }PERHANDLEDATA, *PPERHANDLEDATA; typedef struct _PER_IO_DTATA_ { public: _PER_IO_DTATA_() { clean(); } ~_PER_IO_DTATA_() { clean(); } void clean() { ZeroMemory(&ol, sizeof(ol)); memset(buf, 0, sizeof(buf)); sAccept = INVALID_SOCKET; sListen = INVALID_SOCKET; wsaBuf.buf = buf; wsaBuf.len = MAX_BUFFER; opType = OP_NULL; } public: WSAOVERLAPPED ol; SOCKET sAccept; // Only valid with AcceptEx SOCKET sListen; // Only valid with AcceptEx WSABUF wsaBuf; char buf[MAX_BUFFER]; OPERATIONINFO opType; }PERIODATA, *PPERIODATA; HANDLE hThread[MAX_THREAD] = {0}; PERIODATA* pAcceptData[MAX_ACCEPT] = {0}; int g_nThread = 0; BOOL g_bExitThread = FALSE; LPFN_ACCEPTEX lpfnAcceptEx = NULL; LPFN_GETACCEPTEXSOCKADDRS lpfnGetAcceptExSockAddrs = NULL; GUID GuidAcceptEx = WSAID_ACCEPTEX; GUID GuidGetAcceptExSockAddrs = WSAID_GETACCEPTEXSOCKADDRS; unsigned __stdcall ThreadProc(LPVOID lParam); BOOL PostAccept(PERIODATA* pIoData); int _tmain(int argc, _TCHAR* argv[]) { WORD wVersionRequested = MAKEWORD(2, 2); WSADATA wsaData; if(0 != WSAStartup(wVersionRequested, &wsaData)) { printf("WSAStartup failed with error code: %d/n", GetLastError()); return EXIT_FAILURE; } if(2 != HIBYTE(wsaData.wVersion) || 2 != LOBYTE(wsaData.wVersion)) { printf("Socket version not supported./n"); WSACleanup(); return EXIT_FAILURE; } // Create IOCP HANDLE hIOCP = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, NULL, 0); if(NULL == hIOCP) { printf("CreateIoCompletionPort failed with error code: %d/n", WSAGetLastError()); WSACleanup(); return EXIT_FAILURE; } // Create worker thread SYSTEM_INFO si = {0}; GetSystemInfo(&si); for(int i = 0; i < (int)si.dwNumberOfProcessors+2; i++) { hThread[g_nThread] = (HANDLE)_beginthreadex(NULL, 0, ThreadProc, (LPVOID)hIOCP, 0, NULL); if(NULL == hThread[g_nThread]) { printf("_beginthreadex failed with error code: %d/n", GetLastError()); continue; } ++g_nThread; if(g_nThread > MAX_THREAD) { break; } } // Create listen SOCKET SOCKET sListen = WSASocket(AF_INET, SOCK_STREAM, IPPROTO_TCP, NULL, 0, WSA_FLAG_OVERLAPPED); if(INVALID_SOCKET == sListen) { printf("WSASocket failed with error code: %d/n", WSAGetLastError()); goto EXIT_CODE; } // Associate SOCKET with IOCP if(NULL == CreateIoCompletionPort((HANDLE)sListen, hIOCP, NULL, 0)) { printf("CreateIoCompletionPort failed with error code: %d/n", WSAGetLastError()); if(INVALID_SOCKET != sListen) { closesocket(sListen); sListen = INVALID_SOCKET; } goto EXIT_CODE; } // Bind SOCKET SOCKADDR_IN addr; memset(&addr, 0, sizeof(addr)); addr.sin_family = AF_INET; addr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1"); addr.sin_port = htons(5050); if(SOCKET_ERROR == bind(sListen, (LPSOCKADDR)&addr, sizeof(addr))) { printf("bind failed with error code: %d/n", WSAGetLastError()); if(INVALID_SOCKET != sListen) { closesocket(sListen); sListen = INVALID_SOCKET; } goto EXIT_CODE; } // Start Listen if(SOCKET_ERROR == listen(sListen, 200)) { printf("listen failed with error code: %d/n", WSAGetLastError()); if(INVALID_SOCKET != sListen) { closesocket(sListen); sListen = INVALID_SOCKET; } goto EXIT_CODE; } printf("Server start, wait for client to connect .../n"); DWORD dwBytes = 0; if(SOCKET_ERROR == WSAIoctl(sListen, SIO_GET_EXTENSION_FUNCTION_POINTER, &GuidAcceptEx, sizeof(GuidAcceptEx), &lpfnAcceptEx, sizeof(lpfnAcceptEx), &dwBytes, NULL, NULL)) { printf("WSAIoctl failed with error code: %d/n", WSAGetLastError()); if(INVALID_SOCKET != sListen) { closesocket(sListen); sListen = INVALID_SOCKET; } goto EXIT_CODE; } if(SOCKET_ERROR == WSAIoctl(sListen, SIO_GET_EXTENSION_FUNCTION_POINTER, &GuidGetAcceptExSockAddrs, sizeof(GuidGetAcceptExSockAddrs), &lpfnGetAcceptExSockAddrs, sizeof(lpfnGetAcceptExSockAddrs), &dwBytes, NULL, NULL)) { printf("WSAIoctl failed with error code: %d/n", WSAGetLastError()); if(INVALID_SOCKET != sListen) { closesocket(sListen); sListen = INVALID_SOCKET; } goto EXIT_CODE; } // Post MAX_ACCEPT accept for(int i=0; i<MAX_ACCEPT; i++) { pAcceptData[i] = new PERIODATA; pAcceptData[i]->sListen = sListen; PostAccept(pAcceptData[i]); } // After 1 hour later, Server shutdown. Sleep(1000 * 60 *60); EXIT_CODE: g_bExitThread = TRUE; PostQueuedCompletionStatus(hIOCP, 0, NULL, NULL); WaitForMultipleObjects(g_nThread, hThread, TRUE, INFINITE); for(int i = 0; i < g_nThread; i++) { CloseHandle(hThread[g_nThread]); } for(int i=0; i<MAX_ACCEPT; i++) { if(pAcceptData[i]) { delete pAcceptData[i]; pAcceptData[i] = NULL; } } if(INVALID_SOCKET != sListen) { closesocket(sListen); sListen = INVALID_SOCKET; } CloseHandle(hIOCP); // Close IOCP WSACleanup(); return 0; } BOOL PostAccept(PERIODATA* pIoData) { if(INVALID_SOCKET == pIoData->sListen) { return FALSE; } DWORD dwBytes = 0; pIoData->opType = OP_ACCEPT; pIoData->sAccept = WSASocket(AF_INET, SOCK_STREAM, IPPROTO_TCP, NULL, 0, WSA_FLAG_OVERLAPPED); if(INVALID_SOCKET == pIoData->sAccept) { printf("WSASocket failed with error code: %d/n", WSAGetLastError()); return FALSE; } if(FALSE == lpfnAcceptEx(pIoData->sListen, pIoData->sAccept, pIoData->wsaBuf.buf, pIoData->wsaBuf.len - ((sizeof(SOCKADDR_IN)+16)*2), sizeof(SOCKADDR_IN)+16, sizeof(SOCKADDR_IN)+16, &dwBytes, &(pIoData->ol))) { if(WSA_IO_PENDING != WSAGetLastError()) { printf("lpfnAcceptEx failed with error code: %d/n", WSAGetLastError()); return FALSE; } } return TRUE; } unsigned __stdcall ThreadProc(LPVOID lParam) { HANDLE hIOCP = (HANDLE)lParam; PERHANDLEDATA* pPerHandleData = NULL; PERIODATA* pPerIoData = NULL; WSAOVERLAPPED* lpOverlapped = NULL; DWORD dwTrans = 0; DWORD dwFlags = 0; while(!g_bExitThread) { BOOL bRet = GetQueuedCompletionStatus(hIOCP, &dwTrans, (PULONG_PTR)&pPerHandleData, &lpOverlapped, MAX_TIMEOUT); if(!bRet) { // Timeout and exit thread if(WAIT_TIMEOUT == GetLastError()) { continue; } // Error printf("GetQueuedCompletionStatus failed with error: %d/n", GetLastError()); continue; } else { pPerIoData = CONTAINING_RECORD(lpOverlapped, PERIODATA, ol); if(NULL == pPerIoData) { // Exit thread break; } if((0 == dwTrans) && (OP_READ == pPerIoData->opType || OP_WRITE == pPerIoData->opType)) { // Client leave. printf("Client:

目录
相关文章
|
8天前
|
存储 关系型数据库 分布式数据库
PostgreSQL 18 发布,快来 PolarDB 尝鲜!
PostgreSQL 18 发布,PolarDB for PostgreSQL 全面兼容。新版本支持异步I/O、UUIDv7、虚拟生成列、逻辑复制增强及OAuth认证,显著提升性能与安全。PolarDB-PG 18 支持存算分离架构,融合海量弹性存储与极致计算性能,搭配丰富插件生态,为企业提供高效、稳定、灵活的云数据库解决方案,助力企业数字化转型如虎添翼!
|
6天前
|
存储 人工智能 Java
AI 超级智能体全栈项目阶段二:Prompt 优化技巧与学术分析 AI 应用开发实现上下文联系多轮对话
本文讲解 Prompt 基本概念与 10 个优化技巧,结合学术分析 AI 应用的需求分析、设计方案,介绍 Spring AI 中 ChatClient 及 Advisors 的使用。
330 130
AI 超级智能体全栈项目阶段二:Prompt 优化技巧与学术分析 AI 应用开发实现上下文联系多轮对话
|
18天前
|
弹性计算 关系型数据库 微服务
基于 Docker 与 Kubernetes(K3s)的微服务:阿里云生产环境扩容实践
在微服务架构中,如何实现“稳定扩容”与“成本可控”是企业面临的核心挑战。本文结合 Python FastAPI 微服务实战,详解如何基于阿里云基础设施,利用 Docker 封装服务、K3s 实现容器编排,构建生产级微服务架构。内容涵盖容器构建、集群部署、自动扩缩容、可观测性等关键环节,适配阿里云资源特性与服务生态,助力企业打造低成本、高可靠、易扩展的微服务解决方案。
1331 8
|
5天前
|
监控 JavaScript Java
基于大模型技术的反欺诈知识问答系统
随着互联网与金融科技发展,网络欺诈频发,构建高效反欺诈平台成为迫切需求。本文基于Java、Vue.js、Spring Boot与MySQL技术,设计实现集欺诈识别、宣传教育、用户互动于一体的反欺诈系统,提升公众防范意识,助力企业合规与用户权益保护。
|
17天前
|
机器学习/深度学习 人工智能 前端开发
通义DeepResearch全面开源!同步分享可落地的高阶Agent构建方法论
通义研究团队开源发布通义 DeepResearch —— 首个在性能上可与 OpenAI DeepResearch 相媲美、并在多项权威基准测试中取得领先表现的全开源 Web Agent。
1412 87
|
6天前
|
人工智能 Java API
AI 超级智能体全栈项目阶段一:AI大模型概述、选型、项目初始化以及基于阿里云灵积模型 Qwen-Plus实现模型接入四种方式(SDK/HTTP/SpringAI/langchain4j)
本文介绍AI大模型的核心概念、分类及开发者学习路径,重点讲解如何选择与接入大模型。项目基于Spring Boot,使用阿里云灵积模型(Qwen-Plus),对比SDK、HTTP、Spring AI和LangChain4j四种接入方式,助力开发者高效构建AI应用。
315 122
AI 超级智能体全栈项目阶段一:AI大模型概述、选型、项目初始化以及基于阿里云灵积模型 Qwen-Plus实现模型接入四种方式(SDK/HTTP/SpringAI/langchain4j)
|
5天前
|
JavaScript Java 大数据
基于JavaWeb的销售管理系统设计系统
本系统基于Java、MySQL、Spring Boot与Vue.js技术,构建高效、可扩展的销售管理平台,实现客户、订单、数据可视化等全流程自动化管理,提升企业运营效率与决策能力。
|
7天前
|
弹性计算 安全 数据安全/隐私保护
2025年阿里云域名备案流程(新手图文详细流程)
本文图文详解阿里云账号注册、服务器租赁、域名购买及备案全流程,涵盖企业实名认证、信息模板创建、域名备案提交与管局审核等关键步骤,助您快速完成网站上线前的准备工作。
256 82
2025年阿里云域名备案流程(新手图文详细流程)