Merge pull request #2057 from bugadani/dns

Wake Stack's polling task after queueing a DNS query
2023-10-14 14:17:54 +00:00
parent 575db17264 69bb455c60
commit f7c0ec66d2
1 changed files with 4 additions and 1 deletions
--- 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