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