]>
Commit | Line | Data |
---|---|---|
1 | #ifndef RB_LIB_H | |
2 | #define RB_LIB_H 1 | |
3 | ||
4 | #include <librb-config.h> | |
5 | #include <stdio.h> | |
6 | #include <limits.h> | |
7 | #include <stdarg.h> | |
8 | #include <assert.h> | |
9 | #include <fcntl.h> | |
10 | #include <signal.h> | |
11 | #include <ctype.h> | |
12 | ||
13 | #ifdef __GNUC__ | |
14 | #undef alloca | |
15 | #define alloca __builtin_alloca | |
16 | #else | |
17 | # ifdef _MSC_VER | |
18 | # include <malloc.h> | |
19 | # define alloca _alloca | |
20 | # else | |
21 | # if RB_HAVE_ALLOCA_H | |
22 | # include <alloca.h> | |
23 | # else | |
24 | # ifdef _AIX | |
25 | #pragma alloca | |
26 | # else | |
27 | # ifndef alloca /* predefined by HP cc +Olibcalls */ | |
28 | char *alloca(); | |
29 | # endif | |
30 | # endif | |
31 | # endif | |
32 | # endif | |
33 | #endif /* __GNUC__ */ | |
34 | ||
35 | #ifdef __GNUC__ | |
36 | ||
37 | #ifdef rb_likely | |
38 | #undef rb_likely | |
39 | #endif | |
40 | #ifdef rb_unlikely | |
41 | #undef rb_unlikely | |
42 | #endif | |
43 | ||
44 | #if __GNUC__ == 2 && __GNUC_MINOR__ < 96 | |
45 | # define __builtin_expect(x, expected_value) (x) | |
46 | #endif | |
47 | ||
48 | #define rb_likely(x) __builtin_expect(!!(x), 1) | |
49 | #define rb_unlikely(x) __builtin_expect(!!(x), 0) | |
50 | ||
51 | #else /* !__GNUC__ */ | |
52 | ||
53 | #define UNUSED(x) x | |
54 | ||
55 | #ifdef rb_likely | |
56 | #undef rb_likely | |
57 | #endif | |
58 | #ifdef rb_unlikely | |
59 | #undef rb_unlikely | |
60 | #endif | |
61 | #define rb_likely(x) (x) | |
62 | #define rb_unlikely(x) (x) | |
63 | #endif | |
64 | ||
65 | #ifndef HOSTIPLEN | |
66 | #define HOSTIPLEN 53 | |
67 | #endif | |
68 | ||
69 | ||
70 | /* For those unfamiliar with GNU format attributes, a is the 1 based | |
71 | * argument number of the format string, and b is the 1 based argument | |
72 | * number of the variadic ... */ | |
73 | #ifdef __GNUC__ | |
74 | #define AFP(a,b) __attribute__((format (printf, a, b))) | |
75 | #else | |
76 | #define AFP(a,b) | |
77 | #endif | |
78 | ||
79 | ||
80 | #ifdef __GNUC__ | |
81 | #define slrb_assert(expr) ( \ | |
82 | rb_likely((expr)) || ( \ | |
83 | rb_lib_log( \ | |
84 | "file: %s line: %d (%s): Assertion failed: (%s)", \ | |
85 | __FILE__, __LINE__, __PRETTY_FUNCTION__, #expr), 0) \ | |
86 | ) | |
87 | #else | |
88 | #define slrb_assert(expr) ( \ | |
89 | rb_likely((expr)) || ( \ | |
90 | rb_lib_log( \ | |
91 | "file: %s line: %d: Assertion failed: (%s)", \ | |
92 | __FILE__, __LINE__, #expr), 0) \ | |
93 | ) | |
94 | #endif | |
95 | ||
96 | /* evaluates to true if assertion fails */ | |
97 | #ifdef SOFT_ASSERT | |
98 | #define lrb_assert(expr) (!slrb_assert(expr)) | |
99 | #else | |
100 | #define lrb_assert(expr) (assert(slrb_assert(expr)), 0) | |
101 | #endif | |
102 | ||
103 | #ifdef RB_SOCKADDR_HAS_SA_LEN | |
104 | #define ss_len sa_len | |
105 | #endif | |
106 | ||
107 | #define GET_SS_FAMILY(x) (((const struct sockaddr *)(x))->sa_family) | |
108 | #define SET_SS_FAMILY(x, y) ((((struct sockaddr *)(x))->sa_family) = y) | |
109 | #ifdef RB_SOCKADDR_HAS_SA_LEN | |
110 | #define SET_SS_LEN(x, y) do { \ | |
111 | struct sockaddr *_storage; \ | |
112 | _storage = ((struct sockaddr *)(x));\ | |
113 | _storage->sa_len = (y); \ | |
114 | } while (0) | |
115 | #define GET_SS_LEN(x) (((struct sockaddr *)(x))->sa_len) | |
116 | #else /* !RB_SOCKADDR_HAS_SA_LEN */ | |
117 | #define SET_SS_LEN(x, y) (((struct sockaddr *)(x))->sa_family = ((struct sockaddr *)(x))->sa_family) | |
118 | #define GET_SS_LEN(x) (((struct sockaddr *)(x))->sa_family == AF_INET ? sizeof(struct sockaddr_in) : sizeof(struct sockaddr_in6)) | |
119 | #endif | |
120 | ||
121 | #define GET_SS_PORT(x) (((struct sockaddr *)(x))->sa_family == AF_INET ? ((struct sockaddr_in *)(x))->sin_port : ((struct sockaddr_in6 *)(x))->sin6_port) | |
122 | #define SET_SS_PORT(x, y) do { \ | |
123 | if(((struct sockaddr *)(x))->sa_family == AF_INET) { \ | |
124 | ((struct sockaddr_in *)(x))->sin_port = (y); \ | |
125 | } else { \ | |
126 | ((struct sockaddr_in6 *)(x))->sin6_port = (y); \ | |
127 | } \ | |
128 | } while (0) | |
129 | ||
130 | #ifndef INADDRSZ | |
131 | #define INADDRSZ 4 | |
132 | #endif | |
133 | ||
134 | #ifndef IN6ADDRSZ | |
135 | #define IN6ADDRSZ 16 | |
136 | #endif | |
137 | ||
138 | #ifndef INT16SZ | |
139 | #define INT16SZ 2 | |
140 | #endif | |
141 | ||
142 | #ifndef UINT16_MAX | |
143 | #define UINT16_MAX (65535U) | |
144 | #endif | |
145 | ||
146 | #ifndef UINT32_MAX | |
147 | #define UINT32_MAX (4294967295U) | |
148 | #endif | |
149 | ||
150 | ||
151 | typedef void log_cb(const char *buffer); | |
152 | typedef void restart_cb(const char *buffer); | |
153 | typedef void die_cb(const char *buffer); | |
154 | ||
155 | char *rb_ctime(const time_t, char *, size_t); | |
156 | char *rb_date(const time_t, char *, size_t); | |
157 | void rb_lib_log(const char *, ...); | |
158 | void rb_lib_restart(const char *, ...) __attribute__((noreturn)); | |
159 | void rb_lib_die(const char *, ...); | |
160 | void rb_set_time(void); | |
161 | const char *rb_lib_version(void); | |
162 | ||
163 | void rb_lib_init(log_cb * xilog, restart_cb * irestart, die_cb * idie, int closeall, int maxfds, | |
164 | size_t dh_size, size_t fd_heap_size); | |
165 | void rb_lib_loop(long delay) __attribute__((noreturn)); | |
166 | ||
167 | time_t rb_current_time(void); | |
168 | const struct timeval *rb_current_time_tv(void); | |
169 | pid_t rb_spawn_process(const char *, const char **); | |
170 | ||
171 | char *rb_strtok_r(char *, const char *, char **); | |
172 | ||
173 | int rb_gettimeofday(struct timeval *, void *); | |
174 | ||
175 | void rb_sleep(unsigned int seconds, unsigned int useconds); | |
176 | char *rb_crypt(const char *, const char *); | |
177 | ||
178 | unsigned char *rb_base64_encode(const unsigned char *str, int length); | |
179 | unsigned char *rb_base64_decode(const unsigned char *str, int length, int *ret); | |
180 | int rb_kill(pid_t, int); | |
181 | char *rb_strerror(int); | |
182 | ||
183 | int rb_setenv(const char *, const char *, int); | |
184 | ||
185 | pid_t rb_waitpid(pid_t pid, int *status, int options); | |
186 | pid_t rb_getpid(void); | |
187 | //unsigned int rb_geteuid(void); | |
188 | ||
189 | ||
190 | #include <rb_tools.h> | |
191 | #include <rb_memory.h> | |
192 | #include <rb_commio.h> | |
193 | #include <rb_balloc.h> | |
194 | #include <rb_linebuf.h> | |
195 | #include <rb_event.h> | |
196 | #include <rb_helper.h> | |
197 | #include <rb_rawbuf.h> | |
198 | #include <rb_patricia.h> | |
199 | ||
200 | #endif |