LinuxでUDPでデータを受信したかどうかを判断する方法

2年 ago
優斗, 朝陽
1 minute

Linuxでは、select関数がUDPデータの受信を判断するために利用できます。この関数はプログラムが複数のファイル記述子の状態を監視し、それらのいずれかが読み取り可能(または書き込み可能)になるのを待つことを可能にします。以下に簡単なサンプルコードを示します。

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>

int main() {
    int sockfd;
    struct sockaddr_in addr;
    char buffer[1024];

    // 创建UDP套接字
    sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    
    // 设置地址和端口
    memset(&addr, 0, sizeof(addr));
    addr.sin_family = AF_INET;
    addr.sin_port = htons(1234);
    addr.sin_addr.s_addr = htonl(INADDR_ANY);
    
    // 绑定套接字
    bind(sockfd, (struct sockaddr*)&addr, sizeof(addr));
    
    // 设置超时时间
    struct timeval timeout;
    timeout.tv_sec = 5;  // 设置为5秒
    timeout.tv_usec = 0;
    
    // 设置文件描述符集
    fd_set readfds;
    FD_ZERO(&readfds);
    FD_SET(sockfd, &readfds);
    
    // 使用select函数等待数据
    int result = select(sockfd + 1, &readfds, NULL, NULL, &timeout);
    
    if (result > 0) {
        // 接收数据
        ssize_t numBytes = recvfrom(sockfd, buffer, sizeof(buffer), 0, NULL, NULL);
        
        if (numBytes > 0) {
            // 处理接收到的数据
            printf("Received data: %s\n", buffer);
        }
    } else if (result == 0) {
        // 超时，未接收到数据
        printf("Timeout. No data received.\n");
    } else {
        // 出错
        perror("select");
    }
    
    // 关闭套接字
    close(sockfd);

    return 0;
}

上掲のサンプルコードでは、UDPソケットが最初に作成され、次にアドレスとポートがバインドされます。次に、select 関数を使用してタイムアウト時間とファイル記述子セットを設定し、ここでソケットがファイル記述子セットに追加されます。その上で、select 関数を呼び出してデータが待機し、タイムアウト時間内にデータが受領された場合は、recvfrom 関数を使用してデータを受信して処理されます。タイムアウト時間内にデータが受領されなかったか、select 関数でエラーが発生した場合は、適切な処理が行われます。最後に、ソケットが閉じられます。

実際のアプリケーションでは、必要に応じて適切な処理を行う必要がありますので、上述のサンプルコードはUDPデータを受信したかどうかを判断する方法のみを示しており、UDPデータの具体的な内容は処理していないことにご注意ください。

bannerAds