gbuf.c

/*
 * Copyright 2008-2013 Various Authors
 * Copyright 2008 Timo Hirvonen
 *
 * This code is largely based on strbuf in the GIT version control system.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, see <http://www.gnu.org/licenses/>.
 */

#include "gbuf.h"
#include "options.h"
#include "utils.h"
#include "xmalloc.h"

#include <stdio.h>
#include <stdarg.h>

char gbuf_empty_buffer[1];

static inline void gbuf_init(struct gbuf *buf)
{
	buf->buffer = gbuf_empty_buffer;
	buf->alloc = 0;
	buf->len = 0;
}

void gbuf_grow(struct gbuf *buf, size_t more)
{
	size_t align = 64 - 1;
	size_t alloc = (buf->len + more + 1 + align) & ~align;

	if (alloc > buf->alloc) {
		if (!buf->alloc)
			buf->buffer = NULL;
		buf->alloc = alloc;
		buf->buffer = xrealloc(buf->buffer, buf->alloc);
		// gbuf is not NUL terminated if this was first alloc
		buf->buffer[buf->len] = 0;
	}
}

void gbuf_used(struct gbuf *buf, size_t used)
{
	buf->len += used;
	buf->buffer[buf->len] = 0;
}

void gbuf_free(struct gbuf *buf)
{
	if (buf->alloc)
		free(buf->buffer);
	gbuf_init(buf);
}

void gbuf_add_ch(struct gbuf *buf, char ch)
{
	gbuf_grow(buf, 1);
	buf->buffer[buf->len] = ch;
	gbuf_used(buf, 1);
}

void gbuf_add_uchar(struct gbuf *buf, uchar u)
{
	size_t uchar_len = 0;
	gbuf_grow(buf, 4);
	u_set_char(buf->buffer + buf->len, &uchar_len, u);
	gbuf_used(buf, uchar_len);
}

void gbuf_add_bytes(struct gbuf *buf, const void *data, size_t len)
{
	gbuf_grow(buf, len);
	memcpy(buf->buffer + buf->len, data, len);
	gbuf_used(buf, len);
}

void gbuf_add_str(struct gbuf *buf, const char *str)
{
	int len = strlen(str);

	if (!len)
		return;
	gbuf_grow(buf, len);
	memcpy(buf->buffer + buf->len, str, len);
	gbuf_used(buf, len);
}

static int gbuf_mark_clipped_text(struct gbuf *buf)
{
	int buf_width = u_str_width(buf->buffer);
	int clipped_mark_len = min_u(u_str_width(clipped_text_internal), buf_width);
	int skip = buf_width - clipped_mark_len;
	buf->len = u_skip_chars(buf->buffer, &skip, false);
	gbuf_grow(buf, strlen(clipped_text_internal));
	gbuf_used(buf, u_copy_chars(buf->buffer + buf->len, clipped_text_internal, &clipped_mark_len));
	return skip;
}

void gbuf_add_ustr(struct gbuf *buf, const char *src, int *width)
{
	gbuf_grow(buf, strlen(src));
	size_t copy_bytes = u_copy_chars(buf->buffer + buf->len, src, width);
	gbuf_used(buf, copy_bytes);
	if (src[copy_bytes] != '\0') {
		gbuf_set(buf, ' ', *width);
		*width = gbuf_mark_clipped_text(buf);
	}
}

void gbuf_addf(struct gbuf *buf, const char *fmt, ...)
{
	va_list ap;
	va_start(ap, fmt);
	gbuf_vaddf(buf, fmt, ap);
	va_end(ap);
}

void gbuf_vaddf(struct gbuf *buf, const char *fmt, va_list ap)
{
	va_list ap2;
	int slen;

	va_copy(ap2, ap);
	slen = vsnprintf(buf->buffer + buf->len, buf->alloc - buf->len, fmt, ap);

	if (slen > gbuf_avail(buf)) {
		gbuf_grow(buf, slen);
		slen = vsnprintf(buf->buffer + buf->len, buf->alloc - buf->len, fmt, ap2);
	}
	va_end(ap2);
	gbuf_used(buf, slen);
}

void gbuf_set(struct gbuf *buf, int c, size_t count)
{
	gbuf_grow(buf, count);
	memset(buf->buffer + buf->len, c, count);
	gbuf_used(buf, count);
}

char *gbuf_steal(struct gbuf *buf)
{
	char *b = buf->buffer;
	if (!buf->alloc)
		b = xnew0(char, 1);
	gbuf_init(buf);
	return b;
}