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. 4 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. 10 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. 15 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. */ 20 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" 40 41 #ifdef TEST 42 int global_total_width; 43 #endif 44 45 /* 46 47 @deftypefn Extension int vasprintf (char **@var{resptr}, const char *@var{format}, va_list @var{args}) 48 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}}. 56 57 @end deftypefn 58 59 */ 60 61 static int int_vasprintf (char **, const char *, va_list); 62 63 static int 64 int_vasprintf (char **result, const char *format, va_list args) 65 { 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; 71 72 #ifdef va_copy 73 va_copy (ap, args); 74 #else 75 memcpy ((PTR) &ap, (PTR) &args, sizeof (va_list)); 76 #endif 77 78 while (*p != '\0') 79 { 80 if (*p++ == '%') 81 { 82 while (strchr ("-+ #0", *p)) 83 ++p; 84 if (*p == '*') 85 { 86 ++p; 87 total_width += abs (va_arg (ap, int)); 88 } 89 else 90 total_width += strtoul (p, (char **) &p, 10); 91 if (*p == '.') 92 { 93 ++p; 94 if (*p == '*') 95 { 96 ++p; 97 total_width += abs (va_arg (ap, int)); 98 } 99 else 100 total_width += strtoul (p, (char **) &p, 10); 101 } 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) 107 { 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; 134 } 135 p++; 136 } 137 } 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; 149 } 150 151 int 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 158 { 159 return int_vasprintf (result, format, args); 160 } 161 162 #ifdef TEST 163 static void ATTRIBUTE_PRINTF_1 164 checkit (const char *format, ...) 165 { 166 char *result; 167 VA_OPEN (args, format); 168 VA_FIXEDARG (args, const char *, format); 169 vasprintf (&result, format, args); 170 VA_CLOSE (args); 171 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); 177 178 free (result); 179 } 180 181 extern int main (void); 182 183 int 184 main (void) 185 { 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"); 193 194 return 0; 195 } 196 #endif /* TEST */ 197