commit 223fe1b9e71245c7f481d08a76f781a51a2ed831
author Rui Ueyama <ruiu@google.com> Thu Jun 18 20:29:41 2015 +0000
committer Rui Ueyama <ruiu@google.com> Thu Jun 18 20:29:41 2015 +0000
tree 21e54fdb9e6b0dd785fb91a2da45e8b74e93e26c
parent b95188cb2c820ca6e124ec993b62176355d6f2af

COFF: Fix unsafe memory access.

llvm-svn: 240046

lld/COFF/InputFiles.cpp

diff --git a/lld/COFF/InputFiles.cpp b/lld/COFF/InputFiles.cpp
index 6d084bb..1d4db89 100644
--- a/lld/COFF/InputFiles.cpp
+++ b/lld/COFF/InputFiles.cpp
@@ -256,12 +256,13 @@
     return make_error_code(LLDError::BrokenFile);
   }
 
+  llvm::BumpPtrStringSaver Saver(Alloc);
   for (unsigned I = 0, E = M->getSymbolCount(); I != E; ++I) {
     lto_symbol_attributes Attrs = M->getSymbolAttributes(I);
     if ((Attrs & LTO_SYMBOL_SCOPE_MASK) == LTO_SYMBOL_SCOPE_INTERNAL)
       continue;
 
-    StringRef SymName = M->getSymbolName(I);
+    StringRef SymName = Saver.save(M->getSymbolName(I));
     int SymbolDef = Attrs & LTO_SYMBOL_DEFINITION_MASK;
     if (SymbolDef == LTO_SYMBOL_DEFINITION_UNDEFINED) {
       SymbolBodies.push_back(new (Alloc) Undefined(SymName));