mirror of
https://github.com/apple/swift.git
synced 2025-12-21 12:14:44 +01:00
b4dc611b26
- First, it now knows how to look through PartialApplyInsts in addition to simple ApplyInsts when finding functions to link. - Second, it now consults the external source in all cases where CalleeFunction is non-NULL rather than just if the CalleeFunction is non-NULL and empty. This allows the external source to override functions with pre-existing bodies. Swift SVN r9836
114 lines
4.1 KiB
C++
114 lines
4.1 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 - 2015 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See http://swift.org/LICENSE.txt for license information
|
|
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#define DEBUG_TYPE "link"
|
|
#include "swift/AST/Module.h"
|
|
#include "swift/Serialization/SerializedSILLoader.h"
|
|
#include "swift/SIL/SILExternalSource.h"
|
|
#include "swift/SIL/SILModule.h"
|
|
#include "swift/Subsystems.h"
|
|
#include "llvm/ADT/Statistic.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
|
|
using namespace swift;
|
|
|
|
// To help testing serialization, deserialization, we turn on sil-link-all.
|
|
static llvm::cl::opt<bool>
|
|
EnableLinkAll("sil-link-all", llvm::cl::Hidden, llvm::cl::init(false));
|
|
static llvm::cl::opt<bool>
|
|
EnableLinking("enable-sil-linking", llvm::cl::Hidden, llvm::cl::init(true));
|
|
|
|
STATISTIC(NumFuncLinked, "Number of SIL functions linked");
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Top Level Driver
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
void swift::performSILLinking(SILModule *M) {
|
|
if (!EnableLinking && !EnableLinkAll)
|
|
return;
|
|
|
|
SerializedSILLoader *SILLoader = SerializedSILLoader::create(
|
|
M->getASTContext(), M);
|
|
|
|
SILExternalSource *ExternalSource = M->getExternalSource();
|
|
|
|
SmallVector<SILFunction*, 128> Worklist;
|
|
for (auto &Fn : *M)
|
|
Worklist.push_back(&Fn);
|
|
|
|
while (!Worklist.empty()) {
|
|
auto Fn = Worklist.pop_back_val();
|
|
|
|
for (auto &BB : *Fn) {
|
|
for (auto I = BB.begin(), E = BB.end(); I != E; I++) {
|
|
SILFunction *CalleeFunction = nullptr;
|
|
bool TryLinking = false;
|
|
if (ApplyInst *AI = dyn_cast<ApplyInst>(I)) {
|
|
SILValue Callee = AI->getCallee();
|
|
// Handles FunctionRefInst only.
|
|
if (FunctionRefInst *FRI = dyn_cast<FunctionRefInst>(Callee.getDef()))
|
|
CalleeFunction = FRI->getReferencedFunction();
|
|
|
|
// When EnableLinkAll is true, we always link the Callee.
|
|
TryLinking = EnableLinkAll ? true : AI->isTransparent();
|
|
}
|
|
else if (PartialApplyInst *PAI = dyn_cast<PartialApplyInst>(I)) {
|
|
SILValue Callee = PAI->getCallee();
|
|
// Handles FunctionRefInst only.
|
|
if (FunctionRefInst *FRI = dyn_cast<FunctionRefInst>(Callee.getDef()))
|
|
CalleeFunction = FRI->getReferencedFunction();
|
|
|
|
// When EnableLinkAll is true, we always link the Callee.
|
|
TryLinking = EnableLinkAll ? true : CalleeFunction->isTransparent();
|
|
}
|
|
else if (FunctionRefInst *FRI = dyn_cast<FunctionRefInst>(I)) {
|
|
// When EnableLinkAll is true, we link the function referenced by
|
|
// FunctionRefInst.
|
|
CalleeFunction = EnableLinkAll ? FRI->getReferencedFunction() :
|
|
nullptr;
|
|
TryLinking = EnableLinkAll;
|
|
}
|
|
|
|
if (!CalleeFunction)
|
|
continue;
|
|
|
|
// The ExternalSource may wish to rewrite non-empty bodies.
|
|
if (ExternalSource) {
|
|
if (auto NewFn = ExternalSource->lookupSILFunction(CalleeFunction)) {
|
|
Worklist.push_back(NewFn);
|
|
++NumFuncLinked;
|
|
continue;
|
|
}
|
|
}
|
|
|
|
if (CalleeFunction->empty()) {
|
|
// Try to find the definition in a serialized module when callee is
|
|
// currently empty.
|
|
if (TryLinking) {
|
|
if (auto NewFn = SILLoader->lookupSILFunction(CalleeFunction)) {
|
|
Worklist.push_back(NewFn);
|
|
++NumFuncLinked;
|
|
continue;
|
|
}
|
|
}
|
|
// FIXME: Make sure the declaration has external linkage?
|
|
CalleeFunction->setLinkage(SILLinkage::External);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (EnableLinkAll)
|
|
SILLoader->getAllVTables();
|
|
}
|