57 lines
1.6 KiB
Swift
57 lines
1.6 KiB
Swift
//
|
|
// SDLPuncherActor.swift
|
|
// Tun
|
|
//
|
|
// Created by 安礼成 on 2026/1/7.
|
|
//
|
|
|
|
import Foundation
|
|
import NIOCore
|
|
|
|
actor PolicyRequesterActor {
|
|
nonisolated private let cooldown: Duration = .seconds(5)
|
|
|
|
// identityId
|
|
private var coolingDown: Set<UInt32> = []
|
|
|
|
// 处理各个请求的版本问题, map[identityId] = version
|
|
private var versions: [UInt32: UInt32] = [:]
|
|
// 处理holer
|
|
nonisolated private let querySocketAddress: SocketAddress
|
|
|
|
init(querySocketAddress: SocketAddress) {
|
|
self.querySocketAddress = querySocketAddress
|
|
}
|
|
|
|
// 提交权限请求
|
|
func submitPolicyRequest(using udpHole: SDLUDPHole?, request: inout SDLPolicyRequest) {
|
|
let identityId = request.srcIdentityID
|
|
guard let udpHole, !coolingDown.contains(identityId) else {
|
|
return
|
|
}
|
|
|
|
// 触发一次打洞
|
|
coolingDown.insert(identityId)
|
|
|
|
let version = self.versions[identityId, default: 1]
|
|
request.version = version
|
|
// 更新请求的版本问题
|
|
self.versions[identityId] = version + 1
|
|
// 发送请求
|
|
if let queryData = try? request.serializedData() {
|
|
udpHole.send(type: .policyRequest, data: queryData, remoteAddress: self.querySocketAddress)
|
|
}
|
|
|
|
Task {
|
|
// 启动冷却期
|
|
try? await Task.sleep(for: .seconds(5))
|
|
self.endCooldown(for: identityId)
|
|
}
|
|
}
|
|
|
|
private func endCooldown(for key: UInt32) {
|
|
self.coolingDown.remove(key)
|
|
}
|
|
|
|
}
|