extract.h 9.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225
  1. /*
  2. * Copyright (c) 1992, 1993, 1994, 1995, 1996
  3. * The Regents of the University of California. All rights reserved.
  4. *
  5. * Redistribution and use in source and binary forms, with or without
  6. * modification, are permitted provided that: (1) source code distributions
  7. * retain the above copyright notice and this paragraph in its entirety, (2)
  8. * distributions including binary code include the above copyright notice and
  9. * this paragraph in its entirety in the documentation or other materials
  10. * provided with the distribution, and (3) all advertising materials mentioning
  11. * features or use of this software display the following acknowledgement:
  12. * ``This product includes software developed by the University of California,
  13. * Lawrence Berkeley Laboratory and its contributors.'' Neither the name of
  14. * the University nor the names of its contributors may be used to endorse
  15. * or promote products derived from this software without specific prior
  16. * written permission.
  17. * THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR IMPLIED
  18. * WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF
  19. * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
  20. */
  21. #ifndef _WIN32
  22. #include <arpa/inet.h>
  23. #endif
  24. #include <pcap/pcap-inttypes.h>
  25. #include <pcap/compiler-tests.h>
  26. /*
  27. * Macros to extract possibly-unaligned big-endian integral values.
  28. */
  29. #ifdef LBL_ALIGN
  30. /*
  31. * The processor doesn't natively handle unaligned loads.
  32. */
  33. #if PCAP_IS_AT_LEAST_GNUC_VERSION(2,0) && \
  34. (defined(__alpha) || defined(__alpha__) || \
  35. defined(__mips) || defined(__mips__))
  36. /*
  37. * This is MIPS or Alpha, which don't natively handle unaligned loads,
  38. * but which have instructions that can help when doing unaligned
  39. * loads, and this is GCC 2.0 or later or a compiler that claims to
  40. * be GCC 2.0 or later, which we assume that mean we have
  41. * __attribute__((packed)), which we can use to convince the compiler
  42. * to generate those instructions.
  43. *
  44. * Declare packed structures containing a uint16_t and a uint32_t,
  45. * cast the pointer to point to one of those, and fetch through it;
  46. * the GCC manual doesn't appear to explicitly say that
  47. * __attribute__((packed)) causes the compiler to generate unaligned-safe
  48. * code, but it apppears to do so.
  49. *
  50. * We do this in case the compiler can generate code using those
  51. * instructions to do an unaligned load and pass stuff to "ntohs()" or
  52. * "ntohl()", which might be better than than the code to fetch the
  53. * bytes one at a time and assemble them. (That might not be the
  54. * case on a little-endian platform, such as DEC's MIPS machines and
  55. * Alpha machines, where "ntohs()" and "ntohl()" might not be done
  56. * inline.)
  57. *
  58. * We do this only for specific architectures because, for example,
  59. * at least some versions of GCC, when compiling for 64-bit SPARC,
  60. * generate code that assumes alignment if we do this.
  61. *
  62. * XXX - add other architectures and compilers as possible and
  63. * appropriate.
  64. *
  65. * HP's C compiler, indicated by __HP_cc being defined, supports
  66. * "#pragma unaligned N" in version A.05.50 and later, where "N"
  67. * specifies a number of bytes at which the typedef on the next
  68. * line is aligned, e.g.
  69. *
  70. * #pragma unalign 1
  71. * typedef uint16_t unaligned_uint16_t;
  72. *
  73. * to define unaligned_uint16_t as a 16-bit unaligned data type.
  74. * This could be presumably used, in sufficiently recent versions of
  75. * the compiler, with macros similar to those below. This would be
  76. * useful only if that compiler could generate better code for PA-RISC
  77. * or Itanium than would be generated by a bunch of shifts-and-ORs.
  78. *
  79. * DEC C, indicated by __DECC being defined, has, at least on Alpha,
  80. * an __unaligned qualifier that can be applied to pointers to get the
  81. * compiler to generate code that does unaligned loads and stores when
  82. * dereferencing the pointer in question.
  83. *
  84. * XXX - what if the native C compiler doesn't support
  85. * __attribute__((packed))? How can we get it to generate unaligned
  86. * accesses for *specific* items?
  87. */
  88. typedef struct {
  89. uint16_t val;
  90. } __attribute__((packed)) unaligned_uint16_t;
  91. typedef struct {
  92. uint32_t val;
  93. } __attribute__((packed)) unaligned_uint32_t;
  94. static inline uint16_t
  95. EXTRACT_16BITS(const void *p)
  96. {
  97. return ((uint16_t)ntohs(((const unaligned_uint16_t *)(p))->val));
  98. }
  99. static inline uint32_t
  100. EXTRACT_32BITS(const void *p)
  101. {
  102. return ((uint32_t)ntohl(((const unaligned_uint32_t *)(p))->val));
  103. }
  104. static inline uint64_t
  105. EXTRACT_64BITS(const void *p)
  106. {
  107. return ((uint64_t)(((uint64_t)ntohl(((const unaligned_uint32_t *)(p) + 0)->val)) << 32 | \
  108. ((uint64_t)ntohl(((const unaligned_uint32_t *)(p) + 1)->val)) << 0));
  109. }
  110. #else /* have to do it a byte at a time */
  111. /*
  112. * This isn't a GCC-compatible compiler, we don't have __attribute__,
  113. * or we do but we don't know of any better way with this instruction
  114. * set to do unaligned loads, so do unaligned loads of big-endian
  115. * quantities the hard way - fetch the bytes one at a time and
  116. * assemble them.
  117. */
  118. #define EXTRACT_16BITS(p) \
  119. ((uint16_t)(((uint16_t)(*((const uint8_t *)(p) + 0)) << 8) | \
  120. ((uint16_t)(*((const uint8_t *)(p) + 1)) << 0)))
  121. #define EXTRACT_32BITS(p) \
  122. ((uint32_t)(((uint32_t)(*((const uint8_t *)(p) + 0)) << 24) | \
  123. ((uint32_t)(*((const uint8_t *)(p) + 1)) << 16) | \
  124. ((uint32_t)(*((const uint8_t *)(p) + 2)) << 8) | \
  125. ((uint32_t)(*((const uint8_t *)(p) + 3)) << 0)))
  126. #define EXTRACT_64BITS(p) \
  127. ((uint64_t)(((uint64_t)(*((const uint8_t *)(p) + 0)) << 56) | \
  128. ((uint64_t)(*((const uint8_t *)(p) + 1)) << 48) | \
  129. ((uint64_t)(*((const uint8_t *)(p) + 2)) << 40) | \
  130. ((uint64_t)(*((const uint8_t *)(p) + 3)) << 32) | \
  131. ((uint64_t)(*((const uint8_t *)(p) + 4)) << 24) | \
  132. ((uint64_t)(*((const uint8_t *)(p) + 5)) << 16) | \
  133. ((uint64_t)(*((const uint8_t *)(p) + 6)) << 8) | \
  134. ((uint64_t)(*((const uint8_t *)(p) + 7)) << 0)))
  135. #endif /* must special-case unaligned accesses */
  136. #else /* LBL_ALIGN */
  137. /*
  138. * The processor natively handles unaligned loads, so we can just
  139. * cast the pointer and fetch through it.
  140. */
  141. static inline uint16_t
  142. EXTRACT_16BITS(const void *p)
  143. {
  144. return ((uint16_t)ntohs(*(const uint16_t *)(p)));
  145. }
  146. static inline uint32_t
  147. EXTRACT_32BITS(const void *p)
  148. {
  149. return ((uint32_t)ntohl(*(const uint32_t *)(p)));
  150. }
  151. static inline uint64_t
  152. EXTRACT_64BITS(const void *p)
  153. {
  154. return ((uint64_t)(((uint64_t)ntohl(*((const uint32_t *)(p) + 0))) << 32 | \
  155. ((uint64_t)ntohl(*((const uint32_t *)(p) + 1))) << 0));
  156. }
  157. #endif /* LBL_ALIGN */
  158. #define EXTRACT_24BITS(p) \
  159. ((uint32_t)(((uint32_t)(*((const uint8_t *)(p) + 0)) << 16) | \
  160. ((uint32_t)(*((const uint8_t *)(p) + 1)) << 8) | \
  161. ((uint32_t)(*((const uint8_t *)(p) + 2)) << 0)))
  162. #define EXTRACT_40BITS(p) \
  163. ((uint64_t)(((uint64_t)(*((const uint8_t *)(p) + 0)) << 32) | \
  164. ((uint64_t)(*((const uint8_t *)(p) + 1)) << 24) | \
  165. ((uint64_t)(*((const uint8_t *)(p) + 2)) << 16) | \
  166. ((uint64_t)(*((const uint8_t *)(p) + 3)) << 8) | \
  167. ((uint64_t)(*((const uint8_t *)(p) + 4)) << 0)))
  168. #define EXTRACT_48BITS(p) \
  169. ((uint64_t)(((uint64_t)(*((const uint8_t *)(p) + 0)) << 40) | \
  170. ((uint64_t)(*((const uint8_t *)(p) + 1)) << 32) | \
  171. ((uint64_t)(*((const uint8_t *)(p) + 2)) << 24) | \
  172. ((uint64_t)(*((const uint8_t *)(p) + 3)) << 16) | \
  173. ((uint64_t)(*((const uint8_t *)(p) + 4)) << 8) | \
  174. ((uint64_t)(*((const uint8_t *)(p) + 5)) << 0)))
  175. #define EXTRACT_56BITS(p) \
  176. ((uint64_t)(((uint64_t)(*((const uint8_t *)(p) + 0)) << 48) | \
  177. ((uint64_t)(*((const uint8_t *)(p) + 1)) << 40) | \
  178. ((uint64_t)(*((const uint8_t *)(p) + 2)) << 32) | \
  179. ((uint64_t)(*((const uint8_t *)(p) + 3)) << 24) | \
  180. ((uint64_t)(*((const uint8_t *)(p) + 4)) << 16) | \
  181. ((uint64_t)(*((const uint8_t *)(p) + 5)) << 8) | \
  182. ((uint64_t)(*((const uint8_t *)(p) + 6)) << 0)))
  183. /*
  184. * Macros to extract possibly-unaligned little-endian integral values.
  185. * XXX - do loads on little-endian machines that support unaligned loads?
  186. */
  187. #define EXTRACT_LE_8BITS(p) (*(p))
  188. #define EXTRACT_LE_16BITS(p) \
  189. ((uint16_t)(((uint16_t)(*((const uint8_t *)(p) + 1)) << 8) | \
  190. ((uint16_t)(*((const uint8_t *)(p) + 0)) << 0)))
  191. #define EXTRACT_LE_32BITS(p) \
  192. ((uint32_t)(((uint32_t)(*((const uint8_t *)(p) + 3)) << 24) | \
  193. ((uint32_t)(*((const uint8_t *)(p) + 2)) << 16) | \
  194. ((uint32_t)(*((const uint8_t *)(p) + 1)) << 8) | \
  195. ((uint32_t)(*((const uint8_t *)(p) + 0)) << 0)))
  196. #define EXTRACT_LE_24BITS(p) \
  197. ((uint32_t)(((uint32_t)(*((const uint8_t *)(p) + 2)) << 16) | \
  198. ((uint32_t)(*((const uint8_t *)(p) + 1)) << 8) | \
  199. ((uint32_t)(*((const uint8_t *)(p) + 0)) << 0)))
  200. #define EXTRACT_LE_64BITS(p) \
  201. ((uint64_t)(((uint64_t)(*((const uint8_t *)(p) + 7)) << 56) | \
  202. ((uint64_t)(*((const uint8_t *)(p) + 6)) << 48) | \
  203. ((uint64_t)(*((const uint8_t *)(p) + 5)) << 40) | \
  204. ((uint64_t)(*((const uint8_t *)(p) + 4)) << 32) | \
  205. ((uint64_t)(*((const uint8_t *)(p) + 3)) << 24) | \
  206. ((uint64_t)(*((const uint8_t *)(p) + 2)) << 16) | \
  207. ((uint64_t)(*((const uint8_t *)(p) + 1)) << 8) | \
  208. ((uint64_t)(*((const uint8_t *)(p) + 0)) << 0)))