arch/x86/include/asm/checksum_32.h

   1 /* SPDX-License-Identifier: GPL-2.0 */
   2 #ifndef _ASM_X86_CHECKSUM_32_H
   3 #define _ASM_X86_CHECKSUM_32_H
   4
   5 #include <linux/in6.h>
   6 #include <linux/uaccess.h>
   7
   8 /*
   9  * computes the checksum of a memory block at buff, length len,
  10  * and adds in "sum" (32-bit)
  11  *
  12  * returns a 32-bit number suitable for feeding into itself
  13  * or csum_tcpudp_magic
  14  *
  15  * this function must be called with even lengths, except
  16  * for the last fragment, which may be odd
  17  *
  18  * it's best to have buff aligned on a 32-bit boundary
  19  */
  20 asmlinkage __wsum csum_partial(const void *buff, int len, __wsum sum);
  21
  22 /*
  23  * the same as csum_partial, but copies from src while it
  24  * checksums, and handles user-space pointer exceptions correctly, when needed.
  25  *
  26  * here even more important to align src and dst on a 32-bit (or even
  27  * better 64-bit) boundary
  28  */
  29
  30 asmlinkage __wsum csum_partial_copy_generic(const void *src, void *dst,
  31                                             int len, __wsum sum,
  32                                             int *src_err_ptr, int *dst_err_ptr);
  33
  34 /*
  35  *      Note: when you get a NULL pointer exception here this means someone
  36  *      passed in an incorrect kernel address to one of these functions.
  37  *
  38  *      If you use these functions directly please don't forget the
  39  *      access_ok().
  40  */
  41 static inline __wsum csum_partial_copy_nocheck(const void *src, void *dst, int len)
  42 {
  43         return csum_partial_copy_generic(src, dst, len, 0, NULL, NULL);
  44 }
  45
  46 static inline __wsum csum_and_copy_from_user(const void __user *src,
  47                                              void *dst, int len,
  48                                              __wsum sum, int *err_ptr)
  49 {
  50         __wsum ret;
  51
  52         might_sleep();
  53         if (!user_access_begin(src, len)) {
  54                 if (len)
  55                         *err_ptr = -EFAULT;
  56                 return sum;
  57         }
  58         ret = csum_partial_copy_generic((__force void *)src, dst,
  59                                         len, sum, err_ptr, NULL);
  60         user_access_end();
  61
  62         return ret;
  63 }
  64
  65 /*
  66  *      This is a version of ip_compute_csum() optimized for IP headers,
  67  *      which always checksum on 4 octet boundaries.
  68  *
  69  *      By Jorge Cwik <jorge@laser.satlink.net>, adapted for linux by
  70  *      Arnt Gulbrandsen.
  71  */
  72 static inline __sum16 ip_fast_csum(const void *iph, unsigned int ihl)
  73 {
  74         unsigned int sum;
  75
  76         asm volatile("movl (%1), %0     ;\n"
  77                      "subl $4, %2       ;\n"
  78                      "jbe 2f            ;\n"
  79                      "addl 4(%1), %0    ;\n"
  80                      "adcl 8(%1), %0    ;\n"
  81                      "adcl 12(%1), %0;\n"
  82                      "1:        adcl 16(%1), %0 ;\n"
  83                      "lea 4(%1), %1     ;\n"
  84                      "decl %2   ;\n"
  85                      "jne 1b            ;\n"
  86                      "adcl $0, %0       ;\n"
  87                      "movl %0, %2       ;\n"
  88                      "shrl $16, %0      ;\n"
  89                      "addw %w2, %w0     ;\n"
  90                      "adcl $0, %0       ;\n"
  91                      "notl %0   ;\n"
  92                      "2:                ;\n"
  93         /* Since the input registers which are loaded with iph and ihl
  94            are modified, we must also specify them as outputs, or gcc
  95            will assume they contain their original values. */
  96                      : "=r" (sum), "=r" (iph), "=r" (ihl)
  97                      : "1" (iph), "2" (ihl)
  98                      : "memory");
  99         return (__force __sum16)sum;
 100 }
 101
 102 /*
 103  *      Fold a partial checksum
 104  */
 105
 106 static inline __sum16 csum_fold(__wsum sum)
 107 {
 108         asm("addl %1, %0                ;\n"
 109             "adcl $0xffff, %0   ;\n"
 110             : "=r" (sum)
 111             : "r" ((__force u32)sum << 16),
 112               "0" ((__force u32)sum & 0xffff0000));
 113         return (__force __sum16)(~(__force u32)sum >> 16);
 114 }
 115
 116 static inline __wsum csum_tcpudp_nofold(__be32 saddr, __be32 daddr,
 117                                         __u32 len, __u8 proto,
 118                                         __wsum sum)
 119 {
 120         asm("addl %1, %0        ;\n"
 121             "adcl %2, %0        ;\n"
 122             "adcl %3, %0        ;\n"
 123             "adcl $0, %0        ;\n"
 124             : "=r" (sum)
 125             : "g" (daddr), "g"(saddr),
 126               "g" ((len + proto) << 8), "0" (sum));
 127         return sum;
 128 }
 129
 130 /*
 131  * computes the checksum of the TCP/UDP pseudo-header
 132  * returns a 16-bit checksum, already complemented
 133  */
 134 static inline __sum16 csum_tcpudp_magic(__be32 saddr, __be32 daddr,
 135                                         __u32 len, __u8 proto,
 136                                         __wsum sum)
 137 {
 138         return csum_fold(csum_tcpudp_nofold(saddr, daddr, len, proto, sum));
 139 }
 140
 141 /*
 142  * this routine is used for miscellaneous IP-like checksums, mainly
 143  * in icmp.c
 144  */
 145
 146 static inline __sum16 ip_compute_csum(const void *buff, int len)
 147 {
 148     return csum_fold(csum_partial(buff, len, 0));
 149 }
 150
 151 #define _HAVE_ARCH_IPV6_CSUM
 152 static inline __sum16 csum_ipv6_magic(const struct in6_addr *saddr,
 153                                       const struct in6_addr *daddr,
 154                                       __u32 len, __u8 proto, __wsum sum)
 155 {
 156         asm("addl 0(%1), %0     ;\n"
 157             "adcl 4(%1), %0     ;\n"
 158             "adcl 8(%1), %0     ;\n"
 159             "adcl 12(%1), %0    ;\n"
 160             "adcl 0(%2), %0     ;\n"
 161             "adcl 4(%2), %0     ;\n"
 162             "adcl 8(%2), %0     ;\n"
 163             "adcl 12(%2), %0    ;\n"
 164             "adcl %3, %0        ;\n"
 165             "adcl %4, %0        ;\n"
 166             "adcl $0, %0        ;\n"
 167             : "=&r" (sum)
 168             : "r" (saddr), "r" (daddr),
 169               "r" (htonl(len)), "r" (htonl(proto)), "0" (sum)
 170             : "memory");
 171
 172         return csum_fold(sum);
 173 }
 174
 175 /*
 176  *      Copy and checksum to user
 177  */
 178 static inline __wsum csum_and_copy_to_user(const void *src,
 179                                            void __user *dst,
 180                                            int len, __wsum sum,
 181                                            int *err_ptr)
 182 {
 183         __wsum ret;
 184
 185         might_sleep();
 186         if (user_access_begin(dst, len)) {
 187                 ret = csum_partial_copy_generic(src, (__force void *)dst,
 188                                                 len, sum, NULL, err_ptr);
 189                 user_access_end();
 190                 return ret;
 191         }
 192
 193         if (len)
 194                 *err_ptr = -EFAULT;
 195
 196         return (__force __wsum)-1; /* invalid checksum */
 197 }
 198
 199 #endif /* _ASM_X86_CHECKSUM_32_H */