[thirdparty/git.git] / strbuf.c

#include "cache.h"
#include "refs.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 len = a->len < b->len ? a->len: b->len;
	int cmp = memcmp(a->buf, b->buf, len);
	if (cmp)
		return cmp;
	return a->len < b->len ? -1: 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_expand_dict_cb(struct strbuf *sb, const char *placeholder,
		void *context)
{
	struct strbuf_expand_dict_entry *e = context;
	size_t len;

	for (; e->placeholder && (len = strlen(e->placeholder)); e++) {
		if (!strncmp(placeholder, e->placeholder, len)) {
			if (e->value)
				strbuf_addstr(sb, e->value);
			return len;
		}
	}
	return 0;
}

size_t strbuf_fread(struct strbuf *sb, size_t size, FILE *f)
{
	size_t res;
	size_t oldalloc = sb->alloc;

	strbuf_grow(sb, size);
	res = fread(sb->buf + sb->len, 1, size, f);
	if (res > 0)
		strbuf_setlen(sb, sb->len + res);
	else if (res < 0 && oldalloc == 0)
		strbuf_release(sb);
	return res;
}

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

	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) {
			if (oldalloc == 0)
				strbuf_release(sb);
			else
				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;
}

#define STRBUF_MAXLINK (2*PATH_MAX)

int strbuf_readlink(struct strbuf *sb, const char *path, size_t hint)
{
	size_t oldalloc = sb->alloc;

	if (hint < 32)
		hint = 32;

	while (hint < STRBUF_MAXLINK) {
		int len;

		strbuf_grow(sb, hint);
		len = readlink(path, sb->buf, hint);
		if (len < 0) {
			if (errno != ERANGE)
				break;
		} else if (len < hint) {
			strbuf_setlen(sb, len);
			return 0;
		}

		/* .. the buffer was too small - try again */
		hint *= 2;
	}
	if (oldalloc == 0)
		strbuf_release(sb);
	return -1;
}

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;
}

int strbuf_branchname(struct strbuf *sb, const char *name)
{
	int len = strlen(name);
	if (interpret_branch_name(name, sb) == len)
		return 0;
	strbuf_add(sb, name, len);
	return len;
}

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