WinCE下 WSAWaitForMultipleEvents 事件模型实现并发IO

直接贴代码

void CListenAction::Init()
{
    // 初始化套接字库
    if (!InitSocketLib())
    {
       TRACE(_T("初始化套接字库失败"));
        return ;
    }

    // 创建服务器端监听套接字
    SOCKET listenSocket;
    listenSocket = socket(AF_INET, SOCK_STREAM, 0);
    if (INVALID_SOCKET == listenSocket)
    {
        TRACE(_T("socket error: %d\n"),WSAGetLastError());
        return;
    }

    // 本地地址
    SOCKADDR_IN localAddr;
    localAddr.sin_family = AF_INET;

    // 获取本机IP地址
    char szIPBuf[100];
    memset(szIPBuf, 0, 100);
    CCommUtility::GetLocalIP(szIPBuf, sizeof(szIPBuf));
    localAddr.sin_addr.S_un.S_addr = inet_addr(szIPBuf);

    // 设置端口号
    localAddr.sin_port = htons(SERVER_PORT);

    INT iRetTmp = bind(listenSocket, (SOCKADDR *)&localAddr, sizeof(localAddr));
    if (SOCKET_ERROR == iRetTmp)
    {
        TRACE(_T("bind error: %d\n"),WSAGetLastError());
        return;
    }

    iRetTmp = listen(listenSocket, LISTEN_QUEUE_LENGTH);
    if (SOCKET_ERROR == iRetTmp)
    {
        TRACE(_T("listen error: %d\n"),WSAGetLastError());
        return;
    }

    // 添加监听主套接字到通信对象数组
    SOCKET_OBJ socketObj;
    socketObj.sd = listenSocket;
    socketObj.emConnectType = LISTEN;
    socketObj.emSocketType = TCP_SOCKET;
    m_arraySocket.Add(socketObj);

    // 添加监听事件到事件数组 关联事件
    WSAEVENT listenEvent = WSACreateEvent();
    int iRet = WSAEventSelect(listenSocket, listenEvent, FD_ACCEPT | FD_CLOSE);
    if (iRet != 0)
   {
       TRACE(_T("WSAEventSelect failed: %d\n"),WSAGetLastError());
       return;
   }
    m_arrayEvent.Add(listenEvent);

    // 开始监听事件
    StartListenEvent();
}

 

/// @brief  开始监听套接字事件
VOID CListenAction::StartListenEvent()
{
    // 网络事件
    WSANETWORKEVENTS netEvents;

    // 返回索引
    DWORD dwIndex = 0;
   
    DWORD dwRetTemp;

    // 循环等待网络事件
    while (TRUE)
    {
        // 判断线程是否退出
        dwRetTemp = WaitForSingleObject(m_ExitEvent.m_hObject, 0);
        if (WAIT_OBJECT_0 == dwRetTemp)
        {
            // 退出
            break;
        }

        // 第一个有事件的通信对象索引
        dwIndex = WSAWaitForMultipleEvents(m_arrayEvent.GetCount(),
                                           &m_arrayEvent[0],
                                           FALSE,
                                           NETWORK_EVENT_WAIT_TIME,
                                           FALSE);
        // 失败
        if (WSA_WAIT_FAILED == dwIndex)
        {
            TRACE(_T("WSAWaitForMultipleEvents error: %d\n"),WSAGetLastError());
            return;
        }
        else
        {
            if (WSA_WAIT_TIMEOUT != dwIndex)
            {
                dwIndex -= WSA_WAIT_EVENT_0;
                for (INT i = dwIndex; i < m_arrayEvent.GetCount(); i++)
                {
                    dwIndex = WSAWaitForMultipleEvents(1, &m_arrayEvent[i], FALSE, 0, FALSE);
                    if ((WSA_WAIT_FAILED != dwIndex) && (WSA_WAIT_TIMEOUT != dwIndex))
                    {
                        /* 枚举通信对象上的事件 */
                        WSAEnumNetworkEvents(m_arraySocket[i].sd, m_arrayEvent[i], &netEvents);
                        if (netEvents.lNetworkEvents & FD_ACCEPT)
                        {
                            if (netEvents.iErrorCode[FD_ACCEPT_BIT] != 0)
                            {
                                TRACE(_T("FD_ACCEPT error: %d\n"),netEvents.iErrorCode[FD_ACCEPT_BIT]);
                                break;
                            }

                            // 接收请求连接的客户端
                            AcceptClientConnection(m_arraySocket[i].sd);
                        }
                        if (netEvents.lNetworkEvents & FD_READ)
                        {
                            if (netEvents.iErrorCode[FD_READ_BIT] != 0)
                            {
                                // 记录日志
                                TRACE(_T("FD_READ error: %d\n"),netEvents.iErrorCode[FD_READ_BIT]);
                                break;
                            }

                            // 报告数据到达事件
                            ReportDataArrive(i);
                        }
                        if (netEvents.lNetworkEvents & FD_CLOSE)
                        {
                            if (netEvents.iErrorCode[FD_READ_BIT] != 0)
                            {
                                // 记录日志
                                TRACE(_T("FD_CLOSE error: %d\n"),netEvents.iErrorCode[FD_CLOSE_BIT]);
                                break;
                            }

                            // 断开连接
                            DisConnect(m_arraySocket[i].sd);
                        }

                    }
     else
     {
      TRACE(_T("WSAWaitForMultipleEvents error: %d\n"),WSAGetLastError());
     }
                }

            }
   else
   {
    TRACE(_T("time out\n"));
   }
        }
    }

}