Add send/recvmsg test case

Signed-off-by: Jens Axboe <axboe@kernel.dk>

2 files changed, 169 insertions, 2 deletions

diff --git a/test/Makefile b/test/Makefile
index e4fe57d..b94e29e 100644
--- a/test/Makefile
+++ b/test/Makefile
@@ -2,14 +2,15 @@ CFLAGS ?= -g -O2 -Wall -D_GNU_SOURCE -L../src/
 
 all_targets += poll poll-cancel ring-leak fsync io_uring_setup io_uring_register \
 	       io_uring_enter nop sq-full cq-full 35fa71a030ca-test \
-		917257daa0fe-test b19062a56726-test eeed8b54e0df-test link
+		917257daa0fe-test b19062a56726-test eeed8b54e0df-test link \
+		send_recvmsg
 
 all: $(all_targets)
 
 test_srcs := poll.c poll-cancel.c ring-leak.c fsync.c io_uring_setup.c \
 	io_uring_register.c io_uring_enter.c nop.c sq-full.c cq-full.c \
 	35fa71a030ca-test.c 917257daa0fe-test.c b19062a56726-test.c \
-	eeed8b54e0df-test link
+	eeed8b54e0df-test.c link.c send_recvmsg.c
 
 test_objs := $(patsubst %.c,%.ol,$(test_srcs))
 
@@ -43,6 +44,8 @@ eeed8b54e0df-test: eeed8b54e0df-test.c
 	$(CC) $(CFLAGS) -o $@ eeed8b54e0df-test.c -luring
 link: link.c
 	$(CC) $(CFLAGS) -o $@ link.c -luring
+send_recvmsg: send_recvmsg.c
+	$(CC) $(CFLAGS) -o $@ send_recvmsg.c -luring
 
 clean:
 	rm -f $(all_targets) $(test_objs)
diff --git a/test/send_recvmsg.c b/test/send_recvmsg.c
new file mode 100644
index 0000000..900b0d9
--- /dev/null
+++ b/test/send_recvmsg.c
@@ -0,0 +1,164 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <arpa/inet.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+
+#include "../src/liburing.h"
+
+static char str[] = "This is a test of sendmsg and recvmsg over io_uring!";
+
+#define MAX_MSG	128
+
+#define PORT	10200
+#define HOST	"127.0.0.1"
+
+static int do_recvmsg(void)
+{
+        struct sockaddr_in saddr;
+        char buf[MAX_MSG + 1];
+        struct msghdr msg;
+        struct iovec iov;
+	struct io_uring ring;
+	struct io_uring_cqe *cqe;
+	struct io_uring_sqe *sqe;
+	char *recv_str;
+        int sockfd, ret;
+
+	ret = io_uring_queue_init(1, &ring, 0);
+	if (ret) {
+		printf("queue init fail\n");
+		return 1;
+	}
+
+        sockfd = socket(AF_INET, SOCK_DGRAM, 0);
+        memset(&saddr, 0, sizeof(saddr));
+        saddr.sin_family = AF_INET;
+        saddr.sin_addr.s_addr = htonl(INADDR_ANY);
+        saddr.sin_port = htons(PORT);
+        bind(sockfd, (struct sockaddr *)&saddr, sizeof(saddr));
+
+	memset(&msg, 0, sizeof(msg));
+        msg.msg_namelen = sizeof(struct sockaddr_in);
+
+	iov.iov_base = buf;
+	iov.iov_len = MAX_MSG;
+	msg.msg_iov = &iov;
+	msg.msg_iovlen = 1;
+
+	sqe = io_uring_get_sqe(&ring);
+	memset(sqe, 0, sizeof(*sqe));
+	sqe->opcode = IORING_OP_RECVMSG;
+	sqe->fd = sockfd;
+	sqe->addr = (uint64_t) &msg;
+	sqe->len = 1;
+
+	ret = io_uring_submit(&ring);
+	if (ret <= 0) {
+		printf("submit failed\n");
+		goto err;
+	}
+
+	ret = io_uring_wait_cqe(&ring, &cqe);
+	if (cqe->res < 0) {
+		printf("failed cqe: %d\n", cqe->res);
+		goto err;
+	}
+
+	if (cqe->res -1 != strlen(str)) {
+		printf("got wrong length\n");
+		goto err;
+	}
+
+        recv_str = msg.msg_iov[0].iov_base;
+	if (strcmp(str, recv_str)) {
+		printf("string mismatch\n");
+		goto err;
+	}
+
+	close(sockfd);
+	return 0;
+err:
+	close(sockfd);
+	return 1;
+}
+
+static int do_sendmsg(void)
+{
+	struct sockaddr_in saddr;
+	struct msghdr msg;
+	struct iovec iov;
+	struct io_uring ring;
+	struct io_uring_cqe *cqe;
+	struct io_uring_sqe *sqe;
+	int sockfd, ret;
+
+	ret = io_uring_queue_init(1, &ring, 0);
+	if (ret) {
+		printf("queue init fail\n");
+		return 1;
+	}
+
+	sockfd = socket(AF_INET, SOCK_DGRAM, 0);
+	memset(&saddr, 0, sizeof(saddr));
+	saddr.sin_family = AF_INET;
+	saddr.sin_port = htons(PORT);
+	inet_pton(AF_INET, HOST, &saddr.sin_addr);
+
+	memset(&msg, 0, sizeof(msg));
+	msg.msg_name = &saddr;
+	msg.msg_namelen = sizeof(struct sockaddr_in);
+
+	iov.iov_base = str;
+	iov.iov_len = sizeof(str);
+	msg.msg_iov = &iov;
+	msg.msg_iovlen = 1;
+
+	sqe = io_uring_get_sqe(&ring);
+	memset(sqe, 0, sizeof(*sqe));
+	sqe->opcode = IORING_OP_SENDMSG;
+	sqe->fd = sockfd;
+	sqe->addr = (uint64_t) &msg;
+	sqe->len = 1;
+
+	ret = io_uring_submit(&ring);
+	if (ret <= 0) {
+		printf("submit failed\n");
+		goto err;
+	}
+
+	ret = io_uring_wait_cqe(&ring, &cqe);
+	if (cqe->res < 0) {
+		printf("failed cqe: %d\n", cqe->res);
+		goto err;
+	}
+
+	close(sockfd);
+	return 0;
+err:
+	close(sockfd);
+	return 1;
+}
+
+int main(int argc, char *argv[])
+{
+	pid_t pid;
+	int ret;
+
+	pid = fork();
+	switch (pid) {
+	case -1:
+		printf("fork() failed\n");
+		return 1;
+	case 0:
+		ret = do_sendmsg();
+		break;
+	default:
+		ret = do_recvmsg();
+		break;
+	}
+
+	return ret;
+}