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
55 struct snprintf_state
{
58 unsigned char *theend
;
61 void (*append_char
)(struct snprintf_state
*, unsigned char);
65 #if !defined(HAVE_VSNPRINTF) || defined(TEST_SNPRINTF)
67 sn_reserve (struct snprintf_state
*state
, size_t n
)
69 return state
->s
+ n
> state
->theend
;
73 sn_append_char (struct snprintf_state
*state
, unsigned char c
)
75 if (!sn_reserve (state
, 1))
81 as_reserve (struct snprintf_state
*state
, size_t n
)
83 if (state
->s
+ n
> state
->theend
) {
84 int off
= state
->s
- state
->str
;
87 if (state
->max_sz
&& state
->sz
>= state
->max_sz
)
90 state
->sz
= max(state
->sz
* 2, state
->sz
+ n
);
92 state
->sz
= min(state
->sz
, state
->max_sz
);
93 tmp
= realloc (state
->str
, state
->sz
);
97 state
->s
= state
->str
+ off
;
98 state
->theend
= state
->str
+ state
->sz
- 1;
104 as_append_char (struct snprintf_state
*state
, unsigned char c
)
106 if(!as_reserve (state
, 1))
110 /* longest integer types */
112 #ifdef HAVE_LONG_LONG
113 typedef unsigned long long u_longest
;
114 typedef long long longest
;
116 typedef unsigned long u_longest
;
117 typedef long longest
;
120 #ifndef HAVE_UINTPTR_T
121 typedef u_longest
uintptr_t;
127 pad(struct snprintf_state
*state
, int width
, char c
)
131 (*state
->append_char
)(state
, c
);
137 /* return true if we should use alternatve hex form */
139 use_alternative (int flags
, u_longest num
, unsigned base
)
141 return (flags
& alternate_flag
) && base
== 16 && num
!= 0;
145 append_number(struct snprintf_state
*state
,
146 u_longest num
, unsigned base
, const char *rep
,
147 int width
, int prec
, int flags
, int minusp
)
151 char nstr
[64]; /* enough for <192 bit octal integers */
155 /* given precision, ignore zero flag */
161 /* format number as string */
162 nstart
= sizeof(nstr
);
164 nstr
[--nstart
] = '\0';
167 nstr
[--nstart
] = rep
[n
% base
];
172 /* zero value with zero precision should produce no digits */
173 if(prec
== 0 && num
== 0) {
178 /* figure out what char to use for sign */
181 else if((flags
& plus_flag
))
183 else if((flags
& space_flag
))
188 if((flags
& alternate_flag
) && base
== 8) {
189 /* if necessary, increase the precision to
190 make first digit a zero */
192 /* XXX C99 claims (regarding # and %o) that "if the value and
193 precision are both 0, a single 0 is printed", but there is
194 no such wording for %x. This would mean that %#.o would
195 output "0", but %#.x "". This does not make sense, and is
196 also not what other printf implementations are doing. */
198 if(prec
<= nlen
&& nstr
[nstart
] != '0' && nstr
[nstart
] != '\0')
203 pad | sign | alt | zero | digits
204 sign | alt | zero | digits | pad minus_flag
205 sign | alt | zero | digits zero_flag */
207 /* if not right justifying or padding with zeros, we need to
208 compute the length of the rest of the string, and then pad with
210 if(!(flags
& (minus_flag
| zero_flag
))) {
216 if(use_alternative(flags
, num
, base
))
223 len
+= pad(state
, width
, ' ');
225 if(signchar
!= '\0') {
226 (*state
->append_char
)(state
, signchar
);
229 if(use_alternative(flags
, num
, base
)) {
230 (*state
->append_char
)(state
, '0');
231 (*state
->append_char
)(state
, rep
[10] + 23); /* XXX */
234 if(flags
& zero_flag
) {
235 /* pad to width with zeros */
236 if(prec
- nlen
> width
- len
- nlen
)
237 len
+= pad(state
, prec
- nlen
, '0');
239 len
+= pad(state
, width
- len
- nlen
, '0');
241 /* pad to prec with zeros */
242 len
+= pad(state
, prec
- nlen
, '0');
244 while(nstr
[nstart
] != '\0') {
245 (*state
->append_char
)(state
, nstr
[nstart
++]);
249 if(flags
& minus_flag
)
250 len
+= pad(state
, width
- len
, ' ');
260 append_string (struct snprintf_state
*state
,
261 const unsigned char *arg
,
269 arg
= (const unsigned char*)"(null)";
274 width
-= strlen((const char *)arg
);
275 if(!(flags
& minus_flag
))
276 len
+= pad(state
, width
, ' ');
279 while (prec
-- && *arg
) {
280 (*state
->append_char
) (state
, *arg
++);
285 (*state
->append_char
) (state
, *arg
++);
289 if(flags
& minus_flag
)
290 len
+= pad(state
, width
, ' ');
295 append_char(struct snprintf_state
*state
,
302 while(!(flags
& minus_flag
) && --width
> 0) {
303 (*state
->append_char
) (state
, ' ') ;
306 (*state
->append_char
) (state
, arg
);
308 while((flags
& minus_flag
) && --width
> 0) {
309 (*state
->append_char
) (state
, ' ');
316 * This can't be made into a function...
319 #ifdef HAVE_LONG_LONG
321 #define PARSE_INT_FORMAT(res, arg, unsig) \
322 if (long_long_flag) \
323 res = (unsig long long)va_arg(arg, unsig long long); \
324 else if (long_flag) \
325 res = (unsig long)va_arg(arg, unsig long); \
326 else if (size_t_flag) \
327 res = (unsig long)va_arg(arg, size_t); \
328 else if (short_flag) \
329 res = (unsig short)va_arg(arg, unsig int); \
331 res = (unsig int)va_arg(arg, unsig int)
335 #define PARSE_INT_FORMAT(res, arg, unsig) \
337 res = (unsig long)va_arg(arg, unsig long); \
338 else if (size_t_flag) \
339 res = (unsig long)va_arg(arg, size_t); \
340 else if (short_flag) \
341 res = (unsig short)va_arg(arg, unsig int); \
343 res = (unsig int)va_arg(arg, unsig int)
348 * zyxprintf - return length, as snprintf
352 xyzprintf (struct snprintf_state
*state
, const char *char_format
, va_list ap
)
354 const unsigned char *format
= (const unsigned char *)char_format
;
358 while((c
= *format
++)) {
364 int long_long_flag
= 0;
369 while((c
= *format
++)){
377 flags
|= alternate_flag
;
386 if((flags
& space_flag
) && (flags
& plus_flag
))
389 if((flags
& minus_flag
) && (flags
& zero_flag
))
395 width
= width
* 10 + c
- '0';
399 width
= va_arg(ap
, int);
409 prec
= prec
* 10 + c
- '0';
413 prec
= va_arg(ap
, int);
423 } else if (c
== 'z') {
426 } else if (c
== 'l') {
435 if(c
!= 'd' && c
!= 'i')
436 flags
&= ~(plus_flag
| space_flag
);
440 append_char(state
, va_arg(ap
, int), width
, flags
);
444 len
+= append_string(state
,
445 va_arg(ap
, unsigned char*),
456 PARSE_INT_FORMAT(arg
, ap
, signed);
464 len
+= append_number (state
, num
, 10, "0123456789",
465 width
, prec
, flags
, minusp
);
471 PARSE_INT_FORMAT(arg
, ap
, unsigned);
473 len
+= append_number (state
, arg
, 10, "0123456789",
474 width
, prec
, flags
, 0);
480 PARSE_INT_FORMAT(arg
, ap
, unsigned);
482 len
+= append_number (state
, arg
, 010, "01234567",
483 width
, prec
, flags
, 0);
489 PARSE_INT_FORMAT(arg
, ap
, unsigned);
491 len
+= append_number (state
, arg
, 0x10, "0123456789abcdef",
492 width
, prec
, flags
, 0);
498 PARSE_INT_FORMAT(arg
, ap
, unsigned);
500 len
+= append_number (state
, arg
, 0x10, "0123456789ABCDEF",
501 width
, prec
, flags
, 0);
505 uintptr_t arg
= (uintptr_t)va_arg(ap
, void*);
507 len
+= append_number (state
, arg
, 0x10, "0123456789ABCDEF",
508 width
, prec
, flags
, 0);
512 int *arg
= va_arg(ap
, int*);
513 *arg
= state
->s
- state
->str
;
520 (*state
->append_char
)(state
, c
);
524 (*state
->append_char
)(state
, '%');
525 (*state
->append_char
)(state
, c
);
530 (*state
->append_char
) (state
, c
);
537 #if !defined(HAVE_SNPRINTF) || defined(TEST_SNPRINTF)
538 ROKEN_LIB_FUNCTION
int ROKEN_LIB_CALL
539 rk_snprintf (char *str
, size_t sz
, const char *format
, ...)
544 va_start(args
, format
);
545 ret
= vsnprintf (str
, sz
, format
, args
);
557 va_start(args
, format
);
558 ret2
= vsprintf (tmp
, format
, args
);
560 if (ret
!= ret2
|| strcmp(str
, tmp
) != 0)
570 #if !defined(HAVE_ASPRINTF) || defined(TEST_SNPRINTF)
571 ROKEN_LIB_FUNCTION
int ROKEN_LIB_CALL
572 rk_asprintf (char **ret
, const char *format
, ...)
577 va_start(args
, format
);
578 val
= vasprintf (ret
, format
, args
);
585 tmp
= malloc (val
+ 1);
589 va_start(args
, format
);
590 ret2
= vsprintf (tmp
, format
, args
);
592 if (val
!= ret2
|| strcmp(*ret
, tmp
) != 0)
602 #if !defined(HAVE_ASNPRINTF) || defined(TEST_SNPRINTF)
603 ROKEN_LIB_FUNCTION
int ROKEN_LIB_CALL
604 rk_asnprintf (char **ret
, size_t max_sz
, const char *format
, ...)
609 va_start(args
, format
);
610 val
= vasnprintf (ret
, max_sz
, format
, args
);
616 tmp
= malloc (val
+ 1);
620 ret2
= vsprintf (tmp
, format
, args
);
621 if (val
!= ret2
|| strcmp(*ret
, tmp
) != 0)
632 #if !defined(HAVE_VASPRINTF) || defined(TEST_SNPRINTF)
633 ROKEN_LIB_FUNCTION
int ROKEN_LIB_CALL
634 rk_vasprintf (char **ret
, const char *format
, va_list args
)
636 return vasnprintf (ret
, 0, format
, args
);
641 #if !defined(HAVE_VASNPRINTF) || defined(TEST_SNPRINTF)
642 ROKEN_LIB_FUNCTION
int ROKEN_LIB_CALL
643 rk_vasnprintf (char **ret
, size_t max_sz
, const char *format
, va_list args
)
646 struct snprintf_state state
;
648 state
.max_sz
= max_sz
;
650 state
.str
= malloc(state
.sz
);
651 if (state
.str
== NULL
) {
656 state
.theend
= state
.s
+ state
.sz
- 1;
657 state
.append_char
= as_append_char
;
659 st
= xyzprintf (&state
, format
, args
);
668 tmp
= realloc (state
.str
, st
+1);
680 #if !defined(HAVE_VSNPRINTF) || defined(TEST_SNPRINTF)
681 ROKEN_LIB_FUNCTION
int ROKEN_LIB_CALL
682 rk_vsnprintf (char *str
, size_t sz
, const char *format
, va_list args
)
684 struct snprintf_state state
;
686 unsigned char *ustr
= (unsigned char *)str
;
692 state
.theend
= ustr
+ sz
- (sz
> 0);
693 state
.append_char
= sn_append_char
;
695 ret
= xyzprintf (&state
, format
, args
);
696 if (state
.s
!= NULL
&& sz
!= 0)