Merge tag 'for-linus-5.14-rc1-tag' of git://git.kernel.org/pub/scm/linux/kernel/git...
[linux-2.6-microblaze.git] / lib / iov_iter.c
index 74e20d6..e23123a 100644 (file)
 #define PIPE_PARANOIA /* for now */
 
 /* covers iovec and kvec alike */
-#define iterate_iovec(i, n, __v, __p, skip, STEP) {            \
-       size_t left;                                            \
-       size_t wanted = n;                                      \
+#define iterate_iovec(i, n, base, len, off, __p, STEP) {       \
+       size_t off = 0;                                         \
+       size_t skip = i->iov_offset;                            \
        do {                                                    \
-               __v.iov_len = min(n, __p->iov_len - skip);      \
-               if (likely(__v.iov_len)) {                      \
-                       __v.iov_base = __p->iov_base + skip;    \
-                       left = (STEP);                          \
-                       __v.iov_len -= left;                    \
-                       skip += __v.iov_len;                    \
-                       n -= __v.iov_len;                       \
+               len = min(n, __p->iov_len - skip);              \
+               if (likely(len)) {                              \
+                       base = __p->iov_base + skip;            \
+                       len -= (STEP);                          \
+                       off += len;                             \
+                       skip += len;                            \
+                       n -= len;                               \
                        if (skip < __p->iov_len)                \
                                break;                          \
                }                                               \
                __p++;                                          \
                skip = 0;                                       \
        } while (n);                                            \
-       n = wanted - n;                                         \
+       i->iov_offset = skip;                                   \
+       n = off;                                                \
 }
 
-#define iterate_bvec(i, n, __v, p, skip, STEP) {               \
-       size_t wanted = n;                                      \
+#define iterate_bvec(i, n, base, len, off, p, STEP) {          \
+       size_t off = 0;                                         \
+       unsigned skip = i->iov_offset;                          \
        while (n) {                                             \
                unsigned offset = p->bv_offset + skip;          \
                unsigned left;                                  \
                void *kaddr = kmap_local_page(p->bv_page +      \
                                        offset / PAGE_SIZE);    \
-               __v.iov_base = kaddr + offset % PAGE_SIZE;      \
-               __v.iov_len = min(min(n, p->bv_len - skip),     \
+               base = kaddr + offset % PAGE_SIZE;              \
+               len = min(min(n, (size_t)(p->bv_len - skip)),   \
                     (size_t)(PAGE_SIZE - offset % PAGE_SIZE)); \
                left = (STEP);                                  \
                kunmap_local(kaddr);                            \
-               __v.iov_len -= left;                            \
-               skip += __v.iov_len;                            \
+               len -= left;                                    \
+               off += len;                                     \
+               skip += len;                                    \
                if (skip == p->bv_len) {                        \
                        skip = 0;                               \
                        p++;                                    \
                }                                               \
-               n -= __v.iov_len;                               \
+               n -= len;                                       \
                if (left)                                       \
                        break;                                  \
        }                                                       \
-       n = wanted - n;                                         \
+       i->iov_offset = skip;                                   \
+       n = off;                                                \
 }
 
-#define iterate_xarray(i, n, __v, skip, STEP) {                \
+#define iterate_xarray(i, n, base, len, __off, STEP) {         \
        __label__ __out;                                        \
+       size_t __off = 0;                                       \
        struct page *head = NULL;                               \
-       size_t wanted = n, seg, offset;                         \
-       loff_t start = i->xarray_start + skip;                  \
-       pgoff_t index = start >> PAGE_SHIFT;                    \
+       loff_t start = i->xarray_start + i->iov_offset;         \
+       unsigned offset = start % PAGE_SIZE;                    \
+       pgoff_t index = start / PAGE_SIZE;                      \
        int j;                                                  \
                                                                \
        XA_STATE(xas, i->xarray, index);                        \
                                                                \
-       rcu_read_lock();                                                \
-       xas_for_each(&xas, head, ULONG_MAX) {                           \
-               unsigned left;                                          \
-               if (xas_retry(&xas, head))                              \
-                       continue;                                       \
-               if (WARN_ON(xa_is_value(head)))                         \
-                       break;                                          \
-               if (WARN_ON(PageHuge(head)))                            \
-                       break;                                          \
+       rcu_read_lock();                                        \
+       xas_for_each(&xas, head, ULONG_MAX) {                   \
+               unsigned left;                                  \
+               if (xas_retry(&xas, head))                      \
+                       continue;                               \
+               if (WARN_ON(xa_is_value(head)))                 \
+                       break;                                  \
+               if (WARN_ON(PageHuge(head)))                    \
+                       break;                                  \
                for (j = (head->index < index) ? index - head->index : 0; \
-                    j < thp_nr_pages(head); j++) {                     \
+                    j < thp_nr_pages(head); j++) {             \
                        void *kaddr = kmap_local_page(head + j);        \
-                       offset = (i->xarray_start + skip) % PAGE_SIZE;  \
-                       __v.iov_base = kaddr + offset;                  \
-                       seg = PAGE_SIZE - offset;                       \
-                       __v.iov_len = min(n, seg);                      \
-                       left = (STEP);                                  \
-                       kunmap_local(kaddr);                            \
-                       __v.iov_len -= left;                            \
-                       n -= __v.iov_len;                               \
-                       skip += __v.iov_len;                            \
-                       if (left || n == 0)                             \
-                               goto __out;                             \
-               }                                                       \
+                       base = kaddr + offset;                  \
+                       len = PAGE_SIZE - offset;               \
+                       len = min(n, len);                      \
+                       left = (STEP);                          \
+                       kunmap_local(kaddr);                    \
+                       len -= left;                            \
+                       __off += len;                           \
+                       n -= len;                               \
+                       if (left || n == 0)                     \
+                               goto __out;                     \
+                       offset = 0;                             \
+               }                                               \
        }                                                       \
 __out:                                                         \
        rcu_read_unlock();                                      \
-       n = wanted - n;                                         \
+       i->iov_offset += __off;                                         \
+       n = __off;                                              \
 }
 
-#define __iterate_and_advance(i, n, v, I, K) {                 \
+#define __iterate_and_advance(i, n, base, len, off, I, K) {    \
        if (unlikely(i->count < n))                             \
                n = i->count;                                   \
        if (likely(n)) {                                        \
-               size_t skip = i->iov_offset;                    \
                if (likely(iter_is_iovec(i))) {                 \
                        const struct iovec *iov = i->iov;       \
-                       struct iovec v;                         \
-                       iterate_iovec(i, n, v, iov, skip, (I))  \
+                       void __user *base;                      \
+                       size_t len;                             \
+                       iterate_iovec(i, n, base, len, off,     \
+                                               iov, (I))       \
                        i->nr_segs -= iov - i->iov;             \
                        i->iov = iov;                           \
                } else if (iov_iter_is_bvec(i)) {               \
                        const struct bio_vec *bvec = i->bvec;   \
-                       struct kvec v;                          \
-                       iterate_bvec(i, n, v, bvec, skip, (K))  \
+                       void *base;                             \
+                       size_t len;                             \
+                       iterate_bvec(i, n, base, len, off,      \
+                                               bvec, (K))      \
                        i->nr_segs -= bvec - i->bvec;           \
                        i->bvec = bvec;                         \
                } else if (iov_iter_is_kvec(i)) {               \
                        const struct kvec *kvec = i->kvec;      \
-                       struct kvec v;                          \
-                       iterate_iovec(i, n, v, kvec, skip, (K)) \
+                       void *base;                             \
+                       size_t len;                             \
+                       iterate_iovec(i, n, base, len, off,     \
+                                               kvec, (K))      \
                        i->nr_segs -= kvec - i->kvec;           \
                        i->kvec = kvec;                         \
                } else if (iov_iter_is_xarray(i)) {             \
-                       struct kvec v;                          \
-                       iterate_xarray(i, n, v, skip, (K))      \
+                       void *base;                             \
+                       size_t len;                             \
+                       iterate_xarray(i, n, base, len, off,    \
+                                                       (K))    \
                }                                               \
                i->count -= n;                                  \
-               i->iov_offset = skip;                           \
        }                                                       \
 }
-#define iterate_and_advance(i, n, v, I, K) \
-       __iterate_and_advance(i, n, v, I, ((void)(K),0))
+#define iterate_and_advance(i, n, base, len, off, I, K) \
+       __iterate_and_advance(i, n, base, len, off, I, ((void)(K),0))
 
 static int copyout(void __user *to, const void *from, size_t n)
 {
@@ -453,7 +465,6 @@ void iov_iter_init(struct iov_iter *i, unsigned int direction,
                        size_t count)
 {
        WARN_ON(direction & ~(READ | WRITE));
-       WARN_ON_ONCE(uaccess_kernel());
        *i = (struct iov_iter) {
                .iter_type = ITER_IOVEC,
                .data_source = direction,
@@ -570,51 +581,45 @@ static __wsum csum_and_memcpy(void *to, const void *from, size_t len,
 }
 
 static size_t csum_and_copy_to_pipe_iter(const void *addr, size_t bytes,
-                                        struct csum_state *csstate,
-                                        struct iov_iter *i)
+                                        struct iov_iter *i, __wsum *sump)
 {
        struct pipe_inode_info *pipe = i->pipe;
        unsigned int p_mask = pipe->ring_size - 1;
-       __wsum sum = csstate->csum;
-       size_t off = csstate->off;
+       __wsum sum = *sump;
+       size_t off = 0;
        unsigned int i_head;
-       size_t n, r;
+       size_t r;
 
        if (!sanity(i))
                return 0;
 
-       bytes = n = push_pipe(i, bytes, &i_head, &r);
-       if (unlikely(!n))
-               return 0;
-       do {
-               size_t chunk = min_t(size_t, n, PAGE_SIZE - r);
-               char *p = kmap_atomic(pipe->bufs[i_head & p_mask].page);
-               sum = csum_and_memcpy(p + r, addr, chunk, sum, off);
-               kunmap_atomic(p);
+       bytes = push_pipe(i, bytes, &i_head, &r);
+       while (bytes) {
+               size_t chunk = min_t(size_t, bytes, PAGE_SIZE - r);
+               char *p = kmap_local_page(pipe->bufs[i_head & p_mask].page);
+               sum = csum_and_memcpy(p + r, addr + off, chunk, sum, off);
+               kunmap_local(p);
                i->head = i_head;
                i->iov_offset = r + chunk;
-               n -= chunk;
+               bytes -= chunk;
                off += chunk;
-               addr += chunk;
                r = 0;
                i_head++;
-       } while (n);
-       i->count -= bytes;
-       csstate->csum = sum;
-       csstate->off = off;
-       return bytes;
+       }
+       *sump = sum;
+       i->count -= off;
+       return off;
 }
 
 size_t _copy_to_iter(const void *addr, size_t bytes, struct iov_iter *i)
 {
-       const char *from = addr;
        if (unlikely(iov_iter_is_pipe(i)))
                return copy_pipe_to_iter(addr, bytes, i);
        if (iter_is_iovec(i))
                might_fault();
-       iterate_and_advance(i, bytes, v,
-               copyout(v.iov_base, (from += v.iov_len) - v.iov_len, v.iov_len),
-               memcpy(v.iov_base, (from += v.iov_len) - v.iov_len, v.iov_len)
+       iterate_and_advance(i, bytes, base, len, off,
+               copyout(base, addr + off, len),
+               memcpy(base, addr + off, len)
        )
 
        return bytes;
@@ -631,19 +636,6 @@ static int copyout_mc(void __user *to, const void *from, size_t n)
        return n;
 }
 
-static unsigned long copy_mc_to_page(struct page *page, size_t offset,
-               const char *from, size_t len)
-{
-       unsigned long ret;
-       char *to;
-
-       to = kmap_atomic(page);
-       ret = copy_mc_to_kernel(to + offset, from, len);
-       kunmap_atomic(to);
-
-       return ret;
-}
-
 static size_t copy_mc_pipe_to_iter(const void *addr, size_t bytes,
                                struct iov_iter *i)
 {
@@ -655,25 +647,23 @@ static size_t copy_mc_pipe_to_iter(const void *addr, size_t bytes,
        if (!sanity(i))
                return 0;
 
-       bytes = n = push_pipe(i, bytes, &i_head, &off);
-       if (unlikely(!n))
-               return 0;
-       do {
+       n = push_pipe(i, bytes, &i_head, &off);
+       while (n) {
                size_t chunk = min_t(size_t, n, PAGE_SIZE - off);
+               char *p = kmap_local_page(pipe->bufs[i_head & p_mask].page);
                unsigned long rem;
-
-               rem = copy_mc_to_page(pipe->bufs[i_head & p_mask].page,
-                                           off, addr, chunk);
+               rem = copy_mc_to_kernel(p + off, addr + xfer, chunk);
+               chunk -= rem;
+               kunmap_local(p);
                i->head = i_head;
-               i->iov_offset = off + chunk - rem;
-               xfer += chunk - rem;
+               i->iov_offset = off + chunk;
+               xfer += chunk;
                if (rem)
                        break;
                n -= chunk;
-               addr += chunk;
                off = 0;
                i_head++;
-       } while (n);
+       }
        i->count -= xfer;
        return xfer;
 }
@@ -703,17 +693,13 @@ static size_t copy_mc_pipe_to_iter(const void *addr, size_t bytes,
  */
 size_t _copy_mc_to_iter(const void *addr, size_t bytes, struct iov_iter *i)
 {
-       const char *from = addr;
-
        if (unlikely(iov_iter_is_pipe(i)))
                return copy_mc_pipe_to_iter(addr, bytes, i);
        if (iter_is_iovec(i))
                might_fault();
-       __iterate_and_advance(i, bytes, v,
-               copyout_mc(v.iov_base, (from += v.iov_len) - v.iov_len,
-                          v.iov_len),
-               copy_mc_to_kernel(v.iov_base, (from += v.iov_len)
-                                       - v.iov_len, v.iov_len)
+       __iterate_and_advance(i, bytes, base, len, off,
+               copyout_mc(base, addr + off, len),
+               copy_mc_to_kernel(base, addr + off, len)
        )
 
        return bytes;
@@ -723,16 +709,15 @@ EXPORT_SYMBOL_GPL(_copy_mc_to_iter);
 
 size_t _copy_from_iter(void *addr, size_t bytes, struct iov_iter *i)
 {
-       char *to = addr;
        if (unlikely(iov_iter_is_pipe(i))) {
                WARN_ON(1);
                return 0;
        }
        if (iter_is_iovec(i))
                might_fault();
-       iterate_and_advance(i, bytes, v,
-               copyin((to += v.iov_len) - v.iov_len, v.iov_base, v.iov_len),
-               memcpy((to += v.iov_len) - v.iov_len, v.iov_base, v.iov_len)
+       iterate_and_advance(i, bytes, base, len, off,
+               copyin(addr + off, base, len),
+               memcpy(addr + off, base, len)
        )
 
        return bytes;
@@ -741,15 +726,13 @@ EXPORT_SYMBOL(_copy_from_iter);
 
 size_t _copy_from_iter_nocache(void *addr, size_t bytes, struct iov_iter *i)
 {
-       char *to = addr;
        if (unlikely(iov_iter_is_pipe(i))) {
                WARN_ON(1);
                return 0;
        }
-       iterate_and_advance(i, bytes, v,
-               __copy_from_user_inatomic_nocache((to += v.iov_len) - v.iov_len,
-                                        v.iov_base, v.iov_len),
-               memcpy((to += v.iov_len) - v.iov_len, v.iov_base, v.iov_len)
+       iterate_and_advance(i, bytes, base, len, off,
+               __copy_from_user_inatomic_nocache(addr + off, base, len),
+               memcpy(addr + off, base, len)
        )
 
        return bytes;
@@ -773,16 +756,13 @@ EXPORT_SYMBOL(_copy_from_iter_nocache);
  */
 size_t _copy_from_iter_flushcache(void *addr, size_t bytes, struct iov_iter *i)
 {
-       char *to = addr;
        if (unlikely(iov_iter_is_pipe(i))) {
                WARN_ON(1);
                return 0;
        }
-       iterate_and_advance(i, bytes, v,
-               __copy_from_user_flushcache((to += v.iov_len) - v.iov_len,
-                                        v.iov_base, v.iov_len),
-               memcpy_flushcache((to += v.iov_len) - v.iov_len, v.iov_base,
-                       v.iov_len)
+       iterate_and_advance(i, bytes, base, len, off,
+               __copy_from_user_flushcache(addr + off, base, len),
+               memcpy_flushcache(addr + off, base, len)
        )
 
        return bytes;
@@ -820,9 +800,9 @@ static size_t __copy_page_to_iter(struct page *page, size_t offset, size_t bytes
        if (likely(iter_is_iovec(i)))
                return copy_page_to_iter_iovec(page, offset, bytes, i);
        if (iov_iter_is_bvec(i) || iov_iter_is_kvec(i) || iov_iter_is_xarray(i)) {
-               void *kaddr = kmap_atomic(page);
-               size_t wanted = copy_to_iter(kaddr + offset, bytes, i);
-               kunmap_atomic(kaddr);
+               void *kaddr = kmap_local_page(page);
+               size_t wanted = _copy_to_iter(kaddr + offset, bytes, i);
+               kunmap_local(kaddr);
                return wanted;
        }
        if (iov_iter_is_pipe(i))
@@ -870,9 +850,9 @@ size_t copy_page_from_iter(struct page *page, size_t offset, size_t bytes,
        if (likely(iter_is_iovec(i)))
                return copy_page_from_iter_iovec(page, offset, bytes, i);
        if (iov_iter_is_bvec(i) || iov_iter_is_kvec(i) || iov_iter_is_xarray(i)) {
-               void *kaddr = kmap_atomic(page);
+               void *kaddr = kmap_local_page(page);
                size_t wanted = _copy_from_iter(kaddr + offset, bytes, i);
-               kunmap_atomic(kaddr);
+               kunmap_local(kaddr);
                return wanted;
        }
        WARN_ON(1);
@@ -896,7 +876,9 @@ static size_t pipe_zero(size_t bytes, struct iov_iter *i)
 
        do {
                size_t chunk = min_t(size_t, n, PAGE_SIZE - off);
-               memzero_page(pipe->bufs[i_head & p_mask].page, off, chunk);
+               char *p = kmap_local_page(pipe->bufs[i_head & p_mask].page);
+               memset(p + off, 0, chunk);
+               kunmap_local(p);
                i->head = i_head;
                i->iov_offset = off + chunk;
                n -= chunk;
@@ -911,9 +893,9 @@ size_t iov_iter_zero(size_t bytes, struct iov_iter *i)
 {
        if (unlikely(iov_iter_is_pipe(i)))
                return pipe_zero(bytes, i);
-       iterate_and_advance(i, bytes, v,
-               clear_user(v.iov_base, v.iov_len),
-               memset(v.iov_base, 0, v.iov_len)
+       iterate_and_advance(i, bytes, base, len, count,
+               clear_user(base, len),
+               memset(base, 0, len)
        )
 
        return bytes;
@@ -933,9 +915,9 @@ size_t copy_page_from_iter_atomic(struct page *page, unsigned offset, size_t byt
                WARN_ON(1);
                return 0;
        }
-       iterate_and_advance(i, bytes, v,
-               copyin((p += v.iov_len) - v.iov_len, v.iov_base, v.iov_len),
-               memcpy((p += v.iov_len) - v.iov_len, v.iov_base, v.iov_len)
+       iterate_and_advance(i, bytes, base, len, off,
+               copyin(p + off, base, len),
+               memcpy(p + off, base, len)
        )
        kunmap_atomic(kaddr);
        return bytes;
@@ -1656,28 +1638,18 @@ EXPORT_SYMBOL(iov_iter_get_pages_alloc);
 size_t csum_and_copy_from_iter(void *addr, size_t bytes, __wsum *csum,
                               struct iov_iter *i)
 {
-       char *to = addr;
        __wsum sum, next;
-       size_t off = 0;
        sum = *csum;
        if (unlikely(iov_iter_is_pipe(i) || iov_iter_is_discard(i))) {
                WARN_ON(1);
                return 0;
        }
-       iterate_and_advance(i, bytes, v, ({
-               next = csum_and_copy_from_user(v.iov_base,
-                                              (to += v.iov_len) - v.iov_len,
-                                              v.iov_len);
-               if (next) {
-                       sum = csum_block_add(sum, next, off);
-                       off += v.iov_len;
-               }
-               next ? 0 : v.iov_len;
+       iterate_and_advance(i, bytes, base, len, off, ({
+               next = csum_and_copy_from_user(base, addr + off, len);
+               sum = csum_block_add(sum, next, off);
+               next ? 0 : len;
        }), ({
-               sum = csum_and_memcpy((to += v.iov_len) - v.iov_len,
-                                     v.iov_base, v.iov_len,
-                                     sum, off);
-               off += v.iov_len;
+               sum = csum_and_memcpy(addr + off, base, len, sum, off);
        })
        )
        *csum = sum;
@@ -1689,33 +1661,22 @@ size_t csum_and_copy_to_iter(const void *addr, size_t bytes, void *_csstate,
                             struct iov_iter *i)
 {
        struct csum_state *csstate = _csstate;
-       const char *from = addr;
        __wsum sum, next;
-       size_t off;
-
-       if (unlikely(iov_iter_is_pipe(i)))
-               return csum_and_copy_to_pipe_iter(addr, bytes, _csstate, i);
 
-       sum = csum_shift(csstate->csum, csstate->off);
-       off = 0;
        if (unlikely(iov_iter_is_discard(i))) {
                WARN_ON(1);     /* for now */
                return 0;
        }
-       iterate_and_advance(i, bytes, v, ({
-               next = csum_and_copy_to_user((from += v.iov_len) - v.iov_len,
-                                            v.iov_base,
-                                            v.iov_len);
-               if (next) {
-                       sum = csum_block_add(sum, next, off);
-                       off += v.iov_len;
-               }
-               next ? 0 : v.iov_len;
+
+       sum = csum_shift(csstate->csum, csstate->off);
+       if (unlikely(iov_iter_is_pipe(i)))
+               bytes = csum_and_copy_to_pipe_iter(addr, bytes, i, &sum);
+       else iterate_and_advance(i, bytes, base, len, off, ({
+               next = csum_and_copy_to_user(addr + off, base, len);
+               sum = csum_block_add(sum, next, off);
+               next ? 0 : len;
        }), ({
-               sum = csum_and_memcpy(v.iov_base,
-                                    (from += v.iov_len) - v.iov_len,
-                                    v.iov_len, sum, off);
-               off += v.iov_len;
+               sum = csum_and_memcpy(base, addr + off, len, sum, off);
        })
        )
        csstate->csum = csum_shift(sum, csstate->off);