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