[lld] r303491 - COFF: migrate def parser from LLD to LLVM [2/2]
Martell Malone via llvm-commits
llvm-commits at lists.llvm.org
Sat May 20 12:56:44 PDT 2017
Author: martell
Date: Sat May 20 14:56:44 2017
New Revision: 303491
URL: http://llvm.org/viewvc/llvm-project?rev=303491&view=rev
Log:
COFF: migrate def parser from LLD to LLVM [2/2]
This is split up into two commits.
This commit removes the DEF parser from LLD
See the previous commit for the creation in LLVM.
Reviewers: ruiu
Differential Revision: https://reviews.llvm.org/D32689
Removed:
lld/trunk/COFF/Librarian.cpp
lld/trunk/COFF/ModuleDef.cpp
Modified:
lld/trunk/COFF/CMakeLists.txt
lld/trunk/COFF/Driver.cpp
lld/trunk/COFF/Driver.h
Modified: lld/trunk/COFF/CMakeLists.txt
URL: http://llvm.org/viewvc/llvm-project/lld/trunk/COFF/CMakeLists.txt?rev=303491&r1=303490&r2=303491&view=diff
==============================================================================
--- lld/trunk/COFF/CMakeLists.txt (original)
+++ lld/trunk/COFF/CMakeLists.txt Sat May 20 14:56:44 2017
@@ -14,11 +14,9 @@ add_lld_library(lldCOFF
Error.cpp
ICF.cpp
InputFiles.cpp
- Librarian.cpp
LTO.cpp
MapFile.cpp
MarkLive.cpp
- ModuleDef.cpp
PDB.cpp
Strings.cpp
SymbolTable.cpp
Modified: lld/trunk/COFF/Driver.cpp
URL: http://llvm.org/viewvc/llvm-project/lld/trunk/COFF/Driver.cpp?rev=303491&r1=303490&r2=303491&view=diff
==============================================================================
--- lld/trunk/COFF/Driver.cpp (original)
+++ lld/trunk/COFF/Driver.cpp Sat May 20 14:56:44 2017
@@ -19,6 +19,8 @@
#include "llvm/ADT/Optional.h"
#include "llvm/ADT/StringSwitch.h"
#include "llvm/Object/ArchiveWriter.h"
+#include "llvm/Object/COFFImportFile.h"
+#include "llvm/Object/COFFModuleDefinition.h"
#include "llvm/Option/Arg.h"
#include "llvm/Option/ArgList.h"
#include "llvm/Option/Option.h"
@@ -35,6 +37,7 @@
#include <future>
using namespace llvm;
+using namespace llvm::object;
using namespace llvm::COFF;
using llvm::sys::Process;
using llvm::sys::fs::file_magic;
@@ -419,6 +422,84 @@ static std::string getMapFile(const opt:
return (OutFile.substr(0, OutFile.rfind('.')) + ".map").str();
}
+static std::string getImplibPath() {
+ if (!Config->Implib.empty())
+ return Config->Implib;
+ SmallString<128> Out = StringRef(Config->OutputFile);
+ sys::path::replace_extension(Out, ".lib");
+ return Out.str();
+}
+
+std::vector<COFFShortExport> createCOFFShortExportFromConfig() {
+ std::vector<COFFShortExport> Exports;
+ for (Export &E1 : Config->Exports) {
+ COFFShortExport E2;
+ E2.Name = E1.Name;
+ E2.ExtName = E1.ExtName;
+ E2.Ordinal = E1.Ordinal;
+ E2.Noname = E1.Noname;
+ E2.Data = E1.Data;
+ E2.Private = E1.Private;
+ E2.Constant = E1.Constant;
+ Exports.push_back(E2);
+ }
+ return Exports;
+}
+
+static void createImportLibrary() {
+ std::vector<COFFShortExport> Exports = createCOFFShortExportFromConfig();
+ std::string DLLName = sys::path::filename(Config->OutputFile);
+ std::string Path = getImplibPath();
+ writeImportLibrary(DLLName, Path, Exports, Config->Machine);
+}
+
+static void parseModuleDefs(StringRef Path) {
+ std::unique_ptr<MemoryBuffer> MB = check(
+ MemoryBuffer::getFile(Path, -1, false, true), "could not open " + Path);
+ MemoryBufferRef MBRef = MB->getMemBufferRef();
+
+ Expected<COFFModuleDefinition> Def =
+ parseCOFFModuleDefinition(MBRef, Config->Machine);
+ if (!Def)
+ fatal(errorToErrorCode(Def.takeError()).message());
+
+ COFFModuleDefinition &M = *Def;
+ if (Config->OutputFile.empty())
+ Config->OutputFile = Saver.save(M.OutputFile);
+
+ if (M.ImageBase)
+ Config->ImageBase = M.ImageBase;
+ if (M.StackReserve)
+ Config->StackReserve = M.StackReserve;
+ if (M.StackCommit)
+ Config->StackCommit = M.StackCommit;
+ if (M.HeapReserve)
+ Config->HeapReserve = M.HeapReserve;
+ if (M.HeapCommit)
+ Config->HeapCommit = M.HeapCommit;
+ if (M.MajorImageVersion)
+ Config->MajorImageVersion = M.MajorImageVersion;
+ if (M.MinorImageVersion)
+ Config->MinorImageVersion = M.MinorImageVersion;
+ if (M.MajorOSVersion)
+ Config->MajorOSVersion = M.MajorOSVersion;
+ if (M.MinorOSVersion)
+ Config->MinorOSVersion = M.MinorOSVersion;
+
+ for (COFFShortExport E1 : M.Exports) {
+ Export E2;
+ E2.Name = Saver.save(E1.Name);
+ if (E1.isWeak())
+ E2.ExtName = Saver.save(E1.ExtName);
+ E2.Ordinal = E1.Ordinal;
+ E2.Noname = E1.Noname;
+ E2.Data = E1.Data;
+ E2.Private = E1.Private;
+ E2.Constant = E1.Constant;
+ Config->Exports.push_back(E2);
+ }
+}
+
std::vector<MemoryBufferRef> getArchiveMembers(Archive *File) {
std::vector<MemoryBufferRef> V;
Error Err = Error::success();
@@ -912,9 +993,7 @@ void LinkerDriver::link(ArrayRef<const c
// Handle /def
if (auto *Arg = Args.getLastArg(OPT_deffile)) {
// parseModuleDefs mutates Config object.
- parseModuleDefs(
- takeBuffer(check(MemoryBuffer::getFile(Arg->getValue()),
- Twine("could not open ") + Arg->getValue())));
+ parseModuleDefs(Arg->getValue());
}
// Handle /delayload
@@ -1034,7 +1113,7 @@ void LinkerDriver::link(ArrayRef<const c
// need to create a .lib file.
if (!Config->Exports.empty() || Config->DLL) {
fixupExports();
- writeImportLibrary();
+ createImportLibrary();
assignExportOrdinals();
}
Modified: lld/trunk/COFF/Driver.h
URL: http://llvm.org/viewvc/llvm-project/lld/trunk/COFF/Driver.h?rev=303491&r1=303490&r2=303491&view=diff
==============================================================================
--- lld/trunk/COFF/Driver.h (original)
+++ lld/trunk/COFF/Driver.h Sat May 20 14:56:44 2017
@@ -124,9 +124,6 @@ private:
std::vector<MemoryBufferRef> Resources;
};
-void parseModuleDefs(MemoryBufferRef MB);
-void writeImportLibrary();
-
// Functions below this line are defined in DriverUtils.cpp.
void printHelp(const char *Argv0);
Removed: lld/trunk/COFF/Librarian.cpp
URL: http://llvm.org/viewvc/llvm-project/lld/trunk/COFF/Librarian.cpp?rev=303490&view=auto
==============================================================================
--- lld/trunk/COFF/Librarian.cpp (original)
+++ lld/trunk/COFF/Librarian.cpp (removed)
@@ -1,511 +0,0 @@
-//===- Librarian.cpp ------------------------------------------------------===//
-//
-// The LLVM Linker
-//
-// This file is distributed under the University of Illinois Open Source
-// License. See LICENSE.TXT for details.
-//
-//===----------------------------------------------------------------------===//
-//
-// This file contains functions for the Librarian. The librarian creates and
-// manages libraries of the Common Object File Format (COFF) object files. It
-// primarily is used for creating static libraries and import libraries.
-//
-//===----------------------------------------------------------------------===//
-
-#include "Config.h"
-#include "Driver.h"
-#include "Error.h"
-#include "Symbols.h"
-#include "llvm/Object/Archive.h"
-#include "llvm/Object/ArchiveWriter.h"
-#include "llvm/Object/COFF.h"
-#include "llvm/Support/Path.h"
-
-#include <vector>
-
-using namespace lld::coff;
-using namespace llvm::COFF;
-using namespace llvm::object;
-using namespace llvm;
-
-static bool is32bit() {
- switch (Config->Machine) {
- default:
- llvm_unreachable("unsupported machine");
- case IMAGE_FILE_MACHINE_AMD64:
- return false;
- case IMAGE_FILE_MACHINE_ARMNT:
- case IMAGE_FILE_MACHINE_I386:
- return true;
- }
-}
-
-static uint16_t getImgRelRelocation() {
- switch (Config->Machine) {
- default:
- llvm_unreachable("unsupported machine");
- case IMAGE_FILE_MACHINE_AMD64:
- return IMAGE_REL_AMD64_ADDR32NB;
- case IMAGE_FILE_MACHINE_ARMNT:
- return IMAGE_REL_ARM_ADDR32NB;
- case IMAGE_FILE_MACHINE_I386:
- return IMAGE_REL_I386_DIR32NB;
- }
-}
-
-template <class T> static void append(std::vector<uint8_t> &B, const T &Data) {
- size_t S = B.size();
- B.resize(S + sizeof(T));
- memcpy(&B[S], &Data, sizeof(T));
-}
-
-static void writeStringTable(std::vector<uint8_t> &B,
- ArrayRef<const std::string> Strings) {
- // The COFF string table consists of a 4-byte value which is the size of the
- // table, including the length field itself. This value is followed by the
- // string content itself, which is an array of null-terminated C-style
- // strings. The termination is important as they are referenced to by offset
- // by the symbol entity in the file format.
-
- std::vector<uint8_t>::size_type Pos = B.size();
- std::vector<uint8_t>::size_type Offset = B.size();
-
- // Skip over the length field, we will fill it in later as we will have
- // computed the length while emitting the string content itself.
- Pos += sizeof(uint32_t);
-
- for (const auto &S : Strings) {
- B.resize(Pos + S.length() + 1);
- strcpy(reinterpret_cast<char *>(&B[Pos]), S.c_str());
- Pos += S.length() + 1;
- }
-
- // Backfill the length of the table now that it has been computed.
- support::ulittle32_t Length(B.size() - Offset);
- memcpy(&B[Offset], &Length, sizeof(Length));
-}
-
-static std::string getImplibPath() {
- if (!Config->Implib.empty())
- return Config->Implib;
- SmallString<128> Out = StringRef(Config->OutputFile);
- sys::path::replace_extension(Out, ".lib");
- return Out.str();
-}
-
-static ImportNameType getNameType(StringRef Sym, StringRef ExtName) {
- if (Sym != ExtName)
- return IMPORT_NAME_UNDECORATE;
- if (Config->Machine == I386 && Sym.startswith("_"))
- return IMPORT_NAME_NOPREFIX;
- return IMPORT_NAME;
-}
-
-static std::string replace(StringRef S, StringRef From, StringRef To) {
- size_t Pos = S.find(From);
-
- // From and To may be mangled, but substrings in S may not.
- if (Pos == StringRef::npos && From.startswith("_") && To.startswith("_")) {
- From = From.substr(1);
- To = To.substr(1);
- Pos = S.find(From);
- }
-
- if (Pos == StringRef::npos) {
- error(S + ": replacing '" + From + "' with '" + To + "' failed");
- return "";
- }
- return (Twine(S.substr(0, Pos)) + To + S.substr(Pos + From.size())).str();
-}
-
-static const std::string NullImportDescriptorSymbolName =
- "__NULL_IMPORT_DESCRIPTOR";
-
-namespace {
-// This class constructs various small object files necessary to support linking
-// symbols imported from a DLL. The contents are pretty strictly defined and
-// nearly entirely static. The details of the structures files are defined in
-// WINNT.h and the PE/COFF specification.
-class ObjectFactory {
- using u16 = support::ulittle16_t;
- using u32 = support::ulittle32_t;
-
- BumpPtrAllocator Alloc;
- StringRef DLLName;
- StringRef Library;
- std::string ImportDescriptorSymbolName;
- std::string NullThunkSymbolName;
-
-public:
- ObjectFactory(StringRef S)
- : DLLName(S), Library(S.drop_back(4)),
- ImportDescriptorSymbolName(("__IMPORT_DESCRIPTOR_" + Library).str()),
- NullThunkSymbolName(("\x7f" + Library + "_NULL_THUNK_DATA").str()) {}
-
- // Creates an Import Descriptor. This is a small object file which contains a
- // reference to the terminators and contains the library name (entry) for the
- // import name table. It will force the linker to construct the necessary
- // structure to import symbols from the DLL.
- NewArchiveMember createImportDescriptor(std::vector<uint8_t> &Buffer);
-
- // Creates a NULL import descriptor. This is a small object file whcih
- // contains a NULL import descriptor. It is used to terminate the imports
- // from a specific DLL.
- NewArchiveMember createNullImportDescriptor(std::vector<uint8_t> &Buffer);
-
- // Create a NULL Thunk Entry. This is a small object file which contains a
- // NULL Import Address Table entry and a NULL Import Lookup Table Entry. It
- // is used to terminate the IAT and ILT.
- NewArchiveMember createNullThunk(std::vector<uint8_t> &Buffer);
-
- // Create a short import file which is described in PE/COFF spec 7. Import
- // Library Format.
- NewArchiveMember createShortImport(StringRef Sym, uint16_t Ordinal,
- ImportType Type, ImportNameType NameType);
-};
-}
-
-NewArchiveMember
-ObjectFactory::createImportDescriptor(std::vector<uint8_t> &Buffer) {
- static const uint32_t NumberOfSections = 2;
- static const uint32_t NumberOfSymbols = 7;
- static const uint32_t NumberOfRelocations = 3;
-
- // COFF Header
- coff_file_header Header{
- u16(Config->Machine), u16(NumberOfSections), u32(0),
- u32(sizeof(Header) + (NumberOfSections * sizeof(coff_section)) +
- // .idata$2
- sizeof(coff_import_directory_table_entry) +
- NumberOfRelocations * sizeof(coff_relocation) +
- // .idata$4
- (DLLName.size() + 1)),
- u32(NumberOfSymbols), u16(0),
- u16(is32bit() ? IMAGE_FILE_32BIT_MACHINE : 0),
- };
- append(Buffer, Header);
-
- // Section Header Table
- static const coff_section SectionTable[NumberOfSections] = {
- {{'.', 'i', 'd', 'a', 't', 'a', '$', '2'},
- u32(0),
- u32(0),
- u32(sizeof(coff_import_directory_table_entry)),
- u32(sizeof(coff_file_header) + NumberOfSections * sizeof(coff_section)),
- u32(sizeof(coff_file_header) + NumberOfSections * sizeof(coff_section) +
- sizeof(coff_import_directory_table_entry)),
- u32(0),
- u16(NumberOfRelocations),
- u16(0),
- u32(IMAGE_SCN_ALIGN_4BYTES | IMAGE_SCN_CNT_INITIALIZED_DATA |
- IMAGE_SCN_MEM_READ | IMAGE_SCN_MEM_WRITE)},
- {{'.', 'i', 'd', 'a', 't', 'a', '$', '6'},
- u32(0),
- u32(0),
- u32(DLLName.size() + 1),
- u32(sizeof(coff_file_header) + NumberOfSections * sizeof(coff_section) +
- sizeof(coff_import_directory_table_entry) +
- NumberOfRelocations * sizeof(coff_relocation)),
- u32(0),
- u32(0),
- u16(0),
- u16(0),
- u32(IMAGE_SCN_ALIGN_2BYTES | IMAGE_SCN_CNT_INITIALIZED_DATA |
- IMAGE_SCN_MEM_READ | IMAGE_SCN_MEM_WRITE)},
- };
- append(Buffer, SectionTable);
-
- // .idata$2
- static const coff_import_directory_table_entry ImportDescriptor{
- u32(0), u32(0), u32(0), u32(0), u32(0),
- };
- append(Buffer, ImportDescriptor);
-
- static const coff_relocation RelocationTable[NumberOfRelocations] = {
- {u32(offsetof(coff_import_directory_table_entry, NameRVA)), u32(2),
- u16(getImgRelRelocation())},
- {u32(offsetof(coff_import_directory_table_entry, ImportLookupTableRVA)),
- u32(3), u16(getImgRelRelocation())},
- {u32(offsetof(coff_import_directory_table_entry, ImportAddressTableRVA)),
- u32(4), u16(getImgRelRelocation())},
- };
- append(Buffer, RelocationTable);
-
- // .idata$6
- auto S = Buffer.size();
- Buffer.resize(S + DLLName.size() + 1);
- memcpy(&Buffer[S], DLLName.data(), DLLName.size());
- Buffer[S + DLLName.size()] = '\0';
-
- // Symbol Table
- coff_symbol16 SymbolTable[NumberOfSymbols] = {
- {{{0, 0, 0, 0, 0, 0, 0, 0}},
- u32(0),
- u16(1),
- u16(0),
- IMAGE_SYM_CLASS_EXTERNAL,
- 0},
- {{{'.', 'i', 'd', 'a', 't', 'a', '$', '2'}},
- u32(0),
- u16(1),
- u16(0),
- IMAGE_SYM_CLASS_SECTION,
- 0},
- {{{'.', 'i', 'd', 'a', 't', 'a', '$', '6'}},
- u32(0),
- u16(2),
- u16(0),
- IMAGE_SYM_CLASS_STATIC,
- 0},
- {{{'.', 'i', 'd', 'a', 't', 'a', '$', '4'}},
- u32(0),
- u16(0),
- u16(0),
- IMAGE_SYM_CLASS_SECTION,
- 0},
- {{{'.', 'i', 'd', 'a', 't', 'a', '$', '5'}},
- u32(0),
- u16(0),
- u16(0),
- IMAGE_SYM_CLASS_SECTION,
- 0},
- {{{0, 0, 0, 0, 0, 0, 0, 0}},
- u32(0),
- u16(0),
- u16(0),
- IMAGE_SYM_CLASS_EXTERNAL,
- 0},
- {{{0, 0, 0, 0, 0, 0, 0, 0}},
- u32(0),
- u16(0),
- u16(0),
- IMAGE_SYM_CLASS_EXTERNAL,
- 0},
- };
- reinterpret_cast<StringTableOffset &>(SymbolTable[0].Name).Offset =
- sizeof(uint32_t);
- reinterpret_cast<StringTableOffset &>(SymbolTable[5].Name).Offset =
- sizeof(uint32_t) + ImportDescriptorSymbolName.length() + 1;
- reinterpret_cast<StringTableOffset &>(SymbolTable[6].Name).Offset =
- sizeof(uint32_t) + ImportDescriptorSymbolName.length() + 1 +
- NullImportDescriptorSymbolName.length() + 1;
- append(Buffer, SymbolTable);
-
- // String Table
- writeStringTable(Buffer,
- {ImportDescriptorSymbolName, NullImportDescriptorSymbolName,
- NullThunkSymbolName});
-
- StringRef F{reinterpret_cast<const char *>(Buffer.data()), Buffer.size()};
- return {MemoryBufferRef(F, DLLName)};
-}
-
-NewArchiveMember
-ObjectFactory::createNullImportDescriptor(std::vector<uint8_t> &Buffer) {
- static const uint32_t NumberOfSections = 1;
- static const uint32_t NumberOfSymbols = 1;
-
- // COFF Header
- coff_file_header Header{
- u16(Config->Machine), u16(NumberOfSections), u32(0),
- u32(sizeof(Header) + (NumberOfSections * sizeof(coff_section)) +
- // .idata$3
- sizeof(coff_import_directory_table_entry)),
- u32(NumberOfSymbols), u16(0),
- u16(is32bit() ? IMAGE_FILE_32BIT_MACHINE : 0),
- };
- append(Buffer, Header);
-
- // Section Header Table
- static const coff_section SectionTable[NumberOfSections] = {
- {{'.', 'i', 'd', 'a', 't', 'a', '$', '3'},
- u32(0),
- u32(0),
- u32(sizeof(coff_import_directory_table_entry)),
- u32(sizeof(coff_file_header) +
- (NumberOfSections * sizeof(coff_section))),
- u32(0),
- u32(0),
- u16(0),
- u16(0),
- u32(IMAGE_SCN_ALIGN_4BYTES | IMAGE_SCN_CNT_INITIALIZED_DATA |
- IMAGE_SCN_MEM_READ | IMAGE_SCN_MEM_WRITE)},
- };
- append(Buffer, SectionTable);
-
- // .idata$3
- static const coff_import_directory_table_entry ImportDescriptor{
- u32(0), u32(0), u32(0), u32(0), u32(0),
- };
- append(Buffer, ImportDescriptor);
-
- // Symbol Table
- coff_symbol16 SymbolTable[NumberOfSymbols] = {
- {{{0, 0, 0, 0, 0, 0, 0, 0}},
- u32(0),
- u16(1),
- u16(0),
- IMAGE_SYM_CLASS_EXTERNAL,
- 0},
- };
- reinterpret_cast<StringTableOffset &>(SymbolTable[0].Name).Offset =
- sizeof(uint32_t);
- append(Buffer, SymbolTable);
-
- // String Table
- writeStringTable(Buffer, {NullImportDescriptorSymbolName});
-
- StringRef F{reinterpret_cast<const char *>(Buffer.data()), Buffer.size()};
- return {MemoryBufferRef(F, DLLName)};
-}
-
-NewArchiveMember ObjectFactory::createNullThunk(std::vector<uint8_t> &Buffer) {
- static const uint32_t NumberOfSections = 2;
- static const uint32_t NumberOfSymbols = 1;
- uint32_t VASize = is32bit() ? 4 : 8;
-
- // COFF Header
- coff_file_header Header{
- u16(Config->Machine), u16(NumberOfSections), u32(0),
- u32(sizeof(Header) + (NumberOfSections * sizeof(coff_section)) +
- // .idata$5
- VASize +
- // .idata$4
- VASize),
- u32(NumberOfSymbols), u16(0),
- u16(is32bit() ? IMAGE_FILE_32BIT_MACHINE : 0),
- };
- append(Buffer, Header);
-
- // Section Header Table
- static const coff_section SectionTable[NumberOfSections] = {
- {{'.', 'i', 'd', 'a', 't', 'a', '$', '5'},
- u32(0),
- u32(0),
- u32(VASize),
- u32(sizeof(coff_file_header) + NumberOfSections * sizeof(coff_section)),
- u32(0),
- u32(0),
- u16(0),
- u16(0),
- u32((is32bit() ? IMAGE_SCN_ALIGN_4BYTES : IMAGE_SCN_ALIGN_8BYTES) |
- IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ |
- IMAGE_SCN_MEM_WRITE)},
- {{'.', 'i', 'd', 'a', 't', 'a', '$', '4'},
- u32(0),
- u32(0),
- u32(VASize),
- u32(sizeof(coff_file_header) + NumberOfSections * sizeof(coff_section) +
- VASize),
- u32(0),
- u32(0),
- u16(0),
- u16(0),
- u32((is32bit() ? IMAGE_SCN_ALIGN_4BYTES : IMAGE_SCN_ALIGN_8BYTES) |
- IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ |
- IMAGE_SCN_MEM_WRITE)},
- };
- append(Buffer, SectionTable);
-
- // .idata$5, ILT
- append(Buffer, u32(0));
- if (!is32bit())
- append(Buffer, u32(0));
-
- // .idata$4, IAT
- append(Buffer, u32(0));
- if (!is32bit())
- append(Buffer, u32(0));
-
- // Symbol Table
- coff_symbol16 SymbolTable[NumberOfSymbols] = {
- {{{0, 0, 0, 0, 0, 0, 0, 0}},
- u32(0),
- u16(1),
- u16(0),
- IMAGE_SYM_CLASS_EXTERNAL,
- 0},
- };
- reinterpret_cast<StringTableOffset &>(SymbolTable[0].Name).Offset =
- sizeof(uint32_t);
- append(Buffer, SymbolTable);
-
- // String Table
- writeStringTable(Buffer, {NullThunkSymbolName});
-
- StringRef F{reinterpret_cast<const char *>(Buffer.data()), Buffer.size()};
- return {MemoryBufferRef{F, DLLName}};
-}
-
-NewArchiveMember ObjectFactory::createShortImport(StringRef Sym,
- uint16_t Ordinal,
- ImportType ImportType,
- ImportNameType NameType) {
- size_t ImpSize = DLLName.size() + Sym.size() + 2; // +2 for NULs
- size_t Size = sizeof(coff_import_header) + ImpSize;
- char *Buf = Alloc.Allocate<char>(Size);
- memset(Buf, 0, Size);
- char *P = Buf;
-
- // Write short import library.
- auto *Imp = reinterpret_cast<coff_import_header *>(P);
- P += sizeof(*Imp);
- Imp->Sig2 = 0xFFFF;
- Imp->Machine = Config->Machine;
- Imp->SizeOfData = ImpSize;
- if (Ordinal > 0)
- Imp->OrdinalHint = Ordinal;
- Imp->TypeInfo = (NameType << 2) | ImportType;
-
- // Write symbol name and DLL name.
- memcpy(P, Sym.data(), Sym.size());
- P += Sym.size() + 1;
- memcpy(P, DLLName.data(), DLLName.size());
-
- return {MemoryBufferRef(StringRef(Buf, Size), DLLName)};
-}
-
-// Creates an import library for a DLL. In this function, we first
-// create an empty import library using lib.exe and then adds short
-// import files to that file.
-void lld::coff::writeImportLibrary() {
- std::vector<NewArchiveMember> Members;
-
- std::string Path = getImplibPath();
- std::string DLLName = sys::path::filename(Config->OutputFile);
- ObjectFactory OF(DLLName);
-
- std::vector<uint8_t> ImportDescriptor;
- Members.push_back(OF.createImportDescriptor(ImportDescriptor));
-
- std::vector<uint8_t> NullImportDescriptor;
- Members.push_back(OF.createNullImportDescriptor(NullImportDescriptor));
-
- std::vector<uint8_t> NullThunk;
- Members.push_back(OF.createNullThunk(NullThunk));
-
- for (Export &E : Config->Exports) {
- if (E.Private)
- continue;
-
- ImportType ImportType = IMPORT_CODE;
- if (E.Data)
- ImportType = IMPORT_DATA;
- if (E.Constant)
- ImportType = IMPORT_CONST;
-
- ImportNameType NameType = getNameType(E.SymbolName, E.Name);
- std::string Name = E.ExtName.empty()
- ? std::string(E.SymbolName)
- : replace(E.SymbolName, E.Name, E.ExtName);
- Members.push_back(OF.createShortImport(Name, E.Ordinal, ImportType,
- NameType));
- }
-
- std::pair<StringRef, std::error_code> Result =
- writeArchive(Path, Members, /*WriteSymtab*/ true, object::Archive::K_GNU,
- /*Deterministic*/ true, /*Thin*/ false);
- if (auto EC = Result.second)
- fatal(EC, "failed to write " + Path);
-}
Removed: lld/trunk/COFF/ModuleDef.cpp
URL: http://llvm.org/viewvc/llvm-project/lld/trunk/COFF/ModuleDef.cpp?rev=303490&view=auto
==============================================================================
--- lld/trunk/COFF/ModuleDef.cpp (original)
+++ lld/trunk/COFF/ModuleDef.cpp (removed)
@@ -1,304 +0,0 @@
-//===- COFF/ModuleDef.cpp -------------------------------------------------===//
-//
-// The LLVM Linker
-//
-// This file is distributed under the University of Illinois Open Source
-// License. See LICENSE.TXT for details.
-//
-//===----------------------------------------------------------------------===//
-//
-// Windows-specific.
-// A parser for the module-definition file (.def file).
-// Parsed results are directly written to Config global variable.
-//
-// The format of module-definition files are described in this document:
-// https://msdn.microsoft.com/en-us/library/28d6s79h.aspx
-//
-//===----------------------------------------------------------------------===//
-
-#include "Config.h"
-#include "Error.h"
-#include "Memory.h"
-#include "llvm/ADT/StringRef.h"
-#include "llvm/ADT/StringSwitch.h"
-#include "llvm/Support/StringSaver.h"
-#include "llvm/Support/raw_ostream.h"
-#include <system_error>
-
-using namespace llvm;
-
-namespace lld {
-namespace coff {
-namespace {
-
-enum Kind {
- Unknown,
- Eof,
- Identifier,
- Comma,
- Equal,
- KwBase,
- KwConstant,
- KwData,
- KwExports,
- KwHeapsize,
- KwLibrary,
- KwName,
- KwNoname,
- KwPrivate,
- KwStacksize,
- KwVersion,
-};
-
-struct Token {
- explicit Token(Kind T = Unknown, StringRef S = "") : K(T), Value(S) {}
- Kind K;
- StringRef Value;
-};
-
-static bool isDecorated(StringRef Sym) {
- return Sym.startswith("_") || Sym.startswith("@") || Sym.startswith("?");
-}
-
-class Lexer {
-public:
- explicit Lexer(StringRef S) : Buf(S) {}
-
- Token lex() {
- Buf = Buf.trim();
- if (Buf.empty())
- return Token(Eof);
-
- switch (Buf[0]) {
- case '\0':
- return Token(Eof);
- case ';': {
- size_t End = Buf.find('\n');
- Buf = (End == Buf.npos) ? "" : Buf.drop_front(End);
- return lex();
- }
- case '=':
- Buf = Buf.drop_front();
- return Token(Equal, "=");
- case ',':
- Buf = Buf.drop_front();
- return Token(Comma, ",");
- case '"': {
- StringRef S;
- std::tie(S, Buf) = Buf.substr(1).split('"');
- return Token(Identifier, S);
- }
- default: {
- size_t End = Buf.find_first_of("=,\r\n \t\v");
- StringRef Word = Buf.substr(0, End);
- Kind K = llvm::StringSwitch<Kind>(Word)
- .Case("BASE", KwBase)
- .Case("CONSTANT", KwConstant)
- .Case("DATA", KwData)
- .Case("EXPORTS", KwExports)
- .Case("HEAPSIZE", KwHeapsize)
- .Case("LIBRARY", KwLibrary)
- .Case("NAME", KwName)
- .Case("NONAME", KwNoname)
- .Case("PRIVATE", KwPrivate)
- .Case("STACKSIZE", KwStacksize)
- .Case("VERSION", KwVersion)
- .Default(Identifier);
- Buf = (End == Buf.npos) ? "" : Buf.drop_front(End);
- return Token(K, Word);
- }
- }
- }
-
-private:
- StringRef Buf;
-};
-
-class Parser {
-public:
- explicit Parser(StringRef S) : Lex(S) {}
-
- void parse() {
- do {
- parseOne();
- } while (Tok.K != Eof);
- }
-
-private:
- void read() {
- if (Stack.empty()) {
- Tok = Lex.lex();
- return;
- }
- Tok = Stack.back();
- Stack.pop_back();
- }
-
- void readAsInt(uint64_t *I) {
- read();
- if (Tok.K != Identifier || Tok.Value.getAsInteger(10, *I))
- fatal("integer expected");
- }
-
- void expect(Kind Expected, StringRef Msg) {
- read();
- if (Tok.K != Expected)
- fatal(Msg);
- }
-
- void unget() { Stack.push_back(Tok); }
-
- void parseOne() {
- read();
- switch (Tok.K) {
- case Eof:
- return;
- case KwExports:
- for (;;) {
- read();
- if (Tok.K != Identifier) {
- unget();
- return;
- }
- parseExport();
- }
- case KwHeapsize:
- parseNumbers(&Config->HeapReserve, &Config->HeapCommit);
- return;
- case KwStacksize:
- parseNumbers(&Config->StackReserve, &Config->StackCommit);
- return;
- case KwLibrary:
- case KwName: {
- bool IsDll = Tok.K == KwLibrary; // Check before parseName.
- std::string Name;
- parseName(&Name, &Config->ImageBase);
-
- // Append the appropriate file extension if not already present.
- StringRef Ext = IsDll ? ".dll" : ".exe";
- if (!StringRef(Name).endswith_lower(Ext))
- Name += Ext;
-
- // Set the output file, but don't override /out if it was already passed.
- if (Config->OutputFile.empty())
- Config->OutputFile = Name;
- return;
- }
- case KwVersion:
- parseVersion(&Config->MajorImageVersion, &Config->MinorImageVersion);
- return;
- default:
- fatal("unknown directive: " + Tok.Value);
- }
- }
-
- void parseExport() {
- Export E;
- E.Name = Tok.Value;
- read();
- if (Tok.K == Equal) {
- read();
- if (Tok.K != Identifier)
- fatal("identifier expected, but got " + Tok.Value);
- E.ExtName = E.Name;
- E.Name = Tok.Value;
- } else {
- unget();
- }
-
- if (Config->Machine == I386) {
- if (!isDecorated(E.Name))
- E.Name = Saver.save("_" + E.Name);
- if (!E.ExtName.empty() && !isDecorated(E.ExtName))
- E.ExtName = Saver.save("_" + E.ExtName);
- }
-
- for (;;) {
- read();
- if (Tok.K == Identifier && Tok.Value[0] == '@') {
- Tok.Value.drop_front().getAsInteger(10, E.Ordinal);
- read();
- if (Tok.K == KwNoname) {
- E.Noname = true;
- } else {
- unget();
- }
- continue;
- }
- if (Tok.K == KwData) {
- E.Data = true;
- continue;
- }
- if (Tok.K == KwConstant) {
- warn("CONSTANT keyword is obsolete; use DATA");
- E.Constant = true;
- continue;
- }
- if (Tok.K == KwPrivate) {
- E.Private = true;
- continue;
- }
- unget();
- Config->Exports.push_back(E);
- return;
- }
- }
-
- // HEAPSIZE/STACKSIZE reserve[,commit]
- void parseNumbers(uint64_t *Reserve, uint64_t *Commit) {
- readAsInt(Reserve);
- read();
- if (Tok.K != Comma) {
- unget();
- Commit = nullptr;
- return;
- }
- readAsInt(Commit);
- }
-
- // NAME outputPath [BASE=address]
- void parseName(std::string *Out, uint64_t *Baseaddr) {
- read();
- if (Tok.K == Identifier) {
- *Out = Tok.Value;
- } else {
- *Out = "";
- unget();
- return;
- }
- read();
- if (Tok.K == KwBase) {
- expect(Equal, "'=' expected");
- readAsInt(Baseaddr);
- } else {
- unget();
- *Baseaddr = 0;
- }
- }
-
- // VERSION major[.minor]
- void parseVersion(uint32_t *Major, uint32_t *Minor) {
- read();
- if (Tok.K != Identifier)
- fatal("identifier expected, but got " + Tok.Value);
- StringRef V1, V2;
- std::tie(V1, V2) = Tok.Value.split('.');
- if (V1.getAsInteger(10, *Major))
- fatal("integer expected, but got " + Tok.Value);
- if (V2.empty())
- *Minor = 0;
- else if (V2.getAsInteger(10, *Minor))
- fatal("integer expected, but got " + Tok.Value);
- }
-
- Lexer Lex;
- Token Tok;
- std::vector<Token> Stack;
-};
-
-} // anonymous namespace
-
-void parseModuleDefs(MemoryBufferRef MB) { Parser(MB.getBuffer()).parse(); }
-
-} // namespace coff
-} // namespace lld
More information about the llvm-commits
mailing list