X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ian/git?a=blobdiff_plain;f=udp.c;h=552a58e000d78d372cc5def7de0afa7512785ccb;hb=3c35339bc4d3db033cc44558baf66c80e87fbd85;hp=1e637b6e5be7740ccb53f3678496298a8f31ffe2;hpb=6af9a9842664ba88d89a67b300c1b0970bcf43ee;p=secnet.git

diff --git a/udp.c b/udp.c
index 1e637b6..552a58e 100644
--- a/udp.c
+++ b/udp.c
@@ -19,6 +19,7 @@
 #include <netinet/in.h>
 #include <arpa/inet.h>
 #include "util.h"
+#include "magic.h"
 #include "unaligned.h"
 #include "ipaddr.h"
 #include "magic.h"
@@ -103,8 +104,9 @@ static void udp_afterpoll(void *state, struct pollfd *fds, int nfds)
 	    BUF_ASSERT_FREE(st->rbuf);
 	    BUF_ALLOC(st->rbuf,"udp_afterpoll");
 	    buffer_init(st->rbuf,calculate_max_start_pad());
-	    rv=recvfrom(st->fd, st->rbuf->start, st->rbuf->len, 0,
-			(struct sockaddr *)&from, &fromlen);
+	    rv=recvfrom(st->fd, st->rbuf->start,
+			buf_remaining_space(st->rbuf),
+			0, (struct sockaddr *)&from, &fromlen);
 	    if (rv>0) {
 		st->rbuf->size=rv;
 		if (st->use_proxy) {