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