compat.h 9.42 KB
Newer Older
Nick Mathewson's avatar
Nick Mathewson committed
1
/* Copyright 2003-2004 Roger Dingledinex
Roger Dingledine's avatar
Roger Dingledine committed
2
 * Copyright 2004-2006 Roger Dingledine, Nick Mathewson */
3
4
5
6
7
/* See LICENSE for licensing information */
/* $Id$ */

#ifndef __COMPAT_H
#define __COMPAT_H
8
#define COMPAT_H_ID "$Id$"
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

#include "orconfig.h"
#include "torint.h"
#ifdef MS_WINDOWS
#define WIN32_WINNT 0x400
#define _WIN32_WINNT 0x400
#define WIN32_LEAN_AND_MEAN
#if (_MSC_VER <= 1300)
#include <winsock.h>
#else
#include <winsock2.h>
#include <ws2tcpip.h>
#endif
#endif
#ifdef HAVE_SYS_TYPES_H
#include <sys/types.h>
#endif
26
27
28
#ifdef HAVE_SYS_TIME_H
#include <sys/time.h>
#endif
29
30
31
#ifdef HAVE_TIME_H
#include <time.h>
#endif
32
33
34
35
36
37
38
39
40
41
42
#include <stdarg.h>

#ifndef NULL_REP_IS_ZERO_BYTES
#error "It seems your platform does not represent NULL as zero. We can't cope."
#endif

/* ===== Compiler compatibility */

/* GCC can check printf types on arbitrary functions. */
#ifdef __GNUC__
#define CHECK_PRINTF(formatIdx, firstArg) \
43
   __attribute__ ((format(printf, formatIdx, firstArg)))
44
45
46
47
48
49
50
51
52
53
54
#else
#define CHECK_PRINTF(formatIdx, firstArg)
#endif

/* inline is __inline on windows. */
#ifdef MS_WINDOWS
#define INLINE __inline
#else
#define INLINE inline
#endif

55
56
57
58
59
60
61
62
/* Try to get a reasonable __func__ substitute in place. */
#if defined(_MSC_VER)
/* MSVC compilers before VC7 don't have __func__ at all; later ones call it
 * __FUNCTION__. */
#if _MSC_VER < 1300
#define __func__ "???"
#else
#define __func__ __FUNCTION__
63
64
#endif

65
66
67
68
69
70
71
72
73
74
#else
/* For platforms where autoconf works, make sure __func__ is defined
 * sanely. */
#ifndef HAVE_MACRO__func__
#ifdef HAVE_MACRO__FUNCTION__
#define __func__ __FUNCTION__
#elif HAVE_MACRO__FUNC__
#define __func__ __FUNC__
#else
#define __func__ "???"
75
#endif
76
#endif /* ifndef MAVE_MACRO__func__ */
Nick Mathewson's avatar
Nick Mathewson committed
77
#endif /* if not windows */
78

79
80
81
#if defined(_MSC_VER) && (_MSC_VER < 1300)
/* MSVC versions before 7 apparently don't believe that you can cast uint64_t
 * to double and really mean it. */
Nick Mathewson's avatar
Nick Mathewson committed
82
extern INLINE double U64_TO_DBL(uint64_t x) {
83
84
85
86
87
88
89
90
91
  int64_t i = (int64_t) x;
  return (i < 0) ? ((double) INT64_MAX) : (double) i;
}
#define DBL_TO_U64(x) ((uint64_t)(int64_t) (x))
#else
#define U64_TO_DBL(x) ((double) (x))
#define DBL_TO_U64(x) ((uint64_t) (x))
#endif

92
93
94
95
96
97
/* GCC has several useful attributes. */
#ifdef __GNUC__
#define ATTR_NORETURN __attribute__((noreturn))
#define ATTR_PURE __attribute__((pure))
#define ATTR_MALLOC __attribute__((malloc))
#define ATTR_NONNULL(x) __attribute__((nonnull x))
98
#define PREDICT(exp, val) __builtin_expect((exp), (val))
99
100
101
102
103
#else
#define ATTR_NORETURN
#define ATTR_PURE
#define ATTR_MALLOC
#define ATTR_NONNULL(x)
104
#define PREDICT(exp, val)
105
106
#endif

107
108
109
110
111
112
/* ===== String compatibility */
#ifdef MS_WINDOWS
/* Windows names string functions differently from most other platforms. */
#define strncasecmp strnicmp
#define strcasecmp stricmp
#endif
113
#ifndef HAVE_STRLCAT
114
size_t strlcat(char *dst, const char *src, size_t siz) ATTR_NONNULL((1,2));
115
116
#endif
#ifndef HAVE_STRLCPY
117
size_t strlcpy(char *dst, const char *src, size_t siz) ATTR_NONNULL((1,2));
118
#endif
119

120
#ifdef _MSC_VER
121
122
#define U64_PRINTF_ARG(a) (a)
#define U64_SCANF_ARG(a) (a)
123
#define U64_FORMAT "%I64u"
124
#define U64_LITERAL(n) (n ## ui64)
125
#else
126
#define U64_PRINTF_ARG(a) ((long long unsigned int)(a))
Roger Dingledine's avatar
Roger Dingledine committed
127
#define U64_SCANF_ARG(a) ((long long unsigned int*)(a))
128
#define U64_FORMAT "%llu"
129
#define U64_LITERAL(n) (n ## llu)
130
#endif
131

132
/** Opaque bookkeeping type used for mmap accounting. */
133
134
135
136
typedef struct tor_mmap_t {
  const char *data;
  size_t size;
} tor_mmap_t;
137

138
139
tor_mmap_t *tor_mmap_file(const char *filename) ATTR_NONNULL((1));
void tor_munmap_file(tor_mmap_t *handle) ATTR_NONNULL((1));
140

141
int tor_snprintf(char *str, size_t size, const char *format, ...)
142
143
144
  CHECK_PRINTF(3,4) ATTR_NONNULL((1,3));
int tor_vsnprintf(char *str, size_t size, const char *format, va_list args)
  ATTR_NONNULL((1,3));
145

Nick Mathewson's avatar
Nick Mathewson committed
146
const void *tor_memmem(const void *haystack, size_t hlen, const void *needle,
147
                       size_t nlen)  ATTR_PURE ATTR_NONNULL((1,3));
Nick Mathewson's avatar
Nick Mathewson committed
148

149
#define TOR_ISALPHA(c)   isalpha((int)(unsigned char)(c))
150
#define TOR_ISALNUM(c)   isalnum((int)(unsigned char)(c))
151
152
153
#define TOR_ISSPACE(c)   isspace((int)(unsigned char)(c))
#define TOR_ISXDIGIT(c) isxdigit((int)(unsigned char)(c))
#define TOR_ISDIGIT(c)   isdigit((int)(unsigned char)(c))
Nick Mathewson's avatar
Nick Mathewson committed
154
#define TOR_ISPRINT(c)   isprint((int)(unsigned char)(c))
155
156
#define TOR_ISLOWER(c)   islower((int)(unsigned char)(c))
#define TOR_ISUPPER(c)   isupper((int)(unsigned char)(c))
157

158
159
160
161
162
163
164
#ifdef MS_WINDOWS
#define _SHORT_FILE_ (tor_fix_source_file(__FILE__))
const char *tor_fix_source_file(const char *fname);
#else
#define _SHORT_FILE_ (__FILE__)
#define tor_fix_source_file(s) (s)
#endif
165

166
167
168
169
170
171
172
173
174
175
/* ===== Time compatibility */
#if !defined(HAVE_GETTIMEOFDAY) && !defined(HAVE_STRUCT_TIMEVAL_TV_SEC)
struct timeval {
  time_t tv_sec;
  unsigned int tv_usec;
};
#endif

void tor_gettimeofday(struct timeval *timeval);

176
177
178
179
180
181
182
183
184
185
186
187
#ifdef HAVE_LOCALTIME_R
#define tor_localtime_r localtime_r
#else
struct tm *tor_localtime_r(const time_t *timep, struct tm *result);
#endif

#ifdef HAVE_GMTIME_R
#define tor_gmtime_r gmtime_r
#else
struct tm *tor_gmtime_r(const time_t *timep, struct tm *result);
#endif

188
189
/* ===== File compatibility */
int replace_file(const char *from, const char *to);
190
int touch_file(const char *fname);
191

192
193
194
195
196
197
#ifdef MS_WINDOWS
#define PATH_SEPARATOR "\\"
#else
#define PATH_SEPARATOR "/"
#endif

198
199
/* ===== Net compatibility */
#ifdef MS_WINDOWS
Roger Dingledine's avatar
Roger Dingledine committed
200
/** On Windows, you have to call close() on fds returned by open(),
201
202
203
204
205
206
207
208
209
210
 * and closesocket() on fds returned by socket().  On Unix, everything
 * gets close()'d.  We abstract this difference by always using
 * tor_close_socket to close sockets, and always using close() on
 * files.
 */
#define tor_close_socket(s) closesocket(s)
#else
#define tor_close_socket(s) close(s)
#endif

211
212
213
214
#if (SIZEOF_SOCKLEN_T == 0)
typedef int socklen_t;
#endif

215
struct in_addr;
216
217
int tor_inet_aton(const char *cp, struct in_addr *addr) ATTR_NONNULL((1,2));
int tor_lookup_hostname(const char *name, uint32_t *addr) ATTR_NONNULL((1,2));
218
219
void set_socket_nonblocking(int socket);
int tor_socketpair(int family, int type, int protocol, int fd[2]);
220
int network_init(void);
221
222
223
224
225
226
227
228
229
230
231
232
233
/* For stupid historical reasons, windows sockets have an independent
 * set of errnos, and an independent way to get them.  Also, you can't
 * always believe WSAEWOULDBLOCK.  Use the macros below to compare
 * errnos against expected values, and use tor_socket_errno to find
 * the actual errno after a socket operation fails.
 */
#ifdef MS_WINDOWS
/** Return true if e is EAGAIN or the local equivalent. */
#define ERRNO_IS_EAGAIN(e)           ((e) == EAGAIN || (e) == WSAEWOULDBLOCK)
/** Return true if e is EINPROGRESS or the local equivalent. */
#define ERRNO_IS_EINPROGRESS(e)      ((e) == WSAEINPROGRESS)
/** Return true if e is EINPROGRESS or the local equivalent as returned by
 * a call to connect(). */
234
235
#define ERRNO_IS_CONN_EINPROGRESS(e) \
  ((e) == WSAEINPROGRESS || (e)== WSAEINVAL || (e) == WSAEWOULDBLOCK)
236
237
238
239
240
241
242
/** Return true if e is EAGAIN or another error indicating that a call to
 * accept() has no pending connections to return. */
#define ERRNO_IS_ACCEPT_EAGAIN(e)    ERRNO_IS_EAGAIN(e)
/** Return true if e is EMFILE or another error indicating that a call to
 * accept() has failed because we're out of fds or something. */
#define ERRNO_IS_ACCEPT_RESOURCE_LIMIT(e) \
  ((e) == WSAEMFILE || (e) == WSAENOBUFS)
243
244
/** Return true if e is EADDRINUSE or the local equivalent. */
#define ERRNO_IS_EADDRINUSE(e)      ((e) == WSAEADDRINUSE)
245
246
247
248
249
250
251
252
253
int tor_socket_errno(int sock);
const char *tor_socket_strerror(int e);
#else
#define ERRNO_IS_EAGAIN(e)           ((e) == EAGAIN)
#define ERRNO_IS_EINPROGRESS(e)      ((e) == EINPROGRESS)
#define ERRNO_IS_CONN_EINPROGRESS(e) ((e) == EINPROGRESS)
#define ERRNO_IS_ACCEPT_EAGAIN(e)    ((e) == EAGAIN || (e) == ECONNABORTED)
#define ERRNO_IS_ACCEPT_RESOURCE_LIMIT(e) \
  ((e) == EMFILE || (e) == ENFILE || (e) == ENOBUFS || (e) == ENOMEM)
254
#define ERRNO_IS_EADDRINUSE(e)       ((e) == EADDRINUSE)
255
256
257
258
259
260
261
#define tor_socket_errno(sock)       (errno)
#define tor_socket_strerror(e)       strerror(e)
#endif

/* ===== OS compatibility */
const char *get_uname(void);

262
263
264
265
uint16_t get_uint16(const char *cp) ATTR_PURE ATTR_NONNULL((1));
uint32_t get_uint32(const char *cp) ATTR_PURE ATTR_NONNULL((1));
void set_uint16(char *cp, uint16_t v) ATTR_NONNULL((1));
void set_uint32(char *cp, uint32_t v) ATTR_NONNULL((1));
266

267
int set_max_file_descriptors(unsigned long limit, unsigned long cap);
268
int switch_id(char *user, char *group);
269
270
271
#ifdef HAVE_PWD_H
char *get_user_homedir(const char *username);
#endif
272
273

int spawn_func(int (*func)(void *), void *data);
274
void spawn_exit(void) ATTR_NORETURN;
275

Nick Mathewson's avatar
Nick Mathewson committed
276
#if defined(ENABLE_THREADS) && defined(MS_WINDOWS)
277
278
#define USE_WIN32_THREADS
#define TOR_IS_MULTITHREADED 1
279
280
#elif (defined(ENABLE_THREADS) && defined(HAVE_PTHREAD_H) && \
       defined(HAVE_PTHREAD_CREATE))
281
#define USE_PTHREADS
282
283
284
285
286
#define TOR_IS_MULTITHREADED 1
#else
#undef TOR_IS_MULTITHREADED
#endif

Roger Dingledine's avatar
Roger Dingledine committed
287
288
289
/* Because we use threads instead of processes on most platforms (Windows,
 * Linux, etc), we need locking for them.  On platforms with poor thread
 * support or broken gethostbyname_r, these functions are no-ops. */
290

291
292
293
294
295
296
297
298
299
300
301
typedef struct tor_mutex_t tor_mutex_t;
#ifdef TOR_IS_MULTITHREADED
tor_mutex_t *tor_mutex_new(void);
void tor_mutex_acquire(tor_mutex_t *m);
void tor_mutex_release(tor_mutex_t *m);
void tor_mutex_free(tor_mutex_t *m);
unsigned long tor_get_thread_id(void);
#else
#define tor_mutex_new() ((tor_mutex_t*)tor_malloc(sizeof(int)))
#define tor_mutex_acquire(m) do { } while (0)
#define tor_mutex_release(m) do { } while (0)
Nick Mathewson's avatar
Nick Mathewson committed
302
#define tor_mutex_free(m) do { tor_free(m); } while (0)
303
304
305
#define tor_get_thread_id() (1UL)
#endif

306
#endif
307