65,210
社区成员
发帖
与我相关
我的任务
分享
/******************************************************************
*
* Copyright (c) 2008, xxxxx有限公司
* All rights reserved.
*
* 文件名称:IOCPHeader.h
* 摘 要: IOCP定义文件
*
* 当前版本:1.0
* 作 者:吴会然
* 完成日期:2008-9-16
*
* 取代版本:
* 原 作者:
* 完成日期:
*
******************************************************************/
#ifndef _IOCPHEADER_H_20080916_
#define _IOCPHEADER_H_20080916_
#include <WINSOCK2.H>
#include <windows.h>
#define BUFFER_SIZE 1024
/******************************************************************
* per_handle 数据
*******************************************************************/
typedef struct _PER_HANDLE_DATA
{
SOCKET s; // 对应的套接字句柄
sockaddr_in addr; // 对方的地址
}PER_HANDLE_DATA, *PPER_HANDLE_DATA;
/******************************************************************
* per_io 数据
*******************************************************************/
typedef struct _PER_IO_DATA
{
OVERLAPPED ol; // 重叠结构
char buf[BUFFER_SIZE]; // 数据缓冲区
int nOperationType; // 操作类型
#define OP_READ 1
#define OP_WRITE 2
#define OP_ACCEPT 3
}PER_IO_DATA, *PPER_IO_DATA;
#endif
#include <iostream>
#include <string>
#include "IOCPHeader.h"
using namespace std;
#include <winsock2.h>
#include <stdio.h>
#include <windows.h>
#pragma comment(lib,"ws2_32.lib")
DWORD WINAPI ServerThread(LPVOID lpParam);
int main(int argc, char *argv[])
{
//////////////////////////////////////////////////////////////////////////
WSADATA wsaData;
if (0 != WSAStartup(MAKEWORD(2, 2), &wsaData))
{
printf("Using %s (Status:%s)\n", wsaData.szDescription, wsaData.szSystemStatus);
printf("with API versions: %d.%d to %d.%d",
LOBYTE(wsaData.wVersion), HIBYTE(wsaData.wVersion),
LOBYTE(wsaData.wHighVersion), HIBYTE(wsaData.wHighVersion));
return -1;
}
else
{
printf("Windows sockets 2.2 startup\n");
}
//////////////////////////////////////////////////////////////////////////
int nPort = 5555;
// 创建完成端口对象
// 创建工作线程处理完成端口对象的事件
HANDLE hIocp = ::CreateIoCompletionPort(INVALID_HANDLE_VALUE, 0, 0, 0);
::CreateThread(NULL, 0, ServerThread, (LPVOID)hIocp, 0, 0);
// 创建监听套接字,绑定本地端口,开始监听
SOCKET sListen = ::socket(AF_INET,SOCK_STREAM, 0);
SOCKADDR_IN addr;
addr.sin_family = AF_INET;
addr.sin_port = ::htons(nPort);
addr.sin_addr.S_un.S_addr = INADDR_ANY;
::bind(sListen, (sockaddr *)&addr, sizeof(addr));
::listen(sListen, 5);
printf("iocp demo start......\n");
// 循环处理到来的请求
while (TRUE)
{
// 等待接受未决的连接请求
SOCKADDR_IN saRemote;
int nRemoteLen = sizeof(saRemote);
SOCKET sRemote = ::accept(sListen, (sockaddr *)&saRemote, &nRemoteLen);
// 接受到新连接之后,为它创建一个per_handle数据,并将他们关联到完成端口对象
PPER_HANDLE_DATA pPerHandle = (PPER_HANDLE_DATA)::GlobalAlloc(GPTR, sizeof(PPER_HANDLE_DATA));
if (pPerHandle == NULL)
{
break;
}
pPerHandle->s = sRemote;
memcpy(&pPerHandle->addr, &saRemote, nRemoteLen);
::CreateIoCompletionPort((HANDLE)pPerHandle->s, hIocp, (DWORD)pPerHandle, 0);
// 投递一个接受请求
PPER_IO_DATA pIoData = (PPER_IO_DATA)::GlobalAlloc(GPTR, sizeof(PPER_IO_DATA));
if (pIoData == NULL)
{
break;
}
pIoData->nOperationType = OP_READ;
WSABUF buf;
buf.buf = pIoData->buf;
buf.len = BUFFER_SIZE;
DWORD dwRecv = 0;
DWORD dwFlags = 0;
::WSARecv(pPerHandle->s, &buf, 1, &dwRecv, &dwFlags, &pIoData->ol, NULL);
}
//////////////////////////////////////////////////////////////////////////
ERROR_PROC:
WSACleanup();
//////////////////////////////////////////////////////////////////////////
return 0;
}
/******************************************************************
* 函数介绍:处理完成端口对象事件的线程
* 输入参数:
* 输出参数:
* 返回值 :
*******************************************************************/
DWORD WINAPI ServerThread(LPVOID lpParam)
{
HANDLE hIocp = (HANDLE)lpParam;
if (hIocp == NULL)
{
return -1;
}
DWORD dwTrans = 0;
PPER_HANDLE_DATA pPerHandle;
PPER_IO_DATA pPerIo;
while (TRUE)
{
// 在关联到此完成端口的所有套接字上等待I/O完成
BOOL bRet = ::GetQueuedCompletionStatus(hIocp, &dwTrans, (LPDWORD)&pPerHandle, (LPOVERLAPPED*)&pPerIo, WSA_INFINITE);
if (!bRet) // 发生错误
{
::closesocket(pPerHandle->s);
::GlobalFree(pPerHandle);
::GlobalFree(pPerIo);
cout << "error" << endl;
continue;
}
// 套接字被对方关闭
if (dwTrans == 0 && (pPerIo->nOperationType == OP_READ || pPerIo->nOperationType == OP_WRITE))
{
::closesocket(pPerHandle->s);
::GlobalFree(pPerHandle);
::GlobalFree(pPerIo);
cout << "client closed" << endl;
continue;
}
switch (pPerIo->nOperationType)
{
case OP_READ: // 完成一个接收请求
{
pPerIo->buf[dwTrans] = '\0';
printf("%s\n", pPerIo->buf);
// 继续投递接受操作
WSABUF buf;
buf.buf = pPerIo->buf;
buf.len = BUFFER_SIZE;
pPerIo->nOperationType = OP_READ;
DWORD dwRecv = 0;
DWORD dwFlags = 0;
::WSARecv(pPerHandle->s, &buf, 1, &dwRecv, &dwFlags, &pPerIo->ol, NULL); //******报错
}
break;
case OP_WRITE:
case OP_ACCEPT:
break;
}
}
return 0;
}