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

数学计算

开发平台:

Unix_Linux

  1. /* __gmp_replacement_vsnprintf -- for systems which don't have vsnprintf, or
  2.    only have a broken one.
  3.    THE FUNCTIONS IN THIS FILE ARE FOR INTERNAL USE ONLY.  THEY'RE ALMOST
  4.    CERTAIN TO BE SUBJECT TO INCOMPATIBLE CHANGES OR DISAPPEAR COMPLETELY IN
  5.    FUTURE GNU MP RELEASES.
  6. Copyright 2001, 2002 Free Software Foundation, Inc.
  7. This file is part of the GNU MP Library.
  8. The GNU MP Library is free software; you can redistribute it and/or modify
  9. it under the terms of the GNU Lesser General Public License as published by
  10. the Free Software Foundation; either version 3 of the License, or (at your
  11. option) any later version.
  12. The GNU MP Library is distributed in the hope that it will be useful, but
  13. WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  14. or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
  15. License for more details.
  16. You should have received a copy of the GNU Lesser General Public License
  17. along with the GNU MP Library.  If not, see http://www.gnu.org/licenses/.  */
  18. #include "config.h"
  19. #if ! HAVE_VSNPRINTF   /* only need this file if we don't have vsnprintf */
  20. #define _GNU_SOURCE    /* for strnlen prototype */
  21. #if HAVE_STDARG
  22. #include <stdarg.h>
  23. #else
  24. #include <varargs.h>
  25. #endif
  26. #include <ctype.h>     /* for isdigit */
  27. #include <stddef.h>    /* for ptrdiff_t */
  28. #include <string.h>
  29. #include <stdio.h>     /* for NULL */
  30. #include <stdlib.h>
  31. #if HAVE_FLOAT_H
  32. #include <float.h>     /* for DBL_MAX_10_EXP etc */
  33. #endif
  34. #if HAVE_INTTYPES_H
  35. # include <inttypes.h> /* for intmax_t */
  36. #else
  37. # if HAVE_STDINT_H
  38. #  include <stdint.h>
  39. # endif
  40. #endif
  41. #if HAVE_SYS_TYPES_H
  42. #include <sys/types.h> /* for quad_t */
  43. #endif
  44. #include "gmp.h"
  45. #include "gmp-impl.h"
  46. /* Autoconf notes that AIX 4.3 has a broken strnlen, but fortunately it
  47.    doesn't affect us since __gmp_replacement_vsnprintf is not required on
  48.    that system.  */
  49. #if ! HAVE_STRNLEN
  50. static size_t
  51. strnlen (const char *s, size_t n)
  52. {
  53.   size_t  i;
  54.   for (i = 0; i < n; i++)
  55.     if (s[i] == '')
  56.       break;
  57.   return i;
  58. }
  59. #endif
  60. /* The approach here is to parse the fmt string, and decide how much space
  61.    it requires, then use vsprintf into a big enough buffer.  The space
  62.    calculated isn't an exact amount, but it's certainly no less than
  63.    required.
  64.    This code was inspired by GNU libiberty/vasprintf.c but we support more
  65.    datatypes, when available.
  66.    mingw32 - doesn't have vsnprintf, it seems.  Because gcc is used a full
  67.        set of types are available, but "long double" is just a plain IEEE
  68.        64-bit "double" and LDBL_MAX_EXP_10 is correspondingly defined, so we
  69.        avoid the big 15-bit exponent estimate.  */
  70. int
  71. __gmp_replacement_vsnprintf (char *buf, size_t buf_size,
  72.      const char *orig_fmt, va_list orig_ap)
  73. {
  74.   va_list     ap;
  75.   const char  *fmt;
  76.   size_t      total_width, integer_sizeof, floating_sizeof, len;
  77.   char        fchar, type;
  78.   int         width, prec, seen_prec, double_digits, long_double_digits;
  79.   int         *value;
  80.   /* preserve orig_ap for use after size estimation */
  81.   va_copy (ap, orig_ap);
  82.   fmt = orig_fmt;
  83.   total_width = strlen (fmt) + 1;   /* 1 extra for the '' */
  84.   integer_sizeof = sizeof (long);
  85. #if HAVE_LONG_LONG
  86.   integer_sizeof = MAX (integer_sizeof, sizeof (long long));
  87. #endif
  88. #if HAVE_QUAD_T
  89.   integer_sizeof = MAX (integer_sizeof, sizeof (quad_t));
  90. #endif
  91.   floating_sizeof = sizeof (double);
  92. #if HAVE_LONG_DOUBLE
  93.   floating_sizeof = MAX (floating_sizeof, sizeof (long double));
  94. #endif
  95.   /* IEEE double or VAX G floats have an 11 bit exponent, so the default is
  96.      a maximum 308 decimal digits.  VAX D floats have only an 8 bit
  97.      exponent, but we don't bother trying to detect that directly.  */
  98.   double_digits = 308;
  99. #ifdef DBL_MAX_10_EXP
  100.   /* but in any case prefer a value the compiler says */
  101.   double_digits = DBL_MAX_10_EXP;
  102. #endif
  103.   /* IEEE 128-bit quad, Intel 80-bit temporary, or VAX H floats all have 15
  104.      bit exponents, so the default is a maximum 4932 decimal digits.  */
  105.   long_double_digits = 4932;
  106.   /* but if double == long double, then go with that size */
  107. #if HAVE_LONG_DOUBLE
  108.   if (sizeof (double) == sizeof (long double))
  109.     long_double_digits = double_digits;
  110. #endif
  111. #ifdef LDBL_MAX_10_EXP
  112.   /* but in any case prefer a value the compiler says */
  113.   long_double_digits = LDBL_MAX_10_EXP;
  114. #endif
  115.   for (;;)
  116.     {
  117.       fmt = strchr (fmt, '%');
  118.       if (fmt == NULL)
  119. break;
  120.       fmt++;
  121.       type = '';
  122.       width = 0;
  123.       prec = 6;
  124.       seen_prec = 0;
  125.       value = &width;
  126.       for (;;)
  127. {
  128.   fchar = *fmt++;
  129.   switch (fchar) {
  130.   case 'c':
  131.     /* char, already accounted for by strlen(fmt) */
  132.     goto next;
  133.   case 'd':
  134.   case 'i':
  135.   case 'o':
  136.   case 'x':
  137.   case 'X':
  138.   case 'u':
  139.     /* at most 3 digits per byte in hex, dec or octal, plus a sign */
  140.     total_width += 3 * integer_sizeof + 1;
  141.     switch (type) {
  142.     case 'j':
  143.       /* Let's assume uintmax_t is the same size as intmax_t. */
  144. #if HAVE_INTMAX_T
  145.       (void) va_arg (ap, intmax_t);
  146. #else
  147.       ASSERT_FAIL (intmax_t not available);
  148. #endif
  149.       break;
  150.     case 'l':
  151.       (void) va_arg (ap, long);
  152.       break;
  153.     case 'L':
  154. #if HAVE_LONG_LONG
  155.       (void) va_arg (ap, long long);
  156. #else
  157.       ASSERT_FAIL (long long not available);
  158. #endif
  159.       break;
  160.     case 'q':
  161.       /* quad_t is probably the same as long long, but let's treat
  162.  it separately just to be sure.  Also let's assume u_quad_t
  163.  will be the same size as quad_t.  */
  164. #if HAVE_QUAD_T
  165.       (void) va_arg (ap, quad_t);
  166. #else
  167.       ASSERT_FAIL (quad_t not available);
  168. #endif
  169.       break;
  170.     case 't':
  171. #if HAVE_PTRDIFF_T
  172.       (void) va_arg (ap, ptrdiff_t);
  173. #else
  174.       ASSERT_FAIL (ptrdiff_t not available);
  175. #endif
  176.       break;
  177.     case 'z':
  178.       (void) va_arg (ap, size_t);
  179.       break;
  180.     default:
  181.       /* default is an "int", and this includes h=short and hh=char
  182.  since they're promoted to int in a function call */
  183.       (void) va_arg (ap, int);
  184.       break;
  185.     }
  186.     goto next;
  187.   case 'E':
  188.   case 'e':
  189.   case 'G':
  190.   case 'g':
  191.     /* Requested decimals, sign, point and e, plus an overestimate
  192.        of exponent digits (the assumption is all the float is
  193.        exponent!).  */
  194.     total_width += prec + 3 + floating_sizeof * 3;
  195.     if (type == 'L')
  196.       {
  197. #if HAVE_LONG_DOUBLE
  198. (void) va_arg (ap, long double);
  199. #else
  200. ASSERT_FAIL (long double not available);
  201. #endif
  202.       }
  203.     else
  204.       (void) va_arg (ap, double);
  205.     break;
  206.   case 'f':
  207.     /* Requested decimals, sign and point, and a margin for error,
  208.        then add the maximum digits that can be in the integer part,
  209.        based on the maximum exponent value. */
  210.     total_width += prec + 2 + 10;
  211.     if (type == 'L')
  212.       {
  213. #if HAVE_LONG_DOUBLE
  214. (void) va_arg (ap, long double);
  215. total_width += long_double_digits;
  216. #else
  217. ASSERT_FAIL (long double not available);
  218. #endif
  219.       }
  220.     else
  221.       {
  222. (void) va_arg (ap, double);
  223. total_width += double_digits;
  224.       }
  225.     break;
  226.   case 'h':  /* short or char */
  227.   case 'j':  /* intmax_t */
  228.   case 'L':  /* long long or long double */
  229.   case 'q':  /* quad_t */
  230.   case 't':  /* ptrdiff_t */
  231.   set_type:
  232.     type = fchar;
  233.     break;
  234.   case 'l':
  235.     /* long or long long */
  236.     if (type != 'l')
  237.       goto set_type;
  238.     type = 'L';   /* "ll" means "L" */
  239.     break;
  240.   case 'n':
  241.     /* bytes written, no output as such */
  242.     (void) va_arg (ap, void *);
  243.     goto next;
  244.   case 's':
  245.     /* If no precision was given, then determine the string length
  246.        and put it there, to be added to the total under "next".  If
  247.        a precision was given then that's already the maximum from
  248.        this field, but see whether the string is shorter than that,
  249.        in case the limit was very big.  */
  250.     {
  251.       const char  *s = va_arg (ap, const char *);
  252.       prec = (seen_prec ? strnlen (s, prec) : strlen (s));
  253.     }
  254.     goto next;
  255.   case 'p':
  256.     /* pointer, let's assume at worst it's octal with some padding */
  257.     (void) va_arg (ap, const void *);
  258.     total_width += 3 * sizeof (void *) + 16;
  259.     goto next;
  260.   case '%':
  261.     /* literal %, already accounted for by strlen(fmt) */
  262.     goto next;
  263.   case '#':
  264.     /* showbase, at most 2 for "0x" */
  265.     total_width += 2;
  266.     break;
  267.   case '+':
  268.   case ' ':
  269.     /* sign, already accounted for under numerics */
  270.     break;
  271.   case '-':
  272.     /* left justify, no effect on total width */
  273.     break;
  274.   case '.':
  275.     seen_prec = 1;
  276.     value = &prec;
  277.     break;
  278.   case '*':
  279.     {
  280.       /* negative width means left justify which can be ignored,
  281.  negative prec would be invalid, just use absolute value */
  282.       int n = va_arg (ap, int);
  283.       *value = ABS (n);
  284.     }
  285.     break;
  286.   case '0': case '1': case '2': case '3': case '4':
  287.   case '5': case '6': case '7': case '8': case '9':
  288.     /* process all digits to form a value */
  289.     {
  290.       int  n = 0;
  291.       do {
  292. n = n * 10 + (fchar-'0');
  293. fchar = *fmt++;
  294.       } while (isascii (fchar) && isdigit (fchar));
  295.       fmt--; /* unget the non-digit */
  296.       *value = n;
  297.     }
  298.     break;
  299.   default:
  300.     /* incomplete or invalid % sequence */
  301.     ASSERT (0);
  302.     goto next;
  303.   }
  304. }
  305.     next:
  306.       total_width += width;
  307.       total_width += prec;
  308.     }
  309.   if (total_width <= buf_size)
  310.     {
  311.       vsprintf (buf, orig_fmt, orig_ap);
  312.       len = strlen (buf);
  313.     }
  314.   else
  315.     {
  316.       char  *s;
  317.       s = __GMP_ALLOCATE_FUNC_TYPE (total_width, char);
  318.       vsprintf (s, orig_fmt, orig_ap);
  319.       len = strlen (s);
  320.       if (buf_size != 0)
  321. {
  322.   size_t  copylen = MIN (len, buf_size-1);
  323.   memcpy (buf, s, copylen);
  324.   buf[copylen] = '';
  325. }
  326.       (*__gmp_free_func) (s, total_width);
  327.     }
  328.   /* If total_width was somehow wrong then chances are we've already
  329.      clobbered memory, but maybe this check will still work.  */
  330.   ASSERT_ALWAYS (len < total_width);
  331.   return len;
  332. }
  333. #endif /* ! HAVE_VSNPRINTF */