better-socks 0.1.2

//! Test the tor proxy capabilities
//!
//! This example requires a running tor proxy. The Unix-socket portion is
//! compiled on Unix only.

use tokio::{
    io::{AsyncReadExt, AsyncWriteExt},
    net::TcpStream,
    runtime::Runtime,
};
#[cfg(unix)]
use tokio::net::UnixStream;
use better_socks::{tcp::Socks5Stream, Error};

#[cfg(unix)]
const UNIX_PROXY_ADDR: &str = "/tmp/tor/socket.s";
const TCP_PROXY_ADDR: &str = "127.0.0.1:9050";
const ONION_ADDR: &str = "3g2upl4pq6kufc4m.onion:80"; // DuckDuckGo

async fn connect() -> Result<(), Error> {
    #[cfg(unix)]
    {
        // Tor must listen on a Unix domain socket. Create /tmp/tor owned by tor
        // and add `SocksPort unix:/tmp/tor/socket.s` to torrc.
        let socket = UnixStream::connect(UNIX_PROXY_ADDR).await?;
        let target = Socks5Stream::tor_resolve_with_socket(socket, "duckduckgo.com:0").await?;
        eprintln!("duckduckgo.com = {:?}", target);
        let socket = UnixStream::connect(UNIX_PROXY_ADDR).await?;
        let target = Socks5Stream::tor_resolve_ptr_with_socket(socket, "176.34.155.23:0").await?;
        eprintln!("176.34.155.23 = {:?}", target);
    }

    let socket = TcpStream::connect(TCP_PROXY_ADDR).await?;
    socket.set_nodelay(true)?;
    let mut conn = Socks5Stream::connect_with_socket(socket, ONION_ADDR).await?;
    conn.write_all(b"GET /\n\n").await?;

    let mut buf = Vec::new();
    let n = conn.read_to_end(&mut buf).await?;

    println!("{} bytes read\n\n{}", n, String::from_utf8_lossy(&buf));

    Ok(())
}

fn main() {
    let rt = Runtime::new().unwrap();
    rt.block_on(connect()).unwrap();
}