source: src/main.cc @ c532847

ADTarm-ehast-experimentalenumforall-pointer-decayjacob/cs343-translationnew-ast-unique-exprpthread-emulationqualifiedEnum
Last change on this file since c532847 was e7d6968, checked in by Fangren Yu <f37yu@…>, 4 years ago

Merge branch 'master' of plg.uwaterloo.ca:software/cfa/cfa-cc into master

  • Property mode set to 100644
File size: 26.0 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//
[3e96559]9// Author           : Peter Buhr and Rob Schluntz
[b87a5ed]10// Created On       : Fri May 15 23:12:02 2015
[3e9de01]11// Last Modified By : Peter A. Buhr
12// Last Modified On : Thu Oct  8 18:17:46 2020
13// Update Count     : 637
[b87a5ed]14//
15
[bf2438c]16#include <cxxabi.h>                         // for __cxa_demangle
17#include <execinfo.h>                       // for backtrace, backtrace_symbols
18#include <getopt.h>                         // for no_argument, optind, geto...
[08fc48f]19#include <cassert>                          // for assertf
[bf2438c]20#include <cstdio>                           // for fopen, FILE, fclose, stdin
21#include <cstdlib>                          // for exit, free, abort, EXIT_F...
[bffcd66]22#include <csignal>                          // for signal, SIGABRT, SIGSEGV
[bf2438c]23#include <cstring>                          // for index
[be9288a]24#include <fstream>                          // for ofstream
[bf2438c]25#include <iostream>                         // for operator<<, basic_ostream
[62ce290]26#include <iomanip>
[bf2438c]27#include <iterator>                         // for back_inserter
28#include <list>                             // for list
[08fc48f]29#include <string>                           // for char_traits, operator<<
[e6955b1]30
[bffcd66]31using namespace std;
32
[9ea38de]33#include "AST/Convert.hpp"
[7f38b67a]34#include "CompilationState.h"
[bf2438c]35#include "../config.h"                      // for CFA_LIBDIR
36#include "CodeGen/FixMain.h"                // for FixMain
37#include "CodeGen/FixNames.h"               // for fixNames
38#include "CodeGen/Generate.h"               // for generate
39#include "CodeTools/DeclStats.h"            // for printDeclStats
[3b3491b]40#include "CodeTools/ResolvProtoDump.h"      // for dumpAsResolvProto
[bf2438c]41#include "CodeTools/TrackLoc.h"             // for fillLocations
42#include "Common/CompilerError.h"           // for CompilerError
[7abee38]43#include "Common/Stats.h"
[cbbd5b48]44#include "Common/PassVisitor.h"
[bf2438c]45#include "Common/SemanticError.h"           // for SemanticError
46#include "Common/UnimplementedError.h"      // for UnimplementedError
47#include "Common/utility.h"                 // for deleteAll, filter, printAll
[9f5ecf5]48#include "Concurrency/Waitfor.h"            // for generateWaitfor
[bf2438c]49#include "ControlStruct/ExceptTranslate.h"  // for translateEHM
50#include "ControlStruct/Mutate.h"           // for mutate
51#include "GenPoly/Box.h"                    // for box
52#include "GenPoly/InstantiateGeneric.h"     // for instantiateGeneric
53#include "GenPoly/Lvalue.h"                 // for convertLvalue
54#include "GenPoly/Specialize.h"             // for convertSpecializations
55#include "InitTweak/FixInit.h"              // for fix
56#include "InitTweak/GenInit.h"              // for genInit
57#include "MakeLibCfa.h"                     // for makeLibCfa
58#include "Parser/ParseNode.h"               // for DeclarationNode, buildList
59#include "Parser/TypedefTable.h"            // for TypedefTable
60#include "ResolvExpr/AlternativePrinter.h"  // for AlternativePrinter
61#include "ResolvExpr/Resolver.h"            // for resolve
62#include "SymTab/Validate.h"                // for validate
[bffcd66]63#include "SynTree/LinkageSpec.h"            // for Spec, Cforall, Intrinsic
[bf2438c]64#include "SynTree/Declaration.h"            // for Declaration
65#include "SynTree/Visitor.h"                // for acceptAll
66#include "Tuples/Tuples.h"                  // for expandMemberTuples, expan...
[a5f0529]67#include "Virtual/ExpandCasts.h"            // for expandCasts
[51b7345]68
[4615ac8]69
[3e96559]70static void NewPass( const char * const name ) {
71        Stats::Heap::newPass( name );
[1cb7fab2]72        using namespace Stats::Counters;
[b8665e3]73        {
[3e96559]74                static auto group = build<CounterGroup>( "Pass Visitor" );
75                auto pass = build<CounterGroup>( name, group );
[b8665e3]76                pass_visitor_stats.depth = 0;
[3e96559]77                pass_visitor_stats.avg = build<AverageCounter<double>>( "Average Depth", pass );
78                pass_visitor_stats.max = build<MaxCounter<double>>( "Max Depth", pass );
[b8665e3]79        }
80        {
[3e96559]81                static auto group = build<CounterGroup>( "Syntax Node" );
82                auto pass = build<CounterGroup>( name, group );
83                BaseSyntaxNode::new_nodes = build<SimpleCounter>( "Allocs", pass );
[b8665e3]84        }
[675716e]85}
86
[3e96559]87#define PASS( name, pass )                  \
[ecaeac6e]88        if ( errorp ) { cerr << name << endl; } \
[675716e]89        NewPass(name);                          \
[4f97937]90        Stats::Time::StartBlock(name);          \
91        pass;                                   \
92        Stats::Time::StopBlock();
[0da3e2c]93
[8b7ee09]94LinkageSpec::Spec linkage = LinkageSpec::Cforall;
[0da3e2c]95TypedefTable typedefTable;
[cbaee0d]96DeclarationNode * parseTree = nullptr;                                  // program parse tree
[81419b5]97
[ef22ad6]98static bool waiting_for_gdb = false;                                    // flag to set cfa-cpp to wait for gdb on start
[dee1f89]99
[bffcd66]100static string PreludeDirector = "";
[4dcaed2]101
[e0bd0f9]102static void parse_cmdline( int argc, char *argv[] );
[8b7ee09]103static void parse( FILE * input, LinkageSpec::Spec linkage, bool shouldExit = false );
[e6955b1]104static void dump( list< Declaration * > & translationUnit, ostream & out = cout );
105
[0afffee]106static void backtrace( int start ) {                                    // skip first N stack frames
[74330e7]107        enum { Frames = 50, };                                                          // maximum number of stack frames
[e6955b1]108        void * array[Frames];
[74330e7]109        size_t size = ::backtrace( array, Frames );
[0afffee]110        char ** messages = ::backtrace_symbols( array, size ); // does not demangle names
111
112        *index( messages[0], '(' ) = '\0';                                      // find executable name
113        cerr << "Stack back trace for: " << messages[0] << endl;
[e6955b1]114
[b542bfb]115        // skip last 2 stack frames after main
[74330e7]116        for ( unsigned int i = start; i < size - 2 && messages != nullptr; i += 1 ) {
[e6955b1]117                char * mangled_name = nullptr, * offset_begin = nullptr, * offset_end = nullptr;
[7006ba5]118
119                for ( char * p = messages[i]; *p; p += 1 ) {    // find parantheses and +offset
[0afffee]120                        if ( *p == '(' ) {
[46f6134]121                                mangled_name = p;
[0afffee]122                        } else if ( *p == '+' ) {
[e6955b1]123                                offset_begin = p;
[0afffee]124                        } else if ( *p == ')' ) {
[e6955b1]125                                offset_end = p;
126                                break;
127                        } // if
128                } // for
129
130                // if line contains symbol, attempt to demangle
[b542bfb]131                int frameNo = i - start;
[e6955b1]132                if ( mangled_name && offset_begin && offset_end && mangled_name < offset_begin ) {
[0afffee]133                        *mangled_name++ = '\0';                                         // delimit strings
[e6955b1]134                        *offset_begin++ = '\0';
135                        *offset_end++ = '\0';
136
[0afffee]137                        int status;
[e6955b1]138                        char * real_name = __cxxabiv1::__cxa_demangle( mangled_name, 0, 0, &status );
[0afffee]139                        // bug in __cxa_demangle for single-character lower-case non-mangled names
[e6955b1]140                        if ( status == 0 ) {                                            // demangling successful ?
[b542bfb]141                                cerr << "(" << frameNo << ") " << messages[i] << " : "
[e6955b1]142                                         << real_name << "+" << offset_begin << offset_end << endl;
143                        } else {                                                                        // otherwise, output mangled name
[b542bfb]144                                cerr << "(" << frameNo << ") " << messages[i] << " : "
[0afffee]145                                         << mangled_name << "(/*unknown*/)+" << offset_begin << offset_end << endl;
[e6955b1]146                        } // if
[0afffee]147
[e6955b1]148                        free( real_name );
149                } else {                                                                                // otherwise, print the whole line
[b542bfb]150                        cerr << "(" << frameNo << ") " << messages[i] << endl;
[e6955b1]151                } // if
152        } // for
[b542bfb]153
[e6955b1]154        free( messages );
[b542bfb]155} // backtrace
156
[bffcd66]157#define SIGPARMS int sig __attribute__(( unused )), siginfo_t * sfp __attribute__(( unused )), ucontext_t * cxt __attribute__(( unused ))
158
159static void Signal( int sig, void (*handler)(SIGPARMS), int flags ) {
160        struct sigaction act;
161
162        act.sa_sigaction = (void (*)(int, siginfo_t *, void *))handler;
163        act.sa_flags = flags;
164
165        if ( sigaction( sig, &act, nullptr ) == -1 ) {
166            cerr << "*CFA runtime error* problem installing signal handler, error(" << errno << ") " << strerror( errno ) << endl;
167            _exit( EXIT_FAILURE );
168        } // if
169} // Signal
170
171static void sigSegvBusHandler( SIGPARMS ) {
172        if ( sfp->si_addr == nullptr ) {
173                cerr << "Null pointer (nullptr) dereference." << endl;
174        } else {
175                cerr << (sig == SIGSEGV ? "Segment fault" : "Bus error") << " at memory location " << sfp->si_addr << "." << endl
176                         << "Possible cause is reading outside the address space or writing to a protected area within the address space with an invalid pointer or subscript." << endl;
177        } // if
[b542bfb]178        backtrace( 2 );                                                                         // skip first 2 stack frames
[3e96559]179        abort();                                                                                        // cause core dump for debugging
[e6955b1]180} // sigSegvBusHandler
[0da3e2c]181
[74330e7]182static void sigFpeHandler( SIGPARMS ) {
183        const char * msg;
184
185        switch ( sfp->si_code ) {
186          case FPE_INTDIV: case FPE_FLTDIV: msg = "divide by zero"; break;
187          case FPE_FLTOVF: msg = "overflow"; break;
188          case FPE_FLTUND: msg = "underflow"; break;
189          case FPE_FLTRES: msg = "inexact result"; break;
190          case FPE_FLTINV: msg = "invalid operation"; break;
191          default: msg = "unknown";
192        } // choose
193        cerr << "Computation error " << msg << " at location " << sfp->si_addr << endl
194                 << "Possible cause is constant-expression evaluation invalid." << endl;
195        backtrace( 2 );                                                                         // skip first 2 stack frames
196        abort();                                                                                        // cause core dump for debugging
197} // sigFpeHandler
198
[bffcd66]199static void sigAbortHandler( SIGPARMS ) {
[b542bfb]200        backtrace( 6 );                                                                         // skip first 6 stack frames
[bffcd66]201        Signal( SIGABRT, (void (*)(SIGPARMS))SIG_DFL, SA_SIGINFO );     // reset default signal handler
[9be45a2]202        raise( SIGABRT );                                                                       // reraise SIGABRT
[b542bfb]203} // sigAbortHandler
204
[cbaee0d]205int main( int argc, char * argv[] ) {
[3b8e52c]206        FILE * input;                                                                           // use FILE rather than istream because yyin is FILE
[d08beee]207        ostream * output = & cout;
[e6955b1]208        list< Declaration * > translationUnit;
209
[bffcd66]210        Signal( SIGSEGV, sigSegvBusHandler, SA_SIGINFO );
211        Signal( SIGBUS, sigSegvBusHandler, SA_SIGINFO );
[74330e7]212        Signal( SIGFPE, sigFpeHandler, SA_SIGINFO );
[bffcd66]213        Signal( SIGABRT, sigAbortHandler, SA_SIGINFO );
[b87a5ed]214
[bffcd66]215        // cout << "main" << endl;
[44bca7f]216        // for ( int i = 0; i < argc; i += 1 ) {
[bffcd66]217        //      cout << '\t' << argv[i] << endl;
[44bca7f]218        // } // for
219
[e0bd0f9]220        parse_cmdline( argc, argv );                                            // process command-line arguments
[13de47bc]221        CodeGen::FixMain::setReplaceMain( !nomainp );
[b87a5ed]222
[ef22ad6]223        if ( waiting_for_gdb ) {
[bffcd66]224                cerr << "Waiting for gdb" << endl;
225                cerr << "run :" << endl;
226                cerr << "  gdb attach " << getpid() << endl;
[dee1f89]227                raise(SIGSTOP);
[ef22ad6]228        } // if
[dee1f89]229
[b87a5ed]230        try {
[81419b5]231                // choose to read the program from a file or stdin
[3b8e52c]232                if ( optind < argc ) {                                                  // any commands after the flags ? => input file name
[b87a5ed]233                        input = fopen( argv[ optind ], "r" );
[e0bd0f9]234                        assertf( input, "cannot open %s because %s\n", argv[ optind ], strerror( errno ) );
[b87a5ed]235                        optind += 1;
[3b8e52c]236                } else {                                                                                // no input file name
[b87a5ed]237                        input = stdin;
238                } // if
239
[79eaeb7]240                Stats::Time::StartGlobal();
[3c0d4cd]241                NewPass("Parse");
242                Stats::Time::StartBlock("Parse");
[675716e]243
[159c62e]244                // read in the builtins, extras, and the prelude
[de62360d]245                if ( ! nopreludep ) {                                                   // include gcc builtins
[faf8857]246                        // -l is for initial build ONLY and builtins.cf is not in the lib directory so access it here.
[807ce84]247
[37fe352]248                        assertf( !PreludeDirector.empty(), "Can't find prelude without option --prelude-dir must be used." );
[4dcaed2]249
[807ce84]250                        // Read to gcc builtins, if not generating the cfa library
[37fe352]251                        FILE * gcc_builtins = fopen( (PreludeDirector + "/gcc-builtins.cf").c_str(), "r" );
[6ce3ae9]252                        assertf( gcc_builtins, "cannot open gcc-builtins.cf\n" );
253                        parse( gcc_builtins, LinkageSpec::Compiler );
[81419b5]254
[159c62e]255                        // read the extra prelude in, if not generating the cfa library
[37fe352]256                        FILE * extras = fopen( (PreludeDirector + "/extras.cf").c_str(), "r" );
[3b8e52c]257                        assertf( extras, "cannot open extras.cf\n" );
[f0994a1]258                        parse( extras, LinkageSpec::BuiltinC );
[159c62e]259
[81419b5]260                        if ( ! libcfap ) {
[faf8857]261                                // read the prelude in, if not generating the cfa library
[e523b07]262                                FILE * prelude = fopen( (PreludeDirector + "/prelude.cfa").c_str(), "r" );
263                                assertf( prelude, "cannot open prelude.cfa\n" );
[35304009]264                                parse( prelude, LinkageSpec::Intrinsic );
[fa4805f]265
266                                // Read to cfa builtins, if not generating the cfa library
[37fe352]267                                FILE * builtins = fopen( (PreludeDirector + "/builtins.cf").c_str(), "r" );
[fa4805f]268                                assertf( builtins, "cannot open builtins.cf\n" );
[54d714e]269                                parse( builtins, LinkageSpec::BuiltinCFA );
[b87a5ed]270                        } // if
271                } // if
[81419b5]272
[926af74]273                parse( input, libcfap ? LinkageSpec::Intrinsic : LinkageSpec::Cforall, yydebug );
[71f4e4f]274
[b87a5ed]275                if ( parsep ) {
[e6955b1]276                        parseTree->printList( cout );
[0da3e2c]277                        delete parseTree;
[3e96559]278                        return EXIT_SUCCESS;
[b87a5ed]279                } // if
280
[0da3e2c]281                buildList( parseTree, translationUnit );
282                delete parseTree;
[cbaee0d]283                parseTree = nullptr;
[b87a5ed]284
285                if ( astp ) {
[1ab4ce2]286                        dump( translationUnit );
[3e96559]287                        return EXIT_SUCCESS;
[b87a5ed]288                } // if
289
[036dd5f]290                // Temporary: fill locations after parsing so that every node has a location, for early error messages.
291                // Eventually we should pass the locations from the parser to every node, but this quick and dirty solution
292                // works okay for now.
293                CodeTools::fillLocations( translationUnit );
[3c0d4cd]294                Stats::Time::StopBlock();
[036dd5f]295
[839ccbb]296                // add the assignment statement after the initialization of a type parameter
[675716e]297                PASS( "Validate", SymTab::validate( translationUnit, symtabp ) );
[81419b5]298                if ( symtabp ) {
[46f6134]299                        deleteAll( translationUnit );
[3e96559]300                        return EXIT_SUCCESS;
[b87a5ed]301                } // if
302
[81419b5]303                if ( expraltp ) {
[bff09c8]304                        PassVisitor<ResolvExpr::AlternativePrinter> printer( cout );
[81419b5]305                        acceptAll( translationUnit, printer );
[3e96559]306                        return EXIT_SUCCESS;
[b87a5ed]307                } // if
308
309                if ( validp ) {
[1ab4ce2]310                        dump( translationUnit );
[3e96559]311                        return EXIT_SUCCESS;
[b87a5ed]312                } // if
313
[046a890]314                PASS( "Translate Throws", ControlStruct::translateThrows( translationUnit ) );
[675716e]315                PASS( "Fix Labels", ControlStruct::fixLabels( translationUnit ) );
316                PASS( "Fix Names", CodeGen::fixNames( translationUnit ) );
317                PASS( "Gen Init", InitTweak::genInit( translationUnit ) );
318                PASS( "Expand Member Tuples" , Tuples::expandMemberTuples( translationUnit ) );
[81419b5]319                if ( libcfap ) {
320                        // generate the bodies of cfa library functions
321                        LibCfa::makeLibCfa( translationUnit );
[b87a5ed]322                } // if
323
[fa2de95]324                if ( declstatsp ) {
325                        CodeTools::printDeclStats( translationUnit );
326                        deleteAll( translationUnit );
[3e96559]327                        return EXIT_SUCCESS;
328                } // if
[fa2de95]329
[de62360d]330                if ( bresolvep ) {
[1ab4ce2]331                        dump( translationUnit );
[3e96559]332                        return EXIT_SUCCESS;
[de62360d]333                } // if
334
[76b378d]335                CodeTools::fillLocations( translationUnit );
336
[3b3491b]337                if ( resolvprotop ) {
338                        CodeTools::dumpAsResolvProto( translationUnit );
[3e96559]339                        return EXIT_SUCCESS;
340                } // if
[3b3491b]341
[a77257be]342                if( useNewAST) {
[9ea38de]343                        auto transUnit = convert( move( translationUnit ) );
344                        PASS( "Resolve", ResolvExpr::resolve( transUnit ) );
[490fb92e]345                        if ( exprp ) {
346                                translationUnit = convert( move( transUnit ) );
347                                dump( translationUnit );
348                                return EXIT_SUCCESS;
349                        } // if
350
351                        PASS( "Fix Init", InitTweak::fix(transUnit, buildingLibrary()));
[9ea38de]352                        translationUnit = convert( move( transUnit ) );
[a77257be]353                } else {
354                        PASS( "Resolve", ResolvExpr::resolve( translationUnit ) );
[490fb92e]355                        if ( exprp ) {
356                                dump( translationUnit );
357                                return EXIT_SUCCESS;
358                        }
[4615ac8]359
[490fb92e]360                        PASS( "Fix Init", InitTweak::fix( translationUnit, buildingLibrary() ) );
361                }
[81419b5]362
[71f4e4f]363                // fix ObjectDecl - replaces ConstructorInit nodes
[ca1c11f]364                if ( ctorinitp ) {
365                        dump ( translationUnit );
[3e96559]366                        return EXIT_SUCCESS;
[926af74]367                } // if
[71f4e4f]368
[675716e]369                PASS( "Expand Unique Expr", Tuples::expandUniqueExpr( translationUnit ) ); // xxx - is this the right place for this? want to expand ASAP so tha, sequent passes don't need to worry about double-visiting a unique expr - needs to go after InitTweak::fix so that copy constructed return declarations are reused
[626dbc10]370
[046a890]371                PASS( "Translate Tries" , ControlStruct::translateTries( translationUnit ) );
[6edd210]372
[675716e]373                PASS( "Gen Waitfor" , Concurrency::generateWaitFor( translationUnit ) );
[307a732]374
[675716e]375                PASS( "Convert Specializations",  GenPoly::convertSpecializations( translationUnit ) ); // needs to happen before tuple types are expanded
[9f5ecf5]376
[675716e]377                PASS( "Expand Tuples", Tuples::expandTuples( translationUnit ) ); // xxx - is this the right place for this?
[626dbc10]378
379                if ( tuplep ) {
380                        dump( translationUnit );
[3e96559]381                        return EXIT_SUCCESS;
382                } // if
[141b786]383
[675716e]384                PASS( "Virtual Expand Casts", Virtual::expandCasts( translationUnit ) ); // Must come after translateEHM
[a5f0529]385
[675716e]386                PASS( "Instantiate Generics", GenPoly::instantiateGeneric( translationUnit ) );
[53d3ab4b]387                if ( genericsp ) {
388                        dump( translationUnit );
[3e96559]389                        return EXIT_SUCCESS;
390                } // if
[b4f8808]391
[675716e]392                PASS( "Convert L-Value", GenPoly::convertLvalue( translationUnit ) );
[53d3ab4b]393
[fea7ca7]394                if ( bboxp ) {
395                        dump( translationUnit );
[3e96559]396                        return EXIT_SUCCESS;
[926af74]397                } // if
[675716e]398                PASS( "Box", GenPoly::box( translationUnit ) );
[81419b5]399
[8905f56]400                if ( bcodegenp ) {
401                        dump( translationUnit );
[3e96559]402                        return EXIT_SUCCESS;
403                } // if
[8905f56]404
[13de47bc]405                if ( optind < argc ) {                                                  // any commands after the flags and input file ? => output file name
406                        output = new ofstream( argv[ optind ] );
407                } // if
[0270824]408
[7b15d7a]409                CodeTools::fillLocations( translationUnit );
[62ce290]410                PASS( "Code Gen", CodeGen::generate( translationUnit, *output, ! genproto, prettycodegenp, true, linemarks ) );
[0270824]411
[37fe352]412                CodeGen::FixMain::fix( *output, (PreludeDirector + "/bootloader.c").c_str() );
[e6955b1]413                if ( output != &cout ) {
[b87a5ed]414                        delete output;
415                } // if
[a16764a6]416        } catch ( SemanticErrorException &e ) {
[b87a5ed]417                if ( errorp ) {
[e6955b1]418                        cerr << "---AST at error:---" << endl;
419                        dump( translationUnit, cerr );
420                        cerr << endl << "---End of AST, begin error message:---\n" << endl;
[926af74]421                } // if
[d55d7a6]422                e.print();
[e6955b1]423                if ( output != &cout ) {
[b87a5ed]424                        delete output;
425                } // if
[3e96559]426                return EXIT_FAILURE;
[b87a5ed]427        } catch ( UnimplementedError &e ) {
[e6955b1]428                cout << "Sorry, " << e.get_what() << " is not currently implemented" << endl;
429                if ( output != &cout ) {
[b87a5ed]430                        delete output;
431                } // if
[3e96559]432                return EXIT_FAILURE;
[b87a5ed]433        } catch ( CompilerError &e ) {
[e6955b1]434                cerr << "Compiler Error: " << e.get_what() << endl;
[c850687]435                cerr << "(please report bugs to [REDACTED])" << endl;
[e6955b1]436                if ( output != &cout ) {
[b87a5ed]437                        delete output;
438                } // if
[3e96559]439                return EXIT_FAILURE;
440        } catch ( ... ) {
[bffcd66]441                exception_ptr eptr = current_exception();
[4990812]442                try {
443                        if (eptr) {
[bffcd66]444                                rethrow_exception(eptr);
[3e96559]445                        } else {
[bffcd66]446                                cerr << "Exception Uncaught and Unknown" << endl;
[3e96559]447                        } // if
[bffcd66]448                } catch(const exception& e) {
449                        cerr << "Uncaught Exception \"" << e.what() << "\"\n";
[3e96559]450                } // try
451                return EXIT_FAILURE;
452        } // try
[b87a5ed]453
[39786813]454        deleteAll( translationUnit );
[1cb7fab2]455        Stats::print();
[3e96559]456        return EXIT_SUCCESS;
[d9a0e76]457} // main
[51b7345]458
[0da3e2c]459
[3e9de01]460static const char optstring[] = ":c:ghlLmNnpdOAP:S:twW:D:";
[3e96559]461
[62ce290]462enum { PreludeDir = 128 };
[3e96559]463static struct option long_opts[] = {
[1a69a90]464        { "colors", required_argument, nullptr, 'c' },
465        { "gdb", no_argument, nullptr, 'g' },
[3e96559]466        { "help", no_argument, nullptr, 'h' },
467        { "libcfa", no_argument, nullptr, 'l' },
[62ce290]468        { "linemarks", no_argument, nullptr, 'L' },
[3e96559]469        { "no-main", no_argument, 0, 'm' },
[62ce290]470        { "no-linemarks", no_argument, nullptr, 'N' },
471        { "no-prelude", no_argument, nullptr, 'n' },
[3e96559]472        { "prototypes", no_argument, nullptr, 'p' },
[7215000]473        { "deterministic-out", no_argument, nullptr, 'd' },
[a77257be]474        { "old-ast", no_argument, nullptr, 'O'},
475        { "new-ast", no_argument, nullptr, 'A'},
[62ce290]476        { "print", required_argument, nullptr, 'P' },
477        { "prelude-dir", required_argument, nullptr, PreludeDir },
478        { "statistics", required_argument, nullptr, 'S' },
[3e96559]479        { "tree", no_argument, nullptr, 't' },
480        { "", no_argument, nullptr, 0 },                                        // -w
481        { "", no_argument, nullptr, 0 },                                        // -W
482        { "", no_argument, nullptr, 0 },                                        // -D
483        { nullptr, 0, nullptr, 0 }
484}; // long_opts
485
486static const char * description[] = {
[3e9de01]487        "diagnostic color: never, always, or auto.",            // -c
488        "wait for gdb to attach",                                                       // -g
489        "print help message",                                                           // -h
490        "generate libcfa.c",                                                            // -l
491        "generate line marks",                                                          // -L
492        "do not replace main",                                                          // -m
493        "do not generate line marks",                                           // -N
494        "do not read prelude",                                                          // -n
[62ce290]495        "generate prototypes for prelude functions",            // -p
[3e9de01]496        "only print deterministic output",                  // -d
497        "Use the old-ast",                                                                      // -O
498        "Use the new-ast",                                                                      // -A
499        "print",                                                                                        // -P
[62ce290]500        "<directory> prelude directory for debug/nodebug",      // no flag
501        "<option-list> enable profiling information:\n          counters,heap,time,all,none", // -S
[3e9de01]502        "building cfa standard lib",                                            // -t
503        "",                                                                                                     // -w
504        "",                                                                                                     // -W
505        "",                                                                                                     // -D
[3e96559]506}; // description
507
[0c0f548]508static_assert( sizeof( long_opts ) / sizeof( long_opts[0] ) - 1 == sizeof( description ) / sizeof( description[0] ), "Long opts and description must match" );
[62ce290]509
510static struct Printopts {
511        const char * name;
512        int & flag;
513        int val;
514        const char * descript;
515} printopts[] = {
[0e464f6]516        { "ascodegen", codegenp, true, "print AST as codegen rather than AST" },
517        { "asterr", errorp, true, "print AST on error" },
[62ce290]518        { "declstats", declstatsp, true, "code property statistics" },
519        { "parse", yydebug, true, "yacc (parsing) debug information" },
520        { "pretty", prettycodegenp, true, "prettyprint for ascodegen flag" },
521        { "rproto", resolvprotop, true, "resolver-proto instance" },
[0e464f6]522        { "rsteps", resolvep, true, "print resolver steps" },
523        { "tree", parsep, true, "print parse tree" },
524        // code dumps
525        { "ast", astp, true, "print AST after parsing" },
526        { "symevt", symtabp, true, "print AST after symbol table events" },
527        { "altexpr", expraltp, true, "print alternatives for expressions" },
528        { "astdecl", validp, true, "print AST after declaration validation pass" },
529        { "resolver", bresolvep, true, "print AST before resolver step" },
530        { "astexpr", exprp, true, "print AST after expression analysis" },
531        { "ctordtor", ctorinitp, true, "print AST after ctor/dtor are replaced" },
532        { "tuple", tuplep, true, "print AST after tuple expansion" },
533        { "astgen", genericsp, true, "print AST after instantiate generics" },
534        { "box", bboxp, true, "print AST before box step" },
535        { "codegen", bcodegenp, true, "print AST before code generation" },
[62ce290]536};
537enum { printoptsSize = sizeof( printopts ) / sizeof( printopts[0] ) };
538
[3e96559]539static void usage( char *argv[] ) {
[e0bd0f9]540    cout << "Usage: " << argv[0] << " [options] [input-file (default stdin)] [output-file (default stdout)], where options are:" << endl;
[3e96559]541        int i = 0, j = 1;                                                                       // j skips starting colon
542        for ( ; long_opts[i].name != 0 && optstring[j] != '\0'; i += 1, j += 1 ) {
543                if ( long_opts[i].name[0] != '\0' ) {                   // hidden option, internal usage only
[62ce290]544                        if ( strcmp( long_opts[i].name, "prelude-dir" ) != 0 ) { // flag
545                                cout << "  -" << optstring[j] << ",";
546                        } else {                                                                        // no flag
547                                j -= 1;                                                                 // compensate
548                                cout << "     ";
549                        } // if
550                        cout << " --" << left << setw(12) << long_opts[i].name << "  ";
551                        if ( strcmp( long_opts[i].name, "print" ) == 0 ) {
552                                cout << "one of: " << endl;
553                                for ( int i = 0; i < printoptsSize; i += 1 ) {
554                                        cout << setw(10) << " " << left << setw(10) << printopts[i].name << "  " << printopts[i].descript << endl;
555                                } // for
556                        } else {
557                                cout << description[i] << endl;
558                        } // if
[3e96559]559                } // if
[62ce290]560                if ( optstring[j + 1] == ':' ) j += 1;
[3e96559]561        } // for
562        if ( long_opts[i].name != 0 || optstring[j] != '\0' ) assertf( false, "internal error, mismatch of option flags and names\n" );
563    exit( EXIT_FAILURE );
564} // usage
565
[e0bd0f9]566static void parse_cmdline( int argc, char * argv[] ) {
[0da3e2c]567        opterr = 0;                                                                                     // (global) prevent getopt from printing error messages
568
[c5e5109]569        bool Wsuppress = false, Werror = false;
[0da3e2c]570        int c;
[3e96559]571        while ( (c = getopt_long( argc, argv, optstring, long_opts, nullptr )) != -1 ) {
[0da3e2c]572                switch ( c ) {
[1a69a90]573                  case 'c':                                                                             // diagnostic colors
574                        if ( strcmp( optarg, "always" ) == 0 ) {
575                                ErrorHelpers::colors = ErrorHelpers::Colors::Always;
576                        } else if ( strcmp( optarg, "never" ) == 0 ) {
577                                ErrorHelpers::colors = ErrorHelpers::Colors::Never;
578                        } else if ( strcmp( optarg, "auto" ) == 0 ) {
579                                ErrorHelpers::colors = ErrorHelpers::Colors::Auto;
580                        } // if
581                        break;
[3e96559]582                  case 'h':                                                                             // help message
583                        usage( argv );                                                          // no return
[53d3ab4b]584                        break;
[3e96559]585                  case 'l':                                                                             // generate libcfa.c
[0da3e2c]586                        libcfap = true;
587                        break;
[62ce290]588                  case 'L':                                                                             // generate line marks
[6de43b6]589                        linemarks = true;
[c850687]590                        break;
[3e96559]591                  case 'm':                                                                             // do not replace main
592                        nomainp = true;
[0da3e2c]593                        break;
[62ce290]594                  case 'N':                                                                             // do not generate line marks
[6de43b6]595                        linemarks = false;
[c59bde6]596                        break;
[62ce290]597                  case 'n':                                                                             // do not read prelude
[3e96559]598                        nopreludep = true;
[0da3e2c]599                        break;
[62ce290]600                  case 'p':                                                                             // generate prototypes for prelude functions
601                        genproto = true;
[0da3e2c]602                        break;
[7215000]603                  case 'd':                                     // don't print non-deterministic output
[a77257be]604                        deterministic_output = true;
605                        break;
606                  case 'O':                                     // don't print non-deterministic output
607                        useNewAST = false;
608                        break;
609                  case 'A':                                     // don't print non-deterministic output
610                        useNewAST = true;
[7215000]611                        break;
[62ce290]612                  case 'P':                                                                             // print options
613                        for ( int i = 0;; i += 1 ) {
614                                if ( i == printoptsSize ) {
615                                        cout << "Unknown --print option " << optarg << endl;
616                                        goto Default;
617                                } // if
618                                if ( strcmp( optarg, printopts[i].name ) == 0 ) {
619                                        printopts[i].flag = printopts[i].val;
620                                        break;
621                                } // if
622                        } // for
[0da3e2c]623                        break;
[62ce290]624                  case PreludeDir:                                                              // prelude directory for debug/nodebug, hidden
625                        PreludeDirector = optarg;
[3b3491b]626                        break;
[3e96559]627                  case 'S':                                                                             // enable profiling information, argument comma separated list of names
628                        Stats::parse_params( optarg );
[ebcc940]629                        break;
[dee1f89]630                  case 't':                                                                             // building cfa stdlib
[0da3e2c]631                        treep = true;
632                        break;
[dee1f89]633                  case 'g':                                                                             // wait for gdb
634                        waiting_for_gdb = true;
635                        break;
[3e96559]636                  case 'w':                                                                             // suppress all warnings, hidden
[c5e5109]637                        Wsuppress = true;
[44bca7f]638                        break;
[3e96559]639                  case 'W':                                                                             // coordinate gcc -W with CFA, hidden
[44bca7f]640                        if ( strcmp( optarg, "all" ) == 0 ) {
[68e9ace]641                                SemanticWarning_EnableAll();
[44bca7f]642                        } else if ( strcmp( optarg, "error" ) == 0 ) {
643                                Werror = true;
644                        } else {
645                                char * warning = optarg;
646                                Severity s;
647                                if ( strncmp( optarg, "no-", 3 ) == 0 ) {
648                                        warning += 3;
649                                        s = Severity::Suppress;
650                                } else {
651                                        s = Severity::Warn;
652                                } // if
[68e9ace]653                                SemanticWarning_Set( warning, s );
[44bca7f]654                        } // if
655                        break;
[3e96559]656                  case 'D':                                                                             // ignore -Dxxx, forwarded by cpp, hidden
[0da3e2c]657                        break;
[3e96559]658                  case '?':                                                                             // unknown option
659                        if ( optopt ) {                                                         // short option ?
660                                cout << "Unknown option -" << (char)optopt << endl;
661                        } else {
662                                cout << "Unknown option " << argv[optind - 1] << endl;
663                        } // if
664                        goto Default;
665                  case ':':                                                                             // missing option
[ae47a23]666                        if ( optopt ) {                                                         // short option ?
[3e96559]667                                cout << "Missing option for -" << (char)optopt << endl;
[ae47a23]668                        } else {
[3e96559]669                                cout << "Missing option for " << argv[optind - 1] << endl;
[ae47a23]670                        } // if
[3e96559]671                        goto Default;
672                  Default:
673                  default:
674                        usage( argv );                                                          // no return
[0da3e2c]675                } // switch
676        } // while
[44bca7f]677
678        if ( Werror ) {
[68e9ace]679                SemanticWarning_WarningAsError();
[44bca7f]680        } // if
[c5e5109]681        if ( Wsuppress ) {
682                SemanticWarning_SuppressAll();
683        } // if
[44bca7f]684        // for ( const auto w : WarningFormats ) {
685        //      cout << w.name << ' ' << (int)w.severity << endl;
686        // } // for
[0da3e2c]687} // parse_cmdline
688
[8b7ee09]689static void parse( FILE * input, LinkageSpec::Spec linkage, bool shouldExit ) {
[0da3e2c]690        extern int yyparse( void );
[cbaee0d]691        extern FILE * yyin;
[0da3e2c]692        extern int yylineno;
693
[8b7ee09]694        ::linkage = linkage;                                                            // set globals
[0da3e2c]695        yyin = input;
696        yylineno = 1;
697        int parseStatus = yyparse();
[81419b5]698
699        fclose( input );
[0da3e2c]700        if ( shouldExit || parseStatus != 0 ) {
701                exit( parseStatus );
[81419b5]702        } // if
[0da3e2c]703} // parse
[81419b5]704
[1ab4ce2]705static bool notPrelude( Declaration * decl ) {
706        return ! LinkageSpec::isBuiltin( decl->get_linkage() );
[0da3e2c]707} // notPrelude
[1ab4ce2]708
[e6955b1]709static void dump( list< Declaration * > & translationUnit, ostream & out ) {
710        list< Declaration * > decls;
[926af74]711
[62ce290]712        if ( genproto ) {
[e6955b1]713                filter( translationUnit.begin(), translationUnit.end(), back_inserter( decls ), notPrelude );
[1ab4ce2]714        } else {
715                decls = translationUnit;
[926af74]716        } // if
[1ab4ce2]717
[e39241b]718        // depending on commandline options, either generate code or dump the AST
719        if ( codegenp ) {
[62ce290]720                CodeGen::generate( decls, out, ! genproto, prettycodegenp );
[e39241b]721        } else {
722                printAll( decls, out );
[3e96559]723        } // if
[7f5566b]724        deleteAll( translationUnit );
[0da3e2c]725} // dump
[1ab4ce2]726
[51b7345]727// Local Variables: //
[b87a5ed]728// tab-width: 4 //
729// mode: c++ //
730// compile-command: "make install" //
[51b7345]731// End:  //
Note: See TracBrowser for help on using the repository browser.