4 * Used as the default ->buf value, so that people can always assume
5 * buf is non NULL and ->buf is NUL terminated even for a freshly
8 char strbuf_slopbuf
[1];
10 void strbuf_init(struct strbuf
*sb
, size_t hint
)
12 sb
->alloc
= sb
->len
= 0;
13 sb
->buf
= strbuf_slopbuf
;
15 strbuf_grow(sb
, hint
);
18 void strbuf_release(struct strbuf
*sb
)
26 char *strbuf_detach(struct strbuf
*sb
, size_t *sz
)
28 char *res
= sb
->alloc
? sb
->buf
: NULL
;
35 void strbuf_attach(struct strbuf
*sb
, void *buf
, size_t len
, size_t alloc
)
42 sb
->buf
[sb
->len
] = '\0';
45 void strbuf_grow(struct strbuf
*sb
, size_t extra
)
47 if (sb
->len
+ extra
+ 1 <= sb
->len
)
48 die("you want to use way too much memory");
51 ALLOC_GROW(sb
->buf
, sb
->len
+ extra
+ 1, sb
->alloc
);
54 void strbuf_rtrim(struct strbuf
*sb
)
56 while (sb
->len
> 0 && isspace((unsigned char)sb
->buf
[sb
->len
- 1]))
58 sb
->buf
[sb
->len
] = '\0';
61 int strbuf_cmp(struct strbuf
*a
, struct strbuf
*b
)
64 if (a
->len
< b
->len
) {
65 cmp
= memcmp(a
->buf
, b
->buf
, a
->len
);
66 return cmp
? cmp
: -1;
68 cmp
= memcmp(a
->buf
, b
->buf
, b
->len
);
69 return cmp
? cmp
: a
->len
!= b
->len
;
73 void strbuf_splice(struct strbuf
*sb
, size_t pos
, size_t len
,
74 const void *data
, size_t dlen
)
77 die("you want to use way too much memory");
79 die("`pos' is too far after the end of the buffer");
80 if (pos
+ len
> sb
->len
)
81 die("`pos + len' is too far after the end of the buffer");
84 strbuf_grow(sb
, dlen
- len
);
85 memmove(sb
->buf
+ pos
+ dlen
,
88 memcpy(sb
->buf
+ pos
, data
, dlen
);
89 strbuf_setlen(sb
, sb
->len
+ dlen
- len
);
92 void strbuf_insert(struct strbuf
*sb
, size_t pos
, const void *data
, size_t len
)
94 strbuf_splice(sb
, pos
, 0, data
, len
);
97 void strbuf_remove(struct strbuf
*sb
, size_t pos
, size_t len
)
99 strbuf_splice(sb
, pos
, len
, NULL
, 0);
102 void strbuf_add(struct strbuf
*sb
, const void *data
, size_t len
)
104 strbuf_grow(sb
, len
);
105 memcpy(sb
->buf
+ sb
->len
, data
, len
);
106 strbuf_setlen(sb
, sb
->len
+ len
);
109 void strbuf_addf(struct strbuf
*sb
, const char *fmt
, ...)
115 len
= vsnprintf(sb
->buf
+ sb
->len
, sb
->alloc
- sb
->len
, fmt
, ap
);
120 if (len
> strbuf_avail(sb
)) {
121 strbuf_grow(sb
, len
);
123 len
= vsnprintf(sb
->buf
+ sb
->len
, sb
->alloc
- sb
->len
, fmt
, ap
);
125 if (len
> strbuf_avail(sb
)) {
126 die("this should not happen, your snprintf is broken");
129 strbuf_setlen(sb
, sb
->len
+ len
);
132 size_t strbuf_fread(struct strbuf
*sb
, size_t size
, FILE *f
)
136 strbuf_grow(sb
, size
);
137 res
= fread(sb
->buf
+ sb
->len
, 1, size
, f
);
139 strbuf_setlen(sb
, sb
->len
+ res
);
144 ssize_t
strbuf_read(struct strbuf
*sb
, int fd
, size_t hint
)
146 size_t oldlen
= sb
->len
;
148 strbuf_grow(sb
, hint
? hint
: 8192);
152 cnt
= xread(fd
, sb
->buf
+ sb
->len
, sb
->alloc
- sb
->len
- 1);
154 strbuf_setlen(sb
, oldlen
);
160 strbuf_grow(sb
, 8192);
163 sb
->buf
[sb
->len
] = '\0';
164 return sb
->len
- oldlen
;
167 int strbuf_getline(struct strbuf
*sb
, FILE *fp
, int term
)
176 while ((ch
= fgetc(fp
)) != EOF
) {
180 sb
->buf
[sb
->len
++] = ch
;
182 if (ch
== EOF
&& sb
->len
== 0)
185 sb
->buf
[sb
->len
] = '\0';
189 int strbuf_read_file(struct strbuf
*sb
, const char *path
, size_t hint
)
193 fd
= open(path
, O_RDONLY
);
196 len
= strbuf_read(sb
, fd
, hint
);