sjero.net Git - wget/blob - src/snprintf.c

   1 /* This file is NOT part of Wget, but is used by Wget on the systems
   2    where vsnprintf() is not defined.  It has been written by Patrick
   3    Powell and modified by other people.  All the copyright and other
   4    notices have been left intact.
   5
   6    My changes are documented at the bottom, along with other changes.
   7    I hereby place my modifications to this file under the public
   8    domain.  */
   9
  10 /*
  11  * Copyright Patrick Powell 1995
  12  * This code is based on code written by Patrick Powell (papowell@astart.com)
  13  * It may be used for any purpose as long as this notice remains intact
  14  * on all source code distributions
  15  */
  16
  17 /**************************************************************
  18  * Original:
  19  * Patrick Powell Tue Apr 11 09:48:21 PDT 1995
  20  * A bombproof version of doprnt (dopr) included.
  21  * Sigh.  This sort of thing is always nasty do deal with.  Note that
  22  * the version here does not include floating point...
  23  *
  24  * snprintf() is used instead of sprintf() as it does limit checks
  25  * for string length.  This covers a nasty loophole.
  26  *
  27  * The other functions are there to prevent NULL pointers from
  28  * causing nast effects.
  29  *
  30  * More Recently:
  31  *  Brandon Long <blong@fiction.net> 9/15/96 for mutt 0.43
  32  *  This was ugly.  It is still ugly.  I opted out of floating point
  33  *  numbers, but the formatter understands just about everything
  34  *  from the normal C string format, at least as far as I can tell from
  35  *  the Solaris 2.5 printf(3S) man page.
  36  *
  37  *  Brandon Long <blong@fiction.net> 10/22/97 for mutt 0.87.1
  38  *    Ok, added some minimal floating point support, which means this
  39  *    probably requires libm on most operating systems.  Don't yet
  40  *    support the exponent (e,E) and sigfig (g,G).  Also, fmtint()
  41  *    was pretty badly broken, it just wasn't being exercised in ways
  42  *    which showed it, so that's been fixed.  Also, formated the code
  43  *    to mutt conventions, and removed dead code left over from the
  44  *    original.  Also, there is now a builtin-test, just compile with:
  45  *           gcc -DTEST_SNPRINTF -o snprintf snprintf.c -lm
  46  *    and run snprintf for results.
  47  *
  48  *  Thomas Roessler <roessler@guug.de> 01/27/98 for mutt 0.89i
  49  *    The PGP code was using unsigned hexadecimal formats.
  50  *    Unfortunately, unsigned formats simply didn't work.
  51  *
  52  *  Michael Elkins <me@cs.hmc.edu> 03/05/98 for mutt 0.90.8
  53  *    The original code assumed that both snprintf() and vsnprintf() were
  54  *    missing.  Some systems only have snprintf() but not vsnprintf(), so
  55  *    the code is now broken down under HAVE_SNPRINTF and HAVE_VSNPRINTF.
  56  *
  57  *  Andrew Tridgell (tridge@samba.org) Oct 1998
  58  *    fixed handling of %.0f
  59  *    added test for HAVE_LONG_DOUBLE
  60  *
  61  *  Russ Allbery <rra@stanford.edu> 2000-08-26
  62  *    fixed return value to comply with C99
  63  *    fixed handling of snprintf(NULL, ...)
  64  *
  65  *  Hrvoje Niksic <hniksic@xemacs.org> 2000-11-04
  66  *    include <config.h> instead of "config.h".
  67  *    moved TEST_SNPRINTF stuff out of HAVE_SNPRINTF ifdef.
  68  *    include <stdio.h> for NULL.
  69  *    added support and test cases for long long.
  70  *    don't declare argument types to (v)snprintf if stdarg is not used.
  71  *    use int instead of short int as 2nd arg to va_arg.
  72  *
  73  *  alexk (INN) 2002-08-21
  74  *    use LLONG in fmtfp to handle more characters during floating
  75  *    point conversion.
  76  *
  77  *  herb (Samba) 2002-12-19
  78  *    actually print args for %g and %e
  79  *
  80  *  Hrvoje Niksic <hniksic@xemacs.org> 2005-04-15
  81  *    write function definitions in the ansi2knr-friendly way.
  82  *    if string precision is specified, don't read VALUE past it.
  83  *    fix bug in fmtfp that caused 0.01 to be printed as 0.1.
  84  *    don't include <ctype.h> because none of it is used.
  85  *    interpret precision as number of significant digits with %g
  86  *    omit trailing decimal zeros with %g
  87  *
  88  **************************************************************/
  89
  90 #ifdef HAVE_CONFIG_H
  91 # include <config.h>
  92 #endif
  93
  94 /* For testing purposes, always compile in the code. */
  95 #ifdef TEST_SNPRINTF
  96 # undef HAVE_SNPRINTF
  97 # undef HAVE_VSNPRINTF
  98 # ifndef SIZEOF_LONG_LONG
  99 #  ifdef __GNUC__
 100 #   define SIZEOF_LONG_LONG 8
 101 #  endif
 102 # endif
 103 #endif
 104
 105 #if !defined(HAVE_SNPRINTF) || !defined(HAVE_VSNPRINTF)
 106
 107 #include <string.h>
 108 #include <sys/types.h>
 109 #include <stdio.h>              /* for NULL */
 110
 111 #include <stdarg.h>
 112
 113 #ifdef HAVE_LONG_DOUBLE
 114 #define LDOUBLE long double
 115 #else
 116 #define LDOUBLE double
 117 #endif
 118
 119 #if SIZEOF_LONG_LONG != 0
 120 # define LLONG long long
 121 #else
 122 # define LLONG long
 123 #endif
 124
 125 /* If we're running the test suite, rename snprintf and vsnprintf to
 126    avoid conflicts with the system version.  */
 127 #ifdef TEST_SNPRINTF
 128 # define snprintf test_snprintf
 129 # define vsnprintf test_vsnprintf
 130 #endif
 131
 132 int snprintf (char *str, size_t count, const char *fmt, ...);
 133 int vsnprintf (char *str, size_t count, const char *fmt, va_list arg);
 134
 135 static int dopr (char *buffer, size_t maxlen, const char *format,
 136                  va_list args);
 137 static int fmtstr (char *buffer, size_t *currlen, size_t maxlen,
 138                    const char *value, int flags, int min, int max);
 139 static int fmtint (char *buffer, size_t *currlen, size_t maxlen,
 140                    LLONG value, int base, int min, int max, int flags);
 141 static int fmtfp (char *buffer, size_t *currlen, size_t maxlen,
 142                   LDOUBLE fvalue, int min, int max, int flags);
 143 static int dopr_outch (char *buffer, size_t *currlen, size_t maxlen, char c);
 144
 145 /*
 146  * dopr(): poor man's version of doprintf
 147  */
 148
 149 /* format read states */
 150 #define DP_S_DEFAULT 0
 151 #define DP_S_FLAGS   1
 152 #define DP_S_MIN     2
 153 #define DP_S_DOT     3
 154 #define DP_S_MAX     4
 155 #define DP_S_MOD     5
 156 #define DP_S_MOD_L   6
 157 #define DP_S_CONV    7
 158 #define DP_S_DONE    8
 159
 160 /* format flags - Bits */
 161 #define DP_F_MINUS      (1 << 0)
 162 #define DP_F_PLUS       (1 << 1)
 163 #define DP_F_SPACE      (1 << 2)
 164 #define DP_F_NUM        (1 << 3)
 165 #define DP_F_ZERO       (1 << 4)
 166 #define DP_F_UP         (1 << 5)
 167 #define DP_F_UNSIGNED   (1 << 6)
 168 #define DP_F_FP_G       (1 << 7)
 169
 170 /* Conversion Flags */
 171 #define DP_C_SHORT   1
 172 #define DP_C_LONG    2
 173 #define DP_C_LLONG   3
 174 #define DP_C_LDOUBLE 4
 175
 176 #define char_to_int(p) (p - '0')
 177 #define MAX(p,q) ((p >= q) ? p : q)
 178 #define MIN(p,q) ((p <= q) ? p : q)
 179
 180 static int dopr (char *buffer, size_t maxlen, const char *format, va_list args)
 181 {
 182   char ch;
 183   LLONG value;
 184   LDOUBLE fvalue;
 185   char *strvalue;
 186   int min;
 187   int max;
 188   int state;
 189   int flags;
 190   int cflags;
 191   int total;
 192   size_t currlen;
 193
 194   state = DP_S_DEFAULT;
 195   currlen = flags = cflags = min = 0;
 196   max = -1;
 197   ch = *format++;
 198   total = 0;
 199
 200   while (state != DP_S_DONE)
 201   {
 202     if (ch == '\0')
 203       state = DP_S_DONE;
 204
 205     switch(state)
 206     {
 207     case DP_S_DEFAULT:
 208       if (ch == '%')
 209         state = DP_S_FLAGS;
 210       else
 211         total += dopr_outch (buffer, &currlen, maxlen, ch);
 212       ch = *format++;
 213       break;
 214     case DP_S_FLAGS:
 215       switch (ch)
 216       {
 217       case '-':
 218         flags |= DP_F_MINUS;
 219         ch = *format++;
 220         break;
 221       case '+':
 222         flags |= DP_F_PLUS;
 223         ch = *format++;
 224         break;
 225       case ' ':
 226         flags |= DP_F_SPACE;
 227         ch = *format++;
 228         break;
 229       case '#':
 230         flags |= DP_F_NUM;
 231         ch = *format++;
 232         break;
 233       case '0':
 234         flags |= DP_F_ZERO;
 235         ch = *format++;
 236         break;
 237       default:
 238         state = DP_S_MIN;
 239         break;
 240       }
 241       break;
 242     case DP_S_MIN:
 243       if ('0' <= ch && ch <= '9')
 244       {
 245         min = 10*min + char_to_int (ch);
 246         ch = *format++;
 247       }
 248       else if (ch == '*')
 249       {
 250         min = va_arg (args, int);
 251         ch = *format++;
 252         state = DP_S_DOT;
 253       }
 254       else
 255         state = DP_S_DOT;
 256       break;
 257     case DP_S_DOT:
 258       if (ch == '.')
 259       {
 260         state = DP_S_MAX;
 261         ch = *format++;
 262       }
 263       else
 264         state = DP_S_MOD;
 265       break;
 266     case DP_S_MAX:
 267       if ('0' <= ch && ch <= '9')
 268       {
 269         if (max < 0)
 270           max = 0;
 271         max = 10*max + char_to_int (ch);
 272         ch = *format++;
 273       }
 274       else if (ch == '*')
 275       {
 276         max = va_arg (args, int);
 277         ch = *format++;
 278         state = DP_S_MOD;
 279       }
 280       else
 281         state = DP_S_MOD;
 282       break;
 283     case DP_S_MOD:
 284       switch (ch)
 285       {
 286       case 'h':
 287         cflags = DP_C_SHORT;
 288         ch = *format++;
 289         break;
 290       case 'l':
 291         cflags = DP_C_LONG;
 292         ch = *format++;
 293         break;
 294       case 'L':
 295         cflags = DP_C_LDOUBLE;
 296         ch = *format++;
 297         break;
 298       default:
 299         break;
 300       }
 301       if (cflags != DP_C_LONG)
 302         state = DP_S_CONV;
 303       else
 304         state = DP_S_MOD_L;
 305       break;
 306     case DP_S_MOD_L:
 307       switch (ch)
 308         {
 309         case 'l':
 310           cflags = DP_C_LLONG;
 311           ch = *format++;
 312           break;
 313         default:
 314           break;
 315         }
 316       state = DP_S_CONV;
 317       break;
 318     case DP_S_CONV:
 319       switch (ch)
 320       {
 321       case 'd':
 322       case 'i':
 323         if (cflags == DP_C_SHORT)
 324           value = (short int) va_arg (args, int);
 325         else if (cflags == DP_C_LONG)
 326           value = va_arg (args, long int);
 327         else if (cflags == DP_C_LLONG)
 328           value = va_arg (args, LLONG);
 329         else
 330           value = va_arg (args, int);
 331         total += fmtint (buffer, &currlen, maxlen, value, 10, min, max, flags);
 332         break;
 333       case 'o':
 334         flags |= DP_F_UNSIGNED;
 335         if (cflags == DP_C_SHORT)
 336           value = (unsigned short int) va_arg (args, unsigned int);
 337         else if (cflags == DP_C_LONG)
 338           value = va_arg (args, unsigned long int);
 339         else if (cflags == DP_C_LLONG)
 340           value = va_arg (args, unsigned LLONG);
 341         else
 342           value = va_arg (args, unsigned int);
 343         total += fmtint (buffer, &currlen, maxlen, value, 8, min, max, flags);
 344         break;
 345       case 'u':
 346         flags |= DP_F_UNSIGNED;
 347         if (cflags == DP_C_SHORT)
 348           value = (unsigned short int) va_arg (args, unsigned int);
 349         else if (cflags == DP_C_LONG)
 350           value = va_arg (args, unsigned long int);
 351         else if (cflags == DP_C_LLONG)
 352           value = va_arg (args, unsigned LLONG);
 353         else
 354           value = va_arg (args, unsigned int);
 355         total += fmtint (buffer, &currlen, maxlen, value, 10, min, max, flags);
 356         break;
 357       case 'X':
 358         flags |= DP_F_UP;
 359       case 'x':
 360         flags |= DP_F_UNSIGNED;
 361         if (cflags == DP_C_SHORT)
 362           value = (unsigned short int) va_arg (args, unsigned int);
 363         else if (cflags == DP_C_LONG)
 364           value = va_arg (args, unsigned long int);
 365         else if (cflags == DP_C_LLONG)
 366           value = va_arg (args, unsigned LLONG);
 367         else
 368           value = va_arg (args, unsigned int);
 369         total += fmtint (buffer, &currlen, maxlen, value, 16, min, max, flags);
 370         break;
 371       case 'f':
 372         if (cflags == DP_C_LDOUBLE)
 373           fvalue = va_arg (args, LDOUBLE);
 374         else
 375           fvalue = va_arg (args, double);
 376         total += fmtfp (buffer, &currlen, maxlen, fvalue, min, max, flags);
 377         break;
 378       case 'E':
 379         flags |= DP_F_UP;
 380       case 'e':
 381         if (cflags == DP_C_LDOUBLE)
 382           fvalue = va_arg (args, LDOUBLE);
 383         else
 384           fvalue = va_arg (args, double);
 385         total += fmtfp (buffer, &currlen, maxlen, fvalue, min, max, flags);
 386         break;
 387       case 'G':
 388         flags |= DP_F_UP;
 389       case 'g':
 390         flags |= DP_F_FP_G;
 391         if (cflags == DP_C_LDOUBLE)
 392           fvalue = va_arg (args, LDOUBLE);
 393         else
 394           fvalue = va_arg (args, double);
 395         if (max == 0)
 396           /* C99 says: if precision [for %g] is zero, it is taken as one */
 397           max = 1;
 398         total += fmtfp (buffer, &currlen, maxlen, fvalue, min, max, flags);
 399         break;
 400       case 'c':
 401         total += dopr_outch (buffer, &currlen, maxlen, va_arg (args, int));
 402         break;
 403       case 's':
 404         strvalue = va_arg (args, char *);
 405         total += fmtstr (buffer, &currlen, maxlen, strvalue, flags, min, max);
 406         break;
 407       case 'p':
 408         strvalue = va_arg (args, void *);
 409         total += fmtint (buffer, &currlen, maxlen, (long) strvalue, 16, min,
 410                          max, flags);
 411         break;
 412       case 'n':
 413         if (cflags == DP_C_SHORT)
 414         {
 415           short int *num;
 416           num = va_arg (args, short int *);
 417           *num = currlen;
 418         }
 419         else if (cflags == DP_C_LONG)
 420         {
 421           long int *num;
 422           num = va_arg (args, long int *);
 423           *num = currlen;
 424         }
 425         else if (cflags == DP_C_LLONG)
 426         {
 427           LLONG *num;
 428           num = va_arg (args, LLONG *);
 429           *num = currlen;
 430         }
 431         else
 432         {
 433           int *num;
 434           num = va_arg (args, int *);
 435           *num = currlen;
 436         }
 437         break;
 438       case '%':
 439         total += dopr_outch (buffer, &currlen, maxlen, ch);
 440         break;
 441       case 'w':
 442         /* not supported yet, treat as next char */
 443         ch = *format++;
 444         break;
 445       default:
 446         /* Unknown, skip */
 447         break;
 448       }
 449       ch = *format++;
 450       state = DP_S_DEFAULT;
 451       flags = cflags = min = 0;
 452       max = -1;
 453       break;
 454     case DP_S_DONE:
 455       break;
 456     default:
 457       /* hmm? */
 458       break; /* some picky compilers need this */
 459     }
 460   }
 461   if (buffer != NULL)
 462   {
 463     if (currlen < maxlen - 1)
 464       buffer[currlen] = '\0';
 465     else
 466       buffer[maxlen - 1] = '\0';
 467   }
 468   return total;
 469 }
 470
 471 static int fmtstr (char *buffer, size_t *currlen, size_t maxlen,
 472                    const char *value, int flags, int min, int max)
 473 {
 474   int padlen, strln;     /* amount to pad */
 475   int cnt = 0;
 476   int total = 0;
 477
 478   if (value == 0)
 479   {
 480     value = "(null)";
 481   }
 482
 483   if (max < 0)
 484     strln = strlen (value);
 485   else
 486     /* When precision is specified, don't read VALUE past precision. */
 487     /*strln = strnlen (value, max);*/
 488     for (strln = 0; strln < max && value[strln]; ++strln)
 489       ;
 490   padlen = min - strln;
 491   if (padlen < 0)
 492     padlen = 0;
 493   if (flags & DP_F_MINUS)
 494     padlen = -padlen; /* Left Justify */
 495
 496   while (padlen > 0)
 497   {
 498     total += dopr_outch (buffer, currlen, maxlen, ' ');
 499     --padlen;
 500   }
 501   while (*value && ((max < 0) || (cnt < max)))
 502   {
 503     total += dopr_outch (buffer, currlen, maxlen, *value++);
 504     ++cnt;
 505   }
 506   while (padlen < 0)
 507   {
 508     total += dopr_outch (buffer, currlen, maxlen, ' ');
 509     ++padlen;
 510   }
 511   return total;
 512 }
 513
 514 /* Have to handle DP_F_NUM (ie 0x and 0 alternates) */
 515
 516 static int fmtint (char *buffer, size_t *currlen, size_t maxlen,
 517                    LLONG value, int base, int min, int max, int flags)
 518 {
 519   int signvalue = 0;
 520   unsigned LLONG uvalue;
 521   char convert[24];
 522   int place = 0;
 523   int spadlen = 0; /* amount to space pad */
 524   int zpadlen = 0; /* amount to zero pad */
 525   const char *digits;
 526   int total = 0;
 527
 528   if (max < 0)
 529     max = 0;
 530
 531   uvalue = value;
 532
 533   if(!(flags & DP_F_UNSIGNED))
 534   {
 535     if( value < 0 ) {
 536       signvalue = '-';
 537       uvalue = -value;
 538     }
 539     else
 540       if (flags & DP_F_PLUS)  /* Do a sign (+/i) */
 541         signvalue = '+';
 542     else
 543       if (flags & DP_F_SPACE)
 544         signvalue = ' ';
 545   }
 546
 547   if (flags & DP_F_UP)
 548     /* Should characters be upper case? */
 549     digits = "0123456789ABCDEF";
 550   else
 551     digits = "0123456789abcdef";
 552
 553   do {
 554     convert[place++] = digits[uvalue % (unsigned)base];
 555     uvalue = (uvalue / (unsigned)base );
 556   } while(uvalue && (place < sizeof (convert)));
 557   if (place == sizeof (convert)) place--;
 558   convert[place] = 0;
 559
 560   zpadlen = max - place;
 561   spadlen = min - MAX (max, place) - (signvalue ? 1 : 0);
 562   if (zpadlen < 0) zpadlen = 0;
 563   if (spadlen < 0) spadlen = 0;
 564   if (flags & DP_F_ZERO)
 565   {
 566     zpadlen = MAX(zpadlen, spadlen);
 567     spadlen = 0;
 568   }
 569   if (flags & DP_F_MINUS)
 570     spadlen = -spadlen; /* Left Justifty */
 571
 572 #ifdef DEBUG_SNPRINTF
 573   dprint (1, (debugfile, "zpad: %d, spad: %d, min: %d, max: %d, place: %d\n",
 574       zpadlen, spadlen, min, max, place));
 575 #endif
 576
 577   /* Spaces */
 578   while (spadlen > 0)
 579   {
 580     total += dopr_outch (buffer, currlen, maxlen, ' ');
 581     --spadlen;
 582   }
 583
 584   /* Sign */
 585   if (signvalue)
 586     total += dopr_outch (buffer, currlen, maxlen, signvalue);
 587
 588   /* Zeros */
 589   if (zpadlen > 0)
 590   {
 591     while (zpadlen > 0)
 592     {
 593       total += dopr_outch (buffer, currlen, maxlen, '0');
 594       --zpadlen;
 595     }
 596   }
 597
 598   /* Digits */
 599   while (place > 0)
 600     total += dopr_outch (buffer, currlen, maxlen, convert[--place]);
 601
 602   /* Left Justified spaces */
 603   while (spadlen < 0) {
 604     total += dopr_outch (buffer, currlen, maxlen, ' ');
 605     ++spadlen;
 606   }
 607
 608   return total;
 609 }
 610
 611 static LDOUBLE abs_val (LDOUBLE value)
 612 {
 613   LDOUBLE result = value;
 614
 615   if (value < 0)
 616     result = -value;
 617
 618   return result;
 619 }
 620
 621 static LDOUBLE pow10_int (int exp)
 622 {
 623   LDOUBLE result = 1;
 624
 625   while (exp)
 626   {
 627     result *= 10;
 628     exp--;
 629   }
 630
 631   return result;
 632 }
 633
 634 static LLONG round_int (LDOUBLE value)
 635 {
 636   LLONG intpart;
 637
 638   intpart = value;
 639   value = value - intpart;
 640   if (value >= 0.5)
 641     intpart++;
 642
 643   return intpart;
 644 }
 645
 646 static int fmtfp (char *buffer, size_t *currlen, size_t maxlen,
 647                   LDOUBLE fvalue, int min, int max, int flags)
 648 {
 649   int signvalue = 0;
 650   LDOUBLE ufvalue;
 651   char iconvert[24];
 652   char fconvert[24];
 653   int iplace = 0;
 654   int fplace = 0;
 655   int padlen = 0; /* amount to pad */
 656   int zpadlen = 0;
 657   int total = 0;
 658   LLONG intpart;
 659   LLONG fracpart;
 660   LLONG mask10;
 661   int leadingfrac0s = 0; /* zeros at the start of fractional part */
 662   int omitzeros = 0;
 663   int omitcount = 0;
 664
 665   /*
 666    * AIX manpage says the default is 0, but Solaris says the default
 667    * is 6, and sprintf on AIX defaults to 6
 668    */
 669   if (max < 0)
 670     max = 6;
 671
 672   ufvalue = abs_val (fvalue);
 673
 674   if (fvalue < 0)
 675     signvalue = '-';
 676   else
 677     if (flags & DP_F_PLUS)  /* Do a sign (+/i) */
 678       signvalue = '+';
 679     else
 680       if (flags & DP_F_SPACE)
 681         signvalue = ' ';
 682
 683 #if 0
 684   if (flags & DP_F_UP) caps = 1; /* Should characters be upper case? */
 685 #endif
 686
 687   intpart = ufvalue;
 688
 689   /* With %g precision is the number of significant digits, which
 690      includes the digits in intpart. */
 691   if (flags & DP_F_FP_G)
 692     {
 693       if (intpart != 0)
 694         {
 695           /* For each digit of INTPART, print one less fractional digit. */
 696           LLONG temp = intpart;
 697           for (temp = intpart; temp != 0; temp /= 10)
 698             --max;
 699           if (max < 0)
 700             max = 0;
 701         }
 702       else
 703         {
 704           /* For each leading 0 in fractional part, print one more
 705              fractional digit. */
 706           LDOUBLE temp;
 707           if (ufvalue != 0)
 708             for (temp = ufvalue; temp < 0.1; temp *= 10)
 709               ++max;
 710         }
 711     }
 712
 713   /* C99: trailing zeros are removed from the fractional portion of the
 714      result unless the # flag is specified */
 715   if ((flags & DP_F_FP_G) && !(flags & DP_F_NUM))
 716     omitzeros = 1;
 717
 718 #if SIZEOF_LONG_LONG > 0
 719 # define MAX_DIGITS 18          /* grok more digits with long long */
 720 #else
 721 # define MAX_DIGITS 9           /* just long */
 722 #endif
 723
 724   /*
 725    * Sorry, we only support several digits past the decimal because of
 726    * our conversion method
 727    */
 728   if (max > MAX_DIGITS)
 729     max = MAX_DIGITS;
 730
 731   /* Factor of 10 with the needed number of digits, e.g. 1000 for max==3 */
 732   mask10 = pow10_int (max);
 733
 734   /* We "cheat" by converting the fractional part to integer by
 735    * multiplying by a factor of 10
 736    */
 737   fracpart = round_int (mask10 * (ufvalue - intpart));
 738
 739   if (fracpart >= mask10)
 740   {
 741     intpart++;
 742     fracpart -= mask10;
 743   }
 744   else if (fracpart != 0)
 745     /* If fracpart has less digits than the 10* mask, we need to
 746        manually insert leading 0s.  For example 2.01's fractional part
 747        requires one leading zero to distinguish it from 2.1. */
 748     while (fracpart < mask10 / 10)
 749       {
 750         ++leadingfrac0s;
 751         mask10 /= 10;
 752       }
 753
 754 #ifdef DEBUG_SNPRINTF
 755   dprint (1, (debugfile, "fmtfp: %f =? %d.%d\n", fvalue, intpart, fracpart));
 756 #endif
 757
 758   /* Convert integer part */
 759   do {
 760     iconvert[iplace++] = '0' + intpart % 10;
 761     intpart = (intpart / 10);
 762   } while(intpart && (iplace < sizeof(iconvert)));
 763   if (iplace == sizeof(iconvert)) iplace--;
 764   iconvert[iplace] = 0;
 765
 766   /* Convert fractional part */
 767   do {
 768     fconvert[fplace++] = '0' + fracpart % 10;
 769     fracpart = (fracpart / 10);
 770   } while(fracpart && (fplace < sizeof(fconvert)));
 771   while (leadingfrac0s-- > 0 && fplace < sizeof(fconvert))
 772     fconvert[fplace++] = '0';
 773   if (fplace == sizeof(fconvert)) fplace--;
 774   fconvert[fplace] = 0;
 775   if (omitzeros)
 776     while (omitcount < fplace && fconvert[omitcount] == '0')
 777       ++omitcount;
 778
 779   /* -1 for decimal point, another -1 if we are printing a sign */
 780   padlen = min - iplace - (max - omitcount) - 1 - ((signvalue) ? 1 : 0);
 781   if (!omitzeros)
 782     zpadlen = max - fplace;
 783   if (zpadlen < 0)
 784     zpadlen = 0;
 785   if (padlen < 0)
 786     padlen = 0;
 787   if (flags & DP_F_MINUS)
 788     padlen = -padlen; /* Left Justifty */
 789
 790   if ((flags & DP_F_ZERO) && (padlen > 0))
 791   {
 792     if (signvalue)
 793     {
 794       total += dopr_outch (buffer, currlen, maxlen, signvalue);
 795       --padlen;
 796       signvalue = 0;
 797     }
 798     while (padlen > 0)
 799     {
 800       total += dopr_outch (buffer, currlen, maxlen, '0');
 801       --padlen;
 802     }
 803   }
 804   while (padlen > 0)
 805   {
 806     total += dopr_outch (buffer, currlen, maxlen, ' ');
 807     --padlen;
 808   }
 809   if (signvalue)
 810     total += dopr_outch (buffer, currlen, maxlen, signvalue);
 811
 812   while (iplace > 0)
 813     total += dopr_outch (buffer, currlen, maxlen, iconvert[--iplace]);
 814
 815   /*
 816    * Decimal point.  This should probably use locale to find the correct
 817    * char to print out.
 818    */
 819   if (max > 0 && (fplace > omitcount || zpadlen > 0))
 820   {
 821     total += dopr_outch (buffer, currlen, maxlen, '.');
 822
 823     while (fplace > omitcount)
 824       total += dopr_outch (buffer, currlen, maxlen, fconvert[--fplace]);
 825   }
 826
 827   while (zpadlen > 0)
 828   {
 829     total += dopr_outch (buffer, currlen, maxlen, '0');
 830     --zpadlen;
 831   }
 832
 833   while (padlen < 0)
 834   {
 835     total += dopr_outch (buffer, currlen, maxlen, ' ');
 836     ++padlen;
 837   }
 838
 839   return total;
 840 }
 841
 842 static int dopr_outch (char *buffer, size_t *currlen, size_t maxlen, char c)
 843 {
 844   if (*currlen + 1 < maxlen)
 845     buffer[(*currlen)++] = c;
 846   return 1;
 847 }
 848
 849 #ifndef HAVE_VSNPRINTF
 850 int vsnprintf (char *str, size_t count, const char *fmt, va_list args)
 851 {
 852   if (str != NULL)
 853     str[0] = 0;
 854   return dopr(str, count, fmt, args);
 855 }
 856 #endif /* !HAVE_VSNPRINTF */
 857
 858 #ifndef HAVE_SNPRINTF
 859 int snprintf (char *str, size_t count, const char *fmt,...)
 860 {
 861   va_list ap;
 862   int total;
 863
 864   va_start (ap, fmt);
 865   total = vsnprintf (str, count, fmt, ap);
 866   va_end (ap);
 867   return total;
 868 }
 869 #endif /* !HAVE_SNPRINTF */
 870 #endif /* !HAVE_SNPRINTF || !HAVE_VSNPRINTF */
 871
 872 #ifdef TEST_SNPRINTF
 873
 874 # ifndef LONG_STRING
 875 #  define LONG_STRING 1024
 876 # endif
 877
 878 int main (void)
 879 {
 880   char buf1[LONG_STRING];
 881   char buf2[LONG_STRING];
 882   char *fp_fmt[] = {
 883     /* %f formats */
 884     "%f",
 885     "%-1.5f",
 886     "%1.5f",
 887     "%123.9f",
 888     "%10.5f",
 889     "% 10.5f",
 890     "%+22.9f",
 891     "%+4.9f",
 892     "%01.3f",
 893     "%4f",
 894     "%3.1f",
 895     "%3.2f",
 896     "%.0f",
 897     "%.1f",
 898     "%#10.1f",
 899 #if SIZEOF_LONG_LONG != 0
 900     "%.16f",
 901     "%18.16f",
 902     "%-16.16f",
 903 #endif
 904     /* %g formats */
 905     "%g",
 906     "%1.5g",
 907     "%-1.5g",
 908     "%.9g",
 909     "%123.9g",
 910     "%#123.9g",
 911 #if SIZEOF_LONG_LONG != 0
 912     "%.16g",
 913     "%20.16g",
 914 #endif
 915     NULL
 916   };
 917   double fp_nums[] = { -1.5, 134.21, 91340.2, 341.1234, 0203.9, 0.96, 0.996,
 918                        0.9996, 1.996, 4.136, 0.00205, 0.0001, 321.000009,
 919                        0};
 920   char *int_fmt[] = {
 921     "%-1.5d",
 922     "%1.5d",
 923     "%123.9d",
 924     "%5.5d",
 925     "%10.5d",
 926     "% 10.5d",
 927     "%+22.33d",
 928     "%01.3d",
 929     "%4d",
 930     NULL
 931   };
 932   long int_nums[] = { -1, 134, 91340, 341, 0203, 0};
 933 #if SIZEOF_LONG_LONG != 0
 934   char *llong_fmt[] = {
 935     "%lld",             "%llu",
 936     "%-1.5lld",         "%-1.5llu",
 937     "%1.5lld",          "%1.5llu",
 938     "%123.9lld",        "%123.9llu",
 939     "%5.5lld",          "%5.5llu",
 940     "%10.5lld",         "%10.5llu",
 941     "% 10.5lld",        "% 10.5llu",
 942     "%+22.33lld",       "%+22.33llu",
 943     "%01.3lld",         "%01.3llu",
 944     "%4lld",            "%4llu",
 945     NULL
 946   };
 947   long long llong_nums[] = {
 948     ~(long long)0,              /* all-1 bit pattern */
 949     (~(unsigned long long)0) >> 1, /* largest signed long long */
 950     /* random... */
 951     -150, 134, 91340, 341,
 952     0
 953   };
 954 #endif
 955   int x, y;
 956   int fail = 0;
 957   int num = 0;
 958
 959   printf ("Testing snprintf format codes against system sprintf...\n");
 960
 961   for (x = 0; fp_fmt[x] != NULL ; x++)
 962     for (y = 0; fp_nums[y] != 0 ; y++)
 963     {
 964       snprintf (buf1, sizeof (buf1), fp_fmt[x], fp_nums[y]);
 965       sprintf (buf2, fp_fmt[x], fp_nums[y]);
 966       if (strcmp (buf1, buf2))
 967       {
 968         printf("snprintf doesn't match Format: %s\n\tsnprintf = %s\n\tsprintf  = %s\n",
 969             fp_fmt[x], buf1, buf2);
 970         fail++;
 971       }
 972       num++;
 973     }
 974
 975   for (x = 0; int_fmt[x] != NULL ; x++)
 976     for (y = 0; int_nums[y] != 0 ; y++)
 977     {
 978       snprintf (buf1, sizeof (buf1), int_fmt[x], int_nums[y]);
 979       sprintf (buf2, int_fmt[x], int_nums[y]);
 980       if (strcmp (buf1, buf2))
 981       {
 982         printf("snprintf doesn't match Format: %s\n\tsnprintf = %s\n\tsprintf  = %s\n",
 983             int_fmt[x], buf1, buf2);
 984         fail++;
 985       }
 986       num++;
 987     }
 988
 989 #if SIZEOF_LONG_LONG != 0
 990   for (x = 0; llong_fmt[x] != NULL ; x++)
 991     for (y = 0; llong_nums[y] != 0 ; y++)
 992     {
 993       snprintf (buf1, sizeof (buf1), llong_fmt[x], llong_nums[y]);
 994       sprintf (buf2, llong_fmt[x], llong_nums[y]);
 995       if (strcmp (buf1, buf2))
 996       {
 997         printf("snprintf doesn't match Format: %s\n\tsnprintf = %s\n\tsprintf  = %s\n",
 998             llong_fmt[x], buf1, buf2);
 999         fail++;
1000       }
1001       num++;
1002     }
1003 #endif
1004
1005   printf ("%d tests failed out of %d.\n", fail, num);
1006   return 0;
1007 }
1008 #endif /* TEST_SNPRINTF */