ash: fix BASE###nn bashism for bases 36..64

function                                             old     new   delta
evaluate_string                                      876     932     +56

Signed-off-by: Denys Vlasenko <vda.linux@googlemail.com>

1 files changed, 21 insertions, 3 deletions

diff --git a/shell/math.c b/shell/math.c
index 0c806ad39..af1ab55c0 100644
--- a/shell/math.c
+++ b/shell/math.c
@@ -540,11 +540,29 @@ static arith_t strto_arith_t(const char *nptr, char **endptr)
 	/* bash allows "N#" (empty "nnnn" part) */
 	for (;;) {
 		unsigned digit = (unsigned)*nptr - '0';
-		if (digit >= 10 || digit >= base) {
+		if (digit >= 10) {
+			/* *nptr is not 0..9 */
+			if (*nptr > 'z')
+				break; /* this rejects e.g. $((64#~)) */
+			/* in bases up to 36, case does not matter for a-z */
 			digit = (unsigned)(*nptr | 0x20) - ('a' - 10);
-			if (digit >= base)
-				break;
+			if (base > 36 && *nptr <= '_') {
+				/* otherwise, A-Z,@,_ are 36..61,62,63 */
+				if (*nptr == '@')
+					digit = 62;
+				else if (*nptr == '_')
+					digit = 63;
+				else if (digit < 36) /* A-Z */
+					digit += 36 - 10;
+				else
+					break; /* error, such as [ or \ */
+			}
+			//bb_error_msg("ch:'%c'%d digit:%u", *nptr, *nptr, digit);
+			//if (digit < 10) - example where we need this?
+			//	break;
 		}
+		if (digit >= base)
+			break;
 		/* bash does not check for overflows */
 		n = n * base + digit;
 		nptr++;