Compare commits

..

1 Commits

Author SHA1 Message Date
Andrej Mihajlov 977ec25e79 Kit: implement Codable for IPAddressRange
Signed-off-by: Andrej Mihajlov <and@mullvad.net>
2022-09-26 16:43:17 +02:00
2 changed files with 26 additions and 15 deletions
+23
View File
@@ -27,6 +27,29 @@ extension IPAddressRange: Hashable {
} }
} }
extension IPAddressRange: Codable {
public func encode(to encoder: Encoder) throws {
var container = encoder.singleValueContainer()
try container.encode(self.stringRepresentation)
}
public init(from decoder: Decoder) throws {
let container = try decoder.singleValueContainer()
let value = try container.decode(String.self)
if let ipAddressRange = IPAddressRange(from: value) {
self = ipAddressRange
} else {
let context = DecodingError.Context(
codingPath: container.codingPath,
debugDescription: "Invalid IPAddressRange representation"
)
throw DecodingError.dataCorrupted(context)
}
}
}
extension IPAddressRange { extension IPAddressRange {
public var stringRepresentation: String { public var stringRepresentation: String {
return "\(address)/\(networkPrefixLength)" return "\(address)/\(networkPrefixLength)"
+3 -15
View File
@@ -56,10 +56,6 @@ public class WireGuardAdapter {
/// Adapter state. /// Adapter state.
private var state: State = .stopped private var state: State = .stopped
/// Whether adapter should automatically raise the `reasserting` flag when updating
/// tunnel configuration.
private let shouldHandleReasserting: Bool
/// Tunnel device file descriptor. /// Tunnel device file descriptor.
private var tunnelFileDescriptor: Int32? { private var tunnelFileDescriptor: Int32? {
var ctlInfo = ctl_info() var ctlInfo = ctl_info()
@@ -132,12 +128,9 @@ public class WireGuardAdapter {
/// Designated initializer. /// Designated initializer.
/// - Parameter packetTunnelProvider: an instance of `NEPacketTunnelProvider`. Internally stored /// - Parameter packetTunnelProvider: an instance of `NEPacketTunnelProvider`. Internally stored
/// as a weak reference. /// as a weak reference.
/// - Parameter shouldHandleReasserting: whether adapter should automatically raise the
/// `reasserting` flag when updating tunnel configuration.
/// - Parameter logHandler: a log handler closure. /// - Parameter logHandler: a log handler closure.
public init(with packetTunnelProvider: NEPacketTunnelProvider, shouldHandleReasserting: Bool = true, logHandler: @escaping LogHandler) { public init(with packetTunnelProvider: NEPacketTunnelProvider, logHandler: @escaping LogHandler) {
self.packetTunnelProvider = packetTunnelProvider self.packetTunnelProvider = packetTunnelProvider
self.shouldHandleReasserting = shouldHandleReasserting
self.logHandler = logHandler self.logHandler = logHandler
setupLogHandler() setupLogHandler()
@@ -255,14 +248,9 @@ public class WireGuardAdapter {
// Tell the system that the tunnel is going to reconnect using new WireGuard // Tell the system that the tunnel is going to reconnect using new WireGuard
// configuration. // configuration.
// This will broadcast the `NEVPNStatusDidChange` notification to the GUI process. // This will broadcast the `NEVPNStatusDidChange` notification to the GUI process.
if self.shouldHandleReasserting { self.packetTunnelProvider?.reasserting = true
self.packetTunnelProvider?.reasserting = true
}
defer { defer {
if self.shouldHandleReasserting { self.packetTunnelProvider?.reasserting = false
self.packetTunnelProvider?.reasserting = false
}
} }
do { do {