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

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

samples/sockmap/sockmap_user.c

+39 −16
Original line number Diff line number Diff line
@@ -23,6 +23,7 @@ 
#include <stdbool.h>

#include <signal.h>

#include <fcntl.h>

#include <sys/wait.h>



#include <sys/time.h>

#include <sys/types.h>
@@ -195,7 +196,7 @@ static int msg_loop(int fd, int iov_count, int iov_length, int cnt, 
{

	struct msghdr msg = {0};

	struct iovec *iov;

	int i, flags = 0;

	int i, flags = MSG_NOSIGNAL;



	iov = calloc(iov_count, sizeof(struct iovec));

	if (!iov)
@@ -275,25 +276,47 @@ out_errno: 


static int sendmsg_test(int iov_count, int iov_buf, int cnt, int verbose)

{

	int txpid, rxpid, err = 0;

	struct msg_stats s = {0};

	int err;

	int status;



	err = msg_loop(c1, iov_count, iov_buf, cnt, &s, true);

	if (err) {

		fprintf(stderr,

			"msg_loop_tx: iov_count %i iov_buf %i cnt %i err %i\n",

			iov_count, iov_buf, cnt, err);

		return err;

	}

	errno = 0;



	rxpid = fork();

	if (rxpid == 0) {

		err = msg_loop(p2, iov_count, iov_buf, cnt, &s, false);

		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;

	}



	fprintf(stdout, "sendmsg: TX_bytes %zu RX_bytes %zu\n",

	txpid = fork();

	if (txpid == 0) {

		err = msg_loop(c1, iov_count, iov_buf, cnt, &s, true);

		if (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;

	}



	assert(waitpid(rxpid, &status, 0) == rxpid);

	assert(waitpid(txpid, &status, 0) == txpid);

	return err;

}

CRA Git | Maintained and supported by SUSTech CRA and CCSE