feat: Added check for return type to match function type

alinalihassan · May 28, 2022 · 7c284ff · 7c284ff
1 parent 9510dbd
commit 7c284ff
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/src/liblesma/Backend/Codegen.cpp b/src/liblesma/Backend/Codegen.cpp
@@ -716,10 +716,20 @@ void Codegen::visit(Return *node) {
         visit(inst);
 
     isReturn = true;
-    if (node->getValue() == nullptr)
-        Builder->CreateRetVoid();
-    else
-        Builder->CreateRet(visit(node->getValue()));
+
+    if (node->getValue() == nullptr) {
+        if (Builder->getCurrentFunctionReturnType() == Builder->getVoidTy())
+            Builder->CreateRetVoid();
+        else
+            throw CodegenError(node->getSpan(), "Return type does not match the function return type");
+    }
+    else {
+        auto val = visit(node->getValue());
+        if (Builder->getCurrentFunctionReturnType() == val->getType())
+            Builder->CreateRet(val);
+        else
+            throw CodegenError(node->getSpan(), "Return type does not match the function return type");
+    }
 }
 
 void Codegen::visit(Defer *node) {