1 #include <u.h>
2 #include <libc.h>
3 #include <ip.h>
4 #include <thread.h>
5 #include <sunrpc.h>
6 
7 typedef struct SunMsgUdp SunMsgUdp;
8 struct SunMsgUdp
9 {
10 	SunMsg msg;
11 	Udphdr udp;
12 };
13 
14 typedef struct Arg Arg;
15 struct Arg
16 {
17 	SunSrv *srv;
18 	Channel *creply;
19 	Channel *csync;
20 	int fd;
21 };
22 
23 enum
24 {
25 	UdpMaxRead = 65536+Udphdrsize
26 };
27 static void
28 sunudpread(void *v)
29 {
30 	int n, paraport, port;
31 	uchar *buf;
32 	Arg arg = *(Arg*)v;
33 	SunMsgUdp *msg;
34 	SunSrv *srv;
35 	Udphdr udp;
36 	uchar localip[IPaddrlen];
37 
38 	sendp(arg.csync, 0);
39 	srv = arg.srv;
40 	paraport = -1;
41 
42 	/* 127.1 */
43 	memmove(localip, v4prefix, IPaddrlen);
44 	localip[12] = 127;
45 	localip[15] = 1;
46 
47 	buf = emalloc(UdpMaxRead);
48 	while((n = udpread(arg.fd, &udp, buf, UdpMaxRead)) > 0){
49 		if(arg.srv->chatty)
50 			fprint(2, "udpread got %d (%d) from %I\n", n, Udphdrsize, udp.raddr);
51 		if((srv->localonly || srv->localparanoia) && ipcmp(udp.raddr, localip) != 0){
52 			fprint(2, "dropping message from %I: not local\n", udp.raddr);
53 			continue;
54 		}
55 		if(srv->localparanoia){
56 			port = nhgets(udp.rport);
57 			if(paraport == -1){
58 				fprint(2, "paranoid mode: only %I/%d allowed\n", localip, port);
59 				paraport = port;
60 			}else if(paraport != port){
61 				fprint(2, "dropping message from %I: not port %d\n", udp.raddr, port);
62 				continue;
63 			}
64 		}
65 		msg = emalloc(sizeof(SunMsgUdp));
66 		msg->udp = udp;
67 		msg->msg.data = emalloc(n);
68 		msg->msg.count = n;
69 		memmove(msg->msg.data, buf, n);
70 		msg->msg.creply = arg.creply;
71 		if(arg.srv->chatty)
72 			fprint(2, "message %p count %d\n", msg, msg->msg.count);
73 		sendp(arg.srv->crequest, msg);
74 	}
75 }
76 
77 static void
78 sunudpwrite(void *v)
79 {
80 	uchar *buf;
81 	Arg arg = *(Arg*)v;
82 	SunMsgUdp *msg;
83 
84 	sendp(arg.csync, 0);
85 
86 	buf = emalloc(UdpMaxRead);
87 	while((msg = recvp(arg.creply)) != nil){
88 		if(udpwrite(arg.fd, &msg->udp, msg->msg.data, msg->msg.count) != msg->msg.count)
89 			fprint(2, "udpwrite: %r\n");
90 		free(msg->msg.data);
91 		free(msg);
92 	}
93 }
94 
95 int
96 sunsrvudp(SunSrv *srv, char *address)
97 {
98 	int fd;
99 	char adir[40];
100 	Arg *arg;
101 
102 	fd = announce(address, adir);
103 	if(fd < 0)
104 		return -1;
105 
106 	arg = emalloc(sizeof(Arg));
107 	arg->fd = fd;
108 	arg->srv = srv;
109 	arg->creply = chancreate(sizeof(SunMsg*), 10);
110 	arg->csync = chancreate(sizeof(void*), 10);
111 
112 	proccreate(sunudpread, arg, SunStackSize);
113 	proccreate(sunudpwrite, arg, SunStackSize);
114 	recvp(arg->csync);
115 	recvp(arg->csync);
116 	chanfree(arg->csync);
117 	free(arg);
118 
119 	return 0;
120 }