array subscript

2 files changed, 58 insertions, 0 deletions

diff --git a/main.c b/main.c
index 162d3a4..9a8de99 100644
--- a/main.c
+++ b/main.c
@@ -770,6 +770,18 @@ struct AstNode* parse_postfix_expr(struct Parser* p) {
             struct AstNode* args = parse_arg_list(p);
             expect(p, TK_PAREN_R);
             ret->expr1 = args;
+        } else if (tk == TK_BRACKET_L) {
+            next_token(p);
+            struct AstNode* idx = parse_expr(p);
+            expect(p, TK_BRACKET_R);
+
+            struct AstNode* e = ast_new(AST_DEREF_EXPR);
+            struct AstNode* ptr_expr = ast_new_binary_expr(TK_PLUS, ret, idx);
+            ptr_expr->ty = ret->ty;
+            e->expr1 = ptr_expr;
+            e->ty = ret->ty->to;
+
+            ret = e;
         } else if (tk == TK_DOT) {
             next_token(p);
             char* name = parse_ident(p);
diff --git a/tests/033.sh b/tests/033.sh
new file mode 100644
index 0000000..1fb92ac
--- /dev/null
+++ b/tests/033.sh
@@ -0,0 +1,46 @@
+set -e
+
+cat <<'EOF' > expected
+42
+EOF
+bash ../../test_diff.sh <<'EOF'
+int printf();
+void* calloc();
+
+int main() {
+    int b;
+    int* a = &b;
+    a[0] = 42;
+    printf("%d\n", *a);
+    return 0;
+}
+EOF
+
+cat <<'EOF' > expected
+0 0
+1 1
+2 2
+3 3
+4 4
+5 5
+6 6
+7 7
+8 8
+9 9
+EOF
+bash ../../test_diff.sh <<'EOF'
+int printf();
+void* calloc();
+
+int main() {
+    long* a = calloc(10, sizeof(long));
+    long i = 0;
+    for (i = 0; i < 10; i = i + 1) {
+        a[i] = i;
+    }
+    for (i = 0; i < 10; i = i + 1) {
+        printf("%d %d\n", *(a + i), a[i]);
+    }
+    return 0;
+}
+EOF