Fix lightning-net-tokio sometimes dropping messages

p2pderivatives · Jan 17, 2024 · 9c57564 · 9c57564
1 parent ea5de93
commit 9c57564
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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,