source: src/CodeGen/FixMain.cc

Last change on this file was ca9d65e, checked in by Peter A. Buhr <pabuhr@…>, 3 months ago

second attempt at simplifying SemanticError? messages

  • Property mode set to 100644
File size: 4.7 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 -- Tools to change a Cforall main into a C main.
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     : 1
14//
15
16#include "FixMain.h"
17
18#include <cassert>                 // for assert, assertf
19#include <fstream>                 // for operator<<, basic_ostream::operator<<
20#include <list>                    // for list
21#include <string>                  // for operator<<
22
23#include "AST/Decl.hpp"
24#include "AST/Pass.hpp"
25#include "AST/Type.hpp"
26#include "AST/Vector.hpp"
27#include "Common/SemanticError.h"  // for SemanticError
28#include "CodeGen/GenType.h"       // for GenType
29#include "SymTab/Mangler.h"
30
31namespace CodeGen {
32
33namespace {
34
35struct FindMainCore final {
36        ast::FunctionDecl const * main_declaration = nullptr;
37
38        void previsit( ast::FunctionDecl const * decl ) {
39                if ( isMain( decl ) ) {
40                        if ( main_declaration ) {
41                                SemanticError( decl, "Multiple definition of main routine" );
42                        }
43                        main_declaration = decl;
44                }
45        }
46};
47
48std::string genTypeAt( const ast::vector<ast::Type> & types, size_t at ) {
49        return genType( types[at], "", Options( false, false, false, false ) );
50}
51
52ast::ObjectDecl * makeIntObj(){
53        return new ast::ObjectDecl( CodeLocation(), "",
54                new ast::BasicType( ast::BasicType::SignedInt ) );
55}
56
57ast::ObjectDecl * makeCharStarStarObj() {
58        return new ast::ObjectDecl( CodeLocation(), "",
59                new ast::PointerType(
60                        new ast::PointerType(
61                                new ast::BasicType( ast::BasicType::Char ) ) ) );
62}
63
64std::string getMangledNameOfMain(
65                ast::vector<ast::DeclWithType> && params, ast::ArgumentFlag isVarArgs ) {
66        ast::ptr<ast::FunctionDecl> decl = new ast::FunctionDecl(
67                CodeLocation(),
68                "main",
69                ast::vector<ast::TypeDecl>(),
70                ast::vector<ast::DeclWithType>(),
71                std::move( params ),
72                { makeIntObj() },
73                nullptr,
74                ast::Storage::Classes(),
75                ast::Linkage::Spec(),
76                ast::vector<ast::Attribute>(),
77                ast::Function::Specs(),
78                isVarArgs
79        );
80        return Mangle::mangle( decl.get() );
81}
82
83std::string getMangledNameOf0ParameterMain() {
84        return getMangledNameOfMain( {}, ast::VariableArgs );
85}
86
87std::string getMangledNameOf2ParameterMain() {
88        return getMangledNameOfMain( {
89                makeIntObj(),
90                makeCharStarStarObj(),
91        }, ast::FixedArgs );
92}
93
94bool is_main( const std::string & mangled_name ) {
95        // This breaks if you move it out of the function.
96        static const std::string mangled_mains[] = {
97                getMangledNameOf0ParameterMain(),
98                getMangledNameOf2ParameterMain(),
99                //getMangledNameOf3ParameterMain(),
100        };
101
102        for ( auto main_name : mangled_mains ) {
103                if ( main_name == mangled_name ) return true;
104        }
105        return false;
106}
107
108struct FixLinkageCore final {
109        ast::Linkage::Spec const spec;
110        FixLinkageCore( ast::Linkage::Spec spec ) : spec( spec ) {}
111
112        ast::FunctionDecl const * previsit( ast::FunctionDecl const * decl ) {
113                if ( decl->name != "main" ) return decl;
114                return ast::mutate_field( decl, &ast::FunctionDecl::linkage, spec );
115        }
116};
117
118} // namespace
119
120bool isMain( const ast::FunctionDecl * decl ) {
121        if ( std::string("main") != decl->name ) {
122                return false;
123        }
124        return is_main( Mangle::mangle( decl, Mangle::Type ) );
125}
126
127void fixMainLinkage( ast::TranslationUnit & translationUnit,
128                bool replace_main ) {
129        ast::Linkage::Spec const spec =
130                ( replace_main ) ? ast::Linkage::Cforall : ast::Linkage::C;
131        ast::Pass<FixLinkageCore>::run( translationUnit, spec );
132}
133
134void fixMainInvoke( ast::TranslationUnit & translationUnit,
135                std::ostream &os, const char * bootloader_filename ) {
136
137        ast::Pass<FindMainCore> main_finder;
138        ast::accept_all( translationUnit, main_finder );
139        if ( nullptr == main_finder.core.main_declaration ) return;
140
141        ast::FunctionDecl * main_declaration =
142                ast::mutate( main_finder.core.main_declaration );
143
144        main_declaration->mangleName = Mangle::mangle( main_declaration );
145
146        os << "static inline int invoke_main(int argc, char* argv[], char* envp[]) { (void)argc; (void)argv; (void)envp; return ";
147        os << main_declaration->scopedMangleName() << "(";
148        const auto& params = main_declaration->type->params;
149        switch ( params.size() ) {
150                case 3: os << "(" << genTypeAt(params, 0) << ")argc, (" << genTypeAt(params, 1) << ")argv, (" << genTypeAt(params, 2) << ")envp"; break;
151                case 2: os << "(" << genTypeAt(params, 0) << ")argc, (" << genTypeAt(params, 1) << ")argv"; break;
152                case 0: break;
153                default : assert(false);
154        }
155        os << "); }\n";
156
157        std::ifstream bootloader( bootloader_filename, std::ios::in );
158        assertf( bootloader.is_open(), "cannot open bootloader.c\n" );
159        os << bootloader.rdbuf();
160}
161
162} // namespace CodeGen
Note: See TracBrowser for help on using the repository browser.