Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fixed a race condition in Signal terminal event handling. #267

Merged
merged 4 commits into from
Feb 18, 2017
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 14 additions & 3 deletions Sources/Signal.swift
Original file line number Diff line number Diff line change
Expand Up @@ -313,13 +313,24 @@ public final class Signal<Value, Error: Swift.Error> {
return ActionDisposable { [weak self] in
if let s = self {
s.updateLock.lock()

if case let .alive(snapshot) = s.state {
var observers = snapshot.observers
observers.remove(using: token)
s.state = .alive(AliveState(observers: observers,
retaining: observers.isEmpty ? nil : self))

// Ensure the old signal state snapshot does not deinitialize before
// `updateLock` is released. Otherwise, it might result in a
// deadlock in cases where a `Signal` legitimately receives terminal
// events recursively as a result of the deinitialization of the
// snapshot.
withExtendedLifetime(snapshot) {
Copy link
Member Author

@andersio andersio Feb 17, 2017

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is not strictly necessary, since snapshot is guaranteed to live till the end of the scope. But it serves a documentation purpose, and would be optimised away by the compiler IIRC.

s.state = .alive(AliveState(observers: observers,
retaining: observers.isEmpty ? nil : self))
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thoughts on extracting this to a local newState so we can remove the awkward formatting over two lines?

s.updateLock.unlock()
}
} else {
s.updateLock.unlock()
}
s.updateLock.unlock()
}
}
} else {
Expand Down
17 changes: 16 additions & 1 deletion Tests/ReactiveSwiftTests/ActionSpec.swift
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,7 @@ class ActionSpec: QuickSpec {
action.errors.observeValues { errors.append($0) }
action.completed.observeValues { completedCount += 1 }
}

it("should retain the state property") {
var property: MutableProperty<Bool>? = MutableProperty(false)
weak var weakProperty = property
Expand Down Expand Up @@ -114,6 +114,21 @@ class ActionSpec: QuickSpec {
expect(action.isExecuting.value) == false
}

it("should not deadlock") {
final class ViewModel {
let action2 = Action<(), (), NoError> { SignalProducer(value: ()) }
}

let action1 = Action<(), ViewModel, NoError> { SignalProducer(value: ViewModel()) }

action1.values
.flatMap(.latest) { viewModel in viewModel.action2.values.map { _ in viewModel } }
Copy link
Member Author

@andersio andersio Feb 17, 2017

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The deadlock happens as the observer disposable releases the closure { _ in viewModel } here without releasing the mapped signal's updateLock first. The deinitialization triggers the propagation of terminal event of the Action, which eventually hits the mapped signal.

.observeValues { _ in }

action1.apply().start()
action1.apply().start()
}

describe("completed") {
beforeEach {
enabled.value = true
Expand Down