Add support for binary literals in arithmetic expressions

Author: alexdowad

parser/arithm.c

@@ -74,6 +74,23 @@ static size_t peek_hex_literal(struct mrsh_parser *parser) {
 	return i;
 }
 
+static size_t peek_binary_literal(struct mrsh_parser *parser) {
+	size_t i = 2; // We already know '0b' prefix is there
+
+	while (true) {
+		parser_peek(parser, NULL, i + 1);
+
+		char c = parser->buf.data[i];
+		if (c != '0' && c != '1') {
+			break;
+		}
+
+		++i;
+	}
+
+	return i;
+}
+
 static size_t peek_literal(struct mrsh_parser *parser) {
 	size_t i = 0;
 
@@ -85,9 +102,10 @@ static size_t peek_literal(struct mrsh_parser *parser) {
 		parser_peek(parser, NULL, 2);
 
 		c = parser->buf.data[1];
-		// TODO: 0b prefix
 		if (c == 'x' || c == 'X') {
 			return peek_hex_literal(parser);
+		} else if (c == 'b' || c == 'B') {
+			return peek_binary_literal(parser);
 		}
 	}
 

test/arithm.sh

@@ -4,6 +4,9 @@ echo "1 =" $((1))
 echo "0x0 = " $((0x0))
 echo "0x100 = " $((0x100))
 echo "0xabcd = " $((0xabcd))
+echo "0b0 = " $((0b0))
+echo "0b1010 = " $((0b1010))
+echo "0B1111 = " $((0B1111))
 echo "2*5 =" $((2*5))
 echo "2/5 =" $((2/5))
 echo "2%5 =" $((2%5))