socket的select函数

思路：

1. 初始化一个socket
2. 建立一个socket列表用于管理socket
3. 将初步连接的socket放入列表中
4. 用select判断列表中未处理的socket

Win API版本

1.

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);

2.

// select模型处理过程
// 1）初始化一个套节字集合fdSocket，添加监听套节字句柄到这个集合
fd_set fdSocket;        // 所有可用套节字集合
FD_ZERO(&fdSocket);
FD_SET(sListen, &fdSocket);

3.

while(TRUE)
{
    // 2）将fdSocket集合的一个拷贝fdRead传递给select函数，
    // 当有事件发生时，select函数移除fdRead集合中没有未决I/O操作的套节字句柄，然后返回。
    fd_set fdRead = fdSocket;
    int nRet = ::select(0, &fdRead, NULL, NULL, NULL);
    if(nRet > 0)
    {
        // 3）通过将原来fdSocket集合与select处理过的fdRead集合比较，
        // 确定都有哪些套节字有未决I/O，并进一步处理这些I/O。
        for(int i=0; i<(int)fdSocket.fd_count; i++)
        {
            if(fdSocket.fd_array[i] == sListen)        // （1）监听套节字接收到新连接
            {
                if(fdSocket.fd_count < FD_SETSIZE)
                {
                    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)                        // （2）可读
                {
                    szText[nRecv] = '\0';
                    printf("接收到数据：%s \n", szText);
                }
               else                                // （3）连接关闭、重启或者中断
                {
                    ::closesocket(fdSocket.fd_array[i]);
                    printf("关闭\n");
                    FD_CLR(fdSocket.fd_array[i], &fdSocket);
                }
            }
        }
    }
    else
    {
        printf(" Failed select() \n");
        break;
    }
}

本文系转载而来学习之用.

windows头文件WinSock2.h中对FD_SETSIZE的定义是

#ifndef FD_SETSIZE
#define FD_SETSIZE      64
#endif /* FD_SETSIZE */

如果是偏大点的服务程序,64个socket肯定是太小了,如果想手动地增加FD_SETSIZE的大小,可以在导入WinSock2.h的前一行加入 #define FD_SETSIZE 1024,网上貌似说这个数字不能大于1024.