linux+c测试传输速度

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

#define SERVER_IP "127.0.0.1"
#define PORT 8080
#define BUFFER_SIZE 1024 * 1024 // 1MB

int main() {
    int sock;
    struct sockaddr_in server_addr;
    char buffer[BUFFER_SIZE];
    ssize_t bytes_received;
    clock_t start_time, end_time;

    // 创建 socket
    if ((sock = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
        perror("socket");
        return 1;
    }

    // 设置服务器地址
    memset(&server_addr, 0, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(PORT);
    inet_pton(AF_INET, SERVER_IP, &server_addr.sin_addr);

    // 连接服务器
    if (connect(sock, (struct sockaddr*)&server_addr, sizeof(server_addr)) == -1) {
        perror("connect");
        close(sock);
        return 1;
    }

    // 发送测试数据
    memset(buffer, 'A', BUFFER_SIZE);
    send(sock, buffer, BUFFER_SIZE, 0);

    // 接收测试数据并计时
    start_time = clock();
    bytes_received = recv(sock, buffer, BUFFER_SIZE, 0);
    end_time = clock();

    if (bytes_received == -1) {
        perror("recv");
    } else {
        double duration = (double)(end_time - start_time) / CLOCKS_PER_SEC;
        double speed = (double)bytes_received / duration / 1024 / 1024; // MB/s
        printf("传输速度: %.2f MB/s\n", speed);
    }

    close(sock);
    return 0;
}