checksum.c 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163
  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 <net/checksum.h>
  34. #include <asm/checksum.h>
  35. #include <linux/module.h>
  36. static inline unsigned short from32to16(unsigned long x)
  37. {
  38. /* add up 16-bit and 16-bit for 16+c bit */
  39. x = (x & 0xffff) + (x >> 16);
  40. /* add up carry.. */
  41. x = (x & 0xffff) + (x >> 16);
  42. return x;
  43. }
  44. static unsigned int do_csum(const unsigned char *buff, int len)
  45. {
  46. int odd, count;
  47. unsigned long result = 0;
  48. if (len <= 0)
  49. goto out;
  50. odd = 1 & (unsigned long) buff;
  51. if (odd) {
  52. result = *buff;
  53. len--;
  54. buff++;
  55. }
  56. count = len >> 1; /* nr of 16-bit words.. */
  57. if (count) {
  58. if (2 & (unsigned long) buff) {
  59. result += *(unsigned short *) buff;
  60. count--;
  61. len -= 2;
  62. buff += 2;
  63. }
  64. count >>= 1; /* nr of 32-bit words.. */
  65. if (count) {
  66. unsigned long carry = 0;
  67. do {
  68. unsigned long w = *(unsigned long *) buff;
  69. count--;
  70. buff += 4;
  71. result += carry;
  72. result += w;
  73. carry = (w > result);
  74. } while (count);
  75. result += carry;
  76. result = (result & 0xffff) + (result >> 16);
  77. }
  78. if (len & 2) {
  79. result += *(unsigned short *) buff;
  80. buff += 2;
  81. }
  82. }
  83. if (len & 1)
  84. result += (*buff << 8);
  85. result = from32to16(result);
  86. if (odd)
  87. result = ((result >> 8) & 0xff) | ((result & 0xff) << 8);
  88. out:
  89. return result;
  90. }
  91. /*
  92. * This is a version of ip_compute_csum() optimized for IP headers,
  93. * which always checksum on 4 octet boundaries.
  94. */
  95. __sum16 ip_fast_csum(const void *iph, unsigned int ihl)
  96. {
  97. return (__force __sum16)~do_csum(iph, ihl*4);
  98. }
  99. /*
  100. * computes the checksum of a memory block at buff, length len,
  101. * and adds in "sum" (32-bit)
  102. *
  103. * returns a 32-bit number suitable for feeding into itself
  104. * or csum_tcpudp_magic
  105. *
  106. * this function must be called with even lengths, except
  107. * for the last fragment, which may be odd
  108. *
  109. * it's best to have buff aligned on a 32-bit boundary
  110. */
  111. __wsum csum_partial(const void *buff, int len, __wsum sum)
  112. {
  113. unsigned int result = do_csum(buff, len);
  114. /* add in old sum, and carry.. */
  115. result += sum;
  116. if (sum > result)
  117. result += 1;
  118. return result;
  119. }
  120. EXPORT_SYMBOL(csum_partial);
  121. /*
  122. * this routine is used for miscellaneous IP-like checksums, mainly
  123. * in icmp.c
  124. */
  125. __sum16 ip_compute_csum(const unsigned char *buff, int len)
  126. {
  127. return ~do_csum(buff, len);
  128. }
  129. EXPORT_SYMBOL(ip_compute_csum);
  130. /*
  131. * copy from fs while checksumming, otherwise like csum_partial
  132. */
  133. __wsum
  134. csum_partial_copy_from_user(const char __user *src, char *dst, int len,
  135. int sum, int *csum_err)
  136. {
  137. if (csum_err)
  138. *csum_err = 0;
  139. memcpy(dst, src, len);
  140. return csum_partial(dst, len, sum);
  141. }
  142. EXPORT_SYMBOL(csum_partial_copy_from_user);
  143. /*
  144. * copy from ds while checksumming, otherwise like csum_partial
  145. */
  146. __wsum
  147. csum_partial_copy(const char *src, char *dst, int len, int sum)
  148. {
  149. memcpy(dst, src, len);
  150. return csum_partial(dst, len, sum);
  151. }
  152. EXPORT_SYMBOL(csum_partial_copy);