From 4aca36beea668802b4f7b5220d970000fc1aeef2 Mon Sep 17 00:00:00 2001
From: Lee Smet <lee.smet@hotmail.com>
Date: Fri, 11 Apr 2025 14:31:18 +0200
Subject: [PATCH] Fix message systemd dropping final chunk unexpectedly

Signed-off-by: Lee Smet <lee.smet@hotmail.com>
---
 mycelium/src/message.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/mycelium/src/message.rs b/mycelium/src/message.rs
index d495cb4..c60e768 100644
--- a/mycelium/src/message.rs
+++ b/mycelium/src/message.rs
@@ -440,7 +440,11 @@ where
                     return;
                 }
                 // Check max chunk idx.
-                let max_chunk_idx = message.len.div_ceil(MINIMUM_CHUNK_SIZE);
+                let max_chunk_idx = if message.len == 0 {
+                    0
+                } else {
+                    message.len.div_ceil(MINIMUM_CHUNK_SIZE) - 1
+                };
                 if mc.chunk_idx() > max_chunk_idx {
                     debug!("Dropping CHUNK because index is too high");
                     return;