Commit code. Update time: 2023-06-25

2023-06-25 11:50:55 +08:00
parent 2d74227878
commit 39a5f2fa92
244 changed files with 26147 additions and 0 deletions
@@ -0,0 +1,97 @@
+//
+//  Logger.swift
+//  PacketTunnel
+//
+//  Created by Jerry Bool on 2022/12/6.
+//
+
+import Foundation
+
+class Logger {
+
+    static let ENABLE_LOGGING = true
+
+    static var vpnLogFile: URL? = {
+        FileManager.default.vpnLogFile?.truncate()
+    }()
+
+    static func log(_ message: String, to: URL?) {
+        guard ENABLE_LOGGING,
+              let url = to,
+              let data = message.trimmingCharacters(in: .whitespacesAndNewlines).appending("\n").data(using: .utf8),
+              let fh = try? FileHandle(forUpdating: url)
+        else {
+            return
+        }
+
+        defer {
+            fh.closeFile()
+        }
+
+        fh.seekToEndOfFile()
+        fh.write(data)
+    }
+
+    private static var logFsObject: DispatchSourceFileSystemObject? {
+        didSet {
+            oldValue?.cancel()
+        }
+    }
+
+    private static var logText = ""
+
+    static func tailFile(_ url: URL?, _ update: ((_ logText: String) -> Void)? = nil) {
+
+        // Stop and remove the previous watched content.
+        // (Will implicitely call #stop through `didSet` hook!)
+        logFsObject = nil
+
+        guard let url = url,
+            let fh = try? FileHandle(forReadingFrom: url)
+        else {
+            return
+        }
+
+        let ui = {
+            let data = fh.readDataToEndOfFile()
+
+            if let content = String(data: data, encoding: .utf8) {
+                logText.append(content)
+
+                update?(logText)
+            }
+        }
+
+        logText = ""
+        ui()
+
+        logFsObject = DispatchSource.makeFileSystemObjectSource(
+            fileDescriptor: fh.fileDescriptor,
+            eventMask: [.extend, .delete, .link],
+            queue: .main)
+
+        logFsObject?.setEventHandler {
+            guard let data = logFsObject?.data else {
+                return
+            }
+
+            if data.contains(.delete) || data.contains(.link) {
+                DispatchQueue.main.async {
+                    tailFile(url, update)
+                }
+            }
+
+            if data.contains(.extend) {
+                ui()
+            }
+        }
+
+        logFsObject?.setCancelHandler {
+            try? fh.close()
+
+            logText = ""
+        }
+
+        logFsObject?.resume()
+    }
+}