161 lines
3.7 KiB
C
161 lines
3.7 KiB
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)
|
|
{
|
|
int src_bytes = u_str_print_size(src) - 1;
|
|
gbuf_grow(buf, src_bytes);
|
|
size_t copy_bytes = u_copy_chars(buf->buffer + buf->len, src, width);
|
|
gbuf_used(buf, copy_bytes);
|
|
if (copy_bytes != src_bytes) {
|
|
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;
|
|
}
|