1178825Sdfr/* 2233294Sstas * Copyright (c) 2005 Kungliga Tekniska H��gskolan 3233294Sstas * (Royal Institute of Technology, Stockholm, Sweden). 4233294Sstas * All rights reserved. 5178825Sdfr * 6233294Sstas * Redistribution and use in source and binary forms, with or without 7233294Sstas * modification, are permitted provided that the following conditions 8233294Sstas * are met: 9178825Sdfr * 10233294Sstas * 1. Redistributions of source code must retain the above copyright 11233294Sstas * notice, this list of conditions and the following disclaimer. 12178825Sdfr * 13233294Sstas * 2. Redistributions in binary form must reproduce the above copyright 14233294Sstas * notice, this list of conditions and the following disclaimer in the 15233294Sstas * documentation and/or other materials provided with the distribution. 16178825Sdfr * 17233294Sstas * 3. Neither the name of the Institute nor the names of its contributors 18233294Sstas * may be used to endorse or promote products derived from this software 19233294Sstas * without specific prior written permission. 20178825Sdfr * 21233294Sstas * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND 22233294Sstas * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 23233294Sstas * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 24233294Sstas * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE 25233294Sstas * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 26233294Sstas * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 27233294Sstas * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 28233294Sstas * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 29233294Sstas * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 30233294Sstas * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 31233294Sstas * SUCH DAMAGE. 32178825Sdfr */ 33178825Sdfr 34178825Sdfr#include <config.h> 35178825Sdfr 36178825Sdfr#include <stdarg.h> 37178825Sdfr#include <stdlib.h> 38178825Sdfr#include "roken.h" 39178825Sdfr 40178825Sdfrstruct rk_strpool { 41178825Sdfr char *str; 42178825Sdfr size_t len; 43178825Sdfr}; 44178825Sdfr 45178825Sdfr/* 46178825Sdfr * 47178825Sdfr */ 48178825Sdfr 49233294SstasROKEN_LIB_FUNCTION void ROKEN_LIB_CALL 50178825Sdfrrk_strpoolfree(struct rk_strpool *p) 51178825Sdfr{ 52178825Sdfr if (p->str) { 53178825Sdfr free(p->str); 54178825Sdfr p->str = NULL; 55178825Sdfr } 56178825Sdfr free(p); 57178825Sdfr} 58178825Sdfr 59178825Sdfr/* 60178825Sdfr * 61178825Sdfr */ 62178825Sdfr 63233294SstasROKEN_LIB_FUNCTION struct rk_strpool * ROKEN_LIB_CALL 64178825Sdfrrk_strpoolprintf(struct rk_strpool *p, const char *fmt, ...) 65178825Sdfr{ 66178825Sdfr va_list ap; 67178825Sdfr char *str, *str2; 68178825Sdfr int len; 69178825Sdfr 70178825Sdfr if (p == NULL) { 71178825Sdfr p = malloc(sizeof(*p)); 72178825Sdfr if (p == NULL) 73178825Sdfr return NULL; 74178825Sdfr p->str = NULL; 75178825Sdfr p->len = 0; 76178825Sdfr } 77178825Sdfr va_start(ap, fmt); 78178825Sdfr len = vasprintf(&str, fmt, ap); 79178825Sdfr va_end(ap); 80178825Sdfr if (str == NULL) { 81178825Sdfr rk_strpoolfree(p); 82178825Sdfr return NULL; 83178825Sdfr } 84178825Sdfr str2 = realloc(p->str, len + p->len + 1); 85178825Sdfr if (str2 == NULL) { 86178825Sdfr rk_strpoolfree(p); 87178825Sdfr return NULL; 88178825Sdfr } 89178825Sdfr p->str = str2; 90178825Sdfr memcpy(p->str + p->len, str, len + 1); 91178825Sdfr p->len += len; 92178825Sdfr free(str); 93178825Sdfr return p; 94178825Sdfr} 95178825Sdfr 96178825Sdfr/* 97178825Sdfr * 98178825Sdfr */ 99178825Sdfr 100233294SstasROKEN_LIB_FUNCTION char * ROKEN_LIB_CALL 101178825Sdfrrk_strpoolcollect(struct rk_strpool *p) 102178825Sdfr{ 103233294Sstas char *str; 104233294Sstas if (p == NULL) 105233294Sstas return strdup(""); 106233294Sstas str = p->str; 107178825Sdfr p->str = NULL; 108178825Sdfr free(p); 109178825Sdfr return str; 110178825Sdfr} 111