选择模型2

选择模型

fd_set结构可以把多个套接字连在一起，形成一个套接字集合

typedef struct fd_set{
u_int fd_count;//下面数组的大小
SOCKET fd_array[FD_SETSIZE];//套接字句柄数组
}fd_set;

网络事件：

readfds集合：数据可读，连接关闭，重启或者中断

writefds集合：数据能发送

exceptfds集合：OOB数据可读

设置超时：

如果为null为无限阻塞，知道有网络事件发生

typedef struct timeval{
long tv_sec;//指示等待多少秒
long tv_usec;//指示等待多少毫秒
}timeval;

应用举例

1 初始化fdSocket集合，添加监听套接字句柄

2 将fdSocket集合拷贝fRead传递给select函数，当有事件发生的时候，select函数移除fRead中没有未决IO操作的句柄，然后返回

3 比较原来的fdSocket集合，与select处理过的fdRead集合，确定哪些套接字有未决IO并处理这些IO

4 回到2进行选择

CInitSock theSock;//初始化winsock库
int main()
{
    USHORT nPort=4567;//此服务器监听的端口号
    //创建监听套接字
    SOCKET sListen=::listen(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 0;
    }
    //进入监听模式
    ::listen(sListen,5);
    
    //select模型处理过程
    ////////////////////////////////////////////////////////////////////////////
    // 1  初始化fdSocket集合，添加监听套接字句柄
    fd_set fdSocket;//所有可用套接字集合
    FD_ZERO(&fdSocket);
    FD_SET(sListen,&fdSocket);
    while(TRUE)
    {
        ////////////////////////////////////////////////////////////////////////
        //2 将fdSocket集合拷贝fRead传递给select函数，当有事件发生的时候，select函数移除fRead中没有未决IO操作的句柄，然后返回
        fd_set fdRead = fdSocket;
        int nRet = ::select(0,&fdRead,NULL,NULL,NULL);
        if(nRet>0)
        {
            ///////////////////////////////////////////////////////////////////
            // 3  比较原来的fdSocket集合，与select处理过的fdRead集合，确定哪些套接字有未决IO并处理这些IO
            for(int i=0;i<(int)fdSocket.fd_count;i++)
            {
                if(FD_ISSET(fdSocket.fd_array[i],&fdRead))
                {
                    if(fdSocket.fd_array[i]==sListen)// 1 监听套接字连接新连接
                    {
                        sockaddr_in addrRemote;
                        int nAddrLen = sizeof(addrRemote);
                        SOCKET sNew = ::accept(sListen,(SOCKADDR*)&addrRemote,&nAddrLen);
                        FD_SET(sNew,&fdSocket);
                        printf("接收到连接(%s)\n",::inet_ntoa(addrRemote.sin_addr));
                    }
                    else
                    {
                        printf("Too much connections!\n");
                        continue;
                    }
                }
                else
                {
                    char szText[256];
                    int nRecv = ::recv(fdSocket.fd_array[i],szText,strlen(szText),0);
                    if(nRecv > 0)//可读
                    {
                        szText[nRecv] = '\0';
                        printf("接收到数据：%s\n",szText);
                    }
                    else//连接关闭或者重启
                    {
                        ::closesocket(fdSocket.fd_array[i]);
                        FD_CLR(fdSocket.fd_array[i],&fdSocket);
                    }
                }
            }
        }
        else
        {
            printf("Failed select()!\n");
            break;
        }
    }
    return 0;
}