通过select()实现一个非阻塞的udp套接字

时间:2013-04-20 06:57:32

标签: c++ c sockets select

我想创建一个异步/非阻塞的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);

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

1 个答案:

答案 0 :(得分:1)

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

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

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

由于您没有写任何内容,因此不应使用writefds

接下来,如果输入就绪,您应该确保只是尝试从终端读取。所以,你应该做的事情如下:

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

无论发生什么事,您目前都在尝试阅读。

最后,当您拨打file_buffer电话时,您错误地衡量了recvfrom的尺寸。只有在您已将数据放入其中后,strlen才有效。您应该使用sizeof(file_buffer)