Linux IO模型：IO多路复用

● 应用程序中同时处理多路输入输出流，若采用阻塞模式，得不到预期的目的；

● 若采用非阻塞模式，对多个输入进行轮询，但又太浪费CPU时间；

● 若设置多个进程/线程，分别处理一条数据通路，将新产生进程/线程间的同步与通信问题，使程序变得更加复杂；

● 比较好的方法是使用I/O多路复用技术。

其（select）基本思想是：

○ 先构造一张有关描述符的表（最大1024），然后调用一个函数。

○ 当这些文件描述符中的一个或多个已准备好进行I/O时函数才返回。

○ 函数返回时告诉进程哪个描述符已就绪，可以进行I/O操作。

用select创建并发服务器，可以与多个客户端进行通信（监听键盘、socket、多个acceptfd）

循环服务器：一个客户端可以连接多个客户端，但是不能同时

并发服务器：一个服务器可以同时处理多个客户端的请求

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

// tcp服务器一共有两种文件描述符，一类用于连接，一类用于通信
int main(int argc, char const *argv[])
{
    char buf[128];
    int ret;
    if (argc != 2)
    {
        printf("usage:%s <端口号>\n", argv[0]);
        return -1;
    }

    // 1.创建套接字（socket）
    // socket函数返回值：用于连接的文件描述符
    int sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd < 0)
    {
        perror("socket err");
        return -1;
    }
    printf("sockfd:%d\n", sockfd);

    // 2. 指定网络信息
    struct sockaddr_in saddr, caddr;
    saddr.sin_family = AF_INET;            // IPV4
    saddr.sin_port = htons(atoi(argv[1])); // 端口号
    // saddr.sin_addr.s_addr = inet_addr("0.0.0.0"); // 虚拟机IP地址(自动获取)
    saddr.sin_addr.s_addr = INADDR_ANY; // 虚拟机IP地址
    // INADDR_ANY是一个常量，它指代的是一个特殊的IP地址，即0.0.0.0
    // 在网络编程中，当一个进程需要绑定一个网络端口时，可以使用INADDR_ANY来指定该端口可以接受来自任何IP地址的连接请求
    int len = sizeof(caddr);

    // 3.绑定套接字（bind）
    if (bind(sockfd, (struct sockaddr *)&saddr, sizeof(saddr)) < 0)
    {
        perror("bind err");
        return -1;
    }
    printf("bind ok\n");

    // 4.监听套接字（listen）
    if (listen(sockfd, 6) < 0)
    {
        perror("listen err");
        return -1;
    }
    printf("listen ok\n");

    //******************************************************************//
    // 1)构造一张关于文件描述符的表
    fd_set rfds, tempfds;
    int maxfd;
    // 2)清空表 FD_ZERO
    FD_ZERO(&rfds);
    FD_ZERO(&tempfds);
    // 3)将关心的文件描述符添加到表中 FD_SET
    FD_SET(sockfd, &rfds);
    FD_SET(0, &rfds);
    maxfd = sockfd;
    while (1)
    {
        // 将原来的表复制给新表
        tempfds = rfds;
        struct timeval tm = {1, 0}; // 超时检测
        // 4)调用select函数，监听 select
        ret = select(maxfd + 1, &tempfds, NULL, NULL, 0);
        if (ret < 0)
        {
            perror("select err");
            return -1;
        }
        else if(ret==0)
        {
            printf("time out\n");
        }
        // 5)判断到底是哪一个或者是哪些文件描述符发生了事件 FD_ISSET
        if (FD_ISSET(0, &tempfds))
        {
            fgets(buf, sizeof(buf), stdin);
            printf("keybroad:%s\n", buf);
        }
        if (FD_ISSET(sockfd, &tempfds))
        {
            // 5.接受客户端连接请求（accept）
            // accept函数返回值：用于通信的文件描述符
            int acceptfd = accept(sockfd, (struct sockaddr *)&caddr, &len);
            if (acceptfd < 0)
            {
                perror("accpet err");
                return -1;
            }
            printf("port:%d ip:%s\n", ntohs(caddr.sin_port), inet_ntoa(caddr.sin_addr));
            printf("accpetfd:%d\n", acceptfd);

            // 将用于通信的文件描述符加入表中
            FD_SET(acceptfd, &rfds);
            if (acceptfd > maxfd)
                maxfd = acceptfd;
        }
        for (int i = sockfd + 1; i <= maxfd; i++) // 循环判断哪些文件描述符发生响应
        {
            if (FD_ISSET(i, &tempfds))
            {
                // 6. 接收，发送数据（recv，send）
                ret = recv(i, buf, sizeof(buf), 0);
                if (ret < 0)
                {
                    perror("recv err");
                    break;
                }
                else if (ret == 0)
                {
                    printf("client %d exit\n", i);
                    close(i);         // 关闭退出的客户端文件描述符
                    FD_CLR(i, &rfds); // 将关闭的文件描述符从表中删除
                    while (!FD_ISSET(maxfd, &rfds))
                        maxfd--;
                }
                else
                {
                    printf("client %d: %s\n", i, buf);
                    memset(buf, 0, sizeof(buf));
                }
            }
        }
    }
    //*****************************************************************//
    // 7.关闭套接字（close）
    close(sockfd);

    return 0;
}

使用poll实现client的收发功能

#include <stdio.h>
#include <stdlib.h>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <string.h>
#include <poll.h>
int main(int argc, char const *argv[])
{
    char buf[128];
    if (argc != 3)
    {
        printf("usage:%s <IP地址> <端口号>\n", argv[0]);
        return -1;
    }

    // 1.创建套接字（socket）
    int sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd < 0)
    {
        perror("socket err");
        return -1;
    }
    printf("sockfd:%d\n", sockfd);

    // 2.指定（服务器）网络信息
    struct sockaddr_in saddr;
    saddr.sin_family = AF_INET;                 // IPV4
    saddr.sin_port = htons(atoi(argv[2]));      // 端口号
    saddr.sin_addr.s_addr = inet_addr(argv[1]); // 虚拟机IP地址

    // 3.连接（connect）
    if (connect(sockfd, (struct sockaddr *)&saddr, sizeof(saddr)) < 0)
    {
        perror("connect err");
        return -1;
    }
    printf("connect ok\n");

    // 4.接收发送消息（recv send）
    //************************************************************//
    // 1)创建结构体数组
    struct pollfd fds[2];

    // 2)将关心的文件描述符添加到数组中，并赋予事件
    fds[0].fd = 0;          // 键盘
    fds[0].events = POLLIN; // 想要发生的事件
    fds[1].fd = sockfd;     // 套接字
    fds[1].events = POLLIN; // 想要发生的事件

    // 3)保存数组内最后一个有效元素的下标
    int last = 1;

    // 4)调用poll函数，监听
    while (1)
    {
        int ret = poll(fds, last + 1, 0);
        if (ret < 0)
        {
            perror("poll err");
            return -1;
        }

        // 5)判断结构体内文件描述符实际触发的事件
        if (fds[0].revents == POLLIN)
        {
            // 6)根据不同文件描述符触发的不同事件做对应的逻辑处理
            fgets(buf, sizeof(buf), stdin);
            // printf("keybroad:%s\n", buf);
            send(sockfd, buf, sizeof(buf), 0);
            if (strcmp(buf, "quit\n") == 0)
                break;
        }
        if (fds[1].revents == POLLIN)
        {
            recv(sockfd, buf, sizeof(buf), 0);
            printf("buf:%s\n", buf);
        }
        memset(buf, 0, sizeof(buf));
    }
    //************************************************************//
    // 5.关闭套接字（close）
    close(sockfd);
    return 0;
}