Whoops
[llpp.git] / wsi.ml
blob8cd475cb6ab4afa4f78439a834b514044ff1bd49
1 type cursor =
2 | CURSOR_INHERIT
3 | CURSOR_INFO
4 | CURSOR_CYCLE
5 | CURSOR_CROSSHAIR
6 | CURSOR_TEXT
7 ;;
9 external cloexec : Unix.file_descr -> unit = "ml_cloexec";;
10 external glx : int -> unit = "ml_glx";;
11 external swapb : unit -> unit = "ml_swapb";;
12 external hasdata : Unix.file_descr -> bool = "ml_hasdata";;
13 external toutf8 : int -> string = "ml_keysymtoutf8";;
15 let dolog fmt = Format.kprintf prerr_endline fmt;;
16 let vlog fmt = Format.kprintf ignore fmt;;
18 exception Quit;;
20 let onot = object
21 method display = ()
22 method reshape _ _ = ()
23 method mouse _ _ _ _ _ = ()
24 method motion _ _ = ()
25 method pmotion _ _ = ()
26 method key _ _ = ()
27 method enter _ _ = ()
28 method leave = ()
29 end;;
31 class type t = object
32 method display : unit
33 method reshape : int -> int -> unit
34 method mouse : int -> bool -> int -> int -> int -> unit
35 method motion : int -> int -> unit
36 method pmotion : int -> int -> unit
37 method key : int -> int -> unit
38 method enter : int -> int -> unit
39 method leave : unit
40 end;;
42 type state =
43 { mutable mink : int
44 ; mutable maxk : int
45 ; mutable keymap : int array array
46 ; fifo : (string -> unit) Queue.t
47 ; mutable seq : int
48 ; mutable protoatom : int
49 ; mutable deleatom : int
50 ; mutable idbase : int
51 ; mutable fullscreen : (int -> unit)
52 ; mutable stringatom : int
53 ; mutable t : t
54 ; mutable sock : Unix.file_descr
55 ; mutable w : int
56 ; mutable h : int
57 ; mutable fs : bool
58 ; mutable parent : int
62 let state =
63 { mink = max_int
64 ; maxk = min_int
65 ; keymap = [||]
66 ; fifo = Queue.create ()
67 ; seq = 0
68 ; protoatom = -1
69 ; deleatom = -1
70 ; idbase = -1
71 ; fullscreen = (fun _ -> ())
72 ; sock = Unix.stdin
73 ; t = onot
74 ; w = 900
75 ; h = 900
76 ; fs = false
77 ; stringatom = 31
78 ; parent = -1
82 let w8 s pos i = String.set s pos (Char.chr (i land 0xff));;
84 let w16 s pos i =
85 w8 s pos i;
86 w8 s (pos+1) (i lsr 8);
89 let w32 s pos i =
90 w16 s pos i;
91 w16 s (pos+2) (i lsr 16);
94 let r16 s pos =
95 let rb pos1 = Char.code (String.get s (pos + pos1)) in
96 (rb 0) lor ((rb 1) lsl 8)
99 let r16s s pos =
100 let i = r16 s pos in
101 i - ((i land 0x8000) lsl 1);
104 let r8 s pos = Char.code (String.get s pos);;
106 let r32 s pos =
107 let rb pos1 = Char.code (String.get s (pos + pos1)) in
108 let l = (rb 0) lor ((rb 1) lsl 8)
109 and u = (rb 2) lor ((rb 3) lsl 8) in
110 (u lsl 16) lor l
113 let error fmt = Printf.kprintf failwith fmt;;
115 let readstr sock n =
116 let s = String.create n in
117 let rec loop pos n =
118 let m = Unix.read sock s pos n in
119 if n != m
120 then (
121 ignore (Unix.select [sock] [] [] 0.01);
122 loop (pos + m) (n - m)
124 (* error "read %d returned %d" n m; *)
126 loop 0 n;
130 let sendstr s ?(pos=0) ?(len=String.length s) sock =
131 vlog "%d => %S" state.seq s;
132 state.seq <- state.seq + 1;
133 let n = Unix.send sock s pos len [] in
134 if n != len
135 then error "send %d returned %d" len n;
138 let updkmap sock resp =
139 let syms = r8 resp 1 in
140 let len = r32 resp 4 in
141 let data =
142 if len > 0
143 then readstr sock (4*len)
144 else ""
146 let m = len / syms in
147 state.keymap <- Array.make_matrix
148 (state.maxk - state.mink) syms 0xffffff;
149 let rec loop i = if i = m then () else
150 let k = i*4*syms in
151 let rec loop2 k l = if l = syms then () else
152 let v = r32 data k in
153 state.keymap.(i).(l) <- v;
154 loop2 (k+4) (l+1)
156 loop2 k 0;
157 loop (i+1);
159 loop 0;
162 let sendwithrep sock s f =
163 Queue.push f state.fifo;
164 sendstr s sock;
167 let padcatl ss =
168 let b = Buffer.create 16 in
169 List.iter (Buffer.add_string b) ss;
170 let bl = Buffer.length b in
171 let pl = bl land 3 in
172 if pl != 0
173 then (
174 let pad = "123" in
175 Buffer.add_substring b pad 0 (4 - pl);
177 Buffer.contents b;
180 let padcat s1 s2 = padcatl [s1; s2];;
182 let internreq name onlyifexists =
183 let s = "\016\000\000\000\000\000\000\000" in
184 let s = padcat s name in
185 w8 s 1 (if onlyifexists then 1 else 0);
186 w16 s 2 (String.length s / 4);
187 w16 s 4 (String.length name);
191 let sendintern sock s onlyifexists f =
192 let s = internreq s onlyifexists in
193 sendwithrep sock s f;
196 let createwindowreq wid parent x y w h bw mask =
197 let s = "\001\000\009\000wwwwppppxxyywwhhbwccvvvvmmmmeeee" in
198 w32 s 4 wid;
199 w32 s 8 parent;
200 w16 s 12 x;
201 w16 s 14 y;
202 w16 s 16 w;
203 w16 s 18 h;
204 w16 s 20 bw;
205 w16 s 22 1;
206 w32 s 24 0;
207 w32 s 28 0x800; (* eventmask *)
208 w32 s 32 mask;
212 let mapreq wid =
213 let s = "\008u\002\000wwww" in
214 w32 s 4 wid;
218 let getkeymapreq first count =
219 let s = "\101u\002\000fcuu" in
220 w8 s 4 first;
221 w8 s 5 count;
225 let changepropreq wid prop typ format props =
226 let s = "\018\000llwwwwppppttttfuuuLLLL" in
227 let s = padcat s props in
228 w16 s 2 (String.length s / 4);
229 w32 s 4 wid;
230 w32 s 8 prop;
231 w32 s 12 typ;
232 w8 s 16 format;
233 let ful = String.length props / (match format with
234 | 8 -> 1
235 | 16 -> 2
236 | 32 -> 4
237 | n -> error "no idea what %d means" n)
239 w32 s 20 ful;
243 let openfontreq fid name =
244 let s = "\045ullffffnnuu" in
245 let s = padcat s name in
246 w16 s 2 (String.length s / 4);
247 w32 s 4 fid;
248 w16 s 8 (String.length name);
252 let createglyphcursorreq fid cid cindex =
253 let s = "\094u\008\000ccccffffffffssmmrrggbbRRGGBB" in
254 w32 s 4 cid;
255 w32 s 8 fid;
256 w32 s 12 fid;
257 w16 s 16 cindex;
258 w16 s 18 (cindex+1);
259 w16 s 20 0;
260 w16 s 22 0;
261 w16 s 24 0;
262 w16 s 26 0xffff;
263 w16 s 28 0xffff;
264 w16 s 30 0xffff;
268 let mapwindowreq wid =
269 let s = "\008u\002\000wwww" in
270 w32 s 4 wid;
274 let changewindowattributesreq wid mask attrs =
275 let s = "\002ullwwwwmmmm" in
276 let s = padcat s attrs in
277 w16 s 2 (String.length s / 4);
278 w32 s 4 wid;
279 w32 s 8 mask;
283 let configurewindowreq wid mask values =
284 let s = "\012ullwwwwmmuu" in
285 let s = padcat s values in
286 w16 s 2 (String.length s / 4);
287 w32 s 4 wid;
288 w16 s 8 mask;
292 let s32 n =
293 let s = "1234" in
294 w32 s 0 n;
298 let clientmessage format seq wid typ data =
299 let s = "\033fsswwwwtttt" in
300 let s = padcat s data in
301 w8 s 1 format;
302 w16 s 2 seq;
303 w32 s 4 wid;
304 w32 s 8 typ;
308 let sendeventreq propagate destwid mask data =
309 let s = "\025p\011\000wwwwmmmm" in
310 let s = padcat s data in
311 w8 s 1 propagate;
312 w32 s 4 destwid;
313 w32 s 8 mask;
317 let readresp sock =
318 let resp = readstr sock 32 in
319 let opcode = r8 resp 0 in
320 match opcode land lnot 0x80 with
321 | 0 -> (* error *)
322 let s = resp in
323 let code = r8 s 1
324 and serial = r16 s 2
325 and resid = r32 resp 4
326 and min = r16 s 8
327 and maj = r8 s 10 in
328 error "code=%d serial=%d resid=%#x min=%d maj=%d\n%S"
329 code serial resid min maj resp;
331 | 1 -> (* response *)
332 let rep = Queue.pop state.fifo in
333 rep resp;
335 | 2 -> (* key press *)
336 if Array.length state.keymap > 0
337 then
338 let code = r8 resp 1 in
339 let mask = r16 resp 28 in
340 let index = (mask land 1) lxor ((mask land 2) lsr 1) in
341 let keysym = state.keymap.(code-state.mink).(index) in
342 vlog "keysym = %x %c" keysym (Char.unsafe_chr keysym);
343 state.t#key keysym mask;
345 | 3 -> (* key release *)
346 if Array.length state.keymap > 0
347 then
348 let code = r8 resp 1 in
349 let mask = r16 resp 28 in
350 let index = (mask land 1) lxor ((mask land 2) lsr 1) in
351 let keysym = state.keymap.(code-state.mink).(index) in
352 vlog "release keysym = %x %c mask %#x"
353 keysym (Char.unsafe_chr keysym) mask
355 | 4 -> (* buttonpress *)
356 let n = r8 resp 1
357 and x = r16s resp 24
358 and y = r16s resp 26
359 and m = r16 resp 28 in
360 state.t#mouse n true x y m;
361 vlog "press %d" n
363 | 5 -> (* buttonrelease *)
364 let n = r8 resp 1
365 and x = r16s resp 24
366 and y = r16s resp 26
367 and m = r16 resp 28 in
368 state.t#mouse n false x y m;
369 vlog "release %d %d %d" n x y
371 | 6 -> (* motion *)
372 let x = r16s resp 24 in
373 let y = r16s resp 26 in
374 let m = r16 resp 28 in
375 if m = 0
376 then state.t#pmotion x y
377 else state.t#motion x y;
378 vlog "move %dx%d => %d" x y m
380 | 7 -> (* enter *)
381 let x = r16s resp 24
382 and y = r16s resp 26 in
383 state.t#enter x y;
384 vlog "enter %d %d" x y
386 | 8 -> (* leave *)
387 state.t#leave
389 | 18 -> vlog "unmap";
391 | 19 -> (* map *)
392 if state.parent = -1 && state.w > 0 && state.h > 0
393 then state.t#reshape state.w state.h;
394 state.t#display;
395 vlog "map";
397 | 12 -> (* exposure *)
398 vlog "exposure";
399 state.t#display
401 | 15 -> (* visibility *)
402 let vis = r8 resp 8 in
403 if vis != 2 then state.t#display;
404 vlog "visibility %d" vis;
406 | 34 -> (* mapping *)
407 state.keymap <- [||];
408 let s = getkeymapreq state.mink (state.maxk-state.mink-1) in
409 sendwithrep sock s (updkmap sock);
411 | 33 -> (* clientmessage *)
412 let atom = r32 resp 8 in
413 if atom = state.protoatom
414 then (
415 let atom = r32 resp 12 in
416 if atom = state.deleatom
417 then raise Quit;
419 vlog "atom %#x" atom
421 | 21 -> (* reparent *)
422 state.parent <- r32 resp 24;
423 state.w <- -1;
424 state.h <- -1;
425 vlog "reparent"
427 | 22 -> (* configure *)
428 vlog "configure";
429 let w = r16 resp 20
430 and h = r16 resp 22 in
431 if w != state.w || h != state.h
432 then (
433 state.w <- w;
434 state.h <- h;
435 state.t#reshape w h;
438 | n ->
439 dolog "event %d %S" n resp
442 let readresp sock =
443 let rec loop () =
444 readresp sock;
445 if hasdata sock then loop ();
447 loop ();
450 let hexstr s =
451 let b = Buffer.create 16 in
452 String.iter (fun c ->
453 Buffer.add_string b (Printf.sprintf "%02x" (Char.code c))) s;
454 Buffer.contents b;
457 let reshape w h =
458 if state.fs
459 then
460 state.fullscreen state.idbase
462 let s = "wwuuhhuu" in
463 w32 s 0 w;
464 w32 s 4 h;
465 let s = configurewindowreq state.idbase 0x000c s in
466 vlog "reshape %d %s %d" state.seq (hexstr s) (String.length s);
467 sendstr s state.sock;
470 let setup sock screennum =
471 let s = readstr sock 2 in
472 let n = String.length s in
473 if n != 2
474 then error "failed to read X connection setup response n=%d" n;
475 match s.[0] with
476 | '\000' ->
477 let reasonlen = r8 s 1 in
478 let s = readstr sock 6 in
479 let maj = r16 s 0
480 and min = r16 s 2
481 and add = r16 s 4 in
482 let len = add*4 in
483 let data = readstr sock len in
484 let reason = String.sub data 0 reasonlen in
485 error "X connection failed maj=%d min=%d reason=%S"
486 maj min reason
488 | '\002' -> failwith "X connection setup failed: authentication required";
490 | '\001' ->
491 let s = readstr sock 38 in
492 let maj = r16 s 0
493 and min = r16 s 2
494 and add = r16 s 4
495 and idbase = r32 s 10
496 and idmask = r32 s 14
497 and vlen = r16 s 22
498 and screens = r8 s 26
499 and formats = r8 s 27
500 and minkk = r8 s 32
501 and maxkk = r8 s 33 in
502 let data = readstr sock (4*add-32) in
503 let vendor = String.sub data 0 vlen in
504 let pos = ((vlen+3) land lnot 3) + formats*8 in
506 if screennum >= screens
507 then error "invalid screen %d, max %d" screennum (screens-1);
509 let pos =
510 let s = data in
511 let rec findscreen n pos = if n = screennum then pos else
512 let pos =
513 let ndepths = r8 s (pos+39) in
514 let rec skipdepths n pos = if n = ndepths then pos else
515 let pos =
516 let nvisiuals = r16 s (pos+2) in
517 pos + nvisiuals*24 + 8
519 skipdepths (n+1) pos
521 skipdepths n (pos+40)
523 findscreen (n+1) pos
525 findscreen 0 pos
527 let root = r32 data pos in
528 let w = r16 data (pos+20)
529 and h = r16 data (pos+22) in
530 state.mink <- minkk;
531 state.maxk <- maxkk;
532 state.idbase <- idbase;
533 vlog "vendor = %S, maj=%d min=%d" vendor maj min;
534 vlog "screens = %d formats = %d" screens formats;
535 vlog "minkk = %d maxkk = %d" minkk maxkk;
536 vlog "idbase = %#x idmask = %#x" idbase idmask;
537 vlog "root=%#x %dx%d" root w h;
539 let mask = 0
540 + 0x00000001 (* KeyPress *)
541 (* + 0x00000002 *) (* KeyRelease *)
542 + 0x00000004 (* ButtonPress *)
543 + 0x00000008 (* ButtonRelease *)
544 + 0x00000010 (* EnterWindow *)
545 + 0x00000020 (* LeaveWindow *)
546 + 0x00000040 (* PointerMotion *)
547 (* + 0x00000080 *) (* PointerMotionHint *)
548 (* + 0x00000100 *) (* Button1Motion *)
549 (* + 0x00000200 *) (* Button2Motion *)
550 (* + 0x00000400 *) (* Button3Motion *)
551 (* + 0x00000800 *) (* Button4Motion *)
552 (* + 0x00001000 *) (* Button5Motion *)
553 + 0x00002000 (* ButtonMotion *)
554 (* + 0x00004000 *) (* KeymapState *)
555 + 0x00008000 (* Exposure *)
556 + 0x00010000 (* VisibilityChange *)
557 + 0x00020000 (* StructureNotify *)
558 (* + 0x00040000 *) (* ResizeRedirect *)
559 (* + 0x00080000 *) (* SubstructureNotify *)
560 (* + 0x00100000 *) (* SubstructureRedirect *)
561 (* + 0x00200000 *) (* FocusChange *)
562 (* + 0x00400000 *) (* PropertyChange *)
563 (* + 0x00800000 *) (* ColormapChange *)
564 (* + 0x01000000 *) (* OwnerGrabButton *)
566 let wid = state.idbase in
567 let s = createwindowreq wid root 0 0 state.w state.h 0 mask in
568 sendstr s sock;
570 let s = mapreq wid in
571 sendstr s sock;
573 let s = getkeymapreq state.mink (state.maxk-state.mink) in
574 sendwithrep sock s (updkmap sock);
576 sendintern sock "WM_PROTOCOLS" false (fun resp ->
577 state.protoatom <- r32 resp 8;
578 sendintern sock "WM_DELETE_WINDOW" false (fun resp ->
579 state.deleatom <- r32 resp 8;
580 let s = s32 state.deleatom in
581 let s = changepropreq wid state.protoatom 4 32 s in
582 sendstr s sock;
586 sendintern sock "WM_CLASS" false (fun resp ->
587 let atom = r32 resp 8 in
588 let llpp = "llpp\000llpp\000" in
589 let s = changepropreq wid atom 31 8 llpp in
590 sendstr s sock;
593 let s = openfontreq (wid+1) "cursor" in
594 sendstr s sock;
596 Array.iteri (fun i glyphindex ->
597 let s = createglyphcursorreq (wid+1) (wid+2+i) glyphindex in
598 sendstr s sock;
599 ) [|34;48;50;58;128;152|];
601 sendintern sock "UTF8_STRING" false (fun resp ->
602 state.stringatom <- r32 resp 8;
605 sendintern sock "_NET_WM_STATE" true (fun resp ->
606 let nwmsatom = r32 resp 8 in
607 sendintern sock "_NET_WM_STATE_FULLSCREEN" true (fun resp ->
608 let fsatom = r32 resp 8 in
610 state.fullscreen <-
611 (fun wid ->
612 let data = String.make 20 '\000' in
613 state.fs <- not state.fs;
614 w32 data 0 (if state.fs then 1 else 0);
615 w32 data 4 fsatom;
617 let cm = clientmessage 32 0 wid nwmsatom data in
618 let s = sendeventreq 0 root 0x180000 cm in
619 sendstr s sock;
624 glx wid
626 | c ->
627 error "unknown conection setup response %d" (Char.code c)
630 let getauth haddr dnum =
631 let haddr =
632 if haddr = "localhost"
633 then Unix.gethostname ()
634 else haddr
636 let path = Filename.concat (Sys.getenv "HOME") ".Xauthority" in
637 let ic = open_in_bin path in
638 let input_string ic len =
639 let s = String.create len in
640 really_input ic s 0 len;
643 let r16 s =
644 let rb pos = Char.code (String.get s pos) in
645 (rb 1) lor ((rb 0) lsl 8)
647 let rec find () =
648 let rs () =
649 let s = input_string ic 2 in
650 let n = r16 s in
651 input_string ic n
653 let family = input_string ic 2 in
654 let addr = rs () in
655 let nums = rs () in
656 let num = int_of_string nums in
657 let name = rs () in
658 let data = rs () in
660 vlog "family %S addr %S(%S) num %d(%d) name %S data %S"
661 family addr haddr num dnum name data;
662 if addr = haddr && num = dnum
663 then name, data
664 else find ()
666 let name, data =
667 try find () with _ -> "", ""
669 close_in ic;
670 name, data;
673 let init t =
674 let d = Sys.getenv "DISPLAY" in
675 let colonpos = String.index d ':' in
676 let host = String.sub d 0 colonpos in
677 let dispnum, screennum =
679 let dotpos = String.index_from d (colonpos + 1) '.' in
680 let disp = String.sub d (colonpos + 1) (dotpos - colonpos - 1) in
681 let screen = String.sub d (dotpos + 1) (String.length d - dotpos - 1) in
682 int_of_string disp, int_of_string screen
683 with Not_found ->
684 let disp = String.sub d (colonpos + 1) (String.length d - colonpos - 1) in
685 int_of_string disp, 0
687 let aname, adata = getauth host dispnum in
688 let fd =
689 if String.length host = 0 || host = "unix"
690 then
691 let addr = Unix.ADDR_UNIX ("/tmp/.X11-unix/X" ^ string_of_int dispnum) in
692 let fd = Unix.socket Unix.PF_UNIX Unix.SOCK_STREAM 0 in
693 Unix.connect fd addr;
695 else
696 let h = Unix.gethostbyname host in
697 let addr = h.Unix.h_addr_list.(0) in
698 let port = 6000 + dispnum in
699 let fd = Unix.socket Unix.PF_INET Unix.SOCK_STREAM 0 in
700 Unix.connect fd (Unix.ADDR_INET (addr, port));
703 cloexec fd;
704 let s = "luMMmmnndduu" in
705 let s = padcat s aname in
706 let s = padcat s adata in
707 w16 s 2 11;
708 w16 s 4 0;
709 w16 s 6 (String.length aname);
710 w16 s 8 (String.length adata);
711 sendstr s fd;
712 state.sock <- fd;
713 setup fd screennum;
714 state.t <- t;
718 let settitle s =
719 let s = changepropreq state.idbase 39 state.stringatom 8 s in
720 sendstr s state.sock;
723 let setcursor cursor =
724 let n =
725 match cursor with
726 | CURSOR_INHERIT -> -1
727 | CURSOR_INFO -> 3
728 | CURSOR_CYCLE -> 2
729 | CURSOR_CROSSHAIR -> 0
730 | CURSOR_TEXT -> 5
732 let s = s32 (if n = -1 then 0 else state.idbase+2+n) in
733 let s = changewindowattributesreq state.idbase (*cursor*)0x4000 s in
734 sendstr s state.sock;
737 let fullscreen () =
738 state.fullscreen state.idbase;
741 let withalt mask = mask land 8 != 0;;
742 let withctrl mask = mask land 4 != 0;;
743 let withshift mask = mask land 1 != 0;;