Changes in src/CodeGen/FixNames.cc [aaa1a99a:0270824]

File:

• src/CodeGen/FixNames.cc (modified) (2 diffs)

src/CodeGen/FixNames.cc

@@ -22 +22 @@
 #include "SymTab/Mangler.h"
 #include "OperatorTable.h"
-#include "FixMain.h"
+
+extern std::unique_ptr<FunctionDecl> translation_unit_main_signature;
 
 namespace CodeGen {
@@ -118 +119 @@
 
                 if(is_main( SymTab::Mangler::mangle(functionDecl, true, true) )) {
-                        int nargs = functionDecl->get_functionType()->get_parameters().size();
-                        if( !(nargs == 0 || nargs == 2 || nargs == 3) ) {
-                                throw SemanticError("Main expected to have 0, 2 or 3 arguments\n", functionDecl); 
+                        if(translation_unit_main_signature) { 
+                                throw SemanticError("Multiple definition of main routine\n", functionDecl); 
                         }
+
                         functionDecl->get_statements()->get_kids().push_back( new ReturnStmt( noLabels, new ConstantExpr( Constant( new BasicType( Type::Qualifiers(), BasicType::SignedInt ), "0") ) ) );
-                        CodeGen::FixMain::registerMain( functionDecl );
+                        translation_unit_main_signature.reset( functionDecl->clone() );
                 }
         }