山寨:不是最好的,是最适合我们的!欢迎体验山寨 中文版MSDN

Blog @ Blog

当华美的叶片落尽,生命的脉络才历历可见。 -- 聂鲁达

常用链接

统计

积分与排名

BBS

Blog

Web

最新评论

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;
    
    
if0 != WSAStartup( MAKEWORD( 22 ), &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, 000 );
    ::CreateThread( NULL, 
0, ServerThread, (LPVOID)hIocp, 00 );
    
    
// 创建监听套接字,绑定本地端口,开始监听
    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
->= 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;
}

posted on 2009-05-10 16:37 isabc 阅读(1222) 评论(0)  编辑 收藏 引用 所属分类: 网络编程


只有注册用户登录后才能发表评论。
【推荐】超50万行VC++源码: 大型组态工控、电力仿真CAD与GIS源码库
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理


广告信息(免费广告联系)

中文版MSDN:
欢迎体验