79 lines
2.1 KiB
Swift
79 lines
2.1 KiB
Swift
import Foundation
|
|
|
|
actor DNSLocalService {
|
|
private let publicDnsServers: [String]
|
|
private var onEvent: DNSEventHandler = { _ in }
|
|
|
|
private var currentClient: DNSLocalClient?
|
|
|
|
init(publicDnsServers: [String]) {
|
|
self.publicDnsServers = publicDnsServers
|
|
}
|
|
|
|
func updateEventHandler(_ onEvent: @escaping DNSEventHandler) {
|
|
self.onEvent = onEvent
|
|
}
|
|
|
|
func run() async throws {
|
|
let dnsServer = self.publicDnsServers.randomElement() ?? "223.5.5.5"
|
|
let client = DNSLocalClient(host: dnsServer)
|
|
self.currentClient = client
|
|
|
|
SDLLogger.log("[DNSLocalService] dnsLocalClient started", category: .dns)
|
|
|
|
do {
|
|
try await self.run(client: client)
|
|
self.clearCurrent(client)
|
|
await client.stop()
|
|
} catch is CancellationError {
|
|
self.clearCurrent(client)
|
|
await client.stop()
|
|
throw CancellationError()
|
|
} catch {
|
|
self.clearCurrent(client)
|
|
await client.stop()
|
|
throw error
|
|
}
|
|
}
|
|
|
|
func stop() async {
|
|
let client = self.currentClient
|
|
self.currentClient = nil
|
|
|
|
await client?.stop()
|
|
}
|
|
|
|
func query(tracker: DNSLocalClient.DNSTracker, dnsPayload: Data) async {
|
|
await self.currentClient?.query(tracker: tracker, dnsPayload: dnsPayload)
|
|
}
|
|
|
|
private func run(client: DNSLocalClient) async throws {
|
|
let onEvent = self.onEvent
|
|
|
|
try await withThrowingTaskGroup(of: Void.self) { group in
|
|
defer {
|
|
group.cancelAll()
|
|
}
|
|
|
|
group.addTask {
|
|
try await client.run()
|
|
}
|
|
|
|
group.addTask {
|
|
for try await packet in client.packetFlow {
|
|
try Task.checkCancellation()
|
|
await onEvent(.packet(packet))
|
|
}
|
|
}
|
|
|
|
_ = try await group.next()
|
|
}
|
|
}
|
|
|
|
private func clearCurrent(_ client: DNSLocalClient) {
|
|
if self.currentClient === client {
|
|
self.currentClient = nil
|
|
}
|
|
}
|
|
}
|