1 /* $NetBSD: concat.c,v 1.1.1.2 2014/04/24 12:45:52 pettai Exp $ */
2
3 /*
4 * Copyright (c) 1995, 1996, 1997 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 <krb5/roken.h>
39
40 ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
roken_concat(char * s,size_t len,...)41 roken_concat (char *s, size_t len, ...)
42 {
43 int ret;
44 va_list args;
45
46 va_start(args, len);
47 ret = roken_vconcat (s, len, args);
48 va_end(args);
49 return ret;
50 }
51
52 ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
roken_vconcat(char * s,size_t len,va_list args)53 roken_vconcat (char *s, size_t len, va_list args)
54 {
55 const char *a;
56
57 while ((a = va_arg(args, const char*))) {
58 size_t n = strlen (a);
59
60 if (n >= len)
61 return -1;
62 memcpy (s, a, n);
63 s += n;
64 len -= n;
65 }
66 *s = '\0';
67 return 0;
68 }
69
70 ROKEN_LIB_FUNCTION size_t ROKEN_LIB_CALL
roken_vmconcat(char ** s,size_t max_len,va_list args)71 roken_vmconcat (char **s, size_t max_len, va_list args)
72 {
73 const char *a;
74 char *p, *q;
75 size_t len = 0;
76 *s = NULL;
77 p = malloc(1);
78 if(p == NULL)
79 return 0;
80 len = 1;
81 while ((a = va_arg(args, const char*))) {
82 size_t n = strlen (a);
83
84 if(max_len && len + n > max_len){
85 free(p);
86 return 0;
87 }
88 q = realloc(p, len + n);
89 if(q == NULL){
90 free(p);
91 return 0;
92 }
93 p = q;
94 memcpy (p + len - 1, a, n);
95 len += n;
96 }
97 p[len - 1] = '\0';
98 *s = p;
99 return len;
100 }
101
102 ROKEN_LIB_FUNCTION size_t ROKEN_LIB_CALL
roken_mconcat(char ** s,size_t max_len,...)103 roken_mconcat (char **s, size_t max_len, ...)
104 {
105 size_t ret;
106 va_list args;
107
108 va_start(args, max_len);
109 ret = roken_vmconcat (s, max_len, args);
110 va_end(args);
111 return ret;
112 }
113