blob: 3d105c019aaca9b0bf0ffee5510386a9bc27bb1f (
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
|
//
// Socks5HandshakeNegotiation.swift
// MullvadTransport
//
// Created by pronebird on 20/10/2023.
//
import Foundation
import Network
/// The object handling a handshake negotiation with socks proxy.
struct Socks5HandshakeNegotiation: Sendable {
let connection: NWConnection
let handshake: Socks5Handshake
let onComplete: @Sendable (Socks5HandshakeReply) -> Void
let onFailure: @Sendable (Error) -> Void
func perform() {
connection.send(
content: handshake.rawData,
completion: .contentProcessed { [self] error in
if let error {
onFailure(Socks5Error.remoteConnectionFailure(error))
} else {
readReply()
}
})
}
private func readReply() {
// The length of a handshake reply in bytes.
let replyLength = 2
connection.receive(exactLength: replyLength) { [self] data, _, _, error in
if let error {
onFailure(Socks5Error.remoteConnectionFailure(error))
} else if let data {
do {
onComplete(try parseReply(data: data))
} catch {
onFailure(error)
}
} else {
onFailure(Socks5Error.unexpectedEndOfStream)
}
}
}
private func parseReply(data: Data) throws -> Socks5HandshakeReply {
var iterator = data.makeIterator()
guard let version = iterator.next() else { throw Socks5Error.unexpectedEndOfStream }
guard version == Socks5Constants.socksVersion else { throw Socks5Error.invalidSocksVersion }
guard let rawMethod = iterator.next() else { throw Socks5Error.unexpectedEndOfStream }
// The response code returned by the server when none of the auth methods listed by the client are acceptable.
let authMethodsUnacceptableReplyCode: UInt8 = 0xff
guard rawMethod != authMethodsUnacceptableReplyCode else {
throw Socks5Error.unacceptableAuthMethods
}
guard let method = Socks5AuthenticationMethod(rawValue: rawMethod) else {
throw Socks5Error.unsupportedAuthMethod
}
return Socks5HandshakeReply(method: method)
}
}
|
