使用select函数搭建TCP客户端和服务器

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__ ", __LINE__);\
    perror(msg);\
}while(0)

#define PORT 3333            //1024~49151
#define IP "192.168.10.57"     //ifconfig查看

int updateMaxfd(int maxfd, fd_set readfds);

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

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


    //填充地址信息结构体，真实的地址信息结构体在man 7 ip 
    //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(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
    {
        ERR_MSG("bind");
        return -1;
    }
    printf("bind success __%d__\n", __LINE__);


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

    //设置一个读集合
    //由于fd_set本质上是一个结构体，结构体中有个一个数组，用于存储文件描述符
    //所以不能让该数组为随机值，需要清空
 fd_set readfds, tempfds;        
    FD_ZERO(&readfds);         //清空集合
    FD_ZERO(&tempfds);         //清空集合

    //将需要监测的文件描述符添加到集合中
    FD_SET(0, &readfds);
    FD_SET(sfd, &readfds);

    int maxfd = sfd;

    struct sockaddr_in saveCin[1024-4];     //0,1,2,sfd不会有对应的客户端，所以-4
    struct sockaddr_in cin;                 //存储客户端的地址信息
    socklen_t addrlen = sizeof(cin);
    int newfd = 0;

    int s_res = 0;
    char buf[128] = "";
    ssize_t res = 0;
    while(1)
    {
        tempfds = readfds;
        //调用select，让内核去监测文件描述符集合
        s_res = select(maxfd+1, &tempfds, NULL, NULL, NULL);
        if(s_res < 0)
        {
            ERR_MSG("se