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

use tokio::{
    net::TcpStream,
    io::{AsyncWriteExt, AsyncBufReadExt, BufReader},
};
use async_trait::async_trait;
use crate::{
    prelude::*,
    utils::*,
    target::ToTarget,
    proxy::HTTPProxy,
    tokio::{AsyncProxy, ProxyTcpStream},
};

#[async_trait]
impl AsyncProxy for HTTPProxy {
    async fn connect(&self, addr: impl ToTarget + Send) -> Result<ProxyTcpStream> {
        let request = make_http_connect_request(&addr, &self)?;

        let mut stream = TcpStream::connect((&*self.server, self.port)).await?;
        stream.write_all(request.as_bytes()).await?;
        stream.flush().await?;

        let mut reader = BufReader::new(&mut stream);
        let mut buffer = String::new();

        loop {
            reader.read_line(&mut buffer).await?;

            if buffer.ends_with("\r\n\r\n") {
                break;
            }
        }

        parse_http_response(&buffer.as_bytes())?;

        Ok(ProxyTcpStream {
            stream
        })
    }
}