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
;
14 * Used as the default ->buf value, so that people can always assume
15 * buf is non NULL and ->buf is NUL terminated even for a freshly
18 char strbuf_slopbuf
[1];
20 void strbuf_init(struct strbuf
*sb
, size_t hint
)
22 sb
->alloc
= sb
->len
= 0;
23 sb
->buf
= strbuf_slopbuf
;
25 strbuf_grow(sb
, hint
);
28 void strbuf_release(struct strbuf
*sb
)
36 char *strbuf_detach(struct strbuf
*sb
, size_t *sz
)
38 char *res
= sb
->alloc
? sb
->buf
: NULL
;
45 void strbuf_attach(struct strbuf
*sb
, void *buf
, size_t len
, size_t alloc
)
52 sb
->buf
[sb
->len
] = '\0';
55 void strbuf_grow(struct strbuf
*sb
, size_t extra
)
57 if (sb
->len
+ extra
+ 1 <= sb
->len
)
58 die("you want to use way too much memory");
61 ALLOC_GROW(sb
->buf
, sb
->len
+ extra
+ 1, sb
->alloc
);
64 void strbuf_trim(struct strbuf
*sb
)
67 while (sb
->len
> 0 && isspace((unsigned char)sb
->buf
[sb
->len
- 1]))
69 while (sb
->len
> 0 && isspace(*b
)) {
73 memmove(sb
->buf
, b
, sb
->len
);
74 sb
->buf
[sb
->len
] = '\0';
76 void strbuf_rtrim(struct strbuf
*sb
)
78 while (sb
->len
> 0 && isspace((unsigned char)sb
->buf
[sb
->len
- 1]))
80 sb
->buf
[sb
->len
] = '\0';
83 void strbuf_ltrim(struct strbuf
*sb
)
86 while (sb
->len
> 0 && isspace(*b
)) {
90 memmove(sb
->buf
, b
, sb
->len
);
91 sb
->buf
[sb
->len
] = '\0';
94 void strbuf_tolower(struct strbuf
*sb
)
97 for (i
= 0; i
< sb
->len
; i
++)
98 sb
->buf
[i
] = tolower(sb
->buf
[i
]);
101 struct strbuf
**strbuf_split(const struct strbuf
*sb
, int delim
)
103 int alloc
= 2, pos
= 0;
108 ret
= xcalloc(alloc
, sizeof(struct strbuf
*));
110 while (n
< sb
->buf
+ sb
->len
) {
112 n
= memchr(n
, delim
, sb
->len
- (n
- sb
->buf
));
113 if (pos
+ 1 >= alloc
) {
115 ret
= xrealloc(ret
, sizeof(struct strbuf
*) * alloc
);
118 n
= sb
->buf
+ sb
->len
- 1;
120 t
= xmalloc(sizeof(struct strbuf
));
122 strbuf_add(t
, p
, len
);
130 void strbuf_list_free(struct strbuf
**sbs
)
132 struct strbuf
**s
= sbs
;
141 int strbuf_cmp(const struct strbuf
*a
, const struct strbuf
*b
)
143 int len
= a
->len
< b
->len
? a
->len
: b
->len
;
144 int cmp
= memcmp(a
->buf
, b
->buf
, len
);
147 return a
->len
< b
->len
? -1: a
->len
!= b
->len
;
150 void strbuf_splice(struct strbuf
*sb
, size_t pos
, size_t len
,
151 const void *data
, size_t dlen
)
154 die("you want to use way too much memory");
156 die("`pos' is too far after the end of the buffer");
157 if (pos
+ len
> sb
->len
)
158 die("`pos + len' is too far after the end of the buffer");
161 strbuf_grow(sb
, dlen
- len
);
162 memmove(sb
->buf
+ pos
+ dlen
,
164 sb
->len
- pos
- len
);
165 memcpy(sb
->buf
+ pos
, data
, dlen
);
166 strbuf_setlen(sb
, sb
->len
+ dlen
- len
);
169 void strbuf_insert(struct strbuf
*sb
, size_t pos
, const void *data
, size_t len
)
171 strbuf_splice(sb
, pos
, 0, data
, len
);
174 void strbuf_remove(struct strbuf
*sb
, size_t pos
, size_t len
)
176 strbuf_splice(sb
, pos
, len
, NULL
, 0);
179 void strbuf_add(struct strbuf
*sb
, const void *data
, size_t len
)
181 strbuf_grow(sb
, len
);
182 memcpy(sb
->buf
+ sb
->len
, data
, len
);
183 strbuf_setlen(sb
, sb
->len
+ len
);
186 void strbuf_adddup(struct strbuf
*sb
, size_t pos
, size_t len
)
188 strbuf_grow(sb
, len
);
189 memcpy(sb
->buf
+ sb
->len
, sb
->buf
+ pos
, len
);
190 strbuf_setlen(sb
, sb
->len
+ len
);
193 void strbuf_addf(struct strbuf
*sb
, const char *fmt
, ...)
198 if (!strbuf_avail(sb
))
201 len
= vsnprintf(sb
->buf
+ sb
->len
, sb
->alloc
- sb
->len
, fmt
, ap
);
204 die("your vsnprintf is broken");
205 if (len
> strbuf_avail(sb
)) {
206 strbuf_grow(sb
, len
);
208 len
= vsnprintf(sb
->buf
+ sb
->len
, sb
->alloc
- sb
->len
, fmt
, ap
);
210 if (len
> strbuf_avail(sb
)) {
211 die("this should not happen, your snprintf is broken");
214 strbuf_setlen(sb
, sb
->len
+ len
);
217 void strbuf_expand(struct strbuf
*sb
, const char *format
, expand_fn_t fn
,
224 percent
= strchrnul(format
, '%');
225 strbuf_add(sb
, format
, percent
- format
);
228 format
= percent
+ 1;
230 consumed
= fn(sb
, format
, context
);
234 strbuf_addch(sb
, '%');
238 size_t strbuf_expand_dict_cb(struct strbuf
*sb
, const char *placeholder
,
241 struct strbuf_expand_dict_entry
*e
= context
;
244 for (; e
->placeholder
&& (len
= strlen(e
->placeholder
)); e
++) {
245 if (!strncmp(placeholder
, e
->placeholder
, len
)) {
247 strbuf_addstr(sb
, e
->value
);
254 size_t strbuf_fread(struct strbuf
*sb
, size_t size
, FILE *f
)
257 size_t oldalloc
= sb
->alloc
;
259 strbuf_grow(sb
, size
);
260 res
= fread(sb
->buf
+ sb
->len
, 1, size
, f
);
262 strbuf_setlen(sb
, sb
->len
+ res
);
263 else if (oldalloc
== 0)
268 ssize_t
strbuf_read(struct strbuf
*sb
, int fd
, size_t hint
)
270 size_t oldlen
= sb
->len
;
271 size_t oldalloc
= sb
->alloc
;
273 strbuf_grow(sb
, hint
? hint
: 8192);
277 cnt
= xread(fd
, sb
->buf
+ sb
->len
, sb
->alloc
- sb
->len
- 1);
282 strbuf_setlen(sb
, oldlen
);
288 strbuf_grow(sb
, 8192);
291 sb
->buf
[sb
->len
] = '\0';
292 return sb
->len
- oldlen
;
295 #define STRBUF_MAXLINK (2*PATH_MAX)
297 int strbuf_readlink(struct strbuf
*sb
, const char *path
, size_t hint
)
299 size_t oldalloc
= sb
->alloc
;
304 while (hint
< STRBUF_MAXLINK
) {
307 strbuf_grow(sb
, hint
);
308 len
= readlink(path
, sb
->buf
, hint
);
312 } else if (len
< hint
) {
313 strbuf_setlen(sb
, len
);
317 /* .. the buffer was too small - try again */
325 int strbuf_getline(struct strbuf
*sb
, FILE *fp
, int term
)
334 while ((ch
= fgetc(fp
)) != EOF
) {
338 sb
->buf
[sb
->len
++] = ch
;
340 if (ch
== EOF
&& sb
->len
== 0)
343 sb
->buf
[sb
->len
] = '\0';
347 int strbuf_read_file(struct strbuf
*sb
, const char *path
, size_t hint
)
351 fd
= open(path
, O_RDONLY
);
354 len
= strbuf_read(sb
, fd
, hint
);
362 int strbuf_branchname(struct strbuf
*sb
, const char *name
)
364 int len
= strlen(name
);
365 if (interpret_branch_name(name
, sb
) == len
)
367 strbuf_add(sb
, name
, len
);
371 int strbuf_check_branch_ref(struct strbuf
*sb
, const char *name
)
373 strbuf_branchname(sb
, name
);
374 strbuf_splice(sb
, 0, 0, "refs/heads/", 11);
375 return check_ref_format(sb
->buf
);