mirror of
https://github.com/mailscope/kumomta.git
synced 2026-10-27 16:56:13 +00:00
fix detecting EOF
This commit is contained in:
@@ -310,6 +310,9 @@ impl SmtpServer {
|
||||
// Didn't find a complete line, fill up the rest of the buffer
|
||||
let mut data = [0u8; 1024];
|
||||
let size = self.socket.as_mut().unwrap().read(&mut data).await?;
|
||||
if size == 0 {
|
||||
anyhow::bail!("client disconnected");
|
||||
}
|
||||
self.read_buffer.extend_from_slice(&data[0..size]);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -82,6 +82,10 @@ impl SmtpClient {
|
||||
// Didn't find a complete line, fill up the rest of the buffer
|
||||
let mut data = [0u8; MAX_LINE_LEN];
|
||||
let size = self.socket.as_mut().unwrap().read(&mut data).await?;
|
||||
if size == 0 {
|
||||
self.socket.take();
|
||||
return Err(ClientError::NotConnected);
|
||||
}
|
||||
self.read_buffer.extend_from_slice(&data[0..size]);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user