From 9c575641b41ea8a2e67018ec3f33563220955671 Mon Sep 17 00:00:00 2001
From: Tibo-lg <thibaut@cryptogarage.co.jp>
Date: Wed, 17 Jan 2024 09:20:30 +0900
Subject: [PATCH] Fix lightning-net-tokio sometimes dropping messages

---
 lightning-net-tokio/src/lib.rs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/lightning-net-tokio/src/lib.rs b/lightning-net-tokio/src/lib.rs
index 1aa2cc25a13..937529a4e1c 100644
--- a/lightning-net-tokio/src/lib.rs
+++ b/lightning-net-tokio/src/lib.rs
@@ -502,7 +502,12 @@ impl peer_handler::SocketDescriptor for SocketDescriptor {
 							written_len += res;
 							if written_len == data.len() { return written_len; }
 						},
-						Err(_) => return written_len,
+						Err(ref e) if e.kind() == std::io::ErrorKind::WouldBlock => {
+							continue;
+						}
+						Err(_) => {
+							return written_len;
+						}
 					}
 				},
 				task::Poll::Ready(Err(_)) => return written_len,