Pull utrace into release branch
[linux-2.6] / arch / powerpc / boot / stdio.c
1 /*
2  * Copyright (C) Paul Mackerras 1997.
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version
7  * 2 of the License, or (at your option) any later version.
8  */
9 #include <stdarg.h>
10 #include <stddef.h>
11 #include "string.h"
12 #include "stdio.h"
13 #include "ops.h"
14
15 size_t strnlen(const char * s, size_t count)
16 {
17         const char *sc;
18
19         for (sc = s; count-- && *sc != '\0'; ++sc)
20                 /* nothing */;
21         return sc - s;
22 }
23
24 extern unsigned int __div64_32(unsigned long long *dividend,
25                                unsigned int divisor);
26
27 /* The unnecessary pointer compare is there
28  * to check for type safety (n must be 64bit)
29  */
30 # define do_div(n,base) ({                                              \
31         unsigned int __base = (base);                                   \
32         unsigned int __rem;                                             \
33         (void)(((typeof((n)) *)0) == ((unsigned long long *)0));        \
34         if (((n) >> 32) == 0) {                                         \
35                 __rem = (unsigned int)(n) % __base;                     \
36                 (n) = (unsigned int)(n) / __base;                       \
37         } else                                                          \
38                 __rem = __div64_32(&(n), __base);                       \
39         __rem;                                                          \
40  })
41
42 static int skip_atoi(const char **s)
43 {
44         int i, c;
45
46         for (i = 0; '0' <= (c = **s) && c <= '9'; ++*s)
47                 i = i*10 + c - '0';
48         return i;
49 }
50
51 #define ZEROPAD 1               /* pad with zero */
52 #define SIGN    2               /* unsigned/signed long */
53 #define PLUS    4               /* show plus */
54 #define SPACE   8               /* space if plus */
55 #define LEFT    16              /* left justified */
56 #define SPECIAL 32              /* 0x */
57 #define LARGE   64              /* use 'ABCDEF' instead of 'abcdef' */
58
59 static char * number(char * str, unsigned long long num, int base, int size, int precision, int type)
60 {
61         char c,sign,tmp[66];
62         const char *digits="0123456789abcdefghijklmnopqrstuvwxyz";
63         int i;
64
65         if (type & LARGE)
66                 digits = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ";
67         if (type & LEFT)
68                 type &= ~ZEROPAD;
69         if (base < 2 || base > 36)
70                 return 0;
71         c = (type & ZEROPAD) ? '0' : ' ';
72         sign = 0;
73         if (type & SIGN) {
74                 if ((signed long long)num < 0) {
75                         sign = '-';
76                         num = - (signed long long)num;
77                         size--;
78                 } else if (type & PLUS) {
79                         sign = '+';
80                         size--;
81                 } else if (type & SPACE) {
82                         sign = ' ';
83                         size--;
84                 }
85         }
86         if (type & SPECIAL) {
87                 if (base == 16)
88                         size -= 2;
89                 else if (base == 8)
90                         size--;
91         }
92         i = 0;
93         if (num == 0)
94                 tmp[i++]='0';
95         else while (num != 0) {
96                 tmp[i++] = digits[do_div(num, base)];
97         }
98         if (i > precision)
99                 precision = i;
100         size -= precision;
101         if (!(type&(ZEROPAD+LEFT)))
102                 while(size-->0)
103                         *str++ = ' ';
104         if (sign)
105                 *str++ = sign;
106         if (type & SPECIAL) {
107                 if (base==8)
108                         *str++ = '0';
109                 else if (base==16) {
110                         *str++ = '0';
111                         *str++ = digits[33];
112                 }
113         }
114         if (!(type & LEFT))
115                 while (size-- > 0)
116                         *str++ = c;
117         while (i < precision--)
118                 *str++ = '0';
119         while (i-- > 0)
120                 *str++ = tmp[i];
121         while (size-- > 0)
122                 *str++ = ' ';
123         return str;
124 }
125
126 int vsprintf(char *buf, const char *fmt, va_list args)
127 {
128         int len;
129         unsigned long long num;
130         int i, base;
131         char * str;
132         const char *s;
133
134         int flags;              /* flags to number() */
135
136         int field_width;        /* width of output field */
137         int precision;          /* min. # of digits for integers; max
138                                    number of chars for from string */
139         int qualifier;          /* 'h', 'l', or 'L' for integer fields */
140                                 /* 'z' support added 23/7/1999 S.H.    */
141                                 /* 'z' changed to 'Z' --davidm 1/25/99 */
142
143         
144         for (str=buf ; *fmt ; ++fmt) {
145                 if (*fmt != '%') {
146                         *str++ = *fmt;
147                         continue;
148                 }
149                         
150                 /* process flags */
151                 flags = 0;
152                 repeat:
153                         ++fmt;          /* this also skips first '%' */
154                         switch (*fmt) {
155                                 case '-': flags |= LEFT; goto repeat;
156                                 case '+': flags |= PLUS; goto repeat;
157                                 case ' ': flags |= SPACE; goto repeat;
158                                 case '#': flags |= SPECIAL; goto repeat;
159                                 case '0': flags |= ZEROPAD; goto repeat;
160                                 }
161                 
162                 /* get field width */
163                 field_width = -1;
164                 if ('0' <= *fmt && *fmt <= '9')
165                         field_width = skip_atoi(&fmt);
166                 else if (*fmt == '*') {
167                         ++fmt;
168                         /* it's the next argument */
169                         field_width = va_arg(args, int);
170                         if (field_width < 0) {
171                                 field_width = -field_width;
172                                 flags |= LEFT;
173                         }
174                 }
175
176                 /* get the precision */
177                 precision = -1;
178                 if (*fmt == '.') {
179                         ++fmt;  
180                         if ('0' <= *fmt && *fmt <= '9')
181                                 precision = skip_atoi(&fmt);
182                         else if (*fmt == '*') {
183                                 ++fmt;
184                                 /* it's the next argument */
185                                 precision = va_arg(args, int);
186                         }
187                         if (precision < 0)
188                                 precision = 0;
189                 }
190
191                 /* get the conversion qualifier */
192                 qualifier = -1;
193                 if (*fmt == 'l' && *(fmt + 1) == 'l') {
194                         qualifier = 'q';
195                         fmt += 2;
196                 } else if (*fmt == 'h' || *fmt == 'l' || *fmt == 'L'
197                         || *fmt == 'Z') {
198                         qualifier = *fmt;
199                         ++fmt;
200                 }
201
202                 /* default base */
203                 base = 10;
204
205                 switch (*fmt) {
206                 case 'c':
207                         if (!(flags & LEFT))
208                                 while (--field_width > 0)
209                                         *str++ = ' ';
210                         *str++ = (unsigned char) va_arg(args, int);
211                         while (--field_width > 0)
212                                 *str++ = ' ';
213                         continue;
214
215                 case 's':
216                         s = va_arg(args, char *);
217                         if (!s)
218                                 s = "<NULL>";
219
220                         len = strnlen(s, precision);
221
222                         if (!(flags & LEFT))
223                                 while (len < field_width--)
224                                         *str++ = ' ';
225                         for (i = 0; i < len; ++i)
226                                 *str++ = *s++;
227                         while (len < field_width--)
228                                 *str++ = ' ';
229                         continue;
230
231                 case 'p':
232                         if (field_width == -1) {
233                                 field_width = 2*sizeof(void *);
234                                 flags |= ZEROPAD;
235                         }
236                         str = number(str,
237                                 (unsigned long) va_arg(args, void *), 16,
238                                 field_width, precision, flags);
239                         continue;
240
241
242                 case 'n':
243                         if (qualifier == 'l') {
244                                 long * ip = va_arg(args, long *);
245                                 *ip = (str - buf);
246                         } else if (qualifier == 'Z') {
247                                 size_t * ip = va_arg(args, size_t *);
248                                 *ip = (str - buf);
249                         } else {
250                                 int * ip = va_arg(args, int *);
251                                 *ip = (str - buf);
252                         }
253                         continue;
254
255                 case '%':
256                         *str++ = '%';
257                         continue;
258
259                 /* integer number formats - set up the flags and "break" */
260                 case 'o':
261                         base = 8;
262                         break;
263
264                 case 'X':
265                         flags |= LARGE;
266                 case 'x':
267                         base = 16;
268                         break;
269
270                 case 'd':
271                 case 'i':
272                         flags |= SIGN;
273                 case 'u':
274                         break;
275
276                 default:
277                         *str++ = '%';
278                         if (*fmt)
279                                 *str++ = *fmt;
280                         else
281                                 --fmt;
282                         continue;
283                 }
284                 if (qualifier == 'l') {
285                         num = va_arg(args, unsigned long);
286                         if (flags & SIGN)
287                                 num = (signed long) num;
288                 } else if (qualifier == 'q') {
289                         num = va_arg(args, unsigned long long);
290                         if (flags & SIGN)
291                                 num = (signed long long) num;
292                 } else if (qualifier == 'Z') {
293                         num = va_arg(args, size_t);
294                 } else if (qualifier == 'h') {
295                         num = (unsigned short) va_arg(args, int);
296                         if (flags & SIGN)
297                                 num = (signed short) num;
298                 } else {
299                         num = va_arg(args, unsigned int);
300                         if (flags & SIGN)
301                                 num = (signed int) num;
302                 }
303                 str = number(str, num, base, field_width, precision, flags);
304         }
305         *str = '\0';
306         return str-buf;
307 }
308
309 int sprintf(char * buf, const char *fmt, ...)
310 {
311         va_list args;
312         int i;
313
314         va_start(args, fmt);
315         i=vsprintf(buf,fmt,args);
316         va_end(args);
317         return i;
318 }
319
320 static char sprint_buf[1024];
321
322 int
323 printf(const char *fmt, ...)
324 {
325         va_list args;
326         int n;
327
328         va_start(args, fmt);
329         n = vsprintf(sprint_buf, fmt, args);
330         va_end(args);
331         if (console_ops.write)
332                 console_ops.write(sprint_buf, n);
333         return n;
334 }