Add a timeout on http recv

Author: Ma233

src/lib.rs

@@ -147,48 +147,51 @@ impl PProxy {
 
                 loop {
                     let mut buf = [0u8; 30000];
-                    match stream.read(&mut buf).await {
-                        Err(_) => break,
-                        Ok(0) => break,
-                        Ok(n) => {
-                            response.extend_from_slice(&buf[..n]);
-
-                            if full_length.not_parsed() {
-                                let mut headers = [httparse::EMPTY_HEADER; 1024];
-                                let mut resp_checker = httparse::Response::new(&mut headers);
-                                let res = resp_checker.parse(&response)?;
-                                if res.is_complete() {
-                                    let content_length =
-                                        resp_checker.headers.iter().find_map(|h| {
-                                            if h.name.to_lowercase() != "content-length" {
-                                                return None;
-                                            }
-                                            let Ok(value) = std::str::from_utf8(h.value) else {
-                                                return None;
-                                            };
-                                            value.parse::<usize>().ok()
-                                        });
-
-                                    match content_length {
-                                        Some(content_length) => {
-                                            let header_length = res.unwrap();
-                                            full_length =
-                                                FullLength::Parsed(header_length + content_length)
-                                        }
-                                        None => {
-                                            full_length = FullLength::NotSet;
-                                        }
-                                    }
-                                }
-                            }
 
-                            if let FullLength::Parsed(full_length) = full_length {
-                                if response.len() >= full_length {
-                                    break;
+                    let Ok(Ok(n)) =
+                        timeout(std::time::Duration::from_secs(5), stream.read(&mut buf)).await
+                    else {
+                        break;
+                    };
+
+                    if n == 0 {
+                        break;
+                    }
+
+                    response.extend_from_slice(&buf[..n]);
+
+                    if full_length.not_parsed() {
+                        let mut headers = [httparse::EMPTY_HEADER; 1024];
+                        let mut resp_checker = httparse::Response::new(&mut headers);
+                        let res = resp_checker.parse(&response)?;
+                        if res.is_complete() {
+                            let content_length = resp_checker.headers.iter().find_map(|h| {
+                                if h.name.to_lowercase() != "content-length" {
+                                    return None;
+                                }
+                                let Ok(value) = std::str::from_utf8(h.value) else {
+                                    return None;
+                                };
+                                value.parse::<usize>().ok()
+                            });
+
+                            match content_length {
+                                Some(content_length) => {
+                                    let header_length = res.unwrap();
+                                    full_length = FullLength::Parsed(header_length + content_length)
+                                }
+                                None => {
+                                    full_length = FullLength::NotSet;
                                 }
                             }
                         }
                     }
+
+                    if let FullLength::Parsed(full_length) = full_length {
+                        if response.len() >= full_length {
+                            break;
+                        }
+                    }
                 }
 
                 if response.is_empty() {