blob: bb53c9727a0bd0483e18274db65bdde59f5d91b7 (
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
|
//
// Socks5Authentication.swift
// MullvadTransport
//
// Created by pronebird on 19/10/2023.
//
import Foundation
import Network
/// Authentication methods supported by socks protocol.
enum Socks5AuthenticationMethod: UInt8 {
case notRequired = 0x00
case usernamePassword = 0x02
}
struct Socks5Authentication: Sendable {
let connection: NWConnection
let endpoint: Socks5Endpoint
let configuration: Socks5Configuration
typealias AuthenticationComplete = @Sendable () -> Void
typealias AuthenticationFailure = @Sendable (Error) -> Void
func authenticate(onComplete: @escaping AuthenticationComplete, onFailure: @escaping AuthenticationFailure) {
guard let username = configuration.username, let password = configuration.password else {
onFailure(Socks5Error.invalidUsernameOrPassword)
return
}
let authenticateCommand = Socks5UsernamePasswordCommand(username: username, password: password)
connection.send(
content: authenticateCommand.rawData,
completion: .contentProcessed { error in
if let error {
onFailure(error)
} else {
readNegotiationReply(onComplete: onComplete, onFailure: onFailure)
}
})
}
func readNegotiationReply(
onComplete: @escaping AuthenticationComplete,
onFailure: @escaping AuthenticationFailure
) {
let replySize = MemoryLayout<Socks5UsernamePasswordReply>.size
// Read in one shot, the payload is very small to not care about a reading loop.
connection.receive(exactLength: replySize) { data, _, _, error in
guard let data else {
if let error {
onFailure(error)
} else {
onFailure(Socks5Error.unexpectedEndOfStream)
}
return
}
guard let reply = Socks5UsernamePasswordReply(from: data) else {
onFailure(Socks5Error.unexpectedEndOfStream)
return
}
guard reply.version == Socks5Constants.usernamePasswordAuthenticationProtocol else {
onFailure(Socks5Error.invalidSocksVersion)
return
}
onComplete()
}
}
}
|
