iov_iter: Introduce nofault flag to disable page faults

Introduce a new nofault flag to indicate to iov_iter_get_pages not to
fault in user pages.

This is implemented by passing the FOLL_NOFAULT flag to get_user_pages,
which causes get_user_pages to fail when it would otherwise fault in a
page. We'll use the ->nofault flag to prevent iomap_dio_rw from faulting
in pages when page faults are not allowed.

Signed-off-by: Andreas Gruenbacher <agruenba@redhat.com>
This commit is contained in:
Andreas Gruenbacher 2021-07-12 12:06:14 +02:00
parent 55b8fe703b
commit 3337ab08d0
2 changed files with 16 additions and 5 deletions

View File

@ -35,6 +35,7 @@ struct iov_iter_state {
struct iov_iter { struct iov_iter {
u8 iter_type; u8 iter_type;
bool nofault;
bool data_source; bool data_source;
size_t iov_offset; size_t iov_offset;
size_t count; size_t count;

View File

@ -513,6 +513,7 @@ void iov_iter_init(struct iov_iter *i, unsigned int direction,
WARN_ON(direction & ~(READ | WRITE)); WARN_ON(direction & ~(READ | WRITE));
*i = (struct iov_iter) { *i = (struct iov_iter) {
.iter_type = ITER_IOVEC, .iter_type = ITER_IOVEC,
.nofault = false,
.data_source = direction, .data_source = direction,
.iov = iov, .iov = iov,
.nr_segs = nr_segs, .nr_segs = nr_segs,
@ -1527,13 +1528,17 @@ ssize_t iov_iter_get_pages(struct iov_iter *i,
return 0; return 0;
if (likely(iter_is_iovec(i))) { if (likely(iter_is_iovec(i))) {
unsigned int gup_flags = 0;
unsigned long addr; unsigned long addr;
if (iov_iter_rw(i) != WRITE)
gup_flags |= FOLL_WRITE;
if (i->nofault)
gup_flags |= FOLL_NOFAULT;
addr = first_iovec_segment(i, &len, start, maxsize, maxpages); addr = first_iovec_segment(i, &len, start, maxsize, maxpages);
n = DIV_ROUND_UP(len, PAGE_SIZE); n = DIV_ROUND_UP(len, PAGE_SIZE);
res = get_user_pages_fast(addr, n, res = get_user_pages_fast(addr, n, gup_flags, pages);
iov_iter_rw(i) != WRITE ? FOLL_WRITE : 0,
pages);
if (unlikely(res <= 0)) if (unlikely(res <= 0))
return res; return res;
return (res == n ? len : res * PAGE_SIZE) - *start; return (res == n ? len : res * PAGE_SIZE) - *start;
@ -1649,15 +1654,20 @@ ssize_t iov_iter_get_pages_alloc(struct iov_iter *i,
return 0; return 0;
if (likely(iter_is_iovec(i))) { if (likely(iter_is_iovec(i))) {
unsigned int gup_flags = 0;
unsigned long addr; unsigned long addr;
if (iov_iter_rw(i) != WRITE)
gup_flags |= FOLL_WRITE;
if (i->nofault)
gup_flags |= FOLL_NOFAULT;
addr = first_iovec_segment(i, &len, start, maxsize, ~0U); addr = first_iovec_segment(i, &len, start, maxsize, ~0U);
n = DIV_ROUND_UP(len, PAGE_SIZE); n = DIV_ROUND_UP(len, PAGE_SIZE);
p = get_pages_array(n); p = get_pages_array(n);
if (!p) if (!p)
return -ENOMEM; return -ENOMEM;
res = get_user_pages_fast(addr, n, res = get_user_pages_fast(addr, n, gup_flags, p);
iov_iter_rw(i) != WRITE ? FOLL_WRITE : 0, p);
if (unlikely(res <= 0)) { if (unlikely(res <= 0)) {
kvfree(p); kvfree(p);
*pages = NULL; *pages = NULL;