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

File:

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

src/CodeGen/FixNames.cc

@@ -22 +22 @@
 #include "SymTab/Mangler.h"
 #include "OperatorTable.h"
-
-extern std::unique_ptr<FunctionDecl> translation_unit_main_signature;
+#include "FixMain.h"
 
 namespace CodeGen {
@@ -119 +118 @@
 
                 if(is_main( SymTab::Mangler::mangle(functionDecl, true, true) )) {
-                        if(translation_unit_main_signature) { 
-                                throw SemanticError("Multiple definition of main routine\n", functionDecl); 
+                        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); 
                         }
-
                         functionDecl->get_statements()->get_kids().push_back( new ReturnStmt( noLabels, new ConstantExpr( Constant( new BasicType( Type::Qualifiers(), BasicType::SignedInt ), "0") ) ) );
-                        translation_unit_main_signature.reset( functionDecl->clone() );
+                        CodeGen::FixMain::registerMain( functionDecl );
                 }
         }