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 fa325e9b 2020-01-10 cross  * 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 454ae2ad 2008-05-06 rsc 	p = nil;
 38 454ae2ad 2008-05-06 rsc 	while(_muxnbqrecv(q, &p) && p != nil){
 39 d3df3087 2003-12-06 devnull 		free(p);
 40 454ae2ad 2008-05-06 rsc 		p = nil;
 41 454ae2ad 2008-05-06 rsc 	}
 42 d3df3087 2003-12-06 devnull 	free(q);
 43 d3df3087 2003-12-06 devnull 	mux->readq = nil;
 44 d3df3087 2003-12-06 devnull 	rwakeup(&mux->rpcfork);
 45 d3df3087 2003-12-06 devnull 	qunlock(&mux->lk);
 46 d3df3087 2003-12-06 devnull }
 47 d3df3087 2003-12-06 devnull 
 48 d3df3087 2003-12-06 devnull void
 49 d3df3087 2003-12-06 devnull _muxsendproc(void *v)
 50 d3df3087 2003-12-06 devnull {
 51 d3df3087 2003-12-06 devnull 	Muxqueue *q;
 52 d3df3087 2003-12-06 devnull 	void *p;
 53 d3df3087 2003-12-06 devnull 	Mux *mux;
 54 d3df3087 2003-12-06 devnull 
 55 d3df3087 2003-12-06 devnull 	mux = v;
 56 d3df3087 2003-12-06 devnull 	q = _muxqalloc();
 57 d3df3087 2003-12-06 devnull 
 58 d3df3087 2003-12-06 devnull 	qlock(&mux->lk);
 59 d3df3087 2003-12-06 devnull 	mux->writeq = q;
 60 d3df3087 2003-12-06 devnull 	qlock(&mux->outlk);
 61 d3df3087 2003-12-06 devnull 	rwakeup(&mux->rpcfork);
 62 d3df3087 2003-12-06 devnull 	qunlock(&mux->lk);
 63 d3df3087 2003-12-06 devnull 
 64 d3df3087 2003-12-06 devnull 	while((p = _muxqrecv(q)) != nil)
 65 d3df3087 2003-12-06 devnull 		if(mux->send(mux, p) < 0)
 66 d3df3087 2003-12-06 devnull 			break;
 67 d3df3087 2003-12-06 devnull 	qunlock(&mux->outlk);
 68 d3df3087 2003-12-06 devnull 	qlock(&mux->lk);
 69 d3df3087 2003-12-06 devnull 	_muxqhangup(q);
 70 3a194702 2006-11-04 devnull 	while(_muxnbqrecv(q, &p))
 71 d3df3087 2003-12-06 devnull 		free(p);
 72 d3df3087 2003-12-06 devnull 	free(q);
 73 d3df3087 2003-12-06 devnull 	mux->writeq = nil;
 74 d3df3087 2003-12-06 devnull 	rwakeup(&mux->rpcfork);
 75 d3df3087 2003-12-06 devnull 	qunlock(&mux->lk);
 76 d3df3087 2003-12-06 devnull 	return;
 77 d3df3087 2003-12-06 devnull }
 78 d3df3087 2003-12-06 devnull 
 79 3a194702 2006-11-04 devnull int
 80 3a194702 2006-11-04 devnull _muxrecv(Mux *mux, int canblock, void **vp)
 81 d3df3087 2003-12-06 devnull {
 82 d3df3087 2003-12-06 devnull 	void *p;
 83 3a194702 2006-11-04 devnull 	int ret;
 84 d3df3087 2003-12-06 devnull 
 85 d3df3087 2003-12-06 devnull 	qlock(&mux->lk);
 86 d3df3087 2003-12-06 devnull 	if(mux->readq){
 87 d3df3087 2003-12-06 devnull 		qunlock(&mux->lk);
 88 3a194702 2006-11-04 devnull 		if(canblock){
 89 3a194702 2006-11-04 devnull 			*vp = _muxqrecv(mux->readq);
 90 3a194702 2006-11-04 devnull 			return 1;
 91 3a194702 2006-11-04 devnull 		}
 92 3a194702 2006-11-04 devnull 		return _muxnbqrecv(mux->readq, vp);
 93 d3df3087 2003-12-06 devnull 	}
 94 d3df3087 2003-12-06 devnull 
 95 d3df3087 2003-12-06 devnull 	qlock(&mux->inlk);
 96 d3df3087 2003-12-06 devnull 	qunlock(&mux->lk);
 97 3a194702 2006-11-04 devnull 	if(canblock){
 98 2d2e5c71 2006-06-25 devnull 		p = mux->recv(mux);
 99 3a194702 2006-11-04 devnull 		ret = 1;
100 3a194702 2006-11-04 devnull 	}else{
101 2d2e5c71 2006-06-25 devnull 		if(mux->nbrecv)
102 3a194702 2006-11-04 devnull 			ret = mux->nbrecv(mux, &p);
103 3a194702 2006-11-04 devnull 		else{
104 3a194702 2006-11-04 devnull 			/* send eof, not "no packet ready" */
105 2d2e5c71 2006-06-25 devnull 			p = nil;
106 3a194702 2006-11-04 devnull 			ret = 1;
107 3a194702 2006-11-04 devnull 		}
108 2d2e5c71 2006-06-25 devnull 	}
109 d3df3087 2003-12-06 devnull 	qunlock(&mux->inlk);
110 3a194702 2006-11-04 devnull 	*vp = p;
111 3a194702 2006-11-04 devnull 	return ret;
112 d3df3087 2003-12-06 devnull }
113 d3df3087 2003-12-06 devnull 
114 d3df3087 2003-12-06 devnull int
115 d3df3087 2003-12-06 devnull _muxsend(Mux *mux, void *p)
116 d3df3087 2003-12-06 devnull {
117 d3df3087 2003-12-06 devnull 	qlock(&mux->lk);
118 d3df3087 2003-12-06 devnull /*
119 d3df3087 2003-12-06 devnull 	if(mux->state != VtStateConnected){
120 d3df3087 2003-12-06 devnull 		packetfree(p);
121 d3df3087 2003-12-06 devnull 		werrstr("not connected");
122 d3df3087 2003-12-06 devnull 		qunlock(&mux->lk);
123 d3df3087 2003-12-06 devnull 		return -1;
124 d3df3087 2003-12-06 devnull 	}
125 d3df3087 2003-12-06 devnull */
126 d3df3087 2003-12-06 devnull 	if(mux->writeq){
127 d3df3087 2003-12-06 devnull 		qunlock(&mux->lk);
128 d3df3087 2003-12-06 devnull 		if(_muxqsend(mux->writeq, p) < 0){
129 d3df3087 2003-12-06 devnull 			free(p);
130 d3df3087 2003-12-06 devnull 			return -1;
131 d3df3087 2003-12-06 devnull 		}
132 d3df3087 2003-12-06 devnull 		return 0;
133 d3df3087 2003-12-06 devnull 	}
134 d3df3087 2003-12-06 devnull 
135 d3df3087 2003-12-06 devnull 	qlock(&mux->outlk);
136 d3df3087 2003-12-06 devnull 	qunlock(&mux->lk);
137 d3df3087 2003-12-06 devnull 	if(mux->send(mux, p) < 0){
138 d3df3087 2003-12-06 devnull 		qunlock(&mux->outlk);
139 d3df3087 2003-12-06 devnull 		/* vthangup(mux); */
140 fa325e9b 2020-01-10 cross 		return -1;
141 d3df3087 2003-12-06 devnull 	}
142 d3df3087 2003-12-06 devnull 	qunlock(&mux->outlk);
143 d3df3087 2003-12-06 devnull 	return 0;
144 d3df3087 2003-12-06 devnull }