Commit d7d6437a authored by John Fastabend's avatar John Fastabend Committed by Daniel Borkmann

bpf: sockmap sample, use fork() for send and recv

Currently for SENDMSG tests first send completes then recv runs. This
does not work well for large data sizes and/or many iterations. So
fork the recv and send handler so that we run both send and recv. In
the future we can add a parameter to do more than a single fork of
tx/rx.

With this we can get many GBps of data which helps exercise the
sockmap code.
Signed-off-by: default avatarJohn Fastabend <john.fastabend@gmail.com>
Acked-by: default avatarMartin KaFai Lau <kafai@fb.com>
Signed-off-by: default avatarDaniel Borkmann <daniel@iogearbox.net>
parent eaf8c6ee
...@@ -23,6 +23,7 @@ ...@@ -23,6 +23,7 @@
#include <stdbool.h> #include <stdbool.h>
#include <signal.h> #include <signal.h>
#include <fcntl.h> #include <fcntl.h>
#include <sys/wait.h>
#include <sys/time.h> #include <sys/time.h>
#include <sys/types.h> #include <sys/types.h>
...@@ -195,7 +196,7 @@ static int msg_loop(int fd, int iov_count, int iov_length, int cnt, ...@@ -195,7 +196,7 @@ static int msg_loop(int fd, int iov_count, int iov_length, int cnt,
{ {
struct msghdr msg = {0}; struct msghdr msg = {0};
struct iovec *iov; struct iovec *iov;
int i, flags = 0; int i, flags = MSG_NOSIGNAL;
iov = calloc(iov_count, sizeof(struct iovec)); iov = calloc(iov_count, sizeof(struct iovec));
if (!iov) if (!iov)
...@@ -275,25 +276,47 @@ static int msg_loop(int fd, int iov_count, int iov_length, int cnt, ...@@ -275,25 +276,47 @@ static int msg_loop(int fd, int iov_count, int iov_length, int cnt,
static int sendmsg_test(int iov_count, int iov_buf, int cnt, int verbose) static int sendmsg_test(int iov_count, int iov_buf, int cnt, int verbose)
{ {
int txpid, rxpid, err = 0;
struct msg_stats s = {0}; struct msg_stats s = {0};
int err; int status;
err = msg_loop(c1, iov_count, iov_buf, cnt, &s, true); errno = 0;
if (err) {
fprintf(stderr, rxpid = fork();
"msg_loop_tx: iov_count %i iov_buf %i cnt %i err %i\n", if (rxpid == 0) {
iov_count, iov_buf, cnt, err); err = msg_loop(p2, iov_count, iov_buf, cnt, &s, false);
return err; if (err)
fprintf(stderr,
"msg_loop_rx: iov_count %i iov_buf %i cnt %i err %i\n",
iov_count, iov_buf, cnt, err);
fprintf(stdout, "rx_sendmsg: TX_bytes %zu RX_bytes %zu\n",
s.bytes_sent, s.bytes_recvd);
shutdown(p2, SHUT_RDWR);
shutdown(p1, SHUT_RDWR);
exit(1);
} else if (rxpid == -1) {
perror("msg_loop_rx: ");
return errno;
} }
err = msg_loop(p2, iov_count, iov_buf, cnt, &s, false); txpid = fork();
if (err) if (txpid == 0) {
fprintf(stderr, err = msg_loop(c1, iov_count, iov_buf, cnt, &s, true);
"msg_loop_rx: iov_count %i iov_buf %i cnt %i err %i\n", if (err)
iov_count, iov_buf, cnt, err); fprintf(stderr,
"msg_loop_tx: iov_count %i iov_buf %i cnt %i err %i\n",
iov_count, iov_buf, cnt, err);
fprintf(stdout, "tx_sendmsg: TX_bytes %zu RX_bytes %zu\n",
s.bytes_sent, s.bytes_recvd);
shutdown(c1, SHUT_RDWR);
exit(1);
} else if (txpid == -1) {
perror("msg_loop_tx: ");
return errno;
}
fprintf(stdout, "sendmsg: TX_bytes %zu RX_bytes %zu\n", assert(waitpid(rxpid, &status, 0) == rxpid);
s.bytes_sent, s.bytes_recvd); assert(waitpid(txpid, &status, 0) == txpid);
return err; return err;
} }
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment