enospace

Signed-off-by: Pavel Begunkov <asml.silence@gmail.com>

Author: isilence

examples/zcrx.c

@@ -330,20 +330,38 @@ static void return_buffer(struct io_uring_zcrx_rq *rq_ring,
 	io_uring_smp_store_release(rq_ring->ktail, ++rq_ring->rq_tail);
 }
 
-static void process_recvzc(struct io_uring __attribute__((unused)) *ring,
+static void process_recvzc_error(struct io_uring *ring, int ret)
+{
+	if (ret == -ENOSPC) {
+		size_t left = 0;
+
+		if (cfg_size) {
+			left = cfg_size - received;
+			if (left == 0)
+				t_error(1, 0, "ENOSPC for a finished request");
+		}
+
+		add_recvzc(ring, connfd, left);
+		return;
+	}
+
+	if (ret != 0)
+		t_error(1, 0, "invalid final recvzc ret %i", ret);
+	if (cfg_size && received != cfg_size)
+		t_error(1, 0, "total receive size mismatch %lu / %lu",
+			received, cfg_size);
+	stop = true;
+}
+
+static void process_recvzc(struct io_uring *ring,
 			   struct io_uring_cqe *cqe)
 {
 	const struct io_uring_zcrx_cqe *rcqe;
 	uint64_t mask;
 	char *data;
 
 	if (!(cqe->flags & IORING_CQE_F_MORE)) {
-		if (!cfg_size || cqe->res != 0)
-			t_error(1, 0, "invalid final recvzc ret %i", cqe->res);
-		if (received != cfg_size)
-			t_error(1, 0, "total receive size mismatch %lu / %lu",
-				received, cfg_size);
-		stop = true;
+		process_recvzc_error(ring, cqe->res);
 		return;
 	}
 	if (cqe->res < 0)