linux c测试传输速度

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

#define BUFFER_SIZE 1024 * 1024 // 1MB buffer
#define TEST_DURATION 10 // seconds

int main(int argc, char *argv[]) {
    if (argc != 3) {
        fprintf(stderr, "Usage: %s <server_ip> <port>
", argv[0]);
        exit(1);
    }

    const char *server_ip = argv[1];
    int port = atoi(argv[2]);

    int sockfd;
    struct sockaddr_in serv_addr;
    socklen_t length;

    if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
        perror("socket");
        exit(1);
    }

    memset(&serv_addr, 0, sizeof(serv_addr));
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_port = htons(port);

    if (inet_pton(AF_INET, server_ip, &serv_addr.sin_addr) <= 0) {
        perror("inet_pton");
        exit(1);
    }

    if (connect(sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0) {
        perror("connect");
        exit(1);
    }

    char *buffer = malloc(BUFFER_SIZE);
    if (!buffer) {
        perror("malloc");
        exit(1);
    }
    memset(buffer, 'A', BUFFER_SIZE);

    int total_bytes = 0;
    int bytes_sent;
    time_t start_time = time(NULL);

    while (time(NULL) - start_time < TEST_DURATION) {
        bytes_sent = send(sockfd, buffer, BUFFER_SIZE, 0);
        if (bytes_sent < 0) {
            perror("send");
            break;
        }
        total_bytes += bytes_sent;
    }

    close(sockfd);
    free(buffer);

    double speed = total_bytes / (double)(TEST_DURATION * 1024 * 1024); // MB/s
    printf("TCP Transmission Speed: %.2f MB/s
", speed);

    return 0;
}