io_uring/net: io_async_msghdr caches for sendzc

We already keep io_async_msghdr caches for normal send/recv requests,
use them also for zerocopy send.

Signed-off-by: Pavel Begunkov <asml.silence@gmail.com>
Link: https://lore.kernel.org/r/42fa615b6e0be25f47a685c35d7b5e4f1b03d348.1662639236.git.asml.silence@gmail.com
Signed-off-by: Jens Axboe <axboe@kernel.dk>

diff --git a/io_uring/net.c b/io_uring/net.c
index 12412ac..07f6b9e 100644 (file)
--- a/io_uring/net.c
+++ b/io_uring/net.c
@@ -196,10 +196,9 @@ int io_sendzc_prep_async(struct io_kiocb *req)
 
        if (!zc->addr || req_has_async_data(req))
                return 0;
-       if (io_alloc_async_data(req))
+       io = io_msg_alloc_async_prep(req);
+       if (!io)
                return -ENOMEM;
-
-       io = req->async_data;
        ret = move_addr_to_kernel(zc->addr, zc->addr_len, &io->addr);
        return ret;
 }
@@ -212,9 +211,9 @@ static int io_setup_async_addr(struct io_kiocb *req,
 
        if (!addr || req_has_async_data(req))
                return -EAGAIN;
-       if (io_alloc_async_data(req))
+       io = io_msg_alloc_async(req, issue_flags);
+       if (!io)
                return -ENOMEM;
-       io = req->async_data;
        memcpy(&io->addr, addr, sizeof(io->addr));
        return -EAGAIN;
 }