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 <thread.h>
  4 551445b9 2004-04-21 devnull #include <sunrpc.h>
  5 551445b9 2004-04-21 devnull 
  6 551445b9 2004-04-21 devnull /*
  7 551445b9 2004-04-21 devnull  * Sun RPC server; for now, no reply cache
  8 551445b9 2004-04-21 devnull  */
  9 551445b9 2004-04-21 devnull 
 10 551445b9 2004-04-21 devnull static void sunrpcproc(void*);
 11 551445b9 2004-04-21 devnull static void sunrpcrequestthread(void*);
 12 551445b9 2004-04-21 devnull static void sunrpcreplythread(void*);
 13 551445b9 2004-04-21 devnull static void sunrpcforkthread(void*);
 14 551445b9 2004-04-21 devnull static SunProg *sunfindprog(SunSrv*, SunMsg*, SunRpc*, Channel**);
 15 551445b9 2004-04-21 devnull 
 16 551445b9 2004-04-21 devnull typedef struct Targ Targ;
 17 551445b9 2004-04-21 devnull struct Targ
 18 551445b9 2004-04-21 devnull {
 19 551445b9 2004-04-21 devnull 	void (*fn)(void*);
 20 551445b9 2004-04-21 devnull 	void *arg;
 21 551445b9 2004-04-21 devnull };
 22 551445b9 2004-04-21 devnull 
 23 551445b9 2004-04-21 devnull SunSrv*
 24 551445b9 2004-04-21 devnull sunsrv(void)
 25 551445b9 2004-04-21 devnull {
 26 551445b9 2004-04-21 devnull 	SunSrv *srv;
 27 551445b9 2004-04-21 devnull 
 28 551445b9 2004-04-21 devnull 	srv = emalloc(sizeof(SunSrv));
 29 551445b9 2004-04-21 devnull 	srv->chatty = 0;
 30 551445b9 2004-04-21 devnull 	srv->crequest = chancreate(sizeof(SunMsg*), 16);
 31 551445b9 2004-04-21 devnull 	srv->creply = chancreate(sizeof(SunMsg*), 16);
 32 551445b9 2004-04-21 devnull 	srv->cthread = chancreate(sizeof(Targ), 4);
 33 551445b9 2004-04-21 devnull 
 34 551445b9 2004-04-21 devnull 	proccreate(sunrpcproc, srv, SunStackSize);
 35 551445b9 2004-04-21 devnull 	return srv;
 36 551445b9 2004-04-21 devnull }
 37 551445b9 2004-04-21 devnull 
 38 551445b9 2004-04-21 devnull void
 39 551445b9 2004-04-21 devnull sunsrvprog(SunSrv *srv, SunProg *prog, Channel *c)
 40 551445b9 2004-04-21 devnull {
 41 551445b9 2004-04-21 devnull 	if(srv->nprog%16 == 0){
 42 551445b9 2004-04-21 devnull 		srv->prog = erealloc(srv->prog, (srv->nprog+16)*sizeof(srv->prog[0]));
 43 551445b9 2004-04-21 devnull 		srv->cdispatch = erealloc(srv->cdispatch, (srv->nprog+16)*sizeof(srv->cdispatch[0]));
 44 551445b9 2004-04-21 devnull 	}
 45 551445b9 2004-04-21 devnull 	srv->prog[srv->nprog] = prog;
 46 551445b9 2004-04-21 devnull 	srv->cdispatch[srv->nprog] = c;
 47 551445b9 2004-04-21 devnull 	srv->nprog++;
 48 551445b9 2004-04-21 devnull }
 49 551445b9 2004-04-21 devnull 
 50 551445b9 2004-04-21 devnull static void
 51 551445b9 2004-04-21 devnull sunrpcproc(void *v)
 52 551445b9 2004-04-21 devnull {
 53 551445b9 2004-04-21 devnull 	threadcreate(sunrpcreplythread, v, SunStackSize);
 54 551445b9 2004-04-21 devnull 	threadcreate(sunrpcrequestthread, v, SunStackSize);
 55 551445b9 2004-04-21 devnull 	threadcreate(sunrpcforkthread, v, SunStackSize);
 56 551445b9 2004-04-21 devnull 
 57 551445b9 2004-04-21 devnull }
 58 551445b9 2004-04-21 devnull 
 59 551445b9 2004-04-21 devnull static void
 60 551445b9 2004-04-21 devnull sunrpcforkthread(void *v)
 61 551445b9 2004-04-21 devnull {
 62 551445b9 2004-04-21 devnull 	SunSrv *srv = v;
 63 551445b9 2004-04-21 devnull 	Targ t;
 64 551445b9 2004-04-21 devnull 
 65 551445b9 2004-04-21 devnull 	while(recv(srv->cthread, &t) == 1)
 66 551445b9 2004-04-21 devnull 		threadcreate(t.fn, t.arg, SunStackSize);
 67 551445b9 2004-04-21 devnull }
 68 551445b9 2004-04-21 devnull 
 69 551445b9 2004-04-21 devnull void
 70 551445b9 2004-04-21 devnull sunsrvthreadcreate(SunSrv *srv, void (*fn)(void*), void *arg)
 71 551445b9 2004-04-21 devnull {
 72 551445b9 2004-04-21 devnull 	Targ t;
 73 551445b9 2004-04-21 devnull 
 74 551445b9 2004-04-21 devnull 	t.fn = fn;
 75 551445b9 2004-04-21 devnull 	t.arg = arg;
 76 551445b9 2004-04-21 devnull 	send(srv->cthread, &t);
 77 551445b9 2004-04-21 devnull }
 78 551445b9 2004-04-21 devnull 
 79 551445b9 2004-04-21 devnull static void
 80 551445b9 2004-04-21 devnull sunrpcrequestthread(void *v)
 81 551445b9 2004-04-21 devnull {
 82 551445b9 2004-04-21 devnull 	uchar *p, *ep;
 83 551445b9 2004-04-21 devnull 	Channel *c;
 84 551445b9 2004-04-21 devnull 	SunSrv *srv = v;
 85 551445b9 2004-04-21 devnull 	SunMsg *m;
 86 551445b9 2004-04-21 devnull 	SunProg *pg;
 87 551445b9 2004-04-21 devnull 	SunStatus ok;
 88 551445b9 2004-04-21 devnull 
 89 551445b9 2004-04-21 devnull 	while((m = recvp(srv->crequest)) != nil){
 90 551445b9 2004-04-21 devnull 		/* could look up in cache here? */
 91 551445b9 2004-04-21 devnull 
 92 551445b9 2004-04-21 devnull if(srv->chatty) fprint(2, "sun msg %p count %d\n", m, m->count);
 93 551445b9 2004-04-21 devnull 		m->srv = srv;
 94 551445b9 2004-04-21 devnull 		p = m->data;
 95 551445b9 2004-04-21 devnull 		ep = p+m->count;
 96 551445b9 2004-04-21 devnull 		if(sunrpcunpack(p, ep, &p, &m->rpc) != SunSuccess){
 97 551445b9 2004-04-21 devnull 			fprint(2, "in: %.*H unpack failed\n", m->count, m->data);
 98 551445b9 2004-04-21 devnull 			sunmsgdrop(m);
 99 551445b9 2004-04-21 devnull 			continue;
100 551445b9 2004-04-21 devnull 		}
101 551445b9 2004-04-21 devnull 		if(srv->chatty)
102 551445b9 2004-04-21 devnull 			fprint(2, "in: %B\n", &m->rpc);
103 551445b9 2004-04-21 devnull 
104 551445b9 2004-04-21 devnull 		if(srv->alwaysreject){
105 551445b9 2004-04-21 devnull 			if(srv->chatty)
106 551445b9 2004-04-21 devnull 				fprint(2, "\trejecting\n");
107 551445b9 2004-04-21 devnull 			sunmsgreplyerror(m, SunAuthTooWeak);
108 551445b9 2004-04-21 devnull 			continue;
109 551445b9 2004-04-21 devnull 		}
110 551445b9 2004-04-21 devnull 
111 551445b9 2004-04-21 devnull 		if(!m->rpc.iscall){
112 551445b9 2004-04-21 devnull 			sunmsgreplyerror(m, SunGarbageArgs);
113 551445b9 2004-04-21 devnull 			continue;
114 551445b9 2004-04-21 devnull 		}
115 551445b9 2004-04-21 devnull 
116 551445b9 2004-04-21 devnull 		if((pg = sunfindprog(srv, m, &m->rpc, &c)) == nil){
117 551445b9 2004-04-21 devnull 			/* sunfindprog sent error */
118 551445b9 2004-04-21 devnull 			continue;
119 551445b9 2004-04-21 devnull 		}
120 551445b9 2004-04-21 devnull 
121 551445b9 2004-04-21 devnull 		p = m->rpc.data;
122 551445b9 2004-04-21 devnull 		ep = p+m->rpc.ndata;
123 551445b9 2004-04-21 devnull 		m->call = nil;
124 551445b9 2004-04-21 devnull 		if((ok = suncallunpackalloc(pg, m->rpc.proc<<1, p, ep, &p, &m->call)) != SunSuccess){
125 551445b9 2004-04-21 devnull 			sunmsgreplyerror(m, ok);
126 551445b9 2004-04-21 devnull 			continue;
127 551445b9 2004-04-21 devnull 		}
128 551445b9 2004-04-21 devnull 		m->call->rpc = m->rpc;
129 551445b9 2004-04-21 devnull 
130 551445b9 2004-04-21 devnull 		if(srv->chatty)
131 551445b9 2004-04-21 devnull 			fprint(2, "\t%C\n", m->call);
132 551445b9 2004-04-21 devnull 
133 551445b9 2004-04-21 devnull 		m->pg = pg;
134 551445b9 2004-04-21 devnull 		sendp(c, m);
135 551445b9 2004-04-21 devnull 	}
136 551445b9 2004-04-21 devnull }
137 551445b9 2004-04-21 devnull 
138 551445b9 2004-04-21 devnull static SunProg*
139 551445b9 2004-04-21 devnull sunfindprog(SunSrv *srv, SunMsg *m, SunRpc *rpc, Channel **pc)
140 551445b9 2004-04-21 devnull {
141 551445b9 2004-04-21 devnull 	int i, vlo, vhi;
142 551445b9 2004-04-21 devnull 	SunProg *pg;
143 551445b9 2004-04-21 devnull 
144 551445b9 2004-04-21 devnull 	vlo = 0x7fffffff;
145 551445b9 2004-04-21 devnull 	vhi = -1;
146 551445b9 2004-04-21 devnull 
147 551445b9 2004-04-21 devnull 	for(i=0; i<srv->nprog; i++){
148 551445b9 2004-04-21 devnull 		pg = srv->prog[i];
149 551445b9 2004-04-21 devnull 		if(pg->prog != rpc->prog)
150 551445b9 2004-04-21 devnull 			continue;
151 551445b9 2004-04-21 devnull 		if(pg->vers == rpc->vers){
152 551445b9 2004-04-21 devnull 			*pc = srv->cdispatch[i];
153 551445b9 2004-04-21 devnull 			return pg;
154 551445b9 2004-04-21 devnull 		}
155 551445b9 2004-04-21 devnull 		/* right program, wrong version: record range */
156 551445b9 2004-04-21 devnull 		if(pg->vers < vlo)
157 551445b9 2004-04-21 devnull 			vlo = pg->vers;
158 551445b9 2004-04-21 devnull 		if(pg->vers > vhi)
159 551445b9 2004-04-21 devnull 			vhi = pg->vers;
160 551445b9 2004-04-21 devnull 	}
161 551445b9 2004-04-21 devnull 	if(vhi == -1){
162 551445b9 2004-04-21 devnull 		if(srv->chatty)
163 551445b9 2004-04-21 devnull 			fprint(2, "\tprogram %ud unavailable\n", rpc->prog);
164 551445b9 2004-04-21 devnull 		sunmsgreplyerror(m, SunProgUnavail);
165 551445b9 2004-04-21 devnull 	}else{
166 551445b9 2004-04-21 devnull 		/* putting these in rpc is a botch */
167 551445b9 2004-04-21 devnull 		rpc->low = vlo;
168 551445b9 2004-04-21 devnull 		rpc->high = vhi;
169 551445b9 2004-04-21 devnull 		if(srv->chatty)
170 551445b9 2004-04-21 devnull 			fprint(2, "\tversion %ud unavailable; have %d-%d\n", rpc->vers, vlo, vhi);
171 551445b9 2004-04-21 devnull 		sunmsgreplyerror(m, SunProgMismatch);
172 551445b9 2004-04-21 devnull 	}
173 551445b9 2004-04-21 devnull 	return nil;
174 551445b9 2004-04-21 devnull }
175 551445b9 2004-04-21 devnull 
176 551445b9 2004-04-21 devnull static void
177 551445b9 2004-04-21 devnull sunrpcreplythread(void *v)
178 551445b9 2004-04-21 devnull {
179 551445b9 2004-04-21 devnull 	SunMsg *m;
180 551445b9 2004-04-21 devnull 	SunSrv *srv = v;
181 551445b9 2004-04-21 devnull 
182 551445b9 2004-04-21 devnull 	while((m = recvp(srv->creply)) != nil){
183 551445b9 2004-04-21 devnull 		/* could record in cache here? */
184 551445b9 2004-04-21 devnull 		sendp(m->creply, m);
185 551445b9 2004-04-21 devnull 	}	
186 551445b9 2004-04-21 devnull }
187 551445b9 2004-04-21 devnull 
188 551445b9 2004-04-21 devnull int
189 551445b9 2004-04-21 devnull sunmsgreplyerror(SunMsg *m, SunStatus error)
190 551445b9 2004-04-21 devnull {
191 551445b9 2004-04-21 devnull 	uchar *p, *bp, *ep;
192 551445b9 2004-04-21 devnull 	int n;
193 551445b9 2004-04-21 devnull 
194 551445b9 2004-04-21 devnull 	m->rpc.status = error;
195 551445b9 2004-04-21 devnull 	m->rpc.iscall = 0;
196 551445b9 2004-04-21 devnull 	m->rpc.verf.flavor = SunAuthNone;
197 551445b9 2004-04-21 devnull 	m->rpc.data = nil;
198 551445b9 2004-04-21 devnull 	m->rpc.ndata = 0;
199 551445b9 2004-04-21 devnull 
200 551445b9 2004-04-21 devnull 	if(m->srv->chatty)
201 551445b9 2004-04-21 devnull 		fprint(2, "out: %B\n", &m->rpc);
202 551445b9 2004-04-21 devnull 
203 551445b9 2004-04-21 devnull 	n = sunrpcsize(&m->rpc);
204 551445b9 2004-04-21 devnull 	bp = emalloc(n);
205 551445b9 2004-04-21 devnull 	ep = bp+n;
206 551445b9 2004-04-21 devnull 	p = bp;
207 551445b9 2004-04-21 devnull 	if(sunrpcpack(p, ep, &p, &m->rpc) < 0){
208 551445b9 2004-04-21 devnull 		fprint(2, "sunrpcpack failed\n");
209 551445b9 2004-04-21 devnull 		sunmsgdrop(m);
210 551445b9 2004-04-21 devnull 		return 0;
211 551445b9 2004-04-21 devnull 	}
212 551445b9 2004-04-21 devnull 	if(p != ep){
213 551445b9 2004-04-21 devnull 		fprint(2, "sunmsgreplyerror: rpc sizes didn't work out\n");
214 551445b9 2004-04-21 devnull 		sunmsgdrop(m);
215 551445b9 2004-04-21 devnull 		return 0;
216 551445b9 2004-04-21 devnull 	}
217 551445b9 2004-04-21 devnull 	free(m->data);
218 551445b9 2004-04-21 devnull 	m->data = bp;
219 551445b9 2004-04-21 devnull 	m->count = n;
220 551445b9 2004-04-21 devnull 	sendp(m->srv->creply, m);
221 551445b9 2004-04-21 devnull 	return 0;
222 551445b9 2004-04-21 devnull }
223 551445b9 2004-04-21 devnull 
224 551445b9 2004-04-21 devnull int
225 551445b9 2004-04-21 devnull sunmsgreply(SunMsg *m, SunCall *c)
226 551445b9 2004-04-21 devnull {
227 551445b9 2004-04-21 devnull 	int n1, n2;
228 551445b9 2004-04-21 devnull 	uchar *bp, *p, *ep;
229 551445b9 2004-04-21 devnull 
230 551445b9 2004-04-21 devnull 	c->type = m->call->type+1;
231 551445b9 2004-04-21 devnull 	c->rpc.iscall = 0;
232 551445b9 2004-04-21 devnull 	c->rpc.prog = m->rpc.prog;
233 551445b9 2004-04-21 devnull 	c->rpc.vers = m->rpc.vers;
234 551445b9 2004-04-21 devnull 	c->rpc.proc = m->rpc.proc;
235 551445b9 2004-04-21 devnull 	c->rpc.xid = m->rpc.xid;
236 551445b9 2004-04-21 devnull 
237 551445b9 2004-04-21 devnull 	if(m->srv->chatty){
238 551445b9 2004-04-21 devnull 		fprint(2, "out: %B\n", &c->rpc);
239 551445b9 2004-04-21 devnull 		fprint(2, "\t%C\n", c);
240 551445b9 2004-04-21 devnull 	}
241 551445b9 2004-04-21 devnull 
242 551445b9 2004-04-21 devnull 	n1 = sunrpcsize(&c->rpc);
243 551445b9 2004-04-21 devnull 	n2 = suncallsize(m->pg, c);
244 551445b9 2004-04-21 devnull 
245 551445b9 2004-04-21 devnull 	bp = emalloc(n1+n2);
246 551445b9 2004-04-21 devnull 	ep = bp+n1+n2;
247 551445b9 2004-04-21 devnull 	p = bp;
248 551445b9 2004-04-21 devnull 	if(sunrpcpack(p, ep, &p, &c->rpc) != SunSuccess){
249 551445b9 2004-04-21 devnull 		fprint(2, "sunrpcpack failed\n");
250 551445b9 2004-04-21 devnull 		return sunmsgdrop(m);
251 551445b9 2004-04-21 devnull 	}
252 551445b9 2004-04-21 devnull 	if(suncallpack(m->pg, p, ep, &p, c) != SunSuccess){
253 551445b9 2004-04-21 devnull 		fprint(2, "pg->pack failed\n");
254 551445b9 2004-04-21 devnull 		return sunmsgdrop(m);
255 551445b9 2004-04-21 devnull 	}
256 551445b9 2004-04-21 devnull 	if(p != ep){
257 551445b9 2004-04-21 devnull 		fprint(2, "sunmsgreply: sizes didn't work out\n");
258 551445b9 2004-04-21 devnull 		return sunmsgdrop(m);
259 551445b9 2004-04-21 devnull 	}
260 551445b9 2004-04-21 devnull 	free(m->data);
261 551445b9 2004-04-21 devnull 	m->data = bp;
262 551445b9 2004-04-21 devnull 	m->count = n1+n2;
263 551445b9 2004-04-21 devnull 
264 551445b9 2004-04-21 devnull 	sendp(m->srv->creply, m);
265 551445b9 2004-04-21 devnull 	return 0;
266 551445b9 2004-04-21 devnull }
267 551445b9 2004-04-21 devnull 
268 551445b9 2004-04-21 devnull int
269 551445b9 2004-04-21 devnull sunmsgdrop(SunMsg *m)
270 551445b9 2004-04-21 devnull {
271 551445b9 2004-04-21 devnull 	free(m->data);
272 551445b9 2004-04-21 devnull 	free(m->call);
273 551445b9 2004-04-21 devnull 	memset(m, 0xFB, sizeof *m);
274 551445b9 2004-04-21 devnull 	free(m);
275 551445b9 2004-04-21 devnull 	return 0;
276 551445b9 2004-04-21 devnull }
277 551445b9 2004-04-21 devnull