From 857072dc11b02d7edc4892613b8d176dc273e92f Mon Sep 17 00:00:00 2001 From: rsc Date: Sat, 7 May 2005 22:37:26 +0000 Subject: try to do better with headers --- src/libip/udp.c | 25 +++++++++++++------------ 1 file changed, 13 insertions(+), 12 deletions(-) (limited to 'src/libip/udp.c') diff --git a/src/libip/udp.c b/src/libip/udp.c index 8df03ba0..042aa7e9 100644 --- a/src/libip/udp.c +++ b/src/libip/udp.c @@ -6,23 +6,24 @@ #include #include -/* - * prefix of all v4 addresses - * copied from libip because libc cannot depend on libip - */ -static uchar v4prefix[IPaddrlen] = { - 0, 0, 0, 0, - 0, 0, 0, 0, - 0, 0, 0xff, 0xff, - 0, 0, 0, 0 -}; - long udpread(int fd, Udphdr *hdr, void *buf, long n) { struct sockaddr_in sin; socklen_t len; + len = sizeof sin; + if(getsockname(fd, (struct sockaddr*)&sin, &len) < 0) + return -1; + if(len != sizeof sin){ + werrstr("getsockname acting weird"); + return -1; + } + memset(hdr, 0, sizeof *hdr); + memmove(hdr->laddr, v4prefix, IPaddrlen); + *(u32int*)(hdr->laddr+12) = *(u32int*)&sin.sin_addr; + *(u16int*)hdr->lport = *(u16int*)&sin.sin_port; + len = sizeof sin; n = recvfrom(fd, buf, n, 0, (struct sockaddr*)&sin, &len); if(n < 0) @@ -31,10 +32,10 @@ udpread(int fd, Udphdr *hdr, void *buf, long n) werrstr("recvfrom acting weird"); return -1; } - memset(hdr, 0, sizeof *hdr); memmove(hdr->raddr, v4prefix, IPaddrlen); *(u32int*)(hdr->raddr+12) = *(u32int*)&sin.sin_addr; *(u16int*)hdr->rport = *(u16int*)&sin.sin_port; + return n; } -- cgit v1.2.3