X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ian/git?p=secnet.git;a=blobdiff_plain;f=udp.c;h=98da30d6f66f391ac8944a4e2a835dacf257572e;hp=8ae5671bd64a38a16914997ffe035f65e0b4d26f;hb=ee697dd98d0f9aca6b9df7ffb32d3a15b4e33d6d;hpb=cf5f1149caccf65c700e73fc0e9212c916df3610

diff --git a/udp.c b/udp.c
index 8ae5671..98da30d 100644
--- a/udp.c
+++ b/udp.c
@@ -87,11 +87,7 @@ static int udp_beforepoll(void *state, struct pollfd *fds, int *nfds_io,
 {
     int i;
     struct udp *st=state;
-    if (*nfds_io<st->n_socks) {
-	*nfds_io=st->n_socks;
-	return ERANGE;
-    }
-    *nfds_io=st->n_socks;
+    BEFOREPOLL_WANT_FDS(st->n_socks);
     for (i=0; i<st->n_socks; i++) {
 	fds[i].fd=st->socks[i].fd;
 	fds[i].events=POLLIN;