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