use socks5_impl::protocol::{Address, AsyncStreamOperation, AuthMethod, Reply, Request, Response, handshake};
use std::io;
use tokio::{io::AsyncWriteExt, net::TcpListener};
#[tokio::main]
async fn main() -> io::Result<()> {
let listener = TcpListener::bind("127.0.0.1:5000").await?;
let (mut stream, _) = listener.accept().await?;
let request = handshake::Request::retrieve_from_async_stream(&mut stream).await?;
if request.evaluate_method(AuthMethod::NoAuth) {
let response = handshake::Response::new(AuthMethod::NoAuth);
response.write_to_async_stream(&mut stream).await?;
} else {
let response = handshake::Response::new(AuthMethod::NoAcceptableMethods);
response.write_to_async_stream(&mut stream).await?;
let _ = stream.shutdown().await;
let err = "No available handshake method provided by client";
return Err(io::Error::new(io::ErrorKind::Unsupported, err));
}
let _req = match Request::retrieve_from_async_stream(&mut stream).await {
Ok(req) => req,
Err(err) => {
let resp = Response::new(Reply::GeneralFailure, Address::unspecified());
resp.write_to_async_stream(&mut stream).await?;
let _ = stream.shutdown().await;
return Err(err);
}
};
Ok(())
}