1 056fe1ba 2003-11-23 devnull #include <u.h>
  2 056fe1ba 2003-11-23 devnull #include <libc.h>
  3 056fe1ba 2003-11-23 devnull #include <venti.h>
  4 056fe1ba 2003-11-23 devnull #include <thread.h>
  5 056fe1ba 2003-11-23 devnull #include "queue.h"
  6 056fe1ba 2003-11-23 devnull 
  7 056fe1ba 2003-11-23 devnull enum
  8 056fe1ba 2003-11-23 devnull {
  9 056fe1ba 2003-11-23 devnull 	STACK = 8192,
 10 056fe1ba 2003-11-23 devnull };
 11 056fe1ba 2003-11-23 devnull 
 12 056fe1ba 2003-11-23 devnull typedef struct VtSconn VtSconn;
 13 056fe1ba 2003-11-23 devnull struct VtSconn
 14 056fe1ba 2003-11-23 devnull {
 15 056fe1ba 2003-11-23 devnull 	int ctl;
 16 056fe1ba 2003-11-23 devnull 	char dir[NETPATHLEN];
 17 056fe1ba 2003-11-23 devnull 	VtSrv *srv;
 18 056fe1ba 2003-11-23 devnull 	VtConn *c;
 19 056fe1ba 2003-11-23 devnull };
 20 056fe1ba 2003-11-23 devnull 
 21 056fe1ba 2003-11-23 devnull struct VtSrv
 22 056fe1ba 2003-11-23 devnull {
 23 056fe1ba 2003-11-23 devnull 	int afd;
 24 056fe1ba 2003-11-23 devnull 	int dead;
 25 056fe1ba 2003-11-23 devnull 	char adir[NETPATHLEN];
 26 056fe1ba 2003-11-23 devnull 	Queue *q;	/* Queue(VtReq*) */
 27 056fe1ba 2003-11-23 devnull };
 28 056fe1ba 2003-11-23 devnull 
 29 056fe1ba 2003-11-23 devnull static void listenproc(void*);
 30 056fe1ba 2003-11-23 devnull static void connproc(void*);
 31 056fe1ba 2003-11-23 devnull 
 32 056fe1ba 2003-11-23 devnull VtSrv*
 33 056fe1ba 2003-11-23 devnull vtlisten(char *addr)
 34 056fe1ba 2003-11-23 devnull {
 35 056fe1ba 2003-11-23 devnull 	VtSrv *s;
 36 056fe1ba 2003-11-23 devnull 
 37 056fe1ba 2003-11-23 devnull 	s = vtmallocz(sizeof(VtSrv));
 38 056fe1ba 2003-11-23 devnull 	s->afd = announce(addr, s->adir);
 39 056fe1ba 2003-11-23 devnull 	if(s->afd < 0){
 40 056fe1ba 2003-11-23 devnull 		free(s);
 41 056fe1ba 2003-11-23 devnull 		return nil;
 42 056fe1ba 2003-11-23 devnull 	}
 43 056fe1ba 2003-11-23 devnull 	s->q = _vtqalloc();
 44 056fe1ba 2003-11-23 devnull 	proccreate(listenproc, s, STACK);
 45 056fe1ba 2003-11-23 devnull 	return s;
 46 056fe1ba 2003-11-23 devnull }
 47 056fe1ba 2003-11-23 devnull 
 48 056fe1ba 2003-11-23 devnull static void
 49 056fe1ba 2003-11-23 devnull listenproc(void *v)
 50 056fe1ba 2003-11-23 devnull {
 51 056fe1ba 2003-11-23 devnull 	int ctl;
 52 056fe1ba 2003-11-23 devnull 	char dir[NETPATHLEN];
 53 056fe1ba 2003-11-23 devnull 	VtSrv *srv;
 54 056fe1ba 2003-11-23 devnull 	VtSconn *sc;
 55 056fe1ba 2003-11-23 devnull 
 56 056fe1ba 2003-11-23 devnull 	srv = v;
 57 056fe1ba 2003-11-23 devnull 	for(;;){
 58 056fe1ba 2003-11-23 devnull 		ctl = listen(srv->adir, dir);
 59 056fe1ba 2003-11-23 devnull 		if(ctl < 0){
 60 056fe1ba 2003-11-23 devnull 			srv->dead = 1;
 61 056fe1ba 2003-11-23 devnull 			break;
 62 056fe1ba 2003-11-23 devnull 		}
 63 056fe1ba 2003-11-23 devnull 		sc = vtmallocz(sizeof(VtSconn));
 64 056fe1ba 2003-11-23 devnull 		sc->ctl = ctl;
 65 056fe1ba 2003-11-23 devnull 		sc->srv = srv;
 66 056fe1ba 2003-11-23 devnull 		strcpy(sc->dir, dir);
 67 056fe1ba 2003-11-23 devnull 		proccreate(connproc, sc, STACK);
 68 056fe1ba 2003-11-23 devnull 	}
 69 056fe1ba 2003-11-23 devnull 
 70 056fe1ba 2003-11-23 devnull 	// hangup
 71 056fe1ba 2003-11-23 devnull }
 72 056fe1ba 2003-11-23 devnull 
 73 056fe1ba 2003-11-23 devnull static void
 74 056fe1ba 2003-11-23 devnull connproc(void *v)
 75 056fe1ba 2003-11-23 devnull {
 76 056fe1ba 2003-11-23 devnull 	VtSconn *sc;
 77 056fe1ba 2003-11-23 devnull 	VtConn *c;
 78 056fe1ba 2003-11-23 devnull 	Packet *p;
 79 056fe1ba 2003-11-23 devnull 	VtReq *r;
 80 056fe1ba 2003-11-23 devnull 	int fd;
 81 056fe1ba 2003-11-23 devnull 
 82 056fe1ba 2003-11-23 devnull 	r = nil;
 83 056fe1ba 2003-11-23 devnull 	c = nil;
 84 056fe1ba 2003-11-23 devnull 	sc = v;
 85 056fe1ba 2003-11-23 devnull 	fprint(2, "new call %s on %d\n", sc->dir, sc->ctl);
 86 056fe1ba 2003-11-23 devnull 	fd = accept(sc->ctl, sc->dir);
 87 056fe1ba 2003-11-23 devnull 	close(sc->ctl);
 88 056fe1ba 2003-11-23 devnull 	if(fd < 0){
 89 056fe1ba 2003-11-23 devnull 		fprint(2, "accept %s: %r\n", sc->dir);
 90 056fe1ba 2003-11-23 devnull 		goto out;
 91 056fe1ba 2003-11-23 devnull 	}
 92 056fe1ba 2003-11-23 devnull 
 93 056fe1ba 2003-11-23 devnull 	c = vtconn(fd, fd);
 94 056fe1ba 2003-11-23 devnull 	sc->c = c;
 95 056fe1ba 2003-11-23 devnull 	if(vtversion(c) < 0){
 96 056fe1ba 2003-11-23 devnull 		fprint(2, "vtversion %s: %r\n", sc->dir);
 97 056fe1ba 2003-11-23 devnull 		goto out;
 98 056fe1ba 2003-11-23 devnull 	}
 99 056fe1ba 2003-11-23 devnull 	if(vtsrvhello(c) < 0){
100 056fe1ba 2003-11-23 devnull 		fprint(2, "vtsrvhello %s: %r\n", sc->dir);
101 056fe1ba 2003-11-23 devnull 		goto out;
102 056fe1ba 2003-11-23 devnull 	}
103 056fe1ba 2003-11-23 devnull 
104 056fe1ba 2003-11-23 devnull 	fprint(2, "new proc %s\n", sc->dir);
105 056fe1ba 2003-11-23 devnull 	proccreate(vtsendproc, c, STACK);
106 056fe1ba 2003-11-23 devnull 	qlock(&c->lk);
107 056fe1ba 2003-11-23 devnull 	while(!c->writeq)
108 056fe1ba 2003-11-23 devnull 		rsleep(&c->rpcfork);
109 056fe1ba 2003-11-23 devnull 	qunlock(&c->lk);
110 056fe1ba 2003-11-23 devnull 
111 056fe1ba 2003-11-23 devnull 	while((p = vtrecv(c)) != nil){
112 056fe1ba 2003-11-23 devnull 		r = vtmallocz(sizeof(VtReq));
113 056fe1ba 2003-11-23 devnull 		if(vtfcallunpack(&r->tx, p) < 0){
114 056fe1ba 2003-11-23 devnull 			packetfree(p);
115 056fe1ba 2003-11-23 devnull 			fprint(2, "bad packet on %s: %r\n", sc->dir);
116 056fe1ba 2003-11-23 devnull 			continue;
117 056fe1ba 2003-11-23 devnull 		}
118 056fe1ba 2003-11-23 devnull 		packetfree(p);
119 056fe1ba 2003-11-23 devnull 		if(r->tx.type == VtTgoodbye)
120 056fe1ba 2003-11-23 devnull 			break;
121 056fe1ba 2003-11-23 devnull 		r->rx.tag = r->tx.tag;
122 056fe1ba 2003-11-23 devnull 		r->sc = sc;
123 056fe1ba 2003-11-23 devnull 		if(_vtqsend(sc->srv->q, r) < 0){
124 056fe1ba 2003-11-23 devnull 			fprint(2, "hungup queue\n");
125 056fe1ba 2003-11-23 devnull 			break;
126 056fe1ba 2003-11-23 devnull 		}
127 056fe1ba 2003-11-23 devnull 		r = nil;
128 056fe1ba 2003-11-23 devnull 	}
129 056fe1ba 2003-11-23 devnull 
130 056fe1ba 2003-11-23 devnull 	fprint(2, "eof on %s\n", sc->dir);
131 056fe1ba 2003-11-23 devnull 
132 056fe1ba 2003-11-23 devnull out:
133 056fe1ba 2003-11-23 devnull 	if(r){
134 056fe1ba 2003-11-23 devnull 		vtfcallclear(&r->tx);
135 056fe1ba 2003-11-23 devnull 		vtfree(r);
136 056fe1ba 2003-11-23 devnull 	}
137 056fe1ba 2003-11-23 devnull 	if(c)
138 056fe1ba 2003-11-23 devnull 		vtfreeconn(c);
139 056fe1ba 2003-11-23 devnull 	fprint(2, "freed %s\n", sc->dir);
140 056fe1ba 2003-11-23 devnull 	vtfree(sc);
141 056fe1ba 2003-11-23 devnull 	return;
142 056fe1ba 2003-11-23 devnull }
143 056fe1ba 2003-11-23 devnull 
144 056fe1ba 2003-11-23 devnull VtReq*
145 056fe1ba 2003-11-23 devnull vtgetreq(VtSrv *srv)
146 056fe1ba 2003-11-23 devnull {
147 056fe1ba 2003-11-23 devnull 	return _vtqrecv(srv->q);
148 056fe1ba 2003-11-23 devnull }
149 056fe1ba 2003-11-23 devnull 
150 056fe1ba 2003-11-23 devnull void
151 056fe1ba 2003-11-23 devnull vtrespond(VtReq *r)
152 056fe1ba 2003-11-23 devnull {
153 056fe1ba 2003-11-23 devnull 	Packet *p;
154 056fe1ba 2003-11-23 devnull 	VtSconn *sc;
155 056fe1ba 2003-11-23 devnull 
156 056fe1ba 2003-11-23 devnull 	sc = r->sc;
157 056fe1ba 2003-11-23 devnull 	if(r->rx.tag != r->tx.tag)
158 056fe1ba 2003-11-23 devnull 		abort();
159 056fe1ba 2003-11-23 devnull 	if(r->rx.type != r->tx.type+1 && r->rx.type != VtRerror)
160 056fe1ba 2003-11-23 devnull 		abort();
161 056fe1ba 2003-11-23 devnull 	if((p = vtfcallpack(&r->rx)) == nil){
162 056fe1ba 2003-11-23 devnull 		fprint(2, "fcallpack on %s: %r\n", sc->dir);
163 056fe1ba 2003-11-23 devnull 		packetfree(p);
164 056fe1ba 2003-11-23 devnull 		vtfcallclear(&r->rx);
165 056fe1ba 2003-11-23 devnull 		return;
166 056fe1ba 2003-11-23 devnull 	}
167 056fe1ba 2003-11-23 devnull 	vtsend(sc->c, p);
168 056fe1ba 2003-11-23 devnull 	vtfcallclear(&r->tx);
169 056fe1ba 2003-11-23 devnull 	vtfcallclear(&r->rx);
170 056fe1ba 2003-11-23 devnull 	vtfree(r);
171 056fe1ba 2003-11-23 devnull }
172 056fe1ba 2003-11-23 devnull