[C++/Linux] UDP编程

一. UDP函数

     UDP（用户数据报协议，User Datagram Protocol）是一种无连接的网络协议，用于在互联网上交换数据。它允许应用程序发送数据报给另一端的应用程序，但不保证数据报能成功到达，也就是说，它提供的是一种不可靠的服务。与TCP（传输控制协议）相比，UDP的优势在于它的简单性和低开销，使得它特别适合于那些对实时性要求高的应用，如视频会议和在线游戏。

在编程中，使用UDP进行数据通信涉及到一系列的函数调用。这些函数调用通常通过操作系统提供的网络接口与内核进行交互。下面是一些在UDP编程中常用的函数及其与内核的关系：

1. socket()

• 用途：创建一个新的套接字（socket），它是进行网络通信的基础。
• 与内核的关系：调用这个函数会让操作系统内核创建一个套接字数据结构，并返回一个引用这个结构的文件描述符（FD）。此后，应用程序通过这个文件描述符进行所有网络操作。

2. bind()

• 用途：将套接字与一个特定的IP地址和端口号绑定。对于服务器端应用程序来说，这样做可以让它在特定的端口上监听来自客户端的连接或数据。
• 与内核的关系：此函数将应用程序指定的IP地址和端口号信息注册到内核中，内核会用这些信息来识别到来的数据包是属于哪个应用程序的。

3. sendto()

• 用途：用于向特定的目标地址发送数据。在UDP编程中，每次发送数据时都需要指定目的地的地址和端口号。
• 与内核的关系：应用程序调用sendto()函数后，数据和目标地址信息被传递给内核，内核负责将数据打包成UDP数据报，并通过网络发送出去。

4. recvfrom()

• 用途：用于接收来自任何地址的数据。应用程序可以通过这个函数获取发送方的地址信息。
• 与内核的关系：当UDP数据报到达操作系统时，内核会根据数据报的目的端口号找到对应的套接字，并将数据放入该套接字的接收缓冲区。应用程序通过recvfrom()函数从这个缓冲区读取数据。

5. close()

• 用途：关闭套接字，释放与之相关的资源。
• 与内核的关系：调用close()函数后，应用程序通知内核它已经完成了对该套接字的使用，内核随后会清理与这个套接字相关的资源。

二. TCP编程与UDP编程的区别：

1. 连接性：

   • TCP：面向连接的协议，通信双方需要建立一个稳定的连接，然后才能进行数据传输。
   • UDP：无连接的协议，数据可以直接发送给接收方，不需要建立连接。

2. 可靠性：

   • TCP：提供可靠的数据传输，通过序列号、确认应答、重传机制等确保数据的可靠到达。
   • UDP：不保证数据可靠到达，可能会出现丢包，但UDP在丢包处理上更简单，不进行重传。

3. 速度：

   • TCP：由于建立连接和保证数据可靠，速度相对较慢。
   • UDP：因为没有连接建立和数据可靠性检查，速度相对较快。

4. 数据流：

   • TCP：保证数据的顺序，确保应用层接收到的数据顺序与发送顺序一致。
   • UDP：不保证数据顺序，数据可能会乱序到达。

5. 使用场景：

   • TCP：适用于对数据准确性要求较高的场景，如文件传输、邮件传输等。
   • UDP：适用于对速度要求较高，但可以容忍一定程度数据丢失的场景，如视频会议、在线游戏等。

三. UDP代码例子：

服务器代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>

#define PORT 8080
#define BUFFER_SIZE 1024

// 函数声明
void error_handling(char *message);

int main(int argc, char *argv[]) {
    int server_socket;
    struct sockaddr_in server_address, client_address;
    socklen_t client_address_size = sizeof(client_address);
    char message[BUFFER_SIZE];

    // 创建UDP套接字
    server_socket = socket(PF_INET, SOCK_DGRAM, 0);
    if (server_socket == -1) {
        error_handling("socket() error");
    }

    // 设置服务器地址结构
    memset(&server_address, 0, sizeof(server_address));
    server_address.sin_family = AF_INET;
    server_address.sin_addr.s_addr = htonl(INADDR_ANY);
    server_address.sin_port = htons(PORT);

    // 绑定套接字到端口
    if (bind(server_socket, (struct sockaddr *)&server_address, sizeof(server_address)) == -1) {
        error_handling("bind() error");
    }

    // 服务器主循环
    while (1) {
        // 接收客户端发送的数据
        int str_len = recvfrom(server_socket, message, BUFFER_SIZE, 0, (struct sockaddr *)&client_address, &client_address_size);
        if (str_len == -1) {
            error_handling("recvfrom() error");
        }

        // 打印接收到的消息
        printf("Received from client: %s\n", message);

        // 处理消息（这里简单地转换为大写）
        for (int i = 0; i < str_len; i++) {
            message[i] = toupper(message[i]);
        }

        // 发送处理后的消息回客户端
        sendto(server_socket, message, str_len, 0, (struct sockaddr *)&client_address, client_address_size);
    }

    // 关闭套接字
    close(server_socket);
    return 0;
}

void error_handling(char *message) {
    fputs(message, stderr);
    fputc('\n', stderr);
    exit(1);
}

客户端代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>

#define PORT 8080
#define BUFFER_SIZE 1024

// 函数声明
void error_handling(char *message);

int main(int argc, char *argv[]) {
    int client_socket;
    struct sockaddr_in server_address;
    char message[BUFFER_SIZE];
    int message_len;

    // 创建UDP套接字
    client_socket = socket(PF_INET, SOCK_DGRAM, 0);
    if (client_socket == -1) {
        error_handling("socket() error");
    }

    // 设置服务器地址结构
    memset(&server_address, 0, sizeof(server_address));
    server_address.sin_family = AF_INET;
    server_address.sin_addr.s_addr = inet_addr("127.0.0.1");
    server_address.sin_port = htons(PORT);

    // 循环读取用户输入并发送给服务器
    while (1) {
        printf("Input message(Q to quit): ");
        fgets(message, BUFFER_SIZE, stdin);

        // 检查是否退出
        if (!strcmp(message, "q\n") || !strcmp(message, "Q\n")) {
            break;
        }

        // 发送消息给服务器
        sendto(client_socket, message, strlen(message), 0, (struct sockaddr *)&server_address, sizeof(server_address));

        // 接收服务器响应
        message_len = recvfrom(client_socket, message, BUFFER_SIZE, 0, NULL, 0);
        if (message_len == -1) {
            error_handling("recvfrom() error");
        }

        // 打印服务器响应
        message[message_len] = 0;
        printf("Message from server: %s\n", message);
    }

    // 关闭套接字
    close(client_socket);
    return 0;
}

void error_handling(char *message) {
    fputs(message, stderr);
    fputc('\n', stderr);
    exit(1);
}