windows下简单的IOCP模型迭代回声服务器实例

#include<iostream>
#include<WinSock2.h>
#include<Windows.h>
#include<process.h>
#pragma comment(lib,"ws2_32.lib")
#define BUF_SIZE 100
#define READ  3
#define WRITE 5
using namespace std;
typedef struct {    //客户端信息
SOCKET hclientSocket;
SOCKADDR_IN clientAddr;
}PER_HANDLE_DATA, *LPPER_HANDLE_DATA;
int k, l, m, n;
typedef struct {
OVERLAPPED overlapped;  //I/O缓冲和重叠I/O都会使用  事件对象有两种状态  signaled:完成IO  unsignaled: 未完成
WSABUF wsaBuf;
char buffer[BUF_SIZE];  //数据缓冲区
int rwMode;    // 操作类型：读或写
}PER_IO_DATA,*LPPER_IO_DATA;

unsigned WINAPI EchoThreadMain(LPVOID CompletionPortIO);
void main() {
WSADATA wsadata;
HANDLE hComPort;
SYSTEM_INFO sysInfo;
LPPER_IO_DATA ioInfo;
LPPER_HANDLE_DATA handleInfo;
SOCKET serverSocket;
SOCKADDR_IN serverAddr;
unsigned int  i;
DWORD recvBytes,flags = 0;

if(WSAStartup(MAKEWORD(2, 2), &wsadata)!=0)
cout<<"WSAStartup() error"<<endl;
//创建cp对象（第二个参数为0，所以是新建的一个iocp），最后一个参数为0 创建和CPU核数相同的线程 返回NULL即为创建失败
hComPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, 0, 0);
GetSystemInfo(&sysInfo);  //获取系统信息
for (i = 0; i < sysInfo.dwNumberOfProcessors; i++) //根据核数创建线程并传递给cp对象句柄
{        // 创建安全线程
_beginthreadex(NULL, 0 ,EchoThreadMain, (LPVOID)hComPort, 0, NULL);
}
//最后一个参数是：创建重叠IO非阻塞模式的TCP套接字
serverSocket = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, WSA_FLAG_OVERLAPPED);
if (serverSocket == INVALID_SOCKET)
cout << "socket()  error" << endl;

memset(&serverAddr, 0, sizeof(serverAddr));
serverAddr.sin_family = AF_INET;
serverAddr.sin_addr.s_addr = htonl(INADDR_ANY);
serverAddr.sin_port = htons(8000);

if (bind(serverSocket, (SOCKADDR*)&serverAddr, sizeof(serverAddr)) == SOCKET_ERROR)
cout << "bind () error" << endl;

listen(serverSocket, 5);
cout << "服务器启动成功！" << endl;

while (1) {
SOCKET clientSocket;
SOCKADDR_IN clientAddr;
int addrLen = sizeof(clientAddr);
clientSocket = accept(serverSocket, (SOCKADDR*)&clientAddr, &addrLen);
if (clientSocket != INVALID_SOCKET) cout << "已有客户端连接：" << clientSocket << endl;

handleInfo = (LPPER_HANDLE_DATA)malloc(sizeof(PER_HANDLE_DATA));//开辟地址创建客户信息句柄
handleInfo->hclientSocket = clientSocket;   //将accep函数的客户端套接字赋给句柄
memcpy(&(handleInfo->clientAddr), &clientAddr, addrLen);  //拷贝地址赋给句柄
//将accept传来的客户套接字和客户端地址和已存在的cp对象绑定，最后一个参数为0表示允许和CPU核数相同的线程访问该函数
CreateIoCompletionPort((HANDLE)clientSocket, hComPort, (DWORD)handleInfo, 0);
ioInfo = (LPPER_IO_DATA)malloc(sizeof(PER_IO_DATA));  //创建事件对象等捆绑信息的句柄
memset((&ioInfo->overlapped), 0, sizeof(OVERLAPPED));
ioInfo->wsaBuf.len = BUF_SIZE;
ioInfo->wsaBuf.buf = ioInfo->buffer;
ioInfo->rwMode = READ;
WSARecv(handleInfo->hclientSocket, &(ioInfo->wsaBuf),1, &recvBytes, &flags, &(ioInfo->overlapped), NULL);
//参数：1重叠IO套接字 2保存接受信息的结构体数组 3第二个参数数组的长度     4保存接受消息大小的变量地址 5设置或读取传输特性的消息
//        6事件对象的状态    7Routine函数地址
}
}

unsigned WINAPI EchoThreadMain(LPVOID  pComPort)
{
cout << "线程 NO.  " << GetCurrentThreadId() << endl;
cout << "EchoThreadMain() 初始化" << endl;
HANDLE hComPort = (HANDLE)pComPort;
SOCKET sock;
DWORD bytesTrans;
LPPER_HANDLE_DATA handleInfo;
LPPER_IO_DATA ioInfo;
DWORD flags = 0;
while (1) {
//在操作队列中获取IO完成的客户端信息    为什么第四个参数是ioInfo？因为ioInfo的地址和它第一个成员变量overlapped的地址相同
GetQueuedCompletionStatus(hComPort, &bytesTrans, (PULONG_PTR)&handleInfo,(LPOVERLAPPED*)&ioInfo, INFINITE);
cout << "获取完队列的信息" << endl;
cout << "客户端地址" << handleInfo->hclientSocket << endl;
cout << "线程 NO.  " << GetCurrentThreadId() << endl;
sock = handleInfo->hclientSocket;
if (ioInfo->rwMode == READ) { //判断客户端是读还是写
cout << "message received!" << endl;
if (bytesTrans == 0) {  //传输EOF时
closesocket(sock);
free(handleInfo);
free(ioInfo);
cout << "客户端关闭：" << sock << endl;
continue;
}
memset(&(ioInfo->overlapped), 0, sizeof(OVERLAPPED)); //初始化时间对象的状态
ioInfo->wsaBuf.len = bytesTrans;
ioInfo->rwMode = WRITE;
//客户端socket 缓存数组地址  数组长度  保存实际字节数变量的地址  数据传输特性   时间状态  Routine函数
WSASend(sock, &(ioInfo->wsaBuf), 1, NULL, 0, &(ioInfo->overlapped), NULL);
cout << "发送完数据" << endl;
ioInfo = (LPPER_IO_DATA)malloc(sizeof(PER_IO_DATA));
memset(&(ioInfo->overlapped), 0, sizeof(OVERLAPPED));
ioInfo->wsaBuf.len = BUF_SIZE;
ioInfo->wsaBuf.buf = ioInfo->buffer;
ioInfo->rwMode = READ;
WSARecv(sock, &(ioInfo->wsaBuf), 1, NULL, &flags, &(ioInfo->overlapped), NULL);
cout << "接受完数据" << endl;
}
else {
cout << "message sent!" << endl;
free(ioInfo);
}
}
return 0;
}