进一步理解IOCP的Get和Post

Enic — Sun, 31 Mar 2013 15:53:00 GMT

// filename: CEIocpSocket.cpp
// author: enic
// date: 2013-03-30

#include "CEIocpSocket.h"

#include "CESpinLock.h"

HANDLE CEIocpSocket::ms_hIoComplete = NULL;

CEIocpSocket::~CEIocpSocket()
{
}

E_DWORD CEIocpSocket::InitIocpWorkThreads()
{
    if (ms_hIoComplete != false)
    {
        return E_ERR;
    }

SYSTEM_INFO nativeSysInfo;
::GetNativeSystemInfo(&nativeSysInfo);

E_SIZE nThreadNum = nativeSysInfo.dwNumberOfProcessors*2;

    ms_hIoComplete = ::CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, NULL, nThreadNum);
    Thread_EIocpSocket_StartParameter startParameter;
    memset(&startParameter, NULL, sizeof(startParameter));
    startParameter.hIoComplete = ms_hIoComplete;

    for (E_SIZE n = 0; n < nThreadNum; ++n)
    {
        boost::thread newThread(
            boost::bind(&CEIocpSocket::Thread_EIocpSocket, startParameter));

        EIOCP_OVERLAPPED* pIocpOverlapped = new EIOCP_OVERLAPPED;
        memset(pIocpOverlapped, NULL, sizeof(*pIocpOverlapped));
        pIocpOverlapped->pEIocpSocket = reinterpret_cast(n);
        // PostQueuedCompletionStatus和GetQueuedCompletionStatus只是传递pIocpOverlapped这地址的值
        // 所以对这个地址的内存管理需要程序员自己实现
        BOOL bResult = PostQueuedCompletionStatus(ms_hIoComplete, n, 0, pIocpOverlapped);
        if (FALSE == bResult)
        {
                std::cout << __FUNCTION__ << std::endl;
                std::cout << "error: " << ::GetLastError() << std::endl;
        }

}

return E_OK;
}

void CEIocpSocket::Thread_EIocpSocket(Thread_EIocpSocket_StartParameter pStartParameter)
{
std::cout << __FUNCTION__ << std::endl;

    DWORD nLenTransferred = 0;
    CEIocpSocket* pEIocpSocket = NULL;
    EIOCP_OVERLAPPED* pIocpOverlapped = NULL;

ULONG_PTR ulKey = NULL;
OVERLAPPED* pOverlapped;

    while(true)
    {
        BOOL bResult =
            GetQueuedCompletionStatus(pStartParameter.hIoComplete,
            &nLenTransferred,
            &ulKey,
            &pOverlapped,
            INFINITE);
        pEIocpSocket = reinterpret_cast(ulKey);
        pIocpOverlapped = static_cast(pOverlapped);
        if(FALSE == bResult)
        {
            DWORD ewLastError = ::GetLastError();
            if(WAIT_TIMEOUT == ewLastError)
            {
                continue;
            }
            else
            {
                std::cout << __FUNCTION__ << std::endl;
                std::cout << "error: " << ewLastError << std::endl;
                return;
            }
        }
        else
        {
            std::cout << nLenTransferred << std::endl;
            std::cout << reinterpret_cast(pIocpOverlapped->pEIocpSocket) << std::endl;
            delete pIocpOverlapped;
        }
    }
}

Enic 2013-03-31 23:53 发表评论

完成端口最最最基本的理解

Enic — Thu, 28 Mar 2013 15:25:00 GMT

// IOCP_Learn.cpp : Defines the entry point for the console application.

#include "stdafx.h"

#include

// 根据某结构体中一个成员的地址，计算出外层结构体的地址

//CONTAINING_RECORD

class CTest_CONTAINING_RECORD

{

public:

void Test()

{

CTest_CONTAINING_RECORD* pTest = CONTAINING_RECORD(&m_i, CTest_CONTAINING_RECORD, m_i);

std::cout << reinterpret_cast(this) << std::endl;

std::cout << reinterpret_cast(pTest) << std::endl;

}

private:

int m_i;

};

//OVERLAPPED

namespace iocp

{

enum EM_IOCP_OP_TYPE

{

emIOCP_POST_SEND = 0,

emIOCP_POST_ASYNC_RECV,

emIOCP_POST_ACCEPT,

emIOCP_DISSCONNECT

};

};//~ end of namespace iocp

// 直接继承OVERLAPPED，根据C++对象的内存布局可以确定stIOCP_OVERLAPPED和static_cast(obj_stIOCP_OVERLAPPED)

// 是一样的

struct stIOCP_OVERLAPPED : public OVERLAPPED

{

WSABUF wsabuf;

EM_IOCP_OP_TYPE optype;

};

// 还有一个是关联的key，没有提到，以及关联socket句柄和完成提示

int _tmain(int argc, _TCHAR* argv[])

{

CTest_CONTAINING_RECORD test;

test.Test();

return 0;

}

Enic 2013-03-28 23:25 发表评论

C++博客-Error-随笔分类-iocp

进一步理解IOCP的Get和Post

完成端口最最最基本的理解