2 * Copyright (c) 1995-2003 Kungliga Tekniska Högskolan
3 * (Royal Institute of Technology, Stockholm, Sweden).
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 * notice, this list of conditions and the following disclaimer in the
15 * documentation and/or other materials provided with the distribution.
17 * 3. Neither the name of the Institute nor the names of its contributors
18 * may be used to endorse or promote products derived from this software
19 * without specific prior written permission.
21 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
36 RCSID("$Id: snprintf.c,v 1.35 2003/03/26 10:05:48 joda Exp $");
57 struct snprintf_state {
60 unsigned char *theend;
63 void (*append_char)(struct snprintf_state *, unsigned char);
68 #include "snprintf-test.h"
69 #endif /* TEST_SNPRINTF */
71 #if !defined(HAVE_VSNPRINTF) || defined(TEST_SNPRINTF)
73 sn_reserve (struct snprintf_state *state, size_t n)
75 return state->s + n > state->theend;
79 sn_append_char (struct snprintf_state *state, unsigned char c)
81 if (!sn_reserve (state, 1))
87 as_reserve (struct snprintf_state *state, size_t n)
89 if (state->s + n > state->theend) {
90 int off = state->s - state->str;
93 if (state->max_sz && state->sz >= state->max_sz)
96 state->sz = max(state->sz * 2, state->sz + n);
98 state->sz = min(state->sz, state->max_sz);
99 tmp = realloc (state->str, state->sz);
103 state->s = state->str + off;
104 state->theend = state->str + state->sz - 1;
110 as_append_char (struct snprintf_state *state, unsigned char c)
112 if(!as_reserve (state, 1))
116 /* longest integer types */
118 #ifdef HAVE_LONG_LONG
119 typedef unsigned long long u_longest;
120 typedef long long longest;
122 typedef unsigned long u_longest;
123 typedef long longest;
127 * is # supposed to do anything?
131 use_alternative (int flags, u_longest num, unsigned base)
133 return flags & alternate_flag && (base == 16 || base == 8) && num != 0;
137 append_number(struct snprintf_state *state,
138 u_longest num, unsigned base, const char *rep,
139 int width, int prec, int flags, int minusp)
145 /* given precision, ignore zero flag */
150 /* zero value with zero precision -> "" */
151 if(prec == 0 && n == 0)
154 (*state->append_char)(state, rep[n % base]);
159 /* pad with prec zeros */
161 (*state->append_char)(state, '0');
164 /* add length of alternate prefix (added later) to len */
165 if(use_alternative(flags, num, base))
168 if(flags & zero_flag){
170 if(minusp || (flags & space_flag) || (flags & plus_flag))
173 (*state->append_char)(state, '0');
177 /* add alternate prefix */
178 if(use_alternative(flags, num, base)){
180 (*state->append_char)(state, rep[10] + 23); /* XXX */
181 (*state->append_char)(state, '0');
185 (*state->append_char)(state, '-');
187 } else if(flags & plus_flag) {
188 (*state->append_char)(state, '+');
190 } else if(flags & space_flag) {
191 (*state->append_char)(state, ' ');
194 if(flags & minus_flag)
195 /* swap before padding with spaces */
196 for(i = 0; i < len / 2; i++){
197 char c = state->s[-i-1];
198 state->s[-i-1] = state->s[-len+i];
199 state->s[-len+i] = c;
203 (*state->append_char)(state, ' ');
206 if(!(flags & minus_flag))
207 /* swap after padding with spaces */
208 for(i = 0; i < len / 2; i++){
209 char c = state->s[-i-1];
210 state->s[-i-1] = state->s[-len+i];
211 state->s[-len+i] = c;
221 append_string (struct snprintf_state *state,
222 const unsigned char *arg,
230 arg = (const unsigned char*)"(null)";
235 width -= strlen((const char *)arg);
236 if(!(flags & minus_flag))
238 (*state->append_char) (state, ' ');
242 while (*arg && prec--) {
243 (*state->append_char) (state, *arg++);
248 (*state->append_char) (state, *arg++);
252 if(flags & minus_flag)
254 (*state->append_char) (state, ' ');
261 append_char(struct snprintf_state *state,
268 while(!(flags & minus_flag) && --width > 0) {
269 (*state->append_char) (state, ' ') ;
272 (*state->append_char) (state, arg);
274 while((flags & minus_flag) && --width > 0) {
275 (*state->append_char) (state, ' ');
282 * This can't be made into a function...
285 #ifdef HAVE_LONG_LONG
287 #define PARSE_INT_FORMAT(res, arg, unsig) \
288 if (long_long_flag) \
289 res = (unsig long long)va_arg(arg, unsig long long); \
290 else if (long_flag) \
291 res = (unsig long)va_arg(arg, unsig long); \
292 else if (short_flag) \
293 res = (unsig short)va_arg(arg, unsig int); \
295 res = (unsig int)va_arg(arg, unsig int)
299 #define PARSE_INT_FORMAT(res, arg, unsig) \
301 res = (unsig long)va_arg(arg, unsig long); \
302 else if (short_flag) \
303 res = (unsig short)va_arg(arg, unsig int); \
305 res = (unsig int)va_arg(arg, unsig int)
310 * zyxprintf - return length, as snprintf
314 xyzprintf (struct snprintf_state *state, const char *char_format, va_list ap)
316 const unsigned char *format = (const unsigned char *)char_format;
320 while((c = *format++)) {
325 int long_long_flag = 0;
330 while((c = *format++)){
338 flags |= alternate_flag;
347 if((flags & space_flag) && (flags & plus_flag))
350 if((flags & minus_flag) && (flags & zero_flag))
356 width = width * 10 + c - '0';
360 width = va_arg(ap, int);
370 prec = prec * 10 + c - '0';
374 prec = va_arg(ap, int);
384 } else if (c == 'l') {
395 append_char(state, va_arg(ap, int), width, flags);
399 len += append_string(state,
400 va_arg(ap, unsigned char*),
411 PARSE_INT_FORMAT(arg, ap, signed);
419 len += append_number (state, num, 10, "0123456789",
420 width, prec, flags, minusp);
426 PARSE_INT_FORMAT(arg, ap, unsigned);
428 len += append_number (state, arg, 10, "0123456789",
429 width, prec, flags, 0);
435 PARSE_INT_FORMAT(arg, ap, unsigned);
437 len += append_number (state, arg, 010, "01234567",
438 width, prec, flags, 0);
444 PARSE_INT_FORMAT(arg, ap, unsigned);
446 len += append_number (state, arg, 0x10, "0123456789abcdef",
447 width, prec, flags, 0);
453 PARSE_INT_FORMAT(arg, ap, unsigned);
455 len += append_number (state, arg, 0x10, "0123456789ABCDEF",
456 width, prec, flags, 0);
460 unsigned long arg = (unsigned long)va_arg(ap, void*);
462 len += append_number (state, arg, 0x10, "0123456789ABCDEF",
463 width, prec, flags, 0);
467 int *arg = va_arg(ap, int*);
468 *arg = state->s - state->str;
475 (*state->append_char)(state, c);
479 (*state->append_char)(state, '%');
480 (*state->append_char)(state, c);
485 (*state->append_char) (state, c);
492 #if !defined(HAVE_SNPRINTF) || defined(TEST_SNPRINTF)
494 snprintf (char *str, size_t sz, const char *format, ...)
499 va_start(args, format);
500 ret = vsnprintf (str, sz, format, args);
512 va_start(args, format);
513 ret2 = vsprintf (tmp, format, args);
515 if (ret != ret2 || strcmp(str, tmp))
525 #if !defined(HAVE_ASPRINTF) || defined(TEST_SNPRINTF)
527 asprintf (char **ret, const char *format, ...)
532 va_start(args, format);
533 val = vasprintf (ret, format, args);
540 tmp = malloc (val + 1);
544 va_start(args, format);
545 ret2 = vsprintf (tmp, format, args);
547 if (val != ret2 || strcmp(*ret, tmp))
557 #if !defined(HAVE_ASNPRINTF) || defined(TEST_SNPRINTF)
559 asnprintf (char **ret, size_t max_sz, const char *format, ...)
564 va_start(args, format);
565 val = vasnprintf (ret, max_sz, format, args);
571 tmp = malloc (val + 1);
575 ret2 = vsprintf (tmp, format, args);
576 if (val != ret2 || strcmp(*ret, tmp))
587 #if !defined(HAVE_VASPRINTF) || defined(TEST_SNPRINTF)
589 vasprintf (char **ret, const char *format, va_list args)
591 return vasnprintf (ret, 0, format, args);
596 #if !defined(HAVE_VASNPRINTF) || defined(TEST_SNPRINTF)
598 vasnprintf (char **ret, size_t max_sz, const char *format, va_list args)
601 struct snprintf_state state;
603 state.max_sz = max_sz;
605 state.str = malloc(state.sz);
606 if (state.str == NULL) {
611 state.theend = state.s + state.sz - 1;
612 state.append_char = as_append_char;
614 st = xyzprintf (&state, format, args);
623 tmp = realloc (state.str, st+1);
635 #if !defined(HAVE_VSNPRINTF) || defined(TEST_SNPRINTF)
637 vsnprintf (char *str, size_t sz, const char *format, va_list args)
639 struct snprintf_state state;
641 unsigned char *ustr = (unsigned char *)str;
647 state.theend = ustr + sz - (sz > 0);
648 state.append_char = sn_append_char;
650 ret = xyzprintf (&state, format, args);