checksum.c 4.2 KB

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