Changes in src/SynTree/FunctionDecl.cc [faddbd8:3fe34ae]

File:

• src/SynTree/FunctionDecl.cc (modified) (1 diff)

src/SynTree/FunctionDecl.cc

@@ -23 +23 @@
 #include "InitTweak/InitTweak.h"
 
+extern bool translation_unit_nomain;
+
 FunctionDecl::FunctionDecl( const std::string &name, DeclarationNode::StorageClass sc, LinkageSpec::Spec linkage, FunctionType *type, CompoundStmt *statements, bool isInline, bool isNoreturn, std::list< Attribute * > attributes )
                 : Parent( name, sc, linkage, attributes ), type( type ), statements( statements ) {
         set_isInline( isInline );
         set_isNoreturn( isNoreturn );
-        // this is a brazen hack to force the function "main" to have C linkage
+        // this is a brazen hack to force the function "main" to have Cforall linkage
+        // because we want to replace the main even if it is inside an extern
         if ( name == "main" ) {
-                set_linkage( LinkageSpec::C );
+                set_linkage( translation_unit_nomain ? LinkageSpec::C : LinkageSpec::Cforall );
         } // if
 }