commit 5de2378f7ea44d3a3d9488d37d63cd12b0b8e8b0
author Aaron Ballman <aaron@aaronballman.com> Mon Nov 18 17:33:32 2013 +0000
committer Aaron Ballman <aaron@aaronballman.com> Mon Nov 18 17:33:32 2013 +0000
tree 90c13da43e2d0858a250ed864741c6edffef61b3
parent a8553497a3497e33195374cfea397e11b385dfdd

Fixing a possible memory leak from a failing realloc() call.

llvm-svn: 195018

llvm/lib/Support/SmallPtrSet.cpp

diff --git a/llvm/lib/Support/SmallPtrSet.cpp b/llvm/lib/Support/SmallPtrSet.cpp
index f0fed77..dd417b4 100644
--- a/llvm/lib/Support/SmallPtrSet.cpp
+++ b/llvm/lib/Support/SmallPtrSet.cpp
@@ -202,8 +202,13 @@
   } else if (CurArraySize != RHS.CurArraySize) {
     if (isSmall())
       CurArray = (const void**)malloc(sizeof(void*) * RHS.CurArraySize);
-    else
-      CurArray = (const void**)realloc(CurArray, sizeof(void*)*RHS.CurArraySize);
+    else {
+      const void **T = (const void**)realloc(CurArray,
+                                             sizeof(void*) * RHS.CurArraySize);
+      if (!T)
+        free(CurArray);
+      CurArray = T;
+    }
     assert(CurArray && "Failed to allocate memory?");
   }