IOCP完成端口简单Demo

我会带着你远行 2021-11-09 10:04 294阅读 0赞

//  
///author:Jeson Yang  
///date:2014.09.28  
//  
\#include <WinSock2.h>  
\#include <tchar.h>  
\#include <stdio.h>  
\#pragma comment(lib, "ws2\_32.lib")

\#define TCP\_PORT 10066  
\#define MAX\_BUFF\_LENGTH 1024  
typedef struct PER\_IO\_OPERATION\_DATA  
\{  
WSAOVERLAPPED overlap;

WSABUF buf;  
char szMessage\[MAX\_BUFF\_LENGTH\];  
DWORD numberOfBytesRecved;  
DWORD flag;  
\}PER\_IO\_OPERATION\_DATA,\*LPPER\_IO\_OPERATION\_DATA;

DWORD WINAPI WorkerThread(LPVOID CompletionPortId);  
void main()  
\{  
WORD dwVersion = MAKEWORD(2, 2);  
WSAData wsaData;  
WSAStartup(dwVersion, &wsaData);  
  
//初始化完成端口  
HANDLE completionport = CreateIoCompletionPort(INVALID\_HANDLE\_VALUE, NULL, 0, 0);  
  
SYSTEM\_INFO sysInfo;  
DWORD threadId;  
::GetSystemInfo(&sysInfo);  
for (int i = 0; i < (int)sysInfo.dwNumberOfProcessors; i++)  
\{  
CreateThread(NULL, 0, WorkerThread, completionport, 0, &threadId);  
\}

//创建套接字  
SOCKET sClient;  
SOCKET sScok = socket(AF\_INET, SOCK\_STREAM, IPPROTO\_TCP);

SOCKADDR\_IN localAddr, clientAddr;  
::memset(&localAddr, 0, sizeof(SOCKADDR\_IN));  
  
//绑定套接字  
localAddr.sin\_family = AF\_INET;  
localAddr.sin\_addr.s\_addr = htonl(INADDR\_ANY);//inet\_addr("127.0.0.1");  
localAddr.sin\_port = htons(TCP\_PORT);  
if (::bind(sScok, (SOCKADDR\*)&localAddr, sizeof(sockaddr)))  
\{  
::OutputDebugString(\_T("bind error!"));  
return;  
\}  
  
//开始监听  
::listen(sScok, 1);  
int iaddrSize = sizeof(SOCKADDR\_IN);//此处应该先赋值，或者在accept出填NULL，如果此处为负数，accept将不会死锁在那里。  
PER\_IO\_OPERATION\_DATA \*lpPerData = NULL;  
while (true)  
\{  
//如果有客户请求连接，则继续，否则等待  
sClient = ::accept(sScok, (sockaddr\*)&clientAddr, &iaddrSize);//NULL);

//client中保存用户信息  
printf("Accepted client: %s: %d\\n", inet\_ntoa(clientAddr.sin\_addr), ntohs(clientAddr.sin\_port));  
  
//将这个最新到来的客户套接字和完成端口绑定到一起  
//第三个参数表示传递的参数，这里就传递的客户套接字地址。最后一个参数为0 表示有和CPU一样的进程数。即1个CPU一个线程  
CreateIoCompletionPort((HANDLE)sClient, completionport, (DWORD)sClient, 0);

lpPerData = (LPPER\_IO\_OPERATION\_DATA)::HeapAlloc(::GetProcessHeap(),  
HEAP\_ZERO\_MEMORY,  
sizeof(PER\_IO\_OPERATION\_DATA));  
lpPerData->buf.len = MAX\_BUFF\_LENGTH;  
lpPerData->buf.buf = lpPerData->szMessage;

//异步接收消息，立刻返回。  
::WSARecv(sClient, &lpPerData->buf, 1, &lpPerData->numberOfBytesRecved, &lpPerData->flag,&lpPerData->overlap, NULL);  
\}  
::PostQueuedCompletionStatus(completionport, 0xFFFFFFFF, 0, NULL);  
::CloseHandle(completionport);  
closesocket(sScok);  
WSACleanup();  
\}

//工作者线程有一个参数，是指向完成端口的句柄  
DWORD WINAPI WorkerThread(LPVOID CompletionPortId)  
\{  
HANDLE completionPort = (HANDLE)CompletionPortId;  
DWORD dwByteTransferred;  
SOCKET sClient;  
LPPER\_IO\_OPERATION\_DATA lpPerIOData = NULL;  
while(true)  
\{  
::GetQueuedCompletionStatus(completionPort,  
&dwByteTransferred,  
(PULONG\_PTR)&sClient,  
(LPOVERLAPPED \*)&lpPerIOData,  
INFINITE);  
if (dwByteTransferred == 0xFFFFFFFF)  
\{  
return 0;  
\}

if (dwByteTransferred == 0)  
\{  
::closesocket(sClient);  
::HeapFree(::GetProcessHeap(), HEAP\_ZERO\_MEMORY, lpPerIOData);//释放结构体  
\}  
else  
\{  
lpPerIOData->szMessage\[dwByteTransferred\] = '\\0';  
::send(sClient, lpPerIOData->szMessage, dwByteTransferred, 0);//将接收到的消息返回

memset(lpPerIOData, 0, sizeof(PER\_IO\_OPERATION\_DATA));  
lpPerIOData->buf.len = MAX\_BUFF\_LENGTH;  
lpPerIOData->buf.buf = lpPerIOData->szMessage;  
//异步接收消息，立刻返回。  
::WSARecv(sClient, &lpPerIOData->buf, 1, &lpPerIOData->numberOfBytesRecved, &lpPerIOData->flag,&lpPerIOData->overlap, NULL);  
\}  
\}

return 0;  
\}

转载于:https://www.cnblogs.com/JesonYang/p/6124133.html