Add a simple RAII object, to be used for pushing a cleanup entry and make the insertion point be the cleanup block.
git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@64048 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/CodeGen/CodeGenFunction.h b/lib/CodeGen/CodeGenFunction.h
index 0686271..1f03760 100644
--- a/lib/CodeGen/CodeGenFunction.h
+++ b/lib/CodeGen/CodeGenFunction.h
@@ -132,6 +132,25 @@
/// and return a BasicBlock where cleanup instructions can be added
llvm::BasicBlock *CreateCleanupBlock();
+ /// CleanupScope - RAII object that will create a cleanup block and
+ /// set the insert point to that block. When destructed, it sets the insert
+ /// point to the previous block.
+ class CleanupScope {
+ CodeGenFunction& CGF;
+ llvm::BasicBlock *CurBB;
+
+ public:
+ CleanupScope(CodeGenFunction &cgf)
+ : CGF(cgf), CurBB(CGF.Builder.GetInsertBlock()) {
+ llvm::BasicBlock *FinallyBB = CGF.CreateCleanupBlock();
+ CGF.Builder.SetInsertPoint(FinallyBB);
+ }
+
+ ~CleanupScope() {
+ CGF.Builder.SetInsertPoint(CurBB);
+ }
+ };
+
private:
/// LabelIDs - Track arbitrary ids assigned to labels for use in
/// implementing the GCC address-of-label extension and indirect