完成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.52"   
 #define PORT 6655          
 
 int main(int argc, const char *argv[])
 {
     //创建流式套接字
     int sfd = socket(AF_INET, SOCK_STREAM, 0);
     if(sfd < 0)
     {
         ERR_MSG("socket");
         return -1;
     close(sfd);
     }
 
     //允许端口快速重用   
     int reuse = 1;
     if(setsockopt(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;      //必须填AF_INET;
     sin.sin_port        = htons(PORT);  //端口号的网络字节序，1024~49151
     sin.sin_addr.s_addr = inet_addr(IP);    //本机IP地址的网络字节序，ifconfig 
 
 
     //绑定服务器的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;
     fd_set tempfds;
 
     FD_ZERO(&readfds);  //清空文件描述符集合
     FD_ZERO(&tempfds);
 
     //将需要的文件描述符放入到读集合中
     FD_SET(0, &readfds);
     FD_SET(sfd, &readfds);
 
     int maxfd = sfd;
     int s_res = 0;
     struct sockaddr_in cin;     //存储获取到的客户端的地址信息
     socklen_t addrlen = sizeof(cin);    //真实地址信息结构体的大小
 
     struct sockaddr_in saveCin[1020];
 
     //从已经完成连接的队列头中获取一个客户端的信息，生成一个新的文件描述符
     //生成的新的文件描述符才是与客户端通信的文件描述符
     char buf[128] = "";
     ssize_t res = 0;
     int newfd = 0;
  //更换网络环境后，IP地址会改变
     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++ )
         {
             //当i所表达的文件描述符不在tempfds中
             if( FD_ISSET(i,&tempfds)==0)
             {
                 continue;
             }
 
             if(i==0)
             {
                 //触发键盘输入事件
                 printf("触发键盘输入事件\n");
                 fgets(buf, sizeof(buf), stdin);
                 buf[strlen(buf)-1] = 0;
                 printf("buf:%s\n", buf);
             }
 
             else if(i==sfd)
             {
 
                 printf("客户端连接事件\n");
                 newfd = accept(sfd, (struct sockaddr*)&cin, &addrlen);
                 if(newfd < 0)
                 {
                     ERR_MSG("accept");
                     return -1;
                 }
                 printf("[%s:%d] newfd = %d\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port),newfd);
 
                 saveCin[newfd-4] = cin;
 
                 //将newfd添加到读集合中
                 FD_SET(newfd,&readfds);
 
                 //更新maxfd
                 maxfd = maxfd<newfd?newfd:maxfd;
 
             }
 
             else
             {
 
                 //触发交互事件
                 bzero(buf, sizeof(buf));
 
                 //接收
                 res = recv(i, buf, sizeof(buf), 0);
 
                 if(res < 0)
                 {
                     ERR_MSG("recv");
                     return -1;
                 }
 
                 else if(0 == res)
                 {
                     fprintf(stderr, "[%s:%d] newfd = %d 客户端下线\n", inet_ntoa(saveCin[i-4].sin_addr), ntohs(saveCin[i-4].sin_port),i);
 
                     //将i所代表的文件描述符关闭
                     close(i);
 
                     //将i所代表的文件描述符从集合中剔除
                     FD_CLR(i,&readfds);
 
                     //更新maxfd
                     for(int j=maxfd; j>0 ; j--)
                     {
                         if(FD_ISSET(j,&readfds))
                         {
                             maxfd = j;
                             break;
                         }
                     }
 
                 }
 
                 else
                 {
                     printf("[%s:%d] newfd = %d : %s\n", \
                             inet_ntoa(saveCin[i-4].sin_addr), ntohs(saveCin[i-4].sin_port),i, buf);
                 }
 
             }
 
         }
     }
 
     close(sfd);
 
     return 0;                                                                                                                                                                                                                                                                                                                                                                                                                              

客户端代码如下：

#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.52"   
#define PORT 6655         

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

    //允许端口快速重用   
    int reuse = 1;
    if(setsockopt(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;      //必须填AF_INET;
    sin.sin_port        = htons(PORT);  //端口号的网络字节序，1024~49151
    sin.sin_addr.s_addr = inet_addr(IP);    //本机IP地址的网络字节序，ifconfig 

    //连接服务器
    if(-1==connect(sfd,(struct sockaddr*)&sin,sizeof(sin)))
    {
        ERR_MSG("connect");
        return -1;
    }
   
    
    //创建集合
    fd_set readfds;
    fd_set tempfds;

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

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

    int maxfd = sfd;
    int s_res = 0;

    struct sockaddr_in saveCin[1020];
    
    char buf[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;
        }
        for( int i=0 ; i<=maxfd ; i++ )
        {
            //当i所表达的文件描述符不在tempfds中
            if( FD_ISSET(i,&tempfds)==0)
            {
                continue;
            }

            if(i==0)
            {
                //触发键盘输入事件
                printf("触发键盘输入事件\n");

                bzero(buf,sizeof(buf));
                fgets(buf, sizeof(buf), stdin);
                buf[strlen(buf)-1] = 0;

                //发送
                if(-1 == send(sfd,buf,sizeof(buf),0))       
                {
                    ERR_MSG("send");
                    return -1;
                }
                printf("send success\n");

            }

            else if(i==sfd)
            {
                //触发交互事件
                printf("服务器交互事件\n");
                bzero(buf, sizeof(buf));

                //接收
                res = recv(sfd, buf, sizeof(buf), 0);

                if(res < 0)
                {
                    ERR_MSG("recv"); 
                    return -1;
                }

                else if(0 == res)
                {
                    fprintf(stderr,"服务器关闭\n");
                }

            }

        }

    }


    close(sfd);

    return 0;

}

结果如下：