1 551445b9 2004-04-21 devnull #include <u.h>
2 551445b9 2004-04-21 devnull #include <libc.h>
3 4c06b8ee 2005-05-19 devnull #include <ip.h>
4 551445b9 2004-04-21 devnull #include <thread.h>
5 551445b9 2004-04-21 devnull #include <sunrpc.h>
8 551445b9 2004-04-21 devnull * Sun RPC server; for now, no reply cache
11 551445b9 2004-04-21 devnull static void sunrpcproc(void*);
12 551445b9 2004-04-21 devnull static void sunrpcrequestthread(void*);
13 551445b9 2004-04-21 devnull static void sunrpcreplythread(void*);
14 551445b9 2004-04-21 devnull static void sunrpcforkthread(void*);
15 551445b9 2004-04-21 devnull static SunProg *sunfindprog(SunSrv*, SunMsg*, SunRpc*, Channel**);
17 551445b9 2004-04-21 devnull typedef struct Targ Targ;
18 551445b9 2004-04-21 devnull struct Targ
20 551445b9 2004-04-21 devnull void (*fn)(void*);
21 551445b9 2004-04-21 devnull void *arg;
25 551445b9 2004-04-21 devnull sunsrv(void)
27 551445b9 2004-04-21 devnull SunSrv *srv;
29 551445b9 2004-04-21 devnull srv = emalloc(sizeof(SunSrv));
30 551445b9 2004-04-21 devnull srv->chatty = 0;
31 551445b9 2004-04-21 devnull srv->crequest = chancreate(sizeof(SunMsg*), 16);
32 551445b9 2004-04-21 devnull srv->creply = chancreate(sizeof(SunMsg*), 16);
33 551445b9 2004-04-21 devnull srv->cthread = chancreate(sizeof(Targ), 4);
35 551445b9 2004-04-21 devnull proccreate(sunrpcproc, srv, SunStackSize);
36 551445b9 2004-04-21 devnull return srv;
40 551445b9 2004-04-21 devnull sunsrvprog(SunSrv *srv, SunProg *prog, Channel *c)
42 551445b9 2004-04-21 devnull if(srv->nprog%16 == 0){
43 551445b9 2004-04-21 devnull srv->prog = erealloc(srv->prog, (srv->nprog+16)*sizeof(srv->prog[0]));
44 551445b9 2004-04-21 devnull srv->cdispatch = erealloc(srv->cdispatch, (srv->nprog+16)*sizeof(srv->cdispatch[0]));
46 551445b9 2004-04-21 devnull srv->prog[srv->nprog] = prog;
47 551445b9 2004-04-21 devnull srv->cdispatch[srv->nprog] = c;
48 551445b9 2004-04-21 devnull srv->nprog++;
51 551445b9 2004-04-21 devnull static void
52 551445b9 2004-04-21 devnull sunrpcproc(void *v)
54 551445b9 2004-04-21 devnull threadcreate(sunrpcreplythread, v, SunStackSize);
55 551445b9 2004-04-21 devnull threadcreate(sunrpcrequestthread, v, SunStackSize);
56 551445b9 2004-04-21 devnull threadcreate(sunrpcforkthread, v, SunStackSize);
60 551445b9 2004-04-21 devnull static void
61 551445b9 2004-04-21 devnull sunrpcforkthread(void *v)
63 551445b9 2004-04-21 devnull SunSrv *srv = v;
66 551445b9 2004-04-21 devnull while(recv(srv->cthread, &t) == 1)
67 551445b9 2004-04-21 devnull threadcreate(t.fn, t.arg, SunStackSize);
71 551445b9 2004-04-21 devnull sunsrvthreadcreate(SunSrv *srv, void (*fn)(void*), void *arg)
75 551445b9 2004-04-21 devnull t.fn = fn;
76 551445b9 2004-04-21 devnull t.arg = arg;
77 551445b9 2004-04-21 devnull send(srv->cthread, &t);
80 551445b9 2004-04-21 devnull static void
81 551445b9 2004-04-21 devnull sunrpcrequestthread(void *v)
83 4c06b8ee 2005-05-19 devnull int status;
84 551445b9 2004-04-21 devnull uchar *p, *ep;
85 551445b9 2004-04-21 devnull Channel *c;
86 551445b9 2004-04-21 devnull SunSrv *srv = v;
87 551445b9 2004-04-21 devnull SunMsg *m;
88 551445b9 2004-04-21 devnull SunProg *pg;
89 551445b9 2004-04-21 devnull SunStatus ok;
91 551445b9 2004-04-21 devnull while((m = recvp(srv->crequest)) != nil){
92 551445b9 2004-04-21 devnull /* could look up in cache here? */
94 551445b9 2004-04-21 devnull if(srv->chatty) fprint(2, "sun msg %p count %d\n", m, m->count);
95 551445b9 2004-04-21 devnull m->srv = srv;
96 551445b9 2004-04-21 devnull p = m->data;
97 551445b9 2004-04-21 devnull ep = p+m->count;
98 4c06b8ee 2005-05-19 devnull status = m->rpc.status;
99 551445b9 2004-04-21 devnull if(sunrpcunpack(p, ep, &p, &m->rpc) != SunSuccess){
100 551445b9 2004-04-21 devnull fprint(2, "in: %.*H unpack failed\n", m->count, m->data);
101 551445b9 2004-04-21 devnull sunmsgdrop(m);
102 551445b9 2004-04-21 devnull continue;
104 551445b9 2004-04-21 devnull if(srv->chatty)
105 551445b9 2004-04-21 devnull fprint(2, "in: %B\n", &m->rpc);
106 4c06b8ee 2005-05-19 devnull if(status){
107 4c06b8ee 2005-05-19 devnull sunmsgreplyerror(m, status);
108 4c06b8ee 2005-05-19 devnull continue;
110 551445b9 2004-04-21 devnull if(srv->alwaysreject){
111 551445b9 2004-04-21 devnull if(srv->chatty)
112 551445b9 2004-04-21 devnull fprint(2, "\trejecting\n");
113 551445b9 2004-04-21 devnull sunmsgreplyerror(m, SunAuthTooWeak);
114 551445b9 2004-04-21 devnull continue;
117 551445b9 2004-04-21 devnull if(!m->rpc.iscall){
118 551445b9 2004-04-21 devnull sunmsgreplyerror(m, SunGarbageArgs);
119 551445b9 2004-04-21 devnull continue;
122 551445b9 2004-04-21 devnull if((pg = sunfindprog(srv, m, &m->rpc, &c)) == nil){
123 551445b9 2004-04-21 devnull /* sunfindprog sent error */
124 551445b9 2004-04-21 devnull continue;
127 551445b9 2004-04-21 devnull p = m->rpc.data;
128 551445b9 2004-04-21 devnull ep = p+m->rpc.ndata;
129 551445b9 2004-04-21 devnull m->call = nil;
130 551445b9 2004-04-21 devnull if((ok = suncallunpackalloc(pg, m->rpc.proc<<1, p, ep, &p, &m->call)) != SunSuccess){
131 551445b9 2004-04-21 devnull sunmsgreplyerror(m, ok);
132 551445b9 2004-04-21 devnull continue;
134 551445b9 2004-04-21 devnull m->call->rpc = m->rpc;
136 551445b9 2004-04-21 devnull if(srv->chatty)
137 551445b9 2004-04-21 devnull fprint(2, "\t%C\n", m->call);
139 551445b9 2004-04-21 devnull m->pg = pg;
140 551445b9 2004-04-21 devnull sendp(c, m);
144 551445b9 2004-04-21 devnull static SunProg*
145 551445b9 2004-04-21 devnull sunfindprog(SunSrv *srv, SunMsg *m, SunRpc *rpc, Channel **pc)
147 886a6f6c 2006-05-04 devnull int i, vlo, vhi, any;
148 551445b9 2004-04-21 devnull SunProg *pg;
150 886a6f6c 2006-05-04 devnull vlo = 0;
151 886a6f6c 2006-05-04 devnull vhi = 0;
152 886a6f6c 2006-05-04 devnull any = 0;
154 551445b9 2004-04-21 devnull for(i=0; i<srv->nprog; i++){
155 551445b9 2004-04-21 devnull pg = srv->prog[i];
156 551445b9 2004-04-21 devnull if(pg->prog != rpc->prog)
157 551445b9 2004-04-21 devnull continue;
158 551445b9 2004-04-21 devnull if(pg->vers == rpc->vers){
159 551445b9 2004-04-21 devnull *pc = srv->cdispatch[i];
160 551445b9 2004-04-21 devnull return pg;
162 551445b9 2004-04-21 devnull /* right program, wrong version: record range */
163 886a6f6c 2006-05-04 devnull if(!any++){
164 551445b9 2004-04-21 devnull vlo = pg->vers;
165 551445b9 2004-04-21 devnull vhi = pg->vers;
167 886a6f6c 2006-05-04 devnull if(pg->vers < vlo)
168 886a6f6c 2006-05-04 devnull vlo = pg->vers;
169 886a6f6c 2006-05-04 devnull if(pg->vers > vhi)
170 886a6f6c 2006-05-04 devnull vhi = pg->vers;
173 551445b9 2004-04-21 devnull if(vhi == -1){
174 551445b9 2004-04-21 devnull if(srv->chatty)
175 551445b9 2004-04-21 devnull fprint(2, "\tprogram %ud unavailable\n", rpc->prog);
176 551445b9 2004-04-21 devnull sunmsgreplyerror(m, SunProgUnavail);
178 551445b9 2004-04-21 devnull /* putting these in rpc is a botch */
179 551445b9 2004-04-21 devnull rpc->low = vlo;
180 551445b9 2004-04-21 devnull rpc->high = vhi;
181 551445b9 2004-04-21 devnull if(srv->chatty)
182 551445b9 2004-04-21 devnull fprint(2, "\tversion %ud unavailable; have %d-%d\n", rpc->vers, vlo, vhi);
183 551445b9 2004-04-21 devnull sunmsgreplyerror(m, SunProgMismatch);
185 551445b9 2004-04-21 devnull return nil;
188 551445b9 2004-04-21 devnull static void
189 551445b9 2004-04-21 devnull sunrpcreplythread(void *v)
191 551445b9 2004-04-21 devnull SunMsg *m;
192 551445b9 2004-04-21 devnull SunSrv *srv = v;
194 551445b9 2004-04-21 devnull while((m = recvp(srv->creply)) != nil){
195 551445b9 2004-04-21 devnull /* could record in cache here? */
196 551445b9 2004-04-21 devnull sendp(m->creply, m);
201 551445b9 2004-04-21 devnull sunmsgreplyerror(SunMsg *m, SunStatus error)
203 551445b9 2004-04-21 devnull uchar *p, *bp, *ep;
206 551445b9 2004-04-21 devnull m->rpc.status = error;
207 551445b9 2004-04-21 devnull m->rpc.iscall = 0;
208 551445b9 2004-04-21 devnull m->rpc.verf.flavor = SunAuthNone;
209 551445b9 2004-04-21 devnull m->rpc.data = nil;
210 551445b9 2004-04-21 devnull m->rpc.ndata = 0;
212 551445b9 2004-04-21 devnull if(m->srv->chatty)
213 551445b9 2004-04-21 devnull fprint(2, "out: %B\n", &m->rpc);
215 551445b9 2004-04-21 devnull n = sunrpcsize(&m->rpc);
216 551445b9 2004-04-21 devnull bp = emalloc(n);
217 551445b9 2004-04-21 devnull ep = bp+n;
219 551445b9 2004-04-21 devnull if(sunrpcpack(p, ep, &p, &m->rpc) < 0){
220 551445b9 2004-04-21 devnull fprint(2, "sunrpcpack failed\n");
221 551445b9 2004-04-21 devnull sunmsgdrop(m);
222 551445b9 2004-04-21 devnull return 0;
224 551445b9 2004-04-21 devnull if(p != ep){
225 551445b9 2004-04-21 devnull fprint(2, "sunmsgreplyerror: rpc sizes didn't work out\n");
226 551445b9 2004-04-21 devnull sunmsgdrop(m);
227 551445b9 2004-04-21 devnull return 0;
229 551445b9 2004-04-21 devnull free(m->data);
230 551445b9 2004-04-21 devnull m->data = bp;
231 551445b9 2004-04-21 devnull m->count = n;
232 551445b9 2004-04-21 devnull sendp(m->srv->creply, m);
233 551445b9 2004-04-21 devnull return 0;
237 551445b9 2004-04-21 devnull sunmsgreply(SunMsg *m, SunCall *c)
239 551445b9 2004-04-21 devnull int n1, n2;
240 551445b9 2004-04-21 devnull uchar *bp, *p, *ep;
242 551445b9 2004-04-21 devnull c->type = m->call->type+1;
243 551445b9 2004-04-21 devnull c->rpc.iscall = 0;
244 551445b9 2004-04-21 devnull c->rpc.prog = m->rpc.prog;
245 551445b9 2004-04-21 devnull c->rpc.vers = m->rpc.vers;
246 551445b9 2004-04-21 devnull c->rpc.proc = m->rpc.proc;
247 551445b9 2004-04-21 devnull c->rpc.xid = m->rpc.xid;
249 551445b9 2004-04-21 devnull if(m->srv->chatty){
250 551445b9 2004-04-21 devnull fprint(2, "out: %B\n", &c->rpc);
251 551445b9 2004-04-21 devnull fprint(2, "\t%C\n", c);
254 551445b9 2004-04-21 devnull n1 = sunrpcsize(&c->rpc);
255 551445b9 2004-04-21 devnull n2 = suncallsize(m->pg, c);
257 551445b9 2004-04-21 devnull bp = emalloc(n1+n2);
258 551445b9 2004-04-21 devnull ep = bp+n1+n2;
260 551445b9 2004-04-21 devnull if(sunrpcpack(p, ep, &p, &c->rpc) != SunSuccess){
261 551445b9 2004-04-21 devnull fprint(2, "sunrpcpack failed\n");
262 551445b9 2004-04-21 devnull return sunmsgdrop(m);
264 551445b9 2004-04-21 devnull if(suncallpack(m->pg, p, ep, &p, c) != SunSuccess){
265 551445b9 2004-04-21 devnull fprint(2, "pg->pack failed\n");
266 551445b9 2004-04-21 devnull return sunmsgdrop(m);
268 551445b9 2004-04-21 devnull if(p != ep){
269 551445b9 2004-04-21 devnull fprint(2, "sunmsgreply: sizes didn't work out\n");
270 551445b9 2004-04-21 devnull return sunmsgdrop(m);
272 551445b9 2004-04-21 devnull free(m->data);
273 551445b9 2004-04-21 devnull m->data = bp;
274 551445b9 2004-04-21 devnull m->count = n1+n2;
276 551445b9 2004-04-21 devnull sendp(m->srv->creply, m);
277 551445b9 2004-04-21 devnull return 0;
281 551445b9 2004-04-21 devnull sunmsgdrop(SunMsg *m)
283 551445b9 2004-04-21 devnull free(m->data);
284 551445b9 2004-04-21 devnull free(m->call);
285 551445b9 2004-04-21 devnull memset(m, 0xFB, sizeof *m);
286 551445b9 2004-04-21 devnull free(m);
287 551445b9 2004-04-21 devnull return 0;