//IOCP代码
#pragma comment(lib, "ws2_32.lib")
#include <WinSock2.h>
#include <stdio.h>
#include <iostream>
using namespace std;
#define DATA_BUFSIZE 1024SOCKET ListenSocket;struct PerIOcontext
{OVERLAPPED m_Overlapped;SOCKET m_sockAccept;WSABUF m_wsaBuf;char buffer[DATA_BUFSIZE];
};DWORD WINAPI AcceptThread(LPVOID lpParameter)
{WSADATA wsaData;WSAStartup(MAKEWORD(2,2), &wsaData);ListenSocket = WSASocket(AF_INET, SOCK_STREAM, IPPROTO_TCP, NULL, NULL, WSA_FLAG_OVERLAPPED);SOCKADDR_IN ServerAddr;ServerAddr.sin_family = AF_INET;ServerAddr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);ServerAddr.sin_port = htons(1234);bind(ListenSocket, (LPSOCKADDR)&ServerAddr, sizeof(ServerAddr));listen(ListenSocket, 100);printf("listenning...\n");int i = 0;SOCKADDR_IN ClientAddr;int addr_length = sizeof(ClientAddr);HANDLE completionPort = (HANDLE)lpParameter;while(TRUE){PerIOcontext* perIOcontext = (PerIOcontext*)GlobalAlloc(GPTR, sizeof(PerIOcontext));SOCKET acceptSocket;SOCKADDR_IN acceptAddr;int len = sizeof(acceptAddr);acceptSocket = accept(ListenSocket, (SOCKADDR*)&acceptAddr, &len);printf("接受到客户端连接");if(SOCKET_ERROR == perIOcontext->m_sockAccept){   // 接收客户端失败  cerr << "Accept Socket Error: " << GetLastError() << endl;  system("pause");  return -1;  } perIOcontext->m_wsaBuf.buf = perIOcontext->buffer;perIOcontext->m_wsaBuf.len = 1024;perIOcontext->m_sockAccept = acceptSocket;CreateIoCompletionPort((HANDLE)(perIOcontext->m_sockAccept), completionPort, (DWORD)perIOcontext, 0);DWORD RecvBytes;  DWORD Flags = 0;ZeroMemory(&(perIOcontext->m_Overlapped), sizeof(OVERLAPPED));WSARecv(perIOcontext->m_sockAccept, &(perIOcontext->m_wsaBuf), 1, &RecvBytes, &Flags, &(perIOcontext->m_Overlapped), NULL);}return FALSE;
}DWORD WINAPI ReceiveThread(LPVOID lpParameter)
{HANDLE completionPort = (HANDLE)lpParameter;DWORD BytesTransferred;PerIOcontext* perIOcontext;LPOVERLAPPED IpOverlapped = NULL;while(true){printf("Receive线程进入等待\n");BOOL ret = GetQueuedCompletionStatus(completionPort, &BytesTransferred, (PULONG_PTR)&perIOcontext, &IpOverlapped, INFINITE);printf("Receive线程退出等待\n");if (BytesTransferred == 0){printf("获得字节为0，disconnect\n");}printf("客户端：%s\n", perIOcontext->buffer);memset(perIOcontext->buffer, 0, DATA_BUFSIZE);DWORD RecvBytes;  DWORD Flags = 0;system("pause");WSARecv(perIOcontext->m_sockAccept, &(perIOcontext->m_wsaBuf), 1, &RecvBytes, &Flags, &(perIOcontext->m_Overlapped), NULL);}return FALSE;
}int main()
{HANDLE completionPort = CreateIoCompletionPort( INVALID_HANDLE_VALUE, NULL, 0, 0);  if (NULL == completionPort){    // 创建IO内核对象失败  cout << "CreateIoCompletionPort failed. Error:" << GetLastError() << endl;  system("pause");  return 0;  }HANDLE hThreads[2];hThreads[0] = CreateThread(NULL, 0, AcceptThread, completionPort, NULL, NULL);hThreads[1] = CreateThread(NULL, 0, ReceiveThread, completionPort, NULL, NULL);WaitForMultipleObjects(2, hThreads, TRUE, INFINITE);printf("exit\n");CloseHandle(hThreads[0]);CloseHandle(hThreads[1]);return 0;
}