vasprintf.c revision 77298 
1/* Like vsprintf but provides a pointer to malloc'd storage, which must
2   be freed by the caller.
3   Copyright (C) 1994 Free Software Foundation, Inc.
4
5This file is part of the libiberty library.
6Libiberty is free software; you can redistribute it and/or
7modify it under the terms of the GNU Library General Public
8License as published by the Free Software Foundation; either
9version 2 of the License, or (at your option) any later version.
10
11Libiberty is distributed in the hope that it will be useful,
12but WITHOUT ANY WARRANTY; without even the implied warranty of
13MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14Library General Public License for more details.
15
16You should have received a copy of the GNU Library General Public
17License along with libiberty; see the file COPYING.LIB.  If
18not, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
19Boston, MA 02111-1307, USA.  */
20
21#ifdef HAVE_CONFIG_H
22#include "config.h"
23#endif
24#include <ansidecl.h>
25#ifdef ANSI_PROTOTYPES
26#include <stdarg.h>
27#else
28#include <varargs.h>
29#endif
30#include <stdio.h>
31#include <string.h>
32#ifdef HAVE_STDLIB_H
33#include <stdlib.h>
34#else
35extern unsigned long strtoul ();
36extern PTR malloc ();
37#endif
38#include "libiberty.h"
39
40#ifdef TEST
41int global_total_width;
42#endif
43
44
45static int int_vasprintf PARAMS ((char **, const char *, va_list *));
46
47static int
48int_vasprintf (result, format, args)
49     char **result;
50     const char *format;
51     va_list *args;
52{
53  const char *p = format;
54  /* Add one to make sure that it is never zero, which might cause malloc
55     to return NULL.  */
56  int total_width = strlen (format) + 1;
57  va_list ap;
58
59  memcpy ((PTR) &ap, (PTR) args, sizeof (va_list));
60
61  while (*p != '\0')
62    {
63      if (*p++ == '%')
64	{
65	  while (strchr ("-+ #0", *p))
66	    ++p;
67	  if (*p == '*')
68	    {
69	      ++p;
70	      total_width += abs (va_arg (ap, int));
71	    }
72	  else
73	    total_width += strtoul (p, (char **) &p, 10);
74	  if (*p == '.')
75	    {
76	      ++p;
77	      if (*p == '*')
78		{
79		  ++p;
80		  total_width += abs (va_arg (ap, int));
81		}
82	      else
83	      total_width += strtoul (p, (char **) &p, 10);
84	    }
85	  while (strchr ("hlL", *p))
86	    ++p;
87	  /* Should be big enough for any format specifier except %s and floats.  */
88	  total_width += 30;
89	  switch (*p)
90	    {
91	    case 'd':
92	    case 'i':
93	    case 'o':
94	    case 'u':
95	    case 'x':
96	    case 'X':
97	    case 'c':
98	      (void) va_arg (ap, int);
99	      break;
100	    case 'f':
101	    case 'e':
102	    case 'E':
103	    case 'g':
104	    case 'G':
105	      (void) va_arg (ap, double);
106	      /* Since an ieee double can have an exponent of 307, we'll
107		 make the buffer wide enough to cover the gross case. */
108	      total_width += 307;
109	      break;
110	    case 's':
111	      total_width += strlen (va_arg (ap, char *));
112	      break;
113	    case 'p':
114	    case 'n':
115	      (void) va_arg (ap, char *);
116	      break;
117	    }
118	  p++;
119	}
120    }
121#ifdef TEST
122  global_total_width = total_width;
123#endif
124  *result = malloc (total_width);
125  if (*result != NULL)
126    return vsprintf (*result, format, *args);
127  else
128    return 0;
129}
130
131int
132vasprintf (result, format, args)
133     char **result;
134     const char *format;
135#if defined (_BSD_VA_LIST_) && defined (__FreeBSD__)
136     _BSD_VA_LIST_ args;
137#else
138     va_list args;
139#endif
140{
141  return int_vasprintf (result, format, &args);
142}
143
144#ifdef TEST
145static void checkit PARAMS ((const char *, ...));
146
147static void
148checkit VPARAMS ((const char* format, ...))
149{
150  va_list args;
151  char *result;
152#ifndef ANSI_PROTOTYPES
153  const char *format;
154#endif
155
156  VA_START (args, format);
157
158#ifndef ANSI_PROTOTYPES
159  format = va_arg (args, const char *);
160#endif
161
162  vasprintf (&result, format, args);
163  if (strlen (result) < (size_t) global_total_width)
164    printf ("PASS: ");
165  else
166    printf ("FAIL: ");
167  printf ("%d %s\n", global_total_width, result);
168}
169
170extern int main PARAMS ((void));
171
172int
173main ()
174{
175  checkit ("%d", 0x12345678);
176  checkit ("%200d", 5);
177  checkit ("%.300d", 6);
178  checkit ("%100.150d", 7);
179  checkit ("%s", "jjjjjjjjjiiiiiiiiiiiiiiioooooooooooooooooppppppppppppaa\n\
180777777777777777777333333333333366666666666622222222222777777777777733333");
181  checkit ("%f%s%d%s", 1.0, "foo", 77, "asdjffffffffffffffiiiiiiiiiiixxxxx");
182
183  return 0;
184}
185#endif /* TEST */
186