自己写的一个服务端IOCP例子。希望大家指点下

C/C++ code

// IocpModel.cpp : 定义控制台应用程序的入口点。//#include "stdafx.h"#include "IocpModel.h"#define Op_READ 1#define Op_WRITE 2WSADATA wsa;HANDLE CompletionPort;SYSTEM_INFO Systeminfo;SOCKADDR_IN InternetAddr;HANDLE ThreadHanle;void ChangeHandleData(LPPER_IO_DATA lpPerIoContext,int state){/**    Fucntion Description:        根据传进来的state，来进行改变下一步的IO操作    Parameter：        lpPerIoContext:上一个IO操作的结果        state           上一个IO操作的状态    Return Value:        返回值为空**/    if(OP_READ == state)    {        //完成了读取客户端数据，并且返回数据        lpPerIoContext->OperationType = OP_WRITE;        ZeroMemory(&(lpPerIoContext->Overlapped),sizeof(OVERLAPPED));        //返回数据给客户端        char *sendClientData = "sendClientData";        lpPerIoContext->DataBuf.buf = sendClientData;        lpPerIoContext->DataBuf.len = BUFFER_SIZE;    }    if(OP_WRITE == state)    {            //上一个IO数据发送客户端完成，结束就用 OP_END        //继续接受，叫类型设置为OP_READ,注意清空下 buff        //继续发送，将数据设置为OP_WRITE        lpPerIoContext->OperationType = OP_READ;        lpPerIoContext->DataBuf.buf = lpPerIoContext->buffer;        lpPerIoContext->DataBuf.len = BUFFER_SIZE;    }    if(OP_ACCEPT == state)    {        //立起来的连接，把句柄设为读类型        lpPerIoContext->OperationType = OP_READ;        ZeroMemory(&(lpPerIoContext->Overlapped),sizeof(OVERLAPPED));        ZeroMemory(&(lpPerIoContext->DataBuf),sizeof(lpPerIoContext->DataBuf));        lpPerIoContext->DataBuf.buf = lpPerIoContext->buffer;        lpPerIoContext->DataBuf.len = BUFFER_SIZE;        cout << "数据包操作类型被改为 : OP_READ"<<endl;    }}void SendHandleData(LPPER_IO_DATA lpPerIoContext,LPPER_HANDLE_DATA perHandData){    /**        根据lpPerIoContext对象OperationType来进行下一步的操作    **/    DWORD dwIosize = 0;    int nResult  =0;    DWORD RecvBytes = 0;    DWORD nFlag = 0;    if(lpPerIoContext->OperationType  == OP_WRITE)    {        /*发送数据到客户端*/        nResult = WSASend(perHandData->Socket,&(lpPerIoContext->DataBuf),1,&dwIosize,0,&(lpPerIoContext->Overlapped),NULL);        if((nResult == SOCKET_ERROR) && WSAGetLastError() != ERROR_IO_PENDING )        {            cout<< "WSASend error :"<<WSAGetLastError()<<endl;            ::closesocket(perHandData->Socket);            return;        }    }    else if(lpPerIoContext->OperationType == OP_READ)    {        //清空，准备下个I/O数据        ZeroMemory(&(lpPerIoContext->Overlapped),sizeof(OVERLAPPED));        //注意：这里清空下buffer，要不然数据会跟上一条粘在一起        ZeroMemory(&(lpPerIoContext->buffer),sizeof(lpPerIoContext->buffer));        lpPerIoContext->DataBuf.len = BUFFER_SIZE;        if(SOCKET_ERROR == WSARecv(perHandData->Socket, &(lpPerIoContext->DataBuf), 1,&RecvBytes,&nFlag,&(lpPerIoContext->Overlapped),NULL)){            cout << "WSARecv() failed: " << WSAGetLastError() << endl;            return;        }    }    else if(lpPerIoContext->OperationType == OP_END)    {        ::closesocket(perHandData->Socket);    }}int main(){    SOCKET ListenSock;    SOCKET accpSocket;    PER_HANDLE_DATA *PerHandleData = NULL;    SOCKADDR_IN saRemote;    int RemoteLen;    cout<<"Server start......"<<endl;    WSAStartup(MAKEWORD(2,2),&wsa);    GetSystemInfo(&Systeminfo);    //创建一个IO完成端口    CompletionPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE,NULL,0,0);    if ( CompletionPort == INVALID_HANDLE_VALUE )     {        cout<<"完成端口创建失败!!!!"<<endl;        return 0;    }    //创建IOCP工作线程(根据CPU个数 * 2)    for(int i = 0;i< Systeminfo.dwNumberOfProcessors * 2 ; ++ i)    {        //创建线程        ThreadHanle = CreateThread(NULL,0,ServerWorkerThread,CompletionPort,0,NULL);        CloseHandle(ThreadHanle);    }    //4. 创建一个监听套接字，    ListenSock = WSASocket(AF_INET,SOCK_STREAM,0,NULL,0,WSA_FLAG_OVERLAPPED);    if(ListenSock == INVALID_SOCKET)    {        cout<<"套接字创建 失败!!"<<endl;        return 0;    }    InternetAddr.sin_family = AF_INET;    InternetAddr.sin_port = htons(5000);    InternetAddr.sin_addr.s_addr = inet_addr("127.0.0.1");    if(SOCKET_ERROR == bind(ListenSock,(SOCKADDR*)&InternetAddr,sizeof(InternetAddr)))    {        cout<<"服务器绑定地址信息 失败!! errid:"<<GetLastError()<<endl;        closesocket(ListenSock);        return 0;    }    int ret = listen(ListenSock, 5);    if (ret == SOCKET_ERROR)    {        cout<<"监听套接字失败!! errid:"<<GetLastError()<<endl;        closesocket(ListenSock);        return 0;    }    while(TRUE)    {        //接受连接        RemoteLen = sizeof(saRemote);        //accpSocket = WSAAccept(ListenSock,NULL,NULL,NULL,0);        accpSocket = accept(ListenSock, (SOCKADDR*)&saRemote, &RemoteLen);        if(SOCKET_ERROR == accpSocket)        {            cout<<"接受套接字错误!! errid:"<<GetLastError()<<endl;            closesocket(accpSocket);            return 0;        }        //创建用来和套接字关联句柄信息结构        PerHandleData =(LPPER_HANDLE_DATA)GlobalAlloc(GPTR,sizeof(PER_HANDLE_DATA));        if(PerHandleData == NULL)        {            cout<<"PerHandleData = null!! errid:"<<GetLastError()<<endl;            closesocket(accpSocket);            return 0;        }        cout<< "Socker number: "<< accpSocket << " connected" << endl;        PerHandleData->Socket = accpSocket;        memcpy(&PerHandleData->ClientAddr,&saRemote,RemoteLen);        //将接受的套接字关联到 完成端口        if(CreateIoCompletionPort((HANDLE)accpSocket,CompletionPort,(DWORD)PerHandleData,0) == NULL)        {            cout<<"create complertion port error! errid:"<<GetLastError()<<endl;            return 0;        }        LPPER_IO_DATA PerIoData = (LPPER_IO_DATA)GlobalAlloc(GPTR,sizeof(PER_IO_DATA));;        ChangeHandleData(PerIoData,OP_ACCEPT);        SendHandleData(PerIoData,PerHandleData);    }    return 0;}//创建线程函数DWORD WINAPI ServerWorkerThread(LPVOID lpParam){    HANDLE CompletionPort = (HANDLE)lpParam;    DWORD ByresTransferred;    LPOVERLAPPED lpOverlapped = NULL;    LPPER_HANDLE_DATA PerHandleData = NULL;    //LPPER_IO_DATA PerHandleData = NULL;    LPPER_IO_DATA PerIoData = NULL;    DWORD SendBytes;    DWORD RecvBytes = 0;    DWORD Flags = 0;    BOOL bRet = FALSE;    while (TRUE)    {        bRet = GetQueuedCompletionStatus(CompletionPort,                                         &ByresTransferred,                    //的I/O操作所传送数据的字节数                                         (PULONG_PTR)&PerHandleData,        //用于存放与之关联的Completion键                                         (LPOVERLAPPED*)&lpOverlapped,                                          INFINITE);        if(!bRet)        {            closesocket(PerHandleData->Socket);            ::GlobalFree(PerHandleData);            ::GlobalFree(PerIoData);            ::GlobalFree(lpOverlapped);            cout<< "失去客户端连接！！！！" << bRet <<endl;            return 0;        }        PerIoData = (LPPER_IO_DATA)lpOverlapped;        //先检查一下,看看是否套接字有错误发生        if(0 == ByresTransferred)        {            if(SOCKET_ERROR  == closesocket(PerHandleData->Socket)){                cout<< "ByresTransferred SOCKET_ERROR err:" << GetLastError() << endl;            }            ::GlobalFree(PerHandleData);            ::GlobalFree(PerIoData);            cout<< "发生错误！ error" <<endl;            continue;        }                //处理操作        switch(PerIoData->OperationType)        {        case OP_ACCEPT:            if(ByresTransferred)            {                //第一次连接接受到的数据                ChangeHandleData(PerIoData,OP_ACCEPT);                SendHandleData(PerIoData,PerHandleData);            }            else            {                //连接成功，数据没接受到，继续接受                ChangeHandleData(PerIoData,OP_ACCEPT);                SendHandleData(PerIoData,PerHandleData);            }            break;        case OP_READ:            cout<<"-----------------reader----------------------"<< endl;            cout<<"传送数据的字节数:"<< ByresTransferred  <<endl;            cout<<"接受数据为："<<PerIoData->DataBuf.buf<< endl;            cout<<"---------------------------------------------"<< endl;            ChangeHandleData(PerIoData,OP_READ);            SendHandleData(PerIoData,PerHandleData);            break;        case OP_WRITE:            ChangeHandleData(PerIoData,OP_WRITE);            SendHandleData(PerIoData,PerHandleData);            break;        default:            //其他操作            break;                    }            }    return 0 ;}