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
82
83
|
//
// IPOverrideWrapper.swift
// MullvadREST
//
// Created by Jon Petersson on 2024-02-05.
// Copyright © 2025 Mullvad VPN AB. All rights reserved.
//
import MullvadSettings
import MullvadTypes
public final class IPOverrideWrapper: RelayCacheProtocol {
private let relayCache: RelayCacheProtocol
private let ipOverrideRepository: any IPOverrideRepositoryProtocol
public init(relayCache: RelayCacheProtocol, ipOverrideRepository: any IPOverrideRepositoryProtocol) {
self.relayCache = relayCache
self.ipOverrideRepository = ipOverrideRepository
}
public func read() throws -> StoredRelays {
let cache = try relayCache.read()
let relayResponse = apply(overrides: ipOverrideRepository.fetchAll(), to: cache.relays)
let rawData = try REST.Coding.makeJSONEncoder().encode(relayResponse)
return try StoredRelays(etag: cache.etag, rawData: rawData, updatedAt: cache.updatedAt)
}
public func readPrebundledRelays() throws -> StoredRelays {
let prebundledRelays = try relayCache.readPrebundledRelays()
let relayResponse = apply(overrides: ipOverrideRepository.fetchAll(), to: prebundledRelays.relays)
let rawData = try REST.Coding.makeJSONEncoder().encode(relayResponse)
return try StoredRelays(etag: prebundledRelays.etag, rawData: rawData, updatedAt: prebundledRelays.updatedAt)
}
public func write(record: StoredRelays) throws {
try relayCache.write(record: record)
}
private func apply(
overrides: [IPOverride],
to relayResponse: REST.ServerRelaysResponse
) -> REST.ServerRelaysResponse {
let wireguard = relayResponse.wireguard
let bridge = relayResponse.bridge
let overridenWireguardRelays = wireguard.relays.map { relay in
return apply(overrides: overrides, to: relay)
}
let overridenBridgeRelays = bridge.relays.map { relay in
return apply(overrides: overrides, to: relay)
}
return REST.ServerRelaysResponse(
locations: relayResponse.locations,
wireguard: REST.ServerWireguardTunnels(
ipv4Gateway: wireguard.ipv4Gateway,
ipv6Gateway: wireguard.ipv6Gateway,
portRanges: wireguard.portRanges,
relays: overridenWireguardRelays,
shadowsocksPortRanges: wireguard.shadowsocksPortRanges
),
bridge: REST.ServerBridges(
shadowsocks: bridge.shadowsocks,
relays: overridenBridgeRelays
)
)
}
private func apply<T: AnyRelay>(overrides: [IPOverride], to relay: T) -> T {
return
overrides
.first { $0.hostname == relay.hostname }
.flatMap {
relay.override(
ipv4AddrIn: $0.ipv4Address,
ipv6AddrIn: $0.ipv6Address
)
}
?? relay
}
}
|
