global array

1 files changed, 15 insertions, 2 deletions

diff --git a/src/main.c b/src/main.c
index a67f00c..3c39975 100644
--- a/src/main.c
+++ b/src/main.c
@@ -87,7 +87,7 @@ is_punct(const char *p)
 	if (strchr("=!<>", p[0]) != NULL && p[1] == '=')
 		return 2;
 
-	return (strchr("+-/*()<>,;{}=", p[0]) != NULL);
+	return (strchr("+-/*()<>,;{}[]=", p[0]) != NULL);
 }
 
 static Token *
@@ -324,6 +324,13 @@ global(Token **rest, Token *tok)
 	node->type = NOD_GLOBAL;
 
 	tok = tok->next;
+
+	if (equal(tok, "[")) {
+		tok = tok->next;
+		node->lhs = new_num(get_number(tok));
+		tok = skip(tok->next, "]");
+	}
+
 	if (equal(tok, "=")) {
 		tok = tok->next;
 		node->val = get_number(tok);
@@ -687,7 +694,13 @@ static void
 gen_globaldec(Node *node)
 {
 	printf("@__gl_%.*s\n", node->len, node->loc);
-	printf("\tLIT %04x\n", node->val);
+	if (node->lhs != NULL) {
+		printf("\tLIT ,__gla_%.*s\n", node->len, node->loc);
+		printf("@__gla_%.*s\n", node->len, node->loc);
+		for (int i = 0; i < node->lhs->val; i++)
+			printf("\tLIT %04x\n", node->val);
+	} else
+		printf("\tLIT %04x\n", node->val);
 }
 
 static void