Changeset 13de47bc for src/CodeGen/FixMain.cc

Timestamp:

Jan 12, 2017, 3:28:29 PM (8 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:

aaa1a99a

Parents:

7cc2c8d

Message:

Proper bootloader boilerplate implemented

File:

• src/CodeGen/FixMain.cc (modified) (1 diff)

src/CodeGen/FixMain.cc

@@ +1 @@
+//
+// Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo
+//
+// The contents of this file are covered under the licence agreement in the
+// file "LICENCE" distributed with Cforall.
+//
+// FixMain.cc -- 
+//
+// Author           : Thierry Delisle
+// Created On       : Thr Jan 12 14:11:09 2017
+// Last Modified By : 
+// Last Modified On : 
+// Update Count     : 0
+//
 
+
+#include "FixMain.h"    
+
+#include <fstream>
+#include <iostream>
+
+#include "Common/SemanticError.h"
+#include "SynTree/Declaration.h"
+
+namespace CodeGen {
+        bool FixMain::replace_main = false;
+        std::unique_ptr<FunctionDecl> FixMain::main_signature = nullptr;
+        
+        void FixMain::registerMain(FunctionDecl* functionDecl) 
+        {
+                if(main_signature) { 
+                        throw SemanticError("Multiple definition of main routine\n", functionDecl); 
+                }
+                main_signature.reset( functionDecl->clone() );
+        }
+
+        void FixMain::fix(std::ostream &os, const char* bootloader_filename) {
+                if( main_signature ) {
+                        os << "static inline int invoke_main(int argc, char* argv[], char* envp[]) { (void)argc; (void)argv; (void)envp; return ";
+
+                        os << main_signature->get_scopedMangleName() << "(";
+                        switch(main_signature->get_functionType()->get_parameters().size()) {
+                                case 3: os << "argc, argv, envp"; break;
+                                case 2: os << "argc, argv"; break;
+                                case 0: break;
+                                default : assert(false);
+                        }
+                        os << "); }\n";
+
+                        std::ifstream bootloader(bootloader_filename, std::ios::in);
+                        assertf( bootloader.is_open(), "cannot open bootloader.c\n" );
+                        os << bootloader.rdbuf();
+                }
+        }
+};