Skip to content

test(vsock): add latency test #5603

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
aaron-ang wants to merge 1 commit into
base: main
Choose a base branch
from
+385 −7
Open

test(vsock): add latency test #5603

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
241 changes: 235 additions & 6 deletions tests/host_tools/vsock_helper.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,26 +10,39 @@
#include <sys/types.h>
#include <sys/select.h>
#include <sys/wait.h>
#include <sys/un.h>
#include <linux/vm_sockets.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <unistd.h>
#include <stdio.h>
#include <errno.h>
#include <time.h>


#define BUF_SIZE (16 * 1024)
#define SERVER_ACCEPT_BACKLOG 128


int print_usage() {
fprintf(stderr, "Usage: ./vsock-helper echo <cid> <port>\n");
fprintf(stderr, "Usage: ./vsock-helper <command> <args>\n");
fprintf(stderr, "\n");
fprintf(stderr, " echo connect to an echo server, listening on CID:port.\n");
fprintf(stderr, " STDIN will be piped through to the echo server, and\n");
fprintf(stderr, " data coming from the server will pe sent to STDOUT.\n");
fprintf(stderr, "Commands:\n");
fprintf(stderr, " echo <cid> <port>\n");
fprintf(stderr, " Connect to echo server at CID:port. Pipe STDIN to server,\n");
fprintf(stderr, " server response to STDOUT.\n");
fprintf(stderr, "\n");
fprintf(stderr, " ping <cid> <port> <count> <delay>\n");
fprintf(stderr, " Send <count> ping messages to echo server at CID:port.\n");
fprintf(stderr, " <delay> is the delay in seconds between pings (float).\n");
fprintf(stderr, " Prints RTT for each ping in microseconds.\n");
fprintf(stderr, "\n");
fprintf(stderr, " ping-uds <uds_path> <port> <count> <delay>\n");
fprintf(stderr, " Send <count> ping messages to echo server at <uds_path>:port.\n");
fprintf(stderr, " Uses Unix Domain Socket with Firecracker CONNECT protocol.\n");
fprintf(stderr, " <delay> is the delay in seconds between pings (float).\n");
fprintf(stderr, " Prints RTT for each ping in microseconds.\n");
fprintf(stderr, "\n");
return -1;
}
Expand Down Expand Up @@ -88,9 +101,195 @@ int run_echo(uint32_t cid, uint32_t port) {
}


int run_ping(uint32_t cid, uint32_t port, int count, double delay_sec) {

int sock = socket(AF_VSOCK, SOCK_STREAM, 0);
if (sock < 0) {
perror("socket()");
return -1;
}

struct sockaddr_vm vsock_addr = {
.svm_family = AF_VSOCK,
.svm_port = port,
.svm_cid = cid
};
if (connect(sock, (struct sockaddr*)&vsock_addr, sizeof(vsock_addr)) < 0) {
perror("connect()");
close(sock);
return -1;
}

char ping_msg[64];
memset(ping_msg, 'A', sizeof(ping_msg));

struct timespec delay_ts;
delay_ts.tv_sec = (time_t)delay_sec;
delay_ts.tv_nsec = (long)((delay_sec - (time_t)delay_sec) * 1000000000);

for (int seq = 1; seq <= count; seq++) {
struct timespec start, end;

if (clock_gettime(CLOCK_MONOTONIC, &start) < 0) {
perror("clock_gettime(start)");
close(sock);
return -1;
}

ssize_t sent = write(sock, ping_msg, sizeof(ping_msg));
if (sent != sizeof(ping_msg)) {
perror("write()");
close(sock);
return -1;
}

char pong_buf[64];
size_t total_received = 0;
while (total_received < sizeof(ping_msg)) {
ssize_t received = read(sock, pong_buf + total_received,
sizeof(ping_msg) - total_received);
if (received <= 0) {
perror("read()");
close(sock);
return -1;
}
total_received += received;
}

if (clock_gettime(CLOCK_MONOTONIC, &end) < 0) {
perror("clock_gettime(end)");
close(sock);
return -1;
}

long long start_us = start.tv_sec * 1000000LL + start.tv_nsec / 1000;
long long end_us = end.tv_sec * 1000000LL + end.tv_nsec / 1000;
long long rtt_us = end_us - start_us;

printf("rtt=%.3f us seq=%d\n", (double)rtt_us, seq);
fflush(stdout);

if (seq < count && delay_sec > 0) {
nanosleep(&delay_ts, NULL);
}
}

close(sock);
return 0;
}


int run_ping_uds(const char *uds_path, uint32_t port, int count, double delay_sec) {

int sock = socket(AF_UNIX, SOCK_STREAM, 0);
if (sock < 0) {
perror("socket()");
return -1;
}

struct sockaddr_un unix_addr;
memset(&unix_addr, 0, sizeof(unix_addr));
unix_addr.sun_family = AF_UNIX;
strncpy(unix_addr.sun_path, uds_path, sizeof(unix_addr.sun_path) - 1);

if (connect(sock, (struct sockaddr*)&unix_addr, sizeof(unix_addr)) < 0) {
perror("connect()");
close(sock);
return -1;
}

char connect_msg[64];
int connect_len = snprintf(connect_msg, sizeof(connect_msg), "CONNECT %u\n", port);
if (connect_len < 0 || connect_len >= sizeof(connect_msg)) {
fprintf(stderr, "Error formatting CONNECT message\n");
close(sock);
return -1;
}

ssize_t sent = write(sock, connect_msg, connect_len);
if (sent != connect_len) {
perror("write(CONNECT)");
close(sock);
return -1;
}

char ack_buf[32];
ssize_t ack_received = read(sock, ack_buf, sizeof(ack_buf) - 1);
if (ack_received <= 0) {
perror("read(ack)");
close(sock);
return -1;
}
ack_buf[ack_received] = '\0';

if (strncmp(ack_buf, "OK ", 3) != 0) {
fprintf(stderr, "Invalid acknowledgment: %s\n", ack_buf);
close(sock);
return -1;
}

char ping_msg[64];
memset(ping_msg, 'A', sizeof(ping_msg));

struct timespec delay_ts;
delay_ts.tv_sec = (time_t)delay_sec;
delay_ts.tv_nsec = (long)((delay_sec - (time_t)delay_sec) * 1000000000);

for (int seq = 1; seq <= count; seq++) {
struct timespec start, end;

if (clock_gettime(CLOCK_MONOTONIC, &start) < 0) {
perror("clock_gettime(start)");
close(sock);
return -1;
}

sent = write(sock, ping_msg, sizeof(ping_msg));
if (sent != sizeof(ping_msg)) {
perror("write()");
close(sock);
return -1;
}

char pong_buf[64];
size_t total_received = 0;
while (total_received < sizeof(ping_msg)) {
ssize_t received = read(sock, pong_buf + total_received,
sizeof(ping_msg) - total_received);
if (received <= 0) {
perror("read()");
close(sock);
return -1;
}
total_received += received;
}

if (clock_gettime(CLOCK_MONOTONIC, &end) < 0) {
perror("clock_gettime(end)");
close(sock);
return -1;
}

long long start_us = start.tv_sec * 1000000LL + start.tv_nsec / 1000;
long long end_us = end.tv_sec * 1000000LL + end.tv_nsec / 1000;
long long rtt_us = end_us - start_us;

printf("rtt=%.3f us seq=%d\n", (double)rtt_us, seq);
fflush(stdout);

if (seq < count && delay_sec > 0) {
nanosleep(&delay_ts, NULL);
}
}

close(sock);
return 0;
}


int main(int argc, char **argv) {

if (argc < 3) {
if (argc < 2) {
return print_usage();
}

Expand All @@ -106,5 +305,35 @@ int main(int argc, char **argv) {
return run_echo(cid, port);
}

if (strcmp(argv[1], "ping") == 0) {
if (argc != 6) {
return print_usage();
}
uint32_t cid = atoi(argv[2]);
uint32_t port = atoi(argv[3]);
int count = atoi(argv[4]);
double delay = atof(argv[5]);

if (!cid || !port || count <= 0 || delay < 0) {
return print_usage();
}
return run_ping(cid, port, count, delay);
}

if (strcmp(argv[1], "ping-uds") == 0) {
if (argc != 6) {
return print_usage();
}
const char *uds_path = argv[2];
uint32_t port = atoi(argv[3]);
int count = atoi(argv[4]);
double delay = atof(argv[5]);

if (!port || count <= 0 || delay < 0) {
return print_usage();
}
return run_ping_uds(uds_path, port, count, delay);
}

return print_usage();
}
Loading