arch/x86/lib/csum-wrappers_64.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /*
   3  * Copyright 2002, 2003 Andi Kleen, SuSE Labs.
   4  *
   5  * Wrappers of assembly checksum functions for x86-64.
   6  */
   7 #include <asm/checksum.h>
   8 #include <linux/export.h>
   9 #include <linux/uaccess.h>
  10 #include <asm/smap.h>
  11
  12 /**
  13  * csum_partial_copy_from_user - Copy and checksum from user space.
  14  * @src: source address (user space)
  15  * @dst: destination address
  16  * @len: number of bytes to be copied.
  17  * @isum: initial sum that is added into the result (32bit unfolded)
  18  * @errp: set to -EFAULT for an bad source address.
  19  *
  20  * Returns an 32bit unfolded checksum of the buffer.
  21  * src and dst are best aligned to 64bits.
  22  */
  23 __wsum
  24 csum_partial_copy_from_user(const void __user *src, void *dst,
  25                             int len, __wsum isum, int *errp)
  26 {
  27         might_sleep();
  28         *errp = 0;
  29
  30         if (!likely(access_ok(src, len)))
  31                 goto out_err;
  32
  33         /*
  34          * Why 6, not 7? To handle odd addresses aligned we
  35          * would need to do considerable complications to fix the
  36          * checksum which is defined as an 16bit accumulator. The
  37          * fix alignment code is primarily for performance
  38          * compatibility with 32bit and that will handle odd
  39          * addresses slowly too.
  40          */
  41         if (unlikely((unsigned long)src & 6)) {
  42                 while (((unsigned long)src & 6) && len >= 2) {
  43                         __u16 val16;
  44
  45                         if (__get_user(val16, (const __u16 __user *)src))
  46                                 goto out_err;
  47
  48                         *(__u16 *)dst = val16;
  49                         isum = (__force __wsum)add32_with_carry(
  50                                         (__force unsigned)isum, val16);
  51                         src += 2;
  52                         dst += 2;
  53                         len -= 2;
  54                 }
  55         }
  56         stac();
  57         isum = csum_partial_copy_generic((__force const void *)src,
  58                                 dst, len, isum, errp, NULL);
  59         clac();
  60         if (unlikely(*errp))
  61                 goto out_err;
  62
  63         return isum;
  64
  65 out_err:
  66         *errp = -EFAULT;
  67         memset(dst, 0, len);
  68
  69         return isum;
  70 }
  71 EXPORT_SYMBOL(csum_partial_copy_from_user);
  72
  73 /**
  74  * csum_partial_copy_to_user - Copy and checksum to user space.
  75  * @src: source address
  76  * @dst: destination address (user space)
  77  * @len: number of bytes to be copied.
  78  * @isum: initial sum that is added into the result (32bit unfolded)
  79  * @errp: set to -EFAULT for an bad destination address.
  80  *
  81  * Returns an 32bit unfolded checksum of the buffer.
  82  * src and dst are best aligned to 64bits.
  83  */
  84 __wsum
  85 csum_partial_copy_to_user(const void *src, void __user *dst,
  86                           int len, __wsum isum, int *errp)
  87 {
  88         __wsum ret;
  89
  90         might_sleep();
  91
  92         if (unlikely(!access_ok(dst, len))) {
  93                 *errp = -EFAULT;
  94                 return 0;
  95         }
  96
  97         if (unlikely((unsigned long)dst & 6)) {
  98                 while (((unsigned long)dst & 6) && len >= 2) {
  99                         __u16 val16 = *(__u16 *)src;
 100
 101                         isum = (__force __wsum)add32_with_carry(
 102                                         (__force unsigned)isum, val16);
 103                         *errp = __put_user(val16, (__u16 __user *)dst);
 104                         if (*errp)
 105                                 return isum;
 106                         src += 2;
 107                         dst += 2;
 108                         len -= 2;
 109                 }
 110         }
 111
 112         *errp = 0;
 113         stac();
 114         ret = csum_partial_copy_generic(src, (void __force *)dst,
 115                                         len, isum, NULL, errp);
 116         clac();
 117         return ret;
 118 }
 119 EXPORT_SYMBOL(csum_partial_copy_to_user);
 120
 121 /**
 122  * csum_partial_copy_nocheck - Copy and checksum.
 123  * @src: source address
 124  * @dst: destination address
 125  * @len: number of bytes to be copied.
 126  * @sum: initial sum that is added into the result (32bit unfolded)
 127  *
 128  * Returns an 32bit unfolded checksum of the buffer.
 129  */
 130 __wsum
 131 csum_partial_copy_nocheck(const void *src, void *dst, int len, __wsum sum)
 132 {
 133         return csum_partial_copy_generic(src, dst, len, sum, NULL, NULL);
 134 }
 135 EXPORT_SYMBOL(csum_partial_copy_nocheck);
 136
 137 __sum16 csum_ipv6_magic(const struct in6_addr *saddr,
 138                         const struct in6_addr *daddr,
 139                         __u32 len, __u8 proto, __wsum sum)
 140 {
 141         __u64 rest, sum64;
 142
 143         rest = (__force __u64)htonl(len) + (__force __u64)htons(proto) +
 144                 (__force __u64)sum;
 145
 146         asm("   addq (%[saddr]),%[sum]\n"
 147             "   adcq 8(%[saddr]),%[sum]\n"
 148             "   adcq (%[daddr]),%[sum]\n"
 149             "   adcq 8(%[daddr]),%[sum]\n"
 150             "   adcq $0,%[sum]\n"
 151
 152             : [sum] "=r" (sum64)
 153             : "[sum]" (rest), [saddr] "r" (saddr), [daddr] "r" (daddr));
 154
 155         return csum_fold(
 156                (__force __wsum)add32_with_carry(sum64 & 0xffffffff, sum64>>32));
 157 }
 158 EXPORT_SYMBOL(csum_ipv6_magic);