source: src/CodeGen/FixMain.cc @ bf2438c

aaron-thesisarm-ehcleanup-dtorsdeferred_resndemanglerjacob/cs343-translationjenkins-sandboxnew-astnew-ast-unique-exprnew-envno_listpersistent-indexerresolv-newwith_gc
Last change on this file since bf2438c was bf2438c, checked in by Thierry Delisle <tdelisle@…>, 4 years ago

Cleaned-up some headers using a tool called 'include-what-you-use'

  • Property mode set to 100644
File size: 1.8 KB
Line 
1//
2// Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo
3//
4// The contents of this file are covered under the licence agreement in the
5// file "LICENCE" distributed with Cforall.
6//
7// FixMain.cc --
8//
9// Author           : Thierry Delisle
10// Created On       : Thr Jan 12 14:11:09 2017
11// Last Modified By :
12// Last Modified On :
13// Update Count     : 0
14//
15
16
17#include "FixMain.h"
18
19#include <cassert>                 // for assert, assertf
20#include <fstream>                 // for operator<<, basic_ostream::operator<<
21#include <list>                    // for list
22#include <string>                  // for operator<<
23
24#include "Common/SemanticError.h"  // for SemanticError
25#include "SynTree/Declaration.h"   // for FunctionDecl, operator<<
26#include "SynTree/Type.h"          // for FunctionType
27
28namespace CodeGen {
29        bool FixMain::replace_main = false;
30        std::unique_ptr<FunctionDecl> FixMain::main_signature = nullptr;
31
32        void FixMain::registerMain(FunctionDecl* functionDecl)
33        {
34                if(main_signature) {
35                        throw SemanticError("Multiple definition of main routine\n", functionDecl);
36                }
37                main_signature.reset( functionDecl->clone() );
38        }
39
40        void FixMain::fix(std::ostream &os, const char* bootloader_filename) {
41                if( main_signature ) {
42                        os << "static inline int invoke_main(int argc, char* argv[], char* envp[]) { (void)argc; (void)argv; (void)envp; return ";
43
44                        os << main_signature->get_scopedMangleName() << "(";
45                        switch(main_signature->get_functionType()->get_parameters().size()) {
46                                case 3: os << "argc, argv, envp"; break;
47                                case 2: os << "argc, argv"; break;
48                                case 0: break;
49                                default : assert(false);
50                        }
51                        os << "); }\n";
52
53                        std::ifstream bootloader(bootloader_filename, std::ios::in);
54                        assertf( bootloader.is_open(), "cannot open bootloader.c\n" );
55                        os << bootloader.rdbuf();
56                }
57        }
58};
Note: See TracBrowser for help on using the repository browser.