【C++网络编程】（二）Linux平台下UDP客户/服务端程序

news2024/11/23 15:22:55

Linux平台下UDP客户/服务端程序

图片来源：https://subingwen.cn/linux/udp/

在这里插入图片描述

UDP服务器无法直接检测客户端断开连接。

UDP 服务端

server.cpp

#include <iostream>
#include <cstdlib>      // std::exit
#include <cstring>      // memset
#include <arpa/inet.h>  // socket, bind, recvfrom, sendto，SOCK_DGRAM
#include <unistd.h>     // close

int main()
{
    // 1. 创建 UDP 套接字
    int sockfd = socket(AF_INET, SOCK_DGRAM, 0);  // 创建一个UDP套接字
    if (sockfd == -1)
    {
        perror("socket");  // 错误处理
        std::exit(EXIT_FAILURE);
    }

    // 2. 将socket()返回值和本地的IP端口绑定到一起
    sockaddr_in addr;                     // 用于存储地址信息
    addr.sin_family = AF_INET;            // 地址族，IPv4
    addr.sin_port = htons(10000);         // 转换服务器的端口号10000为网络字节序
    addr.sin_addr.s_addr = INADDR_ANY;    // 绑定到任意IP地址
    // inet_pton(AF_INET, "172.31.108.107", &addr.sin_addr.s_addr); // 指定IP地址

    int ret = bind(sockfd, (sockaddr*)&addr, sizeof(addr)); // 绑定套接字到地址
    if (ret == -1)
    {
        perror("bind");  // 错误处理
        std::exit(EXIT_FAILURE);
    }

    // 3. 和客户端通信
    while (true)
    {
        char buf[1024];  // 接收缓冲区
        memset(buf, 0, sizeof(buf));  // 清零缓冲区
        
        sockaddr_in cliaddr;           // 用于存储客户端地址信息
        socklen_t clilen = sizeof(cliaddr); // 客户端地址结构的大小
        
        // 接收数据
        int len = recvfrom(sockfd, buf, sizeof(buf), 0, (sockaddr*)&cliaddr, &clilen); // 从客户端读取数据
        if (len > 0)
        {
            std::cout << "客户端: " << buf << std::endl;  // 打印客户端发送的消息

            // 回复客户端
            std::string response = "你好, 客户端...\n";
            sendto(sockfd, response.c_str(), response.size(), 0, (sockaddr*)&cliaddr, clilen); // 发送响应
        }
        else
        {
            perror("recvfrom");  // 错误处理
            break;
        }
    }

    close(sockfd);  // 关闭UDP套接字

    return 0;
}

编译与运行

g++ server.cpp -o server
./server

UDP 客户端

client.cpp

#include <iostream>
#include <cstdlib>      // std::exit
#include <cstring>      // memset, strlen
#include <arpa/inet.h>  // socket, sendto, recvfrom, inet_pton, htons
#include <unistd.h>     // close, sleep

int main()
{
    // 1. 创建通信的套接字
    int sockfd = socket(AF_INET, SOCK_DGRAM, 0);  // 创建一个UDP套接字
    if (sockfd == -1)
    {
        perror("socket");  // 错误处理
        std::exit(EXIT_FAILURE);
    }

    // 2. 准备服务器地址信息
    sockaddr_in addr;               // 用于存储服务器地址信息
    addr.sin_family = AF_INET;      // 地址族，IPv4
    addr.sin_port = htons(10000);   // 大端端口转换
    inet_pton(AF_INET, "172.31.108.107", &addr.sin_addr.s_addr); // 将IP地址转换为网络字节顺序

    // 3. 和服务器端通信
    int number = 0;
    while (true)
    {
        // 发送数据
        char buf[1024];                 // 数据缓冲区
        sprintf(buf, "你好, 服务器...\n", number++);  // 格式化字符串
        sendto(sockfd, buf, strlen(buf), 0, (sockaddr*)&addr, sizeof(addr));  // 发送数据
        
        // 接收数据
        memset(buf, 0, sizeof(buf));  // 清空缓冲区
        sockaddr_in recv_addr;         // 用于存储服务器地址信息
        socklen_t addr_len = sizeof(recv_addr);
        int len = recvfrom(sockfd, buf, sizeof(buf), 0, (sockaddr*)&recv_addr, &addr_len);  // 从服务器读取数据
        if (len > 0)
        {
            std::cout << "服务器: " << buf;   // 打印服务器发送的消息
        }
        else
        {
            perror("recvfrom"); 
            break;
        }
        sleep(1);   // 每隔1秒发送一条数据
    }

    close(sockfd);  // 关闭套接字

    return 0;
}