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
42 #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;