anlicheng/punchnet-macos
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix superClient

Browse Source
This commit is contained in:
anlicheng 2026-05-22 12:27:08 +08:00
parent 5e8b403fbf
commit 4d9f6b9a42
3 changed files with 25 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
Tun/Outbound/PacketOutboundActor.swift
View File

@ -109,7 +109,7 @@ actor PacketOutboundActor {
}
}
func stop() async {
func stop() {
self.packetReaderGeneration &+= 1
let packetReaderTask = self.packetReaderTask
@ -125,10 +125,6 @@ actor PacketOutboundActor {
self.policyService.recordOutboundFlow(ipPacket: packet)
}
await self.handleTunRouteDecision(decision)
}
private func handleTunRouteDecision(_ decision: PacketOutboundRouter.RouteDecision) async {
switch decision {
case .loopback(let ipPacketData):
let nePacket = NEPacket(data: ipPacketData, protocolFamily: 2)
@ -174,12 +170,12 @@ actor PacketOutboundActor {
case .superNode(let payload):
await self.sendSuperPacket(type: .data, data: payload)
case .peer(let payload, let session):
SDLLogger.log("[PacketOutboundActor] step 5 send packet by session: \(session)", for: .trace)
SDLLogger.log("[PacketOutboundActor] send packet by session: \(session)", for: .trace)
await self.sendPeerPacket(type: .data, data: payload, remoteAddress: session.natAddress)
self.flowTracer.inc(num: payload.count, type: .p2p)
case .superNodeAndPunch(let payload, let request):
await self.sendSuperPacket(type: .data, data: payload)
SDLLogger.log("[PacketOutboundActor] step 5 send packet by super: \(self.stunSocketAddress)", for: .trace)
SDLLogger.log("[PacketOutboundActor] send packet by super: \(self.stunSocketAddress)", for: .trace)
self.flowTracer.inc(num: payload.count, type: .forward)
if let queryData = await self.puncherActor.makeQueryInfoRequest(request: request) {

16
Tun/Super/SDLSuperClient.swift
View File

@ -145,6 +145,7 @@ actor SDLSuperClient {
let data = try await self.readOnce()
let frames = try self.frameParser.parseFrames(data: data)
for frame in frames {
try Task.checkCancellation()
if let message = SDLQUICCodec.decode(frame: frame) {
self.messageCont.yield(message)
} else {
@ -183,20 +184,27 @@ actor SDLSuperClient {
throw SDLQUICError.connectionCancelled
}
return try await withCheckedThrowingContinuation { cont in
let readContinuation = OnceContinuation<Data, Error>()
return try await withTaskCancellationHandler {
try await withCheckedThrowingContinuation { cont in
readContinuation.set(cont)
connection.receive(minimumIncompleteLength: 1, maximumLength: 64 * 1024) { data, _, isComplete, error in
if let error {
cont.resume(throwing: error)
readContinuation.resume(throwing: error)
return
}
if isComplete {
cont.resume(throwing: SDLQUICError.dataStreamClosed)
readContinuation.resume(throwing: SDLQUICError.dataStreamClosed)
} else {
cont.resume(returning: data ?? Data())
readContinuation.resume(returning: data ?? Data())
}
}
}
} onCancel: {
readContinuation.resume(throwing: CancellationError())
}
}
func stop() {

1
Tun/Super/SDLSuperService.swift
View File

@ -61,6 +61,7 @@ actor SDLSuperService {
await superClient.stop()
}
}
await self.cleanup(superClient)
} catch {
await self.cleanup(superClient)