Support modulo operator

Author: eecheng87

src/cfront.c

@@ -47,6 +47,7 @@ typedef enum {
     T_close_square,  /* ] */
     T_asterisk,      /* '*' */
     T_divide,        /* / */
+    T_mod,           /* % */
     T_bit_or,        /* | */
     T_bit_xor,       /* ^ */
     T_log_and,       /* && */
@@ -338,6 +339,10 @@ token_t get_next_token()
         skip_whitespace();
         return T_lt;
     }
+    if (next_char == '%') {
+        read_char(1);
+        return T_mod;
+    }
     if (next_char == '>') {
         read_char(0);
         if (next_char == '=') {
@@ -876,6 +881,7 @@ int get_operator_prio(opcode_t op)
         return 12;
     case OP_mul:
     case OP_div:
+    case OP_mod:
         return 13;
     default:
         return 0;
@@ -893,6 +899,8 @@ opcode_t get_operator()
         op = OP_mul;
     else if (lex_accept(T_divide))
         op = OP_div;
+    else if (lex_accept(T_mod))
+        op = OP_mod;
     else if (lex_accept(T_lshift))
         op = OP_lshift;
     else if (lex_accept(T_rshift))
@@ -1371,6 +1379,9 @@ int eval_expression_imm(opcode_t op, int op1, int op2)
     case OP_div:
         res = op1 / op2;
         break;
+    case OP_mod:
+        res = op1 % op2;
+        break;
     case OP_lshift:
         res = op1 << op2;
         break;

src/codegen.c

@@ -121,6 +121,7 @@ int get_code_length(ir_instr_t *ii)
     case OP_func_exit:
         return 16;
     case OP_exit:
+    case OP_mod:
         return 12;
     case OP_load_data_address:
     case OP_jz:
@@ -367,6 +368,14 @@ void code_generate()
             if (dump_ir == 1)
                 printf("    x%d /= x%d", dest_reg, OP_reg);
             break;
+        case OP_mod:
+            emit(__div(__AL, OP_reg + 1, OP_reg, dest_reg));
+            emit(__mul(__AL, OP_reg, OP_reg, OP_reg + 1));
+            emit(__sub_r(__AL, dest_reg, dest_reg, OP_reg));
+            /* TODO: support print '%' */
+            if (dump_ir == 1)
+                printf("    x%d = x%d mod x%d", dest_reg, dest_reg, OP_reg);
+            break;
         case OP_negate:
             emit(__rsb_i(__AL, dest_reg, 0, dest_reg));
             if (dump_ir == 1)

src/defs.h

@@ -68,6 +68,7 @@ typedef enum {
     OP_sub,
     OP_mul,
     OP_div, /* signed division */
+    OP_mod, /* modulo */
     OP_lshift,
     OP_rshift,
     OP_log_and,

tests/driver.sh

@@ -69,8 +69,8 @@ expr 55 "1 + (2 + (3 + (4 + (5 + (6 + (7 + (8 + (9 + 10))))))))"
 # expr 21 "+1+20"
 # expr 10 "-15+(+35-10)"
 
-# expr 2 "5 % 3"
-# expr 6 "111 % 7"
+expr 2 "5 % 3"
+expr 6 "111 % 7"
 
 expr 1 "10 > 5"
 expr 1 "3+3 > 5"