Destroy allocated resources on exception.

llvm-svn: 8969

1 files changed, 13 insertions, 3 deletions

diff --git a/llvm/lib/Bytecode/Reader/ReaderWrappers.cpp b/llvm/lib/Bytecode/Reader/ReaderWrappers.cpp
index 99c0e6be9ee..9b6327437d8 100644
--- a/llvm/lib/Bytecode/Reader/ReaderWrappers.cpp
+++ b/llvm/lib/Bytecode/Reader/ReaderWrappers.cpp
@@ -59,8 +59,13 @@ BytecodeFileReader::BytecodeFileReader(const std::string &Filename) {
   if (Buffer == (unsigned char*)MAP_FAILED)
     throw std::string("Error mmapping file!");
 
-  // Parse the bytecode we mmapped in
-  ParseBytecode(Buffer, Length, Filename);
+  try {
+    // Parse the bytecode we mmapped in
+    ParseBytecode(Buffer, Length, Filename);
+  } catch (...) {
+    munmap((char*)Buffer, Length);
+    throw;
+  }
 }
 
 BytecodeFileReader::~BytecodeFileReader() {
@@ -106,7 +111,12 @@ BytecodeBufferReader::BytecodeBufferReader(const unsigned char *Buf,
     ParseBegin = Buffer = Buf;
     MustDelete = false;
   }
-  ParseBytecode(ParseBegin, Length, ModuleID);
+  try {
+    ParseBytecode(ParseBegin, Length, ModuleID);
+  } catch (...) {
+    if (MustDelete) delete [] Buffer;
+    throw;
+  }
 }
 
 BytecodeBufferReader::~BytecodeBufferReader() {