source: src/CodeGen/FixMain.cc @ 5d41e22

//
// 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();
                }
        }
};