8f6d2def2b
Don't include type signatures that are not referenced by some relocation. We don't include this in the -gc-sections settings since we are always building the type section from scratch, just like we do the table elements. In the future we might want to unify the relocation processing which is currently done once for gc-sections and then again for building the sympathetic type and table sections. Differential Revision: https://reviews.llvm.org/D42747 llvm-svn: 323931
158 lines
4.3 KiB
C++
158 lines
4.3 KiB
C++
//===- InputFiles.h ---------------------------------------------*- C++ -*-===//
|
|
//
|
|
// The LLVM Linker
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLD_WASM_INPUT_FILES_H
|
|
#define LLD_WASM_INPUT_FILES_H
|
|
|
|
#include "lld/Common/LLVM.h"
|
|
#include "llvm/ADT/DenseMap.h"
|
|
#include "llvm/ADT/DenseSet.h"
|
|
#include "llvm/Object/Archive.h"
|
|
#include "llvm/Object/Wasm.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
|
|
#include "Symbols.h"
|
|
#include "WriterUtils.h"
|
|
|
|
#include <vector>
|
|
|
|
using llvm::object::Archive;
|
|
using llvm::object::WasmObjectFile;
|
|
using llvm::object::WasmSection;
|
|
using llvm::object::WasmSymbol;
|
|
using llvm::wasm::WasmImport;
|
|
using llvm::wasm::WasmSignature;
|
|
using llvm::wasm::WasmRelocation;
|
|
|
|
namespace lld {
|
|
namespace wasm {
|
|
|
|
class InputChunk;
|
|
class InputFunction;
|
|
class InputSegment;
|
|
|
|
class InputFile {
|
|
public:
|
|
enum Kind {
|
|
ObjectKind,
|
|
ArchiveKind,
|
|
};
|
|
|
|
virtual ~InputFile() {}
|
|
|
|
// Returns the filename.
|
|
StringRef getName() const { return MB.getBufferIdentifier(); }
|
|
|
|
// Reads a file (the constructor doesn't do that).
|
|
virtual void parse() = 0;
|
|
|
|
Kind kind() const { return FileKind; }
|
|
|
|
// An archive file name if this file is created from an archive.
|
|
StringRef ParentName;
|
|
|
|
protected:
|
|
InputFile(Kind K, MemoryBufferRef M) : MB(M), FileKind(K) {}
|
|
MemoryBufferRef MB;
|
|
|
|
private:
|
|
const Kind FileKind;
|
|
};
|
|
|
|
// .a file (ar archive)
|
|
class ArchiveFile : public InputFile {
|
|
public:
|
|
explicit ArchiveFile(MemoryBufferRef M) : InputFile(ArchiveKind, M) {}
|
|
static bool classof(const InputFile *F) { return F->kind() == ArchiveKind; }
|
|
|
|
void addMember(const Archive::Symbol *Sym);
|
|
|
|
void parse() override;
|
|
|
|
private:
|
|
std::unique_ptr<Archive> File;
|
|
llvm::DenseSet<uint64_t> Seen;
|
|
};
|
|
|
|
// .o file (wasm object file)
|
|
class ObjFile : public InputFile {
|
|
public:
|
|
explicit ObjFile(MemoryBufferRef M) : InputFile(ObjectKind, M) {}
|
|
static bool classof(const InputFile *F) { return F->kind() == ObjectKind; }
|
|
|
|
void parse() override;
|
|
|
|
// Returns the underlying wasm file.
|
|
const WasmObjectFile *getWasmObj() const { return WasmObj.get(); }
|
|
|
|
void dumpInfo() const;
|
|
|
|
uint32_t relocateFunctionIndex(uint32_t Original) const;
|
|
uint32_t calcNewIndex(const WasmRelocation &Reloc) const;
|
|
uint32_t calcNewValue(const WasmRelocation &Reloc) const;
|
|
|
|
const WasmSection *CodeSection = nullptr;
|
|
const WasmSection *DataSection = nullptr;
|
|
|
|
std::vector<uint32_t> TypeMap;
|
|
std::vector<bool> TypeIsUsed;
|
|
std::vector<InputSegment *> Segments;
|
|
std::vector<InputFunction *> Functions;
|
|
|
|
ArrayRef<Symbol *> getSymbols() const { return Symbols; }
|
|
|
|
Symbol *getFunctionSymbol(uint32_t Index) const {
|
|
return FunctionSymbols[Index];
|
|
}
|
|
|
|
Symbol *getGlobalSymbol(uint32_t Index) const { return GlobalSymbols[Index]; }
|
|
|
|
private:
|
|
uint32_t relocateVirtualAddress(uint32_t Index) const;
|
|
uint32_t relocateTypeIndex(uint32_t Original) const;
|
|
uint32_t relocateGlobalIndex(uint32_t Original) const;
|
|
uint32_t relocateTableIndex(uint32_t Original) const;
|
|
|
|
Symbol *createDefined(const WasmSymbol &Sym, Symbol::Kind Kind,
|
|
InputChunk *Chunk = nullptr,
|
|
uint32_t Address = UINT32_MAX);
|
|
Symbol *createUndefined(const WasmSymbol &Sym, Symbol::Kind Kind,
|
|
const WasmSignature *Signature = nullptr);
|
|
void initializeSymbols();
|
|
InputSegment *getSegment(const WasmSymbol &WasmSym) const;
|
|
const WasmSignature *getFunctionSig(const WasmSymbol &Sym) const;
|
|
uint32_t getGlobalValue(const WasmSymbol &Sym) const;
|
|
InputFunction *getFunction(const WasmSymbol &Sym) const;
|
|
bool isExcludedByComdat(InputChunk *Chunk) const;
|
|
|
|
// List of all symbols referenced or defined by this file.
|
|
std::vector<Symbol *> Symbols;
|
|
|
|
// List of all function symbols indexed by the function index space
|
|
std::vector<Symbol *> FunctionSymbols;
|
|
|
|
// List of all global symbols indexed by the global index space
|
|
std::vector<Symbol *> GlobalSymbols;
|
|
|
|
uint32_t NumGlobalImports = 0;
|
|
uint32_t NumFunctionImports = 0;
|
|
std::unique_ptr<WasmObjectFile> WasmObj;
|
|
};
|
|
|
|
// Opens a given file.
|
|
llvm::Optional<MemoryBufferRef> readFile(StringRef Path);
|
|
|
|
} // namespace wasm
|
|
|
|
std::string toString(const wasm::InputFile *File);
|
|
|
|
} // namespace lld
|
|
|
|
#endif
|