Release pack windows before reporting out of memory.
[git/git.git] / git-compat-util.h
CommitLineData
4050c0df
JH
1#ifndef GIT_COMPAT_UTIL_H
2#define GIT_COMPAT_UTIL_H
3
8f1d2e6f
JH
4#ifndef FLEX_ARRAY
5#if defined(__GNUC__) && (__GNUC__ < 3)
6#define FLEX_ARRAY 0
7#else
8#define FLEX_ARRAY /* empty */
9#endif
10#endif
11
b4f2a6ac
JH
12#define ARRAY_SIZE(x) (sizeof(x)/sizeof(x[0]))
13
95ca1c6c 14#if !defined(__APPLE__) && !defined(__FreeBSD__)
85023577
JH
15#define _XOPEN_SOURCE 600 /* glibc2 and AIX 5.3L need 500, OpenBSD needs 600 for S_ISLNK() */
16#define _XOPEN_SOURCE_EXTENDED 1 /* AIX 5.3L needs this */
c902c9a6 17#endif
85023577
JH
18#define _GNU_SOURCE
19#define _BSD_SOURCE
20
4050c0df
JH
21#include <unistd.h>
22#include <stdio.h>
23#include <sys/stat.h>
24#include <fcntl.h>
25#include <stddef.h>
26#include <stdlib.h>
27#include <stdarg.h>
28#include <string.h>
29#include <errno.h>
30#include <limits.h>
31#include <sys/param.h>
4050c0df
JH
32#include <sys/types.h>
33#include <dirent.h>
85023577
JH
34#include <sys/time.h>
35#include <time.h>
36#include <signal.h>
37#include <sys/wait.h>
38#include <fnmatch.h>
39#include <sys/poll.h>
40#include <sys/socket.h>
41#include <assert.h>
42#include <regex.h>
43#include <netinet/in.h>
44#include <netinet/tcp.h>
45#include <arpa/inet.h>
46#include <netdb.h>
47#include <pwd.h>
48#include <grp.h>
49
50#ifndef NO_ICONV
51#include <iconv.h>
52#endif
4050c0df 53
d0c2449f
JH
54/* On most systems <limits.h> would have given us this, but
55 * not on some systems (e.g. GNU/Hurd).
56 */
57#ifndef PATH_MAX
58#define PATH_MAX 4096
59#endif
60
4050c0df
JH
61#ifdef __GNUC__
62#define NORETURN __attribute__((__noreturn__))
63#else
64#define NORETURN
65#ifndef __attribute__
66#define __attribute__(x)
67#endif
68#endif
69
70/* General helper functions */
71extern void usage(const char *err) NORETURN;
72extern void die(const char *err, ...) NORETURN __attribute__((format (printf, 1, 2)));
73extern int error(const char *err, ...) __attribute__((format (printf, 1, 2)));
fa39b6b5 74extern void warn(const char *err, ...) __attribute__((format (printf, 1, 2)));
4050c0df 75
39a3f5ea
PB
76extern void set_usage_routine(void (*routine)(const char *err) NORETURN);
77extern void set_die_routine(void (*routine)(const char *err, va_list params) NORETURN);
78extern void set_error_routine(void (*routine)(const char *err, va_list params));
fa39b6b5 79extern void set_warn_routine(void (*routine)(const char *warn, va_list params));
39a3f5ea 80
4050c0df
JH
81#ifdef NO_MMAP
82
83#ifndef PROT_READ
84#define PROT_READ 1
85#define PROT_WRITE 2
86#define MAP_PRIVATE 1
87#define MAP_FAILED ((void*)-1)
88#endif
89
d6779124
SP
90#define mmap git_mmap
91#define munmap git_munmap
92extern void *git_mmap(void *start, size_t length, int prot, int flags, int fd, off_t offset);
93extern int git_munmap(void *start, size_t length);
4050c0df 94
8c82534d
SP
95#define DEFAULT_PACKED_GIT_WINDOW_SIZE (1 * 1024 * 1024)
96
4050c0df
JH
97#else /* NO_MMAP */
98
99#include <sys/mman.h>
8c82534d 100#define DEFAULT_PACKED_GIT_WINDOW_SIZE (32 * 1024 * 1024)
4050c0df
JH
101
102#endif /* NO_MMAP */
103
8c82534d
SP
104#define DEFAULT_PACKED_GIT_LIMIT (256 * 1024 * 1024)
105
4050c0df
JH
106#ifdef NO_SETENV
107#define setenv gitsetenv
108extern int gitsetenv(const char *, const char *, int);
109#endif
110
731043fd
JR
111#ifdef NO_UNSETENV
112#define unsetenv gitunsetenv
113extern void gitunsetenv(const char *);
114#endif
115
4050c0df
JH
116#ifdef NO_STRCASESTR
117#define strcasestr gitstrcasestr
118extern char *gitstrcasestr(const char *haystack, const char *needle);
119#endif
120
817151e6
PE
121#ifdef NO_STRLCPY
122#define strlcpy gitstrlcpy
123extern size_t gitstrlcpy(char *, const char *, size_t);
124#endif
125
97bfeb34
SP
126extern void release_pack_memory(size_t);
127
9befac47
SP
128static inline char* xstrdup(const char *str)
129{
130 char *ret = strdup(str);
97bfeb34
SP
131 if (!ret) {
132 release_pack_memory(strlen(str) + 1);
133 ret = strdup(str);
134 if (!ret)
135 die("Out of memory, strdup failed");
136 }
9befac47
SP
137 return ret;
138}
139
4050c0df
JH
140static inline void *xmalloc(size_t size)
141{
142 void *ret = malloc(size);
4e7a2ecc
JH
143 if (!ret && !size)
144 ret = malloc(1);
97bfeb34
SP
145 if (!ret) {
146 release_pack_memory(size);
147 ret = malloc(size);
148 if (!ret && !size)
149 ret = malloc(1);
150 if (!ret)
151 die("Out of memory, malloc failed");
152 }
aa5481c1
JH
153#ifdef XMALLOC_POISON
154 memset(ret, 0xA5, size);
155#endif
4050c0df
JH
156 return ret;
157}
158
159static inline void *xrealloc(void *ptr, size_t size)
160{
161 void *ret = realloc(ptr, size);
4e7a2ecc
JH
162 if (!ret && !size)
163 ret = realloc(ptr, 1);
97bfeb34
SP
164 if (!ret) {
165 release_pack_memory(size);
166 ret = realloc(ptr, size);
167 if (!ret && !size)
168 ret = realloc(ptr, 1);
169 if (!ret)
170 die("Out of memory, realloc failed");
171 }
4050c0df
JH
172 return ret;
173}
174
175static inline void *xcalloc(size_t nmemb, size_t size)
176{
177 void *ret = calloc(nmemb, size);
4e7a2ecc
JH
178 if (!ret && (!nmemb || !size))
179 ret = calloc(1, 1);
97bfeb34
SP
180 if (!ret) {
181 release_pack_memory(nmemb * size);
182 ret = calloc(nmemb, size);
183 if (!ret && (!nmemb || !size))
184 ret = calloc(1, 1);
185 if (!ret)
186 die("Out of memory, calloc failed");
187 }
4050c0df
JH
188 return ret;
189}
190
1c15afb9
JH
191static inline ssize_t xread(int fd, void *buf, size_t len)
192{
193 ssize_t nr;
194 while (1) {
195 nr = read(fd, buf, len);
196 if ((nr < 0) && (errno == EAGAIN || errno == EINTR))
197 continue;
198 return nr;
199 }
200}
201
202static inline ssize_t xwrite(int fd, const void *buf, size_t len)
203{
204 ssize_t nr;
205 while (1) {
206 nr = write(fd, buf, len);
207 if ((nr < 0) && (errno == EAGAIN || errno == EINTR))
208 continue;
209 return nr;
210 }
211}
212
5bb1cda5 213static inline int has_extension(const char *filename, const char *ext)
83a2b841 214{
5bb1cda5
RS
215 size_t len = strlen(filename);
216 size_t extlen = strlen(ext);
83a2b841
RS
217 return len > extlen && !memcmp(filename + len - extlen, ext, extlen);
218}
219
4050c0df
JH
220/* Sane ctype - no locale, and works with signed chars */
221#undef isspace
222#undef isdigit
223#undef isalpha
224#undef isalnum
225#undef tolower
226#undef toupper
227extern unsigned char sane_ctype[256];
228#define GIT_SPACE 0x01
229#define GIT_DIGIT 0x02
230#define GIT_ALPHA 0x04
231#define sane_istest(x,mask) ((sane_ctype[(unsigned char)(x)] & (mask)) != 0)
232#define isspace(x) sane_istest(x,GIT_SPACE)
233#define isdigit(x) sane_istest(x,GIT_DIGIT)
234#define isalpha(x) sane_istest(x,GIT_ALPHA)
235#define isalnum(x) sane_istest(x,GIT_ALPHA | GIT_DIGIT)
236#define tolower(x) sane_case((unsigned char)(x), 0x20)
237#define toupper(x) sane_case((unsigned char)(x), 0)
238
239static inline int sane_case(int x, int high)
240{
241 if (sane_istest(x, GIT_ALPHA))
242 x = (x & ~0x20) | high;
243 return x;
244}
245
246#endif