summaryrefslogtreecommitdiffhomepage
path: root/ios/MullvadRustRuntime/MullvadConnectionModeProvider.swift
blob: 0504fb06c8714d6ee6e2ffb720d8e5f0b9126211 (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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
//
//  MullvadConnectionModeProvider.swift
//  MullvadRustRuntime
//
//  Created by Marco Nikic on 2025-02-20.
//  Copyright © 2025 Mullvad VPN AB. All rights reserved.
//

import MullvadTypes

public func initAccessMethodSettingsWrapper(methods: [PersistentAccessMethod])
    -> SwiftAccessMethodSettingsWrapper {
    // 1. Get all the built in access methods, it is expected that they are always available
    let directMethod = methods.first(where: { $0.proxyConfiguration == .direct })!
    let bridgesMethod = methods.first(where: { $0.proxyConfiguration == .bridges })!
    let encryptedDNSMethod = methods.first(where: { $0.proxyConfiguration == .encryptedDNS })!

    // 2. Get the custom access methods
    let defaultMethods: [PersistentProxyConfiguration] = [.direct, .bridges, .encryptedDNS]
    let customMethods = methods.filter { defaultMethods.contains($0.proxyConfiguration) == false }

    // 3. Convert the builtin access methods
    let directMethodRaw = convertAccessMethod(accessMethod: directMethod)
    let bridgesMethodRaw = convertAccessMethod(accessMethod: bridgesMethod)
    let encryptedDNSMethodRaw = convertAccessMethod(accessMethod: encryptedDNSMethod)

    var rawCustomMethods = ContiguousArray<UnsafeRawPointer?>([])
    // 4. Convert the custom access methods (all takes different parameters)
    for method in customMethods {
        let rawMethod = convertAccessMethod(accessMethod: method)
        rawCustomMethods.append(rawMethod)
    }

    // 5. Reunite them all in one, and pass it to rust
    return rawCustomMethods.withUnsafeMutableBufferPointer(
        {
            init_access_method_settings_wrapper(
                directMethodRaw,
                bridgesMethodRaw,
                encryptedDNSMethodRaw,
                $0.baseAddress!,
                UInt(customMethods.count)
            )
        }
    )
}

public func convertAccessMethod(accessMethod: PersistentAccessMethod) -> UnsafeMutableRawPointer? {
    switch accessMethod.proxyConfiguration {
    case .direct, .bridges, .encryptedDNS:
        return convert_builtin_access_method_setting(
            accessMethod.id.uuidString,
            accessMethod.name,
            accessMethod.isEnabled,
            accessMethod.kind(),
            nil
        )
    case let .shadowsocks(configuration):
        let serverAddress = configuration.server.rawValue.map { $0 }
        let shadowsocksConfiguration = new_shadowsocks_access_method_setting(
            serverAddress,
            UInt(serverAddress.count),
            configuration.port,
            configuration.password,
            configuration.cipher.rawValue.rawValue
        )
        let shadowsocksMethodRaw = convert_builtin_access_method_setting(
            accessMethod.id.uuidString,
            accessMethod.name,
            accessMethod.isEnabled,
            accessMethod.kind(),
            shadowsocksConfiguration
        )
        return shadowsocksMethodRaw
    case let .socks5(configuration):
        let serverAddress = configuration.server.rawValue.map { $0 }
        let socks5Configuration = new_socks5_access_method_setting(
            serverAddress,
            UInt(serverAddress.count),
            configuration.port,
            configuration.credential?.username,
            configuration.credential?.password
        )
        let socks5MethodRaw = convert_builtin_access_method_setting(
            accessMethod.id.uuidString,
            accessMethod.name,
            accessMethod.isEnabled,
            accessMethod.kind(),
            socks5Configuration
        )
        return socks5MethodRaw
    }
}

fileprivate
extension PersistentAccessMethod {
    func kind() -> UInt8 {
        switch kind {
        case .direct: UInt8(KindDirect.rawValue)
        case .bridges: UInt8(KindBridge.rawValue)
        case .encryptedDNS: UInt8(KindEncryptedDnsProxy.rawValue)
        case .shadowsocks: UInt8(KindShadowsocks.rawValue)
        case .socks5: UInt8(KindSocks5Local.rawValue)
        }
    }
}