Handle func return

1 files changed, 2 insertions, 1 deletions

diff --git a/src/generator.h b/src/generator.h
index 939ab13..205fc01 100644
--- a/src/generator.h
+++ b/src/generator.h
@@ -19,8 +19,9 @@ FILE *output;
 void gen_prologue();
 void gen_prologue_continue(int *bss_done);
 void gen_const_declaration();
-void gen_function_declaration(const char name[], int return_type, int nb_param);
+Type gen_function_declaration(const char name[], int return_type, int nb_param);
 void gen_function_end_declaration();
+void gen_function_return(Type expect, Type actual);
 int gen_function_call(const char name[], int nb_param);
 void gen_declaration(const char name[], int type, Scope scope);
 void gen_check(const char name[], Scope scope);