Rewrite the name of main function to work with TA's broken toolchain

1 files changed, 12 insertions, 6 deletions

diff --git a/src/code-generation.c b/src/code-generation.c
index 0d37f3d..c9067c4 100644
--- a/src/code-generation.c
+++ b/src/code-generation.c
@@ -9,6 +9,7 @@
 #include <inttypes.h>
 #include <stdio.h>
 #include <stdlib.h>
+#include <string.h>
 
 static void generate_global_variable(CcmmcAst *global_decl, CcmmcState *state)
 {
@@ -732,15 +733,20 @@ static void generate_block(
 static void generate_function(CcmmcAst *function, CcmmcState *state)
 {
     fputs("\t.text\n\t.align\t2\n", state->asm_output);
+    const char *func_name = function->child->right_sibling->value_id.name;
+    const char *symbol_name = func_name;
+    // XXX: We have to rewrite some names to workaround TA's broken toolchain
+    if (strcmp(func_name, "main") == 0 || strcmp(func_name, "MAIN") == 0)
+        symbol_name = "_start_MAIN";
     fprintf(state->asm_output,
         "\t.type\t%s, %%function\n"
         "\t.global\t%s\n"
         "%s:\n"
         "\tstp\tlr, fp, [sp, -16]!\n"
         "\tmov\tfp, sp\n",
-        function->child->right_sibling->value_id.name,
-        function->child->right_sibling->value_id.name,
-        function->child->right_sibling->value_id.name);
+        symbol_name,
+        symbol_name,
+        symbol_name);
     CcmmcAst *param_node = function->child->right_sibling->right_sibling;
     CcmmcAst *block_node = param_node->right_sibling;
     generate_block(block_node, state, 0);
@@ -749,9 +755,9 @@ static void generate_function(CcmmcAst *function, CcmmcState *state)
         "\tldp\tlr, fp, [sp], 16\n"
         "\tret\tlr\n"
         "\t.size\t%s, .-%s\n",
-        function->child->right_sibling->value_id.name,
-        function->child->right_sibling->value_id.name,
-        function->child->right_sibling->value_id.name);
+        func_name,
+        symbol_name,
+        symbol_name);
 }
 
 static void generate_program(CcmmcState *state)