yggdrasil-ios/Yggdrasil Network Extension/PacketTunnelProvider.swift

import NetworkExtension
import Foundation
import Yggdrasil

class PacketTunnelProvider: NEPacketTunnelProvider {
    
    var yggdrasil: MobileYggdrasil = MobileYggdrasil()
    var yggdrasilConfig: ConfigurationProxy?
    
    func startYggdrasil() -> Error? {
        var err: Error? = nil
        
        self.setTunnelNetworkSettings(nil) { (error: Error?) -> Void in
            NSLog("Starting Yggdrasil")
            
            if let error = error {
                NSLog("Failed to clear Yggdrasil tunnel network settings: " + error.localizedDescription)
                err = error
            }
            if self.yggdrasilConfig == nil {
                NSLog("No configuration proxy!")
                return
            }
            if let config = self.yggdrasilConfig {
                NSLog("Configuration loaded")
                
                do {
                    try self.yggdrasil.startJSON(config.data())
                } catch {
                    NSLog("Starting Yggdrasil process produced an error: " + error.localizedDescription)
                    return
                }
                
                let address = self.yggdrasil.getAddressString()
                let subnet = self.yggdrasil.getSubnetString()
                
                NSLog("Yggdrasil IPv6 address: " + address)
                NSLog("Yggdrasil IPv6 subnet: " + subnet)
                
                let tunnelNetworkSettings = NEPacketTunnelNetworkSettings(tunnelRemoteAddress: address)
                tunnelNetworkSettings.ipv6Settings = NEIPv6Settings(addresses: [address], networkPrefixLengths: [7])
                tunnelNetworkSettings.ipv6Settings?.includedRoutes = [NEIPv6Route(destinationAddress: "0200::", networkPrefixLength: 7)]
                tunnelNetworkSettings.mtu = NSNumber(integerLiteral: self.yggdrasil.getMTU())
                
                NSLog("Setting tunnel network settings...")
                
                self.setTunnelNetworkSettings(tunnelNetworkSettings) { (error: Error?) -> Void in
                    NSLog("setTunnelNetworkSettings completed successfully")
                    if let error = error {
                        NSLog("Failed to set Yggdrasil tunnel network settings: " + error.localizedDescription)
                        err = error
                    } else {
                        NSLog("Yggdrasil tunnel settings set successfully")
                        
                        if let fd = self.tunnelFileDescriptor {
                            do {
                                try self.yggdrasil.takeOverTUN(fd)
                                NSLog("Yggdrasil taken over TUN successfully")
                            } catch {
                                NSLog("Taking over TUN produced an error: " + error.localizedDescription)
                                err = error
                            }
                        }
                    }
                }
            }
        }
        return err
    }
    
    override func startTunnel(options: [String : NSObject]?, completionHandler: @escaping (Error?) -> Void) {
        if let conf = (self.protocolConfiguration as! NETunnelProviderProtocol).providerConfiguration {
            if let json = conf["json"] as? Data {
                do {
                    self.yggdrasilConfig = try ConfigurationProxy(json: json)
                } catch {
                    NSLog("Error in Yggdrasil startTunnel: Configuration is invalid")
                    return
                }
                if let error = self.startYggdrasil() {
                    NSLog("Error in Yggdrasil startTunnel: " + error.localizedDescription)
                } else {
                    NSLog("Yggdrasil completion handler called")
                    completionHandler(nil)
                }
            } else {
                NSLog("Error in Yggdrasil startTunnel: No configuration JSON found")
            }
        }
    }
    
    override func stopTunnel(with reason: NEProviderStopReason, completionHandler: @escaping () -> Void) {
        try? self.yggdrasil.stop()
        super.stopTunnel(with: reason, completionHandler: completionHandler)
    }
    
    override func handleAppMessage(_ messageData: Data, completionHandler: ((Data?) -> Void)? = nil) {
        let request = String(data: messageData, encoding: .utf8)
        switch request {
        case "summary":
            let pj = self.yggdrasil.getPeersJSON()
            NSLog("JSON: \(pj)")
            var peers: [YggdrasilPeer] = []
            do {
                peers = try JSONDecoder().decode(
                    [YggdrasilPeer].self,
                    from: pj.data(using: .utf8)!
                )
            } catch {
                NSLog("JSON Error: \(error)")
            }
            let summary = YggdrasilSummary(
                address: self.yggdrasil.getAddressString(),
                subnet: self.yggdrasil.getSubnetString(),
                publicKey: self.yggdrasil.getPublicKeyString(),
                enabled: true,
                peers: peers
            )
            if let json = try? JSONEncoder().encode(summary) {
                completionHandler?(json)
            }
            
        default:
            completionHandler?(nil)
        }
    }
}