Merge pull request #2057 from bugadani/dns

Wake Stack's polling task after queueing a DNS query
embassy-rs · Oct 14, 2023 · f7c0ec6 · f7c0ec6
2 parents 575db17 + 69bb455
commit f7c0ec6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/embassy-net/src/lib.rs b/embassy-net/src/lib.rs
@@ -509,7 +509,10 @@ impl<D: Driver> Stack<D> {
             self.with_mut(|s, i| {
                 let socket = s.sockets.get_mut::<dns::Socket>(i.dns_socket);
                 match socket.start_query(s.iface.context(), name, qtype) {
-                    Ok(handle) => Poll::Ready(Ok(handle)),
+                    Ok(handle) => {
+                        s.waker.wake();
+                        Poll::Ready(Ok(handle))
+                    }
                     Err(dns::StartQueryError::NoFreeSlot) => {
                         i.dns_waker.register(cx.waker());
                         Poll::Pending