From: Aaron Ballman <aaron@aaronballman.com>
Date: Mon, 18 Nov 2013 17:33:32 +0000 (+0000)
Subject: Fixing a possible memory leak from a failing realloc() call.
X-Git-Url: http://plrg.eecs.uci.edu/git/?a=commitdiff_plain;h=eae6e546ec5339179b4c7401416fbf2d641a9e90;p=oota-llvm.git

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

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

diff --git a/lib/Support/SmallPtrSet.cpp b/lib/Support/SmallPtrSet.cpp
index f0fed7792ce..dd417b453ef 100644
--- a/lib/Support/SmallPtrSet.cpp
+++ b/lib/Support/SmallPtrSet.cpp
@@ -202,8 +202,13 @@ void SmallPtrSetImpl::CopyFrom(const SmallPtrSetImpl &RHS) {
   } 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?");
   }