7.10网络编程作业：IO多路复用

Select的TCP服务器模型

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<sys/types.h>
#include<sys/stat.h>
#include<fcntl.h>
#include<unistd.h>
#include<errno.h>

#define ERR_MSG(msg) do{\
    fprintf(stderr, "__%d__", __LINE__);\
    perror(msg);\
}while(0)

#define IP ""
#define PORT 8888

int main(int argc, const char *argv[])
{
    //创建流式套接字
    int sfd = socket(AF_INET, SOCK_STREAM, 0);         //套接字是标识符，这里定义了一个sfd文件描述符用于接收socket的返回值
    if(sfd < 0)
    {
        ERR_MSG("socket");
        return -1;
    }

    //允许端口快速被重用
    int reuse = 1;
    if(stesockopt(sfd, SOL_SOCKET, SO_REUSEADDR,&reuse, sizeof(reuse)) < 0)
    {
        ERR_MSG("setsockopt");
        return -1;
    }
    printf("允许端口快速被重用成功\n");

    //填充服务器的地址信息结构体
    struct sockaddr_in sin;
    sin.sin_family         = AF_INET; //规定
    sin.sin_port         = htons(PORT); //端口号的网络字节序， 1024——49151
    sin.sin_addr.s_addr = inet_addr(IP); //本机IP的网络字节序

    //绑定服务器的IP和端口--->必须绑定
    if(bind(sfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
    {
        ERR_MSG("bind");
        return -1;
    }
    printf("bind success\n");

    
    //将套接字设置为被动监听状态，监听是否有客户端连接
    if(listen(sfd, 128) < 0)
    {
        ERR_MSG("listen");
        return -1;
    }
    printf("listen success\n");

    //创建集合
    fd_set readfds, tempfds;

    //fd_set本质上是一个结构体，且结构体中是一个整形数组，若不初始化，其中会存有随机值
    //由于fd_set这个集合需要存放监测的文件描述符，不能让其中随机到有效文件描述符
    //所以需要初始化
    
    //将集合初始化
    FD_ZERO(&readfds);
    FD_ZERO(&tempfds);

    //将需要监测的文件描述符放入集合中
    FD_SET(0, &readfds);
    FD_SET(sfd, &readfds);
    
    int maxfd = sfd;     //存最大的文件描述符 

    int newfd = -1;
    char buf[128] = "";
    ssize_t res = 0;
    int s_res = 0;

    struct sockaddr_in cin;         //存储客户地址信息
    socklen_t addrlen = sizeof(cin);

    while(1)
    {
        tempfds = readfds;         //保留原有信息
        //让内核监测集合中的文件描述符是否准备就绪
        s_res = select(maxfd+1, &tempfds, NULL, NULL, NULL);
        if(s_res < 0)
        {
            ERR_MSG("select");
            return -1;
        }
        else if(0 == s_res)
        {
            printf("time out....\n");
            break;
        }

        printf("__%d__\n", __LINE__);

        //能运行到当前位置，则代表集合中有文件描述符准备就绪
        //判断那个文件描述符准备就绪，走对应的处理函数即可

        //当集合中有某个文件描述符准备就绪了，集合中会只剩下产生事件的文件描述符
        //若0准备就绪，则集合中只会剩下0
        //若双方都准备就绪，择机和中只会剩下0和sfd
        
        //通过循环遍历集合中的所有文件描述符
        //由于sfd和newfd的值不确定，所以从0开始遍历
        for(int i=0; i<=maxfd; i++)
        {
            if(FD_ISSET(i, &tempfds) == 0)
            {
                continue;
            }

            //能运行到当前位置，就说明i所代表的文件描述符在tempfds中

            if(int i=0; i<=maxfd; i++)         //键盘输入事件
            {
                printf("触发键盘输入\n");
                fgets(buf, sizeof(buf), stdin);
                buf[strlen(buf) - 1] = 0;

                printf ("buf: %s\n", buf);
            }
            else if(std == i)         //客户端连接事件
            {
                printf("触发客户端连接事件\n");
                newfd = accept(sfd, (struct sockaddr*)&cin, &addrlen);
                if(newfd < 0)
                {
                    ERR_MSG("accept");
                    continue;
                }
                printf("[%s : %d] newfd=%d 客户端连接成功\n",\
                        inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd);

                //将newfd添加到集合readfds中
                FD_SET(newfd, &readfds);

                //更新maxfd
                maxfd = maxfd>newfd ? maxfd:newfd;
            }
            else         //客户端交互事件
            {
                printf("触发客户端交互事件\n");
                bzero(buf, sizeof(buf));
                //接收数据
                res = recv(i, buf, sizeof(buf), 0);
                if(res < 0)
                {
                    ERR_MSG("recv");
                    continue;
                }
                else if(0 == res)
                {
                    printf("[%s : %d] newfd=%d 客户端下线\n",\
                            inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), i);

                    //关闭文件描述符
                    close(i);

                    //将i所代表的文件描述符删除
                    FD_CLR(i, &readfds);

                    //更新maxfd --> 有可能将集合中最大的文件描述符剔除
                    //从最大的文件描述符开始遍历，直到文件描述符在集合中，则该文件描述符就是最大的那个
                    /*
                     int j = maxfd;
                     for(; j>=0; j--)
                     {
                         if(FD_ISSET(j, &readfds))
                            break;
                     }
                     maxfd = j;
                     */
                    while(!FD_ISSET(maxfd, &readfds) && maxfd-- >= 0);
                    continue;
                }
                printf("[%s : %d] newfd=%d : %s\n", \
                        inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), i, buf);
            }
        }
    }

    //关闭文件描述符
    if(close(sfd) < 0)
    {
        ERR_MSG("close");
        return -1;
    }
    return 0;