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

数学计算

开发平台:

Unix_Linux

  1. /* __gmp_snprintf_funs -- support for gmp_snprintf and gmp_vsnprintf.
  2.    THE FUNCTIONS IN THIS FILE ARE FOR INTERNAL USE ONLY.  THEY'RE ALMOST
  3.    CERTAIN TO BE SUBJECT TO INCOMPATIBLE CHANGES OR DISAPPEAR COMPLETELY IN
  4.    FUTURE GNU MP RELEASES.
  5. Copyright 2001, 2002 Free Software Foundation, Inc.
  6. This file is part of the GNU MP Library.
  7. The GNU MP Library is free software; you can redistribute it and/or modify
  8. it under the terms of the GNU Lesser General Public License as published by
  9. the Free Software Foundation; either version 3 of the License, or (at your
  10. option) any later version.
  11. The GNU MP Library is distributed in the hope that it will be useful, but
  12. WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  13. or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
  14. License for more details.
  15. You should have received a copy of the GNU Lesser General Public License
  16. along with the GNU MP Library.  If not, see http://www.gnu.org/licenses/.  */
  17. #include "config.h"
  18. #if HAVE_STDARG
  19. #include <stdarg.h>
  20. #else
  21. #include <varargs.h>
  22. #endif
  23. #include <stdio.h>
  24. #include <string.h>
  25. #include "gmp.h"
  26. #include "gmp-impl.h"
  27. #if ! HAVE_VSNPRINTF
  28. #define vsnprintf  __gmp_replacement_vsnprintf
  29. #endif
  30. /* glibc 2.0.x vsnprintf returns either -1 or size-1 for an overflow, with
  31.    no indication how big the output would have been.  It's necessary to
  32.    re-run to determine that size.
  33.    "size-1" would mean success from a C99 vsnprintf, and the re-run is
  34.    unnecessary in this case, but we don't bother to try to detect what sort
  35.    of vsnprintf we've got.  size-1 should occur rarely in normal
  36.    circumstances.
  37.    vsnprintf might trash it's given ap (it does for instance in glibc 2.1.3
  38.    on powerpc), so copy it in case we need to use it to probe for the size
  39.    output that would have been produced.  Note there's no need to preserve
  40.    it for our callers, just for ourselves.  */
  41. static int
  42. gmp_snprintf_format (struct gmp_snprintf_t *d, const char *fmt,
  43.                      va_list orig_ap)
  44. {
  45.   int      ret, step, alloc, avail;
  46.   va_list  ap;
  47.   char     *p;
  48.   ASSERT (d->size >= 0);
  49.   avail = d->size;
  50.   if (avail > 1)
  51.     {
  52.       va_copy (ap, orig_ap);
  53.       ret = vsnprintf (d->buf, avail, fmt, ap);
  54.       if (ret == -1)
  55.         {
  56.           ASSERT (strlen (d->buf) == avail-1);
  57.           ret = avail-1;
  58.         }
  59.       step = MIN (ret, avail-1);
  60.       d->size -= step;
  61.       d->buf += step;
  62.       if (ret != avail-1)
  63.         return ret;
  64.       /* probably glibc 2.0.x truncated output, probe for actual size */
  65.       alloc = MAX (128, ret);
  66.     }
  67.   else
  68.     {
  69.       /* no space to write anything, just probe for size */
  70.       alloc = 128;
  71.     }
  72.   do
  73.     {
  74.       alloc *= 2;
  75.       p = __GMP_ALLOCATE_FUNC_TYPE (alloc, char);
  76.       va_copy (ap, orig_ap);
  77.       ret = vsnprintf (p, alloc, fmt, ap);
  78.       (*__gmp_free_func) (p, alloc);
  79.     }
  80.   while (ret == alloc-1 || ret == -1);
  81.   return ret;
  82. }
  83. static int
  84. gmp_snprintf_memory (struct gmp_snprintf_t *d, const char *str, size_t len)
  85. {
  86.   size_t n;
  87.   ASSERT (d->size >= 0);
  88.   if (d->size > 1)
  89.     {
  90.       n = MIN (d->size-1, len);
  91.       memcpy (d->buf, str, n);
  92.       d->buf += n;
  93.       d->size -= n;
  94.     }
  95.   return len;
  96. }
  97. static int
  98. gmp_snprintf_reps (struct gmp_snprintf_t *d, int c, int reps)
  99. {
  100.   size_t n;
  101.   ASSERT (reps >= 0);
  102.   ASSERT (d->size >= 0);
  103.   if (d->size > 1)
  104.     {
  105.       n = MIN (d->size-1, reps);
  106.       memset (d->buf, c, n);
  107.       d->buf += n;
  108.       d->size -= n;
  109.     }
  110.   return reps;
  111. }
  112. static int
  113. gmp_snprintf_final (struct gmp_snprintf_t *d)
  114. {
  115.   if (d->size >= 1)
  116.     d->buf[0] = '';
  117.   return 0;
  118. }
  119. const struct doprnt_funs_t  __gmp_snprintf_funs = {
  120.   (doprnt_format_t) gmp_snprintf_format,
  121.   (doprnt_memory_t) gmp_snprintf_memory,
  122.   (doprnt_reps_t)   gmp_snprintf_reps,
  123.   (doprnt_final_t)  gmp_snprintf_final
  124. };