Changeset 937e51d for src/ControlStruct/MLEMutator.cc

Timestamp:

Jun 26, 2015, 4:00:26 PM (10 years ago)

Author:

Aaron Moss <a3moss@…>

Branches:

ADT, aaron-thesis, arm-eh, ast-experimental, cleanup-dtors, ctor, deferred_resn, demangler, enum, forall-pointer-decay, gc_noraii, jacob/cs343-translation, jenkins-sandbox, master, memory, new-ast, new-ast-unique-expr, new-env, no_list, persistent-indexer, pthread-emulation, qualifiedEnum, resolv-new, string, with_gc

Children:

0df292b, e0ff3e6

Parents:

eb50842 (diff), 1869adf (diff)
Note: this is a merge changeset, the changes displayed below correspond to the merge itself.
Use the (diff) links above to see all the changes relative to each parent.

Message:

Merge pointer to pointer to qualified fix into master

File:

• src/ControlStruct/MLEMutator.cc (modified) (6 diffs)

src/ControlStruct/MLEMutator.cc

@@ -9 +9 @@
 // Author           : Rodolfo G. Esteves
 // Created On       : Mon May 18 07:44:20 2015
-// Last Modified By : Peter A. Buhr
-// Last Modified On : Tue May 19 15:32:26 2015
-// Update Count     : 2
+// Last Modified By : Rob Schluntz
+// Last Modified On : Thu Jun 04 15:12:33 2015
+// Update Count     : 173
 //
 
@@ -19 +19 @@
 #include "MLEMutator.h"
 #include "SynTree/Statement.h"
+#include "SynTree/Expression.h"
 
 namespace ControlStruct {
@@ -26 +27 @@
         }
 
-        CompoundStmt* MLEMutator::mutate( CompoundStmt *cmpndStmt ) {
-                bool labeledBlock = false;
-                if ( !((cmpndStmt->get_labels()).empty()) ) {
-                        labeledBlock = true;
-                        enclosingBlocks.push_back( Entry( cmpndStmt ) );
-                } // if
-
-                std::list< Statement * > &kids = cmpndStmt->get_kids();
+        // break labels have to come after the statement they break out of, 
+        // so mutate a statement, then if they inform us through the breakLabel field
+        // tha they need a place to jump to on a break statement, add the break label 
+        // to the body of statements
+        void MLEMutator::fixBlock( std::list< Statement * > &kids ) {
                 for ( std::list< Statement * >::iterator k = kids.begin(); k != kids.end(); k++ ) {
                         *k = (*k)->acceptMutator(*this);
 
                         if ( ! get_breakLabel().empty() ) {
-                                std::list< Statement * >::iterator next = k; next++;
+                                std::list< Statement * >::iterator next = k+1;
                                 if ( next == kids.end() ) {
                                         std::list<Label> ls; ls.push_back( get_breakLabel() );
                                         kids.push_back( new NullStmt( ls ) );
-                                } else
+                                } else {
                                         (*next)->get_labels().push_back( get_breakLabel() );
+                                }
 
                                 set_breakLabel("");
                         } // if
                 } // for
+        }
+
+        CompoundStmt* MLEMutator::mutate( CompoundStmt *cmpndStmt ) {
+                bool labeledBlock = !(cmpndStmt->get_labels().empty());
+                if ( labeledBlock ) {
+                        Label brkLabel = generator->newLabel("blockBreak");
+                        enclosingBlocks.push_back( Entry( cmpndStmt, brkLabel ) );
+                } // if
+
+                // a child statement may set the break label
+                // - if they do, attach it to the next statement
+                std::list< Statement * > &kids = cmpndStmt->get_kids();
+                fixBlock( kids );
 
                 if ( labeledBlock ) {
                         assert( ! enclosingBlocks.empty() );
-                        if ( ! enclosingBlocks.back().get_breakExit().empty() )
-                                set_breakLabel( enclosingBlocks.back().get_breakExit() );
+                        if ( ! enclosingBlocks.back().useBreakExit().empty() ) {
+                                set_breakLabel( enclosingBlocks.back().useBreakExit() );
+                        }
                         enclosingBlocks.pop_back();
                 } // if
 
-                //mutateAll( cmpndStmt->get_kids(), *this );
                 return cmpndStmt;
         }
 
-        Statement *MLEMutator::mutate( WhileStmt *whileStmt ) {
-                enclosingLoops.push_back( Entry( whileStmt ) );
-                whileStmt->set_body ( whileStmt->get_body()->acceptMutator( *this ) );
-
+        template< typename LoopClass >
+        Statement *MLEMutator::handleLoopStmt( LoopClass *loopStmt ) {
+                // remember this as the most recent enclosing loop, then mutate 
+                // the body of the loop -- this will determine whether brkLabel
+                // and contLabel are used with branch statements
+                // and will recursively do the same to nested loops
+                Label brkLabel = generator->newLabel("loopBreak");
+                Label contLabel = generator->newLabel("loopContinue");
+                enclosingLoops.push_back( Entry( loopStmt, brkLabel, contLabel ) );
+                loopStmt->set_body ( loopStmt->get_body()->acceptMutator( *this ) );
+
+                // sanity check that the enclosing loops have been popped correctly
                 Entry &e = enclosingLoops.back();
-                assert ( e == whileStmt );
-                whileStmt->set_body( mutateLoop( whileStmt->get_body(), e ) );
+                assert ( e == loopStmt );
+
+                // this will take the necessary steps to add definitions of the previous
+                // two labels, if they are used.
+                loopStmt->set_body( mutateLoop( loopStmt->get_body(), e ) );
                 enclosingLoops.pop_back();
 
-                return whileStmt;
-        }
-
-        Statement *MLEMutator::mutate( ForStmt *forStmt ) {
-                enclosingLoops.push_back( Entry( forStmt ) );
-                maybeMutate( forStmt->get_body(), *this );
-
-                Entry &e = enclosingLoops.back();
-                assert ( e == forStmt );
-                forStmt->set_body( mutateLoop( forStmt->get_body(), e ) );
-                enclosingLoops.pop_back();
-
-                return forStmt;
+                return loopStmt;
+        }
+
+        Statement *MLEMutator::mutate( CaseStmt *caseStmt ) {
+                caseStmt->set_condition( maybeMutate( caseStmt->get_condition(), *this ) );
+                fixBlock( caseStmt->get_statements() );
+
+                return caseStmt;
+        }
+
+        template< typename SwitchClass >
+        Statement *MLEMutator::handleSwitchStmt( SwitchClass *switchStmt ) {
+                // generate a label for breaking out of a labeled switch 
+                Label brkLabel = generator->newLabel("switchBreak");
+                enclosingSwitches.push_back( Entry(switchStmt, brkLabel) );
+                mutateAll( switchStmt->get_branches(), *this ); 
+
+                Entry &e = enclosingSwitches.back();
+                assert ( e == switchStmt );
+
+                // only generate break label if labeled break is used
+                if (e.isBreakUsed()) {
+                        // for the purposes of keeping switch statements uniform (i.e. all statements that are 
+                        // direct children of a switch should be CastStmts), append the exit label + break to the 
+                        // last case statement; create a default case if there are no cases
+                        std::list< Statement * > &branches = switchStmt->get_branches();
+                        if ( branches.empty() ) {
+                                branches.push_back( CaseStmt::makeDefault() );
+                        }
+
+                        if ( CaseStmt * c = dynamic_cast< CaseStmt * >( branches.back() ) ) {
+                                std::list<Label> temp; temp.push_back( brkLabel );
+                                c->get_statements().push_back( new BranchStmt( temp, Label(""), BranchStmt::Break ) );
+                        } else assert(0); // as of this point, all branches of a switch are still CaseStmts
+                }
+
+                assert ( enclosingSwitches.back() == switchStmt );
+                enclosingSwitches.pop_back();
+                return switchStmt;
         }
 
         Statement *MLEMutator::mutate( BranchStmt *branchStmt ) throw ( SemanticError ) {
+                std::string originalTarget = branchStmt->get_originalTarget();
+
                 if ( branchStmt->get_type() == BranchStmt::Goto )
                         return branchStmt;
 
                 // test if continue target is a loop
-                if ( branchStmt->get_type() == BranchStmt::Continue && enclosingLoops.empty() )
-                        throw SemanticError( "'continue' outside a loop" );
+                if ( branchStmt->get_type() == BranchStmt::Continue) {
+                        if ( enclosingLoops.empty() ) {
+                                throw SemanticError( "'continue' outside a loop" );
+                        } else if ( std::find( enclosingLoops.begin(), enclosingLoops.end(), (*targetTable)[branchStmt->get_target()] ) == enclosingLoops.end() ) {
+                                throw SemanticError( "'continue' target label must be an enclosing loop: " + originalTarget );
+                        }
+                }
 
                 if ( branchStmt->get_type() == BranchStmt::Break && (enclosingLoops.empty() && enclosingSwitches.empty() && enclosingBlocks.empty() ) )
@@ -98 +154 @@
 
                 if ( targetTable->find( branchStmt->get_target() ) == targetTable->end() )
-                        throw SemanticError("The label defined in the exit loop statement does not exist." );  // shouldn't happen (since that's already checked)
+                        throw SemanticError("The label defined in the exit loop statement does not exist: " + originalTarget );  // shouldn't happen (since that's already checked)
 
                 std::list< Entry >::iterator check;
@@ -106 +162 @@
                                 // neither in loop nor in block, checking if in switch/choose
                                 if ( (check = std::find( enclosingSwitches.begin(), enclosingSwitches.end(), (*targetTable)[branchStmt->get_target()] )) == enclosingSwitches.end() )
-                                        throw SemanticError("The target specified in the exit loop statement does not correspond to an enclosing loop.");
-
+                                        throw SemanticError("The target specified in the exit loop statement does not correspond to an enclosing control structure: " + originalTarget );
+
+                // what about exiting innermost block or switch???
                 if ( enclosingLoops.back() == (*check) )
                         return branchStmt;                              // exit the innermost loop (labels unnecessary)
 
-                Label newLabel;
+                // branch error checks, get the appropriate label name and create a goto
+                Label exitLabel;
                 switch ( branchStmt->get_type() ) {
                   case BranchStmt::Break:
-                        if ( check->get_breakExit() != "" )
-                                newLabel = check->get_breakExit();
-                        else {
-                                newLabel = generator->newLabel();
-                                check->set_breakExit( newLabel );
-                        } // if
-                        break;
+                                assert( check->useBreakExit() != "");
+                                exitLabel = check->useBreakExit();
+                                break;
                   case BranchStmt::Continue:
-                        if ( check->get_contExit() != "" )
-                                newLabel = check->get_contExit();
-                        else {
-                                newLabel = generator->newLabel();
-                                check->set_contExit( newLabel );
-                        } // if
-                        break;
+                                assert( check->useContExit() != "");
+                                exitLabel = check->useContExit();
+                                break;
                   default:
-                        return 0;                                       // shouldn't be here
+                                assert(0);                                      // shouldn't be here
                 } // switch
 
-                return new BranchStmt( std::list<Label>(), newLabel, BranchStmt::Goto );
-        }
-
-
-        Statement *MLEMutator::mutate( SwitchStmt *switchStmt ) {
-                Label brkLabel = generator->newLabel();
-                enclosingSwitches.push_back( Entry(switchStmt, "", brkLabel) );
-                mutateAll( switchStmt->get_branches(), *this ); {
-                        // check if this is necessary (if there is a break to this point, otherwise do not generate
-                        std::list<Label> temp; temp.push_back( brkLabel );
-                        switchStmt->get_branches().push_back( new BranchStmt( temp, Label(""), BranchStmt::Break ) );
-                }
-                assert ( enclosingSwitches.back() == switchStmt );
-                enclosingSwitches.pop_back();
-                return switchStmt;
-        }
-
-        Statement *MLEMutator::mutate( ChooseStmt *switchStmt ) {
-                Label brkLabel = generator->newLabel();
-                enclosingSwitches.push_back( Entry(switchStmt,"", brkLabel) );
-                mutateAll( switchStmt->get_branches(), *this ); {
-                        // check if this is necessary (if there is a break to this point, otherwise do not generate
-                        std::list<Label> temp; temp.push_back( brkLabel );
-                        switchStmt->get_branches().push_back( new BranchStmt( temp, Label(""), BranchStmt::Break ) );
-                }
-                assert ( enclosingSwitches.back() == switchStmt );
-                enclosingSwitches.pop_back();
-                return switchStmt;
+                return new BranchStmt( std::list<Label>(), exitLabel, BranchStmt::Goto );
         }
 
         Statement *MLEMutator::mutateLoop( Statement *bodyLoop, Entry &e ) {
+                // ensure loop body is a block
                 CompoundStmt *newBody;
                 if ( ! (newBody = dynamic_cast<CompoundStmt *>( bodyLoop )) ) {
@@ -170 +194 @@
                 } // if
 
-                Label endLabel = e.get_contExit();
-
-                if ( e.get_breakExit() != "" ) {
-                        if ( endLabel == "" ) endLabel = generator->newLabel();
-                        // check for whether this label is used or not, so as to not generate extraneous gotos
-                        if (e.breakExitUsed)
-                                newBody->get_kids().push_back( new BranchStmt( std::list< Label >(), endLabel, BranchStmt::Goto ) );
-                        // xxx
-                        //std::list< Label > ls; ls.push_back( e.get_breakExit() );
-                        set_breakLabel( e.get_breakExit() );
-                        //newBody->get_kids().push_back( new BranchStmt( ls, "", BranchStmt::Break ) );
-                } // if
-
-                if ( e.get_breakExit() != "" || e.get_contExit() != "" ) {
-                        if (dynamic_cast< NullStmt *>( newBody->get_kids().back() ))
-                                newBody->get_kids().back()->get_labels().push_back( endLabel );
-                        else {
-                                std::list < Label > ls; ls.push_back( endLabel );
-                                newBody->get_kids().push_back( new NullStmt( ls ) );
-                        } // if
-                } // if
+                // only generate these when needed
+
+                if ( e.isContUsed() ) {
+                        // continue label goes in the body as the last statement
+                        std::list< Label > labels; labels.push_back( e.useContExit() );
+                        newBody->get_kids().push_back( new NullStmt( labels ) );                        
+                }
+
+                if ( e.isBreakUsed() ) {
+                        // break label goes after the loop -- it'll get set by the 
+                        // outer mutator if we do this
+                        set_breakLabel( e.useBreakExit() );                     
+                }
 
                 return newBody;
         }
 
-        //*** Entry's methods
-        void MLEMutator::Entry::set_contExit( Label l ) {
-                assert ( contExit == "" || contExit == l );
-                contExit = l;
-        }
-
-        void MLEMutator::Entry::set_breakExit( Label l ) {
-                assert ( breakExit == "" || breakExit == l );
-                breakExit = l;
-        }
+        Statement *MLEMutator::mutate( WhileStmt *whileStmt ) {
+                return handleLoopStmt( whileStmt );
+        }
+
+        Statement *MLEMutator::mutate( ForStmt *forStmt ) {
+                return handleLoopStmt( forStmt );
+        }
+
+        Statement *MLEMutator::mutate( SwitchStmt *switchStmt ) {
+                return handleSwitchStmt( switchStmt );
+        }
+
+        Statement *MLEMutator::mutate( ChooseStmt *switchStmt ) {
+                return handleSwitchStmt( switchStmt );          
+        }
+
 } // namespace ControlStruct