diff options
-rw-r--r-- | lld/lib/Driver/WinLinkDriver.cpp | 8 | ||||
-rw-r--r-- | lld/lib/ReaderWriter/PECOFF/ReaderCOFF.cpp | 11 |
2 files changed, 10 insertions, 9 deletions
diff --git a/lld/lib/Driver/WinLinkDriver.cpp b/lld/lib/Driver/WinLinkDriver.cpp index e6060ff39ab..13b6cb09a4a 100644 --- a/lld/lib/Driver/WinLinkDriver.cpp +++ b/lld/lib/Driver/WinLinkDriver.cpp @@ -350,12 +350,12 @@ bool WinLinkDriver::parse(int argc, const char *argv[], case OPT_entry: // handle /entry - ctx.setEntrySymbolName(ctx.allocateString(inputArg->getValue())); + ctx.setEntrySymbolName(inputArg->getValue()); break; case OPT_libpath: // handle /libpath - ctx.appendInputSearchPath(ctx.allocateString(inputArg->getValue())); + ctx.appendInputSearchPath(inputArg->getValue()); break; case OPT_force: @@ -410,12 +410,12 @@ bool WinLinkDriver::parse(int argc, const char *argv[], case OPT_incl: // handle /incl - ctx.addInitialUndefinedSymbol(ctx.allocateString(inputArg->getValue())); + ctx.addInitialUndefinedSymbol(inputArg->getValue()); break; case OPT_out: // handle /out - ctx.setOutputPath(ctx.allocateString(inputArg->getValue())); + ctx.setOutputPath(inputArg->getValue()); break; case OPT_INPUT: diff --git a/lld/lib/ReaderWriter/PECOFF/ReaderCOFF.cpp b/lld/lib/ReaderWriter/PECOFF/ReaderCOFF.cpp index 847349b763e..8edae006465 100644 --- a/lld/lib/ReaderWriter/PECOFF/ReaderCOFF.cpp +++ b/lld/lib/ReaderWriter/PECOFF/ReaderCOFF.cpp @@ -112,7 +112,7 @@ private: SectionToAtomsT; public: - FileCOFF(const LinkingContext &context, + FileCOFF(const PECOFFLinkingContext &context, std::unique_ptr<llvm::MemoryBuffer> mb, error_code &ec) : File(mb->getBufferIdentifier(), kindObject), _context(context) { llvm::OwningPtr<llvm::object::Binary> bin; @@ -602,7 +602,7 @@ private: ArrayRef<uint8_t> contents; if (error_code ec = _obj->getSectionContents(section, contents)) return ec; - _directives = std::move(ArrayRefToString(contents)); + _directives = _context.allocateString(ArrayRefToString(contents)); } return error_code::success(); } @@ -614,7 +614,7 @@ private: atom_collection_vector<AbsoluteAtom> _absoluteAtoms; // The contents of .drectve section. - std::string _directives; + StringRef _directives; // A map from symbol to its name. All symbols should be in this map except // unnamed ones. @@ -638,7 +638,7 @@ private: std::map<uint32_t, COFFDefinedAtom *>> _definedAtomLocations; mutable llvm::BumpPtrAllocator _alloc; - const LinkingContext &_context; + const PECOFFLinkingContext &_context; }; class BumpPtrStringSaver : public llvm::cl::StringSaver { @@ -714,7 +714,8 @@ private: std::vector<std::unique_ptr<File>> &result) const { // Parse the memory buffer as PECOFF file. error_code ec; - std::unique_ptr<FileCOFF> file(new FileCOFF(_context, std::move(mb), ec)); + std::unique_ptr<FileCOFF> file( + new FileCOFF(_PECOFFLinkingContext, std::move(mb), ec)); if (ec) return ec; |