]>
Commit | Line | Data |
---|---|---|
1 | #ifndef GIT_COMPAT_UTIL_H | |
2 | #define GIT_COMPAT_UTIL_H | |
3 | ||
4 | #define _FILE_OFFSET_BITS 64 | |
5 | ||
6 | #ifndef FLEX_ARRAY | |
7 | #if defined(__GNUC__) && (__GNUC__ < 3) | |
8 | #define FLEX_ARRAY 0 | |
9 | #else | |
10 | #define FLEX_ARRAY /* empty */ | |
11 | #endif | |
12 | #endif | |
13 | ||
14 | #define ARRAY_SIZE(x) (sizeof(x)/sizeof(x[0])) | |
15 | ||
16 | #ifdef __GNUC__ | |
17 | #define TYPEOF(x) (__typeof__(x)) | |
18 | #else | |
19 | #define TYPEOF(x) | |
20 | #endif | |
21 | ||
22 | #define MSB(x, bits) ((x) & TYPEOF(x)(~0ULL << (sizeof(x) * 8 - (bits)))) | |
23 | ||
24 | /* Approximation of the length of the decimal representation of this type. */ | |
25 | #define decimal_length(x) ((int)(sizeof(x) * 2.56 + 0.5) + 1) | |
26 | ||
27 | #if !defined(__APPLE__) && !defined(__FreeBSD__) | |
28 | #define _XOPEN_SOURCE 600 /* glibc2 and AIX 5.3L need 500, OpenBSD needs 600 for S_ISLNK() */ | |
29 | #define _XOPEN_SOURCE_EXTENDED 1 /* AIX 5.3L needs this */ | |
30 | #endif | |
31 | #define _ALL_SOURCE 1 | |
32 | #define _GNU_SOURCE 1 | |
33 | #define _BSD_SOURCE 1 | |
34 | ||
35 | #include <unistd.h> | |
36 | #include <stdio.h> | |
37 | #include <sys/stat.h> | |
38 | #include <fcntl.h> | |
39 | #include <stddef.h> | |
40 | #include <stdlib.h> | |
41 | #include <stdarg.h> | |
42 | #include <string.h> | |
43 | #include <errno.h> | |
44 | #include <limits.h> | |
45 | #include <sys/param.h> | |
46 | #include <sys/types.h> | |
47 | #include <dirent.h> | |
48 | #include <sys/time.h> | |
49 | #include <time.h> | |
50 | #include <signal.h> | |
51 | #include <sys/wait.h> | |
52 | #include <fnmatch.h> | |
53 | #include <sys/poll.h> | |
54 | #include <sys/socket.h> | |
55 | #include <assert.h> | |
56 | #include <regex.h> | |
57 | #include <netinet/in.h> | |
58 | #include <netinet/tcp.h> | |
59 | #include <arpa/inet.h> | |
60 | #include <netdb.h> | |
61 | #include <pwd.h> | |
62 | #include <inttypes.h> | |
63 | #if defined(__CYGWIN__) | |
64 | #undef _XOPEN_SOURCE | |
65 | #include <grp.h> | |
66 | #define _XOPEN_SOURCE 600 | |
67 | #else | |
68 | #undef _ALL_SOURCE /* AIX 5.3L defines a struct list with _ALL_SOURCE. */ | |
69 | #include <grp.h> | |
70 | #define _ALL_SOURCE 1 | |
71 | #endif | |
72 | ||
73 | #ifndef NO_ICONV | |
74 | #include <iconv.h> | |
75 | #endif | |
76 | ||
77 | /* On most systems <limits.h> would have given us this, but | |
78 | * not on some systems (e.g. GNU/Hurd). | |
79 | */ | |
80 | #ifndef PATH_MAX | |
81 | #define PATH_MAX 4096 | |
82 | #endif | |
83 | ||
84 | #ifndef PRIuMAX | |
85 | #define PRIuMAX "llu" | |
86 | #endif | |
87 | ||
88 | #ifdef __GNUC__ | |
89 | #define NORETURN __attribute__((__noreturn__)) | |
90 | #else | |
91 | #define NORETURN | |
92 | #ifndef __attribute__ | |
93 | #define __attribute__(x) | |
94 | #endif | |
95 | #endif | |
96 | ||
97 | /* General helper functions */ | |
98 | extern void usage(const char *err) NORETURN; | |
99 | extern void die(const char *err, ...) NORETURN __attribute__((format (printf, 1, 2))); | |
100 | extern int error(const char *err, ...) __attribute__((format (printf, 1, 2))); | |
101 | extern void warning(const char *err, ...) __attribute__((format (printf, 1, 2))); | |
102 | ||
103 | extern void set_usage_routine(void (*routine)(const char *err) NORETURN); | |
104 | extern void set_die_routine(void (*routine)(const char *err, va_list params) NORETURN); | |
105 | extern void set_error_routine(void (*routine)(const char *err, va_list params)); | |
106 | extern void set_warn_routine(void (*routine)(const char *warn, va_list params)); | |
107 | ||
108 | #ifdef NO_MMAP | |
109 | ||
110 | #ifndef PROT_READ | |
111 | #define PROT_READ 1 | |
112 | #define PROT_WRITE 2 | |
113 | #define MAP_PRIVATE 1 | |
114 | #define MAP_FAILED ((void*)-1) | |
115 | #endif | |
116 | ||
117 | #define mmap git_mmap | |
118 | #define munmap git_munmap | |
119 | extern void *git_mmap(void *start, size_t length, int prot, int flags, int fd, off_t offset); | |
120 | extern int git_munmap(void *start, size_t length); | |
121 | ||
122 | /* This value must be multiple of (pagesize * 2) */ | |
123 | #define DEFAULT_PACKED_GIT_WINDOW_SIZE (1 * 1024 * 1024) | |
124 | ||
125 | #else /* NO_MMAP */ | |
126 | ||
127 | #include <sys/mman.h> | |
128 | ||
129 | /* This value must be multiple of (pagesize * 2) */ | |
130 | #define DEFAULT_PACKED_GIT_WINDOW_SIZE \ | |
131 | (sizeof(void*) >= 8 \ | |
132 | ? 1 * 1024 * 1024 * 1024 \ | |
133 | : 32 * 1024 * 1024) | |
134 | ||
135 | #endif /* NO_MMAP */ | |
136 | ||
137 | #define DEFAULT_PACKED_GIT_LIMIT \ | |
138 | ((1024L * 1024L) * (sizeof(void*) >= 8 ? 8192 : 256)) | |
139 | ||
140 | #ifdef NO_PREAD | |
141 | #define pread git_pread | |
142 | extern ssize_t git_pread(int fd, void *buf, size_t count, off_t offset); | |
143 | #endif | |
144 | ||
145 | #ifdef NO_SETENV | |
146 | #define setenv gitsetenv | |
147 | extern int gitsetenv(const char *, const char *, int); | |
148 | #endif | |
149 | ||
150 | #ifdef NO_UNSETENV | |
151 | #define unsetenv gitunsetenv | |
152 | extern void gitunsetenv(const char *); | |
153 | #endif | |
154 | ||
155 | #ifdef NO_STRCASESTR | |
156 | #define strcasestr gitstrcasestr | |
157 | extern char *gitstrcasestr(const char *haystack, const char *needle); | |
158 | #endif | |
159 | ||
160 | #ifdef NO_STRLCPY | |
161 | #define strlcpy gitstrlcpy | |
162 | extern size_t gitstrlcpy(char *, const char *, size_t); | |
163 | #endif | |
164 | ||
165 | #ifdef NO_STRTOUMAX | |
166 | #define strtoumax gitstrtoumax | |
167 | extern uintmax_t gitstrtoumax(const char *, char **, int); | |
168 | #endif | |
169 | ||
170 | #ifdef NO_HSTRERROR | |
171 | #define hstrerror githstrerror | |
172 | extern const char *githstrerror(int herror); | |
173 | #endif | |
174 | ||
175 | #ifdef NO_MEMMEM | |
176 | #define memmem gitmemmem | |
177 | void *gitmemmem(const void *haystack, size_t haystacklen, | |
178 | const void *needle, size_t needlelen); | |
179 | #endif | |
180 | ||
181 | extern void release_pack_memory(size_t, int); | |
182 | ||
183 | static inline char* xstrdup(const char *str) | |
184 | { | |
185 | char *ret = strdup(str); | |
186 | if (!ret) { | |
187 | release_pack_memory(strlen(str) + 1, -1); | |
188 | ret = strdup(str); | |
189 | if (!ret) | |
190 | die("Out of memory, strdup failed"); | |
191 | } | |
192 | return ret; | |
193 | } | |
194 | ||
195 | static inline void *xmalloc(size_t size) | |
196 | { | |
197 | void *ret = malloc(size); | |
198 | if (!ret && !size) | |
199 | ret = malloc(1); | |
200 | if (!ret) { | |
201 | release_pack_memory(size, -1); | |
202 | ret = malloc(size); | |
203 | if (!ret && !size) | |
204 | ret = malloc(1); | |
205 | if (!ret) | |
206 | die("Out of memory, malloc failed"); | |
207 | } | |
208 | #ifdef XMALLOC_POISON | |
209 | memset(ret, 0xA5, size); | |
210 | #endif | |
211 | return ret; | |
212 | } | |
213 | ||
214 | static inline void *xmemdupz(const void *data, size_t len) | |
215 | { | |
216 | char *p = xmalloc(len + 1); | |
217 | memcpy(p, data, len); | |
218 | p[len] = '\0'; | |
219 | return p; | |
220 | } | |
221 | ||
222 | static inline char *xstrndup(const char *str, size_t len) | |
223 | { | |
224 | char *p = memchr(str, '\0', len); | |
225 | return xmemdupz(str, p ? p - str : len); | |
226 | } | |
227 | ||
228 | static inline void *xrealloc(void *ptr, size_t size) | |
229 | { | |
230 | void *ret = realloc(ptr, size); | |
231 | if (!ret && !size) | |
232 | ret = realloc(ptr, 1); | |
233 | if (!ret) { | |
234 | release_pack_memory(size, -1); | |
235 | ret = realloc(ptr, size); | |
236 | if (!ret && !size) | |
237 | ret = realloc(ptr, 1); | |
238 | if (!ret) | |
239 | die("Out of memory, realloc failed"); | |
240 | } | |
241 | return ret; | |
242 | } | |
243 | ||
244 | static inline void *xcalloc(size_t nmemb, size_t size) | |
245 | { | |
246 | void *ret = calloc(nmemb, size); | |
247 | if (!ret && (!nmemb || !size)) | |
248 | ret = calloc(1, 1); | |
249 | if (!ret) { | |
250 | release_pack_memory(nmemb * size, -1); | |
251 | ret = calloc(nmemb, size); | |
252 | if (!ret && (!nmemb || !size)) | |
253 | ret = calloc(1, 1); | |
254 | if (!ret) | |
255 | die("Out of memory, calloc failed"); | |
256 | } | |
257 | return ret; | |
258 | } | |
259 | ||
260 | static inline void *xmmap(void *start, size_t length, | |
261 | int prot, int flags, int fd, off_t offset) | |
262 | { | |
263 | void *ret = mmap(start, length, prot, flags, fd, offset); | |
264 | if (ret == MAP_FAILED) { | |
265 | if (!length) | |
266 | return NULL; | |
267 | release_pack_memory(length, fd); | |
268 | ret = mmap(start, length, prot, flags, fd, offset); | |
269 | if (ret == MAP_FAILED) | |
270 | die("Out of memory? mmap failed: %s", strerror(errno)); | |
271 | } | |
272 | return ret; | |
273 | } | |
274 | ||
275 | static inline ssize_t xread(int fd, void *buf, size_t len) | |
276 | { | |
277 | ssize_t nr; | |
278 | while (1) { | |
279 | nr = read(fd, buf, len); | |
280 | if ((nr < 0) && (errno == EAGAIN || errno == EINTR)) | |
281 | continue; | |
282 | return nr; | |
283 | } | |
284 | } | |
285 | ||
286 | static inline ssize_t xwrite(int fd, const void *buf, size_t len) | |
287 | { | |
288 | ssize_t nr; | |
289 | while (1) { | |
290 | nr = write(fd, buf, len); | |
291 | if ((nr < 0) && (errno == EAGAIN || errno == EINTR)) | |
292 | continue; | |
293 | return nr; | |
294 | } | |
295 | } | |
296 | ||
297 | static inline int xdup(int fd) | |
298 | { | |
299 | int ret = dup(fd); | |
300 | if (ret < 0) | |
301 | die("dup failed: %s", strerror(errno)); | |
302 | return ret; | |
303 | } | |
304 | ||
305 | static inline FILE *xfdopen(int fd, const char *mode) | |
306 | { | |
307 | FILE *stream = fdopen(fd, mode); | |
308 | if (stream == NULL) | |
309 | die("Out of memory? fdopen failed: %s", strerror(errno)); | |
310 | return stream; | |
311 | } | |
312 | ||
313 | static inline int xmkstemp(char *template) | |
314 | { | |
315 | int fd; | |
316 | ||
317 | fd = mkstemp(template); | |
318 | if (fd < 0) | |
319 | die("Unable to create temporary file: %s", strerror(errno)); | |
320 | return fd; | |
321 | } | |
322 | ||
323 | static inline size_t xsize_t(off_t len) | |
324 | { | |
325 | return (size_t)len; | |
326 | } | |
327 | ||
328 | static inline int has_extension(const char *filename, const char *ext) | |
329 | { | |
330 | size_t len = strlen(filename); | |
331 | size_t extlen = strlen(ext); | |
332 | return len > extlen && !memcmp(filename + len - extlen, ext, extlen); | |
333 | } | |
334 | ||
335 | /* Sane ctype - no locale, and works with signed chars */ | |
336 | #undef isspace | |
337 | #undef isdigit | |
338 | #undef isalpha | |
339 | #undef isalnum | |
340 | #undef tolower | |
341 | #undef toupper | |
342 | extern unsigned char sane_ctype[256]; | |
343 | #define GIT_SPACE 0x01 | |
344 | #define GIT_DIGIT 0x02 | |
345 | #define GIT_ALPHA 0x04 | |
346 | #define sane_istest(x,mask) ((sane_ctype[(unsigned char)(x)] & (mask)) != 0) | |
347 | #define isspace(x) sane_istest(x,GIT_SPACE) | |
348 | #define isdigit(x) sane_istest(x,GIT_DIGIT) | |
349 | #define isalpha(x) sane_istest(x,GIT_ALPHA) | |
350 | #define isalnum(x) sane_istest(x,GIT_ALPHA | GIT_DIGIT) | |
351 | #define tolower(x) sane_case((unsigned char)(x), 0x20) | |
352 | #define toupper(x) sane_case((unsigned char)(x), 0) | |
353 | ||
354 | static inline int sane_case(int x, int high) | |
355 | { | |
356 | if (sane_istest(x, GIT_ALPHA)) | |
357 | x = (x & ~0x20) | high; | |
358 | return x; | |
359 | } | |
360 | ||
361 | static inline int prefixcmp(const char *str, const char *prefix) | |
362 | { | |
363 | return strncmp(str, prefix, strlen(prefix)); | |
364 | } | |
365 | ||
366 | static inline int strtoul_ui(char const *s, int base, unsigned int *result) | |
367 | { | |
368 | unsigned long ul; | |
369 | char *p; | |
370 | ||
371 | errno = 0; | |
372 | ul = strtoul(s, &p, base); | |
373 | if (errno || *p || p == s || (unsigned int) ul != ul) | |
374 | return -1; | |
375 | *result = ul; | |
376 | return 0; | |
377 | } | |
378 | ||
379 | #endif |