专栏首页xingoo, 一个梦想做发明家的程序员WSAEventSelect模型 ---应用实例,重写TCP服务器实例

WSAEventSelect模型 ---应用实例,重写TCP服务器实例

// WSAEvent.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <winsock2.h>
#include <stdio.h>
  
#pragma comment(lib,"WS2_32")
class CInitSock
{
public:
     CInitSock(BYTE minorVer=2,BYTE majorVer=2)
     {
         WSADATA wsaData;
         WORD sockVersion = MAKEWORD(minorVer,majorVer);
         if(::WSAStartup(sockVersion,&wsaData)!=0)
         {
             exit(0);
         }
     }
     ~CInitSock()
     {
         ::WSACleanup();
     }
 };
CInitSock initSock;
int _tmain(int argc, _TCHAR* argv[])
{
    //事件句柄和套接字句柄表
    WSAEVENT eventArray[WSA_MAXIMUM_WAIT_EVENTS];
    SOCKET sockArray[WSA_MAXIMUM_WAIT_EVENTS];
    int nEventTotal = 0;
    USHORT nPort = 4567;
    SOCKET sListen = ::socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
    sockaddr_in sin;
    sin.sin_family = AF_INET;
    sin.sin_port = htons(nPort);
    sin.sin_addr.S_un.S_addr = INADDR_ANY;
    if(::bind(sListen,(sockaddr*)&sin,sizeof(sin))==SOCKET_ERROR)
    {
        printf("Failed bind()\n");
        return -1;
    }
    ::listen(sListen,5);
    //创建事件对象,并关联到新的套接字
    WSAEVENT event = ::WSACreateEvent();
    ::WSAEventSelect(sListen,event,FD_ACCEPT|FD_CLOSE);
    //添加到表中
    eventArray[nEventTotal] = event;
    sockArray[nEventTotal] = sListen;
    nEventTotal++;
    //处理网络事件
    while(TRUE)
    {
        //在所有事件上等待
        int nIndex = ::WSAWaitForMultipleEvents(nEventTotal,eventArray,FALSE,WSA_INFINITE,FALSE);
        //对每个事件调用函数,确定它的状态
        nIndex = nIndex-WSA_WAIT_EVENT_0;
        for(int i=nIndex;i<nEventTotal;i++)
        {
            nIndex = ::WSAWaitForMultipleEvents(1,&eventArray[i],TRUE,1000,FALSE);
            if(nIndex == WSA_WAIT_FAILED || nIndex == WSA_WAIT_TIMEOUT)
            {
                continue;
            }
            else
            {
                //获取到来的通知消息,自动重置受信事件
                WSANETWORKEVENTS event;
                ::WSAEnumNetworkEvents(sockArray[i],eventArray[i],&event);
                if(event.lNetworkEvents & FD_ACCEPT)
                {
                    if(event.iErrorCode[FD_ACCEPT_BIT] == 0)
                    {
                        if(nEventTotal>WSA_MAXIMUM_WAIT_EVENTS)
                        {
                            printf("Too many connections!");
                            continue;
                        }
                        SOCKET sNew = ::accept(sockArray[i],NULL,NULL);
                        WSAEVENT event = ::WSACreateEvent();
                        ::WSAEventSelect(sNew,event,FD_READ|FD_CLOSE|FD_WRITE);
                        //添加到表中
                        eventArray[nEventTotal] = event;
                        sockArray[nEventTotal] = sNew;
                        nEventTotal++;
                    }
                }
                else if(event.lNetworkEvents & FD_READ)
                {
                    if(event.iErrorCode[FD_ACCEPT_BIT] == 0)
                    {
                        char szText[256];
                        int nRecv = ::recv(sockArray[i],szText,strlen(szText),0);
                        if(nRecv > 0)
                        {
                            szText[nRecv] = '\0';
                            printf("接收到数据:%s\n",szText);
                        }
                    }
                }
                else if(event.lNetworkEvents & FD_CLOSE)
                {
                    if(event.iErrorCode[FD_ACCEPT_BIT] == 0)
                    {
                        ::closesocket(sockArray[i]);
                        for(int j=i;j<nEventTotal-1;j++)
                        {
                            eventArray[j] = eventArray[j+1];
                            sockArray[j] = sockArray[j+1];
                        }
                        nEventTotal--;
                    }
                }
                else if(event.lNetworkEvents & FD_WRITE)
                {

                }
            }
        }

    }
    return 0;
}

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 《C++标准程序库》-1

    explicit关键字 这个关键字原则上只对构造函数起作用,抑制阴性转换,例如: class String{   String(int size); //本意...

    用户1154259
  • 大数据之Oozie——源码分析(一)程序入口

    工作中发现在oozie中使用sqoop与在shell中直接调度sqoop性能上有很大的差异。为了更深入的探索其中的缘由,开始了oozie的源码分析之路。今天第...

    用户1154259
  • 基于重叠IO模型的 回显TCP服务器设计

    ---------------------1 套接字对象---------------------- 为每个套接字创建一个SOCKET_OBJ对象,记录与之相关...

    用户1154259
  • jQuery的事件模型

    前几天自己着重读了jQuery1.11.1的源码,又结合了之前对DE事件模型的分析,最后也实现一个简陋的事件模型。 jQuery的事件系统离不开jQuery的缓...

    欲休
  • Android7.0 亮屏流程分析

    经过一系列的调用会来到PowerManagerService中的updatePowerStateLocked()

    朵朵花儿
  • 杨校老师课堂之HTML5动画特效的风车案例

    版权声明:本文为Mryang博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/kese7952/article/deta...

    杨校
  • 玩转编程语言:基于Node.js构建自定义代码生成器

    在真实的软件开发过程中,无论使用何种编程开发语言,都不可避免的会遇到代码重复的问题。如何处理重复的问题,可以选择情怀(手动再敲一遍),也可以选择 Copy-to...

    RiboseYim
  • Flutter 实现下拉刷新上拉加载的示例代码

    https://gitee.com/cxyzy1/flutter_pulldown_refresh

    砸漏
  • Spring Security OAuth2 实现登录互踢

    一个账号只能一处登录,类似的业务需求在现有后管类系统是非常常见的。 但在原有的 spring security oauth2 令牌方法流程(所谓的登录)无法满足...

    冷冷
  • 重置多说配置后的问题,这是不让我从良的节奏啊(附禁用谷歌在线字体的方法)!

    本想今天发文从良,金盆洗手,从此不再折腾博客,安心写文章的。结果,发现多说又不能同步服务器评论到本地了!特么真是怕什么来什么啊!想来这金盆暂时用不着了。。。 想...

    张戈

扫码关注云+社区

领取腾讯云代金券