网络编程select

select模型实现了单线程上处理多个套接字的可能

1.select 函数定义如下

int select(

     int nfds;  

     fd_set* readfds;   //指向一个套接字集合

     fd_set* writefds;

     fd_set* exceptfds;  //指向一个套接字集合，用来检查错误

    const struct timeval* timeout    //结构体  指定函数等待的最长时间   如果为NULL 则最长时间为无限长

);

2.套接字集合

fd_set结构可以把多个套接字连在一起，形成一个套接字集合。而select函数可以测试这个集合中哪些套接字有事件发生

typedef struct fd_set{

    u_int fd_count;

   SOCKET fd_array[FD_SETSIZE];   //套接字句柄数组  套接字集合  默认的是64

}fd_set;  

3.操作fd_set的宏

    FD_ZERO(*set)   很显然，初始化set为空集合，集合在使用前应该清空

    FD_CLR(*set)     从set移除套接字s

    FD_ISSET(s,*set)     检查s是不是set的成员，如果是返回TRUE

    FD_SET(s,*set)   添加套接字到集合

主要程序代码：

#include"initsock.h"
#include<stdio.h>

CInitSock theSock;  //初始化Winsock库

int main()
{
    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()");
        return -1;
    }
    //进入监听模式
    ::listen(sListen,5);

    //select模型处理过程
    //第一步：初始化一个套接字 集合 fdSocket,添加监听套接字句柄到这个集合
    fd_set fdSocket;   //所有可用套接字集合
    FD_ZERO(&fdSocket);
    FD_SET(sListen,&fdSocket);
    while(TRUE)
    {
        //第二步：将fdSocket集合的一个拷贝fdRead传递给select函数
        //当有事件发生时，select函数移除fdRead集合中没有未决I/O操作的套接字句柄，然后返回
        fd_set fdRead=fdSocket;
        int nRet=::select(0,&fdRead,NULL,NULL,NULL);
        if(nRet>0)
        {
            //第三步：通过将原来fdSocket集合与select处理过的fdRead集合比较
            //确定有哪些套接字有未决I/O,并进一步处理这些I/O
            for(int i=0;i<(int)fdSocket.fd_count;i++)
            {
                if(fdSocket.fd_array[i]==sListen) //监听套接字接收到新连接
                {
                    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)
                    {
                        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;
}

其中 initsock

#include<WinSock.h>
#pragma comment(lib,"WS2_32")
class CInitSock
{
public:
    CInitSock(BYTE minorVer=2,BYTE majorVer=2)
    {
        //初始化Wsh_32.dll
        WSADATA wsaData;
        WORD sockVersion=MAKEWORD(minorVer,majorVer);
        if(::WSAStartup(sockVersion,&wsaData)!=0)
        {
            exit(0);
        }
    }
    ~CInitSock()
    {
        ::WSACleanup();
    }
};