Split csum_and_copy_to_iter() by type.

Signed-off-by: David Howells <dhowe...@redhat.com>
---

 lib/iov_iter.c |   68 ++++++++++++++++++++++++++++++++++++++++----------------
 1 file changed, 48 insertions(+), 20 deletions(-)

diff --git a/lib/iov_iter.c b/lib/iov_iter.c
index 8820a9e72815..2f8019e3b09a 100644
--- a/lib/iov_iter.c
+++ b/lib/iov_iter.c
@@ -698,14 +698,15 @@ static __wsum csum_and_memcpy(void *to, const void *from, 
size_t len,
        return csum_block_add(sum, next, off);
 }
 
-static size_t csum_and_copy_to_pipe_iter(const void *addr, size_t bytes,
-                               __wsum *csum, struct iov_iter *i)
+static size_t pipe_csum_and_copy_to_iter(const void *addr, size_t bytes,
+                               void *csump, struct iov_iter *i)
 {
        struct pipe_inode_info *pipe = i->pipe;
        unsigned int p_mask = pipe->ring_size - 1;
        unsigned int i_head;
        size_t n, r;
        size_t off = 0;
+       __wsum *csum = csump;
        __wsum sum = *csum;
 
        if (!sanity(i))
@@ -1914,7 +1915,7 @@ static bool no_csum_and_copy_from_iter_full(void *addr, 
size_t bytes, __wsum *cs
        return false;
 }
 
-static size_t xxx_csum_and_copy_to_iter(const void *addr, size_t bytes, void 
*csump,
+static size_t iovec_csum_and_copy_to_iter(const void *addr, size_t bytes, void 
*csump,
                             struct iov_iter *i)
 {
        const char *from = addr;
@@ -1922,15 +1923,8 @@ static size_t xxx_csum_and_copy_to_iter(const void 
*addr, size_t bytes, void *cs
        __wsum sum, next;
        size_t off = 0;
 
-       if (unlikely(iov_iter_is_pipe(i)))
-               return csum_and_copy_to_pipe_iter(addr, bytes, csum, i);
-
        sum = *csum;
-       if (unlikely(iov_iter_is_discard(i))) {
-               WARN_ON(1);     /* for now */
-               return 0;
-       }
-       iterate_and_advance(i, bytes, v, ({
+       iterate_and_advance_iovec(i, bytes, v, ({
                next = csum_and_copy_to_user((from += v.iov_len) - v.iov_len,
                                             v.iov_base,
                                             v.iov_len);
@@ -1939,24 +1933,58 @@ static size_t xxx_csum_and_copy_to_iter(const void 
*addr, size_t bytes, void *cs
                        off += v.iov_len;
                }
                next ? 0 : v.iov_len;
-       }), ({
+       }));
+       *csum = sum;
+       return bytes;
+}
+
+static size_t bvec_csum_and_copy_to_iter(const void *addr, size_t bytes, void 
*csump,
+                            struct iov_iter *i)
+{
+       const char *from = addr;
+       __wsum *csum = csump;
+       __wsum sum;
+       size_t off = 0;
+
+       sum = *csum;
+       iterate_and_advance_bvec(i, bytes, v, ({
                char *p = kmap_atomic(v.bv_page);
                sum = csum_and_memcpy(p + v.bv_offset,
                                      (from += v.bv_len) - v.bv_len,
                                      v.bv_len, sum, off);
                kunmap_atomic(p);
                off += v.bv_len;
-       }),({
+       }));
+       *csum = sum;
+       return bytes;
+}
+
+static size_t kvec_csum_and_copy_to_iter(const void *addr, size_t bytes, void 
*csump,
+                            struct iov_iter *i)
+{
+       const char *from = addr;
+       __wsum *csum = csump;
+       __wsum sum;
+       size_t off = 0;
+
+       sum = *csum;
+       iterate_and_advance_kvec(i, bytes, v, ({
                sum = csum_and_memcpy(v.iov_base,
                                     (from += v.iov_len) - v.iov_len,
                                     v.iov_len, sum, off);
                off += v.iov_len;
-       })
-       )
+       }));
        *csum = sum;
        return bytes;
 }
 
+static size_t discard_csum_and_copy_to_iter(const void *addr, size_t bytes, 
void *csump,
+                            struct iov_iter *i)
+{
+       WARN_ON(1);     /* for now */
+       return 0;
+}
+
 size_t hash_and_copy_to_iter(const void *addr, size_t bytes, void *hashp,
                struct iov_iter *i)
 {
@@ -2256,7 +2284,7 @@ static const struct iov_iter_ops iovec_iter_ops = {
 #ifdef CONFIG_ARCH_HAS_COPY_MC
        .copy_mc_to_iter                = iovec_copy_mc_to_iter,
 #endif
-       .csum_and_copy_to_iter          = xxx_csum_and_copy_to_iter,
+       .csum_and_copy_to_iter          = iovec_csum_and_copy_to_iter,
        .csum_and_copy_from_iter        = iovec_csum_and_copy_from_iter,
        .csum_and_copy_from_iter_full   = iovec_csum_and_copy_from_iter_full,
 
@@ -2290,7 +2318,7 @@ static const struct iov_iter_ops kvec_iter_ops = {
 #ifdef CONFIG_ARCH_HAS_COPY_MC
        .copy_mc_to_iter                = kvec_copy_mc_to_iter,
 #endif
-       .csum_and_copy_to_iter          = xxx_csum_and_copy_to_iter,
+       .csum_and_copy_to_iter          = kvec_csum_and_copy_to_iter,
        .csum_and_copy_from_iter        = kvec_csum_and_copy_from_iter,
        .csum_and_copy_from_iter_full   = kvec_csum_and_copy_from_iter_full,
 
@@ -2324,7 +2352,7 @@ static const struct iov_iter_ops bvec_iter_ops = {
 #ifdef CONFIG_ARCH_HAS_COPY_MC
        .copy_mc_to_iter                = bvec_copy_mc_to_iter,
 #endif
-       .csum_and_copy_to_iter          = xxx_csum_and_copy_to_iter,
+       .csum_and_copy_to_iter          = bvec_csum_and_copy_to_iter,
        .csum_and_copy_from_iter        = bvec_csum_and_copy_from_iter,
        .csum_and_copy_from_iter_full   = bvec_csum_and_copy_from_iter_full,
 
@@ -2358,7 +2386,7 @@ static const struct iov_iter_ops pipe_iter_ops = {
 #ifdef CONFIG_ARCH_HAS_COPY_MC
        .copy_mc_to_iter                = pipe_copy_mc_to_iter,
 #endif
-       .csum_and_copy_to_iter          = xxx_csum_and_copy_to_iter,
+       .csum_and_copy_to_iter          = pipe_csum_and_copy_to_iter,
        .csum_and_copy_from_iter        = no_csum_and_copy_from_iter,
        .csum_and_copy_from_iter_full   = no_csum_and_copy_from_iter_full,
 
@@ -2392,7 +2420,7 @@ static const struct iov_iter_ops discard_iter_ops = {
 #ifdef CONFIG_ARCH_HAS_COPY_MC
        .copy_mc_to_iter                = discard_copy_to_iter,
 #endif
-       .csum_and_copy_to_iter          = xxx_csum_and_copy_to_iter,
+       .csum_and_copy_to_iter          = discard_csum_and_copy_to_iter,
        .csum_and_copy_from_iter        = no_csum_and_copy_from_iter,
        .csum_and_copy_from_iter_full   = no_csum_and_copy_from_iter_full,
 


Reply via email to