C 中的套接字编程，使用 select() 函数

Question

根据我从this thread 得到的答案，我创建了这个：

    //Server 

    sock_init(); //from SFL, see http://legacy.imatix.com/html/sfl/

    timeout = 50000;

    serv_sock_input[0] = TCP(1234); 
    serv_sock_input[1] = UDP(9876);

    input_protocols[0] = "tcp";
    input_protocols[1] = "udp";

    while (1)
    {
        FD_ZERO(&sock_set);
        for (x = 0; x<number_of_inputs; x++)
        {
            FD_SET(serv_sock_input[x], &sock_set);
        }

        select_timeout.tv_sec = timeout;
        select_timeout.tv_usec = 0;

        if (select(0, &sock_set, NULL, NULL, &select_timeout) == 0)
            printf("No requests");
        else
        {
            for (x = 0; x<number_of_inputs; x++)
            {
                if (FD_ISSET(serv_sock_input[x],&sock_set))
                {
                    printf("\nRequest on port %d: \n", x);
                    if ((strcmp(input_protocols[x],"tcp")) == 0) //in this case, 0 returned == TRUE
                    {
                        accept_socket(serv_sock_input[x]);
                        printf("Input TCP Port %d\n",x);
                        close_socket(serv_sock_input[x]);
                    }
                    else
                    {
                        printf("Input UDP Port %d\n",x);
                    }
                }
            }
        }
    }
    sock_term();
}

---

int TCP (unsigned short port)
{
    int sock;                        
    struct sockaddr_in servAddr; 

    if ((sock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0)
        exit(1);

    memset(&servAddr, 0, sizeof(servAddr));
    servAddr.sin_family = AF_INET;         
    servAddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servAddr.sin_port = htons(port);            

    if (bind(sock, (struct sockaddr *) &servAddr, sizeof(servAddr)) < 0)
        exit(1);

    if (listen(sock, 5) < 0)
        exit(1);

    return sock;
}

---

int UDP (unsigned short port)
{
    int sock;                        /* socket to create */
    struct sockaddr_in servAddr; /* Local address */

    /* Create socket for sending/receiving datagrams */
    if ((sock = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP)) < 0)
        exit(1);

    /* Construct local address structure */
    memset(&servAddr, 0, sizeof(servAddr));   /* Zero out structure */
    servAddr.sin_family = AF_INET;                /* Internet address family */
    servAddr.sin_addr.s_addr = htonl(INADDR_ANY); /* Any incoming interface */
    servAddr.sin_port = htons(port);      /* Local port */

    /* Bind to the local address */
    if (bind(sock, (struct sockaddr *) &servAddr, sizeof(servAddr)) < 0)
        exit(1);


    return sock;
}

---

//Client
sock_init();

    if ((client_sock_output = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0)
        exit(1);

    memset(&client_addr, 0, sizeof(client_addr));
    client_addr.sin_family      = AF_INET;
    client_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
    client_addr.sin_port        = htons(1234);

    if (connect(client_sock_output, (struct sockaddr *) &client_addr, sizeof(client_addr)) < 0)
        exit(1);

    closesocket(client_sock_output);

    sock_term();

当服务器启动时，服务器在if(select(...)) 语句处被阻塞。

因此，当我运行服务器，然后运行客户端时，客户端会连接到服务器（有时在连接之前需要运行几次客户端）。然后if(select...)) 语句不再为真，它继续到 else。

之后，客户端关闭连接，程序关闭。但是，这就是我的问题发生的地方，if(select(...)) 语句总是错误的。我得到这个输出：

Request on port 0:  
Input TCP Port 0  

Request on port 1:  
Input UDP Port 1

此输出将永远重复。怎么不卡在if(select(...))？

Answer 1

你有两个问题：你不明白accept()在TCP中是如何工作的，你需要在UDP中读取传入的数据。

select() 告诉你一个正在监听的套接字有连接要接受，或者正在读取的套接字有数据要读取。

为了让选择停止告诉你这个，你需要实际读取数据或接受连接。

在您的 UDP 分支中，您需要调用 receiv 来实际获取数据。如果你不这样做，select 会一直告诉你你有数据。

在您的 TCP 分支中，您调用 accept_socket。我不知道你的实现是什么，但关闭你刚刚调用 accept() 的套接字很可能是错误的。 accept() 为您返回一个新套接字 - 您应该用于 IO 的那个。如果需要关闭任何东西，那就是那个新的套接字。

Answer 2

请检查你为什么在服务器中有这个。

if (select(0, &sock_set, NULL, NULL, &select_timeout) == 0)

替换为

if (select(maxDescPlus1, &sock_set, NULL, NULL, &select_timeout) == 0)

其中 maxDescPlus1 --> 是要选择的描述符数量加上 1 个值。