vasprintf.c
上传用户:qaz666999
上传日期:2022-08-06
资源大小:2570k
文件大小:3k
源码类别:

数学计算

开发平台:

Unix_Linux

  1. /* gmp_vasprintf -- formatted output to an allocated space.
  2. Copyright 2001 Free Software Foundation, Inc.
  3. This file is part of the GNU MP Library.
  4. The GNU MP Library is free software; you can redistribute it and/or modify
  5. it under the terms of the GNU Lesser General Public License as published by
  6. the Free Software Foundation; either version 3 of the License, or (at your
  7. option) any later version.
  8. The GNU MP Library is distributed in the hope that it will be useful, but
  9. WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  10. or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
  11. License for more details.
  12. You should have received a copy of the GNU Lesser General Public License
  13. along with the GNU MP Library.  If not, see http://www.gnu.org/licenses/.  */
  14. #include "config.h"
  15. #if HAVE_STDARG
  16. #include <stdarg.h>
  17. #else
  18. #include <varargs.h>
  19. #endif
  20. #include <stdio.h>
  21. #include <stdlib.h>
  22. #include <string.h>
  23. #include "gmp.h"
  24. #include "gmp-impl.h"
  25. #if ! HAVE_VSNPRINTF
  26. #define vsnprintf  __gmp_replacement_vsnprintf
  27. #endif
  28. /* vasprintf isn't used since we prefer all GMP allocs to go through
  29.    __gmp_allocate_func, and in particular we don't want the -1 return from
  30.    vasprintf for out-of-memory, instead __gmp_allocate_func should handle
  31.    that.  Using vsnprintf unfortunately means we might have to re-run it if
  32.    our current space is insufficient.
  33.    The initial guess for the needed space is an arbitrary 256 bytes.  If
  34.    that (and any extra GMP_ASPRINTF_T_NEED might give) isn't enough then an
  35.    ISO C99 standard vsnprintf will tell us what we really need.
  36.    GLIBC 2.0.x vsnprintf returns either -1 or space-1 to indicate overflow,
  37.    without giving any indication how much is really needed.  In this case
  38.    keep trying with double the space each time.
  39.    A return of space-1 is success on a C99 vsnprintf, but we're not
  40.    bothering to identify which style vsnprintf we've got, so just take the
  41.    pessimistic option and assume it's glibc 2.0.x.
  42.    Notice the use of ret+2 for the new space in the C99 case.  This ensures
  43.    the next vsnprintf return value will be space-2, which is unambiguously
  44.    successful.  But actually GMP_ASPRINTF_T_NEED() will realloc to even
  45.    bigger than that ret+2.
  46.    vsnprintf might trash it's given ap, so copy it in case we need to use it
  47.    more than once.  See comments with gmp_snprintf_format.  */
  48. static int
  49. gmp_asprintf_format (struct gmp_asprintf_t *d, const char *fmt,
  50.                      va_list orig_ap)
  51. {
  52.   int      ret;
  53.   va_list  ap;
  54.   size_t   space = 256;
  55.   for (;;)
  56.     {
  57.       GMP_ASPRINTF_T_NEED (d, space);
  58.       space = d->alloc - d->size;
  59.       va_copy (ap, orig_ap);
  60.       ret = vsnprintf (d->buf + d->size, space, fmt, ap);
  61.       if (ret == -1)
  62.         {
  63.           ASSERT (strlen (d->buf + d->size) == space-1);
  64.           ret = space-1;
  65.         }
  66.       /* done if output fits in our space */
  67.       if (ret < space-1)
  68.         break;
  69.       if (ret == space-1)
  70.         space *= 2;     /* possible glibc 2.0.x, so double */
  71.       else
  72.         space = ret+2;  /* C99, so now know space required */
  73.     }
  74.   d->size += ret;
  75.   return ret;
  76. }
  77. const struct doprnt_funs_t  __gmp_asprintf_funs = {
  78.   (doprnt_format_t) gmp_asprintf_format,
  79.   (doprnt_memory_t) __gmp_asprintf_memory,
  80.   (doprnt_reps_t)   __gmp_asprintf_reps,
  81.   (doprnt_final_t)  __gmp_asprintf_final
  82. };
  83. int
  84. gmp_vasprintf (char **result, const char *fmt, va_list ap)
  85. {
  86.   struct gmp_asprintf_t  d;
  87.   GMP_ASPRINTF_T_INIT (d, result);
  88.   return __gmp_doprnt (&__gmp_asprintf_funs, &d, fmt, ap);
  89. }