comments

Author: blobcode

src/args.rs

@@ -6,6 +6,7 @@ pub struct Opts {
 }
 
 pub fn parse() -> Opts {
+    // config path
     let path = short('c')
         .long("config")
         .help("Path to config file")

src/config.rs

@@ -25,6 +25,7 @@ pub struct HostToml {
 fn parsehosts(config: ConfigToml) -> HashMap<String, String> {
     // parse list
     let mut hosts = HashMap::new();
+
     // add all "to" and "from" fields to the hashmap
     for host in config.host {
         for from in host.from {

src/main.rs

@@ -1,5 +1,5 @@
 use simple_logger::SimpleLogger;
-use std::{collections::HashMap, error::Error};
+use std::error::Error;
 
 mod args;
 mod config;
@@ -16,14 +16,14 @@ async fn main() -> Result<(), Box<dyn Error>> {
     // get config
     let config = config::parse(opts.path);
 
-    let mut hosts = HashMap::new();
-    hosts.insert("localhost:8000".to_string(), "localhost:8080".to_string());
-
     // server setup
     let server = server::Server {
         port: config.port,
-        hosts,
+        hosts: config.hosts,
     };
+
+    // run server
     server.run().await?;
+
     Ok(())
 }

src/server.rs

@@ -36,6 +36,7 @@ impl Server {
 
 // request handler
 async fn handle(inbound: TcpStream, hosts: HashMap<String, String>) -> Result<(), Box<dyn Error>> {
+    // buffer init
     let mut buf = vec![0; 1024];
     let mut headers = [httparse::EMPTY_HEADER; 16];
     let mut r = httparse::Request::new(&mut headers);
@@ -61,10 +62,12 @@ async fn handle(inbound: TcpStream, hosts: HashMap<String, String>) -> Result<()
     Ok(())
 }
 
-// proxy a tcp stream
+// proxy tcpstream
 async fn proxy(mut inbound: TcpStream, proxy_addr: String) -> Result<(), Box<dyn Error>> {
+    // connect to server
     let mut outbound = TcpStream::connect(proxy_addr).await?;
 
+    // split and merge streams
     let (mut ri, mut wi) = inbound.split();
     let (mut ro, mut wo) = outbound.split();