2 * Copyright (c) 1995-2000 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
34 #include <afsconfig.h>
44 #ifndef HAVE_VASNPRINTF
45 int vasnprintf(char **ret
, size_t max_sz
, const char *format
, va_list args
);
47 #ifndef HAVE_VASPRINTF
48 int vasprintf(char**ret
, const char *format
, va_list args
);
52 #define min(a, b) ((a) > (b) ? (b) : (a))
55 #define max(a, b) ((a) < (b) ? (b) : (a))
73 unsigned char *theend
;
76 int (*append_char
) (struct state
*, unsigned char);
77 int (*reserve
) (struct state
*, size_t);
81 #ifndef HAVE_VSNPRINTF
83 sn_reserve(struct state
*state
, size_t n
)
85 return state
->s
+ n
> state
->theend
;
89 sn_append_char(struct state
*state
, unsigned char c
)
91 if (sn_reserve(state
, 1)) {
101 as_reserve(struct state
*state
, size_t n
)
103 if (state
->s
+ n
> state
->theend
) {
104 int off
= state
->s
- state
->str
;
107 if (state
->max_sz
&& state
->sz
>= state
->max_sz
)
110 state
->sz
= max(state
->sz
* 2, state
->sz
+ n
);
112 state
->sz
= min(state
->sz
, state
->max_sz
);
113 tmp
= realloc(state
->str
, state
->sz
);
117 state
->s
= state
->str
+ off
;
118 state
->theend
= state
->str
+ state
->sz
- 1;
124 as_append_char(struct state
*state
, unsigned char c
)
126 if (as_reserve(state
, 1))
135 append_number(struct state
*state
, unsigned long num
, unsigned base
,
136 char *rep
, int width
, int prec
, int flags
, int minusp
)
141 /* given precision, ignore zero flag */
146 /* zero value with zero precision -> "" */
147 if (prec
== 0 && num
== 0)
150 if ((*state
->append_char
) (state
, rep
[num
% base
]))
156 /* pad with prec zeros */
158 if ((*state
->append_char
) (state
, '0'))
162 /* add length of alternate prefix (added later) to len */
163 if (flags
& alternate_flag
&& (base
== 16 || base
== 8))
166 if (flags
& zero_flag
) {
168 if (minusp
|| (flags
& space_flag
) || (flags
& plus_flag
))
170 while (width
-- > 0) {
171 if ((*state
->append_char
) (state
, '0'))
176 /* add alternate prefix */
177 if (flags
& alternate_flag
&& (base
== 16 || base
== 8)) {
179 if ((*state
->append_char
) (state
, rep
[10] + 23)) /* XXX */
181 if ((*state
->append_char
) (state
, '0'))
186 if ((*state
->append_char
) (state
, '-'))
189 } else if (flags
& plus_flag
) {
190 if ((*state
->append_char
) (state
, '+'))
193 } else if (flags
& space_flag
) {
194 if ((*state
->append_char
) (state
, ' '))
198 if (flags
& minus_flag
)
199 /* swap before padding with spaces */
200 for (i
= 0; i
< len
/ 2; i
++) {
201 char c
= state
->s
[-i
- 1];
202 state
->s
[-i
- 1] = state
->s
[-len
+ i
];
203 state
->s
[-len
+ i
] = c
;
206 while (width
-- > 0) {
207 if ((*state
->append_char
) (state
, ' '))
211 if (!(flags
& minus_flag
))
212 /* swap after padding with spaces */
213 for (i
= 0; i
< len
/ 2; i
++) {
214 char c
= state
->s
[-i
- 1];
215 state
->s
[-i
- 1] = state
->s
[-len
+ i
];
216 state
->s
[-len
+ i
] = c
;
223 append_string(struct state
*state
, unsigned char *arg
, int width
, int prec
,
227 arg
= (unsigned char *)"(null)";
232 width
-= strlen((char *)arg
);
233 if (!(flags
& minus_flag
))
235 if ((*state
->append_char
) (state
, ' '))
238 while (*arg
&& prec
--)
239 if ((*state
->append_char
) (state
, *arg
++))
243 if ((*state
->append_char
) (state
, *arg
++))
246 if (flags
& minus_flag
)
248 if ((*state
->append_char
) (state
, ' '))
254 append_char(struct state
*state
, unsigned char arg
, int width
, int flags
)
256 while (!(flags
& minus_flag
) && --width
> 0)
257 if ((*state
->append_char
) (state
, ' '))
260 if ((*state
->append_char
) (state
, arg
))
262 while ((flags
& minus_flag
) && --width
> 0)
263 if ((*state
->append_char
) (state
, ' '))
270 * This can't be made into a function...
273 #define PARSE_INT_FORMAT(res, arg, unsig) \
275 res = (unsig long)va_arg(arg, unsig long); \
276 else if (short_flag) \
277 res = (unsig short)va_arg(arg, unsig int); \
279 res = (unsig int)va_arg(arg, unsig int)
282 * zyxprintf - return 0 or -1
286 xyzprintf(struct state
*state
, const char *char_format
, va_list ap
)
288 const unsigned char *format
= (const unsigned char *)char_format
;
291 while ((c
= *format
++)) {
300 while ((c
= *format
++)) {
308 flags
|= alternate_flag
;
315 if ((flags
& space_flag
) && (flags
& plus_flag
))
318 if ((flags
& minus_flag
) && (flags
& zero_flag
))
324 width
= width
* 10 + c
- '0';
326 } while (isdigit(c
));
328 width
= va_arg(ap
, int);
338 prec
= prec
* 10 + c
- '0';
340 } while (isdigit(c
));
342 prec
= va_arg(ap
, int);
352 } else if (c
== 'l') {
359 if (append_char(state
, va_arg(ap
, int), width
, flags
))
364 (state
, va_arg(ap
, unsigned char *), width
, prec
, flags
))
373 PARSE_INT_FORMAT(arg
, ap
, signed);
382 (state
, num
, 10, "0123456789", width
, prec
, flags
,
390 PARSE_INT_FORMAT(arg
, ap
, unsigned);
393 (state
, arg
, 10, "0123456789", width
, prec
, flags
, 0))
400 PARSE_INT_FORMAT(arg
, ap
, unsigned);
403 (state
, arg
, 010, "01234567", width
, prec
, flags
, 0))
410 PARSE_INT_FORMAT(arg
, ap
, unsigned);
413 (state
, arg
, 0x10, "0123456789abcdef", width
, prec
,
421 PARSE_INT_FORMAT(arg
, ap
, unsigned);
424 (state
, arg
, 0x10, "0123456789ABCDEF", width
, prec
,
430 unsigned long arg
= (unsigned long)va_arg(ap
, void *);
433 (state
, arg
, 0x10, "0123456789ABCDEF", width
, prec
,
439 int *arg
= va_arg(ap
, int *);
440 *arg
= state
->s
- state
->str
;
447 if ((*state
->append_char
) (state
, c
))
451 if ((*state
->append_char
) (state
, '%')
452 || (*state
->append_char
) (state
, c
))
456 } else if ((*state
->append_char
) (state
, c
))
462 #ifndef HAVE_SNPRINTF
464 snprintf(char *str
, size_t sz
, const char *format
, ...)
469 va_start(args
, format
);
470 ret
= vsnprintf(str
, sz
, format
, args
);
481 ret2
= vsprintf(tmp
, format
, args
);
482 if (ret
!= ret2
|| strcmp(str
, tmp
))
493 #ifndef HAVE_ASPRINTF
495 asprintf(char **ret
, const char *format
, ...)
500 va_start(args
, format
);
501 val
= vasprintf(ret
, format
, args
);
507 tmp
= malloc(val
+ 1);
511 ret2
= vsprintf(tmp
, format
, args
);
512 if (val
!= ret2
|| strcmp(*ret
, tmp
))
523 #ifndef HAVE_ASNPRINTF
525 asnprintf(char **ret
, size_t max_sz
, const char *format
, ...)
530 va_start(args
, format
);
531 val
= vasnprintf(ret
, max_sz
, format
, args
);
537 tmp
= malloc(val
+ 1);
541 ret2
= vsprintf(tmp
, format
, args
);
542 if (val
!= ret2
|| strcmp(*ret
, tmp
))
553 #ifndef HAVE_VASPRINTF
555 vasprintf(char **ret
, const char *format
, va_list args
)
557 return vasnprintf(ret
, 0, format
, args
);
562 #ifndef HAVE_VASNPRINTF
564 vasnprintf(char **ret
, size_t max_sz
, const char *format
, va_list args
)
570 state
.max_sz
= max_sz
;
572 state
.str
= malloc(state
.sz
);
573 if (state
.str
== NULL
) {
578 state
.theend
= state
.s
+ state
.sz
- 1;
579 state
.append_char
= as_append_char
;
580 state
.reserve
= as_reserve
;
582 st
= xyzprintf(&state
, format
, args
);
591 len
= state
.s
- state
.str
;
592 tmp
= realloc(state
.str
, len
+ 1);
604 #ifndef HAVE_VSNPRINTF
606 vsnprintf(char *str
, size_t sz
, const char *format
, va_list args
)
610 unsigned char *ustr
= (unsigned char *)str
;
616 state
.theend
= ustr
+ sz
- 1;
617 state
.append_char
= sn_append_char
;
618 state
.reserve
= sn_reserve
;
620 ret
= xyzprintf(&state
, format
, args
);
625 return state
.s
- state
.str
;