如何修复C语言UDP Socket传命令行参数、大小写转换的代码问题

现有代码核心问题 #

• 客户端问题：
  1. 定义的clientMsg是单字符类型，无法存储完整的命令行参数，发送逻辑也完全没有用到传入的argv[1]内容
  2. 发送逻辑的for循环条件错误写为for (int i = 0; argc; i++)，只要参数数量符合要求就会进入死循环
  3. 客户端填充服务端地址时用了INADDR_ANY，规范写法应该填写服务端实际IP，本地测试可填127.0.0.1
• 服务端问题：
  1. 没有外层循环接收请求，处理单次请求后就直接退出，不满足多轮响应的要求
  2. 大小写转换用的变量c没有初始化，是野值，会引发内存访问异常
  3. 处理完单次请求就关闭socket，不符合持续运行的要求

修正后的完整代码 #

客户端代码 #

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#define PORT 10291
#define MAXLINE 9999
// Client side code
int main(int argc, char *argv[])
{
    int sockfd;
    char buffer[MAXLINE];
    struct sockaddr_in servaddr;
    // 创建socket文件描述符
    if ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0)
    {
        perror("socket creation failed");
        exit(EXIT_FAILURE);
    }
    memset(&servaddr, 0, sizeof(servaddr));
    // 填充服务端信息
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(PORT);
    // 本地测试填127.0.0.1，部署到其他机器改为对应服务端IP即可
    if (inet_pton(AF_INET, "127.0.0.1", &servaddr.sin_addr) <= 0)
    {
        perror("invalid server address");
        exit(EXIT_FAILURE);
    }
    int n, len = sizeof(servaddr);
    if (argc == 2)
    {
        // 直接发送命令行传入的第一个参数
        sendto(sockfd, (const char *) argv[1], strlen(argv[1]),
            MSG_CONFIRM, (const struct sockaddr *) &servaddr,
            sizeof(servaddr));
        printf("Message sent to server. \n");
    }
    else
    {
        printf("Error: please enter the correct number of arguments. \n");
        close(sockfd);
        printf("Disconnected from the server. \n");
        exit(1);
    }
    // 接收服务端返回的结果
    n = recvfrom(sockfd, (char*) buffer, MAXLINE,
        MSG_WAITALL, (struct sockaddr *) &servaddr, &len);
    buffer[n] = '\0';
    printf("Message Received From Server: %s\n", buffer);
    // 关闭socket自动退出
    close(sockfd);
    printf("Disconnected from the Server. \n");
    return 0;
}

服务端代码 #

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <ctype.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#define PORT 10291
#define MAXLINE 9999
// Server Side code
int main(int argc, char *argv[])
{
    int sockfd;
    char buffer[MAXLINE];
    struct sockaddr_in servaddr, cliaddr;
    // 创建socket文件描述符
    if ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0)
    {
        perror("socket creation failed");
        exit(EXIT_FAILURE);
    }
    memset(&servaddr, 0, sizeof(servaddr));
    memset(&cliaddr, 0, sizeof(cliaddr));
    // 填充服务端信息
    servaddr.sin_family = AF_INET;    // IPv4
    servaddr.sin_addr.s_addr = INADDR_ANY;
    servaddr.sin_port = htons(PORT);
    // 绑定socket端口
    if (bind(sockfd, (const struct sockaddr *) &servaddr, sizeof(servaddr)) < 0)
    {
        perror("bind failed");
        exit(EXIT_FAILURE);
    }
    printf("Server running on port %d, press Ctrl+C to exit\n", PORT);
    int len, n;
    len = sizeof(cliaddr);
    // 无限循环处理请求，直到用户按Ctrl+C终止
    while(1) {
        // 接收客户端发送的数据
        n = recvfrom(sockfd, (char*) buffer, MAXLINE,
            MSG_WAITALL, (struct sockaddr *) &cliaddr, &len);
        buffer[n] = '\0';
        // 大小写反转处理
        for(int i = 0; buffer[i] != '\0'; i++) {
            if (isupper(buffer[i]))
                buffer[i] = tolower(buffer[i]);
            else if (islower(buffer[i]))
                buffer[i] = toupper(buffer[i]);
        }
        // 回传处理后的结果给客户端
        sendto(sockfd, (const char *) buffer, strlen(buffer),
            MSG_CONFIRM, (const struct sockaddr *) &cliaddr, len);
        printf("Processed one request from client %s:%d\n", inet_ntoa(cliaddr.sin_addr), ntohs(cliaddr.sin_port));
    }
    // 正常不会运行到此处，Ctrl+C会直接终止程序
    close(sockfd);
    return 0;
}

使用方法 #

• 编译服务端：gcc server.c -o server，运行./server启动服务
• 编译客户端：gcc client.c -o client，运行./client "待处理字符串"，比如./client "HelloWorld123"，会收到返回的hELLOwORLD123，客户端处理完成后自动退出
• 服务端会持续运行接收新请求，按下Ctrl+C即可主动退出

内容的提问来源于stack exchange，提问作者The Coder