nsz Git - libfirm/blob - ir/obstack/obstack_printf.c

   1 #include <stdarg.h>
   2 #include <stdio.h>
   3 #include <stdlib.h>
   4 #include <assert.h>
   5 #include "obstack.h"
   6
   7 #ifdef _WIN32
   8 #define vsnprintf _vsnprintf
   9 #endif
  10
  11 int obstack_vprintf(struct obstack *obst, const char *fmt, va_list ap)
  12 {
  13         char    buf[128];
  14         char   *buffer = buf;
  15         size_t  size   = sizeof(buf);
  16         int     len;
  17
  18         for (;;) {
  19                 va_list tap;
  20                 va_copy(tap, ap);
  21                 len = vsnprintf(buffer, size, fmt, tap);
  22
  23                 /* snprintf should return -1 only in the error case, but older glibcs
  24                  * and probably other systems are buggy in this respect and return -1 if
  25                  * the buffer was too small. We only abort for LARGE unrealistic buffer
  26                  * sizes here */
  27                 if (len < 0) {
  28                         if (buffer != buf)
  29                                 free(buffer);
  30                         if (size > 65536)
  31                                 return -1;
  32                         size *= 2;
  33                 } else if ((size_t)len >= size) {
  34                         /* If we come here more than once, vsnprintf() returned garbage */
  35                         assert(buffer == buf);
  36                         size = (size_t)len + 1;
  37                 } else {
  38                         break;
  39                 }
  40                 buffer = malloc(size);
  41         }
  42
  43         obstack_grow(obst, buffer, len);
  44         if (buffer != buf)
  45                 free(buffer);
  46
  47         return len;
  48 }
  49
  50 int obstack_printf(struct obstack *obst, const char *fmt, ...)
  51 {
  52         va_list ap;
  53         int     res;
  54
  55         va_start(ap, fmt);
  56         res = obstack_vprintf(obst, fmt, ap);
  57         va_end(ap);
  58
  59         return res;
  60 }