diff options
Diffstat (limited to 'third_party/heimdal/lib/roken/strpool.c')
-rw-r--r-- | third_party/heimdal/lib/roken/strpool.c | 114 |
1 files changed, 114 insertions, 0 deletions
diff --git a/third_party/heimdal/lib/roken/strpool.c b/third_party/heimdal/lib/roken/strpool.c new file mode 100644 index 0000000..032bae1 --- /dev/null +++ b/third_party/heimdal/lib/roken/strpool.c @@ -0,0 +1,114 @@ +/* + * Copyright (c) 2005 Kungliga Tekniska Högskolan + * (Royal Institute of Technology, Stockholm, Sweden). + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * 3. Neither the name of the Institute nor the names of its contributors + * may be used to endorse or promote products derived from this software + * without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND + * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS + * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) + * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT + * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY + * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF + * SUCH DAMAGE. + */ + +#include <config.h> + +#include <stdarg.h> +#include <stdlib.h> +#include "roken.h" + +struct rk_strpool { + char *str; + size_t len; /* strlen() of str */ + size_t sz; /* Allocated size */ +}; + +/* + * + */ + +ROKEN_LIB_FUNCTION void ROKEN_LIB_CALL +rk_strpoolfree(struct rk_strpool *p) +{ + if (p && p->str) { + free(p->str); + p->str = NULL; + } + free(p); +} + +/* + * + */ + +ROKEN_LIB_FUNCTION struct rk_strpool * ROKEN_LIB_CALL +rk_strpoolprintf(struct rk_strpool *p, const char *fmt, ...) +{ + va_list ap; + char *str; + int len; + + va_start(ap, fmt); + len = vasprintf(&str, fmt, ap); + va_end(ap); + if (str == NULL) + return rk_strpoolfree(p), NULL; + + if (p == NULL) { + if ((p = malloc(sizeof(*p))) == NULL) + return free(str), NULL; + p->str = str; + p->len = p->sz = len; + return p; + } /* else grow the buffer and append `str', but don't grow too fast */ + + if (len + p->len + 1 > p->sz) { + size_t sz = p->len + len + 9 + (p->sz >> 2); + char *str2; + + if ((str2 = realloc(p->str, sz)) == NULL) + return rk_strpoolfree(p), NULL; + p->str = str2; + p->sz = sz; + } + memcpy(p->str + p->len, str, len + 1); + p->len += len; + free(str); + return p; +} + +/* + * + */ + +ROKEN_LIB_FUNCTION char * ROKEN_LIB_CALL +rk_strpoolcollect(struct rk_strpool *p) +{ + char *str; + if (p == NULL) + return strdup(""); + str = p->str; + p->str = NULL; + free(p); + return str; +} |