一个使用 VC++ 实现 IOCP的示例源码

源码资源网

一个使用 VC++ 实现 IOCP（输入输出完成端口）的示例源码，该示例使用对话框应用程序。
### 1. 创建对话框应用程序
首先，在 Visual Studio 中创建一个新的 MFC 应用程序，选择“对话框基础”模板。
### 2. 添加必要的头文件
在你的对话框类中，添加以下头文件：

1. #include <winsock2.h>
   
2. 
   
3. #include <mswsock.h>
   
4. 
   
5. #include <windows.h>
   
6. 
   
7. #include <iostream>

复制代码

### 3. 定义必要的结构体和全局变量
在对话框类的头文件中，定义 IOCP 相关的结构体和全局变量：

1. #define MAX_CLIENTS 100
   
2. 
   
3. struct ClientContext {
   
4. 
   
5.     SOCKET socket;
   
6. 
   
7.     OVERLAPPED overlapped;
   
8. 
   
9.     char buffer[1024];
   
10. 
    
11.     WSABUF wsabuf;
    
12. 
    
13. };
    
14. 
    
15. HANDLE hIocp;
    
16. 
    
17. ClientContext clients[MAX_CLIENTS];

复制代码

### 4. 初始化 IOCP
在对话框的  `OnInitDialog`  方法中初始化 IOCP：

1. BOOL CYourDialog::OnInitDialog() {
   
2. 
   
3.     CDialogEx::OnInitDialog();
   
4. 
   
5.      // 初始化 Winsock
   
6. 
   
7.     WSADATA wsaData;
   
8. 
   
9.     WSAStartup(MAKEWORD(2, 2), &wsaData);
   
10. 
    
11.      // 创建 IOCP
    
12. 
    
13.     hIocp = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, 0, 0);
    
14. 
    
15.      // 创建监听 socket
    
16. 
    
17.     SOCKET listenSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    
18. 
    
19.     sockaddr_in serverAddr;
    
20. 
    
21.     serverAddr.sin_family = AF_INET;
    
22. 
    
23.     serverAddr.sin_addr.s_addr = INADDR_ANY;
    
24. 
    
25.     serverAddr.sin_port = htons(8080);
    
26. 
    
27.     bind(listenSocket, (SOCKADDR*)&serverAddr, sizeof(serverAddr));
    
28. 
    
29.     listen(listenSocket, SOMAXCONN);
    
30. 
    
31.      // 将监听 socket 关联到 IOCP
    
32. 
    
33.     CreateIoCompletionPort((HANDLE)listenSocket, hIocp, (ULONG_PTR)listenSocket, 0);
    
34. 
    
35.      // 创建线程处理 IOCP
    
36. 
    
37.     CreateThread(NULL, 0, IocpWorkerThread, NULL, 0, NULL);
    
38. 
    
39.      return TRUE;
    
40. 
    
41. }

复制代码

### 5. IOCP 工作线程
实现 IOCP 工作线程，处理完成的 I/O 操作：

1. DWORD WINAPI IocpWorkerThread(LPVOID lpParam) {
   
2. 
   
3.     DWORD bytesTransferred;
   
4. 
   
5.     ULONG_PTR completionKey;
   
6. 
   
7.     LPOVERLAPPED pOverlapped;
   
8. 
   
9.      while (true) {
   
10. 
    
11.         // 等待 I/O 完成
    
12. 
    
13.         BOOL success = GetQueuedCompletionStatus(hIocp, &bytesTransferred, &completionKey, &pOverlapped, INFINITE);
    
14. 
    
15.         if (success) {
    
16. 
    
17.             if (completionKey == (ULONG_PTR)listenSocket) {
    
18. 
    
19.                 // 处理新的连接
    
20. 
    
21.                 SOCKET clientSocket = accept(listenSocket, NULL, NULL);
    
22. 
    
23.                 CreateIoCompletionPort((HANDLE)clientSocket, hIocp, (ULONG_PTR)clientSocket, 0);
    
24. 
    
25.                 // 继续接收数据
    
26. 
    
27.                 ReceiveData(clientSocket);
    
28. 
    
29.             } else {
    
30. 
    
31.                 // 处理已完成的接收或发送
    
32. 
    
33.                 ClientContext* clientContext = (ClientContext*)completionKey;
    
34. 
    
35.                 // 处理数据...
    
36. 
    
37.             }
    
38. 
    
39.         }
    
40. 
    
41.     }
    
42. 
    
43.     return 0;
    
44. 
    
45. }

复制代码

### 6. 接收数据
实现接收数据的函数：

1. void ReceiveData(SOCKET socket) {
   
2. 
   
3.     ClientContext* clientContext = nullptr;
   
4. 
   
5.      // 查找可用的客户端上下文
   
6. 
   
7.     for (int i = 0; i < MAX_CLIENTS; i++) {
   
8. 
   
9.         if (clients[i].socket == 0) {
   
10. 
    
11.             clientContext = &clients[i];
    
12. 
    
13.             break;
    
14. 
    
15.         }
    
16. 
    
17.     }
    
18. 
    
19.      if (clientContext) {
    
20. 
    
21.         clientContext->socket = socket;
    
22. 
    
23.         clientContext->wsabuf.len = sizeof(clientContext->buffer);
    
24. 
    
25.         clientContext->wsabuf.buf = clientContext->buffer;
    
26. 
    
27.          // 提交接收请求
    
28. 
    
29.         DWORD flags = 0;
    
30. 
    
31.         WSARecv(socket, &clientContext->wsabuf, 1, NULL, &flags, &clientContext->overlapped, NULL);
    
32. 
    
33.     }
    
34. 
    
35. }

复制代码

### 7. 清理资源
在对话框销毁时，清理资源：

1. void CYourDialog::OnDestroy() {
   
2. 
   
3.     CDialogEx::OnDestroy();
   
4. 
   
5.      // 关闭 IOCP 和 Winsock
   
6. 
   
7.     CloseHandle(hIocp);
   
8. 
   
9.     WSACleanup();
   
10. 
    
11. }

复制代码

### 8. 编译和运行
确保在项目中链接  `Ws2_32.lib`  库，然后编译并运行该程序。
### 总结
以上是一个简单的 VC++ IOCP 示例，使用对话框应用程序实现了基本的网络连接和数据接收功能。你可以根据需要扩展这个示例，添加更多的功能，比如数据处理、错误处理等。