defines.h 9.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357
  1. #ifndef __DEFINES_H__
  2. #define __DEFINES_H__
  3. #include "config.h"
  4. #ifdef HAVE_STDDEF_H
  5. #include <stddef.h>
  6. #endif
  7. #ifdef HAVE_SYS_TYPES_H
  8. #include <sys/types.h>
  9. #endif
  10. #ifdef WIN32
  11. #include "msvc_inttypes.h"
  12. #include "msvc_stdint.h"
  13. #else
  14. #ifdef HAVE_INTTYPES_H
  15. #include <inttypes.h>
  16. #endif
  17. #ifdef HAVE_STDINT_H
  18. #include <stdint.h>
  19. #endif
  20. #endif /* WIN32 */
  21. #ifdef HAVE_SYS_SOCKET
  22. #include <sys/socket.h>
  23. #endif
  24. #ifdef HAVE_NETINET_IN_H
  25. #include <netinet/in.h>
  26. #endif
  27. #ifdef HAVE_NETINET_IN_SYSTM_H
  28. #include <netinet/in_systm.h>
  29. #endif
  30. #include "tcpr.h"
  31. #ifdef HAVE_BPF
  32. #include <net/bpf.h>
  33. #define PCAP_DONT_INCLUDE_PCAP_BPF_H 1
  34. #endif
  35. #if defined INCLUDE_PCAP_BPF_H_FILE && !defined PCAP_DONT_INCLUDE_PCAP_BPF_H
  36. #include "@INCLUDE_PCAP_BPF_HEADER@"
  37. #define PCAP_DONT_INCLUDE_PCAP_BPF_H 1 /* don't re-include it in pcap.h */
  38. #endif
  39. #include "/usr/include/pcap.h"
  40. /* include our own strlcat/strlcpy? */
  41. #ifndef HAVE_STRLCPY
  42. #include "lib/strlcpy.h"
  43. #endif
  44. /*
  45. * net/bpf.h doesn't include DLT types, but pcap-bpf.h does.
  46. * Unfortunately, pcap-bpf.h also includes things in net/bpf.h
  47. * while also missing some key things (wow, that sucks)
  48. * The result is that I stole the DLT types from pcap-bpf.h and
  49. * put them in here.
  50. */
  51. #include "common/dlt_names.h"
  52. #ifdef HAVE_BOOL_H
  53. #include <bool.h>
  54. #endif
  55. #ifdef HAVE_STDBOOL_H
  56. #include <stdbool.h>
  57. #endif
  58. /* should packet counters be 32 or 64 bit? --enable-64bit */
  59. #ifdef ENABLE_64BITS
  60. #define COUNTER unsigned long long
  61. #define COUNTER_SPEC "%llu"
  62. #else
  63. #define COUNTER unsigned long
  64. #define COUNTER_SPEC "%lu"
  65. #endif
  66. #define COUNTER_OVERFLOW_RISK (((COUNTER)~0) >> 20)
  67. #include "common/list.h"
  68. #include "common/cidr.h"
  69. typedef struct tcpr_ipv4_hdr ipv4_hdr_t;
  70. typedef struct tcpr_ipv6_hdr ipv6_hdr_t;
  71. typedef struct tcpr_tcp_hdr tcp_hdr_t;
  72. typedef struct tcpr_udp_hdr udp_hdr_t;
  73. typedef struct tcpr_icmpv4_hdr icmpv4_hdr_t;
  74. typedef struct tcpr_icmpv6_hdr icmpv6_hdr_t;
  75. typedef struct tcpr_ethernet_hdr eth_hdr_t;
  76. typedef struct tcpr_802_1q_hdr vlan_hdr_t;
  77. typedef struct sll_header sll_hdr_t;
  78. typedef struct tcpr_arp_hdr arp_hdr_t;
  79. typedef struct tcpr_dnsv4_hdr dnsv4_hdr_t;
  80. /* our custom typdefs/structs */
  81. typedef u_char tcpr_macaddr_t[TCPR_ETH_H];
  82. typedef struct tcpr_bpf_s {
  83. char *filter;
  84. int optimize;
  85. struct bpf_program program;
  86. } tcpr_bpf_t;
  87. typedef struct tcpr_xX_s {
  88. #define xX_MODE_INCLUDE 'x'
  89. #define xX_MODE_EXCLUDE 'X'
  90. int mode;
  91. tcpr_list_t *list;
  92. tcpr_cidr_t *cidr;
  93. #define xX_TYPE_LIST 1
  94. #define xX_TYPE_CIDR 2
  95. int type;
  96. } tcpr_xX_t;
  97. /* number of ports 0-65535 */
  98. #define NUM_PORTS 65536
  99. typedef struct tcpr_services_s {
  100. char tcp[NUM_PORTS];
  101. char udp[NUM_PORTS];
  102. } tcpr_services_t;
  103. typedef struct tcpr_speed_s {
  104. /* speed modifiers */
  105. int mode;
  106. #define SPEED_MULTIPLIER 1
  107. #define SPEED_MBPSRATE 2
  108. #define SPEED_PACKETRATE 3
  109. #define SPEED_TOPSPEED 4
  110. #define SPEED_ONEATATIME 5
  111. COUNTER speed;
  112. float multiplier;
  113. int pps_multi;
  114. } tcpr_speed_t;
  115. #define MAX_FILES 1024 /* Max number of files we can pass to tcpreplay */
  116. #define DEFAULT_MTU 1500 /* Max Transmission Unit of standard ethernet
  117. * don't forget *frames* are MTU + L2 header! */
  118. #define MAXPACKET 65549 /* was 16436 linux loopback, but maybe something is bigger then
  119. linux loopback */
  120. #define MAX_SNAPLEN 65535 /* tell libpcap to capture the entire packet */
  121. #define DNS_RESOLVE 1
  122. #define DNS_DONT_RESOLVE 0
  123. #define RESOLVE 0 /* disable dns lookups */
  124. #define BPF_OPTIMIZE 1 /* default is to optimize bpf program */
  125. #define PCAP_TIMEOUT 100 /* 100ms pcap_open_live timeout */
  126. #define DEFAULT_FUZZ_FACTOR 8
  127. /* HP-UX already defines TRUE/FALSE */
  128. #ifndef TRUE
  129. typedef enum bool_e {
  130. FALSE = 0,
  131. TRUE
  132. } bool_t;
  133. #endif
  134. #define EBUF_SIZE 1024 /* size of our error buffers */
  135. #define MAC_SIZE 7 /* size of the mac[] buffer */
  136. typedef enum pad_e {
  137. PAD_PACKET,
  138. TRUNC_PACKET
  139. } pad_t;
  140. #define DNS_QUERY_FLAG 0x8000
  141. typedef enum direction_e {
  142. DIR_UNKNOWN = -1,
  143. DIR_CLIENT = 0,
  144. DIR_SERVER = 1,
  145. DIR_ANY = 2
  146. } direction_t;
  147. typedef enum tcpprep_mode_e {
  148. ERROR_MODE, /* Some kind of error has occurred */
  149. CIDR_MODE, /* single pass, CIDR netblock */
  150. REGEX_MODE, /* single pass, regex */
  151. PORT_MODE, /* single pass, use src/dst ports to split */
  152. MAC_MODE, /* single pass, use src mac to split */
  153. FIRST_MODE, /* single pass, use first seen to split */
  154. AUTO_MODE, /* first pass through in auto mode */
  155. ROUTER_MODE, /* second pass through in router mode */
  156. BRIDGE_MODE, /* second pass through in bridge mode */
  157. SERVER_MODE, /* second pass through in server (router) mode */
  158. CLIENT_MODE /* second pass through in client (router) mode */
  159. } tcpprep_mode_t;
  160. #define BROADCAST_MAC "\xFF\xFF\xFF\xFF\xFF\xFF"
  161. /* MAC macros for printf */
  162. #define MAC_FORMAT "%02X:%02X:%02X:%02X:%02X:%02X"
  163. #define MAC_STR(x) x[0], x[1], x[2], x[3], x[4], x[5]
  164. /* struct timeval print structs */
  165. #ifdef HAVE_DARWIN
  166. /* Darwin defines usec as an __int32_t, not unsigned long. */
  167. #define TIMEVAL_FORMAT "%lus %dusec"
  168. #else
  169. #define TIMEVAL_FORMAT "%lus %luusec"
  170. #endif
  171. #define TIMESPEC_FORMAT "%lus %lunsec"
  172. /* force a word or half-word swap on both Big and Little endian systems */
  173. #ifndef SWAPLONG
  174. #define SWAPLONG(y) \
  175. ((((y)&0xff)<<24) | (((y)&0xff00)<<8) | (((y)&0xff0000)>>8) | (((y)>>24)&0xff))
  176. #endif
  177. #ifndef SWAPSHORT
  178. #define SWAPSHORT(y) \
  179. ( (((y)&0xff)<<8) | ((u_short)((y)&0xff00)>>8) )
  180. #endif
  181. /* converts a 64bit int to network byte order */
  182. #ifndef HAVE_NTOHLL
  183. #ifdef WORDS_BIGENDIAN
  184. #define ntohll(x) (x)
  185. #define htonll(x) (x)
  186. #else
  187. /* stolen from http://www.codeproject.com/cpp/endianness.asp */
  188. #define ntohll(x) (((u_int64_t)(ntohl((int)((x << 32) >> 32))) << 32) | \
  189. (unsigned int)ntohl(((int)(x >> 32))))
  190. #define htonll(x) ntohll(x)
  191. #endif /* WORDS_BIGENDIAN */
  192. #endif
  193. #define DEBUG_INFO 1 /* informational only, lessthan 1 line per packet */
  194. #define DEBUG_BASIC 2 /* limited debugging, one line per packet */
  195. #define DEBUG_DETAIL 3 /* more detailed, a few lines per packet */
  196. #define DEBUG_MORE 4 /* even more detail */
  197. #define DEBUG_CODE 5 /* examines code & values, many lines per packet */
  198. /* Win32 doesn't know about PF_INET6 */
  199. #ifndef PF_INET6
  200. #ifdef AF_INET6
  201. #define PF_INET6 AF_INET6
  202. #else
  203. #define PF_INET6 30 /* stolen from OS/X */
  204. #endif
  205. #endif
  206. /* convert IPv6 Extention Header Len value to bytes */
  207. #define IPV6_EXTLEN_TO_BYTES(x) ((x * 4) + 8)
  208. #ifndef HAVE_UINT8_T
  209. typedef u_int8_t uint8_t
  210. typedef u_int16_t uint16_t
  211. typedef u_int32_t uint32_t
  212. #endif
  213. /* Support for flexible arrays. */
  214. #undef __flexarr
  215. #if defined(__GNUC__) && ((__GNUC__ > 2) || (__GNUC__ == 2 && __GNUC_MINOR__ >= 97))
  216. /* GCC 2.97 supports C99 flexible array members. */
  217. # define __flexarr []
  218. #else
  219. # ifdef __GNUC__
  220. # define __flexarr [0]
  221. # else
  222. # if defined(__STDC_VERSION__) && __STDC_VERSION__ >= 199901L
  223. # define __flexarr []
  224. # elif defined(_WIN32)
  225. /* MS VC++ */
  226. # define __flexarr []
  227. # else
  228. /* Some other non-C99 compiler. Approximate with [1]. */
  229. # define __flexarr [1]
  230. # endif
  231. # endif
  232. #endif
  233. /* Time converters */
  234. #define SEC_TO_MILLISEC(x) (x * 1000)
  235. #define SEC_TO_MICROSEC(x) (x * 1000000)
  236. #define SEC_TO_NANOSEC(x) ((u_int64_t)x * 1000000000)
  237. #define MILLISEC_TO_SEC(x) (x / 1000)
  238. #define MICROSEC_TO_SEC(x) (x / 1000000)
  239. #define NANOSEC_TO_SEC(x) ((u_int64_t)x / 1000000000)
  240. #define TIMEVAL_TO_MILLISEC(x) (((x)->tv_sec * 1000) + ((x)->tv_usec / 1000))
  241. #define TIMEVAL_TO_MICROSEC(x) (((x)->tv_sec * 1000000) + (x)->tv_usec)
  242. #define TIMEVAL_TO_NANOSEC(x) ((u_int64_t)((x)->tv_sec * 1000000000) + ((u_int64_t)(x)->tv_usec * 1000))
  243. #define TIMSTAMP_TO_MICROSEC(x) (TIMEVAL_TO_MICROSEC(x))
  244. #define MILLISEC_TO_TIMEVAL(x, tv) \
  245. do { \
  246. (tv)->tv_sec = (x) / 1000; \
  247. (tv)->tv_usec = (x * 1000) - ((tv)->tv_sec * 1000000); \
  248. } while(0)
  249. #define MICROSEC_TO_TIMEVAL(x, tv) \
  250. do { \
  251. (tv)->tv_sec = (x) / 1000000; \
  252. (tv)->tv_usec = (x) - ((tv)->tv_sec * 1000000); \
  253. } while(0)
  254. #define NANOSEC_TO_TIMEVAL(x, tv) \
  255. do { \
  256. (tv)->tv_sec = (x) / 1000000000; \
  257. (tv)->tv_usec = ((x) % 1000000000) / 1000; \
  258. } while(0)
  259. #define NANOSEC_TO_TIMESPEC(x, ts) \
  260. do { \
  261. (ts)->tv_sec = (x) / 1000000000; \
  262. (ts)->tv_nsec = (x) % 1000000000; \
  263. } while(0)
  264. #define TIMESPEC_TO_MILLISEC(x) (((x)->tv_sec * 1000) + ((x)->tv_nsec / 1000000))
  265. #define TIMESPEC_TO_MICROSEC(x) (((x)->tv_sec * 1000000) + (x)->tv_nsec / 1000)
  266. #define TIMESPEC_TO_NANOSEC(x) ((u_int64_t)((x)->tv_sec * 1000000000) + ((u_int64_t)(x)->tv_nsec))
  267. /*
  268. * Help suppress some compiler warnings
  269. * No problem if variable is actually used
  270. */
  271. #ifdef UNUSED
  272. #elif defined(__GNUC__)
  273. # define UNUSED(x) UNUSED_ ## x __attribute__((unused))
  274. #elif defined(__LCLINT__)
  275. # define UNUSED(x) /*@unused@*/ x
  276. #else
  277. # define UNUSED(x) x
  278. #endif
  279. #ifndef max
  280. # define max(a,b) \
  281. ({ __typeof__ (a) _a = (a); \
  282. __typeof__ (b) _b = (b); \
  283. _a > _b ? _a : _b; })
  284. #endif
  285. #ifndef min
  286. # define min(a,b) \
  287. ({ __typeof__ (a) _a = (a); \
  288. __typeof__ (b) _b = (b); \
  289. _a > _b ? _b : _a; })
  290. #endif
  291. #endif /* DEFINES */