1 d3df3087 2003-12-06 devnull /* Copyright (C) 2003 Russ Cox, Massachusetts Institute of Technology */
2 d3df3087 2003-12-06 devnull /* See COPYRIGHT */
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>
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.
14 d3df3087 2003-12-06 devnull _muxrecvproc(void *v)
17 d3df3087 2003-12-06 devnull Mux *mux;
18 d3df3087 2003-12-06 devnull Muxqueue *q;
21 d3df3087 2003-12-06 devnull q = _muxqalloc();
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);
29 d3df3087 2003-12-06 devnull while((p = mux->recv(mux)) != nil)
30 d3df3087 2003-12-06 devnull if(_muxqsend(q, p) < 0){
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 3a194702 2006-11-04 devnull while(_muxnbqrecv(q, &p))
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);
46 d3df3087 2003-12-06 devnull _muxsendproc(void *v)
48 d3df3087 2003-12-06 devnull Muxqueue *q;
50 d3df3087 2003-12-06 devnull Mux *mux;
53 d3df3087 2003-12-06 devnull q = _muxqalloc();
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);
61 d3df3087 2003-12-06 devnull while((p = _muxqrecv(q)) != nil)
62 d3df3087 2003-12-06 devnull if(mux->send(mux, p) < 0)
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 3a194702 2006-11-04 devnull while(_muxnbqrecv(q, &p))
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);
77 3a194702 2006-11-04 devnull _muxrecv(Mux *mux, int canblock, void **vp)
82 d3df3087 2003-12-06 devnull qlock(&mux->lk);
83 d3df3087 2003-12-06 devnull if(mux->readq){
84 d3df3087 2003-12-06 devnull qunlock(&mux->lk);
85 3a194702 2006-11-04 devnull if(canblock){
86 3a194702 2006-11-04 devnull *vp = _muxqrecv(mux->readq);
87 3a194702 2006-11-04 devnull return 1;
89 3a194702 2006-11-04 devnull return _muxnbqrecv(mux->readq, vp);
92 d3df3087 2003-12-06 devnull qlock(&mux->inlk);
93 d3df3087 2003-12-06 devnull qunlock(&mux->lk);
94 3a194702 2006-11-04 devnull if(canblock){
95 2d2e5c71 2006-06-25 devnull p = mux->recv(mux);
98 2d2e5c71 2006-06-25 devnull if(mux->nbrecv)
99 3a194702 2006-11-04 devnull ret = mux->nbrecv(mux, &p);
101 3a194702 2006-11-04 devnull /* send eof, not "no packet ready" */
102 2d2e5c71 2006-06-25 devnull p = nil;
103 3a194702 2006-11-04 devnull ret = 1;
106 d3df3087 2003-12-06 devnull qunlock(&mux->inlk);
107 3a194702 2006-11-04 devnull *vp = p;
108 3a194702 2006-11-04 devnull return ret;
112 d3df3087 2003-12-06 devnull _muxsend(Mux *mux, void *p)
114 d3df3087 2003-12-06 devnull qlock(&mux->lk);
116 d3df3087 2003-12-06 devnull if(mux->state != VtStateConnected){
117 d3df3087 2003-12-06 devnull packetfree(p);
118 d3df3087 2003-12-06 devnull werrstr("not connected");
119 d3df3087 2003-12-06 devnull qunlock(&mux->lk);
120 d3df3087 2003-12-06 devnull return -1;
123 d3df3087 2003-12-06 devnull if(mux->writeq){
124 d3df3087 2003-12-06 devnull qunlock(&mux->lk);
125 d3df3087 2003-12-06 devnull if(_muxqsend(mux->writeq, p) < 0){
126 d3df3087 2003-12-06 devnull free(p);
127 d3df3087 2003-12-06 devnull return -1;
129 d3df3087 2003-12-06 devnull return 0;
132 d3df3087 2003-12-06 devnull qlock(&mux->outlk);
133 d3df3087 2003-12-06 devnull qunlock(&mux->lk);
134 d3df3087 2003-12-06 devnull if(mux->send(mux, p) < 0){
135 d3df3087 2003-12-06 devnull qunlock(&mux->outlk);
136 d3df3087 2003-12-06 devnull /* vthangup(mux); */
137 d3df3087 2003-12-06 devnull return -1;
139 d3df3087 2003-12-06 devnull qunlock(&mux->outlk);
140 d3df3087 2003-12-06 devnull return 0;