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 d3df3087 2003-12-06 devnull while((p = _muxnbqrecv(q)) != nil)
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 d3df3087 2003-12-06 devnull while((p = _muxnbqrecv(q)) != nil)
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 d3df3087 2003-12-06 devnull _muxrecv(Mux *mux)
81 d3df3087 2003-12-06 devnull qlock(&mux->lk);
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;
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);
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);
100 d3df3087 2003-12-06 devnull vthangup(mux);
102 d3df3087 2003-12-06 devnull return p;
106 d3df3087 2003-12-06 devnull _muxsend(Mux *mux, void *p)
108 d3df3087 2003-12-06 devnull qlock(&mux->lk);
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;
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;
123 d3df3087 2003-12-06 devnull return 0;
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;
133 d3df3087 2003-12-06 devnull qunlock(&mux->outlk);
134 d3df3087 2003-12-06 devnull return 0;