summaryrefslogtreecommitdiff
path: root/DomainDig/IntegrationService.swift
diff options
context:
space:
mode:
Diffstat (limited to 'DomainDig/IntegrationService.swift')
-rw-r--r--DomainDig/IntegrationService.swift74
1 files changed, 49 insertions, 25 deletions
diff --git a/DomainDig/IntegrationService.swift b/DomainDig/IntegrationService.swift
index de24c23..090f564 100644
--- a/DomainDig/IntegrationService.swift
+++ b/DomainDig/IntegrationService.swift
@@ -1,5 +1,6 @@
import Foundation
import Network
+import os
import Observation
import Security
@@ -749,11 +750,17 @@ private enum SMTPClient {
try await channel.sendRaw(body + "\r\n.\r\n")
_ = try await channel.readResponse(expecting: [250])
_ = try await channel.sendCommand("QUIT", expecting: [221])
- channel.cancel()
+ await channel.cancel()
}
}
-private final class SMTPChannel {
+/// An actor, not a MainActor class. The previous shape inherited the project's
+/// MainActor default while running its receive loop on a background dispatch
+/// queue, so `parsedLines`/`lineWaiters`/`receiveBuffer` were declared
+/// main-actor-protected and mutated off it — concurrent mutation while resuming
+/// a `CheckedContinuation` can double-resume, which traps. The actor serialises
+/// all of it and forces the Network callbacks to hop in explicitly. (Issue #27.)
+private actor SMTPChannel {
private let connection: NWConnection
private var parsedLines: [String] = []
private var lineWaiters: [CheckedContinuation<String, Error>] = []
@@ -765,25 +772,35 @@ private final class SMTPChannel {
func start() async throws {
try await withCheckedThrowingContinuation { (continuation: CheckedContinuation<Void, Error>) in
- connection.stateUpdateHandler = { [weak self] state in
+ // The state handler can fire `.ready` and later `.failed` (or
+ // `.failed` twice); resuming a continuation twice traps. The lock
+ // also keeps the closure Sendable-clean without touching actor state
+ // from the connection's queue.
+ let hasResumed = OSAllocatedUnfairLock(initialState: false)
+ connection.stateUpdateHandler = { state in
+ let isFirst: () -> Bool = {
+ hasResumed.withLock { resumed in
+ if resumed { return false }
+ resumed = true
+ return true
+ }
+ }
switch state {
case .ready:
- self?.scheduleReceiveLoop()
- continuation.resume()
+ if isFirst() { continuation.resume() }
case .failed(let error):
- continuation.resume(throwing: error)
+ if isFirst() { continuation.resume(throwing: error) }
+ case .cancelled:
+ if isFirst() { continuation.resume(throwing: IntegrationError.streamClosed) }
default:
break
}
}
connection.start(queue: .global(qos: .utility))
}
- }
-
- private func scheduleReceiveLoop() {
- DispatchQueue.global(qos: .utility).async { [weak self] in
- self?.startReceiveLoop()
- }
+ // Started from the actor once the connection is ready, replacing the old
+ // dispatch-queue hop. TCP buffers anything that arrives in the gap.
+ startReceiveLoop()
}
func cancel() {
@@ -845,26 +862,33 @@ private final class SMTPChannel {
}
private func startReceiveLoop() {
+ // The completion runs on the connection's queue; hop back onto the
+ // actor before touching any state.
connection.receive(minimumIncompleteLength: 1, maximumLength: 4096) { [weak self] data, _, isComplete, error in
guard let self else { return }
-
- if let error {
- self.failWaiters(with: error)
- return
+ Task {
+ await self.handleReceive(data: data, isComplete: isComplete, error: error)
}
+ }
+ }
- if let data, !data.isEmpty {
- self.receiveBuffer.append(data)
- self.flushBuffer()
- }
+ private func handleReceive(data: Data?, isComplete: Bool, error: Error?) {
+ if let error {
+ failWaiters(with: error)
+ return
+ }
- if isComplete {
- self.failWaiters(with: IntegrationError.streamClosed)
- return
- }
+ if let data, !data.isEmpty {
+ receiveBuffer.append(data)
+ flushBuffer()
+ }
- self.startReceiveLoop()
+ if isComplete {
+ failWaiters(with: IntegrationError.streamClosed)
+ return
}
+
+ startReceiveLoop()
}
private func flushBuffer() {