Changeset 0270824 for src/CodeGen

Timestamp:

Jan 12, 2017, 11:31:57 AM (9 years ago)

Author:

Thierry Delisle <tdelisle@…>

Branches:

ADT, aaron-thesis, arm-eh, ast-experimental, cleanup-dtors, deferred_resn, demangler, enum, forall-pointer-decay, jacob/cs343-translation, jenkins-sandbox, master, new-ast, new-ast-unique-expr, new-env, no_list, persistent-indexer, pthread-emulation, qualifiedEnum, resolv-new, with_gc

Children:

3fe34ae

Parents:

075734f

Message:

Replace user main with custom main, prototype

File:

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

src/CodeGen/FixNames.cc

@@ -14 +14 @@
 //
 
+#include <memory>
+
 #include "FixNames.h"
 #include "SynTree/Declaration.h"
@@ -20 +22 @@
 #include "SymTab/Mangler.h"
 #include "OperatorTable.h"
+
+extern std::unique_ptr<FunctionDecl> translation_unit_main_signature;
 
 namespace CodeGen {
@@ -28 +32 @@
 
                 virtual void visit( CompoundStmt *compoundStmt );
-
           private:
                 int scopeLevel = 1;
@@ -34 +37 @@
                 void fixDWT( DeclarationWithType *dwt );
         };
+
+        std::string mangle_main() {
+                FunctionType* main_type;
+                std::unique_ptr<FunctionDecl> mainDecl { new FunctionDecl(
+                        "main", 
+                        DeclarationNode::NoStorageClass, 
+                        LinkageSpec::Cforall, 
+                        main_type = new FunctionType( Type::Qualifiers(), true ), 
+                        nullptr, false, false
+                ) };
+                main_type->get_returnVals().push_back( 
+                        new ObjectDecl( "", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, new BasicType( Type::Qualifiers(), BasicType::SignedInt ), nullptr )
+                );
+
+                auto&& name = SymTab::Mangler::mangle( mainDecl.get() );
+                // std::cerr << name << std::endl;
+                return name;
+        }
+        std::string mangle_main_args() {
+                FunctionType* main_type;
+                std::unique_ptr<FunctionDecl> mainDecl { new FunctionDecl(
+                        "main", 
+                        DeclarationNode::NoStorageClass, 
+                        LinkageSpec::Cforall, 
+                        main_type = new FunctionType( Type::Qualifiers(), false ), 
+                        nullptr, false, false
+                ) };
+                main_type->get_returnVals().push_back( 
+                        new ObjectDecl( "", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, new BasicType( Type::Qualifiers(), BasicType::SignedInt ), nullptr )
+                );
+
+                mainDecl->get_functionType()->get_parameters().push_back( 
+                        new ObjectDecl( "", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, new BasicType( Type::Qualifiers(), BasicType::SignedInt ), nullptr )
+                );
+
+                mainDecl->get_functionType()->get_parameters().push_back( 
+                        new ObjectDecl( "", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, 
+                        new PointerType( Type::Qualifiers(), new PointerType( Type::Qualifiers(), new BasicType( Type::Qualifiers(), BasicType::Char ) ) ), 
+                        nullptr )
+                );
+
+                auto&& name = SymTab::Mangler::mangle( mainDecl.get() );
+                // std::cerr << name << std::endl;
+                return name;
+        }
+
+        bool is_main(const std::string& name) {
+                static std::string mains[] = { 
+                        mangle_main(), 
+                        mangle_main_args()
+                };
+
+                for(const auto& m : mains) {
+                        if( name == m ) return true;
+                }
+                return false;
+        }
 
         void fixNames( std::list< Declaration* > translationUnit ) {
@@ -57 +117 @@
                 Visitor::visit( functionDecl );
                 fixDWT( functionDecl );
+
+                if(is_main( SymTab::Mangler::mangle(functionDecl, true, true) )) {
+                        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") ) ) );
+                        translation_unit_main_signature.reset( functionDecl->clone() );
+                }
         }