From 1c767a0b4ca33d75374d69e681e9cda5fc0d35b5 Mon Sep 17 00:00:00 2001
From: MeexReay <meexreay@gmail.com>
Date: Fri, 13 Sep 2024 23:53:17 +0300
Subject: [PATCH] flake

---
 src/flowgate/server.rs   | 59 ++++++++++++++++++++++++++++++++++++++--
 src/flowgate/ssl_cert.rs | 20 +++-----------
 src/lib.rs               |  3 ++
 src/main.rs              |  4 +--
 4 files changed, 65 insertions(+), 21 deletions(-)
 create mode 100644 src/lib.rs

diff --git a/src/flowgate/server.rs b/src/flowgate/server.rs
index 77cdae0..cd2c9cc 100644
--- a/src/flowgate/server.rs
+++ b/src/flowgate/server.rs
@@ -1,10 +1,10 @@
 use std::{io::{Read, Write}, net::{Shutdown, SocketAddr, TcpListener}, sync::Arc, thread, time::Duration};
 
-use log::{debug, info};
+use log::info;
 use openssl::ssl::{NameType, SniError, SslAcceptor, SslAlert, SslMethod, SslRef};
 use threadpool::ThreadPool;
 
-use crate::Config;
+use super::Config;
 
 pub struct FlowgateServer {
     config: Arc<Config>,
@@ -193,6 +193,61 @@ impl FlowgateServer {
             info!("{} > {} http://{}{}", addr.to_string(), method, host, page);
         }
 
+        if keep_alive {
+            loop {
+                let mut reqst_data: Vec<u8> = vec![0; 4096];
+
+                stream.read(&mut reqst_data).ok()?;
+
+                let reqst = String::from_utf8(reqst_data).ok()?;
+                let reqst = reqst.trim_matches(char::from(0));
+
+                let (head, body) = reqst.split_once("\r\n\r\n")?;
+
+                let mut head_lines = head.split("\r\n");
+
+                let status = head_lines.next()?;
+                let status: Vec<&str> = status.split(" ").collect();
+
+                let mut content_length: usize = 0;
+
+                for l in head_lines {
+                    let (key, value) = l.split_once(": ")?;
+                    let key = key.to_lowercase().replace("-", "_");
+
+                    if key == "content_length" {
+                        content_length = value.parse().ok()?;
+                    }
+                }
+
+                site_stream.write((addr.to_string() + "\n" + reqst).as_bytes()).ok()?;
+
+                if content_length != 0 && content_length > body.len() {
+                    let mut body_data: Vec<u8> = Vec::new();
+                    stream.read_to_end(&mut body_data).ok()?;
+                    site_stream.write_all(&body_data).ok()?;
+                }
+
+                loop {
+                    let mut buf: Vec<u8> = Vec::new();
+                    site_stream.read_to_end(&mut buf).ok()?;
+                    if buf.is_empty() {
+                        break;
+                    }
+                    stream.write_all(&buf).ok()?;
+                }
+
+                let method = status[0];
+                let page = status[1];
+        
+                if https {
+                    info!("{} > {} https://{}{}", addr.to_string(), method, host, page);
+                } else {
+                    info!("{} > {} http://{}{}", addr.to_string(), method, host, page);
+                }
+            }
+        }
+
         site_stream.shutdown(Shutdown::Both).ok()?;
 
         Some(())
diff --git a/src/flowgate/ssl_cert.rs b/src/flowgate/ssl_cert.rs
index 118e000..6481ad1 100644
--- a/src/flowgate/ssl_cert.rs
+++ b/src/flowgate/ssl_cert.rs
@@ -6,22 +6,10 @@ pub struct SslCert {
 }
 
 fn generate_ctx(cert_file: &str, key_file: &str) -> Option<SslContext> {
-    let mut ctx = match SslContext::builder(SslMethod::tls()) {
-        Ok(i) => i,
-        Err(_) => return None,
-    };
-    match ctx.set_private_key_file(&key_file, SslFiletype::PEM) {
-        Ok(i) => i,
-        Err(_) => return None,
-    };
-    match ctx.set_certificate_file(&cert_file, SslFiletype::PEM) {
-        Ok(i) => i,
-        Err(_) => return None,
-    };
-    match ctx.check_private_key() {
-        Ok(i) => i,
-        Err(_) => return None,
-    };
+    let mut ctx = SslContext::builder(SslMethod::tls()).ok()?;
+    ctx.set_private_key_file(&key_file, SslFiletype::PEM).ok()?;
+    ctx.set_certificate_file(&cert_file, SslFiletype::PEM).ok()?;
+    ctx.check_private_key().ok()?;
     Some(ctx.build())
 }
 
diff --git a/src/lib.rs b/src/lib.rs
new file mode 100644
index 0000000..a812153
--- /dev/null
+++ b/src/lib.rs
@@ -0,0 +1,3 @@
+pub mod flowgate;
+
+pub use flowgate::*;
\ No newline at end of file
diff --git a/src/main.rs b/src/main.rs
index 9d6775f..899b681 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,5 +1,3 @@
-mod flowgate;
-
 use flowgate::{Config, FlowgateServer};
 
 fn main() {
@@ -11,4 +9,4 @@ fn main() {
     server.start();
 
     loop {}
-}
+}
\ No newline at end of file