Add SQ/CQ overflow tests

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

3 files changed, 134 insertions, 2 deletions

diff --git a/test/Makefile b/test/Makefile
index faf5fa6..63bf67a 100644
--- a/test/Makefile
+++ b/test/Makefile
@@ -1,12 +1,13 @@
 CFLAGS ?= -g -O2 -Wall -D_GNU_SOURCE -L../src/
 
 all_targets += io_uring-test io_uring-cp poll poll-cancel ring-leak fsync \
-	io_uring_setup io_uring_register io_uring_enter nop
+	io_uring_setup io_uring_register io_uring_enter nop sq-full cq-full
 
 all: $(all_targets)
 
 test_srcs := io_uring-test.c io_uring-cp.c poll.c poll-cancel.c ring-leak.c \
-	fsync.c io_uring_setup.c io_uring_register.c io_uring_enter.c nop.c
+	fsync.c io_uring_setup.c io_uring_register.c io_uring_enter.c nop.c \
+	sq-full.c cq-full.c
 
 test_objs := $(patsubst %.c,%.ol,$(test_srcs))
 
@@ -30,6 +31,10 @@ io_uring_enter: io_uring_enter.c
 	$(CC) $(CFLAGS) -o $@ io_uring_enter.c -luring
 nop: nop.c
 	$(CC) $(CFLAGS) -o $@ nop.c -luring
+sq-full: sq-full.c
+	$(CC) $(CFLAGS) -o $@ sq-full.c -luring
+cq-full: cq-full.c
+	$(CC) $(CFLAGS) -o $@ cq-full.c -luring
 
 clean:
 	rm -f $(all_targets) $(test_objs)
diff --git a/test/cq-full.c b/test/cq-full.c
new file mode 100644
index 0000000..426eb4a
--- /dev/null
+++ b/test/cq-full.c
@@ -0,0 +1,86 @@
+/*
+ * Description: test CQ ring overflow
+ *
+ */
+#include <errno.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <string.h>
+#include <fcntl.h>
+
+#include "../src/liburing.h"
+
+static int queue_n_nops(struct io_uring *ring, int n)
+{
+	struct io_uring_sqe *sqe;
+	int i, ret;
+
+	for (i = 0; i < n; i++) {
+		sqe = io_uring_get_sqe(ring);
+		if (!sqe) {
+			printf("get sqe failed\n");
+			goto err;
+		}
+
+		io_uring_prep_nop(sqe);
+	}
+
+	ret = io_uring_submit(ring);
+	if (ret < n) {
+		printf("Submitted only %d\n", ret);
+		goto err;
+	} else if (ret < 0) {
+		printf("sqe submit failed: %d\n", ret);
+		goto err;
+	}
+
+	return 0;
+err:
+	return 1;
+}
+
+int main(int argc, char *argv[])
+{
+	struct io_uring_cqe *cqe;
+	struct io_uring ring;
+	int i, ret;
+
+	ret = io_uring_queue_init(4, &ring, 0);
+	if (ret) {
+		printf("ring setup failed\n");
+		return 1;
+
+	}
+
+	if (queue_n_nops(&ring, 4))
+		goto err;
+	if (queue_n_nops(&ring, 4))
+		goto err;
+	if (queue_n_nops(&ring, 4))
+		goto err;
+
+	i = 0;
+	do {
+		ret = io_uring_get_completion(&ring, &cqe);
+		if (ret < 0) {
+			printf("wait completion %d\n", ret);
+			goto err;
+		}
+		if (!cqe)
+			break;
+		i++;
+	} while (1);
+
+	if (i != 8 || *ring.cq.koverflow != 4) {
+		printf("CQ overflow fail: %d completions, %u overflow\n", i,
+				*ring.cq.koverflow);
+		goto err;
+	}
+
+	io_uring_queue_exit(&ring);
+	return 0;
+err:
+	io_uring_queue_exit(&ring);
+	return 1;
+}
diff --git a/test/sq-full.c b/test/sq-full.c
new file mode 100644
index 0000000..5bf7f72
--- /dev/null
+++ b/test/sq-full.c
@@ -0,0 +1,41 @@
+/*
+ * Description: test SQ queue full condition
+ *
+ */
+#include <errno.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <string.h>
+#include <fcntl.h>
+
+#include "../src/liburing.h"
+
+int main(int argc, char *argv[])
+{
+	struct io_uring_sqe *sqe;
+	struct io_uring ring;
+	int ret, i;
+
+	ret = io_uring_queue_init(8, &ring, 0);
+	if (ret) {
+		printf("ring setup failed\n");
+		return 1;
+
+	}
+
+	i = 0;
+	while ((sqe = io_uring_get_sqe(&ring)) != NULL)
+		i++;
+
+	if (i != 8) {
+		printf("Got %d SQEs\n", i);
+		goto err;
+	}
+
+	io_uring_queue_exit(&ring);
+	return 0;
+err:
+	io_uring_queue_exit(&ring);
+	return 1;
+}