1 files changed, 8 insertions, 0 deletions
diff --git a/lib/CodeGen/CodeGen.cc b/lib/CodeGen/CodeGen.cc
index e45a3db..772dc82 100644
--- a/lib/CodeGen/CodeGen.cc
+++ b/lib/CodeGen/CodeGen.cc
@@ -77,6 +77,14 @@ CodeGen::CodeGen() {
         activateStackObject(registerStackObject(Addr, TypeName, Line));
     });
 
+    FuncGen->setMissingReturnHandler([this](int Line, const std::string &Name) {
+        Value *Msg = Builder->CreateGlobalString("[Fatal] line %d: FUNCTION %s ended without RETURN\n",
+                                                 "err_missing_return", 0, TheModule.get());
+        Value *NameStr = Builder->CreateGlobalString(Name, "function_name", 0, TheModule.get());
+        RuntimeChecker->emitStderr(Msg, {RuntimeChecker->lineValue(Line), NameStr});
+        Exceptions->emitThrow();
+    });
+
     SetupExternalFunctions();
 }