add quic client

Tun/Punchnet/SDLMessage.swift

@@ -93,10 +93,13 @@ extension SDLStunProbeReply {
 // --MARK: 进来的消息, 这里需要采用代数类型来表示
 enum SDLHoleMessage {
     case data(SDLData)
-    case signal(SDLHoleSignal)
 }
 
-enum SDLHoleSignal {
+enum SDLQUICInboundMessage {
+    // 欢迎消息
+    case welcome(SDLWelcome)
+    
+    // 注册相关
     case registerSuperAck(SDLRegisterSuperAck)
     case registerSuperNak(SDLRegisterSuperNak)
     
@@ -105,25 +108,21 @@ enum SDLHoleSignal {
     
     case stunProbeReply(SDLStunProbeReply)
     
-    case register(SDLRegister)
-    case registerAck(SDLRegisterAck)
+    //case register(SDLRegister)
+    //case registerAck(SDLRegisterAck)
     
     case policyReponse(SDLPolicyResponse)
 }
 
 // 命令类型
 enum SDLEventType: UInt8 {
-    case dropMacs = 0x02
     case natChanged = 0x03
     case sendRegister = 0x04
-    case refreshAuth = 0x05
     case networkShutdown = 0xFF
 }
 
 enum SDLEvent {
-    case dropMacs(SDLDropMacsEvent)
     case natChanged(SDLNatChangedEvent)
     case sendRegister(SDLSendRegisterEvent)
-    case refreshAuth(SDLRefreshAuthEvent)
     case networkShutdown(SDLNetworkShutdownEvent)
 }

Tun/Punchnet/SDLQuicClient.swift

@@ -0,0 +1,193 @@
+//
+//  SDLQuicClient.swift
+//  Tun
+//
+//  Created by 安礼成 on 2026/2/13.
+//
+
+import Foundation
+import NIOCore
+import Network
+
+actor SDLQUICClient {
+    private let transport: SDLQUICTransport
+    nonisolated private let alloctor = ByteBufferAllocator()
+    
+    init(host: String, port: UInt16) {
+        self.transport = SDLQUICTransport(host: host, port: port)
+    }
+    
+    func start() async -> SDLQUICTransport.Event {
+        return await withCheckedContinuation { cont in
+            transport.start { event in
+                cont.resume(returning: event)
+            }
+        }
+    }
+    
+    func receiveStream(maxLen: Int) -> AsyncCompactMapSequence<AsyncStream<Data>, SDLQUICInboundMessage> {
+        return transport.receiveMessageStream(maxLen: maxLen).compactMap { data in
+            var buf = self.alloctor.buffer(bytes: data)
+            return try? Self.decode(buffer: &buf)
+        }
+    }
+    
+    func send(data: Data) {
+        transport.send(data)
+    }
+    
+    // --MARK: 编解码器
+    private static func decode(buffer: inout ByteBuffer) throws -> SDLQUICInboundMessage? {
+        guard let type = buffer.readInteger(as: UInt8.self),
+              let packetType = SDLPacketType(rawValue: type) else {
+            return nil
+        }
+        
+        switch packetType {
+        case .registerSuperAck:
+            guard let bytes = buffer.readBytes(length: buffer.readableBytes),
+               let registerSuperAck = try? SDLRegisterSuperAck(serializedBytes: bytes) else {
+                return nil
+            }
+            return .registerSuperAck(registerSuperAck)
+        case .registerSuperNak:
+            guard let bytes = buffer.readBytes(length: buffer.readableBytes),
+               let registerSuperNak = try? SDLRegisterSuperNak(serializedBytes: bytes) else {
+                return nil
+            }
+            return .registerSuperNak(registerSuperNak)
+        case .peerInfo:
+            guard let bytes = buffer.readBytes(length: buffer.readableBytes),
+               let peerInfo = try? SDLPeerInfo(serializedBytes: bytes) else {
+                return nil
+            }
+            return .peerInfo(peerInfo)
+        case .policyResponse:
+            guard let bytes = buffer.readBytes(length: buffer.readableBytes),
+               let policyResponse = try? SDLPolicyResponse(serializedBytes: bytes) else {
+                return nil
+            }
+            return .policyReponse(policyResponse)
+        case .event:
+            guard let eventVal = buffer.readInteger(as: UInt8.self),
+                  let event = SDLEventType(rawValue: eventVal),
+                  let bytes = buffer.readBytes(length: buffer.readableBytes) else {
+                SDLLogger.shared.log("[SDLUDPHole] decode error 15")
+                return nil
+            }
+            
+            switch event {
+            case .natChanged:
+                guard let natChangedEvent = try? SDLNatChangedEvent(serializedBytes: bytes) else {
+                    SDLLogger.shared.log("[SDLUDPHole] decode error 16")
+                    return nil
+                }
+                return .event(.natChanged(natChangedEvent))
+            case .sendRegister:
+                guard let sendRegisterEvent = try? SDLSendRegisterEvent(serializedBytes: bytes) else {
+                    SDLLogger.shared.log("[SDLUDPHole] decode error 17")
+                    return nil
+                }
+                return .event(.sendRegister(sendRegisterEvent))
+            case .networkShutdown:
+                guard let networkShutdownEvent = try? SDLNetworkShutdownEvent(serializedBytes: bytes) else {
+                    SDLLogger.shared.log("[SDLUDPHole] decode error 18")
+                    return nil
+                }
+                return .event(.networkShutdown(networkShutdownEvent))
+            }
+        default:
+            SDLLogger.shared.log("SDLUDPHole decode miss type: \(type)")
+            
+            return nil
+        }
+    }
+}
+
+final class SDLQUICTransport {
+    enum Event {
+        case ready
+        case failed(Error)
+        case cancelled
+    }
+
+    private let connection: NWConnection
+
+    init(host: String, port: UInt16) {
+        let params = NWParameters(quic: .init())
+        self.connection = NWConnection(host: .init(host), port: .init(rawValue: port)!, using: params)
+    }
+
+    func start(onEvent: @escaping (Event) -> Void) {
+        connection.stateUpdateHandler = { state in
+            switch state {
+            case .ready: onEvent(.ready)
+            case .failed(let e): onEvent(.failed(e))
+            case .cancelled: onEvent(.cancelled)
+            default: break
+            }
+        }
+        connection.start(queue: .global())
+    }
+    
+    func receiveMessageStream(maxLen: Int) -> AsyncStream<Data> {
+        let connection = self.connection
+        
+        return AsyncStream { continuation in
+            var buffer = Data()
+            
+            func tryParse() {
+                while true {
+                    // 至少要有长度
+                    guard buffer.count >= 2 else {
+                        return
+                    }
+                    
+                    let len0 = UInt16(bigEndian: buffer.withUnsafeBytes { $0.load(as: UInt16.self) })
+                    let len = Int(len0)
+                    
+                    // 数据不够一个完整包
+                    guard buffer.count >= 2 + len else {
+                        return
+                    }
+                    
+                    // 取 body
+                    let body = buffer.subdata(in: 2 ..< 2 + len)
+                    continuation.yield(body)
+                    
+                    // 移除已消费
+                    buffer.removeSubrange(0 ..< 2 + len)
+                }
+            }
+            
+            func loopReceive() {
+                connection.receive(minimumIncompleteLength: 1, maximumLength: maxLen) { data, _, _, error in
+                    if let data, !data.isEmpty {
+                        buffer.append(data)
+                        tryParse()
+                    }
+                    if error == nil {
+                        loopReceive()
+                    } else {
+                        continuation.finish()
+                    }
+                }
+            }
+            
+            loopReceive()
+        }
+    }
+
+    func send(_ data: Data) {
+        var len = UInt16(data.count).bigEndian
+        var packet = Data(Data(bytes: &len, count: 2))
+        packet.append(data)
+        
+        connection.send(content: packet, completion: .contentProcessed { _ in })
+    }
+
+    func stop() {
+        connection.cancel()
+    }
+    
+}