2 * Copyright (c) 1995-2002 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
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
;
345 if((flags
& space_flag
) && (flags
& plus_flag
))
348 if((flags
& minus_flag
) && (flags
& zero_flag
))
354 width
= width
* 10 + c
- '0';
358 width
= va_arg(ap
, int);
368 prec
= prec
* 10 + c
- '0';
372 prec
= va_arg(ap
, int);
382 } else if (c
== 'l') {
393 append_char(state
, va_arg(ap
, int), width
, flags
);
397 len
+= append_string(state
,
398 va_arg(ap
, unsigned char*),
409 PARSE_INT_FORMAT(arg
, ap
, signed);
417 len
+= append_number (state
, num
, 10, "0123456789",
418 width
, prec
, flags
, minusp
);
424 PARSE_INT_FORMAT(arg
, ap
, unsigned);
426 len
+= append_number (state
, arg
, 10, "0123456789",
427 width
, prec
, flags
, 0);
433 PARSE_INT_FORMAT(arg
, ap
, unsigned);
435 len
+= append_number (state
, arg
, 010, "01234567",
436 width
, prec
, flags
, 0);
442 PARSE_INT_FORMAT(arg
, ap
, unsigned);
444 len
+= append_number (state
, arg
, 0x10, "0123456789abcdef",
445 width
, prec
, flags
, 0);
451 PARSE_INT_FORMAT(arg
, ap
, unsigned);
453 len
+= append_number (state
, arg
, 0x10, "0123456789ABCDEF",
454 width
, prec
, flags
, 0);
458 unsigned long arg
= (unsigned long)va_arg(ap
, void*);
460 len
+= append_number (state
, arg
, 0x10, "0123456789ABCDEF",
461 width
, prec
, flags
, 0);
465 int *arg
= va_arg(ap
, int*);
466 *arg
= state
->s
- state
->str
;
473 (*state
->append_char
)(state
, c
);
477 (*state
->append_char
)(state
, '%');
478 (*state
->append_char
)(state
, c
);
483 (*state
->append_char
) (state
, c
);
490 #if !defined(HAVE_SNPRINTF) || defined(TEST_SNPRINTF)
492 snprintf (char *str
, size_t sz
, const char *format
, ...)
497 va_start(args
, format
);
498 ret
= vsnprintf (str
, sz
, format
, args
);
510 va_start(args
, format
);
511 ret2
= vsprintf (tmp
, format
, args
);
513 if (ret
!= ret2
|| strcmp(str
, tmp
))
523 #if !defined(HAVE_ASPRINTF) || defined(TEST_SNPRINTF)
525 asprintf (char **ret
, const char *format
, ...)
530 va_start(args
, format
);
531 val
= vasprintf (ret
, format
, args
);
538 tmp
= malloc (val
+ 1);
542 va_start(args
, format
);
543 ret2
= vsprintf (tmp
, format
, args
);
545 if (val
!= ret2
|| strcmp(*ret
, tmp
))
555 #if !defined(HAVE_ASNPRINTF) || defined(TEST_SNPRINTF)
557 asnprintf (char **ret
, size_t max_sz
, const char *format
, ...)
562 va_start(args
, format
);
563 val
= vasnprintf (ret
, max_sz
, format
, args
);
569 tmp
= malloc (val
+ 1);
573 ret2
= vsprintf (tmp
, format
, args
);
574 if (val
!= ret2
|| strcmp(*ret
, tmp
))
585 #if !defined(HAVE_VASPRINTF) || defined(TEST_SNPRINTF)
587 vasprintf (char **ret
, const char *format
, va_list args
)
589 return vasnprintf (ret
, 0, format
, args
);
594 #if !defined(HAVE_VASNPRINTF) || defined(TEST_SNPRINTF)
596 vasnprintf (char **ret
, size_t max_sz
, const char *format
, va_list args
)
599 struct snprintf_state state
;
601 state
.max_sz
= max_sz
;
603 state
.str
= malloc(state
.sz
);
604 if (state
.str
== NULL
) {
609 state
.theend
= state
.s
+ state
.sz
- 1;
610 state
.append_char
= as_append_char
;
612 st
= xyzprintf (&state
, format
, args
);
621 tmp
= realloc (state
.str
, st
+1);
633 #if !defined(HAVE_VSNPRINTF) || defined(TEST_SNPRINTF)
635 vsnprintf (char *str
, size_t sz
, const char *format
, va_list args
)
637 struct snprintf_state state
;
639 unsigned char *ustr
= (unsigned char *)str
;
645 state
.theend
= ustr
+ sz
- (sz
> 0);
646 state
.append_char
= sn_append_char
;
648 ret
= xyzprintf (&state
, format
, args
);