4 int prefixcmp(const char *str
, const char *prefix
)
6 for (; ; str
++, prefix
++)
9 else if (*str
!= *prefix
)
10 return (unsigned char)*prefix
- (unsigned char)*str
;
13 int suffixcmp(const char *str
, const char *suffix
)
15 int len
= strlen(str
), suflen
= strlen(suffix
);
19 return strcmp(str
+ len
- suflen
, suffix
);
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
27 char strbuf_slopbuf
[1];
29 void strbuf_init(struct strbuf
*sb
, size_t hint
)
31 sb
->alloc
= sb
->len
= 0;
32 sb
->buf
= strbuf_slopbuf
;
34 strbuf_grow(sb
, hint
);
37 void strbuf_release(struct strbuf
*sb
)
45 char *strbuf_detach(struct strbuf
*sb
, size_t *sz
)
47 char *res
= sb
->alloc
? sb
->buf
: NULL
;
54 void strbuf_attach(struct strbuf
*sb
, void *buf
, size_t len
, size_t alloc
)
61 sb
->buf
[sb
->len
] = '\0';
64 void strbuf_grow(struct strbuf
*sb
, size_t extra
)
66 if (sb
->len
+ extra
+ 1 <= sb
->len
)
67 die("you want to use way too much memory");
70 ALLOC_GROW(sb
->buf
, sb
->len
+ extra
+ 1, sb
->alloc
);
73 void strbuf_trim(struct strbuf
*sb
)
76 while (sb
->len
> 0 && isspace((unsigned char)sb
->buf
[sb
->len
- 1]))
78 while (sb
->len
> 0 && isspace(*b
)) {
82 memmove(sb
->buf
, b
, sb
->len
);
83 sb
->buf
[sb
->len
] = '\0';
85 void strbuf_rtrim(struct strbuf
*sb
)
87 while (sb
->len
> 0 && isspace((unsigned char)sb
->buf
[sb
->len
- 1]))
89 sb
->buf
[sb
->len
] = '\0';
92 void strbuf_ltrim(struct strbuf
*sb
)
95 while (sb
->len
> 0 && isspace(*b
)) {
99 memmove(sb
->buf
, b
, sb
->len
);
100 sb
->buf
[sb
->len
] = '\0';
103 void strbuf_tolower(struct strbuf
*sb
)
106 for (i
= 0; i
< sb
->len
; i
++)
107 sb
->buf
[i
] = tolower(sb
->buf
[i
]);
110 struct strbuf
**strbuf_split(const struct strbuf
*sb
, int delim
)
112 int alloc
= 2, pos
= 0;
117 ret
= xcalloc(alloc
, sizeof(struct strbuf
*));
119 while (n
< sb
->buf
+ sb
->len
) {
121 n
= memchr(n
, delim
, sb
->len
- (n
- sb
->buf
));
122 if (pos
+ 1 >= alloc
) {
124 ret
= xrealloc(ret
, sizeof(struct strbuf
*) * alloc
);
127 n
= sb
->buf
+ sb
->len
- 1;
129 t
= xmalloc(sizeof(struct strbuf
));
131 strbuf_add(t
, p
, len
);
139 void strbuf_list_free(struct strbuf
**sbs
)
141 struct strbuf
**s
= sbs
;
150 int strbuf_cmp(const struct strbuf
*a
, const struct strbuf
*b
)
152 int len
= a
->len
< b
->len
? a
->len
: b
->len
;
153 int cmp
= memcmp(a
->buf
, b
->buf
, len
);
156 return a
->len
< b
->len
? -1: a
->len
!= b
->len
;
159 void strbuf_splice(struct strbuf
*sb
, size_t pos
, size_t len
,
160 const void *data
, size_t dlen
)
163 die("you want to use way too much memory");
165 die("`pos' is too far after the end of the buffer");
166 if (pos
+ len
> sb
->len
)
167 die("`pos + len' is too far after the end of the buffer");
170 strbuf_grow(sb
, dlen
- len
);
171 memmove(sb
->buf
+ pos
+ dlen
,
173 sb
->len
- pos
- len
);
174 memcpy(sb
->buf
+ pos
, data
, dlen
);
175 strbuf_setlen(sb
, sb
->len
+ dlen
- len
);
178 void strbuf_insert(struct strbuf
*sb
, size_t pos
, const void *data
, size_t len
)
180 strbuf_splice(sb
, pos
, 0, data
, len
);
183 void strbuf_remove(struct strbuf
*sb
, size_t pos
, size_t len
)
185 strbuf_splice(sb
, pos
, len
, NULL
, 0);
188 void strbuf_add(struct strbuf
*sb
, const void *data
, size_t len
)
190 strbuf_grow(sb
, len
);
191 memcpy(sb
->buf
+ sb
->len
, data
, len
);
192 strbuf_setlen(sb
, sb
->len
+ len
);
195 void strbuf_adddup(struct strbuf
*sb
, size_t pos
, size_t len
)
197 strbuf_grow(sb
, len
);
198 memcpy(sb
->buf
+ sb
->len
, sb
->buf
+ pos
, len
);
199 strbuf_setlen(sb
, sb
->len
+ len
);
202 void strbuf_addf(struct strbuf
*sb
, const char *fmt
, ...)
207 if (!strbuf_avail(sb
))
210 len
= vsnprintf(sb
->buf
+ sb
->len
, sb
->alloc
- sb
->len
, fmt
, ap
);
213 die("your vsnprintf is broken");
214 if (len
> strbuf_avail(sb
)) {
215 strbuf_grow(sb
, len
);
217 len
= vsnprintf(sb
->buf
+ sb
->len
, sb
->alloc
- sb
->len
, fmt
, ap
);
219 if (len
> strbuf_avail(sb
)) {
220 die("this should not happen, your snprintf is broken");
223 strbuf_setlen(sb
, sb
->len
+ len
);
226 void strbuf_expand(struct strbuf
*sb
, const char *format
, expand_fn_t fn
,
233 percent
= strchrnul(format
, '%');
234 strbuf_add(sb
, format
, percent
- format
);
237 format
= percent
+ 1;
239 consumed
= fn(sb
, format
, context
);
243 strbuf_addch(sb
, '%');
247 size_t strbuf_expand_dict_cb(struct strbuf
*sb
, const char *placeholder
,
250 struct strbuf_expand_dict_entry
*e
= context
;
253 for (; e
->placeholder
&& (len
= strlen(e
->placeholder
)); e
++) {
254 if (!strncmp(placeholder
, e
->placeholder
, len
)) {
256 strbuf_addstr(sb
, e
->value
);
263 size_t strbuf_fread(struct strbuf
*sb
, size_t size
, FILE *f
)
266 size_t oldalloc
= sb
->alloc
;
268 strbuf_grow(sb
, size
);
269 res
= fread(sb
->buf
+ sb
->len
, 1, size
, f
);
271 strbuf_setlen(sb
, sb
->len
+ res
);
272 else if (oldalloc
== 0)
277 ssize_t
strbuf_read(struct strbuf
*sb
, int fd
, size_t hint
)
279 size_t oldlen
= sb
->len
;
280 size_t oldalloc
= sb
->alloc
;
282 strbuf_grow(sb
, hint
? hint
: 8192);
286 cnt
= xread(fd
, sb
->buf
+ sb
->len
, sb
->alloc
- sb
->len
- 1);
291 strbuf_setlen(sb
, oldlen
);
297 strbuf_grow(sb
, 8192);
300 sb
->buf
[sb
->len
] = '\0';
301 return sb
->len
- oldlen
;
304 #define STRBUF_MAXLINK (2*PATH_MAX)
306 int strbuf_readlink(struct strbuf
*sb
, const char *path
, size_t hint
)
308 size_t oldalloc
= sb
->alloc
;
313 while (hint
< STRBUF_MAXLINK
) {
316 strbuf_grow(sb
, hint
);
317 len
= readlink(path
, sb
->buf
, hint
);
321 } else if (len
< hint
) {
322 strbuf_setlen(sb
, len
);
326 /* .. the buffer was too small - try again */
334 int strbuf_getwholeline(struct strbuf
*sb
, FILE *fp
, int term
)
343 while ((ch
= fgetc(fp
)) != EOF
) {
345 sb
->buf
[sb
->len
++] = ch
;
349 if (ch
== EOF
&& sb
->len
== 0)
352 sb
->buf
[sb
->len
] = '\0';
356 int strbuf_getline(struct strbuf
*sb
, FILE *fp
, int term
)
358 if (strbuf_getwholeline(sb
, fp
, term
))
360 if (sb
->buf
[sb
->len
-1] == term
)
361 strbuf_setlen(sb
, sb
->len
-1);
365 int strbuf_read_file(struct strbuf
*sb
, const char *path
, size_t hint
)
369 fd
= open(path
, O_RDONLY
);
372 len
= strbuf_read(sb
, fd
, hint
);
380 int strbuf_branchname(struct strbuf
*sb
, const char *name
)
382 int len
= strlen(name
);
383 if (interpret_branch_name(name
, sb
) == len
)
385 strbuf_add(sb
, name
, len
);
389 int strbuf_check_branch_ref(struct strbuf
*sb
, const char *name
)
391 strbuf_branchname(sb
, name
);
392 strbuf_splice(sb
, 0, 0, "refs/heads/", 11);
393 return check_ref_format(sb
->buf
);