I/O多路复用—select总结

I/O多路复用—select总结

1.基本概念
I/O多路复用是指系统一旦发现进程指定的一个或者多个I/O条件准备就绪，它就会通知该进程。
I/O多路复用适用与一下场合：
（1）客户程序需要同时处理交互式的输入和服务器之间的网络连接。
（2）客户端需要对多个网络连接作出反应。
（3）TCP服务器需要同时处理多个处于监听状态和多个连接状态的套接字。
（4）服务器需要处理多个网络协议的套接字。
（5）服务器需要同时处理不同的网络服务和协议。
2.select函数

int  select（int nfds, fd_set* readfds, fd_set* writefds, fd_set* execptfds, struct
timeval* timeout）；

第一个参数是它所等的最大文件描述符加一，即它所等的文件描述符的个数。
注意：第二个参数到最后一个参数是输入输出型参数。
第二个参数作为输入参数时，是用户想告诉内核我所关心那个文件描述符上的读事件；当该参数作为输出参数时，是内核想告诉用户那些文件描述符上的读事件就绪。
第三个参数作为输入参数时，是用户想告诉内核我所关心那个文件描述符上的写事件；当该参数作为输出参数时，是内核想告诉用户那些文件描述符上的写事件就绪。
第四个参数作为输入参数时，是用户想告诉内核我所关心那个文件描述符上的异常事件；当该参数作为输出参数时，是内核想告诉用户那些文件描述符上的异常事件就绪。
第五个参数是设置超时的，用来描述一段时间长度，如果在这个时间段内，需要监视的文件描述符没有事件发生则函数返回，返回值为0。
该函数的返回值如果为-1，则表示出错；如果为0，则表示超时；如果大于0，则表示我所关心的文件描述符就绪的个数。
3.处理三种文件描述符的方式
由于readfds、writefds和exceptfds这三个参数的类型都是fd_set文件描述符集，它是通过位图来实现的所以对它的操作只能通过对应函数来执行。
（1）FD_CLR(int fd,fd_set* set);用来清除描述词组set中相关fd的位。
（2）FS_ISSET(int fd,fd_set* set);用来测试描述词组set中相关fd的位是否为真。
（3）FD_SET(int fd,fd_set* set);用来设置描述词组set中相关fd的位。
（4）FD_ZERO(fd_set* set);用来清除文件描述词组中的全部位。
（5）timeout的设置：
NULL：表示函数select没有timeout，它会阻塞等待，直到某个文件描述符上等待的事件就绪。
0：仅检测描述符集的状态，然后立即返回，并不等待外部事件的发生。
一段时间：如果在这段时间里没有事件发生，就会超时返回。
4.select优缺点
优点：与多线程和多进程相比，I/O多路复用的最大优势是系统开销小，系统不需要建立新的进程或线程，从而就不需要去维护这写进程或者线程。
缺点：单个进程所能监视的文件描述符有上限；select需要复制大量的句柄数据结构，产生巨大开销；select参数在每次进入select之前都要设置。
5.代码测试
select_server.c

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

#define SIZE sizeof(fd_set)*8
int fds[SIZE];

static void usage(const char* proc)
{
    printf("Usage:%s [local_ip] [local_port]\n",proc);
}

int startup(const char* ip,int port)
{
    int sock = socket(AF_INET,SOCK_STREAM,0);
    if(sock < 0){
    perror("socket");
    exit(2);
    }

    struct sockaddr_in local;
    local.sin_family = AF_INET;
    local.sin_port = htons(port);
    local.sin_addr.s_addr = inet_addr(ip);
    if(bind(sock,(struct sockaddr*)&local,sizeof(local)) < 0){
    perror("bind");
    exit(3);
    }
    if(listen(sock,10) < 0){
    perror("listen");
    exit(4);
    }

    return sock;
}

int main(int argc,const char* argv[])
{
    if(argc != 3){
    usage(argv[0]);
    return 1;
    }

    int listen_sock = startup(argv[1],atoi(argv[2]));

    int nums = sizeof(fds)/sizeof(fds[0]);
    int i = 0;
    for(;i < nums; i++)
    {
    fds[i] = -1;
    }
    fds[0] = listen_sock;
    while(1){
    int max_fd = -1;
    fd_set rfds;
    FD_ZERO(&rfds);
    for(i = 0;i < nums; i++)
    {
    if(fds[i] > 0){
        FD_SET(fds[i],&rfds);
        if(max_fd < fds[i]){
        max_fd = fds[i];
        }
    }
    }
    struct timeval timeout = {0,0};
    switch(select(max_fd+1,&rfds,NULL,NULL,/*&timeout*/NULL)){
        case 0:
        printf("timeout...\n");
        break;
            case -1:
        perror("select");
        break;
        default:
        {
            for(i = 0;i < nums; i++){
            if(fds[i] == -1){
                continue;
            }
            if(i == 0&&FD_ISSET(listen_sock,&rfds)){
                struct sockaddr_in client;
                socklen_t len = sizeof(client);
                int new_sock = accept(listen_sock,(struct sockaddr*)&client,&len);
                if(new_sock < 0){
                perror("accept");
                continue;
                }
                printf("get a new client:[%s:%d]\n",inet_ntoa(client.sin_addr),ntohs(client.sin_port));
                int j = 1;
                for(;j < nums; j++) {
                if(fds[j] == -1){
                    break;
                }
                }
                if(j == nums){
                printf("client is full!\n");
                close(new_sock);
                continue;
                }
                else{
                fds[j] = new_sock;
                }
            }
            else if(i != 0&&FD_ISSET(fds[i],&rfds)){
                char buf[1024];
                ssize_t s = read(fds[i],buf,sizeof(buf)-1);
                if(s > 0){
                buf[s] = 0;
                printf("client# %s\n",buf);
                }
                else if(s == 0){
                close(fds[i]);
                fds[i] = -1;
                printf("client is quit!\n");
                }
                else{
                perror("read");
                close(fds[i]);
                fds[i] = -1;
                }
            }
            else{
            }
            }
        }
        break;
    }
    }
    return 0;
}

select_client.c

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

static void usage(const char* porc)
{
printf("Usage:%s [server_ip] [server_port]\n",porc);
}

int main(int argc,char* argv[])
{
    if(argc != 3){
    usage(argv[0]); 
    return 1;
    }
    int sock = socket(AF_INET,SOCK_STREAM,0);
    if(sock < 0){
    perror("socket");
    return 2;
    }

    struct sockaddr_in server;
    server.sin_family = AF_INET;
    server.sin_port = htons(atoi(argv[2]));
    server.sin_addr.s_addr = inet_addr(argv[1]);

    if(connect(sock,(struct sockaddr*)&server,sizeof(server))<0){
    perror("connect");
    return 3;
    }

    char buf[1024];
    while(1){
    printf("Please Enter# ");
    fflush(stdout);
    ssize_t s = read(0,buf,sizeof(buf)-1);
    if(s > 0){
        buf[s-1] = 0;
        write(sock,buf,strlen(buf));
    }
    }

    close(sock);
    return 0;
}

6.测试结果