g3proxy: fix icap connection save action

bytedance · Dec 17, 2024 · 83b8a24 · 83b8a24
1 parent b272514
commit 83b8a24
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/g3proxy/src/inspect/imap/forward.rs b/g3proxy/src/inspect/imap/forward.rs
@@ -225,7 +225,7 @@ where
                     let mut body_reader = body.body_reader();
                     let mut sinker = tokio::io::sink();
                     let _ = tokio::io::copy(&mut body_reader, &mut sinker).await;
-                    if body_reader.finished() {
+                    if body_reader.trailer(128).await.is_ok() {
                         body.save_connection().await;
                     }
                 }

diff --git a/g3proxy/src/inspect/smtp/transaction/mod.rs b/g3proxy/src/inspect/smtp/transaction/mod.rs
@@ -366,7 +366,7 @@ impl<'a, SC: ServerConfig> Transaction<'a, SC> {
                     let mut body_reader = body.body_reader();
                     let mut sinker = tokio::io::sink();
                     let _ = tokio::io::copy(&mut body_reader, &mut sinker).await;
-                    if body_reader.finished() {
+                    if body_reader.trailer(128).await.is_ok() {
                         body.save_connection().await;
                     }
                 }