src/console/vtxprintf.c

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