From 96415faa2757d71d86e515d2c1eb068fb94d1f78 Mon Sep 17 00:00:00 2001
From: Tipragot <contact@tipragot.fr>
Date: Wed, 7 Feb 2024 17:09:25 +0100
Subject: [PATCH] Make an error when the connection is disconnected

---
 crates/bevnet/src/lib.rs | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/crates/bevnet/src/lib.rs b/crates/bevnet/src/lib.rs
index d78bc28..4b897cb 100644
--- a/crates/bevnet/src/lib.rs
+++ b/crates/bevnet/src/lib.rs
@@ -197,6 +197,12 @@ impl Connection {
         let receive_buffer = &mut self.receive_buffer[start_index..start_index + len];
         let received_len = self.stream.read(receive_buffer);
         self.receive_filled_len += match received_len {
+            Ok(0) => {
+                return Err(io::Error::new(
+                    io::ErrorKind::ConnectionAborted,
+                    "connection closed by remote peer",
+                ));
+            }
             Ok(n) => n,
             Err(ref e) if e.kind() == io::ErrorKind::WouldBlock => return Ok(false),
             Err(ref e) if e.kind() == io::ErrorKind::Interrupted => return Ok(false),