109 lines
2.9 KiB
Swift
109 lines
2.9 KiB
Swift
import Foundation
|
|
import NIOCore
|
|
|
|
enum SDLUDPHoleKind: Equatable {
|
|
case v4
|
|
case v6
|
|
|
|
func convertAddressType() -> Session.AddressType {
|
|
switch self {
|
|
case .v4:
|
|
return .v4
|
|
case .v6:
|
|
return .v6
|
|
}
|
|
}
|
|
}
|
|
|
|
actor SDLUDPHoleService {
|
|
enum Event {
|
|
case ready(SocketAddress)
|
|
case natType(SDLNATProberActor.NatType)
|
|
case packet(SocketAddress, SDLHoleControlMessage, source: SDLUDPHoleKind)
|
|
case closed(Error)
|
|
}
|
|
|
|
typealias EventHandler = @Sendable (Event) async -> Void
|
|
typealias DataHandler = @Sendable (SDLData) async -> Void
|
|
|
|
private let proberActor: SDLNATProberActor
|
|
|
|
private var onEvent: EventHandler = { _ in }
|
|
private var onData: DataHandler = { _ in }
|
|
private var currentSession: SDLUDPHoleSession?
|
|
private var generation: UInt64 = 0
|
|
|
|
init(proberActor: SDLNATProberActor) {
|
|
self.proberActor = proberActor
|
|
}
|
|
|
|
func updateHandlers(onEvent: @escaping EventHandler, onData: @escaping DataHandler) {
|
|
self.onEvent = onEvent
|
|
self.onData = onData
|
|
}
|
|
|
|
func run(includeV6: Bool = false) async throws {
|
|
let generation = self.nextGeneration()
|
|
let session = SDLUDPHoleSession(
|
|
proberActor: self.proberActor,
|
|
includeV6: includeV6,
|
|
onEvent: { [weak self] event in
|
|
await self?.handleEvent(event, generation: generation)
|
|
},
|
|
onData: self.onData
|
|
)
|
|
|
|
self.currentSession = session
|
|
|
|
do {
|
|
try await session.run()
|
|
self.clearCurrent(session, generation: generation)
|
|
} catch is CancellationError {
|
|
self.clearCurrent(session, generation: generation)
|
|
await session.stop()
|
|
throw CancellationError()
|
|
} catch {
|
|
self.clearCurrent(session, generation: generation)
|
|
await session.stop()
|
|
throw error
|
|
}
|
|
}
|
|
|
|
func stop() async {
|
|
self.generation &+= 1
|
|
|
|
let session = self.currentSession
|
|
self.currentSession = nil
|
|
|
|
await session?.stop()
|
|
await self.proberActor.cancelAll()
|
|
}
|
|
|
|
func send(type: SDLPacketType, data: Data, remoteAddress: SocketAddress) async {
|
|
await self.currentSession?.send(type: type, data: data, remoteAddress: remoteAddress)
|
|
}
|
|
|
|
private func nextGeneration() -> UInt64 {
|
|
self.generation &+= 1
|
|
return self.generation
|
|
}
|
|
|
|
private func clearCurrent(_ session: SDLUDPHoleSession, generation: UInt64) {
|
|
guard self.generation == generation else {
|
|
return
|
|
}
|
|
|
|
if self.currentSession === session {
|
|
self.currentSession = nil
|
|
}
|
|
}
|
|
|
|
private func handleEvent(_ event: Event, generation: UInt64) async {
|
|
guard self.generation == generation else {
|
|
return
|
|
}
|
|
|
|
await self.onEvent(event)
|
|
}
|
|
}
|