C/C++ 使用socket做一个网络性能测试工具-DPDK中文网

要使用 C/C++ 创建一个网络性能测试工具，你可以通过 socket 编程来实现。下面是一个简单的示例，包括客户端和服务器部分，用于测试带宽和延迟。

1. 服务器端代码

#include <iostream>
#include <cstring>
#include <sys/socket.h>
#include <netinet/in.h>
#include <unistd.h>

#define PORT 8080
#define BUFFER_SIZE 1024

int main() {
    int server_fd, new_socket;
    struct sockaddr_in address;
    int opt = 1;
    int addrlen = sizeof(address);
    
    // 创建socket
    if ((server_fd = socket(AF_INET, SOCK_STREAM, 0)) == 0) {
        std::cerr << "Socket failed" << std::endl;
        return -1;
    }
    
    // 附加到端口
    setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
    
    address.sin_family = AF_INET;
    address.sin_addr.s_addr = INADDR_ANY;
    address.sin_port = htons(PORT);

    // 绑定socket
    if (bind(server_fd, (struct sockaddr *)&address, sizeof(address)) < 0) {
        std::cerr << "Bind failed" << std::endl;
        return -1;
    }

    // 开始监听
    if (listen(server_fd, 3) < 0) {
        std::cerr << "Listen failed" << std::endl;
        return -1;
    }

    while (true) {
        // 接受连接
        if ((new_socket = accept(server_fd, (struct sockaddr *)&address,
                                  (socklen_t*)&addrlen)) < 0) {
            std::cerr << "Accept failed" << std::endl;
            continue; 
        }

        char buffer[BUFFER_SIZE] = {0};
        
        while (read(new_socket , buffer, BUFFER_SIZE) > 0) {
            // 可以在这里处理接收到的数据，例如统计字节数或响应等。
            send(new_socket , buffer , strlen(buffer) , 0); // 回显数据，简单的测试功能
        }

        close(new_socket);
    }

    return 0;
}

2. 客户端代码

#include <iostream>
#include <cstring>
#include <sys/socket.h>
#include <netinet/in.h>
#include <unistd.h>

#define PORT 8080
#define BUFFER_SIZE 1024

int main() {
    int sock = 0;
    struct sockaddr_in serv_addr;

    // 创建socket
    if ((sock = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
        std::cerr << "Socket creation error" << std::endl;
        return -1; 
    }

   serv_addr.sin_family = AF_INET; 
   serv_addr.sin_port = htons(PORT); 

   // 转换IPv4地址从文本到二进制格式
   if(inet_pton(AF_INET, "127.0.0.1", &serv_addr.sin_addr) <= 0) { 
       std::cerr << "Invalid address / Address not supported" << std::endl; 
       return -1; 
   } 

   // 尝试连接到服务器
   if (connect(sock, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0) { 
       std::cerr << "Connection Failed" << std::endl; 
       return -1; 
   } 

   char message[BUFFER_SIZE] = "Hello from client";
   
   for(int i=0; i<10000; ++i){ // 发多个包进行性能测试
       send(sock , message , strlen(message), MSG_NOSIGNAL);
       char buffer[BUFFER_SIZE] = {0};
       read(sock , buffer , BUFFER_SIZE);
   }
   
   close(sock);
   
   return 0; 
}