wangjia
6c19b1bd92
上一版试图让 root 扩展把 command.sock 建到用户容器,反而让 startTunnel 抛错、隧道连不上
(root 在用户 home 建 socket、属主/路径与无 root 的 app 对不上,且方向本就错)。回退该路子。
正解:用 NetworkExtension 官方跨进程通道,绕开「root 扩展容器 vs 用户 app 容器」死局——
- 扩展内新增 StatsCollector:连本进程自己的 command.sock(同容器、root 可达),订阅
status/group 缓存最新上下行 + urltest;handleAppMessage 收到 "stats" 即回最新 JSON。
- 主 app StatsClient 改为每秒 NETunnelProviderSession.sendProviderMessage("stats") 拉取、
解析后推 onStats(不再直连 socket);VpnChannel 三处启动点改传 session。
- 扩展代码变更 → CURRENT_PROJECT_VERSION 44→45,让 sysextd 重装。
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
75 lines
3.1 KiB
Swift
75 lines
3.1 KiB
Swift
// StatsClient.swift — 主 app 侧实时统计消费者（macOS）
|
||
//
|
||
// 隧道跑在 PacketTunnel System Extension 进程里（root）。扩展和主 app 的 App Group 容器
|
||
// 不是同一路径（root 的 /var/root/... vs 用户的 /Users/<user>/...），主 app 无法直接连
|
||
// 扩展的 libbox command.sock。改走 NetworkExtension 官方的跨进程通道：
|
||
// 主 app → NETunnelProviderSession.sendProviderMessage("stats")
|
||
// 扩展 → handleAppMessage 回最新统计 JSON（由扩展进程内的 StatsCollector 缓存）
|
||
//
|
||
// 这绕开了 root/用户容器与 socket 属主的死局。字段映射到 pangolin/vpn/stats 契约
|
||
// （见 lib/bridge/vpn_bridge.dart），经 onStats 交给 VpnChannel 推到 Dart 侧。
|
||
|
||
import Foundation
|
||
import NetworkExtension
|
||
|
||
final class StatsClient: NSObject {
|
||
|
||
/// 把契约字典交还 VpnChannel（由其推给当前 statsSink）。主队列回调。
|
||
var onStats: (([String: Any]) -> Void)?
|
||
|
||
private let queue = DispatchQueue(label: "pangolin.stats.client")
|
||
private weak var session: NETunnelProviderSession?
|
||
private var timer: DispatchSourceTimer?
|
||
|
||
// ── 生命周期 ────────────────────────────────────────────────
|
||
|
||
/// 隧道 connected 时调用，传入当前会话。每秒向扩展拉一次统计。
|
||
func start(session: NETunnelProviderSession) {
|
||
queue.async { [weak self] in
|
||
guard let self else { return }
|
||
self.session = session
|
||
self.timer?.cancel()
|
||
let t = DispatchSource.makeTimerSource(queue: self.queue)
|
||
t.schedule(deadline: .now() + 0.2, repeating: 1.0)
|
||
t.setEventHandler { [weak self] in self?.poll() }
|
||
t.resume()
|
||
self.timer = t
|
||
NSLog("[pangolin/stats] polling via sendProviderMessage started")
|
||
}
|
||
}
|
||
|
||
func stop() {
|
||
queue.async { [weak self] in
|
||
guard let self else { return }
|
||
self.timer?.cancel()
|
||
self.timer = nil
|
||
self.session = nil
|
||
self.emit([
|
||
"uploadBytes": 0, "downloadBytes": 0,
|
||
"uploadSpeed": 0.0, "downloadSpeed": 0.0,
|
||
"urltestResults": [[String: Any]](),
|
||
])
|
||
}
|
||
}
|
||
|
||
private func poll() {
|
||
guard let session else { return }
|
||
do {
|
||
try session.sendProviderMessage(Data("stats".utf8)) { [weak self] resp in
|
||
guard let self, let resp, !resp.isEmpty else { return }
|
||
guard let obj = (try? JSONSerialization.jsonObject(with: resp)) as? [String: Any] else {
|
||
NSLog("[pangolin/stats] bad stats payload (%d bytes)", resp.count)
|
||
return
|
||
}
|
||
self.emit(obj)
|
||
}
|
||
} catch {
|
||
NSLog("[pangolin/stats] sendProviderMessage failed: %@", error.localizedDescription)
|
||
}
|
||
}
|
||
|
||
private func emit(_ stats: [String: Any]) {
|
||
DispatchQueue.main.async { [weak self] in self?.onStats?(stats) }
|
||
}
|
||
}
|