]>
Commit | Line | Data |
---|---|---|
812666c8 | 1 | #include "cache.h" |
a2fab531 | 2 | #include "refs.h" |
d1df5743 | 3 | |
698a68be JH |
4 | int prefixcmp(const char *str, const char *prefix) |
5 | { | |
6 | for (; ; str++, prefix++) | |
7 | if (!*prefix) | |
8 | return 0; | |
9 | else if (*str != *prefix) | |
10 | return (unsigned char)*prefix - (unsigned char)*str; | |
11 | } | |
12 | ||
8cc5b290 AP |
13 | int suffixcmp(const char *str, const char *suffix) |
14 | { | |
15 | int len = strlen(str), suflen = strlen(suffix); | |
16 | if (len < suflen) | |
17 | return -1; | |
18 | else | |
19 | return strcmp(str + len - suflen, suffix); | |
20 | } | |
21 | ||
b315c5c0 PH |
22 | /* |
23 | * Used as the default ->buf value, so that people can always assume | |
24 | * buf is non NULL and ->buf is NUL terminated even for a freshly | |
25 | * initialized strbuf. | |
26 | */ | |
27 | char strbuf_slopbuf[1]; | |
28 | ||
917c9a71 PH |
29 | void strbuf_init(struct strbuf *sb, size_t hint) |
30 | { | |
b315c5c0 PH |
31 | sb->alloc = sb->len = 0; |
32 | sb->buf = strbuf_slopbuf; | |
5e7a5d97 | 33 | if (hint) { |
f1696ee3 | 34 | strbuf_grow(sb, hint); |
5e7a5d97 EFL |
35 | sb->buf[0] = '\0'; |
36 | } | |
d1df5743 JH |
37 | } |
38 | ||
917c9a71 PH |
39 | void strbuf_release(struct strbuf *sb) |
40 | { | |
b315c5c0 PH |
41 | if (sb->alloc) { |
42 | free(sb->buf); | |
43 | strbuf_init(sb, 0); | |
44 | } | |
b449f4cf PH |
45 | } |
46 | ||
b315c5c0 | 47 | char *strbuf_detach(struct strbuf *sb, size_t *sz) |
917c9a71 | 48 | { |
b315c5c0 PH |
49 | char *res = sb->alloc ? sb->buf : NULL; |
50 | if (sz) | |
51 | *sz = sb->len; | |
f1696ee3 | 52 | strbuf_init(sb, 0); |
b449f4cf PH |
53 | return res; |
54 | } | |
55 | ||
917c9a71 PH |
56 | void strbuf_attach(struct strbuf *sb, void *buf, size_t len, size_t alloc) |
57 | { | |
58 | strbuf_release(sb); | |
59 | sb->buf = buf; | |
60 | sb->len = len; | |
61 | sb->alloc = alloc; | |
62 | strbuf_grow(sb, 0); | |
63 | sb->buf[sb->len] = '\0'; | |
64 | } | |
65 | ||
66 | void strbuf_grow(struct strbuf *sb, size_t extra) | |
67 | { | |
1368f650 JN |
68 | if (unsigned_add_overflows(extra, 1) || |
69 | unsigned_add_overflows(sb->len, extra + 1)) | |
b449f4cf | 70 | die("you want to use way too much memory"); |
b315c5c0 PH |
71 | if (!sb->alloc) |
72 | sb->buf = NULL; | |
b449f4cf PH |
73 | ALLOC_GROW(sb->buf, sb->len + extra + 1, sb->alloc); |
74 | } | |
75 | ||
eacd6dc5 LS |
76 | void strbuf_trim(struct strbuf *sb) |
77 | { | |
78 | char *b = sb->buf; | |
79 | while (sb->len > 0 && isspace((unsigned char)sb->buf[sb->len - 1])) | |
80 | sb->len--; | |
81 | while (sb->len > 0 && isspace(*b)) { | |
82 | b++; | |
83 | sb->len--; | |
84 | } | |
85 | memmove(sb->buf, b, sb->len); | |
86 | sb->buf[sb->len] = '\0'; | |
87 | } | |
f1696ee3 PH |
88 | void strbuf_rtrim(struct strbuf *sb) |
89 | { | |
90 | while (sb->len > 0 && isspace((unsigned char)sb->buf[sb->len - 1])) | |
91 | sb->len--; | |
92 | sb->buf[sb->len] = '\0'; | |
93 | } | |
94 | ||
eacd6dc5 LS |
95 | void strbuf_ltrim(struct strbuf *sb) |
96 | { | |
97 | char *b = sb->buf; | |
98 | while (sb->len > 0 && isspace(*b)) { | |
99 | b++; | |
100 | sb->len--; | |
101 | } | |
102 | memmove(sb->buf, b, sb->len); | |
103 | sb->buf[sb->len] = '\0'; | |
104 | } | |
105 | ||
2f1d9e2b | 106 | struct strbuf **strbuf_split_buf(const char *str, size_t slen, int delim, int max) |
eacd6dc5 LS |
107 | { |
108 | int alloc = 2, pos = 0; | |
2f1d9e2b | 109 | const char *n, *p; |
eacd6dc5 LS |
110 | struct strbuf **ret; |
111 | struct strbuf *t; | |
112 | ||
113 | ret = xcalloc(alloc, sizeof(struct strbuf *)); | |
2f1d9e2b JK |
114 | p = n = str; |
115 | while (n < str + slen) { | |
eacd6dc5 | 116 | int len; |
28fc3a68 | 117 | if (max <= 0 || pos + 1 < max) |
2f1d9e2b | 118 | n = memchr(n, delim, slen - (n - str)); |
28fc3a68 JK |
119 | else |
120 | n = NULL; | |
eacd6dc5 LS |
121 | if (pos + 1 >= alloc) { |
122 | alloc = alloc * 2; | |
123 | ret = xrealloc(ret, sizeof(struct strbuf *) * alloc); | |
124 | } | |
125 | if (!n) | |
2f1d9e2b | 126 | n = str + slen - 1; |
eacd6dc5 LS |
127 | len = n - p + 1; |
128 | t = xmalloc(sizeof(struct strbuf)); | |
129 | strbuf_init(t, len); | |
130 | strbuf_add(t, p, len); | |
131 | ret[pos] = t; | |
132 | ret[++pos] = NULL; | |
133 | p = ++n; | |
134 | } | |
135 | return ret; | |
136 | } | |
137 | ||
138 | void strbuf_list_free(struct strbuf **sbs) | |
139 | { | |
140 | struct strbuf **s = sbs; | |
141 | ||
142 | while (*s) { | |
143 | strbuf_release(*s); | |
144 | free(*s++); | |
145 | } | |
146 | free(sbs); | |
147 | } | |
148 | ||
9b200fd6 | 149 | int strbuf_cmp(const struct strbuf *a, const struct strbuf *b) |
45f66f64 | 150 | { |
8f024655 AR |
151 | int len = a->len < b->len ? a->len: b->len; |
152 | int cmp = memcmp(a->buf, b->buf, len); | |
153 | if (cmp) | |
154 | return cmp; | |
155 | return a->len < b->len ? -1: a->len != b->len; | |
45f66f64 PH |
156 | } |
157 | ||
917c9a71 PH |
158 | void strbuf_splice(struct strbuf *sb, size_t pos, size_t len, |
159 | const void *data, size_t dlen) | |
160 | { | |
1368f650 | 161 | if (unsigned_add_overflows(pos, len)) |
917c9a71 PH |
162 | die("you want to use way too much memory"); |
163 | if (pos > sb->len) | |
164 | die("`pos' is too far after the end of the buffer"); | |
165 | if (pos + len > sb->len) | |
166 | die("`pos + len' is too far after the end of the buffer"); | |
167 | ||
168 | if (dlen >= len) | |
169 | strbuf_grow(sb, dlen - len); | |
170 | memmove(sb->buf + pos + dlen, | |
171 | sb->buf + pos + len, | |
172 | sb->len - pos - len); | |
173 | memcpy(sb->buf + pos, data, dlen); | |
174 | strbuf_setlen(sb, sb->len + dlen - len); | |
175 | } | |
176 | ||
c76689df PH |
177 | void strbuf_insert(struct strbuf *sb, size_t pos, const void *data, size_t len) |
178 | { | |
179 | strbuf_splice(sb, pos, 0, data, len); | |
180 | } | |
181 | ||
182 | void strbuf_remove(struct strbuf *sb, size_t pos, size_t len) | |
183 | { | |
184 | strbuf_splice(sb, pos, len, NULL, 0); | |
185 | } | |
186 | ||
917c9a71 PH |
187 | void strbuf_add(struct strbuf *sb, const void *data, size_t len) |
188 | { | |
b449f4cf PH |
189 | strbuf_grow(sb, len); |
190 | memcpy(sb->buf + sb->len, data, len); | |
191 | strbuf_setlen(sb, sb->len + len); | |
192 | } | |
193 | ||
91db267e RS |
194 | void strbuf_adddup(struct strbuf *sb, size_t pos, size_t len) |
195 | { | |
196 | strbuf_grow(sb, len); | |
197 | memcpy(sb->buf + sb->len, sb->buf + pos, len); | |
198 | strbuf_setlen(sb, sb->len + len); | |
199 | } | |
200 | ||
917c9a71 PH |
201 | void strbuf_addf(struct strbuf *sb, const char *fmt, ...) |
202 | { | |
b449f4cf | 203 | va_list ap; |
ebeb6090 JK |
204 | va_start(ap, fmt); |
205 | strbuf_vaddf(sb, fmt, ap); | |
206 | va_end(ap); | |
207 | } | |
208 | ||
209 | void strbuf_vaddf(struct strbuf *sb, const char *fmt, va_list ap) | |
210 | { | |
211 | int len; | |
212 | va_list cp; | |
b449f4cf | 213 | |
f141bd80 SP |
214 | if (!strbuf_avail(sb)) |
215 | strbuf_grow(sb, 64); | |
ebeb6090 JK |
216 | va_copy(cp, ap); |
217 | len = vsnprintf(sb->buf + sb->len, sb->alloc - sb->len, fmt, cp); | |
218 | va_end(cp); | |
f141bd80 | 219 | if (len < 0) |
ebeb6090 | 220 | die("BUG: your vsnprintf is broken (returned %d)", len); |
f1696ee3 | 221 | if (len > strbuf_avail(sb)) { |
b449f4cf | 222 | strbuf_grow(sb, len); |
b449f4cf | 223 | len = vsnprintf(sb->buf + sb->len, sb->alloc - sb->len, fmt, ap); |
ebeb6090 JK |
224 | if (len > strbuf_avail(sb)) |
225 | die("BUG: your vsnprintf is broken (insatiable)"); | |
b449f4cf PH |
226 | } |
227 | strbuf_setlen(sb, sb->len + len); | |
d1df5743 JH |
228 | } |
229 | ||
c3a670de MC |
230 | void strbuf_expand(struct strbuf *sb, const char *format, expand_fn_t fn, |
231 | void *context) | |
cde75e59 RS |
232 | { |
233 | for (;;) { | |
c3a670de MC |
234 | const char *percent; |
235 | size_t consumed; | |
cde75e59 RS |
236 | |
237 | percent = strchrnul(format, '%'); | |
238 | strbuf_add(sb, format, percent - format); | |
239 | if (!*percent) | |
240 | break; | |
241 | format = percent + 1; | |
242 | ||
0a0416a3 JK |
243 | if (*format == '%') { |
244 | strbuf_addch(sb, '%'); | |
245 | format++; | |
246 | continue; | |
247 | } | |
248 | ||
c3a670de MC |
249 | consumed = fn(sb, format, context); |
250 | if (consumed) | |
251 | format += consumed; | |
252 | else | |
cde75e59 RS |
253 | strbuf_addch(sb, '%'); |
254 | } | |
255 | } | |
256 | ||
9b864e73 RS |
257 | size_t strbuf_expand_dict_cb(struct strbuf *sb, const char *placeholder, |
258 | void *context) | |
259 | { | |
260 | struct strbuf_expand_dict_entry *e = context; | |
261 | size_t len; | |
262 | ||
263 | for (; e->placeholder && (len = strlen(e->placeholder)); e++) { | |
264 | if (!strncmp(placeholder, e->placeholder, len)) { | |
265 | if (e->value) | |
266 | strbuf_addstr(sb, e->value); | |
267 | return len; | |
268 | } | |
269 | } | |
270 | return 0; | |
271 | } | |
272 | ||
361df5df JK |
273 | void strbuf_addbuf_percentquote(struct strbuf *dst, const struct strbuf *src) |
274 | { | |
275 | int i, len = src->len; | |
276 | ||
277 | for (i = 0; i < len; i++) { | |
278 | if (src->buf[i] == '%') | |
279 | strbuf_addch(dst, '%'); | |
280 | strbuf_addch(dst, src->buf[i]); | |
281 | } | |
282 | } | |
283 | ||
917c9a71 PH |
284 | size_t strbuf_fread(struct strbuf *sb, size_t size, FILE *f) |
285 | { | |
b449f4cf | 286 | size_t res; |
2fc64700 | 287 | size_t oldalloc = sb->alloc; |
b449f4cf PH |
288 | |
289 | strbuf_grow(sb, size); | |
290 | res = fread(sb->buf + sb->len, 1, size, f); | |
2fc64700 | 291 | if (res > 0) |
b449f4cf | 292 | strbuf_setlen(sb, sb->len + res); |
6651c3f7 | 293 | else if (oldalloc == 0) |
2fc64700 | 294 | strbuf_release(sb); |
b449f4cf | 295 | return res; |
d1df5743 JH |
296 | } |
297 | ||
f1696ee3 | 298 | ssize_t strbuf_read(struct strbuf *sb, int fd, size_t hint) |
b449f4cf PH |
299 | { |
300 | size_t oldlen = sb->len; | |
2fc64700 | 301 | size_t oldalloc = sb->alloc; |
b449f4cf | 302 | |
f1696ee3 | 303 | strbuf_grow(sb, hint ? hint : 8192); |
b449f4cf PH |
304 | for (;;) { |
305 | ssize_t cnt; | |
306 | ||
b449f4cf PH |
307 | cnt = xread(fd, sb->buf + sb->len, sb->alloc - sb->len - 1); |
308 | if (cnt < 0) { | |
2fc64700 RS |
309 | if (oldalloc == 0) |
310 | strbuf_release(sb); | |
311 | else | |
312 | strbuf_setlen(sb, oldlen); | |
b449f4cf PH |
313 | return -1; |
314 | } | |
315 | if (!cnt) | |
316 | break; | |
317 | sb->len += cnt; | |
f1696ee3 | 318 | strbuf_grow(sb, 8192); |
b449f4cf PH |
319 | } |
320 | ||
321 | sb->buf[sb->len] = '\0'; | |
322 | return sb->len - oldlen; | |
d1df5743 JH |
323 | } |
324 | ||
b11b7e13 LT |
325 | #define STRBUF_MAXLINK (2*PATH_MAX) |
326 | ||
327 | int strbuf_readlink(struct strbuf *sb, const char *path, size_t hint) | |
328 | { | |
2fc64700 RS |
329 | size_t oldalloc = sb->alloc; |
330 | ||
b11b7e13 LT |
331 | if (hint < 32) |
332 | hint = 32; | |
333 | ||
334 | while (hint < STRBUF_MAXLINK) { | |
335 | int len; | |
336 | ||
337 | strbuf_grow(sb, hint); | |
338 | len = readlink(path, sb->buf, hint); | |
339 | if (len < 0) { | |
340 | if (errno != ERANGE) | |
341 | break; | |
342 | } else if (len < hint) { | |
343 | strbuf_setlen(sb, len); | |
344 | return 0; | |
345 | } | |
346 | ||
347 | /* .. the buffer was too small - try again */ | |
348 | hint *= 2; | |
349 | } | |
2fc64700 RS |
350 | if (oldalloc == 0) |
351 | strbuf_release(sb); | |
b11b7e13 LT |
352 | return -1; |
353 | } | |
354 | ||
c7e4f0d7 | 355 | int strbuf_getwholeline(struct strbuf *sb, FILE *fp, int term) |
917c9a71 | 356 | { |
d1df5743 | 357 | int ch; |
e6c019d0 PH |
358 | |
359 | strbuf_grow(sb, 0); | |
360 | if (feof(fp)) | |
361 | return EOF; | |
b449f4cf PH |
362 | |
363 | strbuf_reset(sb); | |
d1df5743 | 364 | while ((ch = fgetc(fp)) != EOF) { |
b449f4cf PH |
365 | strbuf_grow(sb, 1); |
366 | sb->buf[sb->len++] = ch; | |
c7e4f0d7 BC |
367 | if (ch == term) |
368 | break; | |
d1df5743 | 369 | } |
e6c019d0 PH |
370 | if (ch == EOF && sb->len == 0) |
371 | return EOF; | |
b449f4cf | 372 | |
b449f4cf | 373 | sb->buf[sb->len] = '\0'; |
e6c019d0 | 374 | return 0; |
d1df5743 | 375 | } |
a9390b9f | 376 | |
c7e4f0d7 BC |
377 | int strbuf_getline(struct strbuf *sb, FILE *fp, int term) |
378 | { | |
379 | if (strbuf_getwholeline(sb, fp, term)) | |
380 | return EOF; | |
381 | if (sb->buf[sb->len-1] == term) | |
382 | strbuf_setlen(sb, sb->len-1); | |
383 | return 0; | |
384 | } | |
385 | ||
387e7e19 | 386 | int strbuf_read_file(struct strbuf *sb, const char *path, size_t hint) |
a9390b9f KH |
387 | { |
388 | int fd, len; | |
389 | ||
390 | fd = open(path, O_RDONLY); | |
391 | if (fd < 0) | |
392 | return -1; | |
387e7e19 | 393 | len = strbuf_read(sb, fd, hint); |
a9390b9f KH |
394 | close(fd); |
395 | if (len < 0) | |
396 | return -1; | |
397 | ||
398 | return len; | |
399 | } |