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

解决NE到主App的通讯问题

Browse Source
This commit is contained in:
anlicheng 2026-04-15 11:01:17 +08:00
parent 98386ded25
commit 7e2e744bdb
9 changed files with 150 additions and 75 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Tun/PacketTunnelProvider.swift
View File

@ -21,6 +21,7 @@ class PacketTunnelProvider: NEPacketTunnelProvider {
let shared = UserDefaults(suiteName: "group.com.jihe.punchnetmac")
let msg = shared?.string(forKey: "test_msg")
SDLLogger.log("NE read message: \(msg ?? "failed")", for: .debug)
SDLTunnelAppNotifier.shared.clear()
SDLNotificationCenter.shared.post(.vpnStatusChanged)

38
Tun/Punchnet/Actors/SDLContextActor.swift
View File

@ -70,7 +70,7 @@ actor SDLContextActor {
private var monitorWorker: Task<Void, Never>?
// 内部socket通讯
private var noticeClient: SDLNoticeClient?
// 改为基于 App Group + Darwin Notification 的通知
// 流量统计
nonisolated private let flowTracer = SDLFlowTracer()
@ -114,6 +114,7 @@ actor SDLContextActor {
}
self.readyState = .starting
self.prepareTunnelNotifier()
self.startMonitor()
// 启动arp的定时清理任务
@ -130,13 +131,6 @@ actor SDLContextActor {
SDLLogger.log("[SDLContext] quicClient closed!!!!")
}
await self.supervisor.addWorker(name: "noticeClient") {
let noticeClient = try self.startNoticeClient()
SDLLogger.log("[SDLContext] noticeClient running!!!!")
try await noticeClient.waitClose()
SDLLogger.log("[SDLContext] noticeClient closed!!!!")
}
await self.supervisor.addWorker(name: "udpHole") {
let udpHole = try await self.startUDPHole()
SDLLogger.log("[SDLContext] udp running!!!!")
@ -217,15 +211,12 @@ actor SDLContextActor {
return quicClient
}
private func startNoticeClient() throws -> SDLNoticeClient {
private func prepareTunnelNotifier() {
// 启动noticeClient
let noticeClient = try SDLNoticeClient(noticePort: self.config.noticePort)
noticeClient.start()
// 旧的 UDP NoticeClient 已移除，改为初始化基于 App Group 的通知通道。
SDLTunnelAppNotifier.shared.clear()
SDLLogger.log("[SDLContext] noticeClient started")
self.noticeClient = noticeClient
return noticeClient
SDLLogger.log("[SDLContext] tunnelAppNotifier ready")
}
private func startMonitor() {
@ -386,12 +377,16 @@ actor SDLContextActor {
self.updatePolicyTask?.cancel()
self.updatePolicyTask = nil
self.noticeClient?.stop()
self.noticeClient = nil
self.sessionToken = nil
self.dataCipher = nil
}
private func publishTunnelEvent(level: SDLTunnelAppNotifier.Event.Level = .error,
code: Int? = nil,
message: String) {
SDLTunnelAppNotifier.shared.publish(level: level, code: code, message: message)
}
private func setNatType(natType: SDLNATProberActor.NatType) {
self.natType = natType
}
@ -711,8 +706,7 @@ extension SDLContextActor {
switch errorCode {
case .invalidToken, .nodeDisabled:
self.superRegistrationStateMachine.handleFailure()
let alertNotice = NoticeMessage.alert(alert: errorMessage)
self.noticeClient?.send(data: alertNotice)
self.publishTunnelEvent(level: .fatal, code: Int(errorCode.rawValue), message: errorMessage)
// 报告错误并退出
let error = NSError(domain: "com.jihe.punchnet.tun", code: -1)
self.failReady(error)
@ -720,8 +714,7 @@ extension SDLContextActor {
case .noIpAddress, .networkFault, .internalFault:
self.superRegistrationStateMachine.handleRetryableNak()
let alertNotice = NoticeMessage.alert(alert: errorMessage)
self.noticeClient?.send(data: alertNotice)
self.publishTunnelEvent(level: .warning, code: Int(errorCode.rawValue), message: errorMessage)
}
SDLLogger.log("[SDLContext] Get a SuperNak message exit")
@ -742,8 +735,7 @@ extension SDLContextActor {
self.sendPeerPacket(type: .register, data: registerData, remoteAddress: remoteAddress)
}
case .shutdown(let message):
let alertNotice = NoticeMessage.alert(alert: message)
self.noticeClient?.send(data: alertNotice)
self.publishTunnelEvent(level: .fatal, message: message)
// 报告错误并退出
let error = NSError(domain: "com.jihe.punchnet.tun", code: -2)

6
Tun/Punchnet/SDLConfiguration.swift
View File

@ -52,8 +52,6 @@ public class SDLConfiguration {
let serverIp: String
let stunServers: [String]
let noticePort: Int
lazy var stunSocketAddress: SocketAddress = {
let stunServer = stunServers[0]
return try! SocketAddress.makeAddressResolvingHost(stunServer, port: 1365)
@ -84,7 +82,6 @@ public class SDLConfiguration {
clientId: String,
networkAddress: NetworkAddress,
hostname: String,
noticePort: Int,
accessToken: String,
identityId: UInt32,
exitNode: ExitNode?) {
@ -94,7 +91,6 @@ public class SDLConfiguration {
self.stunServers = stunServers
self.clientId = clientId
self.networkAddress = networkAddress
self.noticePort = noticePort
self.accessToken = accessToken
self.identityId = identityId
self.hostname = hostname
@ -110,7 +106,6 @@ extension SDLConfiguration {
guard let version = options["version"] as? Int,
let serverHost = options["server_host"] as? String,
let stunAssistHost = options["stun_assist_host"] as? String,
let noticePort = options["notice_port"] as? Int,
let accessToken = options["access_token"] as? String,
let identityId = options["identity_id"] as? UInt32,
let clientId = options["client_id"] as? String,
@ -141,7 +136,6 @@ extension SDLConfiguration {
clientId: clientId,
networkAddress: networkAddress,
hostname: hostname,
noticePort: noticePort,
accessToken: accessToken,
identityId: identityId,
exitNode: exitNode)

47
punchnet/Core/SDLTunnelAppEventStore.swift Normal file
View File

@ -0,0 +1,47 @@
//
// SDLTunnelAppEventStore.swift
// punchnet
//
// Created by 安礼成 on 2026/4/15.
//
import Foundation
struct SDLTunnelAppEventStore {
struct Event: Codable, Sendable, Identifiable {
enum Level: String, Codable, Sendable {
case info
case warning
case error
case fatal
}
let id: String
let timestamp: TimeInterval
let level: Level
let code: Int?
let message: String
}
static let appGroupSuiteName = "group.com.jihe.punchnetmac"
static let latestEventKey = "tunnel.latestEvent"
static func loadLatestEvent() -> Event? {
guard let shared = UserDefaults(suiteName: self.appGroupSuiteName),
let data = shared.data(forKey: self.latestEventKey),
let event = try? JSONDecoder().decode(Event.self, from: data) else {
return nil
}
return event
}
static func clearLatestEvent() {
guard let shared = UserDefaults(suiteName: self.appGroupSuiteName) else {
return
}
shared.removeObject(forKey: self.latestEventKey)
shared.synchronize()
}
}

2
punchnet/Core/SystemConfig.swift
View File

@ -36,7 +36,6 @@ struct SystemConfig {
accessToken: String,
identityId: UInt32,
hostname: String,
noticePort: Int,
exitNodeIp: String?) -> [String: NSObject] {
// guard let serverIp = DNSResolver.resolveAddrInfos(serverHost).first,
// let stunAssistIp = DNSResolver.resolveAddrInfos(stunAssistHost).first else {
@ -54,7 +53,6 @@ struct SystemConfig {
"server_host": serverHost as NSObject,
"stun_assist_host": stunAssistHost as NSObject,
"hostname": hostname as NSObject,
"notice_port": noticePort as NSObject,
"network_address": [
"network_id": networkId as NSObject,
"ip": ip as NSObject,

38
punchnet/Views/AppContext.swift
View File

@ -16,8 +16,6 @@ struct AppContextError: Error {
class AppContext {
private var vpnManager = VPNManager.shared
var noticePort: Int
// 调用 "/connect" 之后的网络信息
var networkContext: SDLAPIClient.NetworkContext? = nil
@ -62,8 +60,14 @@ class AppContext {
}
}
init(noticePort: Int) {
self.noticePort = noticePort
var tunnelEvent: SDLTunnelAppEventStore.Event? = nil
@ObservationIgnored
private var lastTunnelEventID: String?
init() {
self.observeTunnelEvent()
self.consumeLatestTunnelEvent()
}
func loginWith(token: String) async throws {
@ -129,7 +133,6 @@ class AppContext {
accessToken: session.accessToken,
identityId: context.identityId,
hostname: context.hostname,
noticePort: noticePort,
exitNodeIp: self.loadExitNodeIp()
)
try await self.vpnManager.enableVpn(options: options)
@ -184,4 +187,29 @@ extension AppContext {
}
}
func dismissTunnelEvent() {
self.tunnelEvent = nil
}
private func observeTunnelEvent() {
SDLNotificationCenter.shared.addObserver(for: .tunnelEventChanged) { [weak self] _ in
self?.consumeLatestTunnelEvent()
}
}
private func consumeLatestTunnelEvent() {
guard let event = SDLTunnelAppEventStore.loadLatestEvent(),
self.lastTunnelEventID != event.id else {
return
}
self.lastTunnelEventID = event.id
self.tunnelEvent = event
SDLTunnelAppEventStore.clearLatestEvent()
}
deinit {
SDLNotificationCenter.shared.removeObserver(for: .tunnelEventChanged)
}
}

18
punchnet/Views/AppRootView.swift
View File

@ -46,6 +46,13 @@ struct AppRootView: View {
.animation(.spring(duration: 0.4), value: updateManager.showUpdateOverlay)
// 关键：macOS 标准毛玻璃背景
.background(VisualEffectView(material: .hudWindow, blendingMode: .behindWindow))
.alert("提示", isPresented: tunnelEventPresented, presenting: appContext.tunnelEvent) { _ in
Button("确定", role: .cancel) {
appContext.dismissTunnelEvent()
}
} message: { event in
Text(event.message)
}
.task {
let checkUpdateResult = await updateManager.checkUpdate(isManual: false)
NSLog("[RootView] checkUpdateResult: \(checkUpdateResult)")
@ -77,4 +84,15 @@ struct AppRootView: View {
.zIndex(100) // 确保更新遮罩永远在最上层
}
private var tunnelEventPresented: Binding<Bool> {
Binding(
get: { self.appContext.tunnelEvent != nil },
set: { isPresented in
if !isPresented {
self.appContext.dismissTunnelEvent()
}
}
)
}
}

2
punchnet/Views/Login/LoginView.swift
View File

@ -292,5 +292,5 @@ struct CustomSecureField: View {
#Preview {
LoginView()
.environment(AppContext(noticePort: 0))
.environment(AppContext())
}

5
punchnet/punchnetApp.swift
View File

@ -29,7 +29,6 @@ struct punchnetApp: App {
@NSApplicationDelegateAdaptor(AppDelegate.self) var appDelegate
private var noticeServer: UDPNoticeCenterServer
@State private var appContext: AppContext
@AppStorage("hasAcceptedPrivacy") var hasAcceptedPrivacy: Bool = false
// UI 控制状态：是否显示弹窗
@ -38,9 +37,7 @@ struct punchnetApp: App {
@State private var vpnManager = VPNManager.shared
init() {
self.noticeServer = UDPNoticeCenterServer()
self.noticeServer.start()
self.appContext = AppContext(noticePort: self.noticeServer.port)
self.appContext = AppContext()
}
var body: some Scene {