serval-dna/xprintf.c
2018-05-07 23:31:15 +09:30

138 lines
3.5 KiB
C

/*
Serval extensible printf.
Copyright (C) 2012 Serval Project Inc.
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, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*/
#include <stdio.h>
#include <stdlib.h>
#include "strbuf.h"
#include "xprintf.h"
/* Primitives.
*/
void xprintf(XPRINTF xfp, const char *fmt, ...)
{
va_list ap;
va_start(ap, fmt);
vxprintf(xfp, fmt, ap);
va_end(ap);
}
void vxprintf(XPRINTF xfp, const char *fmt, va_list ap)
{
(*xfp.func)(xfp.context, fmt, ap);
}
void xputs(const char *str, XPRINTF xpf)
{
xprintf(xpf, "%s", str);
}
void xputc(char c, XPRINTF xpf)
{
xprintf(xpf, "%c", c);
}
size_t xhexdump_line(XPRINTF xpf, const unsigned char *addr, const size_t len, const size_t off)
{
xprintf(xpf, "%04zx :", off);
size_t i;
for (i = 0; i < 16 && off + i < len; i++)
xprintf(xpf, " %02x", addr[off + i]);
for (; i < 16; i++)
xputs(" ", xpf);
xputs(" ", xpf);
for (i = 0; i < 16 && off + i < len; ++i) {
unsigned char c = addr[off + i];
xprintf(xpf, "%c", c >= ' ' && c < 0x7f ? c : '.');
}
return i;
}
void xhexdump(XPRINTF xpf, const unsigned char *addr, const size_t len, const char *line_prefix)
{
size_t off = 0;
while (off < len) {
xputs(line_prefix, xpf);
size_t skip = xhexdump_line(xpf, addr, len, off);
off += skip;
addr += skip;
xputc('\n', xpf);
}
}
/* Implementations for various destinations.
*/
void _cx_vprintf_stdio(void *context, const char *fmt, va_list ap)
{
vfprintf((FILE *)context, fmt, ap);
}
static void grow_mallocbuf(struct mallocbuf *mb, size_t extra)
{
size_t newsize = mb->size + extra;
// Round up to nearest multiple of 1024.
newsize = newsize + 1024 - ((newsize - 1) % 1024 + 1);
char *newbuf = realloc(mb->buffer, newsize);
if (newbuf) {
mb->current += newbuf - mb->buffer;
mb->size = newsize;
mb->buffer = newbuf;
}
}
void _cx_vprintf_mallocbuf(void *context, const char *fmt, va_list ap)
{
struct mallocbuf *mb = (struct mallocbuf *) context;
if (mb->buffer == NULL)
grow_mallocbuf(mb, 1024);
if (mb->current) {
if (mb->current + 1 >= mb->buffer + mb->size)
grow_mallocbuf(mb, 1024);
va_list ap1;
va_copy(ap1, ap);
int n = vsnprintf(mb->current, mb->buffer + mb->size - mb->current, fmt, ap1);
va_end(ap1);
char *newcurrent = mb->current + n;
char *end = mb->buffer + mb->size;
if (newcurrent < end)
mb->current = newcurrent;
else {
grow_mallocbuf(mb, newcurrent - end + 1);
va_list ap1;
va_copy(ap1, ap);
n = vsnprintf(mb->current, mb->buffer + mb->size - mb->current, fmt, ap1);
va_end(ap1);
char *newcurrent = mb->current + n;
char *end = mb->buffer + mb->size;
if (newcurrent < end)
mb->current = newcurrent;
else {
mb->current = mb->buffer + mb->size - 1;
*mb->current = '\0';
}
}
}
}
void _cx_vprintf_strbuf(void *context, const char *fmt, va_list ap)
{
strbuf_vsprintf((strbuf)context, fmt, ap);
}