C实现的UDP压力测试工具

最近在做一个功能模块，目前功能上已经实现，但是性能暂时没有做过测试。

我的功能主要是接收udp消息，并将udp消息进行封装处理，发送到后端处理模块进行处理。

简单写一个小的udp客户端程序，去模拟压力测试。

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

int main(int argc, char **argv)
{
    if (argc != 3)
    {
        printf("Usage: %s $SUM $RATE\n", argv[0]);
        exit(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 = inet_addr("127.0.0.1");
    server_addr.sin_port = htons(4020);

    int fd = socket(AF_INET, SOCK_DGRAM, 0);
    if (fd < 0)
    {
        printf("create socket failed: %s\n", strerror(errno));
        exit(2);
    }

    struct timeval start;
    gettimeofday(&start, NULL);
    unsigned long ustart = start.tv_sec*1000000 + start.tv_usec;

    usleep(1);

    struct timeval now;
    memset(&now, 0, sizeof(struct timeval));
    unsigned long unow = 0;

    int totalUdp = atoi(argv[1]);
    int maxRate = atoi(argv[2]); // caps
    printf("totalUdp: %d\n", totalUdp);
    printf("maxRate: %d\n", maxRate);

    int sum = 0;
    double rate = 0.0;

    char buff[1024*16] = ""; 
    while (sum < totalUdp)
    {
        gettimeofday(&now, NULL);
        unow = now.tv_sec*1000000 + now.tv_usec;
        rate = sum/(unow - ustart + 0.0);
        if (maxRate > 0 && rate * 1000000 > maxRate)
        {
            usleep(1000); // sleep 0.001 second
            continue;
        }
        sprintf(buff, "%d %d %d\0", now.tv_sec, now.tv_usec, sum);
        int res = sendto(fd, buff, strlen(buff), 0, (struct sockaddr*)&server_addr, sizeof(server_addr));
        if (res < 0)
        {
            printf("sendto error: %s\n", strerror(errno));
            exit(3);
        }
        // printf("send %d bytes ok\n", res);
        sum ++;
    }

    close(fd);

    printf("finish %d in %.3f, rate: %.f\n", sum, (unow-ustart+0.0)/1000000, rate*1000000);

    exit(0);
}

编码 && 运行：

[udpdriver@eb6347 test]$ gcc -o main main.c
[udpdriver@eb6347 test]$ ./main
Usage: ./main $SUM $RATE
[udpdriver@eb6347 test]$ ./main 10000 1000
totalUdp: 10000
maxRate: 1000
finish 10000 in 9.999, rate: 1000

程序原理很简单：

在未达到指定的发送消息数量时，不断去计算从开始发送时刻到当前发送时刻的平均发送速率，如果当前发送平均速率小于你指定的最大发送速率，则可以继续发送，否则暂停一小段时间，然后继续尝试发送。

例如：我要测试20k的消息总量，2kcaps，如何运行？

./main 20000 2000

实际将在约10s后完成压力测试（发送）。

我们可以设定不同的模块配置参数（例如修改udp接收缓冲区的大小），以及每条udp消息的长度，每秒发送的caps量来做出一组性能测试结果，用以反映我们模块的性能。

原文链接：https://blog.csdn.net/test1280/article/details/79733708