proxie 0.1.1

Library for making requests through HTTP or SOCKS proxy
Documentation 
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

use tokio::{
    net::TcpStream,
    io::{AsyncWriteExt, AsyncReadExt, BufReader},
};
use async_trait::async_trait;
use crate::{
    prelude::*,
    utils::*,
    error::*,
    target::ToTarget,
    proxy::{SOCKS5Proxy, SOCKS5Command},
    tokio::{AsyncProxy, ProxyTcpStream},
};

#[async_trait]
impl AsyncProxy for SOCKS5Proxy {
    async fn connect(&self, addr: impl ToTarget + Send) -> Result<ProxyTcpStream> {
        let mut stream = TcpStream::connect((&*self.server, self.port)).await?;

        let auth_method = self.tokio_connect(&mut stream).await?;

        if auth_method == 0x02 {
            self.tokio_authenticate(&mut stream).await?;
        }

        self.tokio_request(&mut stream, addr).await?;

        Ok(ProxyTcpStream {
            stream
        })
    }
}

impl SOCKS5Proxy {
    async fn tokio_connect(&self, stream: &mut TcpStream) -> Result<u8> {
        let peer = stream.peer_addr()?;

        let request = make_socks5_initial_request(&self.auth);
        stream.write_all(&request).await?;
        stream.flush().await?;

        let mut reader = BufReader::new(stream);
        let mut buffer = vec![0u8; 2];

        let mut buffer_len = 0;
        while buffer_len < 2 {
            let read = reader.read(&mut buffer[buffer_len..]).await?;
            buffer_len += read;
        }

        if buffer[1] == 0xFF { //NO ACCEPTABLE METHODS
            return Err(Box::new(Socks5HandshakeError {
                socket: peer,
            }));
        }

        Ok(buffer[1])
    }

    async fn tokio_authenticate(&self, stream: &mut TcpStream) -> Result<()> {
        let peer = stream.peer_addr()?;

        let request = match &self.auth {
            Some(auth) => make_socks5_authentication_request(&auth),
            None => return Err(Box::new(Socks5HandshakeError {
                socket: peer,
            })),
        };
        stream.write_all(&request).await?;
        stream.flush().await?;

        let mut reader = BufReader::new(stream);
        let mut buffer = vec![0u8; 2];

        let mut buffer_len = 0;
        while buffer_len < 2 {
            let read = reader.read(&mut buffer[buffer_len..]).await?;
            buffer_len += read;
        }

        if buffer[1] != 0x00 { //Authentication failure
            return Err(Box::new(Socks5HandshakeError {
                socket: peer,
            }));
        }

        Ok(())
    }

    async fn tokio_request(&self, stream: &mut TcpStream, addr: impl ToTarget) -> Result<()> {
        let target = addr.to_target()?;
        let peer = stream.peer_addr()?;

        let request = make_socks5_request(SOCKS5Command::CONNECT, &target)?;
        stream.write_all(&request).await?;
        stream.flush().await?;

        let mut reader = BufReader::new(stream);
        let mut buffer = vec![0u8; 512];

        let mut buffer_len = 0;
        while buffer_len < 4 { //Read until ATYPE
            let read = reader.read(&mut buffer[buffer_len..]).await?;
            buffer_len += read;
        }

        if buffer[3] == 0x01 { //IPv4
            while buffer_len < 10 { //Additional 4 bytes for IPv4 and 2 bytes for port
                let read = reader.read(&mut buffer[buffer_len..]).await?;
                buffer_len += read;
            }
        } else if buffer[3] == 0x04 { //IPv6
            while buffer_len < 22 { //Additional 16 bytes for IPv6 and 2 bytes for port
                let read = reader.read(&mut buffer[buffer_len..]).await?;
                buffer_len += read;
            }
        } else if buffer[3] == 0x03 { //Domain name
            while buffer_len < 5 { //Get domain length
                let read = reader.read(&mut buffer[buffer_len..]).await?;
                buffer_len += read;
            }

            while buffer_len < 7 + buffer[4] as usize { //Additional buffer[4] bytes for domain name and 2 bytes for port
                let read = reader.read(&mut buffer[buffer_len..]).await?;
                buffer_len += read;
            }
        } else { //Malformed packet
            return Err(Box::new(Socks5HandshakeError {
                socket: peer,
            }));
        }

        if buffer[1] != 0x00 { //Request refused
            return Err(Box::new(Socks5RequestError {
                socket: peer,
                error_type: buffer[1],
            }));
        }

        Ok(())
    }
}