1 d3df3087 2003-12-06 devnull /* Copyright (C) 2003 Russ Cox, Massachusetts Institute of Technology */
  2 d3df3087 2003-12-06 devnull /* See COPYRIGHT */
  3 d3df3087 2003-12-06 devnull 
  4 d3df3087 2003-12-06 devnull #include <u.h>
  5 d3df3087 2003-12-06 devnull #include <libc.h>
  6 d3df3087 2003-12-06 devnull #include <mux.h>
  7 d3df3087 2003-12-06 devnull 
  8 d3df3087 2003-12-06 devnull /*
  9 d3df3087 2003-12-06 devnull  * If you fork off two procs running muxrecvproc and muxsendproc,
 10 d3df3087 2003-12-06 devnull  * then muxrecv/muxsend (and thus muxrpc) will never block except on 
 11 d3df3087 2003-12-06 devnull  * rendevouses, which is nice when it's running in one thread of many.
 12 d3df3087 2003-12-06 devnull  */
 13 d3df3087 2003-12-06 devnull void
 14 d3df3087 2003-12-06 devnull _muxrecvproc(void *v)
 15 d3df3087 2003-12-06 devnull {
 16 d3df3087 2003-12-06 devnull 	void *p;
 17 d3df3087 2003-12-06 devnull 	Mux *mux;
 18 d3df3087 2003-12-06 devnull 	Muxqueue *q;
 19 d3df3087 2003-12-06 devnull 
 20 d3df3087 2003-12-06 devnull 	mux = v;
 21 d3df3087 2003-12-06 devnull 	q = _muxqalloc();
 22 d3df3087 2003-12-06 devnull 
 23 d3df3087 2003-12-06 devnull 	qlock(&mux->lk);
 24 d3df3087 2003-12-06 devnull 	mux->readq = q;
 25 d3df3087 2003-12-06 devnull 	qlock(&mux->inlk);
 26 d3df3087 2003-12-06 devnull 	rwakeup(&mux->rpcfork);
 27 d3df3087 2003-12-06 devnull 	qunlock(&mux->lk);
 28 d3df3087 2003-12-06 devnull 
 29 d3df3087 2003-12-06 devnull 	while((p = mux->recv(mux)) != nil)
 30 d3df3087 2003-12-06 devnull 		if(_muxqsend(q, p) < 0){
 31 d3df3087 2003-12-06 devnull 			free(p);
 32 d3df3087 2003-12-06 devnull 			break;
 33 d3df3087 2003-12-06 devnull 		}
 34 d3df3087 2003-12-06 devnull 	qunlock(&mux->inlk);
 35 d3df3087 2003-12-06 devnull 	qlock(&mux->lk);
 36 d3df3087 2003-12-06 devnull 	_muxqhangup(q);
 37 d3df3087 2003-12-06 devnull 	while((p = _muxnbqrecv(q)) != nil)
 38 d3df3087 2003-12-06 devnull 		free(p);
 39 d3df3087 2003-12-06 devnull 	free(q);
 40 d3df3087 2003-12-06 devnull 	mux->readq = nil;
 41 d3df3087 2003-12-06 devnull 	rwakeup(&mux->rpcfork);
 42 d3df3087 2003-12-06 devnull 	qunlock(&mux->lk);
 43 d3df3087 2003-12-06 devnull }
 44 d3df3087 2003-12-06 devnull 
 45 d3df3087 2003-12-06 devnull void
 46 d3df3087 2003-12-06 devnull _muxsendproc(void *v)
 47 d3df3087 2003-12-06 devnull {
 48 d3df3087 2003-12-06 devnull 	Muxqueue *q;
 49 d3df3087 2003-12-06 devnull 	void *p;
 50 d3df3087 2003-12-06 devnull 	Mux *mux;
 51 d3df3087 2003-12-06 devnull 
 52 d3df3087 2003-12-06 devnull 	mux = v;
 53 d3df3087 2003-12-06 devnull 	q = _muxqalloc();
 54 d3df3087 2003-12-06 devnull 
 55 d3df3087 2003-12-06 devnull 	qlock(&mux->lk);
 56 d3df3087 2003-12-06 devnull 	mux->writeq = q;
 57 d3df3087 2003-12-06 devnull 	qlock(&mux->outlk);
 58 d3df3087 2003-12-06 devnull 	rwakeup(&mux->rpcfork);
 59 d3df3087 2003-12-06 devnull 	qunlock(&mux->lk);
 60 d3df3087 2003-12-06 devnull 
 61 d3df3087 2003-12-06 devnull 	while((p = _muxqrecv(q)) != nil)
 62 d3df3087 2003-12-06 devnull 		if(mux->send(mux, p) < 0)
 63 d3df3087 2003-12-06 devnull 			break;
 64 d3df3087 2003-12-06 devnull 	qunlock(&mux->outlk);
 65 d3df3087 2003-12-06 devnull 	qlock(&mux->lk);
 66 d3df3087 2003-12-06 devnull 	_muxqhangup(q);
 67 d3df3087 2003-12-06 devnull 	while((p = _muxnbqrecv(q)) != nil)
 68 d3df3087 2003-12-06 devnull 		free(p);
 69 d3df3087 2003-12-06 devnull 	free(q);
 70 d3df3087 2003-12-06 devnull 	mux->writeq = nil;
 71 d3df3087 2003-12-06 devnull 	rwakeup(&mux->rpcfork);
 72 d3df3087 2003-12-06 devnull 	qunlock(&mux->lk);
 73 d3df3087 2003-12-06 devnull 	return;
 74 d3df3087 2003-12-06 devnull }
 75 d3df3087 2003-12-06 devnull 
 76 d3df3087 2003-12-06 devnull void*
 77 d3df3087 2003-12-06 devnull _muxrecv(Mux *mux)
 78 d3df3087 2003-12-06 devnull {
 79 d3df3087 2003-12-06 devnull 	void *p;
 80 d3df3087 2003-12-06 devnull 
 81 d3df3087 2003-12-06 devnull 	qlock(&mux->lk);
 82 d3df3087 2003-12-06 devnull /*
 83 d3df3087 2003-12-06 devnull 	if(mux->state != VtStateConnected){
 84 d3df3087 2003-12-06 devnull 		werrstr("not connected");
 85 d3df3087 2003-12-06 devnull 		qunlock(&mux->lk);
 86 d3df3087 2003-12-06 devnull 		return nil;
 87 d3df3087 2003-12-06 devnull 	}
 88 d3df3087 2003-12-06 devnull */
 89 d3df3087 2003-12-06 devnull 	if(mux->readq){
 90 d3df3087 2003-12-06 devnull 		qunlock(&mux->lk);
 91 d3df3087 2003-12-06 devnull 		return _muxqrecv(mux->readq);
 92 d3df3087 2003-12-06 devnull 	}
 93 d3df3087 2003-12-06 devnull 
 94 d3df3087 2003-12-06 devnull 	qlock(&mux->inlk);
 95 d3df3087 2003-12-06 devnull 	qunlock(&mux->lk);
 96 d3df3087 2003-12-06 devnull 	p = mux->recv(mux);
 97 d3df3087 2003-12-06 devnull 	qunlock(&mux->inlk);
 98 d3df3087 2003-12-06 devnull /*
 99 d3df3087 2003-12-06 devnull 	if(!p)
100 d3df3087 2003-12-06 devnull 		vthangup(mux);
101 d3df3087 2003-12-06 devnull */
102 d3df3087 2003-12-06 devnull 	return p;
103 d3df3087 2003-12-06 devnull }
104 d3df3087 2003-12-06 devnull 
105 d3df3087 2003-12-06 devnull int
106 d3df3087 2003-12-06 devnull _muxsend(Mux *mux, void *p)
107 d3df3087 2003-12-06 devnull {
108 d3df3087 2003-12-06 devnull 	qlock(&mux->lk);
109 d3df3087 2003-12-06 devnull /*
110 d3df3087 2003-12-06 devnull 	if(mux->state != VtStateConnected){
111 d3df3087 2003-12-06 devnull 		packetfree(p);
112 d3df3087 2003-12-06 devnull 		werrstr("not connected");
113 d3df3087 2003-12-06 devnull 		qunlock(&mux->lk);
114 d3df3087 2003-12-06 devnull 		return -1;
115 d3df3087 2003-12-06 devnull 	}
116 d3df3087 2003-12-06 devnull */
117 d3df3087 2003-12-06 devnull 	if(mux->writeq){
118 d3df3087 2003-12-06 devnull 		qunlock(&mux->lk);
119 d3df3087 2003-12-06 devnull 		if(_muxqsend(mux->writeq, p) < 0){
120 d3df3087 2003-12-06 devnull 			free(p);
121 d3df3087 2003-12-06 devnull 			return -1;
122 d3df3087 2003-12-06 devnull 		}
123 d3df3087 2003-12-06 devnull 		return 0;
124 d3df3087 2003-12-06 devnull 	}
125 d3df3087 2003-12-06 devnull 
126 d3df3087 2003-12-06 devnull 	qlock(&mux->outlk);
127 d3df3087 2003-12-06 devnull 	qunlock(&mux->lk);
128 d3df3087 2003-12-06 devnull 	if(mux->send(mux, p) < 0){
129 d3df3087 2003-12-06 devnull 		qunlock(&mux->outlk);
130 d3df3087 2003-12-06 devnull 		/* vthangup(mux); */
131 d3df3087 2003-12-06 devnull 		return -1;	
132 d3df3087 2003-12-06 devnull 	}
133 d3df3087 2003-12-06 devnull 	qunlock(&mux->outlk);
134 d3df3087 2003-12-06 devnull 	return 0;
135 d3df3087 2003-12-06 devnull }
136 d3df3087 2003-12-06 devnull