checksum.h 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261
  1. /*
  2. * This file is subject to the terms and conditions of the GNU General Public
  3. * License. See the file "COPYING" in the main directory of this archive
  4. * for more details.
  5. *
  6. * Copyright (C) 1995, 96, 97, 98, 99, 2001 by Ralf Baechle
  7. * Copyright (C) 1999 Silicon Graphics, Inc.
  8. * Copyright (C) 2001 Thiemo Seufer.
  9. * Copyright (C) 2002 Maciej W. Rozycki
  10. */
  11. #ifndef _ASM_CHECKSUM_H
  12. #define _ASM_CHECKSUM_H
  13. #include <linux/in6.h>
  14. #include <asm/uaccess.h>
  15. /*
  16. * computes the checksum of a memory block at buff, length len,
  17. * and adds in "sum" (32-bit)
  18. *
  19. * returns a 32-bit number suitable for feeding into itself
  20. * or csum_tcpudp_magic
  21. *
  22. * this function must be called with even lengths, except
  23. * for the last fragment, which may be odd
  24. *
  25. * it's best to have buff aligned on a 32-bit boundary
  26. */
  27. __wsum csum_partial(const void *buff, int len, __wsum sum);
  28. __wsum __csum_partial_copy_user(const void *src, void *dst,
  29. int len, __wsum sum, int *err_ptr);
  30. /*
  31. * this is a new version of the above that records errors it finds in *errp,
  32. * but continues and zeros the rest of the buffer.
  33. */
  34. static inline
  35. __wsum csum_partial_copy_from_user(const void __user *src, void *dst, int len,
  36. __wsum sum, int *err_ptr)
  37. {
  38. might_fault();
  39. return __csum_partial_copy_user((__force void *)src, dst,
  40. len, sum, err_ptr);
  41. }
  42. /*
  43. * Copy and checksum to user
  44. */
  45. #define HAVE_CSUM_COPY_USER
  46. static inline
  47. __wsum csum_and_copy_to_user(const void *src, void __user *dst, int len,
  48. __wsum sum, int *err_ptr)
  49. {
  50. might_fault();
  51. if (access_ok(VERIFY_WRITE, dst, len))
  52. return __csum_partial_copy_user(src, (__force void *)dst,
  53. len, sum, err_ptr);
  54. if (len)
  55. *err_ptr = -EFAULT;
  56. return (__force __wsum)-1; /* invalid checksum */
  57. }
  58. /*
  59. * the same as csum_partial, but copies from user space (but on MIPS
  60. * we have just one address space, so this is identical to the above)
  61. */
  62. __wsum csum_partial_copy_nocheck(const void *src, void *dst,
  63. int len, __wsum sum);
  64. /*
  65. * Fold a partial checksum without adding pseudo headers
  66. */
  67. static inline __sum16 csum_fold(__wsum sum)
  68. {
  69. __asm__(
  70. " .set push # csum_fold\n"
  71. " .set noat \n"
  72. " sll $1, %0, 16 \n"
  73. " addu %0, $1 \n"
  74. " sltu $1, %0, $1 \n"
  75. " srl %0, %0, 16 \n"
  76. " addu %0, $1 \n"
  77. " xori %0, 0xffff \n"
  78. " .set pop"
  79. : "=r" (sum)
  80. : "0" (sum));
  81. return (__force __sum16)sum;
  82. }
  83. /*
  84. * This is a version of ip_compute_csum() optimized for IP headers,
  85. * which always checksum on 4 octet boundaries.
  86. *
  87. * By Jorge Cwik <jorge@laser.satlink.net>, adapted for linux by
  88. * Arnt Gulbrandsen.
  89. */
  90. static inline __sum16 ip_fast_csum(const void *iph, unsigned int ihl)
  91. {
  92. const unsigned int *word = iph;
  93. const unsigned int *stop = word + ihl;
  94. unsigned int csum;
  95. int carry;
  96. csum = word[0];
  97. csum += word[1];
  98. carry = (csum < word[1]);
  99. csum += carry;
  100. csum += word[2];
  101. carry = (csum < word[2]);
  102. csum += carry;
  103. csum += word[3];
  104. carry = (csum < word[3]);
  105. csum += carry;
  106. word += 4;
  107. do {
  108. csum += *word;
  109. carry = (csum < *word);
  110. csum += carry;
  111. word++;
  112. } while (word != stop);
  113. return csum_fold(csum);
  114. }
  115. static inline __wsum csum_tcpudp_nofold(__be32 saddr,
  116. __be32 daddr, unsigned short len, unsigned short proto,
  117. __wsum sum)
  118. {
  119. __asm__(
  120. " .set push # csum_tcpudp_nofold\n"
  121. " .set noat \n"
  122. #ifdef CONFIG_32BIT
  123. " addu %0, %2 \n"
  124. " sltu $1, %0, %2 \n"
  125. " addu %0, $1 \n"
  126. " addu %0, %3 \n"
  127. " sltu $1, %0, %3 \n"
  128. " addu %0, $1 \n"
  129. " addu %0, %4 \n"
  130. " sltu $1, %0, %4 \n"
  131. " addu %0, $1 \n"
  132. #endif
  133. #ifdef CONFIG_64BIT
  134. " daddu %0, %2 \n"
  135. " daddu %0, %3 \n"
  136. " daddu %0, %4 \n"
  137. " dsll32 $1, %0, 0 \n"
  138. " daddu %0, $1 \n"
  139. " dsra32 %0, %0, 0 \n"
  140. #endif
  141. " .set pop"
  142. : "=r" (sum)
  143. : "0" ((__force unsigned long)daddr),
  144. "r" ((__force unsigned long)saddr),
  145. #ifdef __MIPSEL__
  146. "r" ((proto + len) << 8),
  147. #else
  148. "r" (proto + len),
  149. #endif
  150. "r" ((__force unsigned long)sum));
  151. return sum;
  152. }
  153. /*
  154. * computes the checksum of the TCP/UDP pseudo-header
  155. * returns a 16-bit checksum, already complemented
  156. */
  157. static inline __sum16 csum_tcpudp_magic(__be32 saddr, __be32 daddr,
  158. unsigned short len,
  159. unsigned short proto,
  160. __wsum sum)
  161. {
  162. return csum_fold(csum_tcpudp_nofold(saddr, daddr, len, proto, sum));
  163. }
  164. /*
  165. * this routine is used for miscellaneous IP-like checksums, mainly
  166. * in icmp.c
  167. */
  168. static inline __sum16 ip_compute_csum(const void *buff, int len)
  169. {
  170. return csum_fold(csum_partial(buff, len, 0));
  171. }
  172. #define _HAVE_ARCH_IPV6_CSUM
  173. static __inline__ __sum16 csum_ipv6_magic(const struct in6_addr *saddr,
  174. const struct in6_addr *daddr,
  175. __u32 len, unsigned short proto,
  176. __wsum sum)
  177. {
  178. __asm__(
  179. " .set push # csum_ipv6_magic\n"
  180. " .set noreorder \n"
  181. " .set noat \n"
  182. " addu %0, %5 # proto (long in network byte order)\n"
  183. " sltu $1, %0, %5 \n"
  184. " addu %0, $1 \n"
  185. " addu %0, %6 # csum\n"
  186. " sltu $1, %0, %6 \n"
  187. " lw %1, 0(%2) # four words source address\n"
  188. " addu %0, $1 \n"
  189. " addu %0, %1 \n"
  190. " sltu $1, %0, %1 \n"
  191. " lw %1, 4(%2) \n"
  192. " addu %0, $1 \n"
  193. " addu %0, %1 \n"
  194. " sltu $1, %0, %1 \n"
  195. " lw %1, 8(%2) \n"
  196. " addu %0, $1 \n"
  197. " addu %0, %1 \n"
  198. " sltu $1, %0, %1 \n"
  199. " lw %1, 12(%2) \n"
  200. " addu %0, $1 \n"
  201. " addu %0, %1 \n"
  202. " sltu $1, %0, %1 \n"
  203. " lw %1, 0(%3) \n"
  204. " addu %0, $1 \n"
  205. " addu %0, %1 \n"
  206. " sltu $1, %0, %1 \n"
  207. " lw %1, 4(%3) \n"
  208. " addu %0, $1 \n"
  209. " addu %0, %1 \n"
  210. " sltu $1, %0, %1 \n"
  211. " lw %1, 8(%3) \n"
  212. " addu %0, $1 \n"
  213. " addu %0, %1 \n"
  214. " sltu $1, %0, %1 \n"
  215. " lw %1, 12(%3) \n"
  216. " addu %0, $1 \n"
  217. " addu %0, %1 \n"
  218. " sltu $1, %0, %1 \n"
  219. " addu %0, $1 # Add final carry\n"
  220. " .set pop"
  221. : "=r" (sum), "=r" (proto)
  222. : "r" (saddr), "r" (daddr),
  223. "0" (htonl(len)), "1" (htonl(proto)), "r" (sum));
  224. return csum_fold(sum);
  225. }
  226. #endif /* _ASM_CHECKSUM_H */