c++

C++ recvfrom函数在多线程环境中的应用

小樊
82
2024-08-30 00:37:50
栏目: 编程语言

recvfrom() 是一个用于接收来自指定套接字上的数据报的函数,通常用于无连接的UDP协议

以下是一个简单的示例,展示了如何在多线程环境中使用 recvfrom() 函数:

#include<iostream>
#include<thread>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <cstring>

const int BUFFER_SIZE = 1024;
const int PORT = 8080;

void receive_data(int sockfd) {
    char buffer[BUFFER_SIZE];
    struct sockaddr_in client_addr;
    socklen_t client_addr_size = sizeof(client_addr);

    while (true) {
        ssize_t received = recvfrom(sockfd, buffer, BUFFER_SIZE, 0, (struct sockaddr *)&client_addr, &client_addr_size);
        if (received > 0) {
            std::cout << "Received message from client: " << inet_ntoa(client_addr.sin_addr) << ":" << ntohs(client_addr.sin_port)<< std::endl;
            std::cout << "Message: "<< buffer<< std::endl;
        } else {
            std::cerr << "Error receiving data"<< std::endl;
            break;
        }
    }
}

int main() {
    int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if (sockfd == -1) {
        std::cerr << "Error creating socket"<< std::endl;
        return 1;
    }

    struct sockaddr_in server_addr;
    memset(&server_addr, 0, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    server_addr.sin_port = htons(PORT);

    if (bind(sockfd, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1) {
        std::cerr << "Error binding socket"<< std::endl;
        close(sockfd);
        return 1;
    }

    // 创建一个线程来处理接收到的数据报
    std::thread receiver_thread(receive_data, sockfd);

    // 主线程可以执行其他任务,例如发送数据报或等待用户输入
    // ...

    // 等待接收线程完成
    receiver_thread.join();

    close(sockfd);
    return 0;
}

这个示例中,我们创建了一个线程来处理接收到的数据报。主线程可以执行其他任务,例如发送数据报或等待用户输入。当主线程完成其任务时,它会等待接收线程完成,然后关闭套接字并退出程序。

请注意,这个示例仅用于演示目的,实际应用中可能需要根据需求进行更复杂的错误处理和资源管理。

0
看了该问题的人还看了