socket编程入门篇（四）

本篇博客介绍如何用select函数实现并发服务器。

下面贴代码：

服务端代码

#include <stdio.h> 
#include <unistd.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/select.h>
#define SERV_PORT 8888
#define SERV_IP "127.0.0.1"         //本地回环接口
#define LIST 20                             //服务器最大接受连接
#define MAX_FD 10                      //FD_SET支持描述符数量
int main(void)
{
        int sockfd;
        int err;
        int i;
        int connfd;

        int fd_all[MAX_FD]; //保存所有描述符，用于select调用后，判断哪个可读

//下面两个备份原因是select调用后，会发生变化，再次调用select前，需要重新赋值

        fd_set fd_read;                      //FD_SET数据备份

        fd_set fd_select;                    //用于select


        struct timeval timeout;           //超时时间备份
        struct timeval timeout_select;  //用于select


        struct sockaddr_in serv_addr;   //服务器地址
        struct sockaddr_in cli_addr;      //客户端地址
        socklen_t serv_len;
        socklen_t cli_len;

       //超时时间设置
        timeout.tv_sec = 10;
        timeout.tv_usec = 0;


        sockfd = socket(AF_INET, SOCK_STREAM, 0);
        if(sockfd < 0)
        {
                perror("fail to socket");
                exit(1);

        }
        memset(&serv_addr, 0, sizeof(serv_addr));
        memset(&cli_addr, 0, sizeof(cli_addr));


        serv_addr.sin_family = AF_INET;
        serv_addr.sin_port = htons(SERV_PORT);
        serv_addr.sin_addr.s_addr = inet_addr(SERV_IP);


        serv_len = sizeof(serv_addr);


        err = bind(sockfd, (struct sockaddr *)&serv_addr, serv_len);
        if(err < 0)
        {
                perror("fail to bind");
                exit(1);
        }


        err = listen(sockfd, LIST);

        if(err < 0)
        {
                perror("fail to listen");
                exit(1);
        }

        //初始化fd_all数组
        memset(&fd_all, -1, sizeof(fd_all));

        fd_all[0] = sockfd;   //第一个为监听套接字
        FD_ZERO(&fd_read);
        FD_SET(sockfd, &fd_read);  //将监听套接字加入fd_set


        char buf[1024];  //读写缓冲区
        int num;
        int maxfd;
        maxfd = fd_all[0];  //监听的最大套接字
        

        while(1)
        {


                //每次都需要重新赋值
                fd_select = fd_read;
                timeout_select = timeout;

//              err = select(maxfd+1, &fd_select, NULL, NULL, NULL);
                err = select(maxfd+1, &fd_select, NULL, NULL, (struct timeval *)&timeout_select);
                if(err < 0)
                {
                        perror("fail to select");
                        exit(1);
                }
                if(err == 0)
                        printf("timeout\n");
               

                //检测监听套接字是否可读

                if(FD_ISSET(sockfd, &fd_select))
                {
                        //可读，证明有新客户端连接服务器      
                        cli_len = sizeof(cli_addr);
                        connfd = accept(sockfd, (struct sockaddr *)&cli_addr, &cli_len);
                        if(connfd < 0)
                        {
                                perror("fail to accept");
                                exit(1);
                        }

                        //将新连接套接字加入fd_all及fd_read
                        for(i=0; i<MAX_FD; i++)
                        {
                                if(fd_all[i] != -1)
                                {
                                        continue;
                                }

                                else
                                {
                                        fd_all[i] = connfd;
                                        printf("client fd_all[%d] join\n", i);
                                        break;
                                }
                        }
                        FD_SET(connfd, &fd_read);


                        if(maxfd < connfd)
                        {
                                maxfd = connfd;  //更新maxfd
                        }


                }
                //从1开始查看连接套接字是否可读，因为上面已经处理过0（sockfd）

                for(i=1; i<MAX_FD; i++)
                {
                        if(FD_ISSET(fd_all[i], &fd_select))
                        {
                                printf("fd_all[%d] is ok\n", i);


                                num = read(fd_all[i], buf, 1024);
                                if(num > 0)
                                {
                                        if(strncmp("exit", buf, 4) == 0)
                                        {

                                                //客户端退出，关闭套接字，并从监听集合清除
                                                printf("client:fd_all[%d] exit\n", i);
                                                FD_CLR(fd_all[i], &fd_read);
                                                close(fd_all[i]);
                                                fd_all[i] = -1;


                                                continue;


                                        }

                                        //收到 客户端数据并打印

                                        buf[num] = '\0';
                                        printf("receive buf from client fd_all[%d] is: %s\n", i, buf);
                                }

                                //回复客户端
                                num = write(fd_all[i], "ok", sizeof("ok"));
                                if(num < 0)
                                {
                                        perror("fail to write ");
                                        exit(1);
                                }
                                else
                                {
                                        printf("send reply\n");
                                }
                        }

                        else
                        {
                                //printf("no data\n");                  
                        }


                }


        }


        return 0;
}

 select函数解析：

函数原型

int select(int maxfdp,fd_set *readfds,fd_set *writefds,fd_set *errorfds,struct timeval *timeout);

先解释数据结构 struct fd_set ，可以理解为一个集合，实际上是一个位图，每一个特定为来标志相应大小文件描述符，这个集合中存放的是文件描述符(file descriptor)，即文件句柄(也就是位图上的每一位都能与一个打开的文件句柄(文件描述符)建立联系，这个工作由程序员来完成)，这可以是我们所说的普通意义的文件，当然Unix下任何设备、管道、FIFO等都是文件形式，全部包括在内，所以毫无疑问一个socket就是一个文件，socket句柄就是一个文件描述符。fd_set集合可以通过一些宏由人为来操作，程序员通过操作4类宏，来完成最fd_set的操作：

(1)、FD_ZERO(fd_set *)  清空一个文件描述符集合；

(2)、FD_SET(int ,fd_set *)将一个文件描述符添加到一个指定的文件描述符集合中；

(3)、FD_CLR(int ,fd_set*)       将一个给定的文件描述符从集合中删除；

(4)、FD_ISSET(int ,fd_set* )检查集合中指定的文件描述符是否可以读写。

深入的理解select模型的关键点在于理解fd_set，为了说明方便，我们取fd_set长度为1个字节，fd_set中的每一个bit可以对应一个文件描述符fd。则1字节长的fd_set最大可以对应8个fd。

（1）执行fd_set set；FD_ZERO(&set)；则set用位表示为 0000,0000 。

（2）若fd = 5 ，则执行 FD_SET(fd,&set)后，set变为 0001，0000 （第5位置为1）

（3）若再加入fd=2 ，fd=1，则set变为 0001.0011

（4）执行select(6,&set,0,0,0)阻塞等待

（5）若fd=1,fd=2上都发生可读事件，则select返回，此时set变为0000，0011。没有可读事件发生时 fd = 5 被清空。

下面具体解释select的参数：

（1）int maxfdp是一个整数值，是指集合中所有文件描述符的范围，即所有文件描述符的最大值加1，不能错。

（2）fd_set *readfds是指向fd_set结构的指针，这个集合中应该包括文件描述符，我们是要监视这些文件描述符的读变化的，即我们关心是否可以从这些文件中读取数据了，如果这个集合中有一个文件可读，select就会返回一个大于0的值，表示有文件可读；如果没有可读的文件，则根据timeout参数再判断是否超时，若超出timeout的时间，select返回0，若发生错误返回负值。可以传入NULL值，表示不关心任何文件的读变化。

（3）fd_set *writefds是指向fd_set结构的指针，这个集合中应该包括文件描述符，我们是要监视这些文件描述符的写变化的，即我们关心是否可以向这些文件中写入数据了，如果这个集合中有一个文件可写，select就会返回一个大于0的值，表示有文件可写，如果没有可写的文件，则根据timeout参数再判断是否超时，若超出timeout的时间，select返回0，若发生错误返回负值。可以传入NULL值，表示不关心任何文件的写变化。

（4）fd_set *errorfds同上面两个参数的意图，用来监视文件错误异常文件。

（5）struct timeval* timeout是select的超时时间，这个参数至关重要，它可以使select处于三种状态，第一，若将NULL以形参传入，即不传入时间结构，就是将select置于阻塞状态，一定等到监视文件描述符集合中某个文件描述符发生变化为止；第二，若将时间值设为0秒0毫秒，就变成一个纯粹的非阻塞函数，不管文件描述符是否有变化，都立刻返回继续执行，文件无变化返回0，有变化返回一个正值；第三，timeout的值大于0，这就是等待的超时时间，即 select在timeout时间内阻塞，超时时间之内有事件到来就返回了，否则在超时后不管怎样一定返回，返回值同上述。
客户端代码

#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>


#define SERV_PORT 8888
#define SERV_IP "127.0.0.1"
int main()
{
        int sockfd;
        int err;
        int connfd;
        struct sockaddr_in serv_addr;
        struct sockaddr_in cli_addr;
        socklen_t serv_len;
        socklen_t cli_len;

        sockfd = socket(AF_INET, SOCK_STREAM, 0);
        if(sockfd < 0)
        {
                perror("fail to socket");
                exit(1);
        }
        memset(&serv_addr, 0, sizeof(serv_addr));
        memset(&cli_addr, 0, sizeof(cli_addr));


        serv_addr.sin_family = AF_INET;
        serv_addr.sin_port = htons(SERV_PORT);
        serv_addr.sin_addr.s_addr = inet_addr(SERV_IP);


        serv_len = sizeof(serv_addr);

        //客户端不需要绑定，直接连接即可

        err = connect(sockfd, (struct sockaddr *)&serv_addr, serv_len);
        if(err < 0)

        {
                perror("fail to bind");
                exit(1);
        }

        char buf[1024];
        int num;
        while(1)
        {
                sleep(2);
                num = read(STDIN_FILENO, buf, 1024);
                if(num > 0)
                {

                        //exit代表退出
                        if(strncmp("exit", buf, 4) == 0)
                        {
                                write(sockfd, buf, num);
                                break;
                        }

                }
                write(sockfd, buf, num);
        }
                num = read(sockfd, buf, 1024);
                if(num > 0)
                {
                        buf[num] = '\0';
                        printf("server reply: %s\n", buf);
                }
                else
                        printf("error to read\n");

        close(sockfd);
        return 0;
}