Various tweaks

Yggdrasil Network Cross-Platform/ConfigurationProxy.swift

@@ -0,0 +1,252 @@
+//
+//  ConfigurationProxy.swift
+//  YggdrasilNetwork
+//
+//  Created by Neil Alexander on 07/01/2019.
+//
+
+#if canImport(UIKit)
+import UIKit
+#elseif canImport(AppKit)
+import AppKit
+#endif
+import Yggdrasil
+import NetworkExtension
+
+#if os(iOS)
+class PlatformItemSource: NSObject, UIActivityItemSource {
+    func activityViewControllerPlaceholderItem(_ activityViewController: UIActivityViewController) -> Any {
+        return "yggdrasil.conf"
+    }
+    
+    func activityViewController(_ activityViewController: UIActivityViewController, itemForActivityType activityType: UIActivity.ActivityType?) -> Any? {
+        return nil
+    }
+}
+#elseif os(OSX)
+class PlatformItemSource: NSObject {}
+#endif
+
+class ConfigurationProxy: PlatformItemSource {
+
+    private var json: Data? = nil
+    private var dict: [String: Any]? = nil
+    
+    override init() {
+        super.init()
+        self.json = MobileGenerateConfigJSON()
+        do {
+            try self.convertToDict()
+        } catch {
+            NSLog("ConfigurationProxy: Error deserialising JSON (\(error))")
+        }
+        #if os(iOS)
+        self.set("name", inSection: "NodeInfo", to: UIDevice.current.name)
+        #elseif os(OSX)
+        self.set("name", inSection: "NodeInfo", to: Host.current().localizedName ?? "")
+        #endif
+        self.fix()
+    }
+    
+    init(json: Data) throws {
+        super.init()
+        self.json = json
+        try self.convertToDict()
+        self.fix()
+    }
+    
+    private func fix() {
+        self.set("Listen", to: [] as [String])
+        self.set("AdminListen", to: "none")
+        self.set("IfName", to: "dummy")
+        
+        if self.get("AutoStart") == nil {
+            self.set("AutoStart", to: ["WiFi": false, "Mobile": false] as [String: Bool])
+        }
+        
+        let multicastInterfaces = self.get("MulticastInterfaces") as? [[String: Any]] ?? []
+        if multicastInterfaces.count == 0 {
+            self.set("MulticastInterfaces", to: [
+                [
+                    "Regex": "en.*",
+                    "Beacon": true,
+                    "Listen": true,
+                ]
+            ])
+        }
+    }
+    
+    public var multicastBeacons: Bool {
+        get {
+            let multicastInterfaces = self.get("MulticastInterfaces") as? [[String: Any]] ?? []
+            if multicastInterfaces.count == 0 {
+                return false
+            }
+            return multicastInterfaces[0]["Beacon"] as? Bool ?? true
+        }
+        set {
+            var multicastInterfaces = self.get("MulticastInterfaces") as? [[String: Any]] ?? []
+            multicastInterfaces[0]["Beacon"] = newValue
+            self.set("MulticastInterfaces", to: multicastInterfaces)
+        }
+    }
+    
+    public var multicastListen: Bool {
+        get {
+            let multicastInterfaces = self.get("MulticastInterfaces") as? [[String: Any]] ?? []
+            if multicastInterfaces.count == 0 {
+                return false
+            }
+            return multicastInterfaces[0]["Listen"] as? Bool ?? true
+        }
+        set {
+            var multicastInterfaces = self.get("MulticastInterfaces") as? [[String: Any]] ?? []
+            multicastInterfaces[0]["Listen"] = newValue
+            self.set("MulticastInterfaces", to: multicastInterfaces)
+        }
+    }
+    
+    func get(_ key: String) -> Any? {
+        if let dict = self.dict {
+            if dict.keys.contains(key) {
+                return dict[key]
+            }
+        }
+        return nil
+    }
+    
+    func get(_ key: String, inSection section: String) -> Any? {
+        if let dict = self.get(section) as? [String: Any] {
+            if dict.keys.contains(key) {
+                return dict[key]
+            }
+        }
+        return nil
+    }
+    
+    func add(_ value: Any, in key: String) {
+        if self.dict != nil {
+            if self.dict![key] as? [Any] != nil {
+                var temp = self.dict![key] as? [Any] ?? []
+                temp.append(value)
+                self.dict!.updateValue(temp, forKey: key)
+            }
+        }
+    }
+    
+    func remove(_ value: String, from key: String) {
+        if self.dict != nil {
+            if self.dict![key] as? [String] != nil {
+                var temp = self.dict![key] as? [String] ?? []
+                if let index = temp.firstIndex(of: value) {
+                    temp.remove(at: index)
+                }
+                self.dict!.updateValue(temp, forKey: key)
+            }
+        }
+    }
+    
+    func remove(index: Int, from key: String) {
+        if self.dict != nil {
+            if self.dict![key] as? [Any] != nil {
+                var temp = self.dict![key] as? [Any] ?? []
+                temp.remove(at: index)
+                self.dict!.updateValue(temp, forKey: key)
+            }
+        }
+    }
+    
+    func set(_ key: String, to value: Any) {
+        if self.dict != nil {
+            self.dict![key] = value
+        }
+    }
+    
+    func set(_ key: String, inSection section: String, to value: Any?) {
+        if self.dict != nil {
+            if self.dict!.keys.contains(section), let value = value {
+                var temp = self.dict![section] as? [String: Any] ?? [:]
+                temp.updateValue(value, forKey: key)
+                self.dict!.updateValue(temp, forKey: section)
+            }
+        }
+    }
+    
+    func data() -> Data? {
+        do {
+            try self.convertToJson()
+            return self.json
+        } catch {
+            return nil
+        }
+    }
+    
+    func save(to manager: inout NETunnelProviderManager) throws {
+        self.fix()
+        if let data = self.data() {
+            let providerProtocol = NETunnelProviderProtocol()
+            #if os(iOS)
+            providerProtocol.providerBundleIdentifier = "eu.neilalexander.yggdrasil.extension"
+            #elseif os(OSX)
+            providerProtocol.providerBundleIdentifier = "eu.neilalexander.yggdrasilmac.extension"
+            #endif
+            providerProtocol.providerConfiguration = [ "json": data ]
+            providerProtocol.serverAddress = "yggdrasil"
+            providerProtocol.username = self.get("PublicKey") as? String ?? self.get("SigningPublicKey") as? String ?? "(unknown public key)"
+            
+            let disconnectrule = NEOnDemandRuleDisconnect()
+            var rules: [NEOnDemandRule] = [disconnectrule]
+            if self.get("WiFi", inSection: "AutoStart") as? Bool ?? false {
+                let wifirule = NEOnDemandRuleConnect()
+                wifirule.interfaceTypeMatch = .wiFi
+                rules.insert(wifirule, at: 0)
+            }
+            #if canImport(UIKit)
+            if self.get("Mobile", inSection: "AutoStart") as? Bool ?? false {
+                let mobilerule = NEOnDemandRuleConnect()
+                mobilerule.interfaceTypeMatch = .cellular
+                rules.insert(mobilerule, at: 0)
+            }
+            #endif
+            manager.onDemandRules = rules
+            manager.isOnDemandEnabled = rules.count > 1
+            providerProtocol.disconnectOnSleep = rules.count > 1
+            
+            manager.protocolConfiguration = providerProtocol
+            
+            manager.saveToPreferences(completionHandler: { (error:Error?) in
+                if let error = error {
+                    print(error)
+                } else {
+                    print("Save successfully")
+                    NotificationCenter.default.post(name: NSNotification.Name.YggdrasilSettingsUpdated, object: self)
+                }
+            })
+        }
+    }
+    
+    private func convertToDict() throws {
+        self.dict = try JSONSerialization.jsonObject(with: self.json!, options: []) as? [String: Any]
+    }
+    
+    private func convertToJson() throws {
+        self.json = try JSONSerialization.data(withJSONObject: self.dict as Any, options: .prettyPrinted)
+    }
+
+    #if canImport(UIKit)
+    override func activityViewControllerPlaceholderItem(_ activityViewController: UIActivityViewController) -> Any {
+        return "yggdrasil.conf"
+    }
+    
+    override func activityViewController(_ activityViewController: UIActivityViewController, itemForActivityType activityType: UIActivity.ActivityType?) -> Any? {
+        return self.data()
+    }
+    
+    func activityViewController(_ activityViewController: UIActivityViewController, subjectForActivityType activityType: UIActivity.ActivityType?) -> String {
+        if let pubkey = self.get("PublicKey") as? String {
+            return "yggdrasil-\(pubkey).conf.json"
+        }
+        return "yggdrasil.conf.json"
+    }
+    #endif
+}