UDP客户端仅在特定情况下才会卡在recvfrom中

时间:2019-04-03 18:20:24

标签: c++ sockets

我正在编写实现UDP服务器和客户端的C ++代码。

当我编写两个代码时,该代码工作正常,一个用于服务器,另一个用于客户端,如本例所示:https://www.geeksforgeeks.org/udp-server-client-implementation-c/

我想做的是用相同的代码编写一个客户端函数和一个服务器函数。我的想法是选择程序如何与命令行参数一起使用。

问题是,以这种方式实现并在两个运行相同代码,具有不同命令行参数的终端上进行测试,一个用于服务器,另一个用于客户端,当客户端收到服务器响应时,客户端卡在recvfrom中。

#include <unistd.h> 
#include <stdio.h> 
#include <sys/socket.h> 
#include <stdlib.h> 
#include <netinet/in.h> 
#include <string.h> 
#include <arpa/inet.h>
#include <unistd.h>
#define MAXLINE 1024
#define PORT 32000

int send(){
    int sockfd; 
    char buffer[MAXLINE]; 
    char *hello = "Hello from server"; 
    struct sockaddr_in servaddr, cliaddr; 

    // Creating socket file descriptor 
    if ( (sockfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0 ) { 
        perror("socket creation failed"); 
        exit(EXIT_FAILURE); 
    } 

    memset(&servaddr, 0, sizeof(servaddr)); 
    memset(&cliaddr, 0, sizeof(cliaddr)); 

    // Filling server information 
    servaddr.sin_family = AF_INET; // IPv4 
    servaddr.sin_addr.s_addr = inet_addr("127.0.0.1"); 
    servaddr.sin_port = htons(PORT); 

    // Bind the socket with the server address 
    if ( bind(sockfd, (const struct sockaddr *)&servaddr, 
            sizeof(servaddr)) < 0 ) 
    { 
        perror("bind failed"); 
        exit(EXIT_FAILURE); 
    } 

    socklen_t len;
    int n; 
    n = recvfrom(sockfd, (char *)buffer, MAXLINE, 
                MSG_WAITALL, ( struct sockaddr *) &cliaddr, 
                &len); 
    buffer[n] = '\0'; 
    printf("Client : %s\n", buffer); 
    sendto(sockfd, (const char *)hello, strlen(hello), 
        MSG_CONFIRM, (const struct sockaddr *) &cliaddr, 
            len); 
    printf("Hello message sent.\n"); 

    return 0; 
}

int receive(){
    int sockfd; 
    char buffer[MAXLINE]; 
    char *hello = "Hello from client"; 
    struct sockaddr_in   servaddr; 

    // Creating socket file descriptor 
    if ( (sockfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0 ) { 
        perror("socket creation failed"); 
        exit(EXIT_FAILURE); 
    } 

    memset(&servaddr, 0, sizeof(servaddr)); 

    // Filling server information 
    servaddr.sin_family = AF_INET; 
    servaddr.sin_port = htons(PORT); 
    servaddr.sin_addr.s_addr = inet_addr("127.0.0.1"); 

    int n;
    socklen_t len; 

    sendto(sockfd, (const char *)hello, strlen(hello), 
        MSG_CONFIRM, (const struct sockaddr *) &servaddr, 
            sizeof(servaddr)); 
    printf("Hello message sent.\n"); 

    n = recvfrom(sockfd, (char *)buffer, MAXLINE, 
                MSG_WAITALL, (struct sockaddr *) &servaddr, 
                &len); 
    buffer[n] = '\0'; 
    printf("Server : %s\n", buffer); 

    return 0; 
}

int main(int argc, char const *argv[]) {

    int command = atoi(argv[1]);

    if(command == 0){
        send();

    }
    if(command == 1){
        receive();
    }


    return 0;
}

预期结果是这样的,当我在单独的代码上运行客户端和服务器时得到的结果:

服务器端:

  

你好,客户

     

您好消息已发送

客户端:

  

您好消息已发送

     

您好来自服务器

但是运行上面的代码时我得到的是

服务器端:

  

你好,客户

     

您好消息已发送

客户端:

  

您好消息已发送

     

---卡在这里---

我在做什么错了?

1 个答案:

答案 0 :(得分:1)

send()函数中,您没有将len初始化为recvfrom可以存储客户地址的缓冲区长度。

根据recvfrom的手册页:

  

ssize_t recvfrom(int sockfd,void * buf,size_t len,int flags,                          struct sockaddr * src_addr,socklen_t * addrlen);

     

如果src_addr不为NULL,并且基础协议提供了   消息的源地址,该源地址位于   src_addr指向的缓冲区。在这种情况下,addrlen是一个值结果   论点。在调用之前,应将其初始化为   与src_addr关联的缓冲区。返回时,addrlen更新为   包含源地址的实际大小。

它不起作用,因为未正确接收客户地址,因此响应消息被发送到错误的地址。要解决您的问题,只需在调用len之前初始化recvfrom

socklen_t len = sizeof(cliaddr); // The size of the buffer you're passing to store the client address