averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-21 12:14:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #58978 from ahoppen/pr/no-deadlock-cancellation-bigger-change

Browse Source 
[SoruceKit] Audit SwiftASTManager to prevent deadlocks
This commit is contained in:
Alex Hoppen
2023-06-26 13:11:08 +02:00
committed by GitHub
parent da34a4ef88 453f6cdeee
commit 22d34d9b14
2 changed files with 33 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tools/SourceKit/lib/SwiftLang/SwiftASTManager.cpp
View File

@@ -1182,6 +1182,7 @@ void ASTBuildOperation::schedule(WorkQueue Queue) {
/*ExpectedOldState=*/State::Running);
};
SmallVector<SwiftASTConsumerRef, 4> ConsumersToCancel;
{
llvm::sys::ScopedLock L(ConsumersAndResultMtx);
if (Consumers.empty()) {
@@ -1192,24 +1193,25 @@ void ASTBuildOperation::schedule(WorkQueue Queue) {
if (CancellationFlag->load(std::memory_order_relaxed)) {
assert(false && "We should only set the cancellation flag if there "
"are no more consumers");
for (auto &Consumer : Consumers) {
ConsumersToCancel = Consumers;
}
}
for (auto &Consumer : ConsumersToCancel) {
Consumer->cancelled();
}
}
}
std::string Error;
assert(!Result && "We should only be producing a result once");
ASTUnitRef AST = buildASTUnit(Error);
SmallVector<SwiftASTConsumerRef, 4> LocalConsumers;
SmallVector<SwiftASTConsumerRef, 4> ConsumersToInform;
{
llvm::sys::ScopedLock L(ConsumersAndResultMtx);
bool WasCancelled = CancellationFlag->load(std::memory_order_relaxed);
Result.emplace(AST, Error, WasCancelled);
LocalConsumers = Consumers;
ConsumersToInform = Consumers;
Consumers = {};
}
for (auto &Consumer : LocalConsumers) {
for (auto &Consumer : ConsumersToInform) {
informConsumer(Consumer);
}
DidFinishCallback();

28
tools/SourceKit/lib/SwiftLang/SwiftASTManager.h
View File

@@ -135,8 +135,9 @@ public:
typedef IntrusiveRefCntPtr<ASTUnit> ASTUnitRef;
class SwiftASTConsumer : public std::enable_shared_from_this<SwiftASTConsumer> {
/// Mutex guarding all accesses to CancellationRequestCallback
llvm::sys::Mutex CancellationRequestCallbackMtx;
/// Mutex guarding all accesses to \c CancellationRequestCallback and \c
/// IsCancelled.
llvm::sys::Mutex CancellationRequestCallbackAndIsCancelledMtx;
/// A callback that informs the \c ASTBuildOperation, which is producing the
/// AST for this consumer, that the consumer is no longer of interest. Calling
@@ -160,12 +161,17 @@ public:
/// cause the \c ASTBuildOperation to be cancelled if no other consumer is
/// depending on it.
void requestCancellation() {
llvm::sys::ScopedLock L(CancellationRequestCallbackMtx);
Optional<std::function<void(std::shared_ptr<SwiftASTConsumer>)>>
CallbackToCall;
{
llvm::sys::ScopedLock L(CancellationRequestCallbackAndIsCancelledMtx);
IsCancelled = true;
if (CancellationRequestCallback.has_value()) {
(*CancellationRequestCallback)(shared_from_this());
CallbackToCall = CancellationRequestCallback;
CancellationRequestCallback = None;
}
if (CallbackToCall.has_value()) {
(*CallbackToCall)(shared_from_this());
}
}
/// Set a cancellation request callback that informs a \c ASTBuildOperation
@@ -177,20 +183,26 @@ public:
/// called, \c NewCallback will be called immediately.
void setCancellationRequestCallback(
std::function<void(std::shared_ptr<SwiftASTConsumer>)> NewCallback) {
llvm::sys::ScopedLock L(CancellationRequestCallbackMtx);
bool ShouldCallCallback = false;
{
llvm::sys::ScopedLock L(CancellationRequestCallbackAndIsCancelledMtx);
assert(!CancellationRequestCallback.has_value() &&
"Can't set two cancellation callbacks on a SwiftASTConsumer");
if (IsCancelled) {
NewCallback(shared_from_this());
ShouldCallCallback = true;
} else {
CancellationRequestCallback = NewCallback;
}
}
if (ShouldCallCallback) {
NewCallback(shared_from_this());
}
}
/// Removes the cancellation request callback previously set by \c
/// setCancellationRequestCallback.
void removeCancellationRequestCallback() {
llvm::sys::ScopedLock L(CancellationRequestCallbackMtx);
llvm::sys::ScopedLock L(CancellationRequestCallbackAndIsCancelledMtx);
CancellationRequestCallback = None;
}