source: translator/SymTab/Validate.cc @ 17cd4eb

ADTaaron-thesisarm-ehast-experimentalcleanup-dtorsctordeferred_resndemanglerenumforall-pointer-decaygc_noraiijacob/cs343-translationjenkins-sandboxmemorynew-astnew-ast-unique-exprnew-envno_listpersistent-indexerpthread-emulationqualifiedEnumresolv-newstringwith_gc
Last change on this file since 17cd4eb was 17cd4eb, checked in by Peter A. Buhr <pabuhr@…>, 10 years ago

fixed restrict, fixed parameter copy, introduced name table for types, changed variable after to string

  • Property mode set to 100644
File size: 29.6 KB
Line 
1/*
2  The "validate" phase of translation is used to take a syntax tree and convert it into a standard form that aims to be
3  as regular in structure as possible.  Some assumptions can be made regarding the state of the tree after this pass is
4  complete, including:
5
6  - No nested structure or union definitions; any in the input are "hoisted" to the level of the containing struct or
7    union.
8
9  - All enumeration constants have type EnumInstType.
10
11  - The type "void" never occurs in lists of function parameter or return types; neither do tuple types.  A function
12    taking no arguments has no argument types, and tuples are flattened.
13
14  - No context instances exist; they are all replaced by the set of declarations signified by the context, instantiated
15    by the particular set of type arguments.
16
17  - Every declaration is assigned a unique id.
18
19  - No typedef declarations or instances exist; the actual type is substituted for each instance.
20
21  - Each type, struct, and union definition is followed by an appropriate assignment operator.
22
23  - Each use of a struct or union is connected to a complete definition of that struct or union, even if that definition
24    occurs later in the input.
25*/
26
27#include <list>
28#include <iterator>
29#include "Validate.h"
30#include "SynTree/Visitor.h"
31#include "SynTree/Mutator.h"
32#include "SynTree/Type.h"
33#include "SynTree/Statement.h"
34#include "Indexer.h"
35#include "SynTree/TypeSubstitution.h"
36#include "FixFunction.h"
37#include "ImplementationType.h"
38#include "utility.h"
39#include "UniqueName.h"
40#include "AddVisit.h"
41
42
43#define debugPrint( x ) if ( doDebug ) { std::cout << x; }
44
45namespace SymTab {
46    class HoistStruct : public Visitor {
47      public:
48        static void hoistStruct( std::list< Declaration * > &translationUnit );
49 
50        std::list< Declaration * > &get_declsToAdd() { return declsToAdd; }
51 
52        virtual void visit( StructDecl *aggregateDecl );
53        virtual void visit( UnionDecl *aggregateDecl );
54
55        virtual void visit( CompoundStmt *compoundStmt );
56        virtual void visit( IfStmt *ifStmt );
57        virtual void visit( WhileStmt *whileStmt );
58        virtual void visit( ForStmt *forStmt );
59        virtual void visit( SwitchStmt *switchStmt );
60        virtual void visit( ChooseStmt *chooseStmt );
61        virtual void visit( CaseStmt *caseStmt );
62        virtual void visit( CatchStmt *catchStmt );
63      private:
64        HoistStruct();
65
66        template< typename AggDecl > void handleAggregate( AggDecl *aggregateDecl );
67
68        std::list< Declaration * > declsToAdd;
69        bool inStruct;
70    };
71
72    class Pass1 : public Visitor {
73        typedef Visitor Parent;
74        virtual void visit( EnumDecl *aggregateDecl );
75        virtual void visit( FunctionType *func );
76    };
77 
78    class Pass2 : public Indexer {
79        typedef Indexer Parent;
80      public:
81        Pass2( bool doDebug, const Indexer *indexer );
82      private:
83        virtual void visit( StructInstType *structInst );
84        virtual void visit( UnionInstType *unionInst );
85        virtual void visit( ContextInstType *contextInst );
86        virtual void visit( StructDecl *structDecl );
87        virtual void visit( UnionDecl *unionDecl );
88        virtual void visit( TypeInstType *typeInst );
89
90        const Indexer *indexer;
91 
92        typedef std::map< std::string, std::list< StructInstType * > > ForwardStructsType;
93        typedef std::map< std::string, std::list< UnionInstType * > > ForwardUnionsType;
94        ForwardStructsType forwardStructs;
95        ForwardUnionsType forwardUnions;
96    };
97
98    class Pass3 : public Indexer {
99        typedef Indexer Parent;
100      public:
101        Pass3( const Indexer *indexer );
102      private:
103        virtual void visit( ObjectDecl *object );
104        virtual void visit( FunctionDecl *func );
105
106        const Indexer *indexer;
107    };
108
109    class AddStructAssignment : public Visitor {
110      public:
111        static void addStructAssignment( std::list< Declaration * > &translationUnit );
112
113        std::list< Declaration * > &get_declsToAdd() { return declsToAdd; }
114 
115        virtual void visit( StructDecl *structDecl );
116        virtual void visit( UnionDecl *structDecl );
117        virtual void visit( TypeDecl *typeDecl );
118        virtual void visit( ContextDecl *ctxDecl );
119        virtual void visit( FunctionDecl *functionDecl );
120
121        virtual void visit( FunctionType *ftype );
122        virtual void visit( PointerType *ftype );
123 
124        virtual void visit( CompoundStmt *compoundStmt );
125        virtual void visit( IfStmt *ifStmt );
126        virtual void visit( WhileStmt *whileStmt );
127        virtual void visit( ForStmt *forStmt );
128        virtual void visit( SwitchStmt *switchStmt );
129        virtual void visit( ChooseStmt *chooseStmt );
130        virtual void visit( CaseStmt *caseStmt );
131        virtual void visit( CatchStmt *catchStmt );
132
133        AddStructAssignment() : functionNesting( 0 ) {}
134      private:
135        template< typename StmtClass > void visitStatement( StmtClass *stmt );
136 
137        std::list< Declaration * > declsToAdd;
138        std::set< std::string > structsDone;
139        unsigned int functionNesting;                   // current level of nested functions
140    };
141
142    class EliminateTypedef : public Mutator {
143      public:
144        static void eliminateTypedef( std::list< Declaration * > &translationUnit );
145      private:
146        virtual Declaration *mutate( TypedefDecl *typeDecl );
147        virtual TypeDecl *mutate( TypeDecl *typeDecl );
148        virtual DeclarationWithType *mutate( FunctionDecl *funcDecl );
149        virtual ObjectDecl *mutate( ObjectDecl *objDecl );
150        virtual CompoundStmt *mutate( CompoundStmt *compoundStmt );
151        virtual Type *mutate( TypeInstType *aggregateUseType );
152        virtual Expression *mutate( CastExpr *castExpr );
153 
154        std::map< std::string, TypedefDecl * > typedefNames;
155    };
156
157    void validate( std::list< Declaration * > &translationUnit, bool doDebug, const Indexer *indexer ) {
158        Pass1 pass1;
159        Pass2 pass2( doDebug, indexer );
160        Pass3 pass3( indexer );
161        EliminateTypedef::eliminateTypedef( translationUnit );
162        HoistStruct::hoistStruct( translationUnit );
163        acceptAll( translationUnit, pass1 );
164        acceptAll( translationUnit, pass2 );
165        AddStructAssignment::addStructAssignment( translationUnit );
166        acceptAll( translationUnit, pass3 );
167    }
168
169    void validateType( Type *type, const Indexer *indexer ) {
170        Pass1 pass1;
171        Pass2 pass2( false, indexer );
172        Pass3 pass3( indexer );
173        type->accept( pass1 );
174        type->accept( pass2 );
175        type->accept( pass3 );
176    }
177
178    template< typename Visitor >
179    void acceptAndAdd( std::list< Declaration * > &translationUnit, Visitor &visitor, bool addBefore ) {
180        std::list< Declaration * >::iterator i = translationUnit.begin();
181        while ( i != translationUnit.end() ) {
182            (*i)->accept( visitor );
183            std::list< Declaration * >::iterator next = i;
184            next++;
185            if ( ! visitor.get_declsToAdd().empty() ) {
186                translationUnit.splice( addBefore ? i : next, visitor.get_declsToAdd() );
187            } // if
188            i = next;
189        } // while
190    }
191
192    void HoistStruct::hoistStruct( std::list< Declaration * > &translationUnit ) {
193        HoistStruct hoister;
194        acceptAndAdd( translationUnit, hoister, true );
195    }
196
197    HoistStruct::HoistStruct() :  inStruct( false ) {
198    }
199
200    void filter( std::list< Declaration * > &declList, bool (*pred)( Declaration * ), bool doDelete ) {
201        std::list< Declaration * >::iterator i = declList.begin();
202        while ( i != declList.end() ) {
203            std::list< Declaration * >::iterator next = i;
204            ++next;
205            if ( pred( *i ) ) {
206                if ( doDelete ) {
207                    delete *i;
208                } // if
209                declList.erase( i );
210            } // if
211            i = next;
212        } // while
213    }
214
215    bool isStructOrUnion( Declaration *decl ) {
216        return dynamic_cast< StructDecl * >( decl ) || dynamic_cast< UnionDecl * >( decl );
217    }
218
219    template< typename AggDecl >
220    void HoistStruct::handleAggregate( AggDecl *aggregateDecl ) {
221        if ( inStruct ) {
222            // Add elements in stack order corresponding to nesting structure.
223            declsToAdd.push_front( aggregateDecl );
224            Visitor::visit( aggregateDecl );
225        } else {
226            inStruct = true;
227            Visitor::visit( aggregateDecl );
228            inStruct = false;
229        } // if
230        // Always remove the hoisted aggregate from the inner structure.
231        filter( aggregateDecl->get_members(), isStructOrUnion, false );
232    }
233
234    void HoistStruct::visit( StructDecl *aggregateDecl ) {
235        handleAggregate( aggregateDecl );
236    }
237
238    void HoistStruct::visit( UnionDecl *aggregateDecl ) {
239        handleAggregate( aggregateDecl );
240    }
241
242    void HoistStruct::visit( CompoundStmt *compoundStmt ) {
243        addVisit( compoundStmt, *this );
244    }
245
246    void HoistStruct::visit( IfStmt *ifStmt ) {
247        addVisit( ifStmt, *this );
248    }
249
250    void HoistStruct::visit( WhileStmt *whileStmt ) {
251        addVisit( whileStmt, *this );
252    }
253
254    void HoistStruct::visit( ForStmt *forStmt ) {
255        addVisit( forStmt, *this );
256    }
257
258    void HoistStruct::visit( SwitchStmt *switchStmt ) {
259        addVisit( switchStmt, *this );
260    }
261
262    void HoistStruct::visit( ChooseStmt *switchStmt ) {
263        addVisit( switchStmt, *this );
264    }
265
266    void HoistStruct::visit( CaseStmt *caseStmt ) {
267        addVisit( caseStmt, *this );
268    }
269
270    void HoistStruct::visit( CatchStmt *cathStmt ) {
271        addVisit( cathStmt, *this );
272    }
273
274    void Pass1::visit( EnumDecl *enumDecl ) {
275        // Set the type of each member of the enumeration to be EnumConstant
276 
277        for ( std::list< Declaration * >::iterator i = enumDecl->get_members().begin(); i != enumDecl->get_members().end(); ++i ) {
278            ObjectDecl *obj = dynamic_cast< ObjectDecl * >( *i );
279            assert( obj );
280            obj->set_type( new EnumInstType( Type::Qualifiers( true, false, false, false ), enumDecl->get_name() ) );
281        } // for
282        Parent::visit( enumDecl );
283    }
284
285    namespace {
286        template< typename DWTIterator >
287        void fixFunctionList( DWTIterator begin, DWTIterator end, FunctionType *func ) {
288            // the only case in which "void" is valid is where it is the only one in the list; then
289            // it should be removed entirely
290            // other fix ups are handled by the FixFunction class
291            if ( begin == end ) return;
292            FixFunction fixer;
293            DWTIterator i = begin;
294            *i = (*i )->acceptMutator( fixer );
295            if ( fixer.get_isVoid() ) {
296                DWTIterator j = i;
297                ++i;
298                func->get_parameters().erase( j );
299                if ( i != end ) { 
300                    throw SemanticError( "invalid type void in function type ", func );
301                } // if
302            } else {
303                ++i;
304                for ( ; i != end; ++i ) {
305                    FixFunction fixer;
306                    *i = (*i )->acceptMutator( fixer );
307                    if ( fixer.get_isVoid() ) {
308                        throw SemanticError( "invalid type void in function type ", func );
309                    } // if
310                } // for
311            } // if
312        }
313    }
314
315    void Pass1::visit( FunctionType *func ) {
316        // Fix up parameters and return types
317        fixFunctionList( func->get_parameters().begin(), func->get_parameters().end(), func );
318        fixFunctionList( func->get_returnVals().begin(), func->get_returnVals().end(), func );
319        Visitor::visit( func );
320    }
321
322    Pass2::Pass2( bool doDebug, const Indexer *other_indexer ) : Indexer( doDebug ) {
323        if ( other_indexer ) {
324            indexer = other_indexer;
325        } else {
326            indexer = this;
327        } // if
328    }
329
330    void Pass2::visit( StructInstType *structInst ) {
331        Parent::visit( structInst );
332        StructDecl *st = indexer->lookupStruct( structInst->get_name() );
333        // it's not a semantic error if the struct is not found, just an implicit forward declaration
334        if ( st ) {
335            assert( ! structInst->get_baseStruct() || structInst->get_baseStruct()->get_members().empty() || ! st->get_members().empty() );
336            structInst->set_baseStruct( st );
337        } // if
338        if ( ! st || st->get_members().empty() ) {
339            // use of forward declaration
340            forwardStructs[ structInst->get_name() ].push_back( structInst );
341        } // if
342    }
343
344    void Pass2::visit( UnionInstType *unionInst ) {
345        Parent::visit( unionInst );
346        UnionDecl *un = indexer->lookupUnion( unionInst->get_name() );
347        // it's not a semantic error if the union is not found, just an implicit forward declaration
348        if ( un ) {
349            unionInst->set_baseUnion( un );
350        } // if
351        if ( ! un || un->get_members().empty() ) {
352            // use of forward declaration
353            forwardUnions[ unionInst->get_name() ].push_back( unionInst );
354        } // if
355    }
356
357    void Pass2::visit( ContextInstType *contextInst ) {
358        Parent::visit( contextInst );
359        ContextDecl *ctx = indexer->lookupContext( contextInst->get_name() );
360        if ( ! ctx ) {
361            throw SemanticError( "use of undeclared context " + contextInst->get_name() );
362        } // if
363        for ( std::list< TypeDecl * >::const_iterator i = ctx->get_parameters().begin(); i != ctx->get_parameters().end(); ++i ) {
364            for ( std::list< DeclarationWithType * >::const_iterator assert = (*i )->get_assertions().begin(); assert != (*i )->get_assertions().end(); ++assert ) {
365                if ( ContextInstType *otherCtx = dynamic_cast< ContextInstType * >(*assert ) ) {
366                    cloneAll( otherCtx->get_members(), contextInst->get_members() );
367                } else {
368                    contextInst->get_members().push_back( (*assert )->clone() );
369                } // if
370            } // for
371        } // for
372        applySubstitution( ctx->get_parameters().begin(), ctx->get_parameters().end(), contextInst->get_parameters().begin(), ctx->get_members().begin(), ctx->get_members().end(), back_inserter( contextInst->get_members() ) );
373    }
374
375    void Pass2::visit( StructDecl *structDecl ) {
376        if ( ! structDecl->get_members().empty() ) {
377            ForwardStructsType::iterator fwds = forwardStructs.find( structDecl->get_name() );
378            if ( fwds != forwardStructs.end() ) {
379                for ( std::list< StructInstType * >::iterator inst = fwds->second.begin(); inst != fwds->second.end(); ++inst ) {
380                    (*inst )->set_baseStruct( structDecl );
381                } // for
382                forwardStructs.erase( fwds );
383            } // if
384        } // if
385        Indexer::visit( structDecl );
386    }
387
388    void Pass2::visit( UnionDecl *unionDecl ) {
389        if ( ! unionDecl->get_members().empty() ) {
390            ForwardUnionsType::iterator fwds = forwardUnions.find( unionDecl->get_name() );
391            if ( fwds != forwardUnions.end() ) {
392                for ( std::list< UnionInstType * >::iterator inst = fwds->second.begin(); inst != fwds->second.end(); ++inst ) {
393                    (*inst )->set_baseUnion( unionDecl );
394                } // for
395                forwardUnions.erase( fwds );
396            } // if
397        } // if
398        Indexer::visit( unionDecl );
399    }
400
401    void Pass2::visit( TypeInstType *typeInst ) {
402        if ( NamedTypeDecl *namedTypeDecl = lookupType( typeInst->get_name() ) ) {
403            if ( TypeDecl *typeDecl = dynamic_cast< TypeDecl * >( namedTypeDecl ) ) {
404                typeInst->set_isFtype( typeDecl->get_kind() == TypeDecl::Ftype );
405            } // if
406        } // if
407    }
408
409    Pass3::Pass3( const Indexer *other_indexer ) :  Indexer( false ) {
410        if ( other_indexer ) {
411            indexer = other_indexer;
412        } else {
413            indexer = this;
414        } // if
415    }
416
417    void forallFixer( Type *func ) {
418        // Fix up assertions
419        for ( std::list< TypeDecl * >::iterator type = func->get_forall().begin(); type != func->get_forall().end(); ++type ) {
420            std::list< DeclarationWithType * > toBeDone, nextRound;
421            toBeDone.splice( toBeDone.end(), (*type )->get_assertions() );
422            while ( ! toBeDone.empty() ) {
423                for ( std::list< DeclarationWithType * >::iterator assertion = toBeDone.begin(); assertion != toBeDone.end(); ++assertion ) {
424                    if ( ContextInstType *ctx = dynamic_cast< ContextInstType * >( (*assertion )->get_type() ) ) {
425                        for ( std::list< Declaration * >::const_iterator i = ctx->get_members().begin(); i != ctx->get_members().end(); ++i ) {
426                            DeclarationWithType *dwt = dynamic_cast< DeclarationWithType * >( *i );
427                            assert( dwt );
428                            nextRound.push_back( dwt->clone() );
429                        }
430                        delete ctx;
431                    } else {
432                        FixFunction fixer;
433                        *assertion = (*assertion )->acceptMutator( fixer );
434                        if ( fixer.get_isVoid() ) {
435                            throw SemanticError( "invalid type void in assertion of function ", func );
436                        }
437                        (*type )->get_assertions().push_back( *assertion );
438                    }
439                }
440                toBeDone.clear();
441                toBeDone.splice( toBeDone.end(), nextRound );
442            }
443        }
444    }
445
446    void Pass3::visit( ObjectDecl *object ) {
447        forallFixer( object->get_type() );
448        if ( PointerType *pointer = dynamic_cast< PointerType * >( object->get_type() ) ) {
449            forallFixer( pointer->get_base() );
450        } // if
451        Parent::visit( object );
452        object->fixUniqueId();
453    }
454
455    void Pass3::visit( FunctionDecl *func ) {
456        forallFixer( func->get_type() );
457        Parent::visit( func );
458        func->fixUniqueId();
459    }
460
461    static const std::list< std::string > noLabels;
462
463    void AddStructAssignment::addStructAssignment( std::list< Declaration * > &translationUnit ) {
464        AddStructAssignment visitor;
465        acceptAndAdd( translationUnit, visitor, false );
466    }
467
468    template< typename OutputIterator >
469    void makeScalarAssignment( ObjectDecl *srcParam, ObjectDecl *dstParam, DeclarationWithType *member, OutputIterator out ) {
470        ObjectDecl *obj = dynamic_cast<ObjectDecl *>( member ); // PAB: unnamed bit fields are not copied
471        if ( obj != NULL && obj->get_name() == "" && obj->get_bitfieldWidth() != NULL ) return;
472
473        UntypedExpr *assignExpr = new UntypedExpr( new NameExpr( "?=?" ) );
474 
475        UntypedExpr *derefExpr = new UntypedExpr( new NameExpr( "*?" ) );
476        derefExpr->get_args().push_back( new VariableExpr( dstParam ) );
477 
478        // do something special for unnamed members
479        Expression *dstselect = new AddressExpr( new MemberExpr( member, derefExpr ) );
480        assignExpr->get_args().push_back( dstselect );
481 
482        Expression *srcselect = new MemberExpr( member, new VariableExpr( srcParam ) );
483        assignExpr->get_args().push_back( srcselect );
484 
485        *out++ = new ExprStmt( noLabels, assignExpr );
486    }
487
488    template< typename OutputIterator >
489    void makeArrayAssignment( ObjectDecl *srcParam, ObjectDecl *dstParam, DeclarationWithType *member, ArrayType *array, OutputIterator out ) {
490        static UniqueName indexName( "_index" );
491 
492        // for a flexible array member nothing is done -- user must define own assignment
493        if ( ! array->get_dimension() ) return;
494 
495        ObjectDecl *index = new ObjectDecl( indexName.newName(), Declaration::NoStorageClass, LinkageSpec::C, 0, new BasicType( Type::Qualifiers(), BasicType::SignedInt ), 0 );
496        *out++ = new DeclStmt( noLabels, index );
497 
498        UntypedExpr *init = new UntypedExpr( new NameExpr( "?=?" ) );
499        init->get_args().push_back( new AddressExpr( new VariableExpr( index ) ) );
500        init->get_args().push_back( new NameExpr( "0" ) );
501        Statement *initStmt = new ExprStmt( noLabels, init );
502 
503        UntypedExpr *cond = new UntypedExpr( new NameExpr( "?<?" ) );
504        cond->get_args().push_back( new VariableExpr( index ) );
505        cond->get_args().push_back( array->get_dimension()->clone() );
506 
507        UntypedExpr *inc = new UntypedExpr( new NameExpr( "++?" ) );
508        inc->get_args().push_back( new AddressExpr( new VariableExpr( index ) ) );
509 
510        UntypedExpr *assignExpr = new UntypedExpr( new NameExpr( "?=?" ) );
511 
512        UntypedExpr *derefExpr = new UntypedExpr( new NameExpr( "*?" ) );
513        derefExpr->get_args().push_back( new VariableExpr( dstParam ) );
514 
515        Expression *dstselect = new MemberExpr( member, derefExpr );
516        UntypedExpr *dstIndex = new UntypedExpr( new NameExpr( "?+?" ) );
517        dstIndex->get_args().push_back( dstselect );
518        dstIndex->get_args().push_back( new VariableExpr( index ) );
519        assignExpr->get_args().push_back( dstIndex );
520 
521        Expression *srcselect = new MemberExpr( member, new VariableExpr( srcParam ) );
522        UntypedExpr *srcIndex = new UntypedExpr( new NameExpr( "?[?]" ) );
523        srcIndex->get_args().push_back( srcselect );
524        srcIndex->get_args().push_back( new VariableExpr( index ) );
525        assignExpr->get_args().push_back( srcIndex );
526 
527        *out++ = new ForStmt( noLabels, initStmt, cond, inc, new ExprStmt( noLabels, assignExpr ) );
528    }
529
530    Declaration *makeStructAssignment( StructDecl *aggregateDecl, StructInstType *refType, unsigned int functionNesting ) {
531        FunctionType *assignType = new FunctionType( Type::Qualifiers(), false );
532 
533        ObjectDecl *returnVal = new ObjectDecl( "", Declaration::NoStorageClass, LinkageSpec::Cforall, 0, refType->clone(), 0 );
534        assignType->get_returnVals().push_back( returnVal );
535 
536        ObjectDecl *dstParam = new ObjectDecl( "_dst", Declaration::NoStorageClass, LinkageSpec::Cforall, 0, new PointerType( Type::Qualifiers(), refType->clone() ), 0 );
537        assignType->get_parameters().push_back( dstParam );
538 
539        ObjectDecl *srcParam = new ObjectDecl( "_src", Declaration::NoStorageClass, LinkageSpec::Cforall, 0, refType, 0 );
540        assignType->get_parameters().push_back( srcParam );
541
542        // Routines at global scope marked "static" to prevent multiple definitions is separate translation units
543        // because each unit generates copies of the default routines for each aggregate.
544        FunctionDecl *assignDecl = new FunctionDecl( "?=?", functionNesting > 0 ? Declaration::NoStorageClass : Declaration::Static, LinkageSpec::AutoGen, assignType, new CompoundStmt( noLabels ), true );
545        assignDecl->fixUniqueId();
546 
547        for ( std::list< Declaration * >::const_iterator member = aggregateDecl->get_members().begin(); member != aggregateDecl->get_members().end(); ++member ) {
548            if ( DeclarationWithType *dwt = dynamic_cast< DeclarationWithType * >( *member ) ) {
549                if ( ArrayType *array = dynamic_cast< ArrayType * >( dwt->get_type() ) ) {
550                    makeArrayAssignment( srcParam, dstParam, dwt, array, back_inserter( assignDecl->get_statements()->get_kids() ) );
551                } else {
552                    makeScalarAssignment( srcParam, dstParam, dwt, back_inserter( assignDecl->get_statements()->get_kids() ) );
553                } // if
554            } // if
555        } // for
556        assignDecl->get_statements()->get_kids().push_back( new ReturnStmt( noLabels, new VariableExpr( srcParam ) ) );
557 
558        return assignDecl;
559    }
560
561    Declaration *makeUnionAssignment( UnionDecl *aggregateDecl, UnionInstType *refType, unsigned int functionNesting ) {
562        FunctionType *assignType = new FunctionType( Type::Qualifiers(), false );
563 
564        ObjectDecl *returnVal = new ObjectDecl( "", Declaration::NoStorageClass, LinkageSpec::Cforall, 0, refType->clone(), 0 );
565        assignType->get_returnVals().push_back( returnVal );
566 
567        ObjectDecl *dstParam = new ObjectDecl( "_dst", Declaration::NoStorageClass, LinkageSpec::Cforall, 0, new PointerType( Type::Qualifiers(), refType->clone() ), 0 );
568        assignType->get_parameters().push_back( dstParam );
569 
570        ObjectDecl *srcParam = new ObjectDecl( "_src", Declaration::NoStorageClass, LinkageSpec::Cforall, 0, refType, 0 );
571        assignType->get_parameters().push_back( srcParam );
572 
573        // Routines at global scope marked "static" to prevent multiple definitions is separate translation units
574        // because each unit generates copies of the default routines for each aggregate.
575        FunctionDecl *assignDecl = new FunctionDecl( "?=?",  functionNesting > 0 ? Declaration::NoStorageClass : Declaration::Static, LinkageSpec::AutoGen, assignType, new CompoundStmt( noLabels ), true );
576        assignDecl->fixUniqueId();
577 
578        UntypedExpr *copy = new UntypedExpr( new NameExpr( "__builtin_memcpy" ) );
579        copy->get_args().push_back( new VariableExpr( dstParam ) );
580        copy->get_args().push_back( new AddressExpr( new VariableExpr( srcParam ) ) );
581        copy->get_args().push_back( new SizeofExpr( refType->clone() ) );
582
583        assignDecl->get_statements()->get_kids().push_back( new ExprStmt( noLabels, copy ) );
584        assignDecl->get_statements()->get_kids().push_back( new ReturnStmt( noLabels, new VariableExpr( srcParam ) ) );
585 
586        return assignDecl;
587    }
588
589    void AddStructAssignment::visit( StructDecl *structDecl ) {
590        if ( ! structDecl->get_members().empty() && structsDone.find( structDecl->get_name() ) == structsDone.end() ) {
591            StructInstType *structInst = new StructInstType( Type::Qualifiers(), structDecl->get_name() );
592            structInst->set_baseStruct( structDecl );
593            declsToAdd.push_back( makeStructAssignment( structDecl, structInst, functionNesting ) );
594            structsDone.insert( structDecl->get_name() );
595        } // if
596    }
597
598    void AddStructAssignment::visit( UnionDecl *unionDecl ) {
599        if ( ! unionDecl->get_members().empty() ) {
600            UnionInstType *unionInst = new UnionInstType( Type::Qualifiers(), unionDecl->get_name() );
601            unionInst->set_baseUnion( unionDecl );
602            declsToAdd.push_back( makeUnionAssignment( unionDecl, unionInst, functionNesting ) );
603        } // if
604    }
605
606    void AddStructAssignment::visit( TypeDecl *typeDecl ) {
607        CompoundStmt *stmts = 0;
608        TypeInstType *typeInst = new TypeInstType( Type::Qualifiers(), typeDecl->get_name(), false );
609        typeInst->set_baseType( typeDecl );
610        ObjectDecl *src = new ObjectDecl( "_src", Declaration::NoStorageClass, LinkageSpec::Cforall, 0, typeInst->clone(), 0 );
611        ObjectDecl *dst = new ObjectDecl( "_dst", Declaration::NoStorageClass, LinkageSpec::Cforall, 0, new PointerType( Type::Qualifiers(), typeInst->clone() ), 0 );
612        if ( typeDecl->get_base() ) {
613            stmts = new CompoundStmt( std::list< Label >() );
614            UntypedExpr *assign = new UntypedExpr( new NameExpr( "?=?" ) );
615            assign->get_args().push_back( new CastExpr( new VariableExpr( dst ), new PointerType( Type::Qualifiers(), typeDecl->get_base()->clone() ) ) );
616            assign->get_args().push_back( new CastExpr( new VariableExpr( src ), typeDecl->get_base()->clone() ) );
617            stmts->get_kids().push_back( new ReturnStmt( std::list< Label >(), assign ) );
618        } // if
619        FunctionType *type = new FunctionType( Type::Qualifiers(), false );
620        type->get_returnVals().push_back( new ObjectDecl( "", Declaration::NoStorageClass, LinkageSpec::Cforall, 0, typeInst, 0 ) );
621        type->get_parameters().push_back( dst );
622        type->get_parameters().push_back( src );
623        FunctionDecl *func = new FunctionDecl( "?=?", Declaration::NoStorageClass, LinkageSpec::AutoGen, type, stmts, false );
624        declsToAdd.push_back( func );
625    }
626
627    void addDecls( std::list< Declaration * > &declsToAdd, std::list< Statement * > &statements, std::list< Statement * >::iterator i ) {
628        if ( ! declsToAdd.empty() ) {
629            for ( std::list< Declaration * >::iterator decl = declsToAdd.begin(); decl != declsToAdd.end(); ++decl ) {
630                statements.insert( i, new DeclStmt( noLabels, *decl ) );
631            } // for
632            declsToAdd.clear();
633        } // if
634    }
635
636    void AddStructAssignment::visit( FunctionType *) {
637        // ensure that we don't add assignment ops for types defined as part of the function
638    }
639
640    void AddStructAssignment::visit( PointerType *) {
641        // ensure that we don't add assignment ops for types defined as part of the pointer
642    }
643
644    void AddStructAssignment::visit( ContextDecl *) {
645        // ensure that we don't add assignment ops for types defined as part of the context
646    }
647
648    template< typename StmtClass >
649    inline void AddStructAssignment::visitStatement( StmtClass *stmt ) {
650        std::set< std::string > oldStructs = structsDone;
651        addVisit( stmt, *this );
652        structsDone = oldStructs;
653    }
654
655    void AddStructAssignment::visit( FunctionDecl *functionDecl ) {
656        maybeAccept( functionDecl->get_functionType(), *this );
657        acceptAll( functionDecl->get_oldDecls(), *this );
658        functionNesting += 1;
659        maybeAccept( functionDecl->get_statements(), *this );
660        functionNesting -= 1;
661    }
662
663    void AddStructAssignment::visit( CompoundStmt *compoundStmt ) {
664        visitStatement( compoundStmt );
665    }
666
667    void AddStructAssignment::visit( IfStmt *ifStmt ) {
668        visitStatement( ifStmt );
669    }
670
671    void AddStructAssignment::visit( WhileStmt *whileStmt ) {
672        visitStatement( whileStmt );
673    }
674
675    void AddStructAssignment::visit( ForStmt *forStmt ) {
676        visitStatement( forStmt );
677    }
678
679    void AddStructAssignment::visit( SwitchStmt *switchStmt ) {
680        visitStatement( switchStmt );
681    }
682
683    void AddStructAssignment::visit( ChooseStmt *switchStmt ) {
684        visitStatement( switchStmt );
685    }
686
687    void AddStructAssignment::visit( CaseStmt *caseStmt ) {
688        visitStatement( caseStmt );
689    }
690
691    void AddStructAssignment::visit( CatchStmt *cathStmt ) {
692        visitStatement( cathStmt );
693    }
694
695    bool isTypedef( Declaration *decl ) {
696        return dynamic_cast< TypedefDecl * >( decl );
697    }
698
699    void EliminateTypedef::eliminateTypedef( std::list< Declaration * > &translationUnit ) {
700        EliminateTypedef eliminator;
701        mutateAll( translationUnit, eliminator );
702        filter( translationUnit, isTypedef, true );
703    }
704
705    Type *EliminateTypedef::mutate( TypeInstType *typeInst ) {
706        std::map< std::string, TypedefDecl * >::const_iterator def = typedefNames.find( typeInst->get_name() );
707        if ( def != typedefNames.end() ) {
708            Type *ret = def->second->get_base()->clone();
709            ret->get_qualifiers() += typeInst->get_qualifiers();
710            delete typeInst;
711            return ret;
712        } // if
713        return typeInst;
714    }
715
716    Declaration *EliminateTypedef::mutate( TypedefDecl *tyDecl ) {
717        Declaration *ret = Mutator::mutate( tyDecl );
718        typedefNames[ tyDecl->get_name() ] = tyDecl;
719        // When a typedef is a forward declaration:
720        //    typedef struct screen SCREEN;
721        // the declaration portion must be retained:
722        //    struct screen;
723        // because the expansion of the typedef is:
724        //    void rtn( SCREEN *p ) => void rtn( struct screen *p )
725        // hence the type-name "screen" must be defined.
726        // Note, qualifiers on the typedef are superfluous for the forward declaration.
727        if ( StructInstType *aggDecl = dynamic_cast< StructInstType * >( tyDecl->get_base() ) ) {
728            return new StructDecl( aggDecl->get_name() );
729        } else if ( UnionInstType *aggDecl = dynamic_cast< UnionInstType * >( tyDecl->get_base() ) ) {
730            return new UnionDecl( aggDecl->get_name() );
731        } else {
732            return ret;
733        } // if
734    }
735
736    TypeDecl *EliminateTypedef::mutate( TypeDecl *typeDecl ) {
737        std::map< std::string, TypedefDecl * >::iterator i = typedefNames.find( typeDecl->get_name() );
738        if ( i != typedefNames.end() ) {
739            typedefNames.erase( i ) ;
740        } // if
741        return typeDecl;
742    }
743
744    DeclarationWithType *EliminateTypedef::mutate( FunctionDecl *funcDecl ) {
745        std::map< std::string, TypedefDecl * > oldNames = typedefNames;
746        DeclarationWithType *ret = Mutator::mutate( funcDecl );
747        typedefNames = oldNames;
748        return ret;
749    }
750
751    ObjectDecl *EliminateTypedef::mutate( ObjectDecl *objDecl ) {
752        std::map< std::string, TypedefDecl * > oldNames = typedefNames;
753        ObjectDecl *ret = Mutator::mutate( objDecl );
754        typedefNames = oldNames;
755        return ret;
756    }
757
758    Expression *EliminateTypedef::mutate( CastExpr *castExpr ) {
759        std::map< std::string, TypedefDecl * > oldNames = typedefNames;
760        Expression *ret = Mutator::mutate( castExpr );
761        typedefNames = oldNames;
762        return ret;
763    }
764
765    CompoundStmt *EliminateTypedef::mutate( CompoundStmt *compoundStmt ) {
766        std::map< std::string, TypedefDecl * > oldNames = typedefNames;
767        CompoundStmt *ret = Mutator::mutate( compoundStmt );
768        std::list< Statement * >::iterator i = compoundStmt->get_kids().begin();
769        while ( i != compoundStmt->get_kids().end() ) {
770            std::list< Statement * >::iterator next = i;
771            ++next;
772            if ( DeclStmt *declStmt = dynamic_cast< DeclStmt * >( *i ) ) {
773                if ( dynamic_cast< TypedefDecl * >( declStmt->get_decl() ) ) {
774                    delete *i;
775                    compoundStmt->get_kids().erase( i );
776                } // if
777            } // if
778            i = next;
779        } // while
780        typedefNames = oldNames;
781        return ret;
782    }
783} // namespace SymTab
Note: See TracBrowser for help on using the repository browser.