summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--test/Makefile7
-rw-r--r--test/send_recvmsg.c164
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;
+}