IO多路复用select函数TCP客户端搭建

代码如下：

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

#define ERR_MSG(msg) do{\
    fprintf(stderr, "line:%d\n", __LINE__);\
    perror(msg);\
}while(0)

#define IP "192.168.8.90"     //填你们自己的本机IP ifconfig查看
#define PORT 6666             //1024~49151

int main(int argc, const char *argv[])
{
    //创建流式套接字
    int sfd = socket(AF_INET, SOCK_STREAM, 0);
    if(sfd < 0)
    {
        ERR_MSG("socket");
        return -1;
    }


    //填充地址信息结构体
    //根据地址族的不同，地址信息结构体不一致，AF_INET:man 7 ip
    struct sockaddr_in sin;
    sin.sin_family         = AF_INET;         //必须填AF_INET;
    sin.sin_port         = htons(PORT);     //端口号的网络字节序，1024~49151
    sin.sin_addr.s_addr = inet_addr(IP);     //本机IP地址的网络字节序，ifconfig 
    //更换网络环境后，IP地址会改变
    
    //连接服务器
    if(connect(sfd,(struct sockaddr *)&sin,sizeof(sin)) < 0)
    {
        ERR_MSG("connect");
        return -1;
    }
    //创建集合
    fd_set readfds, tempfds;

    FD_ZERO(&readfds);     //清空文件描述符集合
    FD_ZERO(&tempfds);     //清空文件描述符集合

    //将需要的文件描述符放入到读集合中
    FD_SET(0, &readfds);
    FD_SET(sfd, &readfds);

    int maxfd = sfd;     //存储最大文件描述符

    int s_res = 0;

    char buf[128] = "";
    char buf_rec[128] = "";
    ssize_t res = 0;
    int newfd = 0;

    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;
        }

        //若代码能运行到当前位置，则代表集合中有文件描述符准备就绪
        //集合中会只剩下准备就绪的文件描述符
        //例如0号准备就绪，则集合中会只剩下0；
        //sfd准备就绪，则集合中会只剩下sfd;
        //0和sfd均准备就绪，则集合中会剩下0和sfd;
        //如果不将文件描述符对应空间中的数据读取出来，则该事件会一直触发
        for(int i=0; i<=maxfd; i++)
        {
            if(FD_ISSET(i, &tempfds) == 0)     //当i所代表的文件描述符在不在tempfds中。
            {
                continue;
            }
            
            //能够运行到当前位置，则代表i所代表的文件描述符在tempfds中
            //只需要判断i所代表的文件描述符是什么，走对应操作即可
            if(0 == i)
            {
                //触发键盘输入事件
                printf("触发键盘输入事件\n");

                bzero(buf, sizeof(buf));
                fgets(buf,sizeof(buf),stdin);
                buf[strlen(buf)-1] = 0;
                if(send(sfd, buf, sizeof(buf), 0) < 0)
                {
                    ERR_MSG("send");
                    return -1;
                }
                printf("发送成功\n");
            }
            else
            {
                printf("客户端交互事件\n");
                bzero(buf_rec, sizeof(buf_rec));
                //接收
                res = recv(i, buf_rec, sizeof(buf_rec), 0);
                if(res < 0)
                {
                    ERR_MSG("recv");
                    return -1;
                }
                else if(0 == res)
                {
                    fprintf(stderr, "[%s:%d] newfd = %d 服务器下线\n", \
                            inet_ntoa(sin.sin_addr), ntohs(sin.sin_port), sfd);
                    return -1;
                }
                printf("buf_rec:%s\n",buf_rec);
            }


        }

    }

    close(sfd);
    return 0;
}

终端执行结果：

ubuntu@ubuntu:net$ gcc 29test.c 
ubuntu@ubuntu:net$ ./a.out 
客户端交互事件
buf_rec:aaaaa
bbbbbbbb
触发键盘输入事件
发送成功
hhhhhhh
触发键盘输入事件
发送成功
客户端交互事件
buf_rec:99999999
客户端交互事件
[192.168.8.90:6666] newfd = 3 服务器下线