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