c++ - 通过 select() 实现非阻塞 udp 套接字

Question

我想创建一个异步/非阻塞的 udp 客户端服务器应用程序，其中客户端和服务器应该可以在不等待对方轮到的情况下相互聊天。 我开始知道这可以通过 select()...

这是我的服务器(仅提及通信部分):

fd_set readfds,writefds;

while(1){
    FD_ZERO(&readfds);
    FD_ZERO(&writefds);
    FD_SET(sd,&readfds);
    FD_SET(sd,&writefds);

    int rv = select(n, &readfds, NULL, NULL, NULL);
    if(rv==-1) 
    {
        printf("Error in Select!!!\n");
        exit(0);
    }
    if(rv==0) 
    {
        printf("Timeout occurred\n");
    }
    if (FD_ISSET(sd, &readfds))
    {     
        int client_length = (int)sizeof(struct sockaddr_in);
        memset(&buffer,0,sizeof(buffer));
        int bytes_received = recvfrom(sd, buffer,SIZE, 0, (struct sockaddr *)&client, &client_length);
        if (bytes_received < 0) {
            fprintf(stderr, "Could not receive datagram.\n");
            closesocket(sd);
            WSACleanup();
            exit(0);
        }
    }

    printf("\nClient says: %s",buffer);
    printf("\nWrite :");
    fgets(buffer,SIZE,stdin); 
    if(FD_ISSET(sd,&writefds)) {
        int client_length = (int)sizeof(struct sockaddr_in);
        if(sendto(sd, buffer,strlen(buffer), 0, (struct sockaddr *) &client,client_length)<0) {
            printf("Error sending the file! \n");
            printf("%d\n",WSAGetLastError());
            exit(1);
        }
    }
}
closesocket(sd);
WSACleanup();

return 0;

这是我的客户:

fd_set readfds,writefds;
while(1)
{
    FD_ZERO(&readfds);
    FD_ZERO(&writefds);
    FD_SET(cs,&readfds);
    FD_SET(cs,&writefds);
    int rv=select(n,&readfds,&writefds,NULL,NULL);
            if(rv==-1)
    {
        printf("Error in Select!!!\n");
        exit(0);
    }
    if(rv==0)
    {
        printf("Timeout occurred\n");
    }
    printf("\nWrite  ");

    fgets(send_buffer,SIZE,stdin);
    if(FD_ISSET(cs,&writefds))
    {
        int server_length = sizeof(struct sockaddr_in);
        FD_CLR(cs,&writefds);

        if (sendto(cs, send_buffer, (int)strlen(send_buffer) + 1, 0, (struct sockaddr *)&server, server_length) == -1)
        {
            fprintf(stderr, "Error transmitting data.\n");
            closesocket(cs);
            WSACleanup();
            exit(0);
        }
    }

    char file_buffer[SIZE];
    //Reply reception from the server:"Ready to receive file"
    int data2=0;
    if (FD_ISSET(cs, &readfds))
    {
        FD_CLR(cs,&readfds);
        int server_length = sizeof(struct sockaddr_in);
        data2=recvfrom(cs,file_buffer,strlen(file_buffer)-1,0,(struct sockaddr *)&server,&server_length);
        //file_buffer[data2]='\0';
        if(data2<0)
        {
            printf("Server is not on:(\n");
            exit(0);
        }
    }
    //printf("%d",data2);
    printf("\nServer says:");
    for(int i=0;i<data2;i++)
    {
        putchar(file_buffer[i]);
    } 
}

return 0;
}

起初在服务器端我写了:int rv = select(n, &readfds, &writefds, NULL, NULL);

但这导致在服务器初始化时在服务器控制台上打印出一个完整的空数组，而且服务器和客户端之间的通信不正常。 删除“&writefds”确实删除了冗余数据，但不正确的通信问题仍然存在...... 所以如果有人帮助我，我将非常感激......

Answer 1

我发现您的代码存在一些问题。

首先，如果你想等待来自套接字和终端的输入，你应该将这两个 fds 放入 readfds 集合:

FD_SET(cs, &readfds);
FD_SET(stdin, &readfds);

因为你没有做任何写作，你不应该使用 writefds。

接下来，您应该确保仅在终端已准备好输入时才尝试从终端读取。所以，你应该这样做:

if (FD_ISSET(stdin, &readfds)) {
    // Read in from terminal...
}

你目前每次都在尝试阅读，无论发生什么。

最后，您在调用 recvfrom 时错误地测量了 file_buffer 的大小。 strlen 仅在您已经将数据放入其中后才起作用。您应该使用 sizeof(file_buffer)。