Merge commit 'refs/top-bases/girocco' into girocco
[git/repo.git] / wrapper.c
blob93562f03eef21b26945d2d9bbdc96818f4de6567
1 /*
2 * Various trivial helper wrappers around standard functions
3 */
4 #include "cache.h"
6 char *xstrdup(const char *str)
8 char *ret = strdup(str);
9 if (!ret) {
10 release_pack_memory(strlen(str) + 1, -1);
11 ret = strdup(str);
12 if (!ret)
13 die("Out of memory, strdup failed");
15 return ret;
18 void *xmalloc(size_t size)
20 void *ret = malloc(size);
21 if (!ret && !size)
22 ret = malloc(1);
23 if (!ret) {
24 release_pack_memory(size, -1);
25 ret = malloc(size);
26 if (!ret && !size)
27 ret = malloc(1);
28 if (!ret)
29 die("Out of memory, malloc failed");
31 #ifdef XMALLOC_POISON
32 memset(ret, 0xA5, size);
33 #endif
34 return ret;
38 * xmemdupz() allocates (len + 1) bytes of memory, duplicates "len" bytes of
39 * "data" to the allocated memory, zero terminates the allocated memory,
40 * and returns a pointer to the allocated memory. If the allocation fails,
41 * the program dies.
43 void *xmemdupz(const void *data, size_t len)
45 char *p = xmalloc(len + 1);
46 memcpy(p, data, len);
47 p[len] = '\0';
48 return p;
51 char *xstrndup(const char *str, size_t len)
53 char *p = memchr(str, '\0', len);
54 return xmemdupz(str, p ? p - str : len);
57 void *xrealloc(void *ptr, size_t size)
59 void *ret = realloc(ptr, size);
60 if (!ret && !size)
61 ret = realloc(ptr, 1);
62 if (!ret) {
63 release_pack_memory(size, -1);
64 ret = realloc(ptr, size);
65 if (!ret && !size)
66 ret = realloc(ptr, 1);
67 if (!ret)
68 die("Out of memory, realloc failed");
70 return ret;
73 void *xcalloc(size_t nmemb, size_t size)
75 void *ret = calloc(nmemb, size);
76 if (!ret && (!nmemb || !size))
77 ret = calloc(1, 1);
78 if (!ret) {
79 release_pack_memory(nmemb * size, -1);
80 ret = calloc(nmemb, size);
81 if (!ret && (!nmemb || !size))
82 ret = calloc(1, 1);
83 if (!ret)
84 die("Out of memory, calloc failed");
86 return ret;
89 void *xmmap(void *start, size_t length,
90 int prot, int flags, int fd, off_t offset)
92 void *ret = mmap(start, length, prot, flags, fd, offset);
93 if (ret == MAP_FAILED) {
94 if (!length)
95 return NULL;
96 release_pack_memory(length, fd);
97 ret = mmap(start, length, prot, flags, fd, offset);
98 if (ret == MAP_FAILED)
99 die("Out of memory? mmap failed: %s", strerror(errno));
101 return ret;
105 * xread() is the same a read(), but it automatically restarts read()
106 * operations with a recoverable error (EAGAIN and EINTR). xread()
107 * DOES NOT GUARANTEE that "len" bytes is read even if the data is available.
109 ssize_t xread(int fd, void *buf, size_t len)
111 ssize_t nr;
112 while (1) {
113 nr = read(fd, buf, len);
114 if ((nr < 0) && (errno == EAGAIN || errno == EINTR))
115 continue;
116 return nr;
121 * xwrite() is the same a write(), but it automatically restarts write()
122 * operations with a recoverable error (EAGAIN and EINTR). xwrite() DOES NOT
123 * GUARANTEE that "len" bytes is written even if the operation is successful.
125 ssize_t xwrite(int fd, const void *buf, size_t len)
127 ssize_t nr;
128 while (1) {
129 nr = write(fd, buf, len);
130 if ((nr < 0) && (errno == EAGAIN || errno == EINTR))
131 continue;
132 return nr;
136 ssize_t read_in_full(int fd, void *buf, size_t count)
138 char *p = buf;
139 ssize_t total = 0;
141 while (count > 0) {
142 ssize_t loaded = xread(fd, p, count);
143 if (loaded <= 0)
144 return total ? total : loaded;
145 count -= loaded;
146 p += loaded;
147 total += loaded;
150 return total;
153 ssize_t write_in_full(int fd, const void *buf, size_t count)
155 const char *p = buf;
156 ssize_t total = 0;
158 while (count > 0) {
159 ssize_t written = xwrite(fd, p, count);
160 if (written < 0)
161 return -1;
162 if (!written) {
163 errno = ENOSPC;
164 return -1;
166 count -= written;
167 p += written;
168 total += written;
171 return total;
174 int xdup(int fd)
176 int ret = dup(fd);
177 if (ret < 0)
178 die("dup failed: %s", strerror(errno));
179 return ret;
182 FILE *xfdopen(int fd, const char *mode)
184 FILE *stream = fdopen(fd, mode);
185 if (stream == NULL)
186 die("Out of memory? fdopen failed: %s", strerror(errno));
187 return stream;
190 int xmkstemp(char *template)
192 int fd;
194 fd = mkstemp(template);
195 if (fd < 0)
196 die("Unable to create temporary file: %s", strerror(errno));
197 return fd;