2007-08-10 H.J. Lu <hongjiu.lu@intel.com>
[binutils.git] / libiberty / vasprintf.c
blobb6cb94e7a22ddef6472116afc754ba5bad94de54
1 /* Like vsprintf but provides a pointer to malloc'd storage, which must
2 be freed by the caller.
3 Copyright (C) 1994, 2003 Free Software Foundation, Inc.
5 This file is part of the libiberty library.
6 Libiberty is free software; you can redistribute it and/or
7 modify it under the terms of the GNU Library General Public
8 License as published by the Free Software Foundation; either
9 version 2 of the License, or (at your option) any later version.
11 Libiberty is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 Library General Public License for more details.
16 You should have received a copy of the GNU Library General Public
17 License along with libiberty; see the file COPYING.LIB. If
18 not, write to the Free Software Foundation, Inc., 51 Franklin Street - Fifth Floor,
19 Boston, MA 02110-1301, USA. */
21 #ifdef HAVE_CONFIG_H
22 #include "config.h"
23 #endif
24 #include <ansidecl.h>
25 #include <stdarg.h>
26 #if !defined (va_copy) && defined (__va_copy)
27 # define va_copy(d,s) __va_copy((d),(s))
28 #endif
29 #include <stdio.h>
30 #ifdef HAVE_STRING_H
31 #include <string.h>
32 #endif
33 #ifdef HAVE_STDLIB_H
34 #include <stdlib.h>
35 #else
36 extern unsigned long strtoul ();
37 extern PTR malloc ();
38 #endif
39 #include "libiberty.h"
41 #ifdef TEST
42 int global_total_width;
43 #endif
47 @deftypefn Extension int vasprintf (char **@var{resptr}, const char *@var{format}, va_list @var{args})
49 Like @code{vsprintf}, but instead of passing a pointer to a buffer,
50 you pass a pointer to a pointer. This function will compute the size
51 of the buffer needed, allocate memory with @code{malloc}, and store a
52 pointer to the allocated memory in @code{*@var{resptr}}. The value
53 returned is the same as @code{vsprintf} would return. If memory could
54 not be allocated, minus one is returned and @code{NULL} is stored in
55 @code{*@var{resptr}}.
57 @end deftypefn
61 static int int_vasprintf (char **, const char *, va_list);
63 static int
64 int_vasprintf (char **result, const char *format, va_list args)
66 const char *p = format;
67 /* Add one to make sure that it is never zero, which might cause malloc
68 to return NULL. */
69 int total_width = strlen (format) + 1;
70 va_list ap;
72 #ifdef va_copy
73 va_copy (ap, args);
74 #else
75 memcpy ((PTR) &ap, (PTR) &args, sizeof (va_list));
76 #endif
78 while (*p != '\0')
80 if (*p++ == '%')
82 while (strchr ("-+ #0", *p))
83 ++p;
84 if (*p == '*')
86 ++p;
87 total_width += abs (va_arg (ap, int));
89 else
90 total_width += strtoul (p, (char **) &p, 10);
91 if (*p == '.')
93 ++p;
94 if (*p == '*')
96 ++p;
97 total_width += abs (va_arg (ap, int));
99 else
100 total_width += strtoul (p, (char **) &p, 10);
102 while (strchr ("hlL", *p))
103 ++p;
104 /* Should be big enough for any format specifier except %s and floats. */
105 total_width += 30;
106 switch (*p)
108 case 'd':
109 case 'i':
110 case 'o':
111 case 'u':
112 case 'x':
113 case 'X':
114 case 'c':
115 (void) va_arg (ap, int);
116 break;
117 case 'f':
118 case 'e':
119 case 'E':
120 case 'g':
121 case 'G':
122 (void) va_arg (ap, double);
123 /* Since an ieee double can have an exponent of 307, we'll
124 make the buffer wide enough to cover the gross case. */
125 total_width += 307;
126 break;
127 case 's':
128 total_width += strlen (va_arg (ap, char *));
129 break;
130 case 'p':
131 case 'n':
132 (void) va_arg (ap, char *);
133 break;
135 p++;
138 #ifdef va_copy
139 va_end (ap);
140 #endif
141 #ifdef TEST
142 global_total_width = total_width;
143 #endif
144 *result = (char *) malloc (total_width);
145 if (*result != NULL)
146 return vsprintf (*result, format, args);
147 else
148 return -1;
152 vasprintf (char **result, const char *format,
153 #if defined (_BSD_VA_LIST_) && defined (__FreeBSD__)
154 _BSD_VA_LIST_ args)
155 #else
156 va_list args)
157 #endif
159 return int_vasprintf (result, format, args);
162 #ifdef TEST
163 static void ATTRIBUTE_PRINTF_1
164 checkit (const char *format, ...)
166 char *result;
167 VA_OPEN (args, format);
168 VA_FIXEDARG (args, const char *, format);
169 vasprintf (&result, format, args);
170 VA_CLOSE (args);
172 if (strlen (result) < (size_t) global_total_width)
173 printf ("PASS: ");
174 else
175 printf ("FAIL: ");
176 printf ("%d %s\n", global_total_width, result);
178 free (result);
181 extern int main (void);
184 main (void)
186 checkit ("%d", 0x12345678);
187 checkit ("%200d", 5);
188 checkit ("%.300d", 6);
189 checkit ("%100.150d", 7);
190 checkit ("%s", "jjjjjjjjjiiiiiiiiiiiiiiioooooooooooooooooppppppppppppaa\n\
191 777777777777777777333333333333366666666666622222222222777777777777733333");
192 checkit ("%f%s%d%s", 1.0, "foo", 77, "asdjffffffffffffffiiiiiiiiiiixxxxx");
194 return 0;
196 #endif /* TEST */