90 lines
2.7 KiB
Swift
90 lines
2.7 KiB
Swift
//
|
|
// SDLPuncherActor.swift
|
|
// Tun
|
|
//
|
|
// Created by 安礼成 on 2026/1/7.
|
|
//
|
|
|
|
import Foundation
|
|
|
|
actor SDLPuncherActor {
|
|
// dstMac
|
|
private var coolingDown: Set<Data> = []
|
|
private let cooldown: Duration = .seconds(5)
|
|
|
|
private var superClientActor: SDLSuperClientActor?
|
|
private var udpHoleActor: SDLUDPHoleActor?
|
|
|
|
// 处理holer
|
|
private var logger: SDLLogger
|
|
|
|
struct RegisterRequest {
|
|
let srcMac: Data
|
|
let dstMac: Data
|
|
let networkId: UInt32
|
|
}
|
|
|
|
init(logger: SDLLogger) {
|
|
self.logger = logger
|
|
}
|
|
|
|
func setSuperClientActor(superClientActor: SDLSuperClientActor?) {
|
|
self.superClientActor = superClientActor
|
|
}
|
|
|
|
func setUDPHoleActor(udpHoleActor: SDLUDPHoleActor?) {
|
|
self.udpHoleActor = udpHoleActor
|
|
}
|
|
|
|
func submitRegisterRequest(request: RegisterRequest) {
|
|
let dstMac = request.dstMac
|
|
|
|
guard !coolingDown.contains(dstMac) else {
|
|
return
|
|
}
|
|
|
|
// 触发一次打洞
|
|
coolingDown.insert(dstMac)
|
|
|
|
Task {
|
|
await self.tryHole(request: request)
|
|
// 启动冷却期
|
|
try? await Task.sleep(for: .seconds(5))
|
|
self.endCooldown(for: dstMac)
|
|
}
|
|
}
|
|
|
|
private func endCooldown(for key: Data) {
|
|
self.coolingDown.remove(key)
|
|
}
|
|
|
|
private func tryHole(request: RegisterRequest) async {
|
|
var queryInfo = SDLQueryInfo()
|
|
queryInfo.dstMac = request.dstMac
|
|
guard let message = try? await self.superClientActor?.request(type: .queryInfo, data: try queryInfo.serializedData()) else {
|
|
return
|
|
}
|
|
|
|
switch message.packet {
|
|
case .empty:
|
|
self.logger.log("[SDLContext] hole query_info get empty: \(message)", level: .debug)
|
|
case .peerInfo(let peerInfo):
|
|
if let remoteAddress = peerInfo.v4Info.socketAddress() {
|
|
self.logger.log("[SDLContext] hole sock address: \(remoteAddress)", level: .debug)
|
|
// 发送register包
|
|
var register = SDLRegister()
|
|
register.networkID = request.networkId
|
|
register.srcMac = request.srcMac
|
|
register.dstMac = request.dstMac
|
|
|
|
await self.udpHoleActor?.send(type: .register, data: try! register.serializedData(), remoteAddress: remoteAddress)
|
|
} else {
|
|
self.logger.log("[SDLContext] hole sock address is invalid: \(peerInfo.v4Info)", level: .warning)
|
|
}
|
|
default:
|
|
self.logger.log("[SDLContext] hole query_info is packet: \(message)", level: .warning)
|
|
}
|
|
}
|
|
|
|
}
|