checksum.c 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211
  1. /*
  2. *
  3. * INET An implementation of the TCP/IP protocol suite for the LINUX
  4. * operating system. INET is implemented using the BSD Socket
  5. * interface as the means of communication with the user level.
  6. *
  7. * IP/TCP/UDP checksumming routines
  8. *
  9. * Authors: Jorge Cwik, <jorge@laser.satlink.net>
  10. * Arnt Gulbrandsen, <agulbra@nvg.unit.no>
  11. * Tom May, <ftom@netcom.com>
  12. * Andreas Schwab, <schwab@issan.informatik.uni-dortmund.de>
  13. * Lots of code moved from tcp.c and ip.c; see those files
  14. * for more names.
  15. *
  16. * 03/02/96 Jes Sorensen, Andreas Schwab, Roman Hodek:
  17. * Fixed some nasty bugs, causing some horrible crashes.
  18. * A: At some points, the sum (%0) was used as
  19. * length-counter instead of the length counter
  20. * (%1). Thanks to Roman Hodek for pointing this out.
  21. * B: GCC seems to mess up if one uses too many
  22. * data-registers to hold input values and one tries to
  23. * specify d0 and d1 as scratch registers. Letting gcc
  24. * choose these registers itself solves the problem.
  25. *
  26. * This program is free software; you can redistribute it and/or
  27. * modify it under the terms of the GNU General Public License
  28. * as published by the Free Software Foundation; either version
  29. * 2 of the License, or (at your option) any later version.
  30. */
  31. /* Revised by Kenneth Albanowski for m68knommu. Basic problem: unaligned access
  32. kills, so most of the assembly has to go. */
  33. #include <linux/export.h>
  34. #include <net/checksum.h>
  35. #include <asm/byteorder.h>
  36. #ifndef do_csum
  37. static inline unsigned short from32to16(unsigned int x)
  38. {
  39. /* add up 16-bit and 16-bit for 16+c bit */
  40. x = (x & 0xffff) + (x >> 16);
  41. /* add up carry.. */
  42. x = (x & 0xffff) + (x >> 16);
  43. return x;
  44. }
  45. static unsigned int do_csum(const unsigned char *buff, int len)
  46. {
  47. int odd;
  48. unsigned int result = 0;
  49. if (len <= 0)
  50. goto out;
  51. odd = 1 & (unsigned long) buff;
  52. if (odd) {
  53. #ifdef __LITTLE_ENDIAN
  54. result += (*buff << 8);
  55. #else
  56. result = *buff;
  57. #endif
  58. len--;
  59. buff++;
  60. }
  61. if (len >= 2) {
  62. if (2 & (unsigned long) buff) {
  63. result += *(unsigned short *) buff;
  64. len -= 2;
  65. buff += 2;
  66. }
  67. if (len >= 4) {
  68. const unsigned char *end = buff + ((unsigned)len & ~3);
  69. unsigned int carry = 0;
  70. do {
  71. unsigned int w = *(unsigned int *) buff;
  72. buff += 4;
  73. result += carry;
  74. result += w;
  75. carry = (w > result);
  76. } while (buff < end);
  77. result += carry;
  78. result = (result & 0xffff) + (result >> 16);
  79. }
  80. if (len & 2) {
  81. result += *(unsigned short *) buff;
  82. buff += 2;
  83. }
  84. }
  85. if (len & 1)
  86. #ifdef __LITTLE_ENDIAN
  87. result += *buff;
  88. #else
  89. result += (*buff << 8);
  90. #endif
  91. result = from32to16(result);
  92. if (odd)
  93. result = ((result >> 8) & 0xff) | ((result & 0xff) << 8);
  94. out:
  95. return result;
  96. }
  97. #endif
  98. #ifndef ip_fast_csum
  99. /*
  100. * This is a version of ip_compute_csum() optimized for IP headers,
  101. * which always checksum on 4 octet boundaries.
  102. */
  103. __sum16 ip_fast_csum(const void *iph, unsigned int ihl)
  104. {
  105. return (__force __sum16)~do_csum(iph, ihl*4);
  106. }
  107. EXPORT_SYMBOL(ip_fast_csum);
  108. #endif
  109. /*
  110. * computes the checksum of a memory block at buff, length len,
  111. * and adds in "sum" (32-bit)
  112. *
  113. * returns a 32-bit number suitable for feeding into itself
  114. * or csum_tcpudp_magic
  115. *
  116. * this function must be called with even lengths, except
  117. * for the last fragment, which may be odd
  118. *
  119. * it's best to have buff aligned on a 32-bit boundary
  120. */
  121. __wsum csum_partial(const void *buff, int len, __wsum wsum)
  122. {
  123. unsigned int sum = (__force unsigned int)wsum;
  124. unsigned int result = do_csum(buff, len);
  125. /* add in old sum, and carry.. */
  126. result += sum;
  127. if (sum > result)
  128. result += 1;
  129. return (__force __wsum)result;
  130. }
  131. EXPORT_SYMBOL(csum_partial);
  132. /*
  133. * this routine is used for miscellaneous IP-like checksums, mainly
  134. * in icmp.c
  135. */
  136. __sum16 ip_compute_csum(const void *buff, int len)
  137. {
  138. return (__force __sum16)~do_csum(buff, len);
  139. }
  140. EXPORT_SYMBOL(ip_compute_csum);
  141. /*
  142. * copy from fs while checksumming, otherwise like csum_partial
  143. */
  144. __wsum
  145. csum_partial_copy_from_user(const void __user *src, void *dst, int len,
  146. __wsum sum, int *csum_err)
  147. {
  148. int missing;
  149. missing = __copy_from_user(dst, src, len);
  150. if (missing) {
  151. memset(dst + len - missing, 0, missing);
  152. *csum_err = -EFAULT;
  153. } else
  154. *csum_err = 0;
  155. return csum_partial(dst, len, sum);
  156. }
  157. EXPORT_SYMBOL(csum_partial_copy_from_user);
  158. /*
  159. * copy from ds while checksumming, otherwise like csum_partial
  160. */
  161. __wsum
  162. csum_partial_copy(const void *src, void *dst, int len, __wsum sum)
  163. {
  164. memcpy(dst, src, len);
  165. return csum_partial(dst, len, sum);
  166. }
  167. EXPORT_SYMBOL(csum_partial_copy);
  168. #ifndef csum_tcpudp_nofold
  169. static inline u32 from64to32(u64 x)
  170. {
  171. /* add up 32-bit and 32-bit for 32+c bit */
  172. x = (x & 0xffffffff) + (x >> 32);
  173. /* add up carry.. */
  174. x = (x & 0xffffffff) + (x >> 32);
  175. return (u32)x;
  176. }
  177. __wsum csum_tcpudp_nofold(__be32 saddr, __be32 daddr,
  178. unsigned short len,
  179. unsigned short proto,
  180. __wsum sum)
  181. {
  182. unsigned long long s = (__force u32)sum;
  183. s += (__force u32)saddr;
  184. s += (__force u32)daddr;
  185. #ifdef __BIG_ENDIAN
  186. s += proto + len;
  187. #else
  188. s += (proto + len) << 8;
  189. #endif
  190. return (__force __wsum)from64to32(s);
  191. }
  192. EXPORT_SYMBOL(csum_tcpudp_nofold);
  193. #endif