UDP socket编程示例

服务端：

#include <iostream>
#include <cstring>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>

const int BUFFER_SIZE = 1024;

//多线程版本：https://github.com/reedthink/socket-program/blob/master/udp-server.cpp

int main()
{
    int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if (sockfd < 0)
    {
        std::cerr << "Error opening socket" << std::endl;
        return 1;
    }
    struct sockaddr_in servaddr {};
    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr = INADDR_ANY;
    servaddr.sin_port = htons(6868);

    if (bind(sockfd, (const struct sockaddr*)&servaddr, sizeof(servaddr)) < 0)
    {
        std::cerr << "Error on binding" << std::endl;
        return 1;
    }
    std::cout << "UDP server up and running on port 6868" << std::endl;

    char buffer[BUFFER_SIZE];
    struct sockaddr_in clientaddr;
    socklen_t len = sizeof(clientaddr);

    while(true)
    {
        ssize_t recvbytes = recvfrom(sockfd, buffer, BUFFER_SIZE, 0, (struct sockaddr*)&clientaddr, &len);
        if (recvbytes < 0)
        {
            std::cerr << "Error receiving message" << std::endl;
            return 1;
        }
        buffer[recvbytes] = '\0';
        std::cout << "Received message:" << buffer << " from " << inet_ntoa(clientaddr.sin_addr) << std::endl;

        if (sendto(sockfd, buffer, recvbytes, 0, (const struct sockaddr * )&clientaddr, len) < 0)
        {
            std::cerr << "Error sending message" << std::endl;
            return 1;
        }
    }
    close(sockfd);
}

客户端：

#include <iostream>
#include <string>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>

const int BUFFER_SIZE = 1024;

int main() {
    // Create a UDP socket
    int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if (sockfd < 0) {
        std::cerr << "Error opening socket" << std::endl;
        return 1;
    }

    // Set up the server address structure
    struct sockaddr_in servaddr {};
    servaddr.sin_family = AF_INET; // IPv4
    servaddr.sin_addr.s_addr = inet_addr("127.0.0.1"); // Server IP
    servaddr.sin_port = htons(6868); // Port

    // Message to be sent
    std::string message = "Hello, Server!";

    // Send the message to the server
    if (sendto(sockfd, message.c_str(), message.size(), 0, (struct sockaddr *)&servaddr, sizeof(servaddr)) < 0) {
        std::cerr << "Error sending message" << std::endl;
        return 1;
    }

    std::cout << "Message sent to server." << std::endl;

    // Receive the message from the server
    char buffer[BUFFER_SIZE];
    struct sockaddr_in fromaddr;
    socklen_t fromlen = sizeof(fromaddr);

    ssize_t recvbytes = recvfrom(sockfd, buffer, BUFFER_SIZE, 0, (struct sockaddr *)&fromaddr, &fromlen);
    if (recvbytes < 0) {
        std::cerr << "Error receiving message" << std::endl;
        return 1;
    }

    // Null-terminate the buffer
    buffer[recvbytes] = '\0';

    // Print the received message
    std::cout << "Received message: " << buffer << " from " << inet_ntoa(fromaddr.sin_addr) << std::endl;

    // Close the socket
    close(sockfd);

    return 0;
}

区别：

1. 绑定（Bind）:

   • 服务端：通常需要调用bind()函数来绑定到一个特定的端口，这样它才能监听发往该端口的数据。在示例中，服务端绑定到端口6868。
   • 客户端：不需要绑定到一个端口，因为它是主动发起连接的一方。在示例中，客户端没有调用bind()，所以它会使用一个临时的源端口。

2. 接收和发送数据:

   • 服务端：使用recvfrom()函数来接收客户端发送的消息，这个函数可以获取发送方的地址信息。
   • 客户端：使用sendto()发送消息，使用recvfrom()接收服务端的回显消息。

3. 地址信息:

   • 服务端：需要一个服务器地址结构来存储其绑定的地址信息，包括IP地址和端口号。
   • 客户端：需要一个服务器地址结构来指定消息发送的目标地址和端口号。

4. 网络编程模型:

   • 服务端：通常设计为可以持续运行，处理来自不同客户端的请求。
   • 客户端：可能设计为发送一次请求后关闭，或者根据需要发送多次请求。

5. 并发处理:

   • 服务端：在实际应用中，可能需要处理多个客户端的并发连接，这可能涉及到多线程或多进程的使用。
   • 客户端：通常不需要处理并发，除非一个客户端需要同时与多个服务端通信。

相同点:

1. 关闭连接（Close）:

   • 服务端和客户端：在完成数据传输后，都会调用close()函数来关闭socket。

2. 错误处理:

   • 服务端和客户端：都需要对可能发生的错误进行处理，如socket创建失败、绑定失败、发送/接收失败等。