1 257fb626 2006-06-25 devnull /* Copyright (c) 2006 Russ Cox */
3 257fb626 2006-06-25 devnull #include <u.h>
4 150f8802 2006-06-25 devnull #include <sys/select.h>
5 257fb626 2006-06-25 devnull #include <libc.h>
6 257fb626 2006-06-25 devnull #include <draw.h>
7 257fb626 2006-06-25 devnull #include <mouse.h>
8 257fb626 2006-06-25 devnull #include <cursor.h>
9 257fb626 2006-06-25 devnull #include <drawfcall.h>
10 257fb626 2006-06-25 devnull #include <mux.h>
12 257fb626 2006-06-25 devnull int chattydrawclient;
14 257fb626 2006-06-25 devnull static int drawgettag(Mux *mux, void *vmsg);
15 257fb626 2006-06-25 devnull static void* drawrecv(Mux *mux);
16 150f8802 2006-06-25 devnull static void* drawnbrecv(Mux *mux);
17 257fb626 2006-06-25 devnull static int drawsend(Mux *mux, void *vmsg);
18 257fb626 2006-06-25 devnull static int drawsettag(Mux *mux, void *vmsg, uint tag);
19 150f8802 2006-06-25 devnull static int canreadfd(int);
22 257fb626 2006-06-25 devnull _displayconnect(Display *d)
24 257fb626 2006-06-25 devnull int pid, p[2];
26 257fb626 2006-06-25 devnull fmtinstall('W', drawfcallfmt);
27 257fb626 2006-06-25 devnull fmtinstall('H', encodefmt);
29 257fb626 2006-06-25 devnull if(pipe(p) < 0)
30 257fb626 2006-06-25 devnull return -1;
31 257fb626 2006-06-25 devnull if((pid=fork()) < 0){
32 257fb626 2006-06-25 devnull close(p[0]);
33 257fb626 2006-06-25 devnull close(p[1]);
34 257fb626 2006-06-25 devnull return -1;
36 257fb626 2006-06-25 devnull if(pid == 0){
37 257fb626 2006-06-25 devnull close(p[0]);
38 257fb626 2006-06-25 devnull dup(p[1], 0);
39 257fb626 2006-06-25 devnull dup(p[1], 1);
40 257fb626 2006-06-25 devnull /* execl("strace", "strace", "-o", "drawsrv.out", "drawsrv", nil); */
42 19f4cef5 2006-06-26 devnull * The argv0 has no meaning to devdraw.
43 19f4cef5 2006-06-26 devnull * Pass it along only so that the various
44 19f4cef5 2006-06-26 devnull * devdraws in psu -a can be distinguished.
46 19f4cef5 2006-06-26 devnull execl("devdraw", "devdraw", argv0, nil);
47 150f8802 2006-06-25 devnull sysfatal("exec devdraw: %r");
49 257fb626 2006-06-25 devnull close(p[1]);
50 257fb626 2006-06-25 devnull d->srvfd = p[0];
51 257fb626 2006-06-25 devnull return 0;
55 257fb626 2006-06-25 devnull _displaymux(Display *d)
57 257fb626 2006-06-25 devnull if((d->mux = mallocz(sizeof(*d->mux), 1)) == nil)
58 257fb626 2006-06-25 devnull return -1;
60 257fb626 2006-06-25 devnull d->mux->mintag = 1;
61 257fb626 2006-06-25 devnull d->mux->maxtag = 255;
62 257fb626 2006-06-25 devnull d->mux->send = drawsend;
63 257fb626 2006-06-25 devnull d->mux->recv = drawrecv;
64 150f8802 2006-06-25 devnull d->mux->nbrecv = drawnbrecv;
65 257fb626 2006-06-25 devnull d->mux->gettag = drawgettag;
66 257fb626 2006-06-25 devnull d->mux->settag = drawsettag;
67 257fb626 2006-06-25 devnull d->mux->aux = d;
68 257fb626 2006-06-25 devnull muxinit(d->mux);
70 257fb626 2006-06-25 devnull return 0;
73 257fb626 2006-06-25 devnull static int
74 257fb626 2006-06-25 devnull drawsend(Mux *mux, void *vmsg)
77 257fb626 2006-06-25 devnull uchar *msg;
78 257fb626 2006-06-25 devnull Display *d;
80 257fb626 2006-06-25 devnull msg = vmsg;
81 257fb626 2006-06-25 devnull GET(msg, n);
82 257fb626 2006-06-25 devnull d = mux->aux;
83 257fb626 2006-06-25 devnull return write(d->srvfd, msg, n);
86 257fb626 2006-06-25 devnull static void*
87 150f8802 2006-06-25 devnull _drawrecv(Mux *mux, int nb)
90 257fb626 2006-06-25 devnull uchar buf[4], *p;
91 257fb626 2006-06-25 devnull Display *d;
93 257fb626 2006-06-25 devnull d = mux->aux;
94 150f8802 2006-06-25 devnull if(nb && !canreadfd(d->srvfd))
95 257fb626 2006-06-25 devnull return nil;
96 150f8802 2006-06-25 devnull if((n=readn(d->srvfd, buf, 4)) != 4)
97 150f8802 2006-06-25 devnull return nil;
98 257fb626 2006-06-25 devnull GET(buf, n);
99 257fb626 2006-06-25 devnull p = malloc(n);
100 257fb626 2006-06-25 devnull if(p == nil){
101 257fb626 2006-06-25 devnull fprint(2, "out of memory allocating %d in drawrecv\n", n);
102 257fb626 2006-06-25 devnull return nil;
104 257fb626 2006-06-25 devnull memmove(p, buf, 4);
105 150f8802 2006-06-25 devnull if(readn(d->srvfd, p+4, n-4) != n-4)
106 257fb626 2006-06-25 devnull return nil;
107 257fb626 2006-06-25 devnull return p;
110 150f8802 2006-06-25 devnull static void*
111 150f8802 2006-06-25 devnull drawrecv(Mux *mux)
113 150f8802 2006-06-25 devnull return _drawrecv(mux, 0);
116 150f8802 2006-06-25 devnull static void*
117 150f8802 2006-06-25 devnull drawnbrecv(Mux *mux)
119 150f8802 2006-06-25 devnull return _drawrecv(mux, 1);
122 257fb626 2006-06-25 devnull static int
123 257fb626 2006-06-25 devnull drawgettag(Mux *mux, void *vmsg)
125 257fb626 2006-06-25 devnull uchar *msg;
126 40227f1f 2006-07-04 devnull USED(mux);
128 257fb626 2006-06-25 devnull msg = vmsg;
129 257fb626 2006-06-25 devnull return msg[4];
132 257fb626 2006-06-25 devnull static int
133 257fb626 2006-06-25 devnull drawsettag(Mux *mux, void *vmsg, uint tag)
135 257fb626 2006-06-25 devnull uchar *msg;
136 40227f1f 2006-07-04 devnull USED(mux);
138 257fb626 2006-06-25 devnull msg = vmsg;
139 257fb626 2006-06-25 devnull msg[4] = tag;
140 257fb626 2006-06-25 devnull return 0;
143 257fb626 2006-06-25 devnull static int
144 257fb626 2006-06-25 devnull displayrpc(Display *d, Wsysmsg *tx, Wsysmsg *rx, void **freep)
146 257fb626 2006-06-25 devnull int n, nn;
147 257fb626 2006-06-25 devnull void *tpkt, *rpkt;
149 257fb626 2006-06-25 devnull n = sizeW2M(tx);
150 257fb626 2006-06-25 devnull tpkt = malloc(n);
151 257fb626 2006-06-25 devnull if(freep)
152 257fb626 2006-06-25 devnull *freep = nil;
153 257fb626 2006-06-25 devnull if(tpkt == nil)
154 257fb626 2006-06-25 devnull return -1;
155 257fb626 2006-06-25 devnull tx->tag = 0;
156 257fb626 2006-06-25 devnull if(chattydrawclient)
157 257fb626 2006-06-25 devnull fprint(2, "<- %W\n", tx);
158 257fb626 2006-06-25 devnull nn = convW2M(tx, tpkt, n);
159 257fb626 2006-06-25 devnull if(nn != n){
160 257fb626 2006-06-25 devnull free(tpkt);
161 257fb626 2006-06-25 devnull werrstr("drawclient: sizeW2M convW2M mismatch");
162 257fb626 2006-06-25 devnull fprint(2, "%r\n");
163 257fb626 2006-06-25 devnull return -1;
166 19f4cef5 2006-06-26 devnull * This is the only point where we might reschedule.
167 19f4cef5 2006-06-26 devnull * Muxrpc might need to acquire d->mux->lk, which could
168 19f4cef5 2006-06-26 devnull * be held by some other proc (e.g., the one reading from
169 19f4cef5 2006-06-26 devnull * the keyboard via Trdkbd messages). If we need to wait
170 19f4cef5 2006-06-26 devnull * for the lock, don't let other threads from this proc
171 19f4cef5 2006-06-26 devnull * run. This keeps up the appearance that writes to /dev/draw
172 19f4cef5 2006-06-26 devnull * don't cause rescheduling. If you *do* allow rescheduling
173 19f4cef5 2006-06-26 devnull * here, then flushimage(display, 1) happening in two different
174 19f4cef5 2006-06-26 devnull * threads in the same proc can cause a buffer of commands
175 19f4cef5 2006-06-26 devnull * to be written out twice, leading to interesting results
176 19f4cef5 2006-06-26 devnull * on the screen.
178 19f4cef5 2006-06-26 devnull * Threadpin and threadunpin were added to the thread library
179 19f4cef5 2006-06-26 devnull * to solve exactly this problem. Be careful! They are dangerous.
181 19f4cef5 2006-06-26 devnull * _pin and _unpin are aliases for threadpin and threadunpin
182 19f4cef5 2006-06-26 devnull * in a threaded program and are no-ops in unthreaded programs.
185 257fb626 2006-06-25 devnull rpkt = muxrpc(d->mux, tpkt);
186 19f4cef5 2006-06-26 devnull _unpin();
187 257fb626 2006-06-25 devnull free(tpkt);
188 257fb626 2006-06-25 devnull if(rpkt == nil){
189 257fb626 2006-06-25 devnull werrstr("muxrpc: %r");
190 257fb626 2006-06-25 devnull return -1;
192 257fb626 2006-06-25 devnull GET((uchar*)rpkt, n);
193 257fb626 2006-06-25 devnull nn = convM2W(rpkt, n, rx);
194 257fb626 2006-06-25 devnull if(nn != n){
195 257fb626 2006-06-25 devnull free(rpkt);
196 257fb626 2006-06-25 devnull werrstr("drawclient: convM2W packet size mismatch %d %d %.*H", n, nn, n, rpkt);
197 257fb626 2006-06-25 devnull fprint(2, "%r\n");
198 257fb626 2006-06-25 devnull return -1;
200 257fb626 2006-06-25 devnull if(chattydrawclient)
201 257fb626 2006-06-25 devnull fprint(2, "-> %W\n", rx);
202 257fb626 2006-06-25 devnull if(rx->type == Rerror){
203 257fb626 2006-06-25 devnull werrstr("%s", rx->error);
204 257fb626 2006-06-25 devnull free(rpkt);
205 257fb626 2006-06-25 devnull return -1;
207 257fb626 2006-06-25 devnull if(rx->type != tx->type+1){
208 257fb626 2006-06-25 devnull werrstr("packet type mismatch -- tx %d rx %d",
209 257fb626 2006-06-25 devnull tx->type, rx->type);
210 257fb626 2006-06-25 devnull free(rpkt);
211 257fb626 2006-06-25 devnull return -1;
213 257fb626 2006-06-25 devnull if(freep)
214 257fb626 2006-06-25 devnull *freep = rpkt;
216 257fb626 2006-06-25 devnull free(rpkt);
217 257fb626 2006-06-25 devnull return 0;
221 257fb626 2006-06-25 devnull _displayinit(Display *d, char *label, char *winsize)
223 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
225 257fb626 2006-06-25 devnull tx.type = Tinit;
226 150f8802 2006-06-25 devnull tx.label = label;
227 150f8802 2006-06-25 devnull tx.winsize = winsize;
228 257fb626 2006-06-25 devnull return displayrpc(d, &tx, &rx, nil);
232 257fb626 2006-06-25 devnull _displayrdmouse(Display *d, Mouse *m, int *resized)
234 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
236 257fb626 2006-06-25 devnull tx.type = Trdmouse;
237 257fb626 2006-06-25 devnull if(displayrpc(d, &tx, &rx, nil) < 0)
238 257fb626 2006-06-25 devnull return -1;
239 257fb626 2006-06-25 devnull *m = rx.mouse;
240 257fb626 2006-06-25 devnull *resized = rx.resized;
241 257fb626 2006-06-25 devnull return 0;
245 257fb626 2006-06-25 devnull _displayrdkbd(Display *d, Rune *r)
247 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
249 257fb626 2006-06-25 devnull tx.type = Trdkbd;
250 257fb626 2006-06-25 devnull if(displayrpc(d, &tx, &rx, nil) < 0)
251 257fb626 2006-06-25 devnull return -1;
252 257fb626 2006-06-25 devnull *r = rx.rune;
253 257fb626 2006-06-25 devnull return 0;
257 257fb626 2006-06-25 devnull _displaymoveto(Display *d, Point p)
259 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
261 257fb626 2006-06-25 devnull tx.type = Tmoveto;
262 257fb626 2006-06-25 devnull tx.mouse.xy = p;
263 257fb626 2006-06-25 devnull return displayrpc(d, &tx, &rx, nil);
267 257fb626 2006-06-25 devnull _displaycursor(Display *d, Cursor *c)
269 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
271 257fb626 2006-06-25 devnull tx.type = Tcursor;
272 257fb626 2006-06-25 devnull if(c == nil){
273 257fb626 2006-06-25 devnull memset(&tx.cursor, 0, sizeof tx.cursor);
274 257fb626 2006-06-25 devnull tx.arrowcursor = 1;
276 257fb626 2006-06-25 devnull tx.arrowcursor = 0;
277 257fb626 2006-06-25 devnull tx.cursor = *c;
279 257fb626 2006-06-25 devnull return displayrpc(d, &tx, &rx, nil);
283 257fb626 2006-06-25 devnull _displaybouncemouse(Display *d, Mouse *m)
285 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
287 257fb626 2006-06-25 devnull tx.type = Tbouncemouse;
288 257fb626 2006-06-25 devnull tx.mouse = *m;
289 257fb626 2006-06-25 devnull return displayrpc(d, &tx, &rx, nil);
293 257fb626 2006-06-25 devnull _displaylabel(Display *d, char *label)
295 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
297 257fb626 2006-06-25 devnull tx.type = Tlabel;
298 257fb626 2006-06-25 devnull tx.label = label;
299 257fb626 2006-06-25 devnull return displayrpc(d, &tx, &rx, nil);
303 257fb626 2006-06-25 devnull _displayrdsnarf(Display *d)
305 257fb626 2006-06-25 devnull void *p;
306 257fb626 2006-06-25 devnull char *s;
307 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
309 257fb626 2006-06-25 devnull tx.type = Trdsnarf;
310 257fb626 2006-06-25 devnull if(displayrpc(d, &tx, &rx, &p) < 0)
311 257fb626 2006-06-25 devnull return nil;
312 257fb626 2006-06-25 devnull s = strdup(rx.snarf);
313 257fb626 2006-06-25 devnull free(p);
314 257fb626 2006-06-25 devnull return s;
318 257fb626 2006-06-25 devnull _displaywrsnarf(Display *d, char *snarf)
320 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
322 257fb626 2006-06-25 devnull tx.type = Twrsnarf;
323 257fb626 2006-06-25 devnull tx.snarf = snarf;
324 257fb626 2006-06-25 devnull return displayrpc(d, &tx, &rx, nil);
328 257fb626 2006-06-25 devnull _displayrddraw(Display *d, void *v, int n)
330 257fb626 2006-06-25 devnull void *p;
331 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
333 257fb626 2006-06-25 devnull tx.type = Trddraw;
334 257fb626 2006-06-25 devnull tx.count = n;
335 257fb626 2006-06-25 devnull if(displayrpc(d, &tx, &rx, &p) < 0)
336 257fb626 2006-06-25 devnull return -1;
337 257fb626 2006-06-25 devnull memmove(v, rx.data, rx.count);
338 257fb626 2006-06-25 devnull free(p);
339 257fb626 2006-06-25 devnull return rx.count;
343 257fb626 2006-06-25 devnull _displaywrdraw(Display *d, void *v, int n)
345 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
347 257fb626 2006-06-25 devnull tx.type = Twrdraw;
348 257fb626 2006-06-25 devnull tx.count = n;
349 257fb626 2006-06-25 devnull tx.data = v;
350 257fb626 2006-06-25 devnull if(displayrpc(d, &tx, &rx, nil) < 0)
351 257fb626 2006-06-25 devnull return -1;
352 257fb626 2006-06-25 devnull return rx.count;
356 257fb626 2006-06-25 devnull _displaytop(Display *d)
358 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
360 257fb626 2006-06-25 devnull tx.type = Ttop;
361 257fb626 2006-06-25 devnull return displayrpc(d, &tx, &rx, nil);
365 257fb626 2006-06-25 devnull _displayresize(Display *d, Rectangle r)
367 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
369 257fb626 2006-06-25 devnull tx.type = Tresize;
370 257fb626 2006-06-25 devnull tx.rect = r;
371 257fb626 2006-06-25 devnull return displayrpc(d, &tx, &rx, nil);
374 150f8802 2006-06-25 devnull static int
375 150f8802 2006-06-25 devnull canreadfd(int fd)
377 150f8802 2006-06-25 devnull fd_set rs, ws, xs;
378 150f8802 2006-06-25 devnull struct timeval tv;
380 150f8802 2006-06-25 devnull FD_ZERO(&rs);
381 150f8802 2006-06-25 devnull FD_ZERO(&ws);
382 150f8802 2006-06-25 devnull FD_ZERO(&xs);
383 150f8802 2006-06-25 devnull FD_SET(fd, &rs);
384 150f8802 2006-06-25 devnull FD_SET(fd, &xs);
385 150f8802 2006-06-25 devnull tv.tv_sec = 0;
386 150f8802 2006-06-25 devnull tv.tv_usec = 0;
387 150f8802 2006-06-25 devnull if(select(fd+1, &rs, &ws, &xs, &tv) < 0)
388 150f8802 2006-06-25 devnull return 0;
389 150f8802 2006-06-25 devnull if(FD_ISSET(fd, &rs) || FD_ISSET(fd, &xs))
390 150f8802 2006-06-25 devnull return 1;
391 150f8802 2006-06-25 devnull return 0;