mirror of
https://git.kernel.org/pub/scm/linux/kernel/git/stable/linux.git
synced 2024-12-29 17:25:38 +00:00
io_uring/net: harden multishot termination case for recv
[ Upstream commitc314094cb4
] If the recv returns zero, or an error, then it doesn't matter if more data has already been received for this buffer. A condition like that should terminate the multishot receive. Rather than pass in the collected return value, pass in whether to terminate or keep the recv going separately. Note that this isn't a bug right now, as the only way to get there is via setting MSG_WAITALL with multishot receive. And if an application does that, then -EINVAL is returned anyway. But it seems like an easy bug to introduce, so let's make it a bit more explicit. Link: https://github.com/axboe/liburing/issues/1246 Cc: stable@vger.kernel.org Fixes:b3fdea6ecb
("io_uring: multishot recv") Signed-off-by: Jens Axboe <axboe@kernel.dk> Signed-off-by: Sasha Levin <sashal@kernel.org>
This commit is contained in:
parent
fd7c4608ca
commit
5a306c8641
@ -893,6 +893,7 @@ int io_recv(struct io_kiocb *req, unsigned int issue_flags)
|
||||
int ret, min_ret = 0;
|
||||
bool force_nonblock = issue_flags & IO_URING_F_NONBLOCK;
|
||||
size_t len = sr->len;
|
||||
bool mshot_finished;
|
||||
|
||||
if (!(req->flags & REQ_F_POLLED) &&
|
||||
(sr->flags & IORING_RECVSEND_POLL_FIRST))
|
||||
@ -957,6 +958,7 @@ int io_recv(struct io_kiocb *req, unsigned int issue_flags)
|
||||
req_set_fail(req);
|
||||
}
|
||||
|
||||
mshot_finished = ret <= 0;
|
||||
if (ret > 0)
|
||||
ret += sr->done_io;
|
||||
else if (sr->done_io)
|
||||
@ -968,7 +970,7 @@ int io_recv(struct io_kiocb *req, unsigned int issue_flags)
|
||||
if (msg.msg_inq)
|
||||
cflags |= IORING_CQE_F_SOCK_NONEMPTY;
|
||||
|
||||
if (!io_recv_finish(req, &ret, cflags, ret <= 0, issue_flags))
|
||||
if (!io_recv_finish(req, &ret, cflags, mshot_finished, issue_flags))
|
||||
goto retry_multishot;
|
||||
|
||||
return ret;
|
||||
|
Loading…
Reference in New Issue
Block a user