Recovery memory from TypeLocBuilders during crashes.

git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@127932 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 6 insertions, 1 deletions

diff --git a/lib/Sema/TypeLocBuilder.h b/lib/Sema/TypeLocBuilder.h
index 1e0a7c4546..7b241871b6 100644
--- a/lib/Sema/TypeLocBuilder.h
+++ b/lib/Sema/TypeLocBuilder.h
@@ -18,6 +18,7 @@
 #include "clang/AST/TypeLoc.h"
 #include "llvm/ADT/SmallVector.h"
 #include "clang/AST/ASTContext.h"
+#include "llvm/Support/CrashRecoveryContext.h"
 
 namespace clang {
 
@@ -42,9 +43,13 @@ class TypeLocBuilder {
   /// The inline buffer.
   char InlineBuffer[InlineCapacity];
 
+  llvm::CrashRecoveryContextCleanupRegistrar cleanupBuffer;
+
  public:
   TypeLocBuilder()
-    : Buffer(InlineBuffer), Capacity(InlineCapacity), Index(InlineCapacity)
+    : Buffer(InlineBuffer), Capacity(InlineCapacity), Index(InlineCapacity),
+      cleanupBuffer(llvm::CrashRecoveryContextCleanup::create<TypeLocBuilder>(
+            this, llvm::CrashRecoveryContextCleanup::DestructorCleanup))
   {}
 
   ~TypeLocBuilder() {