Added one more check in ReturnExpr when void type is involved.

mzweilin · Nov 19, 2013 · 586224d · 586224d
1 parent 5ac258b
commit 586224d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/wx4ed/src/pp4/ast_stmt.cc b/wx4ed/src/pp4/ast_stmt.cc
@@ -193,8 +193,10 @@ void ReturnStmt::Check() {
     if (!given->IsEquivalentTo(expected))
         ReportError::ReturnMismatch(this, given, expected);
 
+    EmptyExpr *ee = dynamic_cast<EmptyExpr*>(expr);
+    if (ee != NULL && expected != Type::voidType)
     //if (given == Type::errorType)
-    //    ReportError::ReturnMismatch(this, Type::voidType, expected);
+        ReportError::ReturnMismatch(this, Type::voidType, expected);
 }
 
 PrintStmt::PrintStmt(List<Expr*> *a) {