From 0cb8a70adca2ba3cb5f5d05ce17540564cc7a6d4 Mon Sep 17 00:00:00 2001
From: "Michael J. Spencer" <bigcheesegs@gmail.com>
Date: Sat, 10 Sep 2016 01:42:43 +0000
Subject: [PATCH] [ELF] Fix memory leak in BinaryFile handling.

llvm-svn: 281129
---
 lld/ELF/SymbolTable.cpp | 1 +
 lld/ELF/SymbolTable.h   | 1 +
 2 files changed, 2 insertions(+)

diff --git a/lld/ELF/SymbolTable.cpp b/lld/ELF/SymbolTable.cpp
index 74892075a153..baeb99d46946 100644
--- a/lld/ELF/SymbolTable.cpp
+++ b/lld/ELF/SymbolTable.cpp
@@ -55,6 +55,7 @@ void SymbolTable<ELFT>::addFile(std::unique_ptr<InputFile> File) {
 
   // Binary file
   if (auto *F = dyn_cast<BinaryFile>(FileP)) {
+    BinaryFiles.emplace_back(cast<BinaryFile>(File.release()));
     addFile(F->createELF<ELFT>());
     return;
   }
diff --git a/lld/ELF/SymbolTable.h b/lld/ELF/SymbolTable.h
index 8a7bba16eccf..f8d601c4b759 100644
--- a/lld/ELF/SymbolTable.h
+++ b/lld/ELF/SymbolTable.h
@@ -127,6 +127,7 @@ private:
 
   // The symbol table owns all file objects.
   std::vector<std::unique_ptr<ArchiveFile>> ArchiveFiles;
+  std::vector<std::unique_ptr<BinaryFile>> BinaryFiles;
   std::vector<std::unique_ptr<ObjectFile<ELFT>>> ObjectFiles;
   std::vector<std::unique_ptr<LazyObjectFile>> LazyObjectFiles;
   std::vector<std::unique_ptr<SharedFile<ELFT>>> SharedFiles;