1/*	$NetBSD: concat.c,v 1.2 2017/01/28 21:31:50 christos 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
40ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
41roken_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
52ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
53roken_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
70ROKEN_LIB_FUNCTION size_t ROKEN_LIB_CALL
71roken_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
102ROKEN_LIB_FUNCTION size_t ROKEN_LIB_CALL
103roken_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