linux網絡編程之select

#include<stdio.h>
#include<sys/select.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<unistd.h>
#include<stdlib.h>

void main()
{
    
    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_addr.s_addr = inet_addr("127.0.0.1");
    addr.sin_port = htons(3001);


    int sock = socket(AF_INET,SOCK_STREAM,0);
    if(-1==sock)
    {
        perror("socket error");
        return;
    }
    printf("listen fd = %d\n",sock);
    
    int reuse = 1;
    setsockopt(sock,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse));


    if(bind(sock,(struct sockaddr*)&addr,sizeof(addr))<0)
    {
        perror("bind error");
        return;
    }


    listen(sock,5);


    fd_set read_set;
    fd_set write_set;
    
    FD_ZERO(&read_set);
    FD_ZERO(&write_set);




    FD_SET(sock,&read_set);
    int max = sock+1;
    
    while(1)
    {
        fd_set temp_read_set;
        fd_set temp_write_set;


        memcpy(&temp_read_set,&read_set,sizeof(temp_read_set));
        memcpy(&temp_write_set,&write_set,sizeof(temp_write_set));


        struct timeval tv;
        tv.tv_sec = 2;
        tv.tv_usec = 0;
        int ret = select(FD_SETSIZE,&temp_read_set,NULL,NULL,&tv);
        if(ret==-1)
        {
            perror("select error");
            break;    
        }


        int i=0;
        for(;i<FD_SETSIZE;++i)
        {
            char buf[100];
            memset(buf,0,100);
            if(FD_ISSET(i,&temp_read_set))
            {
                if(i==sock)
                {
                    struct sockaddr_in client_addr;
                    int sockaddr_in_len = sizeof(client_addr);
                    int client_sock = accept(sock,(struct sockaddr*)&client_addr,&sockaddr_in_len);
                    FD_SET(client_sock,&read_set);
                    FD_SET(client_sock,&write_set);
                    printf("accept new conn");
                    if(client_sock>max-1)
                    {
                        max = client_sock + 1;
                    }
                }
                else
                {
                    recv(i,buf,10,0);
                    printf("recv data: %s\n",buf);
                }
            }
            else if(FD_ISSET(i,&temp_write_set))
            {


            }
        }
         
    }
    
    int i=0;
    for(;i<FD_SETSIZE;++i)
    {
        if(FD_ISSET(i,&read_set))
        {
            close(i);
        }
    }
}