1 /* 2 * Copyright (c) 2005 Kungliga Tekniska H�gskolan 3 * (Royal Institute of Technology, Stockholm, Sweden). 4 * All rights reserved. 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions 8 * are met: 9 * 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 13 * 2. Redistributions in binary form must reproduce the above copyright 14 * notice, this list of conditions and the following disclaimer in the 15 * documentation and/or other materials provided with the distribution. 16 * 17 * 3. Neither the name of the Institute nor the names of its contributors 18 * may be used to endorse or promote products derived from this software 19 * without specific prior written permission. 20 * 21 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND 22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 24 * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE 25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 31 * SUCH DAMAGE. 32 */ 33 34 #ifdef HAVE_CONFIG_H 35 #include <config.h> 36 RCSID("$Id: strpool.c 21005 2007-06-08 01:54:35Z lha $"); 37 #endif 38 39 #include <stdarg.h> 40 #include <stdlib.h> 41 #include "roken.h" 42 43 struct rk_strpool { 44 char *str; 45 size_t len; 46 }; 47 48 /* 49 * 50 */ 51 52 void ROKEN_LIB_FUNCTION 53 rk_strpoolfree(struct rk_strpool *p) 54 { 55 if (p->str) { 56 free(p->str); 57 p->str = NULL; 58 } 59 free(p); 60 } 61 62 /* 63 * 64 */ 65 66 struct rk_strpool * ROKEN_LIB_FUNCTION 67 rk_strpoolprintf(struct rk_strpool *p, const char *fmt, ...) 68 { 69 va_list ap; 70 char *str, *str2; 71 int len; 72 73 if (p == NULL) { 74 p = malloc(sizeof(*p)); 75 if (p == NULL) 76 return NULL; 77 p->str = NULL; 78 p->len = 0; 79 } 80 va_start(ap, fmt); 81 len = vasprintf(&str, fmt, ap); 82 va_end(ap); 83 if (str == NULL) { 84 rk_strpoolfree(p); 85 return NULL; 86 } 87 str2 = realloc(p->str, len + p->len + 1); 88 if (str2 == NULL) { 89 rk_strpoolfree(p); 90 return NULL; 91 } 92 p->str = str2; 93 memcpy(p->str + p->len, str, len + 1); 94 p->len += len; 95 free(str); 96 return p; 97 } 98 99 /* 100 * 101 */ 102 103 char * ROKEN_LIB_FUNCTION 104 rk_strpoolcollect(struct rk_strpool *p) 105 { 106 char *str = p->str; 107 p->str = NULL; 108 free(p); 109 return str; 110 } 111