]> git.ipfire.org Git - people/ms/u-boot.git/blob - lib/vsprintf.c
NAND: Fix integer overflow in ONFI detection of chips >= 4GiB
[people/ms/u-boot.git] / lib / vsprintf.c
1 /*
2 * linux/lib/vsprintf.c
3 *
4 * Copyright (C) 1991, 1992 Linus Torvalds
5 */
6
7 /* vsprintf.c -- Lars Wirzenius & Linus Torvalds. */
8 /*
9 * Wirzenius wrote this portably, Torvalds fucked it up :-)
10 */
11
12 #include <stdarg.h>
13 #include <linux/types.h>
14 #include <linux/string.h>
15 #include <linux/ctype.h>
16
17 #include <common.h>
18 #if !defined (CONFIG_PANIC_HANG)
19 #include <command.h>
20 #endif
21
22 #include <div64.h>
23 # define NUM_TYPE long long
24 #define noinline __attribute__((noinline))
25
26 const char hex_asc[] = "0123456789abcdef";
27 #define hex_asc_lo(x) hex_asc[((x) & 0x0f)]
28 #define hex_asc_hi(x) hex_asc[((x) & 0xf0) >> 4]
29
30 static inline char *pack_hex_byte(char *buf, u8 byte)
31 {
32 *buf++ = hex_asc_hi(byte);
33 *buf++ = hex_asc_lo(byte);
34 return buf;
35 }
36
37 unsigned long simple_strtoul(const char *cp,char **endp,unsigned int base)
38 {
39 unsigned long result = 0,value;
40
41 if (*cp == '0') {
42 cp++;
43 if ((*cp == 'x') && isxdigit(cp[1])) {
44 base = 16;
45 cp++;
46 }
47 if (!base) {
48 base = 8;
49 }
50 }
51 if (!base) {
52 base = 10;
53 }
54 while (isxdigit(*cp) && (value = isdigit(*cp) ? *cp-'0' : (islower(*cp)
55 ? toupper(*cp) : *cp)-'A'+10) < base) {
56 result = result*base + value;
57 cp++;
58 }
59 if (endp)
60 *endp = (char *)cp;
61 return result;
62 }
63
64 long simple_strtol(const char *cp,char **endp,unsigned int base)
65 {
66 if(*cp=='-')
67 return -simple_strtoul(cp+1,endp,base);
68 return simple_strtoul(cp,endp,base);
69 }
70
71 int ustrtoul(const char *cp, char **endp, unsigned int base)
72 {
73 unsigned long result = simple_strtoul(cp, endp, base);
74 switch (**endp) {
75 case 'G' :
76 result *= 1024;
77 /* fall through */
78 case 'M':
79 result *= 1024;
80 /* fall through */
81 case 'K':
82 case 'k':
83 result *= 1024;
84 if ((*endp)[1] == 'i') {
85 if ((*endp)[2] == 'B')
86 (*endp) += 3;
87 else
88 (*endp) += 2;
89 }
90 }
91 return result;
92 }
93
94 unsigned long long simple_strtoull (const char *cp, char **endp, unsigned int base)
95 {
96 unsigned long long result = 0, value;
97
98 if (*cp == '0') {
99 cp++;
100 if ((*cp == 'x') && isxdigit (cp[1])) {
101 base = 16;
102 cp++;
103 }
104 if (!base) {
105 base = 8;
106 }
107 }
108 if (!base) {
109 base = 10;
110 }
111 while (isxdigit (*cp) && (value = isdigit (*cp)
112 ? *cp - '0'
113 : (islower (*cp) ? toupper (*cp) : *cp) - 'A' + 10) < base) {
114 result = result * base + value;
115 cp++;
116 }
117 if (endp)
118 *endp = (char *) cp;
119 return result;
120 }
121
122 /* we use this so that we can do without the ctype library */
123 #define is_digit(c) ((c) >= '0' && (c) <= '9')
124
125 static int skip_atoi(const char **s)
126 {
127 int i=0;
128
129 while (is_digit(**s))
130 i = i*10 + *((*s)++) - '0';
131 return i;
132 }
133
134 /* Decimal conversion is by far the most typical, and is used
135 * for /proc and /sys data. This directly impacts e.g. top performance
136 * with many processes running. We optimize it for speed
137 * using code from
138 * http://www.cs.uiowa.edu/~jones/bcd/decimal.html
139 * (with permission from the author, Douglas W. Jones). */
140
141 /* Formats correctly any integer in [0,99999].
142 * Outputs from one to five digits depending on input.
143 * On i386 gcc 4.1.2 -O2: ~250 bytes of code. */
144 static char* put_dec_trunc(char *buf, unsigned q)
145 {
146 unsigned d3, d2, d1, d0;
147 d1 = (q>>4) & 0xf;
148 d2 = (q>>8) & 0xf;
149 d3 = (q>>12);
150
151 d0 = 6*(d3 + d2 + d1) + (q & 0xf);
152 q = (d0 * 0xcd) >> 11;
153 d0 = d0 - 10*q;
154 *buf++ = d0 + '0'; /* least significant digit */
155 d1 = q + 9*d3 + 5*d2 + d1;
156 if (d1 != 0) {
157 q = (d1 * 0xcd) >> 11;
158 d1 = d1 - 10*q;
159 *buf++ = d1 + '0'; /* next digit */
160
161 d2 = q + 2*d2;
162 if ((d2 != 0) || (d3 != 0)) {
163 q = (d2 * 0xd) >> 7;
164 d2 = d2 - 10*q;
165 *buf++ = d2 + '0'; /* next digit */
166
167 d3 = q + 4*d3;
168 if (d3 != 0) {
169 q = (d3 * 0xcd) >> 11;
170 d3 = d3 - 10*q;
171 *buf++ = d3 + '0'; /* next digit */
172 if (q != 0)
173 *buf++ = q + '0'; /* most sign. digit */
174 }
175 }
176 }
177 return buf;
178 }
179 /* Same with if's removed. Always emits five digits */
180 static char* put_dec_full(char *buf, unsigned q)
181 {
182 /* BTW, if q is in [0,9999], 8-bit ints will be enough, */
183 /* but anyway, gcc produces better code with full-sized ints */
184 unsigned d3, d2, d1, d0;
185 d1 = (q>>4) & 0xf;
186 d2 = (q>>8) & 0xf;
187 d3 = (q>>12);
188
189 /*
190 * Possible ways to approx. divide by 10
191 * gcc -O2 replaces multiply with shifts and adds
192 * (x * 0xcd) >> 11: 11001101 - shorter code than * 0x67 (on i386)
193 * (x * 0x67) >> 10: 1100111
194 * (x * 0x34) >> 9: 110100 - same
195 * (x * 0x1a) >> 8: 11010 - same
196 * (x * 0x0d) >> 7: 1101 - same, shortest code (on i386)
197 */
198
199 d0 = 6*(d3 + d2 + d1) + (q & 0xf);
200 q = (d0 * 0xcd) >> 11;
201 d0 = d0 - 10*q;
202 *buf++ = d0 + '0';
203 d1 = q + 9*d3 + 5*d2 + d1;
204 q = (d1 * 0xcd) >> 11;
205 d1 = d1 - 10*q;
206 *buf++ = d1 + '0';
207
208 d2 = q + 2*d2;
209 q = (d2 * 0xd) >> 7;
210 d2 = d2 - 10*q;
211 *buf++ = d2 + '0';
212
213 d3 = q + 4*d3;
214 q = (d3 * 0xcd) >> 11; /* - shorter code */
215 /* q = (d3 * 0x67) >> 10; - would also work */
216 d3 = d3 - 10*q;
217 *buf++ = d3 + '0';
218 *buf++ = q + '0';
219 return buf;
220 }
221 /* No inlining helps gcc to use registers better */
222 static noinline char* put_dec(char *buf, unsigned NUM_TYPE num)
223 {
224 while (1) {
225 unsigned rem;
226 if (num < 100000)
227 return put_dec_trunc(buf, num);
228 rem = do_div(num, 100000);
229 buf = put_dec_full(buf, rem);
230 }
231 }
232
233 #define ZEROPAD 1 /* pad with zero */
234 #define SIGN 2 /* unsigned/signed long */
235 #define PLUS 4 /* show plus */
236 #define SPACE 8 /* space if plus */
237 #define LEFT 16 /* left justified */
238 #define SMALL 32 /* Must be 32 == 0x20 */
239 #define SPECIAL 64 /* 0x */
240
241 static char *number(char *buf, unsigned NUM_TYPE num, int base, int size, int precision, int type)
242 {
243 /* we are called with base 8, 10 or 16, only, thus don't need "G..." */
244 static const char digits[16] = "0123456789ABCDEF"; /* "GHIJKLMNOPQRSTUVWXYZ"; */
245
246 char tmp[66];
247 char sign;
248 char locase;
249 int need_pfx = ((type & SPECIAL) && base != 10);
250 int i;
251
252 /* locase = 0 or 0x20. ORing digits or letters with 'locase'
253 * produces same digits or (maybe lowercased) letters */
254 locase = (type & SMALL);
255 if (type & LEFT)
256 type &= ~ZEROPAD;
257 sign = 0;
258 if (type & SIGN) {
259 if ((signed NUM_TYPE) num < 0) {
260 sign = '-';
261 num = - (signed NUM_TYPE) num;
262 size--;
263 } else if (type & PLUS) {
264 sign = '+';
265 size--;
266 } else if (type & SPACE) {
267 sign = ' ';
268 size--;
269 }
270 }
271 if (need_pfx) {
272 size--;
273 if (base == 16)
274 size--;
275 }
276
277 /* generate full string in tmp[], in reverse order */
278 i = 0;
279 if (num == 0)
280 tmp[i++] = '0';
281 /* Generic code, for any base:
282 else do {
283 tmp[i++] = (digits[do_div(num,base)] | locase);
284 } while (num != 0);
285 */
286 else if (base != 10) { /* 8 or 16 */
287 int mask = base - 1;
288 int shift = 3;
289 if (base == 16) shift = 4;
290 do {
291 tmp[i++] = (digits[((unsigned char)num) & mask] | locase);
292 num >>= shift;
293 } while (num);
294 } else { /* base 10 */
295 i = put_dec(tmp, num) - tmp;
296 }
297
298 /* printing 100 using %2d gives "100", not "00" */
299 if (i > precision)
300 precision = i;
301 /* leading space padding */
302 size -= precision;
303 if (!(type & (ZEROPAD+LEFT)))
304 while(--size >= 0)
305 *buf++ = ' ';
306 /* sign */
307 if (sign)
308 *buf++ = sign;
309 /* "0x" / "0" prefix */
310 if (need_pfx) {
311 *buf++ = '0';
312 if (base == 16)
313 *buf++ = ('X' | locase);
314 }
315 /* zero or space padding */
316 if (!(type & LEFT)) {
317 char c = (type & ZEROPAD) ? '0' : ' ';
318 while (--size >= 0)
319 *buf++ = c;
320 }
321 /* hmm even more zero padding? */
322 while (i <= --precision)
323 *buf++ = '0';
324 /* actual digits of result */
325 while (--i >= 0)
326 *buf++ = tmp[i];
327 /* trailing space padding */
328 while (--size >= 0)
329 *buf++ = ' ';
330 return buf;
331 }
332
333 static char *string(char *buf, char *s, int field_width, int precision, int flags)
334 {
335 int len, i;
336
337 if (s == 0)
338 s = "<NULL>";
339
340 len = strnlen(s, precision);
341
342 if (!(flags & LEFT))
343 while (len < field_width--)
344 *buf++ = ' ';
345 for (i = 0; i < len; ++i)
346 *buf++ = *s++;
347 while (len < field_width--)
348 *buf++ = ' ';
349 return buf;
350 }
351
352 #ifdef CONFIG_CMD_NET
353 static char *mac_address_string(char *buf, u8 *addr, int field_width,
354 int precision, int flags)
355 {
356 char mac_addr[6 * 3]; /* (6 * 2 hex digits), 5 colons and trailing zero */
357 char *p = mac_addr;
358 int i;
359
360 for (i = 0; i < 6; i++) {
361 p = pack_hex_byte(p, addr[i]);
362 if (!(flags & SPECIAL) && i != 5)
363 *p++ = ':';
364 }
365 *p = '\0';
366
367 return string(buf, mac_addr, field_width, precision, flags & ~SPECIAL);
368 }
369
370 static char *ip6_addr_string(char *buf, u8 *addr, int field_width,
371 int precision, int flags)
372 {
373 char ip6_addr[8 * 5]; /* (8 * 4 hex digits), 7 colons and trailing zero */
374 char *p = ip6_addr;
375 int i;
376
377 for (i = 0; i < 8; i++) {
378 p = pack_hex_byte(p, addr[2 * i]);
379 p = pack_hex_byte(p, addr[2 * i + 1]);
380 if (!(flags & SPECIAL) && i != 7)
381 *p++ = ':';
382 }
383 *p = '\0';
384
385 return string(buf, ip6_addr, field_width, precision, flags & ~SPECIAL);
386 }
387
388 static char *ip4_addr_string(char *buf, u8 *addr, int field_width,
389 int precision, int flags)
390 {
391 char ip4_addr[4 * 4]; /* (4 * 3 decimal digits), 3 dots and trailing zero */
392 char temp[3]; /* hold each IP quad in reverse order */
393 char *p = ip4_addr;
394 int i, digits;
395
396 for (i = 0; i < 4; i++) {
397 digits = put_dec_trunc(temp, addr[i]) - temp;
398 /* reverse the digits in the quad */
399 while (digits--)
400 *p++ = temp[digits];
401 if (i != 3)
402 *p++ = '.';
403 }
404 *p = '\0';
405
406 return string(buf, ip4_addr, field_width, precision, flags & ~SPECIAL);
407 }
408 #endif
409
410 /*
411 * Show a '%p' thing. A kernel extension is that the '%p' is followed
412 * by an extra set of alphanumeric characters that are extended format
413 * specifiers.
414 *
415 * Right now we handle:
416 *
417 * - 'M' For a 6-byte MAC address, it prints the address in the
418 * usual colon-separated hex notation
419 * - 'I' [46] for IPv4/IPv6 addresses printed in the usual way (dot-separated
420 * decimal for v4 and colon separated network-order 16 bit hex for v6)
421 * - 'i' [46] for 'raw' IPv4/IPv6 addresses, IPv6 omits the colons, IPv4 is
422 * currently the same
423 *
424 * Note: The difference between 'S' and 'F' is that on ia64 and ppc64
425 * function pointers are really function descriptors, which contain a
426 * pointer to the real address.
427 */
428 static char *pointer(const char *fmt, char *buf, void *ptr, int field_width, int precision, int flags)
429 {
430 if (!ptr)
431 return string(buf, "(null)", field_width, precision, flags);
432
433 #ifdef CONFIG_CMD_NET
434 switch (*fmt) {
435 case 'm':
436 flags |= SPECIAL;
437 /* Fallthrough */
438 case 'M':
439 return mac_address_string(buf, ptr, field_width, precision, flags);
440 case 'i':
441 flags |= SPECIAL;
442 /* Fallthrough */
443 case 'I':
444 if (fmt[1] == '6')
445 return ip6_addr_string(buf, ptr, field_width, precision, flags);
446 if (fmt[1] == '4')
447 return ip4_addr_string(buf, ptr, field_width, precision, flags);
448 flags &= ~SPECIAL;
449 break;
450 }
451 #endif
452 flags |= SMALL;
453 if (field_width == -1) {
454 field_width = 2*sizeof(void *);
455 flags |= ZEROPAD;
456 }
457 return number(buf, (unsigned long) ptr, 16, field_width, precision, flags);
458 }
459
460 /**
461 * vsprintf - Format a string and place it in a buffer
462 * @buf: The buffer to place the result into
463 * @fmt: The format string to use
464 * @args: Arguments for the format string
465 *
466 * This function follows C99 vsprintf, but has some extensions:
467 * %pS output the name of a text symbol
468 * %pF output the name of a function pointer
469 * %pR output the address range in a struct resource
470 *
471 * The function returns the number of characters written
472 * into @buf.
473 *
474 * Call this function if you are already dealing with a va_list.
475 * You probably want sprintf() instead.
476 */
477 int vsprintf(char *buf, const char *fmt, va_list args)
478 {
479 unsigned NUM_TYPE num;
480 int base;
481 char *str;
482
483 int flags; /* flags to number() */
484
485 int field_width; /* width of output field */
486 int precision; /* min. # of digits for integers; max
487 number of chars for from string */
488 int qualifier; /* 'h', 'l', or 'L' for integer fields */
489 /* 'z' support added 23/7/1999 S.H. */
490 /* 'z' changed to 'Z' --davidm 1/25/99 */
491 /* 't' added for ptrdiff_t */
492
493 str = buf;
494
495 for (; *fmt ; ++fmt) {
496 if (*fmt != '%') {
497 *str++ = *fmt;
498 continue;
499 }
500
501 /* process flags */
502 flags = 0;
503 repeat:
504 ++fmt; /* this also skips first '%' */
505 switch (*fmt) {
506 case '-': flags |= LEFT; goto repeat;
507 case '+': flags |= PLUS; goto repeat;
508 case ' ': flags |= SPACE; goto repeat;
509 case '#': flags |= SPECIAL; goto repeat;
510 case '0': flags |= ZEROPAD; goto repeat;
511 }
512
513 /* get field width */
514 field_width = -1;
515 if (is_digit(*fmt))
516 field_width = skip_atoi(&fmt);
517 else if (*fmt == '*') {
518 ++fmt;
519 /* it's the next argument */
520 field_width = va_arg(args, int);
521 if (field_width < 0) {
522 field_width = -field_width;
523 flags |= LEFT;
524 }
525 }
526
527 /* get the precision */
528 precision = -1;
529 if (*fmt == '.') {
530 ++fmt;
531 if (is_digit(*fmt))
532 precision = skip_atoi(&fmt);
533 else if (*fmt == '*') {
534 ++fmt;
535 /* it's the next argument */
536 precision = va_arg(args, int);
537 }
538 if (precision < 0)
539 precision = 0;
540 }
541
542 /* get the conversion qualifier */
543 qualifier = -1;
544 if (*fmt == 'h' || *fmt == 'l' || *fmt == 'L' ||
545 *fmt == 'Z' || *fmt == 'z' || *fmt == 't') {
546 qualifier = *fmt;
547 ++fmt;
548 if (qualifier == 'l' && *fmt == 'l') {
549 qualifier = 'L';
550 ++fmt;
551 }
552 }
553
554 /* default base */
555 base = 10;
556
557 switch (*fmt) {
558 case 'c':
559 if (!(flags & LEFT))
560 while (--field_width > 0)
561 *str++ = ' ';
562 *str++ = (unsigned char) va_arg(args, int);
563 while (--field_width > 0)
564 *str++ = ' ';
565 continue;
566
567 case 's':
568 str = string(str, va_arg(args, char *), field_width, precision, flags);
569 continue;
570
571 case 'p':
572 str = pointer(fmt+1, str,
573 va_arg(args, void *),
574 field_width, precision, flags);
575 /* Skip all alphanumeric pointer suffixes */
576 while (isalnum(fmt[1]))
577 fmt++;
578 continue;
579
580 case 'n':
581 if (qualifier == 'l') {
582 long * ip = va_arg(args, long *);
583 *ip = (str - buf);
584 } else {
585 int * ip = va_arg(args, int *);
586 *ip = (str - buf);
587 }
588 continue;
589
590 case '%':
591 *str++ = '%';
592 continue;
593
594 /* integer number formats - set up the flags and "break" */
595 case 'o':
596 base = 8;
597 break;
598
599 case 'x':
600 flags |= SMALL;
601 case 'X':
602 base = 16;
603 break;
604
605 case 'd':
606 case 'i':
607 flags |= SIGN;
608 case 'u':
609 break;
610
611 default:
612 *str++ = '%';
613 if (*fmt)
614 *str++ = *fmt;
615 else
616 --fmt;
617 continue;
618 }
619 if (qualifier == 'L') /* "quad" for 64 bit variables */
620 num = va_arg(args, unsigned long long);
621 else if (qualifier == 'l') {
622 num = va_arg(args, unsigned long);
623 if (flags & SIGN)
624 num = (signed long) num;
625 } else if (qualifier == 'Z' || qualifier == 'z') {
626 num = va_arg(args, size_t);
627 } else if (qualifier == 't') {
628 num = va_arg(args, ptrdiff_t);
629 } else if (qualifier == 'h') {
630 num = (unsigned short) va_arg(args, int);
631 if (flags & SIGN)
632 num = (signed short) num;
633 } else {
634 num = va_arg(args, unsigned int);
635 if (flags & SIGN)
636 num = (signed int) num;
637 }
638 str = number(str, num, base, field_width, precision, flags);
639 }
640 *str = '\0';
641 return str-buf;
642 }
643
644 /**
645 * sprintf - Format a string and place it in a buffer
646 * @buf: The buffer to place the result into
647 * @fmt: The format string to use
648 * @...: Arguments for the format string
649 *
650 * The function returns the number of characters written
651 * into @buf.
652 *
653 * See the vsprintf() documentation for format string extensions over C99.
654 */
655 int sprintf(char * buf, const char *fmt, ...)
656 {
657 va_list args;
658 int i;
659
660 va_start(args, fmt);
661 i=vsprintf(buf,fmt,args);
662 va_end(args);
663 return i;
664 }
665
666 void panic(const char *fmt, ...)
667 {
668 va_list args;
669 va_start(args, fmt);
670 vprintf(fmt, args);
671 putc('\n');
672 va_end(args);
673 #if defined (CONFIG_PANIC_HANG)
674 hang();
675 #else
676 udelay (100000); /* allow messages to go out */
677 do_reset (NULL, 0, 0, NULL);
678 #endif
679 }