时间戳

首页 > 解决方案 > UDP 客户端仅在特定情况下卡在 recvfrom

c++ - UDP 客户端仅在特定情况下卡在 recvfrom

问题描述

我正在编写实现 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;
}

预期的结果是这样的，我在分开的代码上运行客户端和服务器时得到：

服务器端：

来自客户的你好

已发送您好消息

客户端：

已发送您好消息

来自服务器的你好

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

服务器端：

来自客户的你好

已发送您好消息

客户端：

已发送您好消息

---卡在这里---

我究竟做错了什么？

标签： c++sockets

解决方案

在您的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

推荐阅读