1 files changed, 100 insertions, 0 deletions
diff --git a/Sources/WireGuardKit/Endpoint.swift b/Sources/WireGuardKit/Endpoint.swift
new file mode 100644
index 0000000..07529e3
--- /dev/null
+++ b/Sources/WireGuardKit/Endpoint.swift
@@ -0,0 +1,100 @@
+// SPDX-License-Identifier: MIT
+// Copyright © 2018-2019 WireGuard LLC. All Rights Reserved.
+
+import Foundation
+import Network
+
+public struct Endpoint {
+    public let host: NWEndpoint.Host
+    public let port: NWEndpoint.Port
+
+    public init(host: NWEndpoint.Host, port: NWEndpoint.Port) {
+        self.host = host
+        self.port = port
+    }
+}
+
+extension Endpoint: Equatable {
+    public static func == (lhs: Endpoint, rhs: Endpoint) -> Bool {
+        return lhs.host == rhs.host && lhs.port == rhs.port
+    }
+}
+
+extension Endpoint: Hashable {
+    public func hash(into hasher: inout Hasher) {
+        hasher.combine(host)
+        hasher.combine(port)
+    }
+}
+
+extension Endpoint {
+    public var stringRepresentation: String {
+        switch host {
+        case .name(let hostname, _):
+            return "\(hostname):\(port)"
+        case .ipv4(let address):
+            return "\(address):\(port)"
+        case .ipv6(let address):
+            return "[\(address)]:\(port)"
+        @unknown default:
+            fatalError()
+        }
+    }
+
+    public init?(from string: String) {
+        // Separation of host and port is based on 'parse_endpoint' function in
+        // https://git.zx2c4.com/wireguard-tools/tree/src/config.c
+        guard !string.isEmpty else { return nil }
+        let startOfPort: String.Index
+        let hostString: String
+        if string.first! == "[" {
+            // Look for IPv6-style endpoint, like [::1]:80
+            let startOfHost = string.index(after: string.startIndex)
+            guard let endOfHost = string.dropFirst().firstIndex(of: "]") else { return nil }
+            let afterEndOfHost = string.index(after: endOfHost)
+            guard string[afterEndOfHost] == ":" else { return nil }
+            startOfPort = string.index(after: afterEndOfHost)
+            hostString = String(string[startOfHost ..< endOfHost])
+        } else {
+            // Look for an IPv4-style endpoint, like 127.0.0.1:80
+            guard let endOfHost = string.firstIndex(of: ":") else { return nil }
+            startOfPort = string.index(after: endOfHost)
+            hostString = String(string[string.startIndex ..< endOfHost])
+        }
+        guard let endpointPort = NWEndpoint.Port(String(string[startOfPort ..< string.endIndex])) else { return nil }
+        let invalidCharacterIndex = hostString.unicodeScalars.firstIndex { char in
+            return !CharacterSet.urlHostAllowed.contains(char)
+        }
+        guard invalidCharacterIndex == nil else { return nil }
+        host = NWEndpoint.Host(hostString)
+        port = endpointPort
+    }
+}
+
+extension Endpoint {
+    public func hasHostAsIPAddress() -> Bool {
+        switch host {
+        case .name:
+            return false
+        case .ipv4:
+            return true
+        case .ipv6:
+            return true
+        @unknown default:
+            fatalError()
+        }
+    }
+
+    public func hostname() -> String? {
+        switch host {
+        case .name(let hostname, _):
+            return hostname
+        case .ipv4:
+            return nil
+        case .ipv6:
+            return nil
+        @unknown default:
+            fatalError()
+        }
+    }
+}