1 /* 2 * Copyright (c) 2014-2018, ARM Limited and Contributors. All rights reserved. 3 * 4 * SPDX-License-Identifier: BSD-3-Clause 5 */ 6 7 #include <assert.h> 8 #include <stdarg.h> 9 #include <stdbool.h> 10 #include <stdint.h> 11 12 #include <common/debug.h> 13 14 #define get_num_va_args(_args, _lcount) \ 15 (((_lcount) > 1) ? va_arg(_args, long long int) : \ 16 (((_lcount) == 1) ? va_arg(_args, long int) : \ 17 va_arg(_args, int))) 18 19 #define get_unum_va_args(_args, _lcount) \ 20 (((_lcount) > 1) ? va_arg(_args, unsigned long long int) : \ 21 (((_lcount) == 1) ? va_arg(_args, unsigned long int) : \ 22 va_arg(_args, unsigned int))) 23 24 static int string_print(const char *str) 25 { 26 int count = 0; 27 28 assert(str != NULL); 29 30 for ( ; *str != '\0'; str++) { 31 (void)putchar(*str); 32 count++; 33 } 34 35 return count; 36 } 37 38 static int unsigned_num_print(unsigned long long int unum, unsigned int radix, 39 char padc, int padn) 40 { 41 /* Just need enough space to store 64 bit decimal integer */ 42 char num_buf[20]; 43 int i = 0, count = 0; 44 unsigned int rem; 45 46 do { 47 rem = unum % radix; 48 if (rem < 0xa) 49 num_buf[i] = '0' + rem; 50 else 51 num_buf[i] = 'a' + (rem - 0xa); 52 i++; 53 unum /= radix; 54 } while (unum > 0U); 55 56 if (padn > 0) { 57 while (i < padn) { 58 (void)putchar(padc); 59 count++; 60 padn--; 61 } 62 } 63 64 while (--i >= 0) { 65 (void)putchar(num_buf[i]); 66 count++; 67 } 68 69 return count; 70 } 71 72 /******************************************************************* 73 * Reduced format print for Trusted firmware. 74 * The following type specifiers are supported by this print 75 * %x - hexadecimal format 76 * %s - string format 77 * %d or %i - signed decimal format 78 * %u - unsigned decimal format 79 * %p - pointer format 80 * 81 * The following length specifiers are supported by this print 82 * %l - long int (64-bit on AArch64) 83 * %ll - long long int (64-bit on AArch64) 84 * %z - size_t sized integer formats (64 bit on AArch64) 85 * 86 * The following padding specifiers are supported by this print 87 * %0NN - Left-pad the number with 0s (NN is a decimal number) 88 * 89 * The print exits on all other formats specifiers other than valid 90 * combinations of the above specifiers. 91 *******************************************************************/ 92 int vprintf(const char *fmt, va_list args) 93 { 94 int l_count; 95 long long int num; 96 unsigned long long int unum; 97 char *str; 98 char padc = '\0'; /* Padding character */ 99 int padn; /* Number of characters to pad */ 100 int count = 0; /* Number of printed characters */ 101 102 while (*fmt != '\0') { 103 l_count = 0; 104 padn = 0; 105 106 if (*fmt == '%') { 107 fmt++; 108 /* Check the format specifier */ 109 loop: 110 switch (*fmt) { 111 case 'i': /* Fall through to next one */ 112 case 'd': 113 num = get_num_va_args(args, l_count); 114 if (num < 0) { 115 (void)putchar('-'); 116 unum = (unsigned long long int)-num; 117 padn--; 118 } else 119 unum = (unsigned long long int)num; 120 121 count += unsigned_num_print(unum, 10, 122 padc, padn); 123 break; 124 case 's': 125 str = va_arg(args, char *); 126 count += string_print(str); 127 break; 128 case 'p': 129 unum = (uintptr_t)va_arg(args, void *); 130 if (unum > 0U) { 131 count += string_print("0x"); 132 padn -= 2; 133 } 134 135 count += unsigned_num_print(unum, 16, 136 padc, padn); 137 break; 138 case 'x': 139 unum = get_unum_va_args(args, l_count); 140 count += unsigned_num_print(unum, 16, 141 padc, padn); 142 break; 143 case 'z': 144 if (sizeof(size_t) == 8U) 145 l_count = 2; 146 147 fmt++; 148 goto loop; 149 case 'l': 150 l_count++; 151 fmt++; 152 goto loop; 153 case 'u': 154 unum = get_unum_va_args(args, l_count); 155 count += unsigned_num_print(unum, 10, 156 padc, padn); 157 break; 158 case '0': 159 padc = '0'; 160 padn = 0; 161 fmt++; 162 163 for (;;) { 164 char ch = *fmt; 165 if ((ch < '0') || (ch > '9')) { 166 goto loop; 167 } 168 padn = (padn * 10) + (ch - '0'); 169 fmt++; 170 } 171 assert(0); /* Unreachable */ 172 default: 173 /* Exit on any other format specifier */ 174 return -1; 175 } 176 fmt++; 177 continue; 178 } 179 (void)putchar(*fmt); 180 fmt++; 181 count++; 182 } 183 184 return count; 185 } 186 187 int printf(const char *fmt, ...) 188 { 189 int count; 190 va_list va; 191 192 va_start(va, fmt); 193 count = vprintf(fmt, va); 194 va_end(va); 195 196 return count; 197 } 198