I don't know if you need to handle break/continue specially during the AST parse. Once you do flow analysis and have your basic blocks, it should be pretty easy to place your lexical defers into the right spots, on the graph edges.