NE: Simplify DNS resolution

Signed-off-by: Roopesh Chander <roop@roopc.net>

1 files changed, 3 insertions, 17 deletions

diff --git a/WireGuard/WireGuardNetworkExtension/PacketTunnelProvider.swift b/WireGuard/WireGuardNetworkExtension/PacketTunnelProvider.swift
index 8d71f0b..c418ebc 100644
--- a/WireGuard/WireGuardNetworkExtension/PacketTunnelProvider.swift
+++ b/WireGuard/WireGuardNetworkExtension/PacketTunnelProvider.swift
@@ -58,11 +58,9 @@ class PacketTunnelProvider: NEPacketTunnelProvider {
         errorNotifier.tunnelName = tunnelName
 
         let endpoints = tunnelConfiguration.peers.map { $0.endpoint }
-        guard let resolvedEndpoints = resolveDomainNames(endpoints: endpoints) else {
-            wg_log(.error, staticMessage: "Starting tunnel failed: DNS resolution failure")
-            let dnsError = PacketTunnelProviderError.dnsResolutionFailure
-            errorNotifier.notify(dnsError)
-            startTunnelCompletionHandler(dnsError)
+        guard let resolvedEndpoints = DNSResolver.resolveSync(endpoints: endpoints) else {
+            errorNotifier.notify(PacketTunnelProviderError.dnsResolutionFailure)
+            startTunnelCompletionHandler(PacketTunnelProviderError.dnsResolutionFailure)
             return
         }
         assert(endpoints.count == resolvedEndpoints.count)
@@ -167,18 +165,6 @@ class PacketTunnelProvider: NEPacketTunnelProvider {
         }
     }
 
-    private func resolveDomainNames(endpoints: [Endpoint?]) -> [Endpoint?]? {
-        do {
-            return try DNSResolver.resolveSync(endpoints: endpoints)
-        } catch DNSResolverError.dnsResolutionFailed(let hostnames) {
-            wg_log(.error, message: "DNS resolution failed for the following hostnames: \(hostnames.joined(separator: ", "))")
-        } catch {
-            // There can be no other errors from DNSResolver.resolveSync()
-            fatalError()
-        }
-        return nil
-    }
-
     private func connect(interfaceName: String, settings: String, fileDescriptor: Int32) -> Int32 {
         return withStringsAsGoStrings(interfaceName, settings) { return wgTurnOn($0.0, $0.1, fileDescriptor) }
     }