Slit lib/Linker in two.
A linker normally has two stages: symbol resolution and "moving stuff". In lib/Linker there is the complication of lazy linking some globals, but it was still far more mixed than it needed to. This splits the linker into a lower level IRMover and the linker proper. The IRMover just takes a list of globals to move and a callback that lets the user control what is lazy linked. The main motivation is that now tools/gold (and soon lld) can use their own symbol resolution to instruct IRMover what to do. llvm-svn: 255254
This commit is contained in:
parent
b949b9c01b
commit
caabe22832
78
llvm/include/llvm/Linker/IRMover.h
Normal file
78
llvm/include/llvm/Linker/IRMover.h
Normal file
@ -0,0 +1,78 @@
|
|||||||
|
//===- IRMover.h ------------------------------------------------*- C++ -*-===//
|
||||||
|
//
|
||||||
|
// The LLVM Compiler Infrastructure
|
||||||
|
//
|
||||||
|
// This file is distributed under the University of Illinois Open Source
|
||||||
|
// License. See LICENSE.TXT for details.
|
||||||
|
//
|
||||||
|
//===----------------------------------------------------------------------===//
|
||||||
|
|
||||||
|
#ifndef LLVM_LINKER_IRMOVER_H
|
||||||
|
#define LLVM_LINKER_IRMOVER_H
|
||||||
|
|
||||||
|
#include "llvm/ADT/ArrayRef.h"
|
||||||
|
#include "llvm/ADT/DenseSet.h"
|
||||||
|
#include "llvm/IR/DiagnosticInfo.h"
|
||||||
|
|
||||||
|
namespace llvm {
|
||||||
|
class GlobalValue;
|
||||||
|
class Module;
|
||||||
|
class StructType;
|
||||||
|
class Type;
|
||||||
|
|
||||||
|
class IRMover {
|
||||||
|
struct StructTypeKeyInfo {
|
||||||
|
struct KeyTy {
|
||||||
|
ArrayRef<Type *> ETypes;
|
||||||
|
bool IsPacked;
|
||||||
|
KeyTy(ArrayRef<Type *> E, bool P);
|
||||||
|
KeyTy(const StructType *ST);
|
||||||
|
bool operator==(const KeyTy &that) const;
|
||||||
|
bool operator!=(const KeyTy &that) const;
|
||||||
|
};
|
||||||
|
static StructType *getEmptyKey();
|
||||||
|
static StructType *getTombstoneKey();
|
||||||
|
static unsigned getHashValue(const KeyTy &Key);
|
||||||
|
static unsigned getHashValue(const StructType *ST);
|
||||||
|
static bool isEqual(const KeyTy &LHS, const StructType *RHS);
|
||||||
|
static bool isEqual(const StructType *LHS, const StructType *RHS);
|
||||||
|
};
|
||||||
|
|
||||||
|
public:
|
||||||
|
class IdentifiedStructTypeSet {
|
||||||
|
// The set of opaque types is the composite module.
|
||||||
|
DenseSet<StructType *> OpaqueStructTypes;
|
||||||
|
|
||||||
|
// The set of identified but non opaque structures in the composite module.
|
||||||
|
DenseSet<StructType *, StructTypeKeyInfo> NonOpaqueStructTypes;
|
||||||
|
|
||||||
|
public:
|
||||||
|
void addNonOpaque(StructType *Ty);
|
||||||
|
void switchToNonOpaque(StructType *Ty);
|
||||||
|
void addOpaque(StructType *Ty);
|
||||||
|
StructType *findNonOpaque(ArrayRef<Type *> ETypes, bool IsPacked);
|
||||||
|
bool hasType(StructType *Ty);
|
||||||
|
};
|
||||||
|
|
||||||
|
IRMover(Module &M, DiagnosticHandlerFunction DiagnosticHandler);
|
||||||
|
|
||||||
|
typedef std::function<void(GlobalValue &)> ValueAdder;
|
||||||
|
/// Move in the provide values. The source is destroyed.
|
||||||
|
/// Returns true on error.
|
||||||
|
bool move(Module &Src, ArrayRef<GlobalValue *> ValuesToLink,
|
||||||
|
std::function<void(GlobalValue &GV, ValueAdder Add)> AddLazyFor);
|
||||||
|
Module &getModule() { return Composite; }
|
||||||
|
|
||||||
|
DiagnosticHandlerFunction getDiagnosticHandler() const {
|
||||||
|
return DiagnosticHandler;
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
Module &Composite;
|
||||||
|
IdentifiedStructTypeSet IdentifiedStructTypes;
|
||||||
|
DiagnosticHandlerFunction DiagnosticHandler;
|
||||||
|
};
|
||||||
|
|
||||||
|
} // End llvm namespace
|
||||||
|
|
||||||
|
#endif
|
@ -10,11 +10,9 @@
|
|||||||
#ifndef LLVM_LINKER_LINKER_H
|
#ifndef LLVM_LINKER_LINKER_H
|
||||||
#define LLVM_LINKER_LINKER_H
|
#define LLVM_LINKER_LINKER_H
|
||||||
|
|
||||||
#include "llvm/ADT/ArrayRef.h"
|
|
||||||
#include "llvm/ADT/DenseMap.h"
|
|
||||||
#include "llvm/ADT/DenseSet.h"
|
|
||||||
#include "llvm/IR/DiagnosticInfo.h"
|
#include "llvm/IR/DiagnosticInfo.h"
|
||||||
#include "llvm/IR/FunctionInfo.h"
|
#include "llvm/IR/FunctionInfo.h"
|
||||||
|
#include "llvm/Linker/IRMover.h"
|
||||||
|
|
||||||
namespace llvm {
|
namespace llvm {
|
||||||
class Module;
|
class Module;
|
||||||
@ -26,41 +24,9 @@ class Type;
|
|||||||
/// module since it is assumed that the user of this class will want to do
|
/// module since it is assumed that the user of this class will want to do
|
||||||
/// something with it after the linking.
|
/// something with it after the linking.
|
||||||
class Linker {
|
class Linker {
|
||||||
|
IRMover Mover;
|
||||||
|
|
||||||
public:
|
public:
|
||||||
struct StructTypeKeyInfo {
|
|
||||||
struct KeyTy {
|
|
||||||
ArrayRef<Type *> ETypes;
|
|
||||||
bool IsPacked;
|
|
||||||
KeyTy(ArrayRef<Type *> E, bool P);
|
|
||||||
KeyTy(const StructType *ST);
|
|
||||||
bool operator==(const KeyTy &that) const;
|
|
||||||
bool operator!=(const KeyTy &that) const;
|
|
||||||
};
|
|
||||||
static StructType *getEmptyKey();
|
|
||||||
static StructType *getTombstoneKey();
|
|
||||||
static unsigned getHashValue(const KeyTy &Key);
|
|
||||||
static unsigned getHashValue(const StructType *ST);
|
|
||||||
static bool isEqual(const KeyTy &LHS, const StructType *RHS);
|
|
||||||
static bool isEqual(const StructType *LHS, const StructType *RHS);
|
|
||||||
};
|
|
||||||
|
|
||||||
typedef DenseSet<StructType *, StructTypeKeyInfo> NonOpaqueStructTypeSet;
|
|
||||||
typedef DenseSet<StructType *> OpaqueStructTypeSet;
|
|
||||||
|
|
||||||
struct IdentifiedStructTypeSet {
|
|
||||||
// The set of opaque types is the composite module.
|
|
||||||
OpaqueStructTypeSet OpaqueStructTypes;
|
|
||||||
|
|
||||||
// The set of identified but non opaque structures in the composite module.
|
|
||||||
NonOpaqueStructTypeSet NonOpaqueStructTypes;
|
|
||||||
|
|
||||||
void addNonOpaque(StructType *Ty);
|
|
||||||
void switchToNonOpaque(StructType *Ty);
|
|
||||||
void addOpaque(StructType *Ty);
|
|
||||||
StructType *findNonOpaque(ArrayRef<Type *> ETypes, bool IsPacked);
|
|
||||||
bool hasType(StructType *Ty);
|
|
||||||
};
|
|
||||||
|
|
||||||
enum Flags {
|
enum Flags {
|
||||||
None = 0,
|
None = 0,
|
||||||
OverrideFromSrc = (1 << 0),
|
OverrideFromSrc = (1 << 0),
|
||||||
@ -88,15 +54,8 @@ public:
|
|||||||
unsigned Flags = Flags::None);
|
unsigned Flags = Flags::None);
|
||||||
|
|
||||||
DiagnosticHandlerFunction getDiagnosticHandler() const {
|
DiagnosticHandlerFunction getDiagnosticHandler() const {
|
||||||
return DiagnosticHandler;
|
return Mover.getDiagnosticHandler();
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
|
||||||
Module &Composite;
|
|
||||||
|
|
||||||
IdentifiedStructTypeSet IdentifiedStructTypes;
|
|
||||||
|
|
||||||
DiagnosticHandlerFunction DiagnosticHandler;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
/// Create a new module with exported local functions renamed and promoted
|
/// Create a new module with exported local functions renamed and promoted
|
||||||
|
@ -1,4 +1,5 @@
|
|||||||
add_llvm_library(LLVMLinker
|
add_llvm_library(LLVMLinker
|
||||||
|
IRMover.cpp
|
||||||
LinkModules.cpp
|
LinkModules.cpp
|
||||||
|
|
||||||
ADDITIONAL_HEADER_DIRS
|
ADDITIONAL_HEADER_DIRS
|
||||||
|
1398
llvm/lib/Linker/IRMover.cpp
Normal file
1398
llvm/lib/Linker/IRMover.cpp
Normal file
File diff suppressed because it is too large
Load Diff
25
llvm/lib/Linker/LinkDiagnosticInfo.h
Normal file
25
llvm/lib/Linker/LinkDiagnosticInfo.h
Normal file
@ -0,0 +1,25 @@
|
|||||||
|
//===- LinkDiagnosticInfo.h -------------------------------------*- C++ -*-===//
|
||||||
|
//
|
||||||
|
// The LLVM Compiler Infrastructure
|
||||||
|
//
|
||||||
|
// This file is distributed under the University of Illinois Open Source
|
||||||
|
// License. See LICENSE.TXT for details.
|
||||||
|
//
|
||||||
|
//===----------------------------------------------------------------------===//
|
||||||
|
|
||||||
|
#ifndef LLVM_LIB_LINKER_LINK_DIAGNOSTIC_INFO_H
|
||||||
|
#define LLVM_LIB_LINKER_LINK_DIAGNOSTIC_INFO_H
|
||||||
|
|
||||||
|
#include "llvm/IR/DiagnosticInfo.h"
|
||||||
|
|
||||||
|
namespace llvm {
|
||||||
|
class LinkDiagnosticInfo : public DiagnosticInfo {
|
||||||
|
const Twine &Msg;
|
||||||
|
|
||||||
|
public:
|
||||||
|
LinkDiagnosticInfo(DiagnosticSeverity Severity, const Twine &Msg);
|
||||||
|
void print(DiagnosticPrinter &DP) const override;
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif
|
File diff suppressed because it is too large
Load Diff
@ -1,16 +1,37 @@
|
|||||||
; RUN: llvm-link %s %S/Inputs/alias.ll -S -o - | FileCheck %s
|
; RUN: llvm-link %s %S/Inputs/alias.ll -S -o - | FileCheck --check-prefix=C1 %s
|
||||||
; RUN: llvm-link %S/Inputs/alias.ll %s -S -o - | FileCheck %s
|
; RUN: llvm-link %S/Inputs/alias.ll %s -S -o - | FileCheck --check-prefix=C2 %s
|
||||||
|
|
||||||
|
; FIXME:
|
||||||
|
; The C1 direction is incorrect.
|
||||||
|
; When moving an alias to an existing module and we want to discard the aliasee
|
||||||
|
; (the C2 case), the IRMover knows to copy the aliasee as internal.
|
||||||
|
; When moving a replacement to an aliasee to a module that has an alias (C1),
|
||||||
|
; a replace all uses with blindly changes the alias.
|
||||||
|
; The C1 case doesn't happen when using a system linker with a plugin because
|
||||||
|
; the linker does full symbol resolution first.
|
||||||
|
; Given that this is a problem only with llvm-link and its 1 module at a time
|
||||||
|
; linking, it should probably learn to changes the aliases in the destination
|
||||||
|
; before using the IRMover.
|
||||||
|
|
||||||
@foo = weak global i32 0
|
@foo = weak global i32 0
|
||||||
; CHECK-DAG: @foo = alias i32, i32* @zed
|
; C1-DAG: @foo = alias i32, i32* @zed
|
||||||
|
; C2-DAG: @foo = alias i32, i32* @zed
|
||||||
|
|
||||||
@bar = alias i32, i32* @foo
|
@bar = alias i32, i32* @foo
|
||||||
; CHECK-DAG: @bar = alias i32, i32* @foo
|
; C1-DAG: @bar = alias i32, i32* @foo
|
||||||
|
|
||||||
|
; C2-DAG: @foo.1 = internal global i32 0
|
||||||
|
; C2-DAG: @bar = alias i32, i32* @foo.1
|
||||||
|
|
||||||
@foo2 = weak global i32 0
|
@foo2 = weak global i32 0
|
||||||
; CHECK-DAG: @foo2 = alias i16, bitcast (i32* @zed to i16*)
|
; C1-DAG: @foo2 = alias i16, bitcast (i32* @zed to i16*)
|
||||||
|
; C2-DAG: @foo2 = alias i16, bitcast (i32* @zed to i16*)
|
||||||
|
|
||||||
@bar2 = alias i32, i32* @foo2
|
@bar2 = alias i32, i32* @foo2
|
||||||
; CHECK-DAG: @bar2 = alias i32, bitcast (i16* @foo2 to i32*)
|
; C1-DAG: @bar2 = alias i32, bitcast (i16* @foo2 to i32*)
|
||||||
|
|
||||||
; CHECK-DAG: @zed = global i32 42
|
; C2-DAG: @foo2.2 = internal global i32 0
|
||||||
|
; C2-DAG: @bar2 = alias i32, i32* @foo2.2
|
||||||
|
|
||||||
|
; C1-DAG: @zed = global i32 42
|
||||||
|
; C2-DAG: @zed = global i32 42
|
||||||
|
@ -11,4 +11,4 @@ define void @foo() {
|
|||||||
ret void
|
ret void
|
||||||
}
|
}
|
||||||
|
|
||||||
; CHECK: declare void @foo(){{$}}
|
; CHECK: declare extern_weak void @foo(){{$}}
|
||||||
|
@ -29,7 +29,7 @@
|
|||||||
#include "llvm/IR/LegacyPassManager.h"
|
#include "llvm/IR/LegacyPassManager.h"
|
||||||
#include "llvm/IR/Module.h"
|
#include "llvm/IR/Module.h"
|
||||||
#include "llvm/IR/Verifier.h"
|
#include "llvm/IR/Verifier.h"
|
||||||
#include "llvm/Linker/Linker.h"
|
#include "llvm/Linker/IRMover.h"
|
||||||
#include "llvm/MC/SubtargetFeature.h"
|
#include "llvm/MC/SubtargetFeature.h"
|
||||||
#include "llvm/Object/FunctionIndexObjectFile.h"
|
#include "llvm/Object/FunctionIndexObjectFile.h"
|
||||||
#include "llvm/Object/IRObjectFile.h"
|
#include "llvm/Object/IRObjectFile.h"
|
||||||
@ -62,6 +62,17 @@ struct claimed_file {
|
|||||||
void *handle;
|
void *handle;
|
||||||
std::vector<ld_plugin_symbol> syms;
|
std::vector<ld_plugin_symbol> syms;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
struct ResolutionInfo {
|
||||||
|
bool IsLinkonceOdr = true;
|
||||||
|
bool UnnamedAddr = true;
|
||||||
|
GlobalValue::VisibilityTypes Visibility = GlobalValue::DefaultVisibility;
|
||||||
|
bool CommonInternal = false;
|
||||||
|
bool UseCommon = false;
|
||||||
|
unsigned CommonSize = 0;
|
||||||
|
unsigned CommonAlign = 0;
|
||||||
|
claimed_file *CommonFile = nullptr;
|
||||||
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
static ld_plugin_status discard_message(int level, const char *format, ...) {
|
static ld_plugin_status discard_message(int level, const char *format, ...) {
|
||||||
@ -81,6 +92,7 @@ static ld_plugin_message message = discard_message;
|
|||||||
static Reloc::Model RelocationModel = Reloc::Default;
|
static Reloc::Model RelocationModel = Reloc::Default;
|
||||||
static std::string output_name = "";
|
static std::string output_name = "";
|
||||||
static std::list<claimed_file> Modules;
|
static std::list<claimed_file> Modules;
|
||||||
|
static StringMap<ResolutionInfo> ResInfo;
|
||||||
static std::vector<std::string> Cleanup;
|
static std::vector<std::string> Cleanup;
|
||||||
static llvm::TargetOptions TargetOpts;
|
static llvm::TargetOptions TargetOpts;
|
||||||
|
|
||||||
@ -332,6 +344,18 @@ static void diagnosticHandlerForContext(const DiagnosticInfo &DI,
|
|||||||
diagnosticHandler(DI);
|
diagnosticHandler(DI);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static GlobalValue::VisibilityTypes
|
||||||
|
getMinVisibility(GlobalValue::VisibilityTypes A,
|
||||||
|
GlobalValue::VisibilityTypes B) {
|
||||||
|
if (A == GlobalValue::HiddenVisibility)
|
||||||
|
return A;
|
||||||
|
if (B == GlobalValue::HiddenVisibility)
|
||||||
|
return B;
|
||||||
|
if (A == GlobalValue::ProtectedVisibility)
|
||||||
|
return A;
|
||||||
|
return B;
|
||||||
|
}
|
||||||
|
|
||||||
/// Called by gold to see whether this file is one that our plugin can handle.
|
/// Called by gold to see whether this file is one that our plugin can handle.
|
||||||
/// We'll try to open it and register all the symbols with add_symbol if
|
/// We'll try to open it and register all the symbols with add_symbol if
|
||||||
/// possible.
|
/// possible.
|
||||||
@ -411,8 +435,22 @@ static ld_plugin_status claim_file_hook(const ld_plugin_input_file *file,
|
|||||||
|
|
||||||
const GlobalValue *GV = Obj->getSymbolGV(Sym.getRawDataRefImpl());
|
const GlobalValue *GV = Obj->getSymbolGV(Sym.getRawDataRefImpl());
|
||||||
|
|
||||||
|
ResolutionInfo &Res = ResInfo[sym.name];
|
||||||
|
|
||||||
sym.visibility = LDPV_DEFAULT;
|
sym.visibility = LDPV_DEFAULT;
|
||||||
if (GV) {
|
if (GV) {
|
||||||
|
Res.UnnamedAddr &= GV->hasUnnamedAddr();
|
||||||
|
Res.IsLinkonceOdr &= GV->hasLinkOnceLinkage();
|
||||||
|
if (GV->hasCommonLinkage()) {
|
||||||
|
Res.CommonAlign = std::max(Res.CommonAlign, GV->getAlignment());
|
||||||
|
const DataLayout &DL = GV->getParent()->getDataLayout();
|
||||||
|
uint64_t Size = DL.getTypeAllocSize(GV->getType()->getElementType());
|
||||||
|
if (Size >= Res.CommonSize) {
|
||||||
|
Res.CommonSize = Size;
|
||||||
|
Res.CommonFile = &cf;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Res.Visibility = getMinVisibility(Res.Visibility, GV->getVisibility());
|
||||||
switch (GV->getVisibility()) {
|
switch (GV->getVisibility()) {
|
||||||
case GlobalValue::DefaultVisibility:
|
case GlobalValue::DefaultVisibility:
|
||||||
sym.visibility = LDPV_DEFAULT;
|
sym.visibility = LDPV_DEFAULT;
|
||||||
@ -466,27 +504,6 @@ static ld_plugin_status claim_file_hook(const ld_plugin_input_file *file,
|
|||||||
return LDPS_OK;
|
return LDPS_OK;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void keepGlobalValue(GlobalValue &GV,
|
|
||||||
std::vector<GlobalAlias *> &KeptAliases) {
|
|
||||||
assert(!GV.hasLocalLinkage());
|
|
||||||
|
|
||||||
if (auto *GA = dyn_cast<GlobalAlias>(&GV))
|
|
||||||
KeptAliases.push_back(GA);
|
|
||||||
|
|
||||||
switch (GV.getLinkage()) {
|
|
||||||
default:
|
|
||||||
break;
|
|
||||||
case GlobalValue::LinkOnceAnyLinkage:
|
|
||||||
GV.setLinkage(GlobalValue::WeakAnyLinkage);
|
|
||||||
break;
|
|
||||||
case GlobalValue::LinkOnceODRLinkage:
|
|
||||||
GV.setLinkage(GlobalValue::WeakODRLinkage);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
assert(!GV.isDiscardableIfUnused());
|
|
||||||
}
|
|
||||||
|
|
||||||
static void internalize(GlobalValue &GV) {
|
static void internalize(GlobalValue &GV) {
|
||||||
if (GV.isDeclarationForLinker())
|
if (GV.isDeclarationForLinker())
|
||||||
return; // We get here if there is a matching asm definition.
|
return; // We get here if there is a matching asm definition.
|
||||||
@ -494,33 +511,6 @@ static void internalize(GlobalValue &GV) {
|
|||||||
GV.setLinkage(GlobalValue::InternalLinkage);
|
GV.setLinkage(GlobalValue::InternalLinkage);
|
||||||
}
|
}
|
||||||
|
|
||||||
static void drop(GlobalValue &GV) {
|
|
||||||
if (auto *F = dyn_cast<Function>(&GV)) {
|
|
||||||
F->deleteBody();
|
|
||||||
F->setComdat(nullptr); // Should deleteBody do this?
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (auto *Var = dyn_cast<GlobalVariable>(&GV)) {
|
|
||||||
Var->setInitializer(nullptr);
|
|
||||||
Var->setLinkage(
|
|
||||||
GlobalValue::ExternalLinkage); // Should setInitializer do this?
|
|
||||||
Var->setComdat(nullptr); // and this?
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
auto &Alias = cast<GlobalAlias>(GV);
|
|
||||||
Module &M = *Alias.getParent();
|
|
||||||
PointerType &Ty = *cast<PointerType>(Alias.getType());
|
|
||||||
GlobalValue::LinkageTypes L = Alias.getLinkage();
|
|
||||||
auto *Var =
|
|
||||||
new GlobalVariable(M, Ty.getElementType(), /*isConstant*/ false, L,
|
|
||||||
/*Initializer*/ nullptr);
|
|
||||||
Var->takeName(&Alias);
|
|
||||||
Alias.replaceAllUsesWith(Var);
|
|
||||||
Alias.eraseFromParent();
|
|
||||||
}
|
|
||||||
|
|
||||||
static const char *getResolutionName(ld_plugin_symbol_resolution R) {
|
static const char *getResolutionName(ld_plugin_symbol_resolution R) {
|
||||||
switch (R) {
|
switch (R) {
|
||||||
case LDPR_UNKNOWN:
|
case LDPR_UNKNOWN:
|
||||||
@ -547,58 +537,6 @@ static const char *getResolutionName(ld_plugin_symbol_resolution R) {
|
|||||||
llvm_unreachable("Unknown resolution");
|
llvm_unreachable("Unknown resolution");
|
||||||
}
|
}
|
||||||
|
|
||||||
namespace {
|
|
||||||
class LocalValueMaterializer final : public ValueMaterializer {
|
|
||||||
DenseSet<GlobalValue *> &Dropped;
|
|
||||||
DenseMap<GlobalObject *, GlobalObject *> LocalVersions;
|
|
||||||
|
|
||||||
public:
|
|
||||||
LocalValueMaterializer(DenseSet<GlobalValue *> &Dropped) : Dropped(Dropped) {}
|
|
||||||
Value *materializeDeclFor(Value *V) override;
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
Value *LocalValueMaterializer::materializeDeclFor(Value *V) {
|
|
||||||
auto *GO = dyn_cast<GlobalObject>(V);
|
|
||||||
if (!GO)
|
|
||||||
return nullptr;
|
|
||||||
|
|
||||||
auto I = LocalVersions.find(GO);
|
|
||||||
if (I != LocalVersions.end())
|
|
||||||
return I->second;
|
|
||||||
|
|
||||||
if (!Dropped.count(GO))
|
|
||||||
return nullptr;
|
|
||||||
|
|
||||||
Module &M = *GO->getParent();
|
|
||||||
GlobalValue::LinkageTypes L = GO->getLinkage();
|
|
||||||
GlobalObject *Declaration;
|
|
||||||
if (auto *F = dyn_cast<Function>(GO)) {
|
|
||||||
Declaration = Function::Create(F->getFunctionType(), L, "", &M);
|
|
||||||
} else {
|
|
||||||
auto *Var = cast<GlobalVariable>(GO);
|
|
||||||
Declaration = new GlobalVariable(M, Var->getType()->getElementType(),
|
|
||||||
Var->isConstant(), L,
|
|
||||||
/*Initializer*/ nullptr);
|
|
||||||
}
|
|
||||||
Declaration->takeName(GO);
|
|
||||||
Declaration->copyAttributesFrom(GO);
|
|
||||||
|
|
||||||
GO->setLinkage(GlobalValue::InternalLinkage);
|
|
||||||
GO->setName(Declaration->getName());
|
|
||||||
Dropped.erase(GO);
|
|
||||||
GO->replaceAllUsesWith(Declaration);
|
|
||||||
|
|
||||||
LocalVersions[Declaration] = GO;
|
|
||||||
|
|
||||||
return GO;
|
|
||||||
}
|
|
||||||
|
|
||||||
static Constant *mapConstantToLocalCopy(Constant *C, ValueToValueMapTy &VM,
|
|
||||||
LocalValueMaterializer *Materializer) {
|
|
||||||
return MapValue(C, VM, RF_IgnoreMissingEntries, nullptr, Materializer);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void freeSymName(ld_plugin_symbol &Sym) {
|
static void freeSymName(ld_plugin_symbol &Sym) {
|
||||||
free(Sym.name);
|
free(Sym.name);
|
||||||
free(Sym.comdat_key);
|
free(Sym.comdat_key);
|
||||||
@ -640,7 +578,8 @@ getFunctionIndexForFile(claimed_file &F, ld_plugin_input_file &Info) {
|
|||||||
static std::unique_ptr<Module>
|
static std::unique_ptr<Module>
|
||||||
getModuleForFile(LLVMContext &Context, claimed_file &F,
|
getModuleForFile(LLVMContext &Context, claimed_file &F,
|
||||||
ld_plugin_input_file &Info, raw_fd_ostream *ApiFile,
|
ld_plugin_input_file &Info, raw_fd_ostream *ApiFile,
|
||||||
StringSet<> &Internalize, StringSet<> &Maybe) {
|
StringSet<> &Internalize, StringSet<> &Maybe,
|
||||||
|
std::vector<GlobalValue *> &Keep) {
|
||||||
|
|
||||||
if (get_symbols(F.handle, F.syms.size(), F.syms.data()) != LDPS_OK)
|
if (get_symbols(F.handle, F.syms.size(), F.syms.data()) != LDPS_OK)
|
||||||
message(LDPL_FATAL, "Failed to get symbol information");
|
message(LDPL_FATAL, "Failed to get symbol information");
|
||||||
@ -668,11 +607,12 @@ getModuleForFile(LLVMContext &Context, claimed_file &F,
|
|||||||
SmallPtrSet<GlobalValue *, 8> Used;
|
SmallPtrSet<GlobalValue *, 8> Used;
|
||||||
collectUsedGlobalVariables(M, Used, /*CompilerUsed*/ false);
|
collectUsedGlobalVariables(M, Used, /*CompilerUsed*/ false);
|
||||||
|
|
||||||
DenseSet<GlobalValue *> Drop;
|
|
||||||
std::vector<GlobalAlias *> KeptAliases;
|
|
||||||
|
|
||||||
unsigned SymNum = 0;
|
unsigned SymNum = 0;
|
||||||
for (auto &ObjSym : Obj.symbols()) {
|
for (auto &ObjSym : Obj.symbols()) {
|
||||||
|
GlobalValue *GV = Obj.getSymbolGV(ObjSym.getRawDataRefImpl());
|
||||||
|
if (GV && GV->hasAppendingLinkage())
|
||||||
|
Keep.push_back(GV);
|
||||||
|
|
||||||
if (shouldSkip(ObjSym.getFlags()))
|
if (shouldSkip(ObjSym.getFlags()))
|
||||||
continue;
|
continue;
|
||||||
ld_plugin_symbol &Sym = F.syms[SymNum];
|
ld_plugin_symbol &Sym = F.syms[SymNum];
|
||||||
@ -684,20 +624,37 @@ getModuleForFile(LLVMContext &Context, claimed_file &F,
|
|||||||
if (options::generate_api_file)
|
if (options::generate_api_file)
|
||||||
*ApiFile << Sym.name << ' ' << getResolutionName(Resolution) << '\n';
|
*ApiFile << Sym.name << ' ' << getResolutionName(Resolution) << '\n';
|
||||||
|
|
||||||
GlobalValue *GV = Obj.getSymbolGV(ObjSym.getRawDataRefImpl());
|
|
||||||
if (!GV) {
|
if (!GV) {
|
||||||
freeSymName(Sym);
|
freeSymName(Sym);
|
||||||
continue; // Asm symbol.
|
continue; // Asm symbol.
|
||||||
}
|
}
|
||||||
|
|
||||||
if (Resolution != LDPR_PREVAILING_DEF_IRONLY && GV->hasCommonLinkage()) {
|
ResolutionInfo &Res = ResInfo[Sym.name];
|
||||||
// Common linkage is special. There is no single symbol that wins the
|
if (Resolution == LDPR_PREVAILING_DEF_IRONLY_EXP && !Res.IsLinkonceOdr)
|
||||||
// resolution. Instead we have to collect the maximum alignment and size.
|
Resolution = LDPR_PREVAILING_DEF;
|
||||||
// The IR linker does that for us if we just pass it every common GV.
|
|
||||||
// We still have to keep track of LDPR_PREVAILING_DEF_IRONLY so we
|
GV->setUnnamedAddr(Res.UnnamedAddr);
|
||||||
// internalize once the IR linker has done its job.
|
GV->setVisibility(Res.Visibility);
|
||||||
freeSymName(Sym);
|
|
||||||
continue;
|
// Override gold's resolution for common symbols. We want the largest
|
||||||
|
// one to win.
|
||||||
|
if (GV->hasCommonLinkage()) {
|
||||||
|
cast<GlobalVariable>(GV)->setAlignment(Res.CommonAlign);
|
||||||
|
if (Resolution == LDPR_PREVAILING_DEF_IRONLY)
|
||||||
|
Res.CommonInternal = true;
|
||||||
|
|
||||||
|
if (Resolution == LDPR_PREVAILING_DEF_IRONLY ||
|
||||||
|
Resolution == LDPR_PREVAILING_DEF)
|
||||||
|
Res.UseCommon = true;
|
||||||
|
|
||||||
|
if (Res.CommonFile == &F && Res.UseCommon) {
|
||||||
|
if (Res.CommonInternal)
|
||||||
|
Resolution = LDPR_PREVAILING_DEF_IRONLY;
|
||||||
|
else
|
||||||
|
Resolution = LDPR_PREVAILING_DEF;
|
||||||
|
} else {
|
||||||
|
Resolution = LDPR_PREEMPTED_IR;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
switch (Resolution) {
|
switch (Resolution) {
|
||||||
@ -707,40 +664,37 @@ getModuleForFile(LLVMContext &Context, claimed_file &F,
|
|||||||
case LDPR_RESOLVED_IR:
|
case LDPR_RESOLVED_IR:
|
||||||
case LDPR_RESOLVED_EXEC:
|
case LDPR_RESOLVED_EXEC:
|
||||||
case LDPR_RESOLVED_DYN:
|
case LDPR_RESOLVED_DYN:
|
||||||
assert(GV->isDeclarationForLinker());
|
case LDPR_PREEMPTED_IR:
|
||||||
|
case LDPR_PREEMPTED_REG:
|
||||||
break;
|
break;
|
||||||
|
|
||||||
case LDPR_UNDEF:
|
case LDPR_UNDEF:
|
||||||
if (!GV->isDeclarationForLinker()) {
|
if (!GV->isDeclarationForLinker())
|
||||||
assert(GV->hasComdat());
|
assert(GV->hasComdat());
|
||||||
Drop.insert(GV);
|
|
||||||
}
|
|
||||||
break;
|
break;
|
||||||
|
|
||||||
case LDPR_PREVAILING_DEF_IRONLY: {
|
case LDPR_PREVAILING_DEF_IRONLY: {
|
||||||
keepGlobalValue(*GV, KeptAliases);
|
Keep.push_back(GV);
|
||||||
if (!Used.count(GV)) {
|
// The IR linker has to be able to map this value to a declaration,
|
||||||
// Since we use the regular lib/Linker, we cannot just internalize GV
|
// so we can only internalize after linking.
|
||||||
// now or it will not be copied to the merged module. Instead we force
|
if (!Used.count(GV))
|
||||||
// it to be copied and then internalize it.
|
|
||||||
Internalize.insert(GV->getName());
|
Internalize.insert(GV->getName());
|
||||||
}
|
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
case LDPR_PREVAILING_DEF:
|
case LDPR_PREVAILING_DEF:
|
||||||
keepGlobalValue(*GV, KeptAliases);
|
Keep.push_back(GV);
|
||||||
break;
|
// There is a non IR use, so we have to force optimizations to keep this.
|
||||||
|
switch (GV->getLinkage()) {
|
||||||
case LDPR_PREEMPTED_IR:
|
default:
|
||||||
// Gold might have selected a linkonce_odr and preempted a weak_odr.
|
break;
|
||||||
// In that case we have to make sure we don't end up internalizing it.
|
case GlobalValue::LinkOnceAnyLinkage:
|
||||||
if (!GV->isDiscardableIfUnused())
|
GV->setLinkage(GlobalValue::WeakAnyLinkage);
|
||||||
Maybe.erase(GV->getName());
|
break;
|
||||||
|
case GlobalValue::LinkOnceODRLinkage:
|
||||||
// fall-through
|
GV->setLinkage(GlobalValue::WeakODRLinkage);
|
||||||
case LDPR_PREEMPTED_REG:
|
break;
|
||||||
Drop.insert(GV);
|
}
|
||||||
break;
|
break;
|
||||||
|
|
||||||
case LDPR_PREVAILING_DEF_IRONLY_EXP: {
|
case LDPR_PREVAILING_DEF_IRONLY_EXP: {
|
||||||
@ -748,9 +702,8 @@ getModuleForFile(LLVMContext &Context, claimed_file &F,
|
|||||||
// reason is that this GV might have a copy in another module
|
// reason is that this GV might have a copy in another module
|
||||||
// and in that module the address might be significant, but that
|
// and in that module the address might be significant, but that
|
||||||
// copy will be LDPR_PREEMPTED_IR.
|
// copy will be LDPR_PREEMPTED_IR.
|
||||||
if (GV->hasLinkOnceODRLinkage())
|
Maybe.insert(GV->getName());
|
||||||
Maybe.insert(GV->getName());
|
Keep.push_back(GV);
|
||||||
keepGlobalValue(*GV, KeptAliases);
|
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -758,19 +711,6 @@ getModuleForFile(LLVMContext &Context, claimed_file &F,
|
|||||||
freeSymName(Sym);
|
freeSymName(Sym);
|
||||||
}
|
}
|
||||||
|
|
||||||
ValueToValueMapTy VM;
|
|
||||||
LocalValueMaterializer Materializer(Drop);
|
|
||||||
for (GlobalAlias *GA : KeptAliases) {
|
|
||||||
// Gold told us to keep GA. It is possible that a GV usied in the aliasee
|
|
||||||
// expression is being dropped. If that is the case, that GV must be copied.
|
|
||||||
Constant *Aliasee = GA->getAliasee();
|
|
||||||
Constant *Replacement = mapConstantToLocalCopy(Aliasee, VM, &Materializer);
|
|
||||||
GA->setAliasee(Replacement);
|
|
||||||
}
|
|
||||||
|
|
||||||
for (auto *GV : Drop)
|
|
||||||
drop(*GV);
|
|
||||||
|
|
||||||
return Obj.takeModule();
|
return Obj.takeModule();
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -941,7 +881,7 @@ static ld_plugin_status allSymbolsReadHook(raw_fd_ostream *ApiFile) {
|
|||||||
Context.setDiagnosticHandler(diagnosticHandlerForContext, nullptr, true);
|
Context.setDiagnosticHandler(diagnosticHandlerForContext, nullptr, true);
|
||||||
|
|
||||||
std::unique_ptr<Module> Combined(new Module("ld-temp.o", Context));
|
std::unique_ptr<Module> Combined(new Module("ld-temp.o", Context));
|
||||||
Linker L(*Combined, diagnosticHandler);
|
IRMover L(*Combined, diagnosticHandler);
|
||||||
|
|
||||||
std::string DefaultTriple = sys::getDefaultTargetTriple();
|
std::string DefaultTriple = sys::getDefaultTargetTriple();
|
||||||
|
|
||||||
@ -951,15 +891,15 @@ static ld_plugin_status allSymbolsReadHook(raw_fd_ostream *ApiFile) {
|
|||||||
ld_plugin_input_file File;
|
ld_plugin_input_file File;
|
||||||
if (get_input_file(F.handle, &File) != LDPS_OK)
|
if (get_input_file(F.handle, &File) != LDPS_OK)
|
||||||
message(LDPL_FATAL, "Failed to get file information");
|
message(LDPL_FATAL, "Failed to get file information");
|
||||||
|
std::vector<GlobalValue *> Keep;
|
||||||
std::unique_ptr<Module> M =
|
std::unique_ptr<Module> M =
|
||||||
getModuleForFile(Context, F, File, ApiFile, Internalize, Maybe);
|
getModuleForFile(Context, F, File, ApiFile, Internalize, Maybe, Keep);
|
||||||
if (!options::triple.empty())
|
if (!options::triple.empty())
|
||||||
M->setTargetTriple(options::triple.c_str());
|
M->setTargetTriple(options::triple.c_str());
|
||||||
else if (M->getTargetTriple().empty()) {
|
else if (M->getTargetTriple().empty())
|
||||||
M->setTargetTriple(DefaultTriple);
|
M->setTargetTriple(DefaultTriple);
|
||||||
}
|
|
||||||
|
|
||||||
if (L.linkInModule(*M))
|
if (L.move(*M, Keep, [](GlobalValue &, IRMover::ValueAdder) {}))
|
||||||
message(LDPL_FATAL, "Failed to link module");
|
message(LDPL_FATAL, "Failed to link module");
|
||||||
if (release_input_file(F.handle) != LDPS_OK)
|
if (release_input_file(F.handle) != LDPS_OK)
|
||||||
message(LDPL_FATAL, "Failed to release file information");
|
message(LDPL_FATAL, "Failed to release file information");
|
||||||
|
Loading…
x
Reference in New Issue
Block a user