Add support to left-pad with zeroes in tf_printf

Add support to formats %i, %d, %p, %x and %u for left-padding numbers
with zeroes (e.g. `%08x`).

Change-Id: Ifd4795a82a8d83da2c00b44b9e482a2d9be797e3
Signed-off-by: Antonio Nino Diaz <antonio.ninodiaz@arm.com>
This commit is contained in:
Antonio Nino Diaz 2017-12-15 10:36:20 +00:00
parent 842c00eb44
commit 84816dcd7c
1 changed files with 36 additions and 7 deletions

View File

@ -1,5 +1,5 @@
/* /*
* Copyright (c) 2014-2016, ARM Limited and Contributors. All rights reserved. * Copyright (c) 2014-2017, ARM Limited and Contributors. All rights reserved.
* *
* SPDX-License-Identifier: BSD-3-Clause * SPDX-License-Identifier: BSD-3-Clause
*/ */
@ -31,7 +31,8 @@ void tf_string_print(const char *str)
putchar(*str++); putchar(*str++);
} }
static void unsigned_num_print(unsigned long long int unum, unsigned int radix) static void unsigned_num_print(unsigned long long int unum, unsigned int radix,
char padc, int padn)
{ {
/* Just need enough space to store 64 bit decimal integer */ /* Just need enough space to store 64 bit decimal integer */
unsigned char num_buf[20]; unsigned char num_buf[20];
@ -45,6 +46,12 @@ static void unsigned_num_print(unsigned long long int unum, unsigned int radix)
num_buf[i++] = 'a' + (rem - 0xa); num_buf[i++] = 'a' + (rem - 0xa);
} while (unum /= radix); } while (unum /= radix);
if (padn > 0) {
while (i < padn--) {
putchar(padc);
}
}
while (--i >= 0) while (--i >= 0)
putchar(num_buf[i]); putchar(num_buf[i]);
} }
@ -63,6 +70,9 @@ static void unsigned_num_print(unsigned long long int unum, unsigned int radix)
* %ll - long long int (64-bit on AArch64) * %ll - long long int (64-bit on AArch64)
* %z - size_t sized integer formats (64 bit on AArch64) * %z - size_t sized integer formats (64 bit on AArch64)
* *
* The following padding specifiers are supported by this print
* %0NN - Left-pad the number with 0s (NN is a decimal number)
*
* The print exits on all other formats specifiers other than valid * The print exits on all other formats specifiers other than valid
* combinations of the above specifiers. * combinations of the above specifiers.
*******************************************************************/ *******************************************************************/
@ -72,9 +82,12 @@ void tf_vprintf(const char *fmt, va_list args)
long long int num; long long int num;
unsigned long long int unum; unsigned long long int unum;
char *str; char *str;
char padc = 0; /* Padding character */
int padn; /* Number of characters to pad */
while (*fmt) { while (*fmt) {
l_count = 0; l_count = 0;
padn = 0;
if (*fmt == '%') { if (*fmt == '%') {
fmt++; fmt++;
@ -87,10 +100,11 @@ loop:
if (num < 0) { if (num < 0) {
putchar('-'); putchar('-');
unum = (unsigned long long int)-num; unum = (unsigned long long int)-num;
padn--;
} else } else
unum = (unsigned long long int)num; unum = (unsigned long long int)num;
unsigned_num_print(unum, 10); unsigned_num_print(unum, 10, padc, padn);
break; break;
case 's': case 's':
str = va_arg(args, char *); str = va_arg(args, char *);
@ -98,14 +112,16 @@ loop:
break; break;
case 'p': case 'p':
unum = (uintptr_t)va_arg(args, void *); unum = (uintptr_t)va_arg(args, void *);
if (unum) if (unum) {
tf_string_print("0x"); tf_string_print("0x");
padn -= 2;
}
unsigned_num_print(unum, 16); unsigned_num_print(unum, 16, padc, padn);
break; break;
case 'x': case 'x':
unum = get_unum_va_args(args, l_count); unum = get_unum_va_args(args, l_count);
unsigned_num_print(unum, 16); unsigned_num_print(unum, 16, padc, padn);
break; break;
case 'z': case 'z':
if (sizeof(size_t) == 8) if (sizeof(size_t) == 8)
@ -119,8 +135,21 @@ loop:
goto loop; goto loop;
case 'u': case 'u':
unum = get_unum_va_args(args, l_count); unum = get_unum_va_args(args, l_count);
unsigned_num_print(unum, 10); unsigned_num_print(unum, 10, padc, padn);
break; break;
case '0':
padc = '0';
padn = 0;
fmt++;
while (1) {
char ch = *fmt;
if (ch < '0' || ch > '9') {
goto loop;
}
padn = (padn * 10) + (ch - '0');
fmt++;
}
default: default:
/* Exit on any other format specifier */ /* Exit on any other format specifier */
return; return;