SMTP: Handle empty fields in process request

Signed-off-by: Mroik <mroik@delayed.space>

1 files changed, 6 insertions, 1 deletions

diff --git a/src/smtp_server.rs b/src/smtp_server.rs
index bee00e9..91d732c 100644
--- a/src/smtp_server.rs
+++ b/src/smtp_server.rs
@@ -104,7 +104,12 @@ impl SessionHandler {
             // message
             if self.state == SessionState::AwaitingMailInput {
                 if buffer.starts_with(".\r\n") {
-                    self.process_mail().await?;
+                    if let Reply::InvalidCommand = self.process_mail().await? {
+                        writer
+                            .write_all(Reply::InvalidCommand.to_string().as_bytes())
+                            .await?;
+                        continue;
+                    }
                     writer
                         .write_all(Reply::Completed("Ok").to_string().as_bytes())
                         .await?;