Blog @ Blog

當(dāng)華美的葉片落盡，生命的脈絡(luò)才歷歷可見。－－聶魯達(dá)

導(dǎo)航

公告

留言簿(20)

隨筆分類(156)

文章分類(1)

隨筆檔案(198)

文章檔案(2)

相冊(cè)

Meet or get-together

閱讀排行榜

評(píng)論排行榜

常用鏈接

統(tǒng)計(jì)

隨筆 - 198
文章 - 2
評(píng)論 - 156
引用 - 0

積分與排名

積分 - 662726
排名 - 24

BBS

Sohu Club

Blog

Web

CSDN
VC KBase

IOCP簡單實(shí)現(xiàn)代碼[轉(zhuǎn)]

/******************************************************************
*
* Copyright (c) 2008, xxxxx有限公司
* All rights reserved.
*
* 文件名稱：IOCPHeader.h
* 摘   要： IOCP定義文件
*
* 當(dāng)前版本：1.0
* 作    者：吳會(huì)然
* 完成日期：2008-9-16
*
* 取代版本：
* 原  作者：
* 完成日期：
*
******************************************************************/

ifndef _IOCPHEADER_H_20080916_
define _IOCPHEADER_H_20080916_

include <WINSOCK2.H>
include <windows.h>

define BUFFER_SIZE 1024

/******************************************************************
* per_handle 數(shù)據(jù)
*******************************************************************/
typedef struct _PER_HANDLE_DATA
{
    SOCKET      s;      // 對(duì)應(yīng)的套接字句柄
    sockaddr_in addr;   // 對(duì)方的地址

}PER_HANDLE_DATA, *PPER_HANDLE_DATA;

/******************************************************************
* per_io 數(shù)據(jù)
*******************************************************************/
typedef struct _PER_IO_DATA
{
    OVERLAPPED  ol;                 // 重疊結(jié)構(gòu)
    char        buf[BUFFER_SIZE];   // 數(shù)據(jù)緩沖區(qū)
    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
*
* 當(dāng)前版本：1.0
* 作    者：吳會(huì)然
* 完成日期：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;

    // 創(chuàng)建完成端口對(duì)象
    // 創(chuàng)建工作線程處理完成端口對(duì)象的事件
    HANDLE hIocp = ::CreateIoCompletionPort( INVALID_HANDLE_VALUE, 0, 0, 0 );
    ::CreateThread( NULL, 0, ServerThread, (LPVOID)hIocp, 0, 0 );

    // 創(chuàng)建監(jiān)聽套接字，綁定本地端口，開始監(jiān)聽
    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" );

    // 循環(huán)處理到來的請(qǐng)求
    while ( TRUE )
    {
        // 等待接受未決的連接請(qǐng)求
        SOCKADDR_IN saRemote;
        int nRemoteLen = sizeof( saRemote );
        SOCKET sRemote = ::accept( sListen, (sockaddr *)&saRemote, &nRemoteLen );

        // 接受到新連接之后，為它創(chuàng)建一個(gè)per_handle數(shù)據(jù)，并將他們關(guān)聯(lián)到完成端口對(duì)象
        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 );

        // 投遞一個(gè)接受請(qǐng)求
        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;
}

/******************************************************************
* 函數(shù)介紹：處理完成端口對(duì)象事件的線程
* 輸入?yún)?shù)：
* 輸出參數(shù)：
* 返回值  ：
*******************************************************************/
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 )
    {
        // 在關(guān)聯(lián)到此完成端口的所有套接字上等待I/O完成
        BOOL bRet = ::GetQueuedCompletionStatus( hIocp, &dwTrans, (LPDWORD)&pPerHandle, (LPOVERLAPPED*)&pPerIo, WSA_INFINITE );
        if( !bRet )     // 發(fā)生錯(cuò)誤
        {
            ::closesocket( pPerHandle->s );
            ::GlobalFree( pPerHandle );
            ::GlobalFree( pPerIo );

            cout << "error" << endl;
            continue;
        }

        // 套接字被對(duì)方關(guān)閉
        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:       // 完成一個(gè)接收請(qǐng)求
            {
                pPerIo->buf[dwTrans] = '\0';
                printf( "%s\n", pPerIo->buf );

                // 繼續(xù)投遞接受操作
                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 閱讀(1443) 評(píng)論(0) 編輯收藏引用所屬分類: 網(wǎng)絡(luò)編程

只有注冊(cè)用戶登錄后才能發(fā)表評(píng)論。


相關(guān)文章: IOCP簡單實(shí)現(xiàn)代碼[轉(zhuǎn)] 用協(xié)議分析工具學(xué)習(xí)TCP/IP ADO之Command對(duì)象全解[轉(zhuǎn)] ARP協(xié)議詳解及如何編程實(shí)現(xiàn)ARP攻擊[轉(zhuǎn)] TCP/IP 協(xié)議的結(jié)構(gòu)與運(yùn)行原理[圖] TCP頭結(jié)構(gòu) 讓你真正了解局域網(wǎng)技術(shù)概述[轉(zhuǎn)] 實(shí)例ＩＰ　ＵＤＰ頭結(jié)構(gòu)

網(wǎng)站導(dǎo)航: 博客園 IT新聞 BlogJava 博問 Chat2DB 管理

青青草原综合久久大伊人导航_色综合久久天天综合_日日噜噜夜夜狠狠久久丁香五月_热久久这里只有精品