io_uring/net: Avoid msghdr on op_connect/op_bind async data
[ Upstream commit 3979840cd858f30f43ea9f4e7f7f1f56de82d698 ] This fixes a memory leak due to the lack of the cleanup hook for the iovec. The stable backport differs from upstream by dropping the io_connect_bpf_populate hunk, which didn't exist at the time and by fixing the merge conflict due to the introduction of io_bind_file_create. Both IORING_OP_CONNECT and IORING_OP_BIND reuse the msghdr object just to store the sockaddr. Beyond allocating a much larger object than needed, msghdr can also wrap an iovec, which will be recycled unnecessarily. This uses the sockaddr directly. Cc: stable@vger.kernel.org Signed-off-by: Gabriel Krisman Bertazi <krisman@suse.de> Link: https://patch.msgid.link/20260602215327.1885109-2-krisman@suse.de Signed-off-by: Jens Axboe <axboe@kernel.dk> Signed-off-by: Gabriel Krisman Bertazi <krisman@suse.de> Signed-off-by: Sasha Levin <sashal@kernel.org>
This commit is contained in:
parent
c9acdc466e
commit
7e00cafa33
2 changed files with 20 additions and 20 deletions
|
|
@ -1733,7 +1733,7 @@ int io_socket(struct io_kiocb *req, unsigned int issue_flags)
|
|||
int io_connect_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe)
|
||||
{
|
||||
struct io_connect *conn = io_kiocb_to_cmd(req, struct io_connect);
|
||||
struct io_async_msghdr *io;
|
||||
struct sockaddr_storage *addr;
|
||||
|
||||
if (sqe->len || sqe->buf_index || sqe->rw_flags || sqe->splice_fd_in)
|
||||
return -EINVAL;
|
||||
|
|
@ -1742,17 +1742,17 @@ int io_connect_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe)
|
|||
conn->addr_len = READ_ONCE(sqe->addr2);
|
||||
conn->in_progress = conn->seen_econnaborted = false;
|
||||
|
||||
io = io_msg_alloc_async(req);
|
||||
if (unlikely(!io))
|
||||
addr = io_uring_alloc_async_data(NULL, req);
|
||||
if (unlikely(!addr))
|
||||
return -ENOMEM;
|
||||
|
||||
return move_addr_to_kernel(conn->addr, conn->addr_len, &io->addr);
|
||||
return move_addr_to_kernel(conn->addr, conn->addr_len, addr);
|
||||
}
|
||||
|
||||
int io_connect(struct io_kiocb *req, unsigned int issue_flags)
|
||||
{
|
||||
struct io_connect *connect = io_kiocb_to_cmd(req, struct io_connect);
|
||||
struct io_async_msghdr *io = req->async_data;
|
||||
struct sockaddr_storage *addr = req->async_data;
|
||||
unsigned file_flags;
|
||||
int ret;
|
||||
bool force_nonblock = issue_flags & IO_URING_F_NONBLOCK;
|
||||
|
|
@ -1766,8 +1766,7 @@ int io_connect(struct io_kiocb *req, unsigned int issue_flags)
|
|||
|
||||
file_flags = force_nonblock ? O_NONBLOCK : 0;
|
||||
|
||||
ret = __sys_connect_file(req->file, &io->addr, connect->addr_len,
|
||||
file_flags);
|
||||
ret = __sys_connect_file(req->file, addr, connect->addr_len, file_flags);
|
||||
if ((ret == -EAGAIN || ret == -EINPROGRESS || ret == -ECONNABORTED)
|
||||
&& force_nonblock) {
|
||||
if (ret == -EINPROGRESS) {
|
||||
|
|
@ -1796,7 +1795,6 @@ get_sock_err:
|
|||
out:
|
||||
if (ret < 0)
|
||||
req_set_fail(req);
|
||||
io_req_msg_cleanup(req, issue_flags);
|
||||
io_req_set_res(req, ret, 0);
|
||||
return IOU_COMPLETE;
|
||||
}
|
||||
|
|
@ -1806,15 +1804,15 @@ out:
|
|||
* which in turn end up in mnt_want_write() which will grab the fs
|
||||
* percpu start write sem. This can trigger a lockdep warning.
|
||||
*/
|
||||
static int io_bind_file_create(const struct io_async_msghdr *io, int addr_len)
|
||||
static int io_bind_file_create(const struct sockaddr_storage *addr, int addr_len)
|
||||
{
|
||||
const struct sockaddr_un *sun;
|
||||
|
||||
if (io->addr.ss_family != AF_UNIX)
|
||||
if (addr->ss_family != AF_UNIX)
|
||||
return 0;
|
||||
if (addr_len <= offsetof(struct sockaddr_un, sun_path))
|
||||
return 0;
|
||||
sun = (const struct sockaddr_un *) &io->addr;
|
||||
sun = (const struct sockaddr_un *) addr;
|
||||
return sun->sun_path[0] != '\0';
|
||||
}
|
||||
|
||||
|
|
@ -1822,7 +1820,7 @@ int io_bind_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe)
|
|||
{
|
||||
struct io_bind *bind = io_kiocb_to_cmd(req, struct io_bind);
|
||||
struct sockaddr __user *uaddr;
|
||||
struct io_async_msghdr *io;
|
||||
struct sockaddr_storage *addr;
|
||||
int ret;
|
||||
|
||||
if (sqe->len || sqe->buf_index || sqe->rw_flags || sqe->splice_fd_in)
|
||||
|
|
@ -1831,21 +1829,23 @@ int io_bind_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe)
|
|||
uaddr = u64_to_user_ptr(READ_ONCE(sqe->addr));
|
||||
bind->addr_len = READ_ONCE(sqe->addr2);
|
||||
|
||||
io = io_msg_alloc_async(req);
|
||||
if (unlikely(!io))
|
||||
addr = io_uring_alloc_async_data(NULL, req);
|
||||
if (unlikely(!addr))
|
||||
return -ENOMEM;
|
||||
ret = move_addr_to_kernel(uaddr, bind->addr_len, &io->addr);
|
||||
|
||||
ret = move_addr_to_kernel(uaddr, bind->addr_len, addr);
|
||||
if (unlikely(ret))
|
||||
return ret;
|
||||
if (io_bind_file_create(io, bind->addr_len))
|
||||
if (io_bind_file_create(addr, bind->addr_len))
|
||||
req->flags |= REQ_F_FORCE_ASYNC;
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
int io_bind(struct io_kiocb *req, unsigned int issue_flags)
|
||||
{
|
||||
struct io_bind *bind = io_kiocb_to_cmd(req, struct io_bind);
|
||||
struct io_async_msghdr *io = req->async_data;
|
||||
struct sockaddr_storage *addr = req->async_data;
|
||||
struct socket *sock;
|
||||
int ret;
|
||||
|
||||
|
|
@ -1853,7 +1853,7 @@ int io_bind(struct io_kiocb *req, unsigned int issue_flags)
|
|||
if (unlikely(!sock))
|
||||
return -ENOTSOCK;
|
||||
|
||||
ret = __sys_bind_socket(sock, &io->addr, bind->addr_len);
|
||||
ret = __sys_bind_socket(sock, addr, bind->addr_len);
|
||||
if (ret < 0)
|
||||
req_set_fail(req);
|
||||
io_req_set_res(req, ret, 0);
|
||||
|
|
|
|||
|
|
@ -203,7 +203,7 @@ const struct io_issue_def io_issue_defs[] = {
|
|||
.unbound_nonreg_file = 1,
|
||||
.pollout = 1,
|
||||
#if defined(CONFIG_NET)
|
||||
.async_size = sizeof(struct io_async_msghdr),
|
||||
.async_size = sizeof(struct sockaddr_storage),
|
||||
.prep = io_connect_prep,
|
||||
.issue = io_connect,
|
||||
#else
|
||||
|
|
@ -503,7 +503,7 @@ const struct io_issue_def io_issue_defs[] = {
|
|||
.needs_file = 1,
|
||||
.prep = io_bind_prep,
|
||||
.issue = io_bind,
|
||||
.async_size = sizeof(struct io_async_msghdr),
|
||||
.async_size = sizeof(struct sockaddr_storage),
|
||||
#else
|
||||
.prep = io_eopnotsupp_prep,
|
||||
#endif
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue