source: src/main.cc @ d5f1cfc

ADTaaron-thesisarm-ehast-experimentalcleanup-dtorsctordeferred_resndemanglerenumforall-pointer-decaygc_noraiijacob/cs343-translationjenkins-sandboxmemorynew-astnew-ast-unique-exprnew-envno_listpersistent-indexerpthread-emulationqualifiedEnumresolv-newwith_gc
Last change on this file since d5f1cfc was 2a7e29b, checked in by Rob Schluntz <rschlunt@…>, 8 years ago

fix crash when taking cfa-cpp input from stdin

  • Property mode set to 100644
File size: 10.3 KB
RevLine 
[b87a5ed]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//
[71f4e4f]7// main.cc --
[b87a5ed]8//
[843054c2]9// Author           : Richard C. Bilson
[b87a5ed]10// Created On       : Fri May 15 23:12:02 2015
[71f4e4f]11// Last Modified By : Rob Schluntz
[9e2c1f0]12// Last Modified On : Fri May 06 15:59:09 2016
[7b937575]13// Update Count     : 203
[b87a5ed]14//
15
[51b7345]16#include <iostream>
17#include <fstream>
[b87a5ed]18#include <cstdlib>
[51b7345]19#include <cstdio>
[b87a5ed]20#include <getopt.h>
[51b7345]21#include "Parser/Parser.h"
22#include "Parser/ParseNode.h"
23#include "Parser/LinkageSpec.h"
24#include "SynTree/Declaration.h"
25#include "SynTree/Visitor.h"
26#include "GenPoly/Lvalue.h"
27#include "GenPoly/Specialize.h"
28#include "GenPoly/Box.h"
29#include "GenPoly/CopyParams.h"
30#include "CodeGen/Generate.h"
31#include "CodeGen/FixNames.h"
32#include "ControlStruct/Mutate.h"
33#include "Tuples/Mutate.h"
34#include "Tuples/FunctionChecker.h"
35#include "SymTab/Mangler.h"
36#include "SymTab/Indexer.h"
37#include "SymTab/Validate.h"
38#include "ResolvExpr/AlternativePrinter.h"
39#include "ResolvExpr/Resolver.h"
40#include "MakeLibCfa.h"
41#include "InitTweak/Mutate.h"
[a0fdbd5]42#include "InitTweak/GenInit.h"
[71f4e4f]43#include "InitTweak/FixInit.h"
[711eee5]44#include "InitTweak/FixGlobalInit.h"
[51b7345]45//#include "Explain/GenProlog.h"
46//#include "Try/Visit.h"
47
[d3b7937]48#include "Common/SemanticError.h"
49#include "Common/UnimplementedError.h"
[51b7345]50
51#include "../config.h"
52
53using namespace std;
54
[81419b5]55#define OPTPRINT(x) \
56        if ( errorp ) std::cerr << x << std::endl;
57
[1ab4ce2]58static void parse( FILE * input, LinkageSpec::Type t, bool shouldExit = false );
[f77f12e2]59static void dump( std::list< Declaration * > & translationUnit, std::ostream & out = std::cout );
[81419b5]60
[b87a5ed]61bool
62        astp = false,
[de62360d]63        bresolvep = false,
[fea7ca7]64        bboxp = false,
[ca1c11f]65        ctorinitp = false,
[b87a5ed]66        exprp = false,
67        expraltp = false,
68        grammarp = false,
69        libcfap = false,
[de62360d]70        nopreludep = false,
[1ab4ce2]71        noprotop = false,
[de62360d]72        parsep = false,
[b87a5ed]73        resolvep = false,                                                                       // used in AlternativeFinder
74        symtabp = false,
[d3b7937]75        treep = false,
[b87a5ed]76        validp = false,
[de62360d]77        errorp = false,
78        codegenp = false;
[b87a5ed]79
[fea7ca7]80enum { Ast, Bbox, Bresolver, CtorInitFix, Expr, ExprAlt, Grammar, LibCFA, Nopreamble, Parse, Prototypes, Resolver, Symbol, Tree, Validate, };
[b87a5ed]81
82static struct option long_opts[] = {
83        { "ast", no_argument, 0, Ast },
[fea7ca7]84        { "before-box", no_argument, 0, Bbox },
[de62360d]85        { "before-resolver", no_argument, 0, Bresolver },
[ca1c11f]86        { "ctorinitfix", no_argument, 0, CtorInitFix },
[b87a5ed]87        { "expr", no_argument, 0, Expr },
88        { "expralt", no_argument, 0, ExprAlt },
89        { "grammar", no_argument, 0, Grammar },
90        { "libcfa", no_argument, 0, LibCFA },
[1ab4ce2]91        { "no-preamble", no_argument, 0, Nopreamble },
[b1d6dd5]92        { "parse", no_argument, 0, Parse },
[1ab4ce2]93        { "no-prototypes", no_argument, 0, Prototypes },
[b87a5ed]94        { "resolver", no_argument, 0, Resolver },
95        { "symbol", no_argument, 0, Symbol },
[d3b7937]96        { "tree", no_argument, 0, Tree },
[b1d6dd5]97        { "validate", no_argument, 0, Validate },
[b87a5ed]98        { 0, 0, 0, 0 }
99};
[51b7345]100
[8c17ab0]101int main( int argc, char *argv[] ) {
[b87a5ed]102        FILE *input;
103        std::ostream *output = &std::cout;
104        int long_index;
[7b937575]105        std::list< Declaration * > translationUnit;
[d029162e]106        const char *filename = NULL;
[b87a5ed]107
108        opterr = 0;                                                                                     // prevent getopt from printing error messages
[71f4e4f]109
[b87a5ed]110        int c;
[9e2c1f0]111        while ( (c = getopt_long( argc, argv, "abBcefglnpqrstvyzD:F:", long_opts, &long_index )) != -1 ) {
[b87a5ed]112                switch ( c ) {
113                  case Ast:
114                  case 'a':                                                                             // dump AST
115                        astp = true;
116                        break;
[de62360d]117                  case Bresolver:
118                  case 'b':                                                                             // print before resolver steps
119                        bresolvep = true;
120                        break;
[fea7ca7]121                  case 'B':                                                                             // print before resolver steps
122                        bboxp = true;
123                        break;
124                  case CtorInitFix:
125                  case 'c':
[ca1c11f]126                        ctorinitp = true;
127                        break;
[b87a5ed]128                  case Expr:
129                  case 'e':                                                                             // dump AST after expression analysis
130                        exprp = true;
131                        break;
132                  case ExprAlt:
133                  case 'f':                                                                             // print alternatives for expressions
134                        expraltp = true;
135                        break;
136                  case Grammar:
137                  case 'g':                                                                             // bison debugging info (grammar rules)
138                        grammarp = true;
139                        break;
140                  case LibCFA:
141                  case 'l':                                                                             // generate libcfa.c
142                        libcfap = true;
143                        break;
144                  case Nopreamble:
145                  case 'n':                                                                             // do not read preamble
[de62360d]146                        nopreludep = true;
[b87a5ed]147                        break;
148                  case Prototypes:
149                  case 'p':                                                                             // generate prototypes for preamble functions
[1ab4ce2]150                        noprotop = true;
[b87a5ed]151                        break;
152                  case Parse:
153                  case 'q':                                                                             // dump parse tree
154                        parsep = true;
155                        break;
156                  case Resolver:
157                  case 'r':                                                                             // print resolver steps
158                        resolvep = true;
159                        break;
160                  case Symbol:
161                  case 's':                                                                             // print symbol table events
162                        symtabp = true;
163                        break;
[d3b7937]164                  case Tree:
165                  case 't':                                                                             // build in tree
166                        treep = true;
167                        break;
[b1d6dd5]168                  case 'v':                                                                             // dump AST after decl validation pass
[b87a5ed]169                        validp = true;
170                        break;
171                  case 'y':
172                        errorp = true;
173                        break;
174                  case 'z':
175                        codegenp = true;
176                        break;
177                  case 'D':                                                                             // ignore -Dxxx
178                        break;
[7b937575]179                  case 'F':                                                                             // source file-name without suffix
180                        filename = optarg;
181                        break;
[b87a5ed]182                  case '?':
183                        cout << "Unknown option: '" << (char)optopt << "'" << endl;
184                        exit(1);
185                  default:
186                        abort();
187                } // switch
188        } // while
189
190        try {
[81419b5]191                // choose to read the program from a file or stdin
[b87a5ed]192                if ( optind < argc ) {
193                        input = fopen( argv[ optind ], "r" );
194                        if ( ! input ) {
[711eee5]195                                std::cout << "Error: can't open " << argv[ optind ] << std::endl;
[b87a5ed]196                                exit( 1 );
197                        } // if
[d029162e]198                        // if running cfa-cpp directly, might forget to pass -F option (and really shouldn't have to)
199                        if ( filename == NULL ) filename = argv[ optind ];
[37024fd]200                        // prelude filename comes in differently
201                        if ( libcfap ) filename = "prelude.cf";
[b87a5ed]202                        optind += 1;
203                } else {
204                        input = stdin;
[2a7e29b]205                        // if running cfa-cpp directly, might forget to pass -F option. Since this takes from stdin, pass
206                        // a fake name along
207                        if ( filename == NULL ) filename = "stdin";
[b87a5ed]208                } // if
209
210                if ( optind < argc ) {
211                        output = new ofstream( argv[ optind ] );
212                } // if
[71f4e4f]213
[b87a5ed]214                Parser::get_parser().set_debug( grammarp );
[81419b5]215
216                // read in the builtins and the prelude
[de62360d]217                if ( ! nopreludep ) {                                                   // include gcc builtins
[faf8857]218                        // -l is for initial build ONLY and builtins.cf is not in the lib directory so access it here.
[d3b7937]219                        FILE * builtins = fopen( libcfap | treep ? "builtins.cf" : CFA_LIBDIR "/builtins.cf", "r" );
[b87a5ed]220                        if ( builtins == NULL ) {
[d3b7937]221                                std::cerr << "Error: can't open builtins.cf" << std::endl;
[b87a5ed]222                                exit( 1 );
223                        } // if
224
[81419b5]225                        parse( builtins, LinkageSpec::Compiler );
226
227                        if ( ! libcfap ) {
[faf8857]228                                // read the prelude in, if not generating the cfa library
[d3b7937]229                                FILE * prelude = fopen( treep ? "prelude.cf" : CFA_LIBDIR "/prelude.cf", "r" );
[81419b5]230                                if ( prelude == NULL ) {
[d3b7937]231                                        std::cerr << "Error: can't open prelude.cf" << std::endl;
[81419b5]232                                        exit( 1 );
233                                } // if
[71f4e4f]234
[35304009]235                                parse( prelude, LinkageSpec::Intrinsic );
[b87a5ed]236                        } // if
237                } // if
[81419b5]238
[71f4e4f]239                parse( input, libcfap ? LinkageSpec::Intrinsic : LinkageSpec::Cforall, grammarp );
240
[b87a5ed]241                if ( parsep ) {
242                        Parser::get_parser().get_parseTree()->printList( std::cout );
243                        Parser::get_parser().freeTree();
244                        return 0;
245                } // if
246
247                buildList( Parser::get_parser().get_parseTree(), translationUnit );
248
249                Parser::get_parser().freeTree();
250                if ( astp ) {
[1ab4ce2]251                        dump( translationUnit );
[b87a5ed]252                        return 0;
253                } // if
254
[839ccbb]255                // add the assignment statement after the initialization of a type parameter
[81419b5]256                OPTPRINT( "validate" )
257                SymTab::validate( translationUnit, symtabp );
258                if ( symtabp ) {
[b87a5ed]259                        return 0;
260                } // if
261
[81419b5]262                if ( expraltp ) {
263                        ResolvExpr::AlternativePrinter printer( std::cout );
264                        acceptAll( translationUnit, printer );
[b87a5ed]265                        return 0;
266                } // if
267
268                if ( validp ) {
[1ab4ce2]269                        dump( translationUnit );
[b87a5ed]270                        return 0;
271                } // if
272
[81419b5]273                OPTPRINT( "mutate" )
274                ControlStruct::mutate( translationUnit );
[71f4e4f]275                OPTPRINT( "fixNames" )
[81419b5]276                CodeGen::fixNames( translationUnit );
[711eee5]277                OPTPRINT( "fixGlobalInit" );
[03e5d14]278                InitTweak::fixGlobalInit( translationUnit, filename, libcfap || treep );
[f1e012b]279                OPTPRINT( "tweakInit" )
[a0fdbd5]280                InitTweak::genInit( translationUnit );
[b87a5ed]281
[81419b5]282                if ( libcfap ) {
283                        // generate the bodies of cfa library functions
284                        LibCfa::makeLibCfa( translationUnit );
[b87a5ed]285                } // if
286
[de62360d]287                if ( bresolvep ) {
[1ab4ce2]288                        dump( translationUnit );
[de62360d]289                        return 0;
290                } // if
291
[81419b5]292                OPTPRINT( "resolve" )
[b87a5ed]293                ResolvExpr::resolve( translationUnit );
[81419b5]294                if ( exprp ) {
[1ab4ce2]295                        dump( translationUnit );
[ca1c11f]296                        return 0;
[81419b5]297                }
298
[f1e012b]299                OPTPRINT( "fixInit" )
[71f4e4f]300                // fix ObjectDecl - replaces ConstructorInit nodes
301                InitTweak::fix( translationUnit );
[ca1c11f]302                if ( ctorinitp ) {
303                        dump ( translationUnit );
304                        return 0;
305                }
[71f4e4f]306
[81419b5]307                OPTPRINT( "copyParams" );
[b87a5ed]308                GenPoly::copyParams( translationUnit );
[81419b5]309                OPTPRINT( "convertSpecializations" )
[698664b3]310                GenPoly::convertSpecializations( translationUnit );
[81419b5]311                OPTPRINT( "convertLvalue" )
[b87a5ed]312                GenPoly::convertLvalue( translationUnit );
[fea7ca7]313
314                if ( bboxp ) {
315                        dump( translationUnit );
316                        return 0;
317                }
[81419b5]318                OPTPRINT( "box" )
[b87a5ed]319                GenPoly::box( translationUnit );
[81419b5]320
[2871210]321                // print tree right before code generation
[81419b5]322                if ( codegenp ) {
[1ab4ce2]323                        dump( translationUnit );
[81419b5]324                        return 0;
325                } // if
[b87a5ed]326
[1ab4ce2]327                CodeGen::generate( translationUnit, *output, ! noprotop );
[b87a5ed]328
329                if ( output != &std::cout ) {
330                        delete output;
331                } // if
332        } catch ( SemanticError &e ) {
333                if ( errorp ) {
[f77f12e2]334                        std::cerr << "---AST at error:---" << std::endl;
335                        dump( translationUnit, std::cerr );
336                        std::cerr << std::endl << "---End of AST, begin error message:---\n" << std::endl;
[b87a5ed]337                }
[81419b5]338                e.print( std::cerr );
[b87a5ed]339                if ( output != &std::cout ) {
340                        delete output;
341                } // if
342                return 1;
343        } catch ( UnimplementedError &e ) {
344                std::cout << "Sorry, " << e.get_what() << " is not currently implemented" << std::endl;
345                if ( output != &std::cout ) {
346                        delete output;
347                } // if
348                return 1;
349        } catch ( CompilerError &e ) {
350                std::cerr << "Compiler Error: " << e.get_what() << std::endl;
351                std::cerr << "(please report bugs to " << std::endl;
352                if ( output != &std::cout ) {
353                        delete output;
354                } // if
355                return 1;
356        } // try
357
[39786813]358        deleteAll( translationUnit );
[b87a5ed]359        return 0;
[d9a0e76]360} // main
[51b7345]361
[1ab4ce2]362static void parse( FILE * input, LinkageSpec::Type linkage, bool shouldExit ) {
[81419b5]363        Parser::get_parser().set_linkage( linkage );
364        Parser::get_parser().parse( input );
365
366        fclose( input );
367        if ( shouldExit || Parser::get_parser().get_parseStatus() != 0 ) {
368                exit( Parser::get_parser().get_parseStatus() );
369        } // if
370}
371
[1ab4ce2]372static bool notPrelude( Declaration * decl ) {
373        return ! LinkageSpec::isBuiltin( decl->get_linkage() );
374}
375
[f77f12e2]376static void dump( std::list< Declaration * > & translationUnit, std::ostream & out ) {
[1ab4ce2]377        std::list< Declaration * > decls;
378        if ( noprotop ) {
[71f4e4f]379                filter( translationUnit.begin(), translationUnit.end(),
[1ab4ce2]380                                std::back_inserter( decls ), notPrelude );
381        } else {
382                decls = translationUnit;
383        }
384
[f77f12e2]385        printAll( decls, out );
[7f5566b]386        deleteAll( translationUnit );
[1ab4ce2]387}
388
389
[51b7345]390// Local Variables: //
[b87a5ed]391// tab-width: 4 //
392// mode: c++ //
393// compile-command: "make install" //
[51b7345]394// End:  //
Note: See TracBrowser for help on using the repository browser.