Fix massive resource leaks in the bytecode reader. Reading a bytecode file

with ParseBytecodeFile used to leak both a ModuleProvider (and related
bytecode parser stuff attached to it) AND a file descriptor, which was
never closed.  This prevented gccld/llvm-ld/llvm-link from linking together
apps with more that ~252 .bc files on darwin.

llvm-svn: 30912

lib/Bytecode/Reader/Reader.h

@@ -248,6 +248,12 @@ protected:
   /// @brief Parse a string constants block
   void ParseStringConstants(unsigned NumEntries, ValueTable &Tab);
 
+  /// @brief Release our memory.
+  void freeState() {
+    freeTable(FunctionValues);
+    freeTable(ModuleValues);
+  }
+  
 /// @}
 /// @name Data
 /// @{
@@ -467,12 +473,6 @@ private:
   /// given constant.
   void ResolveReferencesToConstant(Constant *C, unsigned Typ, unsigned Slot);
 
-  /// @brief Release our memory.
-  void freeState() {
-    freeTable(FunctionValues);
-    freeTable(ModuleValues);
-  }
-
   /// @brief Free a table, making sure to free the ValueList in the table.
   void freeTable(ValueTable &Tab) {
     while (!Tab.empty()) {

lib/Bytecode/Reader/ReaderWrappers.cpp

@@ -44,15 +44,17 @@ namespace {
   public:
     BytecodeFileReader(const std::string &Filename, llvm::BytecodeHandler* H=0);
     bool read(std::string* ErrMsg);
+    
+    void freeState() {
+      BytecodeReader::freeState();
+      mapFile.close();
+    }
   };
 }
 
 BytecodeFileReader::BytecodeFileReader(const std::string &Filename,
                                        llvm::BytecodeHandler* H)
-  : BytecodeReader(H)
+  : BytecodeReader(H), fileName(Filename) {
-  , fileName(Filename)
-  , mapFile()
-{
 }
 
 bool BytecodeFileReader::read(std::string* ErrMsg) {
@@ -63,10 +65,7 @@ bool BytecodeFileReader::read(std::string* ErrMsg) {
     return true;
   }
   unsigned char* buffer = reinterpret_cast<unsigned char*>(mapFile.base());
-  if (ParseBytecode(buffer, mapFile.size(), fileName, ErrMsg)) {
+  return ParseBytecode(buffer, mapFile.size(), fileName, ErrMsg);
-    return true;
-  }
-  return false;
 }
 
 //===----------------------------------------------------------------------===//
@@ -102,13 +101,8 @@ BytecodeBufferReader::BytecodeBufferReader(const unsigned char *buf,
                                            unsigned len,
                                            const std::string &modID,
                                            llvm::BytecodeHandler *H)
-  : BytecodeReader(H)
+  : BytecodeReader(H), Buffer(0), Buf(buf), Length(len), ModuleID(modID)
-  , Buffer(0)
+  , MustDelete(false) {
-  , Buf(buf)
-  , Length(len)
-  , ModuleID(modID)
-  , MustDelete(false)
-{
 }
 
 BytecodeBufferReader::~BytecodeBufferReader() {
@@ -316,9 +310,10 @@ Module *llvm::ParseBytecodeBuffer(const unsigned char *Buffer, unsigned Length,
                                   std::string *ErrMsg) {
   ModuleProvider *MP = 
     getBytecodeBufferModuleProvider(Buffer, Length, ModuleID, ErrMsg, 0);
-  if (!MP)
+  if (!MP) return 0;
-    return 0;
+  Module *M = MP->releaseModule();
-  return MP->releaseModule();
+  delete MP;
+  return M;
 }
 
 /// getBytecodeModuleProvider - lazy function-at-a-time loading from a file
@@ -347,9 +342,10 @@ llvm::getBytecodeModuleProvider(const std::string &Filename,
 Module *llvm::ParseBytecodeFile(const std::string &Filename,
                                 std::string *ErrMsg) {
   ModuleProvider* MP = getBytecodeModuleProvider(Filename, ErrMsg);
-  if (!MP)
+  if (!MP) return 0;
-    return 0;
+  Module *M = MP->releaseModule();
-  return MP->releaseModule();
+  delete MP;
+  return M;
 }
 
 // AnalyzeBytecodeFile - analyze one file
@@ -358,13 +354,13 @@ Module* llvm::AnalyzeBytecodeFile(
   BytecodeAnalysis& bca,        ///< Statistical output
   std::string *ErrMsg,          ///< Error output
   std::ostream* output          ///< Dump output
-)
+) {
-{
   BytecodeHandler* AH = createBytecodeAnalyzerHandler(bca,output);
   ModuleProvider* MP = getBytecodeModuleProvider(Filename, ErrMsg, AH);
-  if (!MP)
+  if (!MP) return 0;
-    return 0;
+  Module *M = MP->releaseModule();
-  return MP->releaseModule();
+  delete MP;
+  return M;
 }
 
 // AnalyzeBytecodeBuffer - analyze a buffer
@@ -380,9 +376,10 @@ Module* llvm::AnalyzeBytecodeBuffer(
   BytecodeHandler* hdlr = createBytecodeAnalyzerHandler(bca, output);
   ModuleProvider* MP = 
     getBytecodeBufferModuleProvider(Buffer, Length, ModuleID, ErrMsg, hdlr);
-  if (!MP)
+  if (!MP) return 0;
-    return 0;
+  Module *M = MP->releaseModule();
-  return MP->releaseModule();
+  delete MP;
+  return M;
 }
 
 bool llvm::GetBytecodeDependentLibraries(const std::string &fname,
@@ -396,6 +393,7 @@ bool llvm::GetBytecodeDependentLibraries(const std::string &fname,
   Module* M = MP->releaseModule();
   deplibs = M->getLibraries();
   delete M;
+  delete MP;
   return false;
 }
 
@@ -423,13 +421,16 @@ bool llvm::GetBytecodeSymbols(const sys::Path& fName,
 
   // Get the module from the provider
   Module* M = MP->materializeModule();
-  if (M == 0) 
+  if (M == 0) {
+    delete MP;
     return true;
+  }
 
   // Get the symbols
   getSymbols(M, symbols);
 
-  // Done with the module
+  // Done with the module.
+  delete MP;
   return true;
 }