blob: 86d0bb898f0dbf38876497cb2769c0d4df0b8f81 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
//
// WireguardCommand.swift
// PacketTunnel
//
// Created by pronebird on 24/06/2019.
// Copyright © 2019 Mullvad VPN AB. All rights reserved.
//
import Foundation
import Network
struct WireguardPeer: Hashable {
let endpoint: AnyIPEndpoint
let publicKey: Data
}
extension WireguardPeer {
func withReresolvedEndpoint() -> Result<WireguardPeer, Error> {
self.endpoint.withReresolvedIP()
.map { WireguardPeer(endpoint: $0, publicKey: self.publicKey) }
}
}
extension WireguardPeer {
/// Returns a 0.0.0.0/0 for IPv4 and ::0/0 for IPv6
var anyAllowedIP: IPAddressRange {
switch endpoint {
case .ipv4:
return IPAddressRange(address: IPv4Address.any, networkPrefixLength: 0)
case .ipv6:
return IPAddressRange(address: IPv6Address.any, networkPrefixLength: 0)
}
}
}
enum WireguardCommand {
case privateKey(WireguardPrivateKey)
case listenPort(UInt16)
case replacePeers
case peer(WireguardPeer)
case allowedIP(IPAddressRange)
}
extension WireguardCommand {
func toRawWireguardCommand() -> String {
switch self {
case .privateKey(let privateKey):
let keyString = privateKey.rawRepresentation.hexEncodedString()
return "private_key=\(keyString)"
case .listenPort(let port):
return "listen_port=\(port)"
case .replacePeers:
return "replace_peers=true"
case .peer(let peer):
let keyString = peer.publicKey.hexEncodedString()
let endpointString = peer.endpoint.wireguardStringRepresentation
return ["public_key=\(keyString)", "endpoint=\(endpointString)"]
.joined(separator: "\n")
case .allowedIP(let ipAddressRange):
return "allowed_ip=\(ipAddressRange)"
}
}
}
extension Array where Element == WireguardCommand {
func toRawWireguardConfigString() -> String {
map { $0.toRawWireguardCommand() }
.joined(separator: "\n")
}
}
|
