[thirdparty/git.git] / strbuf.c

#include "cache.h"

int prefixcmp(const char *str, const char *prefix)
{
	for (; ; str++, prefix++)
		if (!*prefix)
			return 0;
		else if (*str != *prefix)
			return (unsigned char)*prefix - (unsigned char)*str;
}

/*
 * Used as the default ->buf value, so that people can always assume
 * buf is non NULL and ->buf is NUL terminated even for a freshly
 * initialized strbuf.
 */
char strbuf_slopbuf[1];

void strbuf_init(struct strbuf *sb, size_t hint)
{
	sb->alloc = sb->len = 0;
	sb->buf = strbuf_slopbuf;
	if (hint)
		strbuf_grow(sb, hint);
}

void strbuf_release(struct strbuf *sb)
{
	if (sb->alloc) {
		free(sb->buf);
		strbuf_init(sb, 0);
	}
}

char *strbuf_detach(struct strbuf *sb, size_t *sz)
{
	char *res = sb->alloc ? sb->buf : NULL;
	if (sz)
		*sz = sb->len;
	strbuf_init(sb, 0);
	return res;
}

void strbuf_attach(struct strbuf *sb, void *buf, size_t len, size_t alloc)
{
	strbuf_release(sb);
	sb->buf   = buf;
	sb->len   = len;
	sb->alloc = alloc;
	strbuf_grow(sb, 0);
	sb->buf[sb->len] = '\0';
}

void strbuf_grow(struct strbuf *sb, size_t extra)
{
	if (sb->len + extra + 1 <= sb->len)
		die("you want to use way too much memory");
	if (!sb->alloc)
		sb->buf = NULL;
	ALLOC_GROW(sb->buf, sb->len + extra + 1, sb->alloc);
}

void strbuf_trim(struct strbuf *sb)
{
	char *b = sb->buf;
	while (sb->len > 0 && isspace((unsigned char)sb->buf[sb->len - 1]))
		sb->len--;
	while (sb->len > 0 && isspace(*b)) {
		b++;
		sb->len--;
	}
	memmove(sb->buf, b, sb->len);
	sb->buf[sb->len] = '\0';
}
void strbuf_rtrim(struct strbuf *sb)
{
	while (sb->len > 0 && isspace((unsigned char)sb->buf[sb->len - 1]))
		sb->len--;
	sb->buf[sb->len] = '\0';
}

void strbuf_ltrim(struct strbuf *sb)
{
	char *b = sb->buf;
	while (sb->len > 0 && isspace(*b)) {
		b++;
		sb->len--;
	}
	memmove(sb->buf, b, sb->len);
	sb->buf[sb->len] = '\0';
}

void strbuf_tolower(struct strbuf *sb)
{
	int i;
	for (i = 0; i < sb->len; i++)
		sb->buf[i] = tolower(sb->buf[i]);
}

struct strbuf **strbuf_split(const struct strbuf *sb, int delim)
{
	int alloc = 2, pos = 0;
	char *n, *p;
	struct strbuf **ret;
	struct strbuf *t;

	ret = xcalloc(alloc, sizeof(struct strbuf *));
	p = n = sb->buf;
	while (n < sb->buf + sb->len) {
		int len;
		n = memchr(n, delim, sb->len - (n - sb->buf));
		if (pos + 1 >= alloc) {
			alloc = alloc * 2;
			ret = xrealloc(ret, sizeof(struct strbuf *) * alloc);
		}
		if (!n)
			n = sb->buf + sb->len - 1;
		len = n - p + 1;
		t = xmalloc(sizeof(struct strbuf));
		strbuf_init(t, len);
		strbuf_add(t, p, len);
		ret[pos] = t;
		ret[++pos] = NULL;
		p = ++n;
	}
	return ret;
}

void strbuf_list_free(struct strbuf **sbs)
{
	struct strbuf **s = sbs;

	while (*s) {
		strbuf_release(*s);
		free(*s++);
	}
	free(sbs);
}

int strbuf_cmp(const struct strbuf *a, const struct strbuf *b)
{
	int cmp;
	if (a->len < b->len) {
		cmp = memcmp(a->buf, b->buf, a->len);
		return cmp ? cmp : -1;
	} else {
		cmp = memcmp(a->buf, b->buf, b->len);
		return cmp ? cmp : a->len != b->len;
	}
}

void strbuf_splice(struct strbuf *sb, size_t pos, size_t len,
				   const void *data, size_t dlen)
{
	if (pos + len < pos)
		die("you want to use way too much memory");
	if (pos > sb->len)
		die("`pos' is too far after the end of the buffer");
	if (pos + len > sb->len)
		die("`pos + len' is too far after the end of the buffer");

	if (dlen >= len)
		strbuf_grow(sb, dlen - len);
	memmove(sb->buf + pos + dlen,
			sb->buf + pos + len,
			sb->len - pos - len);
	memcpy(sb->buf + pos, data, dlen);
	strbuf_setlen(sb, sb->len + dlen - len);
}

void strbuf_insert(struct strbuf *sb, size_t pos, const void *data, size_t len)
{
	strbuf_splice(sb, pos, 0, data, len);
}

void strbuf_remove(struct strbuf *sb, size_t pos, size_t len)
{
	strbuf_splice(sb, pos, len, NULL, 0);
}

void strbuf_add(struct strbuf *sb, const void *data, size_t len)
{
	strbuf_grow(sb, len);
	memcpy(sb->buf + sb->len, data, len);
	strbuf_setlen(sb, sb->len + len);
}

void strbuf_adddup(struct strbuf *sb, size_t pos, size_t len)
{
	strbuf_grow(sb, len);
	memcpy(sb->buf + sb->len, sb->buf + pos, len);
	strbuf_setlen(sb, sb->len + len);
}

void strbuf_addf(struct strbuf *sb, const char *fmt, ...)
{
	int len;
	va_list ap;

	if (!strbuf_avail(sb))
		strbuf_grow(sb, 64);
	va_start(ap, fmt);
	len = vsnprintf(sb->buf + sb->len, sb->alloc - sb->len, fmt, ap);
	va_end(ap);
	if (len < 0)
		die("your vsnprintf is broken");
	if (len > strbuf_avail(sb)) {
		strbuf_grow(sb, len);
		va_start(ap, fmt);
		len = vsnprintf(sb->buf + sb->len, sb->alloc - sb->len, fmt, ap);
		va_end(ap);
		if (len > strbuf_avail(sb)) {
			die("this should not happen, your snprintf is broken");
		}
	}
	strbuf_setlen(sb, sb->len + len);
}

void strbuf_expand(struct strbuf *sb, const char *format, expand_fn_t fn,
		   void *context)
{
	for (;;) {
		const char *percent;
		size_t consumed;

		percent = strchrnul(format, '%');
		strbuf_add(sb, format, percent - format);
		if (!*percent)
			break;
		format = percent + 1;

		consumed = fn(sb, format, context);
		if (consumed)
			format += consumed;
		else
			strbuf_addch(sb, '%');
	}
}

size_t strbuf_fread(struct strbuf *sb, size_t size, FILE *f)
{
	size_t res;

	strbuf_grow(sb, size);
	res = fread(sb->buf + sb->len, 1, size, f);
	if (res > 0) {
		strbuf_setlen(sb, sb->len + res);
	}
	return res;
}

ssize_t strbuf_read(struct strbuf *sb, int fd, size_t hint)
{
	size_t oldlen = sb->len;

	strbuf_grow(sb, hint ? hint : 8192);
	for (;;) {
		ssize_t cnt;

		cnt = xread(fd, sb->buf + sb->len, sb->alloc - sb->len - 1);
		if (cnt < 0) {
			strbuf_setlen(sb, oldlen);
			return -1;
		}
		if (!cnt)
			break;
		sb->len += cnt;
		strbuf_grow(sb, 8192);
	}

	sb->buf[sb->len] = '\0';
	return sb->len - oldlen;
}

int strbuf_getline(struct strbuf *sb, FILE *fp, int term)
{
	int ch;

	strbuf_grow(sb, 0);
	if (feof(fp))
		return EOF;

	strbuf_reset(sb);
	while ((ch = fgetc(fp)) != EOF) {
		if (ch == term)
			break;
		strbuf_grow(sb, 1);
		sb->buf[sb->len++] = ch;
	}
	if (ch == EOF && sb->len == 0)
		return EOF;

	sb->buf[sb->len] = '\0';
	return 0;
}

int strbuf_read_file(struct strbuf *sb, const char *path, size_t hint)
{
	int fd, len;

	fd = open(path, O_RDONLY);
	if (fd < 0)
		return -1;
	len = strbuf_read(sb, fd, hint);
	close(fd);
	if (len < 0)
		return -1;

	return len;
}
Commit	Line	Data
812666c8	1	#include "cache.h"
d1df5743	2
698a68be JH	3	int prefixcmp(const char str, const char prefix)
	4	{
	5	for (; ; str++, prefix++)
	6	if (!*prefix)
	7	return 0;
	8	else if (str != prefix)
	9	return (unsigned char)prefix - (unsigned char)str;
	10	}
	11
b315c5c0 PH	12	/*
	13	* Used as the default ->buf value, so that people can always assume
	14	* buf is non NULL and ->buf is NUL terminated even for a freshly
	15	* initialized strbuf.
	16	*/
	17	char strbuf_slopbuf[1];
	18
917c9a71 PH	19	void strbuf_init(struct strbuf *sb, size_t hint)
917c9a71 PH	20	{
b315c5c0 PH	21	sb->alloc = sb->len = 0;
b315c5c0 PH	22	sb->buf = strbuf_slopbuf;
f1696ee3 PH	23	if (hint)
f1696ee3 PH	24	strbuf_grow(sb, hint);
d1df5743 JH	25	}
d1df5743 JH	26
917c9a71 PH	27	void strbuf_release(struct strbuf *sb)
917c9a71 PH	28	{
b315c5c0 PH	29	if (sb->alloc) {
	30	free(sb->buf);
	31	strbuf_init(sb, 0);
	32	}
b449f4cf PH	33	}
b449f4cf PH	34
b315c5c0	35	char strbuf_detach(struct strbuf sb, size_t *sz)
917c9a71	36	{
b315c5c0 PH	37	char *res = sb->alloc ? sb->buf : NULL;
	38	if (sz)
	39	*sz = sb->len;
f1696ee3	40	strbuf_init(sb, 0);
b449f4cf PH	41	return res;
	42	}
	43
917c9a71 PH	44	void strbuf_attach(struct strbuf sb, void buf, size_t len, size_t alloc)
	45	{
	46	strbuf_release(sb);
	47	sb->buf = buf;
	48	sb->len = len;
	49	sb->alloc = alloc;
	50	strbuf_grow(sb, 0);
	51	sb->buf[sb->len] = '\0';
	52	}
	53
	54	void strbuf_grow(struct strbuf *sb, size_t extra)
	55	{
b449f4cf PH	56	if (sb->len + extra + 1 <= sb->len)
b449f4cf PH	57	die("you want to use way too much memory");
b315c5c0 PH	58	if (!sb->alloc)
b315c5c0 PH	59	sb->buf = NULL;
b449f4cf PH	60	ALLOC_GROW(sb->buf, sb->len + extra + 1, sb->alloc);
	61	}
	62
eacd6dc5 LS	63	void strbuf_trim(struct strbuf *sb)
	64	{
	65	char *b = sb->buf;
	66	while (sb->len > 0 && isspace((unsigned char)sb->buf[sb->len - 1]))
	67	sb->len--;
	68	while (sb->len > 0 && isspace(*b)) {
	69	b++;
	70	sb->len--;
	71	}
	72	memmove(sb->buf, b, sb->len);
	73	sb->buf[sb->len] = '\0';
	74	}
f1696ee3 PH	75	void strbuf_rtrim(struct strbuf *sb)
	76	{
	77	while (sb->len > 0 && isspace((unsigned char)sb->buf[sb->len - 1]))
	78	sb->len--;
	79	sb->buf[sb->len] = '\0';
	80	}
	81
eacd6dc5 LS	82	void strbuf_ltrim(struct strbuf *sb)
	83	{
	84	char *b = sb->buf;
	85	while (sb->len > 0 && isspace(*b)) {
	86	b++;
	87	sb->len--;
	88	}
	89	memmove(sb->buf, b, sb->len);
	90	sb->buf[sb->len] = '\0';
	91	}
	92
	93	void strbuf_tolower(struct strbuf *sb)
	94	{
	95	int i;
	96	for (i = 0; i < sb->len; i++)
	97	sb->buf[i] = tolower(sb->buf[i]);
	98	}
	99
	100	struct strbuf *strbuf_split(const struct strbuf sb, int delim)
	101	{
	102	int alloc = 2, pos = 0;
	103	char n, p;
	104	struct strbuf **ret;
	105	struct strbuf *t;
	106
	107	ret = xcalloc(alloc, sizeof(struct strbuf *));
	108	p = n = sb->buf;
	109	while (n < sb->buf + sb->len) {
	110	int len;
	111	n = memchr(n, delim, sb->len - (n - sb->buf));
	112	if (pos + 1 >= alloc) {
	113	alloc = alloc * 2;
	114	ret = xrealloc(ret, sizeof(struct strbuf ) alloc);
	115	}
	116	if (!n)
	117	n = sb->buf + sb->len - 1;
	118	len = n - p + 1;
	119	t = xmalloc(sizeof(struct strbuf));
	120	strbuf_init(t, len);
	121	strbuf_add(t, p, len);
	122	ret[pos] = t;
	123	ret[++pos] = NULL;
	124	p = ++n;
	125	}
	126	return ret;
	127	}
	128
	129	void strbuf_list_free(struct strbuf **sbs)
	130	{
	131	struct strbuf **s = sbs;
	132
	133	while (*s) {
	134	strbuf_release(*s);
	135	free(*s++);
	136	}
	137	free(sbs);
	138	}
	139
9b200fd6	140	int strbuf_cmp(const struct strbuf a, const struct strbuf b)
45f66f64 PH	141	{
	142	int cmp;
	143	if (a->len < b->len) {
	144	cmp = memcmp(a->buf, b->buf, a->len);
	145	return cmp ? cmp : -1;
	146	} else {
	147	cmp = memcmp(a->buf, b->buf, b->len);
	148	return cmp ? cmp : a->len != b->len;
	149	}
	150	}
	151
917c9a71 PH	152	void strbuf_splice(struct strbuf *sb, size_t pos, size_t len,
	153	const void *data, size_t dlen)
	154	{
	155	if (pos + len < pos)
	156	die("you want to use way too much memory");
	157	if (pos > sb->len)
	158	die("`pos' is too far after the end of the buffer");
	159	if (pos + len > sb->len)
	160	die("`pos + len' is too far after the end of the buffer");
	161
	162	if (dlen >= len)
	163	strbuf_grow(sb, dlen - len);
	164	memmove(sb->buf + pos + dlen,
	165	sb->buf + pos + len,
	166	sb->len - pos - len);
	167	memcpy(sb->buf + pos, data, dlen);
	168	strbuf_setlen(sb, sb->len + dlen - len);
	169	}
	170
c76689df PH	171	void strbuf_insert(struct strbuf sb, size_t pos, const void data, size_t len)
	172	{
	173	strbuf_splice(sb, pos, 0, data, len);
	174	}
	175
	176	void strbuf_remove(struct strbuf *sb, size_t pos, size_t len)
	177	{
	178	strbuf_splice(sb, pos, len, NULL, 0);
	179	}
	180
917c9a71 PH	181	void strbuf_add(struct strbuf sb, const void data, size_t len)
917c9a71 PH	182	{
b449f4cf PH	183	strbuf_grow(sb, len);
	184	memcpy(sb->buf + sb->len, data, len);
	185	strbuf_setlen(sb, sb->len + len);
	186	}
	187
91db267e RS	188	void strbuf_adddup(struct strbuf *sb, size_t pos, size_t len)
	189	{
	190	strbuf_grow(sb, len);
	191	memcpy(sb->buf + sb->len, sb->buf + pos, len);
	192	strbuf_setlen(sb, sb->len + len);
	193	}
	194
917c9a71 PH	195	void strbuf_addf(struct strbuf sb, const char fmt, ...)
917c9a71 PH	196	{
b449f4cf PH	197	int len;
	198	va_list ap;
	199
f141bd80 SP	200	if (!strbuf_avail(sb))
f141bd80 SP	201	strbuf_grow(sb, 64);
b449f4cf PH	202	va_start(ap, fmt);
	203	len = vsnprintf(sb->buf + sb->len, sb->alloc - sb->len, fmt, ap);
	204	va_end(ap);
f141bd80 SP	205	if (len < 0)
f141bd80 SP	206	die("your vsnprintf is broken");
f1696ee3	207	if (len > strbuf_avail(sb)) {
b449f4cf PH	208	strbuf_grow(sb, len);
	209	va_start(ap, fmt);
	210	len = vsnprintf(sb->buf + sb->len, sb->alloc - sb->len, fmt, ap);
	211	va_end(ap);
f1696ee3	212	if (len > strbuf_avail(sb)) {
b449f4cf PH	213	die("this should not happen, your snprintf is broken");
	214	}
	215	}
	216	strbuf_setlen(sb, sb->len + len);
d1df5743 JH	217	}
d1df5743 JH	218
c3a670de MC	219	void strbuf_expand(struct strbuf sb, const char format, expand_fn_t fn,
c3a670de MC	220	void *context)
cde75e59 RS	221	{
cde75e59 RS	222	for (;;) {
c3a670de MC	223	const char *percent;
c3a670de MC	224	size_t consumed;
cde75e59 RS	225
	226	percent = strchrnul(format, '%');
	227	strbuf_add(sb, format, percent - format);
	228	if (!*percent)
	229	break;
	230	format = percent + 1;
	231
c3a670de MC	232	consumed = fn(sb, format, context);
	233	if (consumed)
	234	format += consumed;
	235	else
cde75e59 RS	236	strbuf_addch(sb, '%');
	237	}
	238	}
	239
917c9a71 PH	240	size_t strbuf_fread(struct strbuf sb, size_t size, FILE f)
917c9a71 PH	241	{
b449f4cf PH	242	size_t res;
	243
	244	strbuf_grow(sb, size);
	245	res = fread(sb->buf + sb->len, 1, size, f);
	246	if (res > 0) {
	247	strbuf_setlen(sb, sb->len + res);
d1df5743	248	}
b449f4cf	249	return res;
d1df5743 JH	250	}
d1df5743 JH	251
f1696ee3	252	ssize_t strbuf_read(struct strbuf *sb, int fd, size_t hint)
b449f4cf PH	253	{
	254	size_t oldlen = sb->len;
	255
f1696ee3	256	strbuf_grow(sb, hint ? hint : 8192);
b449f4cf PH	257	for (;;) {
	258	ssize_t cnt;
	259
b449f4cf PH	260	cnt = xread(fd, sb->buf + sb->len, sb->alloc - sb->len - 1);
	261	if (cnt < 0) {
	262	strbuf_setlen(sb, oldlen);
	263	return -1;
	264	}
	265	if (!cnt)
	266	break;
	267	sb->len += cnt;
f1696ee3	268	strbuf_grow(sb, 8192);
b449f4cf PH	269	}
	270
	271	sb->buf[sb->len] = '\0';
	272	return sb->len - oldlen;
d1df5743 JH	273	}
d1df5743 JH	274
e6c019d0	275	int strbuf_getline(struct strbuf sb, FILE fp, int term)
917c9a71	276	{
d1df5743	277	int ch;
e6c019d0 PH	278
	279	strbuf_grow(sb, 0);
	280	if (feof(fp))
	281	return EOF;
b449f4cf PH	282
b449f4cf PH	283	strbuf_reset(sb);
d1df5743 JH	284	while ((ch = fgetc(fp)) != EOF) {
	285	if (ch == term)
	286	break;
b449f4cf PH	287	strbuf_grow(sb, 1);
b449f4cf PH	288	sb->buf[sb->len++] = ch;
d1df5743	289	}
e6c019d0 PH	290	if (ch == EOF && sb->len == 0)
e6c019d0 PH	291	return EOF;
b449f4cf	292
b449f4cf	293	sb->buf[sb->len] = '\0';
e6c019d0	294	return 0;
d1df5743	295	}
a9390b9f	296
387e7e19	297	int strbuf_read_file(struct strbuf sb, const char path, size_t hint)
a9390b9f KH	298	{
	299	int fd, len;
	300
	301	fd = open(path, O_RDONLY);
	302	if (fd < 0)
	303	return -1;
387e7e19	304	len = strbuf_read(sb, fd, hint);
a9390b9f KH	305	close(fd);
	306	if (len < 0)
	307	return -1;
	308
	309	return len;
	310	}