ref: 1018d7a7091fd5e0433f501a3474cdc3fec322cc
dir: /appl/cmd/mouse.b/
implement mouse; # ported from plan 9's aux/mouse include "sys.m"; sys: Sys; sprint, fprint, sleep: import sys; include "draw.m"; stderr: ref Sys->FD; mouse: module { init: fn(ctxt: ref Draw->Context, argv: list of string); }; Sleep500: con 500; Sleep1000: con 1000; Sleep2000: con 2000; TIMEOUT: con 5000; fail := "fail:"; usage() { fprint(stderr, "usage: mouse [type]\n"); raise fail+"usage"; } write(fd: ref Sys->FD, buf: array of byte, n: int): int { if (debug) { sys->fprint(stderr, "write(%d) ", fd.fd); for (i := 0; i < len buf; i++) { sys->fprint(stderr, "'%c' ", int buf[i]); } sys->fprint(stderr, "\n"); } return sys->write(fd, buf, n); } speeds := array[] of {"b1200", "b2400", "b4800", "b9600"}; debug := 0; can9600 := 0; init(nil: ref Draw->Context, argv: list of string) { sys = load Sys Sys->PATH; stderr = sys->fildes(2); { if (argv == nil) usage(); argv = tl argv; while (argv != nil && len (arg := hd argv) > 1 && arg[0] == '-') { case arg[1] { 'D' => debug = 1; * => usage(); } argv = tl argv; } if (len argv > 1) usage(); p: string; if (argv == nil) p = mouseprobe(); else p = hd argv; if (p != nil && !isnum(p)) { mouseconfig(p); return; } if (p == nil) { serial("0"); serial("1"); fprint(stderr, "mouse: no mouse detected\n"); } else { err := serial(p); fprint(stderr, "mouse: %s\n", err); } } exception{ # this could be taken out so the shell could # get an indication that the command has failed. "fail:*" => ; } } # probe for a serial mouse on port p; # return some an error string if not found. serial(p: string): string { baud := 0; f := sys->sprint("/dev/eia%sctl", p); if ((ctl := sys->open(f, Sys->ORDWR)) == nil) return sprint("can't open %s - %r\n", f); f = sys->sprint("/dev/eia%s", p); if ((data := sys->open(f, Sys->ORDWR)) == nil) return sprint("can't open %s - %r\n", f); if(debug) fprint(stderr, "ctl=%d, data=%d\n", ctl.fd, data.fd); if(debug) fprint(stderr, "MorW()\n"); mtype := MorW(ctl, data); if (mtype == 0) { if(debug) return "no mouse detected"; if(debug) fprint(stderr, "C()\n"); mtype = C(ctl, data); } if (mtype == 0) return "no mouse detected on port "+p; if(debug)fprint(stderr, "done eia setup\n"); mt := "serial " + p; case mtype { * => return "unknown mouse type"; 'C' => if(debug) fprint(stderr, "Logitech 5 byte mouse\n"); Cbaud(ctl, data, baud); 'W' => if(debug) fprint(stderr, "Type W mouse\n"); Wbaud(ctl, data, baud); 'M' => if(debug) fprint(stderr, "Microsoft compatible mouse\n"); mt += " M"; } mouseconfig(mt); return nil; } mouseconfig(mt: string) { if ((conf := sys->open("/dev/mousectl", Sys->OWRITE)) == nil) { fprint(stderr, "mouse: can't open mousectl - %r\n"); raise fail+"open mousectl"; } if(debug) fprint(stderr, "opened mousectl\n"); if (write(conf, array of byte mt, len array of byte mt) < 0) { fprint(stderr, "mouse: error setting mouse type - %r\n"); raise fail+"write conf"; } fprint(stderr, "mouse: configured as '%s'\n", mt); } isnum(s: string): int { for (i := 0; i < len s; i++) if (s[i] < '0' || s[i] > '9') return 0; return 1; } mouseprobe(): string { if ((probe := sys->open("/dev/mouseprobe", Sys->OREAD)) == nil) { fprint(stderr, "mouse: can't open mouseprobe - %r\n"); return nil; } buf := array[64] of byte; n := sys->read(probe, buf, len buf); if (n <= 0) return nil; if (buf[n - 1] == byte '\n') n--; if(debug) fprint(stderr, "mouse probe detected mouse of type '%s'\n", string buf[0:n]); return string buf[0:n]; } readbyte(fd: ref Sys->FD): int { buf := array[1] of byte; (n, err) := timedread(fd, buf, 1, 200); if (n < 0) { if (err == nil) return -1; fprint(stderr, "mouse: readbyte failed - %s\n", err); raise fail+"read failed"; } return int buf[0]; } slowread(fd: ref Sys->FD, buf: array of byte, nbytes: int, msg: string): int { for (i := 0; i < nbytes; i++) { if ((c := readbyte(fd)) == -1) break; buf[i] = byte c; } if(debug) dumpbuf(buf[0:i], msg); return i; } dumpbuf(buf: array of byte, msg: string) { sys->fprint(stderr, "%s", msg); for (i := 0; i < len buf; i++) sys->fprint(stderr, "#%ux ", int buf[i]); sys->fprint(stderr, "\n"); } toggleRTS(fd: ref Sys->FD) { # reset the mouse (toggle RTS) # must be >100mS writes(fd, "d0"); sleep(10); writes(fd, "r0"); sleep(Sleep500); writes(fd, "d1"); sleep(10); writes(fd, "r1"); sleep(Sleep500); } setupeia(fd: ref Sys->FD, baud, bits: string) { # set the speed to 1200/2400/4800/9600 baud, # 7/8-bit data, one stop bit and no parity (abaud, abits) := (array of byte baud, array of byte bits); if(debug)sys->fprint(stderr, "setupeia(%s,%s)\n", baud, bits); write(fd, abaud, len abaud); write(fd, abits, len abits); writes(fd, "s1"); writes(fd, "pn"); } # check for types M, M3 & W # # we talk to all these mice using 1200 baud MorW(ctl, data: ref Sys->FD): int { # set up for type M, V or W # flush any pending data setupeia(ctl, "b1200", "l7"); toggleRTS(ctl); if(debug)sys->fprint(stderr, "toggled RTS\n"); buf := array[256] of byte; while (slowread(data, buf, len buf, "flush: ") > 0) ; if(debug) sys->fprint(stderr, "done slowread\n"); toggleRTS(ctl); # see if there's any data from the mouse # (type M, V and W mice) c := slowread(data, buf, len buf, "check M: "); # type M, V and W mice return "M" or "M3" after reset. # check for type W by sending a 'Send Standard Configuration' # command, "*?". if (c > 0 && int buf[0] == 'M') { writes(data, "*?"); c = slowread(data, buf, len buf, "check W: "); # 4 bytes back indicates a type W mouse if (c == 4) { if (int buf[1] & (1<<4)) can9600 = 1; setupeia(ctl, "b1200", "l8"); writes(data, "*U"); slowread(data, buf, len buf, "check W: "); return 'W'; } return 'M'; } return 0; } # check for type C by seeing if it responds to the status # command "s". the mouse is at an unknown speed so we # have to check all possible speeds. C(ctl, data: ref Sys->FD): int { buf := array[256] of byte; for (s := speeds; len s > 0; s = s[1:]) { if (debug) sys->print("%s\n", s[0]); setupeia(ctl, s[0], "l8"); writes(data, "s"); c := slowread(data, buf, len buf, "check C: "); if (c >= 1 && (int buf[0] & 16rbf) == 16r0f) { sleep(100); writes(data, "*n"); sleep(100); setupeia(ctl, "b1200", "l8"); writes(data, "s"); c = slowread(data, buf, len buf, "recheck C: "); if (c >= 1 && (int buf[0] & 16rbf) == 16r0f) { writes(data, "U"); return 'C'; } } sleep(100); } return 0; } Cbaud(ctl, data: ref Sys->FD, baud: int) { buf := array[2] of byte; case baud { 0 or 1200 => return; 2400 => buf[1] = byte 'o'; 4800 => buf[1] = byte 'p'; 9600 => buf[1] = byte 'q'; * => fprint(stderr, "mouse: can't set baud rate, mouse at 1200\n"); return; } buf[0] = byte '*'; sleep(100); write(data, buf, 2); sleep(100); write(data, buf, 2); setupeia(ctl, sys->sprint("b%d", baud), "l8"); } Wbaud(ctl, data: ref Sys->FD, baud: int) { case baud { 0 or 1200 => return; * => if (baud == 9600 && can9600) break; fprint(stderr, "mouse: can't set baud rate, mouse at 1200\n"); return; } writes(data, "*q"); setupeia(ctl, "b9600", "l8"); slowread(data, array[32] of byte, 32, "setbaud: "); } readproc(fd: ref Sys->FD, buf: array of byte, n: int, pidch: chan of int, ch: chan of (int, string)) { s: string; pidch <-= sys->pctl(0, nil); n = sys->read(fd, buf, n); if (n < 0) s = sys->sprint("read: %r"); ch <-= (n, s); } sleepproc(t: int, pidch: chan of int, ch: chan of (int, string)) { pidch <-= sys->pctl(0, nil); sys->sleep(t); ch <-= (-1, nil); } timedread(fd: ref Sys->FD, buf: array of byte, n: int, t: int): (int, string) { pidch := chan of int; retch := chan of (int, string); spawn readproc(fd, buf, n, pidch, retch); wpid := <-pidch; spawn sleepproc(t, pidch, retch); spid := <-pidch; (nr, err) := <-retch; if (nr == -1 && err == nil) kill(wpid); else kill(spid); return (nr, err); } kill(pid: int) { if ((fd := sys->open("#p/"+string pid+"/ctl", Sys->OWRITE)) == nil) { fprint(stderr, "couldn't kill %d: %r\n", pid); return; } sys->write(fd, array of byte "kill", 4); } writes(fd: ref Sys->FD, s: string): int { a := array of byte s; return write(fd, a, len a); }