1d0983205SRoger A. Faulkner /*
2d0983205SRoger A. Faulkner * Copyright 2009 Sun Microsystems, Inc. All rights reserved.
3d0983205SRoger A. Faulkner * Use is subject to license terms.
4d0983205SRoger A. Faulkner */
5d0983205SRoger A. Faulkner
6d0983205SRoger A. Faulkner /*
7d0983205SRoger A. Faulkner * Copyright (c) 2004 Darren Tucker.
8*4322dd90SRobert Mustacchi * Copyright 2022 Oxide Computer Company
9d0983205SRoger A. Faulkner *
10d0983205SRoger A. Faulkner * Based originally on asprintf.c from OpenBSD:
11d0983205SRoger A. Faulkner * Copyright (c) 1997 Todd C. Miller <Todd.Miller@courtesan.com>
12d0983205SRoger A. Faulkner *
13d0983205SRoger A. Faulkner * Permission to use, copy, modify, and distribute this software for any
14d0983205SRoger A. Faulkner * purpose with or without fee is hereby granted, provided that the above
15d0983205SRoger A. Faulkner * copyright notice and this permission notice appear in all copies.
16d0983205SRoger A. Faulkner *
17d0983205SRoger A. Faulkner * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
18d0983205SRoger A. Faulkner * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
19d0983205SRoger A. Faulkner * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
20d0983205SRoger A. Faulkner * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
21d0983205SRoger A. Faulkner * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
22d0983205SRoger A. Faulkner * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
23d0983205SRoger A. Faulkner * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
24d0983205SRoger A. Faulkner */
25d0983205SRoger A. Faulkner
26d0983205SRoger A. Faulkner #include <lint.h>
27d0983205SRoger A. Faulkner #include <stdio.h>
28d0983205SRoger A. Faulkner #include <stdarg.h>
29d0983205SRoger A. Faulkner #include <stdlib.h>
30d0983205SRoger A. Faulkner #include <string.h>
31d0983205SRoger A. Faulkner #include <limits.h>
32d0983205SRoger A. Faulkner #include <errno.h>
33d0983205SRoger A. Faulkner
34d0983205SRoger A. Faulkner #define INIT_SZ 128
35d0983205SRoger A. Faulkner
36d0983205SRoger A. Faulkner int
vasprintf(char ** str,const char * format,va_list ap)37d0983205SRoger A. Faulkner vasprintf(char **str, const char *format, va_list ap)
38d0983205SRoger A. Faulkner {
39d0983205SRoger A. Faulkner char string[INIT_SZ];
40d0983205SRoger A. Faulkner char *newstr;
41d0983205SRoger A. Faulkner int ret;
42d0983205SRoger A. Faulkner size_t len;
43d0983205SRoger A. Faulkner
44d0983205SRoger A. Faulkner *str = NULL;
45d0983205SRoger A. Faulkner ret = vsnprintf(string, INIT_SZ, format, ap);
46d0983205SRoger A. Faulkner if (ret < 0) /* retain the value of errno from vsnprintf() */
47d0983205SRoger A. Faulkner return (-1);
48d0983205SRoger A. Faulkner if (ret < INIT_SZ) {
49d0983205SRoger A. Faulkner len = ret + 1;
50d0983205SRoger A. Faulkner if ((newstr = malloc(len)) == NULL)
51d0983205SRoger A. Faulkner return (-1); /* retain errno from malloc() */
52*4322dd90SRobert Mustacchi /*
53*4322dd90SRobert Mustacchi * Prior versions of this used strlcpy. This has two problems.
54*4322dd90SRobert Mustacchi * One, it doesn't handle embedded '\0' characters. Secondly,
55*4322dd90SRobert Mustacchi * it's recalculating the length we already know. Please do not
56*4322dd90SRobert Mustacchi * use a string-based copying function.
57*4322dd90SRobert Mustacchi */
58*4322dd90SRobert Mustacchi (void) memcpy(newstr, string, len);
59d0983205SRoger A. Faulkner *str = newstr;
60d0983205SRoger A. Faulkner return (ret);
61d0983205SRoger A. Faulkner }
62d0983205SRoger A. Faulkner /*
63d0983205SRoger A. Faulkner * We will perform this loop more than once only if some other
64d0983205SRoger A. Faulkner * thread modifies one of the vasprintf() arguments after our
65d0983205SRoger A. Faulkner * previous call to vsnprintf().
66d0983205SRoger A. Faulkner */
67d0983205SRoger A. Faulkner for (;;) {
68d0983205SRoger A. Faulkner if (ret == INT_MAX) { /* Bad length */
69d0983205SRoger A. Faulkner errno = ENOMEM;
70d0983205SRoger A. Faulkner return (-1);
71d0983205SRoger A. Faulkner }
72d0983205SRoger A. Faulkner len = ret + 1;
73d0983205SRoger A. Faulkner if ((newstr = malloc(len)) == NULL)
74d0983205SRoger A. Faulkner return (-1); /* retain errno from malloc() */
75d0983205SRoger A. Faulkner ret = vsnprintf(newstr, len, format, ap);
76d0983205SRoger A. Faulkner if (ret < 0) { /* retain errno from vsnprintf() */
77d0983205SRoger A. Faulkner free(newstr);
78d0983205SRoger A. Faulkner return (-1);
79d0983205SRoger A. Faulkner }
80d0983205SRoger A. Faulkner if (ret < len) {
81d0983205SRoger A. Faulkner *str = newstr;
82d0983205SRoger A. Faulkner return (ret);
83d0983205SRoger A. Faulkner }
84d0983205SRoger A. Faulkner free(newstr);
85d0983205SRoger A. Faulkner }
86d0983205SRoger A. Faulkner }
87d0983205SRoger A. Faulkner
88d0983205SRoger A. Faulkner int
asprintf(char ** str,const char * format,...)89d0983205SRoger A. Faulkner asprintf(char **str, const char *format, ...)
90d0983205SRoger A. Faulkner {
91d0983205SRoger A. Faulkner va_list ap;
92d0983205SRoger A. Faulkner int ret;
93d0983205SRoger A. Faulkner
94d0983205SRoger A. Faulkner *str = NULL;
95d0983205SRoger A. Faulkner va_start(ap, format);
96d0983205SRoger A. Faulkner ret = vasprintf(str, format, ap);
97d0983205SRoger A. Faulkner va_end(ap);
98d0983205SRoger A. Faulkner
99d0983205SRoger A. Faulkner return (ret);
100d0983205SRoger A. Faulkner }
101