blob: e7b92c34b2ef774dad079e946c14a793d9e93ad2 (
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
|
//
// URLRequestProxy.swift
// PacketTunnel
//
// Created by pronebird on 03/02/2023.
// Copyright © 2023 Mullvad VPN AB. All rights reserved.
//
import Foundation
import MullvadREST
import MullvadTransport
import MullvadTypes
public final class URLRequestProxy {
/// Serial queue used for synchronizing access to class members.
private let dispatchQueue: DispatchQueue
private let transportProvider: RESTTransportProvider
/// List of all proxied network requests bypassing VPN.
private var proxiedRequests: [UUID: Cancellable] = [:]
public init(
dispatchQueue: DispatchQueue,
transportProvider: RESTTransportProvider
) {
self.dispatchQueue = dispatchQueue
self.transportProvider = transportProvider
}
public func sendRequest(
_ proxyRequest: ProxyURLRequest,
completionHandler: @escaping (ProxyURLResponse) -> Void
) {
dispatchQueue.async {
guard let transportProvider = self.transportProvider.makeTransport() else { return }
// The task sent by `transport.sendRequest` comes in an already resumed state
let task = transportProvider.sendRequest(proxyRequest.urlRequest) { [weak self] data, response, error in
guard let self else { return }
// However there is no guarantee about which queue the execution resumes on
// Use `dispatchQueue` to guarantee thread safe access to `proxiedRequests`
dispatchQueue.async {
let response = ProxyURLResponse(data: data, response: response, error: error)
_ = self.removeRequest(identifier: proxyRequest.id)
completionHandler(response)
}
}
// All tasks should have unique identifiers, but if not, cancel the task scheduled
// earlier.
let oldTask = self.addRequest(identifier: proxyRequest.id, task: task)
oldTask?.cancel()
}
}
public func cancelRequest(identifier: UUID) {
dispatchQueue.async {
let task = self.removeRequest(identifier: identifier)
task?.cancel()
}
}
private func addRequest(identifier: UUID, task: Cancellable) -> Cancellable? {
dispatchPrecondition(condition: .onQueue(dispatchQueue))
return proxiedRequests.updateValue(task, forKey: identifier)
}
private func removeRequest(identifier: UUID) -> Cancellable? {
dispatchPrecondition(condition: .onQueue(dispatchQueue))
return proxiedRequests.removeValue(forKey: identifier)
}
}
|
