1 | // SPDX-License-Identifier: GPL-2.0 |
2 | #include <linux/kernel.h> |
3 | #include <linux/errno.h> |
4 | #include <linux/fs.h> |
5 | #include <linux/file.h> |
6 | #include <linux/mm.h> |
7 | #include <linux/slab.h> |
8 | #include <linux/namei.h> |
9 | #include <linux/io_uring.h> |
10 | |
11 | #include <uapi/linux/fadvise.h> |
12 | #include <uapi/linux/io_uring.h> |
13 | |
14 | #include "io_uring.h" |
15 | #include "advise.h" |
16 | |
17 | struct io_fadvise { |
18 | struct file *file; |
19 | u64 offset; |
20 | u32 len; |
21 | u32 advice; |
22 | }; |
23 | |
24 | struct io_madvise { |
25 | struct file *file; |
26 | u64 addr; |
27 | u32 len; |
28 | u32 advice; |
29 | }; |
30 | |
31 | int io_madvise_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) |
32 | { |
33 | #if defined(CONFIG_ADVISE_SYSCALLS) && defined(CONFIG_MMU) |
34 | struct io_madvise *ma = io_kiocb_to_cmd(req, struct io_madvise); |
35 | |
36 | if (sqe->buf_index || sqe->off || sqe->splice_fd_in) |
37 | return -EINVAL; |
38 | |
39 | ma->addr = READ_ONCE(sqe->addr); |
40 | ma->len = READ_ONCE(sqe->len); |
41 | ma->advice = READ_ONCE(sqe->fadvise_advice); |
42 | req->flags |= REQ_F_FORCE_ASYNC; |
43 | return 0; |
44 | #else |
45 | return -EOPNOTSUPP; |
46 | #endif |
47 | } |
48 | |
49 | int io_madvise(struct io_kiocb *req, unsigned int issue_flags) |
50 | { |
51 | #if defined(CONFIG_ADVISE_SYSCALLS) && defined(CONFIG_MMU) |
52 | struct io_madvise *ma = io_kiocb_to_cmd(req, struct io_madvise); |
53 | int ret; |
54 | |
55 | WARN_ON_ONCE(issue_flags & IO_URING_F_NONBLOCK); |
56 | |
57 | ret = do_madvise(current->mm, start: ma->addr, len_in: ma->len, behavior: ma->advice); |
58 | io_req_set_res(req, res: ret, cflags: 0); |
59 | return IOU_OK; |
60 | #else |
61 | return -EOPNOTSUPP; |
62 | #endif |
63 | } |
64 | |
65 | static bool io_fadvise_force_async(struct io_fadvise *fa) |
66 | { |
67 | switch (fa->advice) { |
68 | case POSIX_FADV_NORMAL: |
69 | case POSIX_FADV_RANDOM: |
70 | case POSIX_FADV_SEQUENTIAL: |
71 | return false; |
72 | default: |
73 | return true; |
74 | } |
75 | } |
76 | |
77 | int io_fadvise_prep(struct io_kiocb *req, const struct io_uring_sqe *sqe) |
78 | { |
79 | struct io_fadvise *fa = io_kiocb_to_cmd(req, struct io_fadvise); |
80 | |
81 | if (sqe->buf_index || sqe->addr || sqe->splice_fd_in) |
82 | return -EINVAL; |
83 | |
84 | fa->offset = READ_ONCE(sqe->off); |
85 | fa->len = READ_ONCE(sqe->len); |
86 | fa->advice = READ_ONCE(sqe->fadvise_advice); |
87 | if (io_fadvise_force_async(fa)) |
88 | req->flags |= REQ_F_FORCE_ASYNC; |
89 | return 0; |
90 | } |
91 | |
92 | int io_fadvise(struct io_kiocb *req, unsigned int issue_flags) |
93 | { |
94 | struct io_fadvise *fa = io_kiocb_to_cmd(req, struct io_fadvise); |
95 | int ret; |
96 | |
97 | WARN_ON_ONCE(issue_flags & IO_URING_F_NONBLOCK && io_fadvise_force_async(fa)); |
98 | |
99 | ret = vfs_fadvise(file: req->file, offset: fa->offset, len: fa->len, advice: fa->advice); |
100 | if (ret < 0) |
101 | req_set_fail(req); |
102 | io_req_set_res(req, res: ret, cflags: 0); |
103 | return IOU_OK; |
104 | } |
105 | |