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