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 b65a69c1 2008-06-30 rsc int chattydrawclient = 0;
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 3a194702 2006-11-04 devnull static int drawnbrecv(Mux *mux, void**);
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 e6c837d6 2009-07-15 rsc char *devdraw;
38 257fb626 2006-06-25 devnull close(p[0]);
39 257fb626 2006-06-25 devnull dup(p[1], 0);
40 257fb626 2006-06-25 devnull dup(p[1], 1);
41 257fb626 2006-06-25 devnull /* execl("strace", "strace", "-o", "drawsrv.out", "drawsrv", nil); */
43 19f4cef5 2006-06-26 devnull * The argv0 has no meaning to devdraw.
44 19f4cef5 2006-06-26 devnull * Pass it along only so that the various
45 19f4cef5 2006-06-26 devnull * devdraws in psu -a can be distinguished.
46 b61a5ce8 2008-07-01 rsc * The NOLIBTHREADDAEMONIZE keeps devdraw from
47 b61a5ce8 2008-07-01 rsc * forking before threadmain. OS X hates it when
50 4aad1a32 2009-04-30 rsc * If client didn't use ARGBEGIN, argv0 == nil.
51 4aad1a32 2009-04-30 rsc * Can't send nil through because OS X expects
52 4aad1a32 2009-04-30 rsc * argv[0] to be non-nil. Also, OS X apparently
53 4aad1a32 2009-04-30 rsc * expects argv[0] to be a valid executable name,
54 4aad1a32 2009-04-30 rsc * so "(argv0)" is not okay. Use "devdraw"
57 b61a5ce8 2008-07-01 rsc putenv("NOLIBTHREADDAEMONIZE", "1");
58 e6c837d6 2009-07-15 rsc devdraw = getenv("DEVDRAW");
59 e6c837d6 2009-07-15 rsc if(devdraw == nil)
60 e6c837d6 2009-07-15 rsc devdraw = "devdraw";
61 4aad1a32 2009-04-30 rsc if(argv0 == nil)
62 e6c837d6 2009-07-15 rsc argv0 = devdraw;
63 e6c837d6 2009-07-15 rsc execl(devdraw, argv0, argv0, "(devdraw)", nil);
64 150f8802 2006-06-25 devnull sysfatal("exec devdraw: %r");
66 257fb626 2006-06-25 devnull close(p[1]);
67 257fb626 2006-06-25 devnull d->srvfd = p[0];
68 257fb626 2006-06-25 devnull return 0;
72 257fb626 2006-06-25 devnull _displaymux(Display *d)
74 257fb626 2006-06-25 devnull if((d->mux = mallocz(sizeof(*d->mux), 1)) == nil)
75 257fb626 2006-06-25 devnull return -1;
77 257fb626 2006-06-25 devnull d->mux->mintag = 1;
78 257fb626 2006-06-25 devnull d->mux->maxtag = 255;
79 257fb626 2006-06-25 devnull d->mux->send = drawsend;
80 257fb626 2006-06-25 devnull d->mux->recv = drawrecv;
81 150f8802 2006-06-25 devnull d->mux->nbrecv = drawnbrecv;
82 257fb626 2006-06-25 devnull d->mux->gettag = drawgettag;
83 257fb626 2006-06-25 devnull d->mux->settag = drawsettag;
84 257fb626 2006-06-25 devnull d->mux->aux = d;
85 257fb626 2006-06-25 devnull muxinit(d->mux);
87 257fb626 2006-06-25 devnull return 0;
90 257fb626 2006-06-25 devnull static int
91 257fb626 2006-06-25 devnull drawsend(Mux *mux, void *vmsg)
94 257fb626 2006-06-25 devnull uchar *msg;
95 257fb626 2006-06-25 devnull Display *d;
97 257fb626 2006-06-25 devnull msg = vmsg;
98 257fb626 2006-06-25 devnull GET(msg, n);
99 257fb626 2006-06-25 devnull d = mux->aux;
100 257fb626 2006-06-25 devnull return write(d->srvfd, msg, n);
103 3a194702 2006-11-04 devnull static int
104 3a194702 2006-11-04 devnull _drawrecv(Mux *mux, int canblock, void **vp)
107 257fb626 2006-06-25 devnull uchar buf[4], *p;
108 257fb626 2006-06-25 devnull Display *d;
110 257fb626 2006-06-25 devnull d = mux->aux;
111 3a194702 2006-11-04 devnull *vp = nil;
112 3a194702 2006-11-04 devnull if(!canblock && !canreadfd(d->srvfd))
113 3a194702 2006-11-04 devnull return 0;
114 150f8802 2006-06-25 devnull if((n=readn(d->srvfd, buf, 4)) != 4)
115 3a194702 2006-11-04 devnull return 1;
116 257fb626 2006-06-25 devnull GET(buf, n);
117 257fb626 2006-06-25 devnull p = malloc(n);
118 257fb626 2006-06-25 devnull if(p == nil){
119 257fb626 2006-06-25 devnull fprint(2, "out of memory allocating %d in drawrecv\n", n);
120 3a194702 2006-11-04 devnull return 1;
122 257fb626 2006-06-25 devnull memmove(p, buf, 4);
123 3a194702 2006-11-04 devnull if(readn(d->srvfd, p+4, n-4) != n-4){
124 3a194702 2006-11-04 devnull free(p);
125 3a194702 2006-11-04 devnull return 1;
127 3a194702 2006-11-04 devnull *vp = p;
128 3a194702 2006-11-04 devnull return 1;
131 150f8802 2006-06-25 devnull static void*
132 150f8802 2006-06-25 devnull drawrecv(Mux *mux)
134 3a194702 2006-11-04 devnull void *p;
135 3a194702 2006-11-04 devnull _drawrecv(mux, 1, &p);
136 3a194702 2006-11-04 devnull return p;
139 3a194702 2006-11-04 devnull static int
140 3a194702 2006-11-04 devnull drawnbrecv(Mux *mux, void **vp)
142 3a194702 2006-11-04 devnull return _drawrecv(mux, 0, vp);
145 257fb626 2006-06-25 devnull static int
146 257fb626 2006-06-25 devnull drawgettag(Mux *mux, void *vmsg)
148 257fb626 2006-06-25 devnull uchar *msg;
149 40227f1f 2006-07-04 devnull USED(mux);
151 257fb626 2006-06-25 devnull msg = vmsg;
152 257fb626 2006-06-25 devnull return msg[4];
155 257fb626 2006-06-25 devnull static int
156 257fb626 2006-06-25 devnull drawsettag(Mux *mux, void *vmsg, uint tag)
158 257fb626 2006-06-25 devnull uchar *msg;
159 40227f1f 2006-07-04 devnull USED(mux);
161 257fb626 2006-06-25 devnull msg = vmsg;
162 257fb626 2006-06-25 devnull msg[4] = tag;
163 257fb626 2006-06-25 devnull return 0;
166 257fb626 2006-06-25 devnull static int
167 257fb626 2006-06-25 devnull displayrpc(Display *d, Wsysmsg *tx, Wsysmsg *rx, void **freep)
169 257fb626 2006-06-25 devnull int n, nn;
170 257fb626 2006-06-25 devnull void *tpkt, *rpkt;
172 257fb626 2006-06-25 devnull n = sizeW2M(tx);
173 257fb626 2006-06-25 devnull tpkt = malloc(n);
174 257fb626 2006-06-25 devnull if(freep)
175 257fb626 2006-06-25 devnull *freep = nil;
176 257fb626 2006-06-25 devnull if(tpkt == nil)
177 257fb626 2006-06-25 devnull return -1;
178 257fb626 2006-06-25 devnull tx->tag = 0;
179 257fb626 2006-06-25 devnull if(chattydrawclient)
180 257fb626 2006-06-25 devnull fprint(2, "<- %W\n", tx);
181 257fb626 2006-06-25 devnull nn = convW2M(tx, tpkt, n);
182 257fb626 2006-06-25 devnull if(nn != n){
183 257fb626 2006-06-25 devnull free(tpkt);
184 257fb626 2006-06-25 devnull werrstr("drawclient: sizeW2M convW2M mismatch");
185 257fb626 2006-06-25 devnull fprint(2, "%r\n");
186 257fb626 2006-06-25 devnull return -1;
189 19f4cef5 2006-06-26 devnull * This is the only point where we might reschedule.
190 19f4cef5 2006-06-26 devnull * Muxrpc might need to acquire d->mux->lk, which could
191 19f4cef5 2006-06-26 devnull * be held by some other proc (e.g., the one reading from
192 19f4cef5 2006-06-26 devnull * the keyboard via Trdkbd messages). If we need to wait
193 19f4cef5 2006-06-26 devnull * for the lock, don't let other threads from this proc
194 19f4cef5 2006-06-26 devnull * run. This keeps up the appearance that writes to /dev/draw
195 19f4cef5 2006-06-26 devnull * don't cause rescheduling. If you *do* allow rescheduling
196 19f4cef5 2006-06-26 devnull * here, then flushimage(display, 1) happening in two different
197 19f4cef5 2006-06-26 devnull * threads in the same proc can cause a buffer of commands
198 19f4cef5 2006-06-26 devnull * to be written out twice, leading to interesting results
199 19f4cef5 2006-06-26 devnull * on the screen.
201 19f4cef5 2006-06-26 devnull * Threadpin and threadunpin were added to the thread library
202 19f4cef5 2006-06-26 devnull * to solve exactly this problem. Be careful! They are dangerous.
204 19f4cef5 2006-06-26 devnull * _pin and _unpin are aliases for threadpin and threadunpin
205 19f4cef5 2006-06-26 devnull * in a threaded program and are no-ops in unthreaded programs.
208 257fb626 2006-06-25 devnull rpkt = muxrpc(d->mux, tpkt);
209 19f4cef5 2006-06-26 devnull _unpin();
210 257fb626 2006-06-25 devnull free(tpkt);
211 257fb626 2006-06-25 devnull if(rpkt == nil){
212 257fb626 2006-06-25 devnull werrstr("muxrpc: %r");
213 257fb626 2006-06-25 devnull return -1;
215 257fb626 2006-06-25 devnull GET((uchar*)rpkt, n);
216 257fb626 2006-06-25 devnull nn = convM2W(rpkt, n, rx);
217 257fb626 2006-06-25 devnull if(nn != n){
218 257fb626 2006-06-25 devnull free(rpkt);
219 257fb626 2006-06-25 devnull werrstr("drawclient: convM2W packet size mismatch %d %d %.*H", n, nn, n, rpkt);
220 257fb626 2006-06-25 devnull fprint(2, "%r\n");
221 257fb626 2006-06-25 devnull return -1;
223 257fb626 2006-06-25 devnull if(chattydrawclient)
224 257fb626 2006-06-25 devnull fprint(2, "-> %W\n", rx);
225 257fb626 2006-06-25 devnull if(rx->type == Rerror){
226 257fb626 2006-06-25 devnull werrstr("%s", rx->error);
227 257fb626 2006-06-25 devnull free(rpkt);
228 257fb626 2006-06-25 devnull return -1;
230 257fb626 2006-06-25 devnull if(rx->type != tx->type+1){
231 257fb626 2006-06-25 devnull werrstr("packet type mismatch -- tx %d rx %d",
232 257fb626 2006-06-25 devnull tx->type, rx->type);
233 257fb626 2006-06-25 devnull free(rpkt);
234 257fb626 2006-06-25 devnull return -1;
236 257fb626 2006-06-25 devnull if(freep)
237 257fb626 2006-06-25 devnull *freep = rpkt;
239 257fb626 2006-06-25 devnull free(rpkt);
240 257fb626 2006-06-25 devnull return 0;
244 257fb626 2006-06-25 devnull _displayinit(Display *d, char *label, char *winsize)
246 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
248 257fb626 2006-06-25 devnull tx.type = Tinit;
249 150f8802 2006-06-25 devnull tx.label = label;
250 150f8802 2006-06-25 devnull tx.winsize = winsize;
251 257fb626 2006-06-25 devnull return displayrpc(d, &tx, &rx, nil);
255 257fb626 2006-06-25 devnull _displayrdmouse(Display *d, Mouse *m, int *resized)
257 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
259 257fb626 2006-06-25 devnull tx.type = Trdmouse;
260 257fb626 2006-06-25 devnull if(displayrpc(d, &tx, &rx, nil) < 0)
261 257fb626 2006-06-25 devnull return -1;
262 257fb626 2006-06-25 devnull *m = rx.mouse;
263 257fb626 2006-06-25 devnull *resized = rx.resized;
264 257fb626 2006-06-25 devnull return 0;
268 257fb626 2006-06-25 devnull _displayrdkbd(Display *d, Rune *r)
270 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
272 257fb626 2006-06-25 devnull tx.type = Trdkbd;
273 257fb626 2006-06-25 devnull if(displayrpc(d, &tx, &rx, nil) < 0)
274 257fb626 2006-06-25 devnull return -1;
275 257fb626 2006-06-25 devnull *r = rx.rune;
276 257fb626 2006-06-25 devnull return 0;
280 257fb626 2006-06-25 devnull _displaymoveto(Display *d, Point p)
282 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
284 257fb626 2006-06-25 devnull tx.type = Tmoveto;
285 257fb626 2006-06-25 devnull tx.mouse.xy = p;
286 257fb626 2006-06-25 devnull return displayrpc(d, &tx, &rx, nil);
290 257fb626 2006-06-25 devnull _displaycursor(Display *d, Cursor *c)
292 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
294 257fb626 2006-06-25 devnull tx.type = Tcursor;
295 257fb626 2006-06-25 devnull if(c == nil){
296 257fb626 2006-06-25 devnull memset(&tx.cursor, 0, sizeof tx.cursor);
297 257fb626 2006-06-25 devnull tx.arrowcursor = 1;
299 257fb626 2006-06-25 devnull tx.arrowcursor = 0;
300 257fb626 2006-06-25 devnull tx.cursor = *c;
302 257fb626 2006-06-25 devnull return displayrpc(d, &tx, &rx, nil);
306 257fb626 2006-06-25 devnull _displaybouncemouse(Display *d, Mouse *m)
308 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
310 257fb626 2006-06-25 devnull tx.type = Tbouncemouse;
311 257fb626 2006-06-25 devnull tx.mouse = *m;
312 257fb626 2006-06-25 devnull return displayrpc(d, &tx, &rx, nil);
316 257fb626 2006-06-25 devnull _displaylabel(Display *d, char *label)
318 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
320 257fb626 2006-06-25 devnull tx.type = Tlabel;
321 257fb626 2006-06-25 devnull tx.label = label;
322 257fb626 2006-06-25 devnull return displayrpc(d, &tx, &rx, nil);
326 257fb626 2006-06-25 devnull _displayrdsnarf(Display *d)
328 257fb626 2006-06-25 devnull void *p;
329 257fb626 2006-06-25 devnull char *s;
330 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
332 257fb626 2006-06-25 devnull tx.type = Trdsnarf;
333 257fb626 2006-06-25 devnull if(displayrpc(d, &tx, &rx, &p) < 0)
334 257fb626 2006-06-25 devnull return nil;
335 257fb626 2006-06-25 devnull s = strdup(rx.snarf);
336 257fb626 2006-06-25 devnull free(p);
337 257fb626 2006-06-25 devnull return s;
341 257fb626 2006-06-25 devnull _displaywrsnarf(Display *d, char *snarf)
343 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
345 257fb626 2006-06-25 devnull tx.type = Twrsnarf;
346 257fb626 2006-06-25 devnull tx.snarf = snarf;
347 257fb626 2006-06-25 devnull return displayrpc(d, &tx, &rx, nil);
351 257fb626 2006-06-25 devnull _displayrddraw(Display *d, void *v, int n)
353 257fb626 2006-06-25 devnull void *p;
354 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
356 257fb626 2006-06-25 devnull tx.type = Trddraw;
357 257fb626 2006-06-25 devnull tx.count = n;
358 257fb626 2006-06-25 devnull if(displayrpc(d, &tx, &rx, &p) < 0)
359 257fb626 2006-06-25 devnull return -1;
360 257fb626 2006-06-25 devnull memmove(v, rx.data, rx.count);
361 257fb626 2006-06-25 devnull free(p);
362 257fb626 2006-06-25 devnull return rx.count;
366 257fb626 2006-06-25 devnull _displaywrdraw(Display *d, void *v, int n)
368 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
370 257fb626 2006-06-25 devnull tx.type = Twrdraw;
371 257fb626 2006-06-25 devnull tx.count = n;
372 257fb626 2006-06-25 devnull tx.data = v;
373 257fb626 2006-06-25 devnull if(displayrpc(d, &tx, &rx, nil) < 0)
374 257fb626 2006-06-25 devnull return -1;
375 257fb626 2006-06-25 devnull return rx.count;
379 257fb626 2006-06-25 devnull _displaytop(Display *d)
381 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
383 257fb626 2006-06-25 devnull tx.type = Ttop;
384 257fb626 2006-06-25 devnull return displayrpc(d, &tx, &rx, nil);
388 257fb626 2006-06-25 devnull _displayresize(Display *d, Rectangle r)
390 257fb626 2006-06-25 devnull Wsysmsg tx, rx;
392 257fb626 2006-06-25 devnull tx.type = Tresize;
393 257fb626 2006-06-25 devnull tx.rect = r;
394 257fb626 2006-06-25 devnull return displayrpc(d, &tx, &rx, nil);
397 150f8802 2006-06-25 devnull static int
398 150f8802 2006-06-25 devnull canreadfd(int fd)
400 150f8802 2006-06-25 devnull fd_set rs, ws, xs;
401 150f8802 2006-06-25 devnull struct timeval tv;
403 150f8802 2006-06-25 devnull FD_ZERO(&rs);
404 150f8802 2006-06-25 devnull FD_ZERO(&ws);
405 150f8802 2006-06-25 devnull FD_ZERO(&xs);
406 150f8802 2006-06-25 devnull FD_SET(fd, &rs);
407 150f8802 2006-06-25 devnull FD_SET(fd, &xs);
408 150f8802 2006-06-25 devnull tv.tv_sec = 0;
409 150f8802 2006-06-25 devnull tv.tv_usec = 0;
410 150f8802 2006-06-25 devnull if(select(fd+1, &rs, &ws, &xs, &tv) < 0)
411 150f8802 2006-06-25 devnull return 0;
412 150f8802 2006-06-25 devnull if(FD_ISSET(fd, &rs) || FD_ISSET(fd, &xs))
413 150f8802 2006-06-25 devnull return 1;
414 150f8802 2006-06-25 devnull return 0;