protocols/mdns: Ensure to set non_blocking on tokio udp socket

See tokio-rs/tokio#3016 for details.
libp2p · Nov 2, 2020 · 6f51985 · 6f51985
1 parent cdfb8df
commit 6f51985
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/protocols/mdns/src/service.rs b/protocols/mdns/src/service.rs
@@ -298,8 +298,9 @@ impl fmt::Debug for $service_name {
 #[cfg(feature = "async-std")]
 codegen!("async-std", MdnsService, async_std::net::UdpSocket, (|socket| Ok::<_, std::io::Error>(async_std::net::UdpSocket::from(socket))));
 
+// Note: Tokio's UdpSocket::from_std does not set the socket into non-blocking mode.
 #[cfg(feature = "tokio")]
-codegen!("tokio", TokioMdnsService, tokio::net::UdpSocket, (|socket| tokio::net::UdpSocket::from_std(socket)));
+codegen!("tokio", TokioMdnsService, tokio::net::UdpSocket, (|socket: std::net::UdpSocket| { socket.set_nonblocking(true); tokio::net::UdpSocket::from_std(socket) }));
 
 
 /// A valid mDNS packet received by the service.