arch/powerpc/boot/stdio.c

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