1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
#include <u.h>
#define NOPLAN9DEFINES
#include <libc.h>
#include <ip.h>
#include <sys/socket.h>
#include <netinet/in.h>
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)
return -1;
if(len != sizeof sin){
werrstr("recvfrom acting weird");
return -1;
}
memmove(hdr->raddr, v4prefix, IPaddrlen);
*(u32int*)(hdr->raddr+12) = *(u32int*)&sin.sin_addr;
*(u16int*)hdr->rport = *(u16int*)&sin.sin_port;
return n;
}
long
udpwrite(int fd, Udphdr *hdr, void *buf, long n)
{
struct sockaddr_in sin;
memset(&sin, 0, sizeof sin);
sin.sin_family = AF_INET;
*(u32int*)&sin.sin_addr = *(u32int*)(hdr->raddr+12);
*(u16int*)&sin.sin_port = *(u16int*)hdr->rport;
return sendto(fd, buf, n, 0, (struct sockaddr*)&sin, sizeof sin);
}
|