162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel/* 262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel * $Id: printbuf.c,v 1.5 2006/01/26 02:16:28 mclark Exp $ 362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel * 462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel * Copyright (c) 2004, 2005 Metaparadigm Pte. Ltd. 562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel * Michael Clark <michael@metaparadigm.com> 662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel * 762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel * This library is free software; you can redistribute it and/or modify 862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel * it under the terms of the MIT license. See COPYING for details. 962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel * 1062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel * 1162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel * Copyright (c) 2008-2009 Yahoo! Inc. All rights reserved. 1262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel * The copyrights to the contents of this file are licensed under the MIT License 1362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel * (http://www.opensource.org/licenses/mit-license.php) 1462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel */ 1562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 1662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#include "config.h" 1762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 1862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#include <stdio.h> 1962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#include <stdlib.h> 2062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#include <string.h> 2162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 2262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#ifdef HAVE_STDARG_H 2362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel# include <stdarg.h> 2462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#else /* !HAVE_STDARG_H */ 2562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel# error Not enough var arg support! 2662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#endif /* HAVE_STDARG_H */ 2762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 2862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#include "debug.h" 2962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#include "printbuf.h" 3062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 3162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudelstatic int printbuf_extend(struct printbuf *p, int min_size); 3262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 3362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudelstruct printbuf* printbuf_new(void) 3462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel{ 3562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel struct printbuf *p; 3662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 3762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel p = (struct printbuf*)calloc(1, sizeof(struct printbuf)); 3862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if(!p) return NULL; 3962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel p->size = 32; 4062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel p->bpos = 0; 4162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if(!(p->buf = (char*)malloc(p->size))) { 4262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel free(p); 4362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel return NULL; 4462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel } 4562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel return p; 4662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel} 4762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 4862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 4962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel/** 5062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel * Extend the buffer p so it has a size of at least min_size. 5162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel * 5262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel * If the current size is large enough, nothing is changed. 5362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel * 5462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel * Note: this does not check the available space! The caller 5562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel * is responsible for performing those calculations. 5662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel */ 5762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudelstatic int printbuf_extend(struct printbuf *p, int min_size) 5862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel{ 5962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel char *t; 6062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel int new_size; 6162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 6262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if (p->size >= min_size) 6362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel return 0; 6462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 6562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel new_size = p->size * 2; 6662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if (new_size < min_size + 8) 6762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel new_size = min_size + 8; 6862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#ifdef PRINTBUF_DEBUG 6962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel MC_DEBUG("printbuf_memappend: realloc " 7062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel "bpos=%d min_size=%d old_size=%d new_size=%d\n", 7162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel p->bpos, min_size, p->size, new_size); 7262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#endif /* PRINTBUF_DEBUG */ 7362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if(!(t = (char*)realloc(p->buf, new_size))) 7462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel return -1; 7562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel p->size = new_size; 7662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel p->buf = t; 7762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel return 0; 7862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel} 7962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 8062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudelint printbuf_memappend(struct printbuf *p, const char *buf, int size) 8162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel{ 8262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if (p->size <= p->bpos + size + 1) { 8362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if (printbuf_extend(p, p->bpos + size + 1) < 0) 8462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel return -1; 8562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel } 8662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel memcpy(p->buf + p->bpos, buf, size); 8762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel p->bpos += size; 8862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel p->buf[p->bpos]= '\0'; 8962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel return size; 9062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel} 9162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 9262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudelint printbuf_memset(struct printbuf *pb, int offset, int charvalue, int len) 9362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel{ 9462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel int size_needed; 9562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 9662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if (offset == -1) 9762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel offset = pb->bpos; 9862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel size_needed = offset + len; 9962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if (pb->size < size_needed) 10062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel { 10162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if (printbuf_extend(pb, size_needed) < 0) 10262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel return -1; 10362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel } 10462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 10562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel memset(pb->buf + offset, charvalue, len); 10662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if (pb->bpos < size_needed) 10762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel pb->bpos = size_needed; 10862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 10962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel return 0; 11062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel} 11162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 11262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#if !defined(HAVE_VSNPRINTF) && defined(_MSC_VER) 11362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel# define vsnprintf _vsnprintf 11462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#elif !defined(HAVE_VSNPRINTF) /* !HAVE_VSNPRINTF */ 11562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel# error Need vsnprintf! 11662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#endif /* !HAVE_VSNPRINTF && defined(WIN32) */ 11762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 11862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#if !defined(HAVE_VASPRINTF) 11962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel/* CAW: compliant version of vasprintf */ 12062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudelstatic int vasprintf(char **buf, const char *fmt, va_list ap) 12162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel{ 12262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#ifndef WIN32 12362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel static char _T_emptybuffer = '\0'; 12462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#endif /* !defined(WIN32) */ 12562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel int chars; 12662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel char *b; 12762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 12862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if(!buf) { return -1; } 12962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 13062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#ifdef WIN32 13162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel chars = _vscprintf(fmt, ap)+1; 13262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#else /* !defined(WIN32) */ 13362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel /* CAW: RAWR! We have to hope to god here that vsnprintf doesn't overwrite 13462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel our buffer like on some 64bit sun systems.... but hey, its time to move on */ 13562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel chars = vsnprintf(&_T_emptybuffer, 0, fmt, ap)+1; 13662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if(chars < 0) { chars *= -1; } /* CAW: old glibc versions have this problem */ 13762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#endif /* defined(WIN32) */ 13862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 13962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel b = (char*)malloc(sizeof(char)*chars); 14062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if(!b) { return -1; } 14162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 14262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if((chars = vsprintf(b, fmt, ap)) < 0) 14362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel { 14462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel free(b); 14562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel } else { 14662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel *buf = b; 14762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel } 14862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 14962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel return chars; 15062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel} 15162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel#endif /* !HAVE_VASPRINTF */ 15262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 15362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudelint sprintbuf(struct printbuf *p, const char *msg, ...) 15462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel{ 15562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel va_list ap; 15662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel char *t; 15762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel int size; 15862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel char buf[128]; 15962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 16062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel /* user stack buffer first */ 16162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel va_start(ap, msg); 16262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel size = vsnprintf(buf, 128, msg, ap); 16362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel va_end(ap); 16462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel /* if string is greater than stack buffer, then use dynamic string 16562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel with vasprintf. Note: some implementation of vsnprintf return -1 16662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if output is truncated whereas some return the number of bytes that 16762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel would have been written - this code handles both cases. */ 16862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if(size == -1 || size > 127) { 16962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel va_start(ap, msg); 17062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if((size = vasprintf(&t, msg, ap)) < 0) { va_end(ap); return -1; } 17162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel va_end(ap); 17262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel printbuf_memappend(p, t, size); 17362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel free(t); 17462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel return size; 17562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel } else { 17662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel printbuf_memappend(p, buf, size); 17762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel return size; 17862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel } 17962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel} 18062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 18162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudelvoid printbuf_reset(struct printbuf *p) 18262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel{ 18362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel p->buf[0] = '\0'; 18462874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel p->bpos = 0; 18562874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel} 18662874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel 18762874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudelvoid printbuf_free(struct printbuf *p) 18862874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel{ 18962874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel if(p) { 19062874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel free(p->buf); 19162874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel free(p); 19262874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel } 19362874b3b227d7dc3db44065741cd05d4d8f9dc48Thierry Strudel} 194