select实现IO多路复用

//服务器端
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <string.h>
#include <dirent.h>
#include <fcntl.h>

int main(int argc, char const *argv[])
{
    if (argc != 2)
    {
        printf("argv %s <port>", argv[0]);
        return -1;
    }
    //1.创建套接字文件
    int sockfd;
    int acceptfd;
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd < 0)
    {
        perror("socket err");
        return -1;
    }
    printf("socket ok %d\n", sockfd);

    //填充ipv4的通信结构体
    struct sockaddr_in serveraddr, caddr;
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(atoi(argv[1]));
    //serveraddr.sin_addr.s_addr=inet_addr("0.0.0.0");//设置自动获取主机ip
    serveraddr.sin_addr.s_addr = INADDR_ANY; //或者hton(INADDR_ANY)

    socklen_t len = sizeof(caddr);
    //2.绑定套接字文件
    if (bind(sockfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr)) < 0)
    {
        perror("bind err");
        return -1;
    }
    printf("bind ok.\n");

    //3.listen 监听 将主动套接字变为被动等待
    if (listen(sockfd, 5) < 0)
    {
        perror("listen err");
        return -1;
    }
    printf("listen ok.\n");

    fd_set readfds, tempfds;
    //清空表
    FD_ZERO(&readfds);
    //添加关心的文件描述符
    FD_SET(0, &readfds);
    FD_SET(sockfd, &readfds);

    char buf[128];
    int ret;
    int maxfd = sockfd;
    int i;
    int recvbyte;

    while (1)
    {
        tempfds = readfds;
        ret = select(maxfd + 1, &tempfds, NULL, NULL, NULL);
        if (ret < 0)
        {
            perror("select err");
            return -1;
        }
        for (i = 0; i <= maxfd; i++)
        {
            if (FD_ISSET(i, &tempfds))
            {
                if (i == 0)
                {
                    fgets(buf, sizeof(buf), stdin);
                    printf("key:%s\n", buf);
                    for(int j=4;j<=maxfd;j++)
                    {
                        if(FD_ISSET(j,&readfds))
                        {
                            send(j,buf,sizeof(buf),0);
                        }
                    }
                }
                else if (i == sockfd)
                {
                    acceptfd = accept(sockfd, (struct sockaddr *)&caddr, &len);
                    if (acceptfd < 0)
                    {
                        perror("accept err");
                        return -1;
                    }
                    //输出查看链接的客户端的ip和端口
                    printf("ip:%s,prot:%d\n", inet_ntoa(caddr.sin_addr), ntohs(caddr.sin_port));
                    FD_SET(acceptfd, &readfds);
                    if (maxfd < acceptfd)
                    {
                        maxfd = acceptfd;
                    }
                }
                else
                {
                    recvbyte=recv(i,buf,sizeof(buf),0);
                    if(recvbyte<0)
                    {
                        perror("recv err");
                        return -1;
                    }
                    else if(recvbyte==0)
                    {
                        printf("client exit\n");
                        close(i);
                        FD_CLR(i, &readfds);
                    }
                    else
                    {
                        printf("%d,buf:%s\n",i, buf);
                    }
                }    
            }  
        }
    }
    close(sockfd);
    return 0;
}





//客户端
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <string.h>

int main(int argc, char const *argv[])
{
    if (argc != 3)
    {
        printf("please input %s <ip> <port>\n", argv[0]);
        return -1;
    }
    //1.创建套接字 socket
    int sockfd;
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd < 0)
    {
        perror("socket err.");
        return -1;
    }
    printf("socket ok %d\n", sockfd);

    //填充ipv4的通信结构体  服务器端ip和端口
    struct sockaddr_in serveraddr;
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(atoi(argv[2]));
    serveraddr.sin_addr.s_addr = inet_addr(argv[1]);

    //&serveraddr -->struct sockaddr_in *

    //2.请求链接  connect
    if (connect(sockfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr)) < 0)
    {
        perror("connect err.");
        return -1;
    }
    printf("connect ok.\n");

    //3.循环收发消息
    char buf[128];
    int recvbyte;
    pid_t pid = fork();
    if (pid < 0)
    {
        perror("fork err.");
        return -1;
    }
    else if (pid == 0)
    {
        while (1)
        {

            fgets(buf, sizeof(buf), stdin);
            if (buf[strlen(buf) - 1] == '\n')
                buf[strlen(buf) - 1] = '\0';
            //发送消息
            send(sockfd, buf, sizeof(buf), 0);
        }
    }
    else
    {
        while (1)
        {
            //接收消息
            recvbyte = recv(sockfd, buf, sizeof(buf), 0);
            if (recvbyte < 0)
            {
                perror("recv err.");
                // return -1;
            }

            printf("buf:%s\n", buf);
        }
    }
    close(sockfd);
    return 0;
}