io_uring: fix multishot accept ordering

Similar to multishot poll, drop multishot accept when CQE overflow occurs.

Signed-off-by: Dylan Yudaken <dylany@fb.com>
Link: https://lore.kernel.org/r/20220630091231.1456789-10-dylany@fb.com
Signed-off-by: Jens Axboe <axboe@kernel.dk>

diff --git a/io_uring/net.c b/io_uring/net.c
index 601955f..e1eaf90 100644 (file)
--- a/io_uring/net.c
+++ b/io_uring/net.c
@@ -656,11 +656,14 @@ retry:
                return IOU_OK;
        }
 
-       if (ret < 0)
-               return ret;
-       if (io_post_aux_cqe(ctx, req->cqe.user_data, ret, IORING_CQE_F_MORE, true))
+       if (ret >= 0 &&
+           io_post_aux_cqe(ctx, req->cqe.user_data, ret, IORING_CQE_F_MORE, false))
                goto retry;
-       return -ECANCELED;
+
+       io_req_set_res(req, ret, 0);
+       if (req->flags & REQ_F_POLLED)
+               return IOU_STOP_MULTISHOT;
+       return IOU_OK;
 }
 
 int io_socket_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe)