16 files changed, 12 insertions, 3 deletions

diff --git a/res/test_add.tpc b/res/test_add_ok.tpc
index 848130c..848130c 100644
--- a/res/test_add.tpc
+++ b/res/test_add_ok.tpc
diff --git a/res/test_array.tpc b/res/test_array_ok.tpc
index a968c22..a968c22 100644
--- a/res/test_array.tpc
+++ b/res/test_array_ok.tpc
diff --git a/res/test_global.tpc b/res/test_global_ok.tpc
index d8b7d45..d8b7d45 100644
--- a/res/test_global.tpc
+++ b/res/test_global_ok.tpc
diff --git a/res/test_if.tpc b/res/test_if_ok.tpc
index 054811b..054811b 100644
--- a/res/test_if.tpc
+++ b/res/test_if_ok.tpc
diff --git a/res/test_mul.tpc b/res/test_mul_ok.tpc
index 62a8080..62a8080 100644
--- a/res/test_mul.tpc
+++ b/res/test_mul_ok.tpc
diff --git a/res/test_nomain_ko.tpc b/res/test_nomain_ko.tpc
new file mode 100644
index 0000000..570c5ca
--- /dev/null
+++ b/res/test_nomain_ko.tpc
@@ -0,0 +1,6 @@
+const c = 0;
+
+/* not a main */
+void f(void) {
+  print(c);
+}
diff --git a/res/test_parameters.tpc b/res/test_parameters_ok.tpc
index 479fb60..479fb60 100644
--- a/res/test_parameters.tpc
+++ b/res/test_parameters_ok.tpc
diff --git a/res/test_read.tpc b/res/test_read_ok.tpc
index 6ad3632..6ad3632 100644
--- a/res/test_read.tpc
+++ b/res/test_read_ok.tpc
diff --git a/res/test_return.tpc b/res/test_return_ok.tpc
index f450203..f450203 100644
--- a/res/test_return.tpc
+++ b/res/test_return_ok.tpc
diff --git a/res/ko_sample.tpc b/res/test_sample_ko.tpc
index 9dc344f..9dc344f 100644
--- a/res/ko_sample.tpc
+++ b/res/test_sample_ko.tpc
diff --git a/res/ok_sample.tpc b/res/test_sample_ok.tpc
index c8d3d9d..c8d3d9d 100644
--- a/res/ok_sample.tpc
+++ b/res/test_sample_ok.tpc
diff --git a/res/test-table-symboles.tpc b/res/test_table_symbols_ok.tpc
index b2cc31e..b2cc31e 100644
--- a/res/test-table-symboles.tpc
+++ b/res/test_table_symbols_ok.tpc
diff --git a/res/test_while.tpc b/res/test_while_ok.tpc
index 25d04c6..25d04c6 100644
--- a/res/test_while.tpc
+++ b/res/test_while_ok.tpc
diff --git a/src/generator.c b/src/generator.c
index bb95e08..44fdeab 100644
--- a/src/generator.c
+++ b/src/generator.c
@@ -79,7 +79,7 @@ void gen_prologue_continue(int *bss_done) {
   *bss_done = 1;
 }
 
-void gen_const_declaration() {
+void gen_epilogue() {
   fprintf(output, "\n_start:\n");
   fprintf(output, "push rbp\nmov rbp, rsp\n\n");
   fprintf(output, "call main\n");
@@ -92,6 +92,9 @@ void gen_const_declaration() {
   loc_display_table();
   fprintf(output, ";function table\n");
   fun_display_table();
+
+  // require a main function
+  fun_lookup("main", 0);
 }
 
 void gen_const(const char name[], int value, Scope scope) {
diff --git a/src/generator.h b/src/generator.h
index 20a83f9..9d6dd65 100644
--- a/src/generator.h
+++ b/src/generator.h
@@ -16,7 +16,7 @@ FILE *output;
 
 void gen_prologue();
 void gen_prologue_continue(int *bss_done);
-void gen_const_declaration();
+void gen_epilogue();
 void gen_const(const char name[], int value, Scope scope);
 
 Type gen_function_declaration(const char name[], int return_type);
diff --git a/src/tpc.y b/src/tpc.y
index 59fa9d1..74d31c6 100644
--- a/src/tpc.y
+++ b/src/tpc.y
@@ -58,7 +58,7 @@ static char fname[64];
 
 %%
 Prog:                               { gen_prologue(); }
-  DeclConsts DeclVars DeclFoncts    { gen_const_declaration(); }
+  DeclConsts DeclVars DeclFoncts    { gen_epilogue(); }
 ;
 DeclConsts:
   DeclConsts CONST ListConst ';'