IOCP简单实现代码[转]

IOCP简单实现代码[转]

/* *****************************************************************
*
* 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

/* *****************************************************************
*
* Copyright (c) 2008, xxxxx有限公司
* All rights reserved.
*
* 文件名称：main.cpp
* 摘   要： iocp demo 
*
* 当前版本：1.0
* 作    者：吴会然
* 完成日期：2008-9-16
*
* 取代版本：
* 原  作者：
* 完成日期：
*
***************************************************************** */

include  < iostream >
include  < string >
include  " IOCPHeader.h "
using   namespace  std;

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  =   20055 ;
    
     //  创建完成端口对象
     //  创建工作线程处理完成端口对象的事件
    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 & nb  sp; ==  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 ;
}