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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
#[allow(dead_code)]
use std::{
error::Error,
fmt,
net::{Ipv4Addr, Ipv6Addr, SocketAddr, SocketAddrV4, SocketAddrV6, ToSocketAddrs},
};
use tokio::{
io::{AsyncReadExt, AsyncRead, AsyncWrite},
};
pub const VERSION5: u8 = 0x05;
pub const RESERVED: u8 = 0x00;
pub enum Command {
Connect = 0x01,
Bind = 0x02,
UdpAssosiate = 0x3,
}
impl Command {
pub fn from(byte: usize) -> Option<Command> {
match byte {
1 => Some(Command::Connect),
2 => Some(Command::Bind),
3 => Some(Command::UdpAssosiate),
_ => None,
}
}
}
#[derive(PartialEq)]
pub enum AddrType {
V4 = 0x01,
Domain = 0x03,
V6 = 0x04,
}
impl AddrType {
pub fn from(byte: usize) -> Option<AddrType> {
match byte {
1 => Some(AddrType::V4),
3 => Some(AddrType::Domain),
4 => Some(AddrType::V6),
_ => None,
}
}
pub async fn get_socket_addrs<S: AsyncRead + AsyncWrite + Unpin>(
socket: &mut S,
) -> Result<Vec<SocketAddr>, Box<dyn Error>> {
let mut addr_type = [0u8; 1];
socket.read(&mut addr_type).await?;
let addr_type = AddrType::from(addr_type[0] as usize);
if let None = addr_type {
Err(Response::AddrTypeNotSupported)?;
}
let addr_type = addr_type.unwrap();
let addr;
if let AddrType::Domain = addr_type {
let mut dlen = [0u8; 1];
socket.read_exact(&mut dlen).await?;
let mut domain = vec![0u8; dlen[0] as usize];
socket.read_exact(&mut domain).await?;
addr = domain;
} else if let AddrType::V4 = addr_type {
let mut v4 = [0u8; 4];
socket.read_exact(&mut v4).await?;
addr = Vec::from(v4);
} else {
let mut v6 = [0u8; 16];
socket.read_exact(&mut v6).await?;
addr = Vec::from(v6);
}
let mut port = [0u8; 2];
socket.read_exact(&mut port).await?;
let port = (u16::from(port[0]) << 8) | u16::from(port[1]);
match addr_type {
AddrType::V6 => {
let new_addr = (0..8)
.map(|x| (u16::from(addr[(x * 2)]) << 8) | u16::from(addr[(x * 2) + 1]))
.collect::<Vec<u16>>();
Ok(vec![SocketAddr::from(SocketAddrV6::new(
Ipv6Addr::new(
new_addr[0],
new_addr[1],
new_addr[2],
new_addr[3],
new_addr[4],
new_addr[5],
new_addr[6],
new_addr[7],
),
port,
0,
0,
))])
}
AddrType::V4 => Ok(vec![SocketAddr::from(SocketAddrV4::new(
Ipv4Addr::new(addr[0], addr[1], addr[2], addr[3]),
port,
))]),
AddrType::Domain => {
let mut domain = String::from_utf8_lossy(&addr[..]).to_string();
domain.push_str(&":");
domain.push_str(&port.to_string());
Ok(domain.to_socket_addrs()?.collect())
}
}
}
}
#[allow(dead_code)]
#[derive(Debug)]
pub enum Response {
Success = 0x00,
Failure = 0x01,
RuleFailure = 0x02,
NetworkUnreachable = 0x03,
HostUnreachable = 0x04,
ConnectionRefused = 0x05,
TtlExpired = 0x06,
CommandNotSupported = 0x07,
AddrTypeNotSupported = 0x08,
}
impl Error for Response {}
impl fmt::Display for Response {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "Error: {}", self)
}
}
#[derive(PartialEq)]
pub enum AuthMethod {
NoAuth = 0x00,
UserPass = 0x02,
NoMethods = 0xFF,
}
impl AuthMethod {
fn from(byte: u8) -> AuthMethod {
if byte == (AuthMethod::NoAuth as u8) {
AuthMethod::NoAuth
} else if byte == (AuthMethod::UserPass as u8) {
AuthMethod::UserPass
} else {
AuthMethod::NoMethods
}
}
pub async fn get_available_methods<S: AsyncRead + AsyncWrite + Unpin>(
methods_count: u8,
socket: &mut S,
) -> Result<Vec<AuthMethod>, Box<dyn Error>> {
let mut methods: Vec<AuthMethod> = Vec::with_capacity(methods_count as usize);
for _ in 0..methods_count {
let mut method = [0u8; 1];
socket.read_exact(&mut method).await?;
methods.push(AuthMethod::from(method[0]));
}
Ok(methods)
}
}