fix possible infinite loop

Author: neevek

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "omnip"
-version = "0.7.5"
+version = "0.7.6"
 edition = "2021"
 
 [lib]

Cargo.toml

@@ -30,6 +30,11 @@ impl HttpReq {
         loop {
             let mut tmp_buffer = [0u8; 256];
             let len = utils::read_from_stream(outbound_stream, &mut tmp_buffer).await?;
+            if len == 0 {
+                log::error!("failed to read request after retrying for 10 times");
+                return Err(ProxyError::BadRequest);
+            }
+
             buffer.extend_from_slice(&tmp_buffer[..len]);
             let len = buffer.len();
             if len > 4 {

src/http/http_req.rs

@@ -86,8 +86,13 @@ impl SocksReq {
                 _ => {}
             }
 
-            let read_len = utils::read_from_stream(outbound_stream, &mut buf).await?;
-            if !resp_parser.advance(&buf[..read_len]) {
+            let len = utils::read_from_stream(outbound_stream, &mut buf).await?;
+            if len == 0 {
+                log::error!("failed to read request after retrying for 10 times");
+                return Err(ProxyError::BadRequest);
+            }
+
+            if !resp_parser.advance(&buf[..len]) {
                 error!(
                     "connect failed: {:?}, dst_addr: {}",
                     resp_parser.state(),