Windows/Linux桌面全链路压测选型手册

Python ```python import socket import threading def proxy_thread(client_sock, addr): 接收客户端数据 client_data = client_sock.recv(1024) if not client_data: client_sock.close() return 解析客户端数据，获取目标地址和端口 host, port = client_data.split(b':') server_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) server_sock.connect((host.decode('utf-8'), int(port))) 转发客户端数据到目标服务器 server_sock.sendall(client_data) 接收目标服务器的响应 server_data = server_sock.recv(1024) if not server_data: server_sock.close() client_sock.close() return 转发目标服务器的响应给客户端 client_sock.sendall(server_data) 关闭连接 server_sock.close() client_sock.close() def main(): 创建协同伙伴服务器套接字 proxy_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 绑定到监听端口 proxy_sock.bind(('', 1080)) 开始监听传入连接 proxy_sock.listen(5) while True: 接受传入连接 client_sock, addr = proxy_sock.accept() 创建一个线程来处理协同伙伴连接 thread = threading.Thread(target=proxy_thread, args=(client_sock, addr)) thread.start() if __name__ == '__main__': main() ``` C++ ```cpp include include include include include include include using namespace std; void proxy_thread(int client_sock, const string &addr) { // 解析客户端数据，获取目标地址和端口 size_t pos = addr.find(':'); string host = addr.substr(0, pos); int port = stoi(addr.substr(pos + 1)); // 创建到目标服务器的套接字 int server_sock = socket(AF_INET, SOCK_STREAM, 0); if (server_sock cerr close(client_sock); return; } // 连接到目标服务器 struct sockaddr_in server_addr; memset(&server_addr, 0, sizeof(server_addr)); server_addr.sin_family = AF_INET; server_addr.sin_port = htons(port); server_addr.sin_addr.s_addr = inet_addr(host.c_str()); if (connect(server_sock, (struct sockaddr )&server_addr, sizeof(server_addr)) cerr close(client_sock); close(server_sock); return; } // 转发客户端数据到目标服务器 char buf[1024]; while (true) { int n = recv(client_sock, buf, sizeof(buf), 0); if (n break; } send(server_sock, buf, n, 0); } // 接收目标服务器的响应 while (true) { int n = recv(server_sock, buf, sizeof(buf), 0); if (n break; } send(client_sock, buf, n, 0); } // 关闭连接 close(client_sock); close(server_sock); } int main() { // 创建业务伙伴服务器套接字 int proxy_sock = socket(AF_INET, SOCK_STREAM, 0); if (proxy_sock cerr return 1; } // 设置复用选项 int reuseaddr = 1; if (setsockopt(proxy_sock, SOL_SOCKET, SO_REUSEADDR, &reuseaddr, sizeof(reuseaddr)) cerr return 1; } // 绑定到监听端口 struct sockaddr_in proxy_addr; memset(&proxy_addr, 0, sizeof(proxy_addr)); proxy_addr.sin_family = AF_INET; proxy_addr.sin_port = htons(1080); proxy_addr.sin_addr.s_addr = INADDR_ANY; if (bind(proxy_sock, (struct sockaddr )&proxy_addr, sizeof(proxy_addr)) cerr return 1; } // 开始监听传入连接 if (listen(proxy_sock, 5) cerr return 1; } while (true) { // 接受传入连接 sockaddr_in client_addr; socklen_t client_addr_len = sizeof(client_addr); int client_sock = accept(proxy_sock, (struct sockaddr )&client_addr, &client_addr_len); if (client_sock cerr continue; } // 读取客户端数据，获取目标地址和端口 char buf[1024]; int n = recv(client_sock, buf, sizeof(buf), 0); if (n cerr close(client_sock); continue; } string addr(buf, n); // 创建一个线程来处理合作伙伴连接 thread t(proxy_thread, client_sock, addr); t.detach(); } // 关闭合作伙伴服务器套接字 close(proxy_sock); return 0; } ```