]>
Commit | Line | Data |
---|---|---|
1 | /* $OpenBSD: asprintf.c,v 1.8 2002/02/19 19:39:36 millert Exp $ */ | |
2 | ||
3 | /* | |
4 | * Copyright (c) 1997 Todd C. Miller <Todd.Miller@courtesan.com> | |
5 | * All rights reserved. | |
6 | * | |
7 | * Redistribution and use in source and binary forms, with or without | |
8 | * modification, are permitted provided that the following conditions | |
9 | * are met: | |
10 | * 1. Redistributions of source code must retain the above copyright | |
11 | * notice, this list of conditions and the following disclaimer. | |
12 | * 2. Redistributions in binary form must reproduce the above copyright | |
13 | * notice, this list of conditions and the following disclaimer in the | |
14 | * documentation and/or other materials provided with the distribution. | |
15 | * 3. The name of the author may not be used to endorse or promote products | |
16 | * derived from this software without specific prior written permission. | |
17 | * | |
18 | * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, | |
19 | * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY | |
20 | * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL | |
21 | * THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, | |
22 | * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, | |
23 | * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; | |
24 | * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, | |
25 | * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR | |
26 | * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF | |
27 | * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
28 | */ | |
29 | ||
30 | #include <sys/cdefs.h> | |
31 | __FBSDID("$FreeBSD: src/lib/libc/stdio/asprintf.c,v 1.13 2002/09/26 13:09:48 tjr Exp $"); | |
32 | ||
33 | #include "xlocale_private.h" | |
34 | ||
35 | #include <stdio.h> | |
36 | #include <stdlib.h> | |
37 | #include <errno.h> | |
38 | #include <stdarg.h> | |
39 | ||
40 | #include "local.h" | |
41 | ||
42 | int | |
43 | asprintf(char **str, char const *fmt, ...) | |
44 | { | |
45 | int ret; | |
46 | va_list ap; | |
47 | FILE f; | |
48 | struct __sFILEX ext; | |
49 | ||
50 | f._file = -1; | |
51 | f._flags = __SWR | __SSTR | __SALC; | |
52 | f._bf._base = f._p = (unsigned char *)malloc(128); | |
53 | if (f._bf._base == NULL) { | |
54 | *str = NULL; | |
55 | errno = ENOMEM; | |
56 | return (-1); | |
57 | } | |
58 | f._bf._size = f._w = 127; /* Leave room for the NUL */ | |
59 | f._extra = &ext; | |
60 | INITEXTRA(&f); | |
61 | va_start(ap, fmt); | |
62 | ret = __vfprintf(&f, __current_locale(), fmt, ap); /* Use unlocked __vfprintf */ | |
63 | va_end(ap); | |
64 | if (ret < 0) { | |
65 | free(f._bf._base); | |
66 | *str = NULL; | |
67 | errno = ENOMEM; | |
68 | return (-1); | |
69 | } | |
70 | *f._p = '\0'; | |
71 | *str = (char *)f._bf._base; | |
72 | return (ret); | |
73 | } | |
74 | ||
75 | int | |
76 | asprintf_l(char **str, locale_t loc, char const *fmt, ...) | |
77 | { | |
78 | int ret; | |
79 | va_list ap; | |
80 | FILE f; | |
81 | struct __sFILEX ext; | |
82 | ||
83 | NORMALIZE_LOCALE(loc); | |
84 | f._file = -1; | |
85 | f._flags = __SWR | __SSTR | __SALC; | |
86 | f._bf._base = f._p = (unsigned char *)malloc(128); | |
87 | if (f._bf._base == NULL) { | |
88 | *str = NULL; | |
89 | errno = ENOMEM; | |
90 | return (-1); | |
91 | } | |
92 | f._bf._size = f._w = 127; /* Leave room for the NUL */ | |
93 | f._extra = &ext; | |
94 | INITEXTRA(&f); | |
95 | va_start(ap, fmt); | |
96 | ret = __vfprintf(&f, loc, fmt, ap); /* Use unlocked __vfprintf */ | |
97 | va_end(ap); | |
98 | if (ret < 0) { | |
99 | free(f._bf._base); | |
100 | *str = NULL; | |
101 | errno = ENOMEM; | |
102 | return (-1); | |
103 | } | |
104 | *f._p = '\0'; | |
105 | *str = (char *)f._bf._base; | |
106 | return (ret); | |
107 | } |