xref: /openbsd-src/lib/libc/stdio/asprintf.c (revision c5f4fad510dd427c0c20c0f4d164f60ce24651b6)
1 /*	$OpenBSD: asprintf.c,v 1.12 2005/04/30 09:25:17 espie Exp $	*/
2 
3 /*
4  * Copyright (c) 1997 Todd C. Miller <Todd.Miller@courtesan.com>
5  *
6  * Permission to use, copy, modify, and distribute this software for any
7  * purpose with or without fee is hereby granted, provided that the above
8  * copyright notice and this permission notice appear in all copies.
9  *
10  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17  */
18 
19 #if defined(LIBC_SCCS) && !defined(lint)
20 static char rcsid[] = "$OpenBSD: asprintf.c,v 1.12 2005/04/30 09:25:17 espie Exp $";
21 #endif /* LIBC_SCCS and not lint */
22 
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <errno.h>
26 #include <stdarg.h>
27 #include "local.h"
28 
29 int
30 asprintf(char **str, const char *fmt, ...)
31 {
32 	int ret;
33 	va_list ap;
34 	FILE f;
35 	struct __sfileext fext;
36 	unsigned char *_base;
37 
38 	_FILEEXT_SETUP(&f, &fext);
39 	va_start(ap, fmt);
40 	f._file = -1;
41 	f._flags = __SWR | __SSTR | __SALC;
42 	f._bf._base = f._p = (unsigned char *)malloc(128);
43 	if (f._bf._base == NULL)
44 		goto err;
45 	f._bf._size = f._w = 127;		/* Leave room for the NUL */
46 	ret = vfprintf(&f, fmt, ap);
47 	if (ret == -1)
48 		goto err;
49 	*f._p = '\0';
50 	_base = realloc(f._bf._base, ret + 1);
51 	if (_base == NULL)
52 		goto err;
53 	*str = (char *)_base;
54 	va_end(ap);
55 	return (ret);
56 
57 err:
58 	va_end(ap);
59 	if (f._bf._base) {
60 		free(f._bf._base);
61 		f._bf._base = NULL;
62 	}
63 	*str = NULL;
64 	errno = ENOMEM;
65 	return (-1);
66 }
67