compat.h 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709
  1. /* Copyright (c) 2003-2004, Roger Dingledine
  2. * Copyright (c) 2004-2006, Roger Dingledine, Nick Mathewson.
  3. * Copyright (c) 2007-2012, The Tor Project, Inc. */
  4. /* See LICENSE for licensing information */
  5. #ifndef TOR_COMPAT_H
  6. #define TOR_COMPAT_H
  7. #include "orconfig.h"
  8. #include "torint.h"
  9. #ifdef _WIN32
  10. #ifndef _WIN32_WINNT
  11. #define _WIN32_WINNT 0x0501
  12. #endif
  13. #define WIN32_LEAN_AND_MEAN
  14. #if defined(_MSC_VER) && (_MSC_VER < 1300)
  15. #include <winsock.h>
  16. #else
  17. #include <winsock2.h>
  18. #include <ws2tcpip.h>
  19. #endif
  20. #endif
  21. #ifdef HAVE_SYS_PARAM_H
  22. #include <sys/param.h>
  23. #endif
  24. #ifdef HAVE_SYS_TYPES_H
  25. #include <sys/types.h>
  26. #endif
  27. #ifdef HAVE_SYS_TIME_H
  28. #include <sys/time.h>
  29. #endif
  30. #ifdef HAVE_TIME_H
  31. #include <time.h>
  32. #endif
  33. #ifdef HAVE_STRING_H
  34. #include <string.h>
  35. #endif
  36. #if defined(HAVE_PTHREAD_H) && !defined(_WIN32)
  37. #include <pthread.h>
  38. #endif
  39. #include <stdarg.h>
  40. #ifdef HAVE_SYS_RESOURCE_H
  41. #include <sys/resource.h>
  42. #endif
  43. #ifdef HAVE_SYS_SOCKET_H
  44. #include <sys/socket.h>
  45. #endif
  46. #ifdef HAVE_NETINET_IN_H
  47. #include <netinet/in.h>
  48. #endif
  49. #ifdef HAVE_NETINET6_IN6_H
  50. #include <netinet6/in6.h>
  51. #endif
  52. #include <stdio.h>
  53. #if defined (WINCE)
  54. #include <fcntl.h>
  55. #include <io.h>
  56. #include <math.h>
  57. #include <projects.h>
  58. #define snprintf _snprintf
  59. /* this is not exported as W .... */
  60. #define SHGetPathFromIDListW SHGetPathFromIDList
  61. /* wcecompat has vasprintf */
  62. #define HAVE_VASPRINTF
  63. /* no service here */
  64. #ifdef NT_SERVICE
  65. #undef NT_SERVICE
  66. #endif
  67. #endif // WINCE
  68. #ifndef NULL_REP_IS_ZERO_BYTES
  69. #error "It seems your platform does not represent NULL as zero. We can't cope."
  70. #endif
  71. #if 'a'!=97 || 'z'!=122 || 'A'!=65 || ' '!=32
  72. #error "It seems that you encode characters in something other than ASCII."
  73. #endif
  74. /* ===== Compiler compatibility */
  75. /* GCC can check printf types on arbitrary functions. */
  76. #ifdef __GNUC__
  77. #define CHECK_PRINTF(formatIdx, firstArg) \
  78. __attribute__ ((format(printf, formatIdx, firstArg)))
  79. #else
  80. #define CHECK_PRINTF(formatIdx, firstArg)
  81. #endif
  82. /* inline is __inline on windows. */
  83. #ifdef _WIN32
  84. #define INLINE __inline
  85. #else
  86. #define INLINE inline
  87. #endif
  88. /* Try to get a reasonable __func__ substitute in place. */
  89. #if defined(_MSC_VER)
  90. /* MSVC compilers before VC7 don't have __func__ at all; later ones call it
  91. * __FUNCTION__. */
  92. #if _MSC_VER < 1300
  93. #define __func__ "???"
  94. #else
  95. #define __func__ __FUNCTION__
  96. #endif
  97. #else
  98. /* For platforms where autoconf works, make sure __func__ is defined
  99. * sanely. */
  100. #ifndef HAVE_MACRO__func__
  101. #ifdef HAVE_MACRO__FUNCTION__
  102. #define __func__ __FUNCTION__
  103. #elif HAVE_MACRO__FUNC__
  104. #define __func__ __FUNC__
  105. #else
  106. #define __func__ "???"
  107. #endif
  108. #endif /* ifndef MAVE_MACRO__func__ */
  109. #endif /* if not windows */
  110. #if defined(_MSC_VER) && (_MSC_VER < 1300)
  111. /* MSVC versions before 7 apparently don't believe that you can cast uint64_t
  112. * to double and really mean it. */
  113. extern INLINE double U64_TO_DBL(uint64_t x) {
  114. int64_t i = (int64_t) x;
  115. return (i < 0) ? ((double) INT64_MAX) : (double) i;
  116. }
  117. #define DBL_TO_U64(x) ((uint64_t)(int64_t) (x))
  118. #else
  119. #define U64_TO_DBL(x) ((double) (x))
  120. #define DBL_TO_U64(x) ((uint64_t) (x))
  121. #endif
  122. /* GCC has several useful attributes. */
  123. #if defined(__GNUC__) && __GNUC__ >= 3
  124. #define ATTR_NORETURN __attribute__((noreturn))
  125. #define ATTR_CONST __attribute__((const))
  126. #define ATTR_MALLOC __attribute__((malloc))
  127. #define ATTR_NORETURN __attribute__((noreturn))
  128. /* Alas, nonnull is not at present a good idea for us. We'd like to get
  129. * warnings when we pass NULL where we shouldn't (which nonnull does, albeit
  130. * spottily), but we don't want to tell the compiler to make optimizations
  131. * with the assumption that the argument can't be NULL (since this would make
  132. * many of our checks go away, and make our code less robust against
  133. * programming errors). Unfortunately, nonnull currently does both of these
  134. * things, and there's no good way to split them up.
  135. *
  136. * #define ATTR_NONNULL(x) __attribute__((nonnull x)) */
  137. #define ATTR_NONNULL(x)
  138. /** Macro: Evaluates to <b>exp</b> and hints the compiler that the value
  139. * of <b>exp</b> will probably be true.
  140. *
  141. * In other words, "if (PREDICT_LIKELY(foo))" is the same as "if (foo)",
  142. * except that it tells the compiler that the branch will be taken most of the
  143. * time. This can generate slightly better code with some CPUs.
  144. */
  145. #define PREDICT_LIKELY(exp) __builtin_expect(!!(exp), 1)
  146. /** Macro: Evaluates to <b>exp</b> and hints the compiler that the value
  147. * of <b>exp</b> will probably be false.
  148. *
  149. * In other words, "if (PREDICT_UNLIKELY(foo))" is the same as "if (foo)",
  150. * except that it tells the compiler that the branch will usually not be
  151. * taken. This can generate slightly better code with some CPUs.
  152. */
  153. #define PREDICT_UNLIKELY(exp) __builtin_expect(!!(exp), 0)
  154. #else
  155. #define ATTR_NORETURN
  156. #define ATTR_CONST
  157. #define ATTR_MALLOC
  158. #define ATTR_NORETURN
  159. #define ATTR_NONNULL(x)
  160. #define PREDICT_LIKELY(exp) (exp)
  161. #define PREDICT_UNLIKELY(exp) (exp)
  162. #endif
  163. /** Expands to a syntactically valid empty statement. */
  164. #define STMT_NIL (void)0
  165. /** Expands to a syntactically valid empty statement, explicitly (void)ing its
  166. * argument. */
  167. #define STMT_VOID(a) while (0) { (void)(a); }
  168. #ifdef __GNUC__
  169. /** STMT_BEGIN and STMT_END are used to wrap blocks inside macros so that
  170. * the macro can be used as if it were a single C statement. */
  171. #define STMT_BEGIN (void) ({
  172. #define STMT_END })
  173. #elif defined(sun) || defined(__sun__)
  174. #define STMT_BEGIN if (1) {
  175. #define STMT_END } else STMT_NIL
  176. #else
  177. #define STMT_BEGIN do {
  178. #define STMT_END } while (0)
  179. #endif
  180. /* ===== String compatibility */
  181. #ifdef _WIN32
  182. /* Windows names string functions differently from most other platforms. */
  183. #define strncasecmp _strnicmp
  184. #define strcasecmp _stricmp
  185. #endif
  186. #ifndef HAVE_STRLCAT
  187. size_t strlcat(char *dst, const char *src, size_t siz) ATTR_NONNULL((1,2));
  188. #endif
  189. #ifndef HAVE_STRLCPY
  190. size_t strlcpy(char *dst, const char *src, size_t siz) ATTR_NONNULL((1,2));
  191. #endif
  192. #ifdef _MSC_VER
  193. /** Casts the uint64_t value in <b>a</b> to the right type for an argument
  194. * to printf. */
  195. #define U64_PRINTF_ARG(a) (a)
  196. /** Casts the uint64_t* value in <b>a</b> to the right type for an argument
  197. * to scanf. */
  198. #define U64_SCANF_ARG(a) (a)
  199. /** Expands to a literal uint64_t-typed constant for the value <b>n</b>. */
  200. #define U64_LITERAL(n) (n ## ui64)
  201. #define I64_PRINTF_ARG(a) (a)
  202. #define I64_SCANF_ARG(a) (a)
  203. #define I64_LITERAL(n) (n ## i64)
  204. #else
  205. #define U64_PRINTF_ARG(a) ((long long unsigned int)(a))
  206. #define U64_SCANF_ARG(a) ((long long unsigned int*)(a))
  207. #define U64_LITERAL(n) (n ## llu)
  208. #define I64_PRINTF_ARG(a) ((long long signed int)(a))
  209. #define I64_SCANF_ARG(a) ((long long signed int*)(a))
  210. #define I64_LITERAL(n) (n ## ll)
  211. #endif
  212. #if defined(_MSC_VER) || defined(__MINGW32__) || defined(__MINGW64__)
  213. /** The formatting string used to put a uint64_t value in a printf() or
  214. * scanf() function. See also U64_PRINTF_ARG and U64_SCANF_ARG. */
  215. #define U64_FORMAT "%I64u"
  216. #define I64_FORMAT "%I64d"
  217. #else
  218. #define U64_FORMAT "%llu"
  219. #define I64_FORMAT "%lld"
  220. #endif
  221. #if (SIZEOF_INTPTR_T == SIZEOF_INT)
  222. #define INTPTR_T_FORMAT "%d"
  223. #define INTPTR_PRINTF_ARG(x) ((int)(x))
  224. #elif (SIZEOF_INTPTR_T == SIZEOF_LONG)
  225. #define INTPTR_T_FORMAT "%ld"
  226. #define INTPTR_PRINTF_ARG(x) ((long)(x))
  227. #elif (SIZEOF_INTPTR_T == 8)
  228. #define INTPTR_T_FORMAT I64_FORMAT
  229. #define INTPTR_PRINTF_ARG(x) I64_PRINTF_ARG(x)
  230. #else
  231. #error Unknown: SIZEOF_INTPTR_T
  232. #endif
  233. /** Represents an mmaped file. Allocated via tor_mmap_file; freed with
  234. * tor_munmap_file. */
  235. typedef struct tor_mmap_t {
  236. const char *data; /**< Mapping of the file's contents. */
  237. size_t size; /**< Size of the file. */
  238. /* None of the fields below should be accessed from outside compat.c */
  239. #ifdef HAVE_SYS_MMAN_H
  240. size_t mapping_size; /**< Size of the actual mapping. (This is this file
  241. * size, rounded up to the nearest page.) */
  242. #elif defined _WIN32
  243. HANDLE mmap_handle;
  244. #endif
  245. } tor_mmap_t;
  246. tor_mmap_t *tor_mmap_file(const char *filename) ATTR_NONNULL((1));
  247. void tor_munmap_file(tor_mmap_t *handle) ATTR_NONNULL((1));
  248. int tor_snprintf(char *str, size_t size, const char *format, ...)
  249. CHECK_PRINTF(3,4) ATTR_NONNULL((1,3));
  250. int tor_vsnprintf(char *str, size_t size, const char *format, va_list args)
  251. CHECK_PRINTF(3,0) ATTR_NONNULL((1,3));
  252. int tor_asprintf(char **strp, const char *fmt, ...)
  253. CHECK_PRINTF(2,3);
  254. int tor_vasprintf(char **strp, const char *fmt, va_list args)
  255. CHECK_PRINTF(2,0);
  256. const void *tor_memmem(const void *haystack, size_t hlen, const void *needle,
  257. size_t nlen) ATTR_NONNULL((1,3));
  258. static const void *tor_memstr(const void *haystack, size_t hlen,
  259. const char *needle) ATTR_NONNULL((1,3));
  260. static INLINE const void *
  261. tor_memstr(const void *haystack, size_t hlen, const char *needle)
  262. {
  263. return tor_memmem(haystack, hlen, needle, strlen(needle));
  264. }
  265. /* Much of the time when we're checking ctypes, we're doing spec compliance,
  266. * which all assumes we're doing ASCII. */
  267. #define DECLARE_CTYPE_FN(name) \
  268. static int TOR_##name(char c); \
  269. extern const uint32_t TOR_##name##_TABLE[]; \
  270. static INLINE int TOR_##name(char c) { \
  271. uint8_t u = c; \
  272. return !!(TOR_##name##_TABLE[(u >> 5) & 7] & (1 << (u & 31))); \
  273. }
  274. DECLARE_CTYPE_FN(ISALPHA)
  275. DECLARE_CTYPE_FN(ISALNUM)
  276. DECLARE_CTYPE_FN(ISSPACE)
  277. DECLARE_CTYPE_FN(ISDIGIT)
  278. DECLARE_CTYPE_FN(ISXDIGIT)
  279. DECLARE_CTYPE_FN(ISPRINT)
  280. DECLARE_CTYPE_FN(ISLOWER)
  281. DECLARE_CTYPE_FN(ISUPPER)
  282. extern const char TOR_TOUPPER_TABLE[];
  283. extern const char TOR_TOLOWER_TABLE[];
  284. #define TOR_TOLOWER(c) (TOR_TOLOWER_TABLE[(uint8_t)c])
  285. #define TOR_TOUPPER(c) (TOR_TOUPPER_TABLE[(uint8_t)c])
  286. char *tor_strtok_r_impl(char *str, const char *sep, char **lasts);
  287. #ifdef HAVE_STRTOK_R
  288. #define tor_strtok_r(str, sep, lasts) strtok_r(str, sep, lasts)
  289. #else
  290. #define tor_strtok_r(str, sep, lasts) tor_strtok_r_impl(str, sep, lasts)
  291. #endif
  292. #ifdef _WIN32
  293. #define SHORT_FILE__ (tor_fix_source_file(__FILE__))
  294. const char *tor_fix_source_file(const char *fname);
  295. #else
  296. #define SHORT_FILE__ (__FILE__)
  297. #define tor_fix_source_file(s) (s)
  298. #endif
  299. /* ===== Time compatibility */
  300. #if !defined(HAVE_GETTIMEOFDAY) && !defined(HAVE_STRUCT_TIMEVAL_TV_SEC)
  301. /** Implementation of timeval for platforms that don't have it. */
  302. struct timeval {
  303. time_t tv_sec;
  304. unsigned int tv_usec;
  305. };
  306. #endif
  307. void tor_gettimeofday(struct timeval *timeval);
  308. struct tm *tor_localtime_r(const time_t *timep, struct tm *result);
  309. struct tm *tor_gmtime_r(const time_t *timep, struct tm *result);
  310. #ifndef timeradd
  311. /** Replacement for timeradd on platforms that do not have it: sets tvout to
  312. * the sum of tv1 and tv2. */
  313. #define timeradd(tv1,tv2,tvout) \
  314. do { \
  315. (tvout)->tv_sec = (tv1)->tv_sec + (tv2)->tv_sec; \
  316. (tvout)->tv_usec = (tv1)->tv_usec + (tv2)->tv_usec; \
  317. if ((tvout)->tv_usec >= 1000000) { \
  318. (tvout)->tv_usec -= 1000000; \
  319. (tvout)->tv_sec++; \
  320. } \
  321. } while (0)
  322. #endif
  323. #ifndef timersub
  324. /** Replacement for timersub on platforms that do not have it: sets tvout to
  325. * tv1 minus tv2. */
  326. #define timersub(tv1,tv2,tvout) \
  327. do { \
  328. (tvout)->tv_sec = (tv1)->tv_sec - (tv2)->tv_sec; \
  329. (tvout)->tv_usec = (tv1)->tv_usec - (tv2)->tv_usec; \
  330. if ((tvout)->tv_usec < 0) { \
  331. (tvout)->tv_usec += 1000000; \
  332. (tvout)->tv_sec--; \
  333. } \
  334. } while (0)
  335. #endif
  336. #ifndef timercmp
  337. /** Replacement for timersub on platforms that do not have it: returns true
  338. * iff the relational operator "op" makes the expression tv1 op tv2 true.
  339. *
  340. * Note that while this definition should work for all boolean opeators, some
  341. * platforms' native timercmp definitions do not support >=, <=, or ==. So
  342. * don't use those.
  343. */
  344. #define timercmp(tv1,tv2,op) \
  345. (((tv1)->tv_sec == (tv2)->tv_sec) ? \
  346. ((tv1)->tv_usec op (tv2)->tv_usec) : \
  347. ((tv1)->tv_sec op (tv2)->tv_sec))
  348. #endif
  349. /* ===== File compatibility */
  350. int tor_open_cloexec(const char *path, int flags, unsigned mode);
  351. FILE *tor_fopen_cloexec(const char *path, const char *mode);
  352. int replace_file(const char *from, const char *to);
  353. int touch_file(const char *fname);
  354. typedef struct tor_lockfile_t tor_lockfile_t;
  355. tor_lockfile_t *tor_lockfile_lock(const char *filename, int blocking,
  356. int *locked_out);
  357. void tor_lockfile_unlock(tor_lockfile_t *lockfile);
  358. off_t tor_fd_getpos(int fd);
  359. int tor_fd_seekend(int fd);
  360. #ifdef _WIN32
  361. #define PATH_SEPARATOR "\\"
  362. #else
  363. #define PATH_SEPARATOR "/"
  364. #endif
  365. /* ===== Net compatibility */
  366. #if (SIZEOF_SOCKLEN_T == 0)
  367. typedef int socklen_t;
  368. #endif
  369. #ifdef _WIN32
  370. /* XXX Actually, this should arguably be SOCKET; we use intptr_t here so that
  371. * any inadvertant checks for the socket being <= 0 or > 0 will probably
  372. * still work. */
  373. #define tor_socket_t intptr_t
  374. #define TOR_SOCKET_T_FORMAT INTPTR_T_FORMAT
  375. #define SOCKET_OK(s) ((SOCKET)(s) != INVALID_SOCKET)
  376. #define TOR_INVALID_SOCKET INVALID_SOCKET
  377. #else
  378. /** Type used for a network socket. */
  379. #define tor_socket_t int
  380. #define TOR_SOCKET_T_FORMAT "%d"
  381. /** Macro: true iff 's' is a possible value for a valid initialized socket. */
  382. #define SOCKET_OK(s) ((s) >= 0)
  383. /** Error/uninitialized value for a tor_socket_t. */
  384. #define TOR_INVALID_SOCKET (-1)
  385. #endif
  386. int tor_close_socket(tor_socket_t s);
  387. tor_socket_t tor_open_socket(int domain, int type, int protocol);
  388. tor_socket_t tor_accept_socket(tor_socket_t sockfd, struct sockaddr *addr,
  389. socklen_t *len);
  390. int get_n_open_sockets(void);
  391. #define tor_socket_send(s, buf, len, flags) send(s, buf, len, flags)
  392. #define tor_socket_recv(s, buf, len, flags) recv(s, buf, len, flags)
  393. /** Implementation of struct in6_addr for platforms that do not have it.
  394. * Generally, these platforms are ones without IPv6 support, but we want to
  395. * have a working in6_addr there anyway, so we can use it to parse IPv6
  396. * addresses. */
  397. #if !defined(HAVE_STRUCT_IN6_ADDR)
  398. struct in6_addr
  399. {
  400. union {
  401. uint8_t u6_addr8[16];
  402. uint16_t u6_addr16[8];
  403. uint32_t u6_addr32[4];
  404. } in6_u;
  405. #define s6_addr in6_u.u6_addr8
  406. #define s6_addr16 in6_u.u6_addr16
  407. #define s6_addr32 in6_u.u6_addr32
  408. };
  409. #endif
  410. /** @{ */
  411. /** Many BSD variants seem not to define these. */
  412. #if defined(__APPLE__) || defined(__darwin__) || defined(__FreeBSD__) \
  413. || defined(__NetBSD__) || defined(__OpenBSD__)
  414. #ifndef s6_addr16
  415. #define s6_addr16 __u6_addr.__u6_addr16
  416. #endif
  417. #ifndef s6_addr32
  418. #define s6_addr32 __u6_addr.__u6_addr32
  419. #endif
  420. #endif
  421. /** @} */
  422. #ifndef HAVE_SA_FAMILY_T
  423. typedef uint16_t sa_family_t;
  424. #endif
  425. /** @{ */
  426. /** Apparently, MS and Solaris don't define s6_addr16 or s6_addr32; these
  427. * macros get you a pointer to s6_addr32 or local equivalent. */
  428. #ifdef HAVE_STRUCT_IN6_ADDR_S6_ADDR32
  429. #define S6_ADDR32(x) ((uint32_t*)(x).s6_addr32)
  430. #else
  431. #define S6_ADDR32(x) ((uint32_t*)((char*)&(x).s6_addr))
  432. #endif
  433. #ifdef HAVE_STRUCT_IN6_ADDR_S6_ADDR16
  434. #define S6_ADDR16(x) ((uint16_t*)(x).s6_addr16)
  435. #else
  436. #define S6_ADDR16(x) ((uint16_t*)((char*)&(x).s6_addr))
  437. #endif
  438. /** @} */
  439. /** Implementation of struct sockaddr_in6 on platforms that do not have
  440. * it. See notes on struct in6_addr. */
  441. #if !defined(HAVE_STRUCT_SOCKADDR_IN6)
  442. struct sockaddr_in6 {
  443. sa_family_t sin6_family;
  444. uint16_t sin6_port;
  445. // uint32_t sin6_flowinfo;
  446. struct in6_addr sin6_addr;
  447. // uint32_t sin6_scope_id;
  448. };
  449. #endif
  450. int tor_inet_aton(const char *cp, struct in_addr *addr) ATTR_NONNULL((1,2));
  451. const char *tor_inet_ntop(int af, const void *src, char *dst, size_t len);
  452. int tor_inet_pton(int af, const char *src, void *dst);
  453. int tor_lookup_hostname(const char *name, uint32_t *addr) ATTR_NONNULL((1,2));
  454. void set_socket_nonblocking(tor_socket_t socket);
  455. int tor_socketpair(int family, int type, int protocol, tor_socket_t fd[2]);
  456. int network_init(void);
  457. /* For stupid historical reasons, windows sockets have an independent
  458. * set of errnos, and an independent way to get them. Also, you can't
  459. * always believe WSAEWOULDBLOCK. Use the macros below to compare
  460. * errnos against expected values, and use tor_socket_errno to find
  461. * the actual errno after a socket operation fails.
  462. */
  463. #if defined(_WIN32)
  464. /** Expands to WSA<b>e</b> on Windows, and to <b>e</b> elsewhere. */
  465. #define SOCK_ERRNO(e) WSA##e
  466. /** Return true if e is EAGAIN or the local equivalent. */
  467. #define ERRNO_IS_EAGAIN(e) ((e) == EAGAIN || (e) == WSAEWOULDBLOCK)
  468. /** Return true if e is EINPROGRESS or the local equivalent. */
  469. #define ERRNO_IS_EINPROGRESS(e) ((e) == WSAEINPROGRESS)
  470. /** Return true if e is EINPROGRESS or the local equivalent as returned by
  471. * a call to connect(). */
  472. #define ERRNO_IS_CONN_EINPROGRESS(e) \
  473. ((e) == WSAEINPROGRESS || (e)== WSAEINVAL || (e) == WSAEWOULDBLOCK)
  474. /** Return true if e is EAGAIN or another error indicating that a call to
  475. * accept() has no pending connections to return. */
  476. #define ERRNO_IS_ACCEPT_EAGAIN(e) ERRNO_IS_EAGAIN(e)
  477. /** Return true if e is EMFILE or another error indicating that a call to
  478. * accept() has failed because we're out of fds or something. */
  479. #define ERRNO_IS_ACCEPT_RESOURCE_LIMIT(e) \
  480. ((e) == WSAEMFILE || (e) == WSAENOBUFS)
  481. /** Return true if e is EADDRINUSE or the local equivalent. */
  482. #define ERRNO_IS_EADDRINUSE(e) ((e) == WSAEADDRINUSE)
  483. int tor_socket_errno(tor_socket_t sock);
  484. const char *tor_socket_strerror(int e);
  485. #else
  486. #define SOCK_ERRNO(e) e
  487. #define ERRNO_IS_EAGAIN(e) ((e) == EAGAIN)
  488. #define ERRNO_IS_EINPROGRESS(e) ((e) == EINPROGRESS)
  489. #define ERRNO_IS_CONN_EINPROGRESS(e) ((e) == EINPROGRESS)
  490. #define ERRNO_IS_ACCEPT_EAGAIN(e) ((e) == EAGAIN || (e) == ECONNABORTED)
  491. #define ERRNO_IS_ACCEPT_RESOURCE_LIMIT(e) \
  492. ((e) == EMFILE || (e) == ENFILE || (e) == ENOBUFS || (e) == ENOMEM)
  493. #define ERRNO_IS_EADDRINUSE(e) ((e) == EADDRINUSE)
  494. #define tor_socket_errno(sock) (errno)
  495. #define tor_socket_strerror(e) strerror(e)
  496. #endif
  497. /** Specified SOCKS5 status codes. */
  498. typedef enum {
  499. SOCKS5_SUCCEEDED = 0x00,
  500. SOCKS5_GENERAL_ERROR = 0x01,
  501. SOCKS5_NOT_ALLOWED = 0x02,
  502. SOCKS5_NET_UNREACHABLE = 0x03,
  503. SOCKS5_HOST_UNREACHABLE = 0x04,
  504. SOCKS5_CONNECTION_REFUSED = 0x05,
  505. SOCKS5_TTL_EXPIRED = 0x06,
  506. SOCKS5_COMMAND_NOT_SUPPORTED = 0x07,
  507. SOCKS5_ADDRESS_TYPE_NOT_SUPPORTED = 0x08,
  508. } socks5_reply_status_t;
  509. /* ===== Insecure rng */
  510. void tor_init_weak_random(unsigned seed);
  511. long tor_weak_random(void);
  512. #define TOR_RAND_MAX (RAND_MAX)
  513. /* ===== OS compatibility */
  514. const char *get_uname(void);
  515. uint16_t get_uint16(const void *cp) ATTR_NONNULL((1));
  516. uint32_t get_uint32(const void *cp) ATTR_NONNULL((1));
  517. uint64_t get_uint64(const void *cp) ATTR_NONNULL((1));
  518. void set_uint16(void *cp, uint16_t v) ATTR_NONNULL((1));
  519. void set_uint32(void *cp, uint32_t v) ATTR_NONNULL((1));
  520. void set_uint64(void *cp, uint64_t v) ATTR_NONNULL((1));
  521. /* These uint8 variants are defined to make the code more uniform. */
  522. #define get_uint8(cp) (*(const uint8_t*)(cp))
  523. static void set_uint8(void *cp, uint8_t v);
  524. static INLINE void
  525. set_uint8(void *cp, uint8_t v)
  526. {
  527. *(uint8_t*)cp = v;
  528. }
  529. #if !defined(HAVE_RLIM_T)
  530. typedef unsigned long rlim_t;
  531. #endif
  532. int set_max_file_descriptors(rlim_t limit, int *max);
  533. int tor_disable_debugger_attach(void);
  534. int switch_id(const char *user);
  535. #ifdef HAVE_PWD_H
  536. char *get_user_homedir(const char *username);
  537. #endif
  538. int get_parent_directory(char *fname);
  539. char *make_path_absolute(char *fname);
  540. char **get_environment(void);
  541. int spawn_func(void (*func)(void *), void *data);
  542. void spawn_exit(void) ATTR_NORETURN;
  543. #if defined(ENABLE_THREADS) && defined(_WIN32)
  544. #define USE_WIN32_THREADS
  545. #define TOR_IS_MULTITHREADED 1
  546. #elif (defined(ENABLE_THREADS) && defined(HAVE_PTHREAD_H) && \
  547. defined(HAVE_PTHREAD_CREATE))
  548. #define USE_PTHREADS
  549. #define TOR_IS_MULTITHREADED 1
  550. #else
  551. #undef TOR_IS_MULTITHREADED
  552. #endif
  553. int compute_num_cpus(void);
  554. /* Because we use threads instead of processes on most platforms (Windows,
  555. * Linux, etc), we need locking for them. On platforms with poor thread
  556. * support or broken gethostbyname_r, these functions are no-ops. */
  557. /** A generic lock structure for multithreaded builds. */
  558. typedef struct tor_mutex_t {
  559. #if defined(USE_WIN32_THREADS)
  560. /** Windows-only: on windows, we implement locks with CRITICAL_SECTIONS. */
  561. CRITICAL_SECTION mutex;
  562. #elif defined(USE_PTHREADS)
  563. /** Pthreads-only: with pthreads, we implement locks with
  564. * pthread_mutex_t. */
  565. pthread_mutex_t mutex;
  566. #else
  567. /** No-threads only: Dummy variable so that tor_mutex_t takes up space. */
  568. int _unused;
  569. #endif
  570. } tor_mutex_t;
  571. int tor_mlockall(void);
  572. #ifdef TOR_IS_MULTITHREADED
  573. tor_mutex_t *tor_mutex_new(void);
  574. void tor_mutex_init(tor_mutex_t *m);
  575. void tor_mutex_acquire(tor_mutex_t *m);
  576. void tor_mutex_release(tor_mutex_t *m);
  577. void tor_mutex_free(tor_mutex_t *m);
  578. void tor_mutex_uninit(tor_mutex_t *m);
  579. unsigned long tor_get_thread_id(void);
  580. void tor_threads_init(void);
  581. #else
  582. #define tor_mutex_new() ((tor_mutex_t*)tor_malloc(sizeof(int)))
  583. #define tor_mutex_init(m) STMT_NIL
  584. #define tor_mutex_acquire(m) STMT_VOID(m)
  585. #define tor_mutex_release(m) STMT_NIL
  586. #define tor_mutex_free(m) STMT_BEGIN tor_free(m); STMT_END
  587. #define tor_mutex_uninit(m) STMT_NIL
  588. #define tor_get_thread_id() (1UL)
  589. #define tor_threads_init() STMT_NIL
  590. #endif
  591. void set_main_thread(void);
  592. int in_main_thread(void);
  593. #ifdef TOR_IS_MULTITHREADED
  594. #if 0
  595. typedef struct tor_cond_t tor_cond_t;
  596. tor_cond_t *tor_cond_new(void);
  597. void tor_cond_free(tor_cond_t *cond);
  598. int tor_cond_wait(tor_cond_t *cond, tor_mutex_t *mutex);
  599. void tor_cond_signal_one(tor_cond_t *cond);
  600. void tor_cond_signal_all(tor_cond_t *cond);
  601. #endif
  602. #endif
  603. /** Macros for MIN/MAX. Never use these when the arguments could have
  604. * side-effects.
  605. * {With GCC extensions we could probably define a safer MIN/MAX. But
  606. * depending on that safety would be dangerous, since not every platform
  607. * has it.}
  608. **/
  609. #ifndef MAX
  610. #define MAX(a,b) ( ((a)<(b)) ? (b) : (a) )
  611. #endif
  612. #ifndef MIN
  613. #define MIN(a,b) ( ((a)>(b)) ? (b) : (a) )
  614. #endif
  615. /* Platform-specific helpers. */
  616. #ifdef _WIN32
  617. char *format_win32_error(DWORD err);
  618. #endif
  619. /*for some reason my compiler doesn't have these version flags defined
  620. a nice homework assignment for someone one day is to define the rest*/
  621. //these are the values as given on MSDN
  622. #ifdef _WIN32
  623. #ifndef VER_SUITE_EMBEDDEDNT
  624. #define VER_SUITE_EMBEDDEDNT 0x00000040
  625. #endif
  626. #ifndef VER_SUITE_SINGLEUSERTS
  627. #define VER_SUITE_SINGLEUSERTS 0x00000100
  628. #endif
  629. #endif
  630. #endif