1*90685d2cSjp161948 /*
2*90685d2cSjp161948 * Copyright (c) 2004 Darren Tucker.
3*90685d2cSjp161948 *
4*90685d2cSjp161948 * Based originally on asprintf.c from OpenBSD:
5*90685d2cSjp161948 * Copyright (c) 1997 Todd C. Miller <Todd.Miller@courtesan.com>
6*90685d2cSjp161948 *
7*90685d2cSjp161948 * Permission to use, copy, modify, and distribute this software for any
8*90685d2cSjp161948 * purpose with or without fee is hereby granted, provided that the above
9*90685d2cSjp161948 * copyright notice and this permission notice appear in all copies.
10*90685d2cSjp161948 *
11*90685d2cSjp161948 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
12*90685d2cSjp161948 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
13*90685d2cSjp161948 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
14*90685d2cSjp161948 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
15*90685d2cSjp161948 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
16*90685d2cSjp161948 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
17*90685d2cSjp161948 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18*90685d2cSjp161948 */
19*90685d2cSjp161948
20*90685d2cSjp161948 #pragma ident "%Z%%M% %I% %E% SMI"
21*90685d2cSjp161948
22*90685d2cSjp161948 #include "includes.h"
23*90685d2cSjp161948
24*90685d2cSjp161948 #ifndef HAVE_VASPRINTF
25*90685d2cSjp161948
26*90685d2cSjp161948 #include <errno.h>
27*90685d2cSjp161948 #include <stdarg.h>
28*90685d2cSjp161948 #include <stdlib.h>
29*90685d2cSjp161948
30*90685d2cSjp161948 #ifndef VA_COPY
31*90685d2cSjp161948 #ifdef HAVE_VA_COPY
32*90685d2cSjp161948 #define VA_COPY(dest, src) va_copy(dest, src)
33*90685d2cSjp161948 #else
34*90685d2cSjp161948 #ifdef HAVE___VA_COPY
35*90685d2cSjp161948 #define VA_COPY(dest, src) __va_copy(dest, src)
36*90685d2cSjp161948 #else
37*90685d2cSjp161948 #define VA_COPY(dest, src) (dest) = (src)
38*90685d2cSjp161948 #endif
39*90685d2cSjp161948 #endif
40*90685d2cSjp161948 #endif
41*90685d2cSjp161948
42*90685d2cSjp161948 #define INIT_SZ 128
43*90685d2cSjp161948
44*90685d2cSjp161948 int
vasprintf(char ** str,const char * fmt,va_list ap)45*90685d2cSjp161948 vasprintf(char **str, const char *fmt, va_list ap)
46*90685d2cSjp161948 {
47*90685d2cSjp161948 int ret = -1;
48*90685d2cSjp161948 va_list ap2;
49*90685d2cSjp161948 char *string, *newstr;
50*90685d2cSjp161948 size_t len;
51*90685d2cSjp161948
52*90685d2cSjp161948 VA_COPY(ap2, ap);
53*90685d2cSjp161948 if ((string = malloc(INIT_SZ)) == NULL)
54*90685d2cSjp161948 goto fail;
55*90685d2cSjp161948
56*90685d2cSjp161948 ret = vsnprintf(string, INIT_SZ, fmt, ap2);
57*90685d2cSjp161948 if (ret >= 0 && ret < INIT_SZ) { /* succeeded with initial alloc */
58*90685d2cSjp161948 *str = string;
59*90685d2cSjp161948 } else if (ret == INT_MAX || ret < 0) { /* Bad length */
60*90685d2cSjp161948 free(string);
61*90685d2cSjp161948 goto fail;
62*90685d2cSjp161948 } else { /* bigger than initial, realloc allowing for nul */
63*90685d2cSjp161948 len = (size_t)ret + 1;
64*90685d2cSjp161948 if ((newstr = realloc(string, len)) == NULL) {
65*90685d2cSjp161948 free(string);
66*90685d2cSjp161948 goto fail;
67*90685d2cSjp161948 } else {
68*90685d2cSjp161948 va_end(ap2);
69*90685d2cSjp161948 VA_COPY(ap2, ap);
70*90685d2cSjp161948 ret = vsnprintf(newstr, len, fmt, ap2);
71*90685d2cSjp161948 if (ret >= 0 && (size_t)ret < len) {
72*90685d2cSjp161948 *str = newstr;
73*90685d2cSjp161948 } else { /* failed with realloc'ed string, give up */
74*90685d2cSjp161948 free(newstr);
75*90685d2cSjp161948 goto fail;
76*90685d2cSjp161948 }
77*90685d2cSjp161948 }
78*90685d2cSjp161948 }
79*90685d2cSjp161948 va_end(ap2);
80*90685d2cSjp161948 return (ret);
81*90685d2cSjp161948
82*90685d2cSjp161948 fail:
83*90685d2cSjp161948 *str = NULL;
84*90685d2cSjp161948 errno = ENOMEM;
85*90685d2cSjp161948 va_end(ap2);
86*90685d2cSjp161948 return (-1);
87*90685d2cSjp161948 }
88*90685d2cSjp161948 #endif
89*90685d2cSjp161948
90*90685d2cSjp161948 #ifndef HAVE_ASPRINTF
91*90685d2cSjp161948 int
asprintf(char ** str,const char * fmt,...)92*90685d2cSjp161948 asprintf(char **str, const char *fmt, ...)
93*90685d2cSjp161948 {
94*90685d2cSjp161948 va_list ap;
95*90685d2cSjp161948 int ret;
96*90685d2cSjp161948
97*90685d2cSjp161948 *str = NULL;
98*90685d2cSjp161948 va_start(ap, fmt);
99*90685d2cSjp161948 ret = vasprintf(str, fmt, ap);
100*90685d2cSjp161948 va_end(ap);
101*90685d2cSjp161948
102*90685d2cSjp161948 return (ret);
103*90685d2cSjp161948 }
104*90685d2cSjp161948 #endif
105