Formatting some stuff / cleaned up

src/data.rs

@@ -128,12 +128,8 @@ impl DataHandler {
 
             stream.write_all(&data).unwrap();
             let _ = stream.flush();
-            //println!("[+] Relayed -> {}", host);
-
-            //return Some(stream); Instead of returning we will now set stream object
             self.relay_stream = Some(stream);
             return Some(0);
         }
     }
-
 } // DataHandler

src/http.rs

@@ -21,8 +21,8 @@ pub struct RelayRequest<'a> {
     http_body: Option<String>,
 }
 
-/*
+/* Unused HTTP helper functions
-fn get_req_headers<'a>(&self, data: &'a Vec<u8>) -> Option<Vec<Header<'a>>> {
+pub fn get_req_headers<'a>(data: &'a Vec<u8>) -> Option<Vec<Header<'a>>> {
 
     let mut headers = [httparse::EMPTY_HEADER; 128];
     let mut request = httparse::Request::new(&mut headers);
@@ -35,7 +35,7 @@ fn get_req_headers<'a>(&self, data: &'a Vec<u8>) -> Option<Vec<Header<'a>>> {
     None
 }
 
-fn get_res_headers(data: &Vec<u8>) -> Option<Vec<Header>> {
+pub fn get_res_headers(data: &Vec<u8>) -> Option<Vec<Header>> {
 
     let mut headers = [httparse::EMPTY_HEADER; 128];
     let mut response = httparse::Response::new(&mut headers);
@@ -48,7 +48,7 @@ fn get_res_headers(data: &Vec<u8>) -> Option<Vec<Header>> {
     None
 }
 
-fn get_host(&self, data: &Vec<u8>) -> Option<String> {
+pub fn get_host(data: &Vec<u8>) -> Option<String> {
 
     return Some("137.220.37.67".to_string());
     let mut headers = [httparse::EMPTY_HEADER; 128];
@@ -78,7 +78,7 @@ fn get_host(&self, data: &Vec<u8>) -> Option<String> {
     }
 }
 
-fn get_cookie(&self, data: &Vec<u8>) -> Option<String> {
+pub fn get_cookie(data: &Vec<u8>) -> Option<String> {
 
     let mut headers = [httparse::EMPTY_HEADER; 128];
 
@@ -107,7 +107,6 @@ fn get_cookie(&self, data: &Vec<u8>) -> Option<String> {
     }
 }
 */
-
 pub fn http_req_verbose(data: &Vec<u8>, mode: u8) {
 
     let req_info = get_request(&data);
@@ -306,7 +305,6 @@ pub fn get_http_body(data: &Vec<u8>, headers: Vec<Header>) -> Option<Vec<u8>> {
                 } else {
                     return Some(blob);
                 }
-
             }
         }
     }

src/lib.rs

@@ -23,7 +23,6 @@ struct RelayConfig {
     ssl_private_key_path: String,
     ssl_cert_path: String,
     verbose_level: i8,
-
 }
 
 pub trait HandlerCallbacks {
@@ -44,7 +43,6 @@ pub struct SSLRelay<H>
 where
     H: HandlerCallbacks + std::marker::Sync + std::marker::Send + 'static,
 {
-    
     config: Option<RelayConfig>,
     handlers: Option<H>,
 }