snprintf.c revision 269257
1/* snprintf - compatibility implementation of snprintf, vsnprintf
2 *
3 * Copyright (c) 2013, NLnet Labs. All rights reserved.
4 *
5 * This software is open source.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 *
11 * Redistributions of source code must retain the above copyright notice,
12 * this list of conditions and the following disclaimer.
13 *
14 * Redistributions in binary form must reproduce the above copyright notice,
15 * this list of conditions and the following disclaimer in the documentation
16 * and/or other materials provided with the distribution.
17 *
18 * Neither the name of the NLNET LABS nor the names of its contributors may
19 * be used to endorse or promote products derived from this software without
20 * specific prior written permission.
21 *
22 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
23 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
24 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
25 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
26 * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
27 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
28 * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
29 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
30 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
31 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
32 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
33 */
34
35#include "config.h"
36#include <stdio.h>
37#include <ctype.h>
38#include <string.h>
39#include <stdarg.h>
40#include <stdlib.h>
41#include <errno.h>
42#ifdef HAVE_STDINT_H
43#include <stdint.h>
44#endif
45
46/* for test */
47/* #define SNPRINTF_TEST 1 */
48#ifdef SNPRINTF_TEST
49#define snprintf my_snprintf
50#define vsnprintf my_vsnprintf
51#endif /* SNPRINTF_TEST */
52
53int snprintf(char* str, size_t size, const char* format, ...);
54int vsnprintf(char* str, size_t size, const char* format, va_list arg);
55
56/**
57 * Very portable snprintf implementation, limited in functionality,
58 * esp. for %[capital] %[nonportable] and so on.  Reduced float functionality,
59 * mostly in formatting and range (e+-16), for %f and %g.
60 *
61 * %s, %d, %u, %i, %x, %c, %n and %% are fully supported.
62 *   This includes width, precision, flags 0- +, and *(arg for wid,prec).
63 * %f, %g, %m, %p have reduced support, support for wid,prec,flags,*, but
64 *   less floating point range, no %e formatting for %g.
65 */
66int snprintf(char* str, size_t size, const char* format, ...)
67{
68	int r;
69	va_list args;
70	va_start(args, format);
71	r = vsnprintf(str, size, format, args);
72	va_end(args);
73	return r;
74}
75
76/** add padding to string */
77static void
78print_pad(char** at, size_t* left, int* ret, char p, int num)
79{
80	while(num--) {
81		if(*left > 1) {
82			*(*at)++ = p;
83			(*left)--;
84		}
85		(*ret)++;
86	}
87}
88
89/** get negative symbol, 0 if none */
90static char
91get_negsign(int negative, int plus, int space)
92{
93	if(negative)
94		return '-';
95	if(plus)
96		return '+';
97	if(space)
98		return ' ';
99	return 0;
100}
101
102#define PRINT_DEC_BUFSZ 32 /* 20 is enough for 64 bit decimals */
103/** print decimal into buffer, returns length */
104static int
105print_dec(char* buf, int max, unsigned int value)
106{
107	int i = 0;
108	if(value == 0) {
109		if(max > 0) {
110			buf[0] = '0';
111			i = 1;
112		}
113	} else while(value && i < max) {
114		buf[i++] = '0' + value % 10;
115		value /= 10;
116	}
117	return i;
118}
119
120/** print long decimal into buffer, returns length */
121static int
122print_dec_l(char* buf, int max, unsigned long value)
123{
124	int i = 0;
125	if(value == 0) {
126		if(max > 0) {
127			buf[0] = '0';
128			i = 1;
129		}
130	} else while(value && i < max) {
131		buf[i++] = '0' + value % 10;
132		value /= 10;
133	}
134	return i;
135}
136
137/** print long decimal into buffer, returns length */
138static int
139print_dec_ll(char* buf, int max, unsigned long long value)
140{
141	int i = 0;
142	if(value == 0) {
143		if(max > 0) {
144			buf[0] = '0';
145			i = 1;
146		}
147	} else while(value && i < max) {
148		buf[i++] = '0' + value % 10;
149		value /= 10;
150	}
151	return i;
152}
153
154/** print hex into buffer, returns length */
155static int
156print_hex(char* buf, int max, unsigned int value)
157{
158	const char* h = "0123456789abcdef";
159	int i = 0;
160	if(value == 0) {
161		if(max > 0) {
162			buf[0] = '0';
163			i = 1;
164		}
165	} else while(value && i < max) {
166		buf[i++] = h[value & 0x0f];
167		value >>= 4;
168	}
169	return i;
170}
171
172/** print long hex into buffer, returns length */
173static int
174print_hex_l(char* buf, int max, unsigned long value)
175{
176	const char* h = "0123456789abcdef";
177	int i = 0;
178	if(value == 0) {
179		if(max > 0) {
180			buf[0] = '0';
181			i = 1;
182		}
183	} else while(value && i < max) {
184		buf[i++] = h[value & 0x0f];
185		value >>= 4;
186	}
187	return i;
188}
189
190/** print long long hex into buffer, returns length */
191static int
192print_hex_ll(char* buf, int max, unsigned long long value)
193{
194	const char* h = "0123456789abcdef";
195	int i = 0;
196	if(value == 0) {
197		if(max > 0) {
198			buf[0] = '0';
199			i = 1;
200		}
201	} else while(value && i < max) {
202		buf[i++] = h[value & 0x0f];
203		value >>= 4;
204	}
205	return i;
206}
207
208/** copy string into result, reversed */
209static void
210spool_str_rev(char** at, size_t* left, int* ret, const char* buf, int len)
211{
212	int i = len;
213	while(i) {
214		if(*left > 1) {
215			*(*at)++ = buf[--i];
216			(*left)--;
217		} else --i;
218		(*ret)++;
219	}
220}
221
222/** copy string into result */
223static void
224spool_str(char** at, size_t* left, int* ret, const char* buf, int len)
225{
226	int i;
227	for(i=0; i<len; i++) {
228		if(*left > 1) {
229			*(*at)++ = buf[i];
230			(*left)--;
231		}
232		(*ret)++;
233	}
234}
235
236/** print number formatted */
237static void
238print_num(char** at, size_t* left, int* ret, int minw, int precision,
239	int prgiven, int zeropad, int minus, int plus, int space,
240	int zero, int negative, char* buf, int len)
241{
242	int w = len; /* excludes minus sign */
243	char s = get_negsign(negative, plus, space);
244	if(minus) {
245		/* left adjust the number into the field, space padding */
246		/* calc numw = [sign][zeroes][number] */
247		int numw = w;
248		if(precision == 0 && zero) numw = 0;
249		if(numw < precision) numw = precision;
250		if(s) numw++;
251
252		/* sign */
253		if(s) print_pad(at, left, ret, s, 1);
254
255		/* number */
256		if(precision == 0 && zero) {
257			/* "" for the number */
258		} else {
259			if(w < precision)
260				print_pad(at, left, ret, '0', precision - w);
261			spool_str_rev(at, left, ret, buf, len);
262		}
263		/* spaces */
264		if(numw < minw)
265			print_pad(at, left, ret, ' ', minw - numw);
266	} else {
267		/* pad on the left of the number */
268		/* calculate numw has width of [sign][zeroes][number] */
269		int numw = w;
270		if(precision == 0 && zero) numw = 0;
271		if(numw < precision) numw = precision;
272		if(!prgiven && zeropad && numw < minw) numw = minw;
273		else if(s) numw++;
274
275		/* pad with spaces */
276		if(numw < minw)
277			print_pad(at, left, ret, ' ', minw - numw);
278		/* print sign (and one less zeropad if so) */
279		if(s) {
280			print_pad(at, left, ret, s, 1);
281			numw--;
282		}
283		/* pad with zeroes */
284		if(w < numw)
285			print_pad(at, left, ret, '0', numw - w);
286		if(precision == 0 && zero)
287			return;
288		/* print the characters for the value */
289		spool_str_rev(at, left, ret, buf, len);
290	}
291}
292
293/** print %d and %i */
294static void
295print_num_d(char** at, size_t* left, int* ret, int value,
296	int minw, int precision, int prgiven, int zeropad, int minus,
297	int plus, int space)
298{
299	char buf[PRINT_DEC_BUFSZ];
300	int negative = (value < 0);
301	int zero = (value == 0);
302	int len = print_dec(buf, (int)sizeof(buf),
303		(unsigned int)(negative?-value:value));
304	print_num(at, left, ret, minw, precision, prgiven, zeropad, minus,
305		plus, space, zero, negative, buf, len);
306}
307
308/** print %ld and %li */
309static void
310print_num_ld(char** at, size_t* left, int* ret, long value,
311	int minw, int precision, int prgiven, int zeropad, int minus,
312	int plus, int space)
313{
314	char buf[PRINT_DEC_BUFSZ];
315	int negative = (value < 0);
316	int zero = (value == 0);
317	int len = print_dec_l(buf, (int)sizeof(buf),
318		(unsigned long)(negative?-value:value));
319	print_num(at, left, ret, minw, precision, prgiven, zeropad, minus,
320		plus, space, zero, negative, buf, len);
321}
322
323/** print %lld and %lli */
324static void
325print_num_lld(char** at, size_t* left, int* ret, long long value,
326	int minw, int precision, int prgiven, int zeropad, int minus,
327	int plus, int space)
328{
329	char buf[PRINT_DEC_BUFSZ];
330	int negative = (value < 0);
331	int zero = (value == 0);
332	int len = print_dec_ll(buf, (int)sizeof(buf),
333		(unsigned long long)(negative?-value:value));
334	print_num(at, left, ret, minw, precision, prgiven, zeropad, minus,
335		plus, space, zero, negative, buf, len);
336}
337
338/** print %u */
339static void
340print_num_u(char** at, size_t* left, int* ret, unsigned int value,
341	int minw, int precision, int prgiven, int zeropad, int minus,
342	int plus, int space)
343{
344	char buf[PRINT_DEC_BUFSZ];
345	int negative = 0;
346	int zero = (value == 0);
347	int len = print_dec(buf, (int)sizeof(buf), value);
348	print_num(at, left, ret, minw, precision, prgiven, zeropad, minus,
349		plus, space, zero, negative, buf, len);
350}
351
352/** print %lu */
353static void
354print_num_lu(char** at, size_t* left, int* ret, unsigned long value,
355	int minw, int precision, int prgiven, int zeropad, int minus,
356	int plus, int space)
357{
358	char buf[PRINT_DEC_BUFSZ];
359	int negative = 0;
360	int zero = (value == 0);
361	int len = print_dec_l(buf, (int)sizeof(buf), value);
362	print_num(at, left, ret, minw, precision, prgiven, zeropad, minus,
363		plus, space, zero, negative, buf, len);
364}
365
366/** print %llu */
367static void
368print_num_llu(char** at, size_t* left, int* ret, unsigned long long value,
369	int minw, int precision, int prgiven, int zeropad, int minus,
370	int plus, int space)
371{
372	char buf[PRINT_DEC_BUFSZ];
373	int negative = 0;
374	int zero = (value == 0);
375	int len = print_dec_ll(buf, (int)sizeof(buf), value);
376	print_num(at, left, ret, minw, precision, prgiven, zeropad, minus,
377		plus, space, zero, negative, buf, len);
378}
379
380/** print %x */
381static void
382print_num_x(char** at, size_t* left, int* ret, unsigned int value,
383	int minw, int precision, int prgiven, int zeropad, int minus,
384	int plus, int space)
385{
386	char buf[PRINT_DEC_BUFSZ];
387	int negative = 0;
388	int zero = (value == 0);
389	int len = print_hex(buf, (int)sizeof(buf), value);
390	print_num(at, left, ret, minw, precision, prgiven, zeropad, minus,
391		plus, space, zero, negative, buf, len);
392}
393
394/** print %lx */
395static void
396print_num_lx(char** at, size_t* left, int* ret, unsigned long value,
397	int minw, int precision, int prgiven, int zeropad, int minus,
398	int plus, int space)
399{
400	char buf[PRINT_DEC_BUFSZ];
401	int negative = 0;
402	int zero = (value == 0);
403	int len = print_hex_l(buf, (int)sizeof(buf), value);
404	print_num(at, left, ret, minw, precision, prgiven, zeropad, minus,
405		plus, space, zero, negative, buf, len);
406}
407
408/** print %llx */
409static void
410print_num_llx(char** at, size_t* left, int* ret, unsigned long long value,
411	int minw, int precision, int prgiven, int zeropad, int minus,
412	int plus, int space)
413{
414	char buf[PRINT_DEC_BUFSZ];
415	int negative = 0;
416	int zero = (value == 0);
417	int len = print_hex_ll(buf, (int)sizeof(buf), value);
418	print_num(at, left, ret, minw, precision, prgiven, zeropad, minus,
419		plus, space, zero, negative, buf, len);
420}
421
422/** print %llp */
423static void
424print_num_llp(char** at, size_t* left, int* ret, void* value,
425	int minw, int precision, int prgiven, int zeropad, int minus,
426	int plus, int space)
427{
428	char buf[PRINT_DEC_BUFSZ];
429	int negative = 0;
430	int zero = (value == 0);
431#if defined(UINTPTR_MAX) && defined(UINT32_MAX) && (UINTPTR_MAX == UINT32_MAX)
432	/* avoid warning about upcast on 32bit systems */
433	unsigned long long llvalue = (unsigned long)value;
434#else
435	unsigned long long llvalue = (unsigned long long)value;
436#endif
437	int len = print_hex_ll(buf, (int)sizeof(buf), llvalue);
438	if(zero) {
439		buf[0]=')';
440		buf[1]='l';
441		buf[2]='i';
442		buf[3]='n';
443		buf[4]='(';
444		len = 5;
445	} else {
446		/* put '0x' in front of the (reversed) buffer result */
447		if(len < PRINT_DEC_BUFSZ)
448			buf[len++] = 'x';
449		if(len < PRINT_DEC_BUFSZ)
450			buf[len++] = '0';
451	}
452	print_num(at, left, ret, minw, precision, prgiven, zeropad, minus,
453		plus, space, zero, negative, buf, len);
454}
455
456#define PRINT_FLOAT_BUFSZ 64 /* xx.yy with 20.20 about the max */
457/** spool remainder after the decimal point to buffer, in reverse */
458static int
459print_remainder(char* buf, int max, double r, int prec)
460{
461	unsigned long long cap = 1;
462	unsigned long long value;
463	int len, i;
464	if(prec > 19) prec = 19; /* max we can do */
465	if(max < prec) return 0;
466	for(i=0; i<prec; i++) {
467		cap *= 10;
468	}
469	r *= (double)cap;
470	value = (unsigned long long)r;
471	/* see if we need to round up */
472	if(((unsigned long long)((r - (double)value)*10.0)) >= 5) {
473		value++;
474		/* that might carry to numbers before the comma, if so,
475		 * just ignore that rounding. failure because 64bitprintout */
476		if(value >= cap)
477			value = cap-1;
478	}
479	len = print_dec_ll(buf, max, value);
480	while(len < prec) { /* pad with zeroes, e.g. if 0.0012 */
481		buf[len++] = '0';
482	}
483	if(len < max)
484		buf[len++] = '.';
485	return len;
486}
487
488/** spool floating point to buffer */
489static int
490print_float(char* buf, int max, double value, int prec)
491{
492	/* as xxx.xxx  if prec==0, no '.', with prec decimals after . */
493	/* no conversion for NAN and INF, because we do not want to require
494	   linking with -lm. */
495	/* Thus, the conversions use 64bit integers to convert the numbers,
496	 * which makes 19 digits before and after the decimal point the max */
497	unsigned long long whole = (unsigned long long)value;
498	double remain = value - (double)whole;
499	int len = 0;
500	if(prec != 0)
501		len = print_remainder(buf, max, remain, prec);
502	len += print_dec_ll(buf+len, max-len, whole);
503	return len;
504}
505
506/** print %f */
507static void
508print_num_f(char** at, size_t* left, int* ret, double value,
509	int minw, int precision, int prgiven, int zeropad, int minus,
510	int plus, int space)
511{
512	char buf[PRINT_FLOAT_BUFSZ];
513	int negative = (value < 0);
514	int zero = 0;
515	int len;
516	if(!prgiven) precision = 6;
517	len = print_float(buf, (int)sizeof(buf), negative?-value:value,
518		precision);
519	print_num(at, left, ret, minw, 1, 0, zeropad, minus,
520		plus, space, zero, negative, buf, len);
521}
522
523/* rudimentary %g support */
524static int
525print_float_g(char* buf, int max, double value, int prec)
526{
527	unsigned long long whole = (unsigned long long)value;
528	double remain = value - (double)whole;
529	int before = 0;
530	int len = 0;
531
532	/* number of digits before the decimal point */
533	while(whole > 0) {
534		before++;
535		whole /= 10;
536	}
537	whole = (unsigned long long)value;
538
539	if(prec > before && remain != 0.0) {
540		/* see if the last decimals are zero, if so, skip them */
541		len = print_remainder(buf, max, remain, prec-before);
542		while(len > 0 && buf[0]=='0') {
543			memmove(buf, buf+1, --len);
544		}
545	}
546	len += print_dec_ll(buf+len, max-len, whole);
547	return len;
548}
549
550
551/** print %g */
552static void
553print_num_g(char** at, size_t* left, int* ret, double value,
554	int minw, int precision, int prgiven, int zeropad, int minus,
555	int plus, int space)
556{
557	char buf[PRINT_FLOAT_BUFSZ];
558	int negative = (value < 0);
559	int zero = 0;
560	int len;
561	if(!prgiven) precision = 6;
562	if(precision == 0) precision = 1;
563	len = print_float_g(buf, (int)sizeof(buf), negative?-value:value,
564		precision);
565	print_num(at, left, ret, minw, 1, 0, zeropad, minus,
566		plus, space, zero, negative, buf, len);
567}
568
569
570/** strnlen (compat implementation) */
571static int
572my_strnlen(const char* s, int max)
573{
574	int i;
575	for(i=0; i<max; i++)
576		if(s[i]==0)
577			return i;
578	return max;
579}
580
581/** print %s */
582static void
583print_str(char** at, size_t* left, int* ret, char* s,
584	int minw, int precision, int prgiven, int minus)
585{
586	int w;
587	/* with prec: no more than x characters from this string, stop at 0 */
588	if(prgiven)
589		w = my_strnlen(s, precision);
590	else	w = (int)strlen(s); /* up to the nul */
591	if(w < minw && !minus)
592		print_pad(at, left, ret, ' ', minw - w);
593	spool_str(at, left, ret, s, w);
594	if(w < minw && minus)
595		print_pad(at, left, ret, ' ', minw - w);
596}
597
598/** print %c */
599static void
600print_char(char** at, size_t* left, int* ret, int c,
601	int minw, int minus)
602{
603	if(1 < minw && !minus)
604		print_pad(at, left, ret, ' ', minw - 1);
605	print_pad(at, left, ret, c, 1);
606	if(1 < minw && minus)
607		print_pad(at, left, ret, ' ', minw - 1);
608}
609
610
611/**
612 * Print to string.
613 * str: string buffer for result. result will be null terminated.
614 * size: size of the buffer. null is put inside buffer.
615 * format: printf format string.
616 * arg: '...' arguments to print.
617 * returns number of characters. a null is printed after this.
618 * return number of bytes that would have been written
619 *	   if the buffer had been large enough.
620 *
621 * supported format specifiers:
622 * 	%s, %u, %d, %x, %i, %f, %g, %c, %p, %n.
623 * 	length: l, ll (for d, u, x).
624 * 	precision: 6.6d (for d, u, x)
625 * 		%f, %g precisions, 0.3f
626 * 		%20s, '.*s'
627 * 	and %%.
628 */
629int vsnprintf(char* str, size_t size, const char* format, va_list arg)
630{
631	char* at = str;
632	size_t left = size;
633	int ret = 0;
634	const char* fmt = format;
635	int conv, minw, precision, prgiven, zeropad, minus, plus, space, length;
636	while(*fmt) {
637		/* copy string before % */
638		while(*fmt && *fmt!='%') {
639			if(left > 1) {
640				*at++ = *fmt++;
641				left--;
642			} else fmt++;
643			ret++;
644		}
645
646		/* see if we are at end */
647		if(!*fmt) break;
648
649		/* fetch next argument % designation from format string */
650		fmt++; /* skip the '%' */
651
652		/********************************/
653		/* get the argument designation */
654		/********************************/
655		/* we must do this vararg stuff inside this function for
656		 * portability.  Hence, get_designation, and print_designation
657		 * are not their own functions. */
658
659		/* printout designation:
660		 * conversion specifier: x, d, u, s, c, n, m, p
661		 * flags: # not supported
662		 *        0 zeropad (on the left)
663		 *	  - left adjust (right by default)
664		 *	  ' ' printspace for positive number (in - position).
665		 *	  + alwayssign
666		 * fieldwidth: [1-9][0-9]* minimum field width.
667		 * 	if this is * then type int next argument specifies the minwidth.
668		 * 	if this is negative, the - flag is set (with positive width).
669		 * precision: period[digits]*, %.2x.
670		 * 	if this is * then type int next argument specifies the precision.
671		 *	just '.' or negative value means precision=0.
672		 *		this is mindigits to print for d, i, u, x
673		 *		this is aftercomma digits for f
674		 *		this is max number significant digits for g
675		 *		maxnumber characters to be printed for s
676		 * length: 0-none (int), 1-l (long), 2-ll (long long)
677		 * 	notsupported: hh (char), h (short), L (long double), q, j, z, t
678		 * Does not support %m$ and *m$ argument designation as array indices.
679		 * Does not support %#x
680		 *
681		 */
682		minw = 0;
683		precision = 1;
684		prgiven = 0;
685		zeropad = 0;
686		minus = 0;
687		plus = 0;
688		space = 0;
689		length = 0;
690
691		/* get flags in any order */
692		for(;;) {
693			if(*fmt == '0')
694				zeropad = 1;
695			else if(*fmt == '-')
696				minus = 1;
697			else if(*fmt == '+')
698				plus = 1;
699			else if(*fmt == ' ')
700				space = 1;
701			else break;
702			fmt++;
703		}
704
705		/* field width */
706		if(*fmt == '*') {
707			fmt++; /* skip char */
708			minw = va_arg(arg, int);
709			if(minw < 0) {
710				minus = 1;
711				minw = -minw;
712			}
713		} else while(*fmt >= '0' && *fmt <= '9') {
714			minw = minw*10 + (*fmt++)-'0';
715		}
716
717		/* precision */
718		if(*fmt == '.') {
719			fmt++; /* skip period */
720			prgiven = 1;
721			precision = 0;
722			if(*fmt == '*') {
723				fmt++; /* skip char */
724				precision = va_arg(arg, int);
725				if(precision < 0)
726					precision = 0;
727			} else while(*fmt >= '0' && *fmt <= '9') {
728				precision = precision*10 + (*fmt++)-'0';
729			}
730		}
731
732		/* length */
733		if(*fmt == 'l') {
734			fmt++; /* skip char */
735			length = 1;
736			if(*fmt == 'l') {
737				fmt++; /* skip char */
738				length = 2;
739			}
740		}
741
742		/* get the conversion */
743		if(!*fmt) conv = 0;
744		else	conv = *fmt++;
745
746		/***********************************/
747		/* print that argument designation */
748		/***********************************/
749		switch(conv) {
750		case 'i':
751		case 'd':
752			if(length == 0)
753			    print_num_d(&at, &left, &ret, va_arg(arg, int),
754				minw, precision, prgiven, zeropad, minus, plus, space);
755			else if(length == 1)
756			    print_num_ld(&at, &left, &ret, va_arg(arg, long),
757				minw, precision, prgiven, zeropad, minus, plus, space);
758			else if(length == 2)
759			    print_num_lld(&at, &left, &ret,
760				va_arg(arg, long long),
761				minw, precision, prgiven, zeropad, minus, plus, space);
762			break;
763		case 'u':
764			if(length == 0)
765			    print_num_u(&at, &left, &ret,
766				va_arg(arg, unsigned int),
767				minw, precision, prgiven, zeropad, minus, plus, space);
768			else if(length == 1)
769			    print_num_lu(&at, &left, &ret,
770				va_arg(arg, unsigned long),
771				minw, precision, prgiven, zeropad, minus, plus, space);
772			else if(length == 2)
773			    print_num_llu(&at, &left, &ret,
774				va_arg(arg, unsigned long long),
775				minw, precision, prgiven, zeropad, minus, plus, space);
776			break;
777		case 'x':
778			if(length == 0)
779			    print_num_x(&at, &left, &ret,
780				va_arg(arg, unsigned int),
781				minw, precision, prgiven, zeropad, minus, plus, space);
782			else if(length == 1)
783			    print_num_lx(&at, &left, &ret,
784				va_arg(arg, unsigned long),
785				minw, precision, prgiven, zeropad, minus, plus, space);
786			else if(length == 2)
787			    print_num_llx(&at, &left, &ret,
788				va_arg(arg, unsigned long long),
789				minw, precision, prgiven, zeropad, minus, plus, space);
790			break;
791		case 's':
792			print_str(&at, &left, &ret, va_arg(arg, char*),
793				minw, precision, prgiven, minus);
794			break;
795		case 'c':
796			print_char(&at, &left, &ret, va_arg(arg, int),
797				minw, minus);
798			break;
799		case 'n':
800			*va_arg(arg, int*) = ret;
801			break;
802		case 'm':
803			print_str(&at, &left, &ret, strerror(errno),
804				minw, precision, prgiven, minus);
805			break;
806		case 'p':
807			print_num_llp(&at, &left, &ret, va_arg(arg, void*),
808				minw, precision, prgiven, zeropad, minus, plus, space);
809			break;
810		case '%':
811			print_pad(&at, &left, &ret, '%', 1);
812			break;
813		case 'f':
814			print_num_f(&at, &left, &ret, va_arg(arg, double),
815				minw, precision, prgiven, zeropad, minus, plus, space);
816			break;
817		case 'g':
818			print_num_g(&at, &left, &ret, va_arg(arg, double),
819				minw, precision, prgiven, zeropad, minus, plus, space);
820			break;
821		/* unknown */
822		default:
823		case 0: break;
824		}
825	}
826
827	/* zero terminate */
828	if(left > 0)
829		*at = 0;
830	return ret;
831}
832
833#ifdef SNPRINTF_TEST
834
835/** do tests */
836#undef snprintf
837#define DOTEST(bufsz, result, retval, ...) do { \
838	char buf[bufsz]; \
839	printf("now test %s\n", #__VA_ARGS__); \
840	int r=my_snprintf(buf, sizeof(buf), __VA_ARGS__); \
841	if(r != retval || strcmp(buf, result) != 0) { \
842		printf("error test(%s) was \"%s\":%d\n", \
843			""#bufsz", "#result", "#retval", "#__VA_ARGS__, \
844			buf, r); \
845		exit(1); \
846		} \
847	r=snprintf(buf, sizeof(buf), __VA_ARGS__); \
848	if(r != retval || strcmp(buf, result) != 0) { \
849		printf("error test(%s) differs with system, \"%s\":%d\n", \
850			""#bufsz", "#result", "#retval", "#__VA_ARGS__, \
851			buf, r); \
852		exit(1); \
853		} \
854	printf("test(\"%s\":%d) passed\n", buf, r); \
855	} while(0);
856
857/** test program */
858int main(void)
859{
860	int x = 0;
861
862	/* bufsize, expectedstring, expectedretval, snprintf arguments */
863	DOTEST(1024, "hello", 5, "hello");
864	DOTEST(1024, "h", 1, "h");
865	/* warning from gcc for format string, but it does work
866	 * DOTEST(1024, "", 0, ""); */
867
868	DOTEST(3, "he", 5, "hello");
869	DOTEST(1, "", 7, "%d", 7823089);
870
871	/* test positive numbers */
872	DOTEST(1024, "0", 1, "%d", 0);
873	DOTEST(1024, "1", 1, "%d", 1);
874	DOTEST(1024, "9", 1, "%d", 9);
875	DOTEST(1024, "15", 2, "%d", 15);
876	DOTEST(1024, "ab15cd", 6, "ab%dcd", 15);
877	DOTEST(1024, "167", 3, "%d", 167);
878	DOTEST(1024, "7823089", 7, "%d", 7823089);
879	DOTEST(1024, " 12", 3, "%3d", 12);
880	DOTEST(1024, "012", 3, "%.3d", 12);
881	DOTEST(1024, "012", 3, "%3.3d", 12);
882	DOTEST(1024, "012", 3, "%03d", 12);
883	DOTEST(1024, " 012", 4, "%4.3d", 12);
884	DOTEST(1024, "", 0, "%.0d", 0);
885
886	/* test negative numbers */
887	DOTEST(1024, "-1", 2, "%d", -1);
888	DOTEST(1024, "-12", 3, "%3d", -12);
889	DOTEST(1024, " -2", 3, "%3d", -2);
890	DOTEST(1024, "-012", 4, "%.3d", -12);
891	DOTEST(1024, "-012", 4, "%3.3d", -12);
892	DOTEST(1024, "-012", 4, "%4.3d", -12);
893	DOTEST(1024, " -012", 5, "%5.3d", -12);
894	DOTEST(1024, "-12", 3, "%03d", -12);
895	DOTEST(1024, "-02", 3, "%03d", -2);
896	DOTEST(1024, "-15", 3, "%d", -15);
897	DOTEST(1024, "-7307", 5, "%d", -7307);
898	DOTEST(1024, "-12  ", 5, "%-5d", -12);
899	DOTEST(1024, "-00012", 6, "%-.5d", -12);
900
901	/* test + and space flags */
902	DOTEST(1024, "+12", 3, "%+d", 12);
903	DOTEST(1024, " 12", 3, "% d", 12);
904
905	/* test %u */
906	DOTEST(1024, "12", 2, "%u", 12);
907	DOTEST(1024, "0", 1, "%u", 0);
908	DOTEST(1024, "4294967295", 10, "%u", 0xffffffff);
909
910	/* test %x */
911	DOTEST(1024, "0", 1, "%x", 0);
912	DOTEST(1024, "c", 1, "%x", 12);
913	DOTEST(1024, "12ab34cd", 8, "%x", 0x12ab34cd);
914
915	/* test %llu, %lld */
916	DOTEST(1024, "18446744073709551615", 20, "%llu",
917		(long long)0xffffffffffffffff);
918	DOTEST(1024, "-9223372036854775808", 20, "%lld",
919		(long long)0x8000000000000000);
920	DOTEST(1024, "9223372036854775808", 19, "%llu",
921		(long long)0x8000000000000000);
922
923	/* test %s */
924	DOTEST(1024, "hello", 5, "%s", "hello");
925	DOTEST(1024, "     hello", 10, "%10s", "hello");
926	DOTEST(1024, "hello     ", 10, "%-10s", "hello");
927	DOTEST(1024, "he", 2, "%.2s", "hello");
928	DOTEST(1024, "  he", 4, "%4.2s", "hello");
929	DOTEST(1024, "   h", 4, "%4.2s", "h");
930
931	/* test %c */
932	DOTEST(1024, "a", 1, "%c", 'a');
933	/* warning from gcc for format string, but it does work
934	   DOTEST(1024, "    a", 5, "%5c", 'a');
935	   DOTEST(1024, "a", 1, "%.0c", 'a'); */
936
937	/* test %n */
938	DOTEST(1024, "hello", 5, "hello%n", &x);
939	if(x != 5) { printf("the %%n failed\n"); exit(1); }
940
941	/* test %m */
942	errno = 0;
943	DOTEST(1024, "Success", 7, "%m");
944
945	/* test %p */
946	DOTEST(1024, "0x10", 4, "%p", (void*)0x10);
947	DOTEST(1024, "(nil)", 5, "%p", (void*)0x0);
948
949	/* test %% */
950	DOTEST(1024, "%", 1, "%%");
951
952	/* test %f */
953	DOTEST(1024, "0.000000", 8, "%f", 0.0);
954	DOTEST(1024, "0.00", 4, "%.2f", 0.0);
955	/* differs, "-0.00" DOTEST(1024, "0.00", 4, "%.2f", -0.0); */
956	DOTEST(1024, "234.00", 6, "%.2f", 234.005);
957	DOTEST(1024, "8973497.1246", 12, "%.4f", 8973497.12456);
958	DOTEST(1024, "-12.000000", 10, "%f", -12.0);
959	DOTEST(1024, "6", 1, "%.0f", 6.0);
960
961	DOTEST(1024, "6", 1, "%g", 6.0);
962	DOTEST(1024, "6.1", 3, "%g", 6.1);
963	DOTEST(1024, "6.15", 4, "%g", 6.15);
964
965	/* These format strings are from the code of NSD, Unbound, ldns */
966
967	DOTEST(1024, "abcdef", 6, "%s", "abcdef");
968	DOTEST(1024, "005", 3, "%03u", 5);
969	DOTEST(1024, "12345", 5, "%03u", 12345);
970	DOTEST(1024, "5", 1, "%d", 5);
971	DOTEST(1024, "(nil)", 5, "%p", NULL);
972	DOTEST(1024, "12345", 5, "%ld", (long)12345);
973	DOTEST(1024, "12345", 5, "%lu", (long)12345);
974	DOTEST(1024, "       12345", 12, "%12u", (unsigned)12345);
975	DOTEST(1024, "12345", 5, "%u", (unsigned)12345);
976	DOTEST(1024, "12345", 5, "%llu", (unsigned long long)12345);
977	DOTEST(1024, "12345", 5, "%x", 0x12345);
978	DOTEST(1024, "12345", 5, "%llx", (long long)0x12345);
979	DOTEST(1024, "012345", 6, "%6.6d", 12345);
980	DOTEST(1024, "012345", 6, "%6.6u", 12345);
981	DOTEST(1024, "1234.54", 7, "%g", 1234.54);
982	DOTEST(1024, "123456789.54", 12, "%.12g", 123456789.54);
983	DOTEST(1024, "3456789123456.54", 16, "%.16g", 3456789123456.54);
984	/* %24g does not work with 24 digits, not enough accuracy,
985	 * the first 16 digits are correct */
986	DOTEST(1024, "12345", 5, "%3.3d", 12345);
987	DOTEST(1024, "000", 3, "%3.3d", 0);
988	DOTEST(1024, "001", 3, "%3.3d", 1);
989	DOTEST(1024, "012", 3, "%3.3d", 12);
990	DOTEST(1024, "-012", 4, "%3.3d", -12);
991	DOTEST(1024, "he", 2, "%.2s", "hello");
992	DOTEST(1024, "helloworld", 10, "%s%s", "hello", "world");
993	DOTEST(1024, "he", 2, "%.*s", 2, "hello");
994	DOTEST(1024, "  hello", 7, "%*s", 7, "hello");
995	DOTEST(1024, "hello  ", 7, "%*s", -7, "hello");
996	DOTEST(1024, "0", 1, "%c", '0');
997	DOTEST(1024, "A", 1, "%c", 'A');
998	DOTEST(1024, "", 1, "%c", 0);
999	DOTEST(1024, "\010", 1, "%c", 8);
1000	DOTEST(1024, "%", 1, "%%");
1001	DOTEST(1024, "0a", 2, "%02x", 0x0a);
1002	DOTEST(1024, "bd", 2, "%02x", 0xbd);
1003	DOTEST(1024, "12", 2, "%02ld", (long)12);
1004	DOTEST(1024, "02", 2, "%02ld", (long)2);
1005	DOTEST(1024, "02", 2, "%02u", (unsigned)2);
1006	DOTEST(1024, "765432", 6, "%05u", (unsigned)765432);
1007	DOTEST(1024, "10.234", 6, "%0.3f", 10.23421);
1008	DOTEST(1024, "123456.234", 10, "%0.3f", 123456.23421);
1009	DOTEST(1024, "123456789.234", 13, "%0.3f", 123456789.23421);
1010	DOTEST(1024, "123456.23", 9, "%.2f", 123456.23421);
1011	DOTEST(1024, "123456", 6, "%.0f", 123456.23421);
1012	DOTEST(1024, "0123", 4, "%.4x", 0x0123);
1013	DOTEST(1024, "00000123", 8, "%.8x", 0x0123);
1014	DOTEST(1024, "ffeb0cde", 8, "%.8x", 0xffeb0cde);
1015	DOTEST(1024, " 987654321", 10, "%10lu", (unsigned long)987654321);
1016	DOTEST(1024, "   987654321", 12, "%12lu", (unsigned long)987654321);
1017	DOTEST(1024, "987654321", 9, "%i", 987654321);
1018	DOTEST(1024, "-87654321", 9, "%i", -87654321);
1019	DOTEST(1024, "hello           ", 16, "%-16s", "hello");
1020	DOTEST(1024, "                ", 16, "%-16s", "");
1021	DOTEST(1024, "a               ", 16, "%-16s", "a");
1022	DOTEST(1024, "foobarfoobar    ", 16, "%-16s", "foobarfoobar");
1023	DOTEST(1024, "foobarfoobarfoobar", 18, "%-16s", "foobarfoobarfoobar");
1024
1025	/* combined expressions */
1026	DOTEST(1024, "foo 1.0 size 512 edns", 21,
1027		"foo %s size %d %s%s", "1.0", 512, "", "edns");
1028	DOTEST(15, "foo 1.0 size 5", 21,
1029		"foo %s size %d %s%s", "1.0", 512, "", "edns");
1030	DOTEST(1024, "packet 1203ceff id", 18,
1031		"packet %2.2x%2.2x%2.2x%2.2x id", 0x12, 0x03, 0xce, 0xff);
1032	DOTEST(1024, "/tmp/testbound_123abcd.tmp", 26, "/tmp/testbound_%u%s%s.tmp", 123, "ab", "cd");
1033
1034	return 0;
1035}
1036#endif /* SNPRINTF_TEST */
1037