poll_fd: add missing fd registration for UDP and ICMP

Signed-off-by: Samuel Thibault <samuel.thibault@ens-lyon.org>

2 files changed, 2 insertions, 0 deletions

diff --git a/src/ip_icmp.c b/src/ip_icmp.c
index 9917f36..9fba653 100644
--- a/src/ip_icmp.c
+++ b/src/ip_icmp.c
@@ -95,6 +95,7 @@ static int icmp_send(struct socket *so, struct mbuf *m, int hlen)
     if (so->s == -1) {
         return -1;
     }
+    so->slirp->cb->register_poll_fd(so->s, so->slirp->opaque);
 
     if (slirp_bind_outbound(so, AF_INET) != 0) {
         // bind failed - close socket
diff --git a/src/udp.c b/src/udp.c
index 0dd1f67..767ca85 100644
--- a/src/udp.c
+++ b/src/udp.c
@@ -328,6 +328,7 @@ int udp_attach(struct socket *so, unsigned short af)
         so->so_expire = curtime + SO_EXPIRE;
         insque(so, &so->slirp->udb);
     }
+    so->slirp->cb->register_poll_fd(so->s, so->slirp->opaque);
     return (so->s);
 }