/*
* Initialize the internals of an vsb.
* If buf is non-NULL, it points to a static or already-allocated string
* big enough to hold at least length characters.
*/
static struct vsb *
VSB_newbuf(struct vsb *s, char *buf, int length, int flags)
{
memset(s, 0, sizeof(*s));
s->magic = VSB_MAGIC;
s->s_flags = flags;
s->s_size = length;
s->s_buf = buf;
if ((s->s_flags & VSB_AUTOEXTEND) == 0) {
KASSERT(s->s_size > 1,
("attempt to create a too small vsb"));
}
if (s->s_buf != NULL)
return (s);
if ((flags & VSB_AUTOEXTEND) != 0)
s->s_size = VSB_extendsize(s->s_size);
s->s_buf = SBMALLOC(s->s_size);
if (s->s_buf == NULL)
return (NULL);
VSB_SETFLAG(s, VSB_DYNAMIC);
return (s);
}
/*
* Extend an vsb.
*/
static int
VSB_extend(struct vsb *s, int addlen)
{
char *newbuf;
int newsize;
if (!VSB_CANEXTEND(s))
return (-1);
newsize = VSB_extendsize(s->s_size + addlen);
newbuf = SBMALLOC(newsize);
if (newbuf == NULL)
return (-1);
memcpy(newbuf, s->s_buf, s->s_size);
if (VSB_ISDYNAMIC(s))
SBFREE(s->s_buf);
else
VSB_SETFLAG(s, VSB_DYNAMIC);
s->s_buf = newbuf;
s->s_size = newsize;
return (0);
}
/*
* Extend an vsb.
*/
static ssize_t
VSB_extend(struct vsb *s, ssize_t addlen)
{
char *newbuf;
ssize_t newsize;
if (!VSB_CANEXTEND(s))
return (-1);
newsize = VSB_extendsize(s->s_size + addlen);
if (VSB_ISDYNAMIC(s))
newbuf = realloc(s->s_buf, newsize);
else
newbuf = SBMALLOC(newsize);
if (newbuf == NULL)
return (-1);
if (!VSB_ISDYNAMIC(s)) {
memcpy(newbuf, s->s_buf, s->s_size);
VSB_SETFLAG(s, VSB_DYNAMIC);
}
s->s_buf = newbuf;
s->s_size = newsize;
return (0);
}
