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 551445b9 2004-04-21 devnull if((srv->localonly || srv->localparanoia) && ipcmp(udp.raddr, localip) != 0){
52 551445b9 2004-04-21 devnull fprint(2, "dropping message from %I: not local\n", udp.raddr);
53 551445b9 2004-04-21 devnull continue;
55 551445b9 2004-04-21 devnull if(srv->localparanoia){
56 551445b9 2004-04-21 devnull port = nhgets(udp.rport);
57 551445b9 2004-04-21 devnull if(paraport == -1){
58 551445b9 2004-04-21 devnull fprint(2, "paranoid mode: only %I/%d allowed\n", localip, port);
59 551445b9 2004-04-21 devnull paraport = port;
60 551445b9 2004-04-21 devnull }else if(paraport != port){
61 551445b9 2004-04-21 devnull fprint(2, "dropping message from %I: not port %d\n", udp.raddr, port);
62 551445b9 2004-04-21 devnull continue;
65 551445b9 2004-04-21 devnull msg = emalloc(sizeof(SunMsgUdp));
66 551445b9 2004-04-21 devnull msg->udp = udp;
67 551445b9 2004-04-21 devnull msg->msg.data = emalloc(n);
68 551445b9 2004-04-21 devnull msg->msg.count = n;
69 551445b9 2004-04-21 devnull memmove(msg->msg.data, buf, n);
70 551445b9 2004-04-21 devnull msg->msg.creply = arg.creply;
71 551445b9 2004-04-21 devnull if(arg.srv->chatty)
72 551445b9 2004-04-21 devnull fprint(2, "message %p count %d\n", msg, msg->msg.count);
73 551445b9 2004-04-21 devnull sendp(arg.srv->crequest, msg);
77 551445b9 2004-04-21 devnull static void
78 551445b9 2004-04-21 devnull sunudpwrite(void *v)
80 551445b9 2004-04-21 devnull uchar *buf;
81 551445b9 2004-04-21 devnull Arg arg = *(Arg*)v;
82 551445b9 2004-04-21 devnull SunMsgUdp *msg;
84 551445b9 2004-04-21 devnull sendp(arg.csync, 0);
86 551445b9 2004-04-21 devnull buf = emalloc(UdpMaxRead);
87 551445b9 2004-04-21 devnull while((msg = recvp(arg.creply)) != nil){
88 551445b9 2004-04-21 devnull if(udpwrite(arg.fd, &msg->udp, msg->msg.data, msg->msg.count) != msg->msg.count)
89 551445b9 2004-04-21 devnull fprint(2, "udpwrite: %r\n");
90 551445b9 2004-04-21 devnull free(msg->msg.data);
91 551445b9 2004-04-21 devnull free(msg);
96 551445b9 2004-04-21 devnull sunsrvudp(SunSrv *srv, char *address)
99 551445b9 2004-04-21 devnull char adir[40];
100 551445b9 2004-04-21 devnull Arg *arg;
102 551445b9 2004-04-21 devnull fd = announce(address, adir);
103 551445b9 2004-04-21 devnull if(fd < 0)
104 551445b9 2004-04-21 devnull return -1;
106 551445b9 2004-04-21 devnull arg = emalloc(sizeof(Arg));
107 551445b9 2004-04-21 devnull arg->fd = fd;
108 551445b9 2004-04-21 devnull arg->srv = srv;
109 551445b9 2004-04-21 devnull arg->creply = chancreate(sizeof(SunMsg*), 10);
110 551445b9 2004-04-21 devnull arg->csync = chancreate(sizeof(void*), 10);
112 551445b9 2004-04-21 devnull proccreate(sunudpread, arg, SunStackSize);
113 551445b9 2004-04-21 devnull proccreate(sunudpwrite, arg, SunStackSize);
114 551445b9 2004-04-21 devnull recvp(arg->csync);
115 551445b9 2004-04-21 devnull recvp(arg->csync);
116 551445b9 2004-04-21 devnull chanfree(arg->csync);
117 551445b9 2004-04-21 devnull free(arg);
119 551445b9 2004-04-21 devnull return 0;