mirror of
https://github.com/apple/swift.git
synced 2025-12-21 12:14:44 +01:00
ec70054c93
This adjusts the runtime function declaration handling to track the owning module for the well known functions. This allows us to ensure that we are able to properly identify if the symbol should be imported or not when building the shared libraries. This will require a subsequent tweak to allow for checking for static library linkage to ensure that we do not mark the symbol as DLLImport when doing static linking.
113 lines
4.0 KiB
C++
113 lines
4.0 KiB
C++
//===--- Link.cpp - Link in transparent SILFunctions from module ----------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See https://swift.org/LICENSE.txt for license information
|
|
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "swift/SILOptimizer/PassManager/Passes.h"
|
|
#include "swift/SILOptimizer/PassManager/Transforms.h"
|
|
#include "swift/SIL/SILModule.h"
|
|
#include "swift/Serialization/SerializedSILLoader.h"
|
|
|
|
using namespace swift;
|
|
|
|
static llvm::cl::opt<bool> LinkEmbeddedRuntime("link-embedded-runtime",
|
|
llvm::cl::init(true));
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Top Level Driver
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
namespace {
|
|
|
|
/// Copies code from the standard library into the user program to enable
|
|
/// optimizations.
|
|
class SILLinker : public SILModuleTransform {
|
|
SILModule::LinkingMode LinkMode;
|
|
|
|
public:
|
|
explicit SILLinker(SILModule::LinkingMode LinkMode) : LinkMode(LinkMode) {}
|
|
|
|
void run() override {
|
|
SILModule &M = *getModule();
|
|
for (auto &Fn : M)
|
|
if (M.linkFunction(&Fn, LinkMode))
|
|
invalidateAnalysis(&Fn, SILAnalysis::InvalidationKind::Everything);
|
|
|
|
// In embedded Swift, the stdlib contains all the runtime functions needed
|
|
// (swift_retain, etc.). Link them in so they can be referenced in IRGen.
|
|
if (M.getOptions().EmbeddedSwift && LinkEmbeddedRuntime) {
|
|
linkEmbeddedRuntimeFromStdlib();
|
|
}
|
|
}
|
|
|
|
void linkEmbeddedRuntimeFromStdlib() {
|
|
using namespace RuntimeConstants;
|
|
#define FUNCTION(ID, MODULE, NAME, CC, AVAILABILITY, RETURNS, ARGS, ATTRS, \
|
|
EFFECT, MEMORY_EFFECTS) \
|
|
linkEmbeddedRuntimeFunctionByName(#NAME, EFFECT); \
|
|
if (getModule()->getASTContext().hadError()) \
|
|
return;
|
|
|
|
#define RETURNS(...)
|
|
#define ARGS(...)
|
|
#define NO_ARGS
|
|
#define ATTRS(...)
|
|
#define NO_ATTRS
|
|
#define EFFECT(...) { __VA_ARGS__ }
|
|
#define MEMORY_EFFECTS(...)
|
|
#define UNKNOWN_MEMEFFECTS
|
|
|
|
#include "swift/Runtime/RuntimeFunctions.def"
|
|
|
|
// swift_retainCount is not part of private contract between the compiler and runtime, but we still need to link it
|
|
linkEmbeddedRuntimeFunctionByName("swift_retainCount", { RefCounting });
|
|
}
|
|
|
|
void linkEmbeddedRuntimeFunctionByName(StringRef name,
|
|
ArrayRef<RuntimeEffect> effects) {
|
|
SILModule &M = *getModule();
|
|
|
|
bool allocating = false;
|
|
for (RuntimeEffect rt : effects)
|
|
if (rt == RuntimeEffect::Allocating || rt == RuntimeEffect::Deallocating)
|
|
allocating = true;
|
|
|
|
// Don't link allocating runtime functions in -no-allocations mode.
|
|
if (M.getOptions().NoAllocations && allocating) return;
|
|
|
|
// Bail if runtime function is already loaded.
|
|
if (M.lookUpFunction(name)) return;
|
|
|
|
SILFunction *Fn =
|
|
M.getSILLoader()->lookupSILFunction(name, SILLinkage::PublicExternal);
|
|
if (!Fn) return;
|
|
|
|
if (M.linkFunction(Fn, LinkMode))
|
|
invalidateAnalysis(Fn, SILAnalysis::InvalidationKind::Everything);
|
|
|
|
// Make sure that dead-function-elimination doesn't remove runtime functions.
|
|
// TODO: lazily emit runtime functions in IRGen so that we don't have to
|
|
// rely on dead-stripping in the linker to remove unused runtime
|
|
// functions.
|
|
if (Fn->isDefinition())
|
|
Fn->setLinkage(SILLinkage::Public);
|
|
}
|
|
};
|
|
} // end anonymous namespace
|
|
|
|
|
|
SILTransform *swift::createMandatorySILLinker() {
|
|
return new SILLinker(SILModule::LinkingMode::LinkNormal);
|
|
}
|
|
|
|
SILTransform *swift::createPerformanceSILLinker() {
|
|
return new SILLinker(SILModule::LinkingMode::LinkAll);
|
|
}
|