ref: 21947578303cdb0fcb4b0cdfe80aa9dab5f6b447
dir: /appl/wm/logon.b/
implement WmLogon;
#
# Logon program for Wm environment
#
include "sys.m";
sys: Sys;
include "draw.m";
draw: Draw;
Screen, Display, Image, Context, Point, Rect: import draw;
ctxt: ref Context;
include "tk.m";
tk: Tk;
include "tkclient.m";
tkclient: Tkclient;
include "readdir.m";
include "arg.m";
include "sh.m";
include "newns.m";
include "keyring.m";
include "security.m";
WmLogon: module
{
init: fn(ctxt: ref Draw->Context, argv: list of string);
};
cfg := array[] of {
"label .p -bitmap @/icons/inferno.bit -borderwidth 2 -relief raised",
"frame .l -bg red",
"label .l.u -fg black -bg silver -text {User Name:} -anchor w",
"pack .l.u -fill x",
"frame .e",
"entry .e.u -bg white",
"pack .e.u -fill x",
"frame .f -borderwidth 2 -relief raised",
"pack .l .e -side left -in .f",
"pack .p .f -fill x",
"bind .e.u <Key-\n> {send cmd ok}",
"focus .e.u"
};
listcfg := array[] of {
"frame .f",
"listbox .f.lb -yscrollcommand {.f.sb set}",
"scrollbar .f.sb -orient vertical -command {.f.lb yview}",
"button .login -text {Login} -command {send cmd login}",
"pack .f.sb .f.lb -in .f -side left -fill both -expand 1",
"pack .f -side top -anchor center -fill y -expand 1",
"pack .login -side top",
# "pack propagate . 0",
};
init(actxt: ref Draw->Context, args: list of string)
{
sys = load Sys Sys->PATH;
draw = load Draw Draw->PATH;
tk = load Tk Tk->PATH;
tkclient = load Tkclient Tkclient->PATH;
if(tkclient == nil){
sys->fprint(stderr(), "logon: cannot load %s: %r\n", Tkclient->PATH);
raise "fail:bad module";
}
sys->pctl(Sys->NEWPGRP|Sys->FORKFD, nil);
tkclient->init();
ctxt = actxt;
dolist := 0;
usr := "";
nsfile := "namespace";
arg := load Arg Arg->PATH;
if(arg != nil){
arg->init(args);
arg->setusage("logon [-l] [-n namespace] [-u user]");
while((opt := arg->opt()) != 0){
case opt{
'u' =>
usr = arg->earg();
'l' =>
dolist = 1;
'n' =>
nsfile = arg->earg();
* =>
arg->usage();
}
}
args = arg->argv();
arg = nil;
} else
args = nil;
if(ctxt == nil)
sys->fprint(stderr(), "logon: must run under a window manager\n");
(ctlwin, nil) := tkclient->toplevel(ctxt, nil, nil, Tkclient->Plain);
if(sys->fprint(ctlwin.ctxt.connfd, "request") == -1){
sys->fprint(stderr(), "logon: must be run as principal wm application\n");
raise "fail:lack of control";
}
if(dolist)
usr = chooseuser(ctxt);
if (usr == nil || !logon(usr)) {
(panel, cmd) := makepanel(ctxt, cfg);
stop := chan of int;
spawn tkclient->handler(panel, stop);
for(;;) {
tk->cmd(panel, "focus .e.u; update");
<-cmd;
usr = tk->cmd(panel, ".e.u get");
if(usr == "") {
notice("You must supply a user name to login");
continue;
}
if(logon(usr)) {
panel = nil;
stop <-= 1;
break;
}
tk->cmd(panel, ".e.u delete 0 end");
}
}
ok: int;
if(nsfile != nil){
(ok, nil) = sys->stat(nsfile);
if(ok < 0){
nsfile = nil;
(ok, nil) = sys->stat("namespace");
}
}else
(ok, nil) = sys->stat("namespace");
if(ok >= 0) {
ns := load Newns Newns->PATH;
if(ns == nil)
notice("failed to load namespace builder");
else if ((nserr := ns->newns(nil, nsfile)) != nil)
notice("namespace error:\n"+nserr);
}
tkclient->wmctl(ctlwin, "endcontrol");
errch := chan of string;
spawn exec(ctxt, args, errch);
err := <-errch;
if (err != nil) {
sys->fprint(stderr(), "logon: %s\n", err);
raise "fail:exec failed";
}
}
makepanel(ctxt: ref Draw->Context, cmds: array of string): (ref Tk->Toplevel, chan of string)
{
(t, nil) := tkclient->toplevel(ctxt, "-bg silver", nil, Tkclient->Plain);
cmd := chan of string;
tk->namechan(t, cmd, "cmd");
for(i := 0; i < len cmds; i++)
tk->cmd(t, cmds[i]);
err := tk->cmd(t, "variable lasterr");
if(err != nil) {
sys->fprint(stderr(), "logon: tk error: %s\n", err);
raise "fail:config error";
}
tk->cmd(t, "update");
centre(t);
tkclient->startinput(t, "kbd" :: "ptr" :: nil);
tkclient->onscreen(t, "onscreen");
return (t, cmd);
}
exec(ctxt: ref Draw->Context, argv: list of string, errch: chan of string)
{
sys->pctl(sys->NEWFD, 0 :: 1 :: 2 :: nil);
{
argv = "/dis/wm/toolbar.dis" :: nil;
cmd := load Command hd argv;
if (cmd == nil) {
errch <-= sys->sprint("cannot load %s: %r", hd argv);
} else {
errch <-= nil;
spawn cmd->init(ctxt, argv);
}
}exception{
"fail:*" =>
exit;
}
}
logon(user: string): int
{
userdir := "/usr/"+user;
if(sys->chdir(userdir) < 0) {
notice("There is no home directory for \""+
user+"\"\nmounted on this machine");
return 0;
}
chmod("/chan", Sys->DMDIR|8r777);
chmod("/chan/wmrect", 8r666);
chmod("/chan/wmctl", 8r666);
#
# Set the user id
#
fd := sys->open("/dev/user", sys->OWRITE);
if(fd == nil) {
notice(sys->sprint("failed to open /dev/user: %r"));
return 0;
}
b := array of byte user;
if(sys->write(fd, b, len b) < 0) {
notice("failed to write /dev/user\nwith error "+sys->sprint("%r"));
return 0;
}
return 1;
}
chmod(file: string, mode: int): int
{
d := sys->nulldir;
d.mode = mode;
if(sys->wstat(file, d) < 0){
notice(sys->sprint("failed to chmod %s: %r", file));
return -1;
}
return 0;
}
chooseuser(ctxt: ref Draw->Context): string
{
(t, cmd) := makepanel(ctxt, listcfg);
usrlist := getusers();
if(usrlist == nil)
usrlist = "inferno" :: nil;
for(; usrlist != nil; usrlist = tl usrlist)
tkcmd(t, ".f.lb insert end '" + hd usrlist);
tkcmd(t, "update");
stop := chan of int;
spawn tkclient->handler(t, stop);
u := "";
for(;;){
<-cmd;
sel := tkcmd(t, ".f.lb curselection");
if(sel == nil)
continue;
u = tkcmd(t, ".f.lb get " + sel);
if(u != nil)
break;
}
stop <-= 1;
return u;
}
getusers(): list of string
{
readdir := load Readdir Readdir->PATH;
if(readdir == nil)
return nil;
(dirs, nil) := readdir->init("/usr", Readdir->NAME);
n: list of string;
for (i := len dirs -1; i >=0; i--)
if (dirs[i].qid.qtype & Sys->QTDIR)
n = dirs[i].name :: n;
return n;
}
notecmd := array[] of {
"frame .f",
"label .f.l -bitmap error -foreground red",
"button .b -text Continue -command {send cmd done}",
"focus .f",
"bind .f <Key-\n> {send cmd done}",
"pack .f.l .f.m -side left -expand 1",
"pack .f .b",
"pack propagate . 0",
};
centre(t: ref Tk->Toplevel)
{
org: Point;
ir := tk->rect(t, ".", Tk->Border|Tk->Required);
org.x = t.screenr.dx() / 2 - ir.dx() / 2;
org.y = t.screenr.dy() / 3 - ir.dy() / 2;
#sys->print("ir: %d %d %d %d\n", ir.min.x, ir.min.y, ir.max.x, ir.max.y);
if (org.y < 0)
org.y = 0;
tk->cmd(t, ". configure -x " + string org.x + " -y " + string org.y);
}
notice(message: string)
{
(t, nil) := tkclient->toplevel(ctxt, "-borderwidth 2 -relief raised", nil, Tkclient->Plain);
cmd := chan of string;
tk->namechan(t, cmd, "cmd");
tk->cmd(t, "label .f.m -anchor nw -text '"+message);
for(i := 0; i < len notecmd; i++)
tk->cmd(t, notecmd[i]);
centre(t);
tkclient->onscreen(t, "onscreen");
tkclient->startinput(t, "kbd"::"ptr"::nil);
stop := chan of int;
spawn tkclient->handler(t, stop);
tk->cmd(t, "update; cursor -default");
<-cmd;
stop <-= 1;
}
tkcmd(t: ref Tk->Toplevel, cmd: string): string
{
s := tk->cmd(t, cmd);
if (s != nil && s[0] == '!') {
sys->print("%s\n", cmd);
sys->print("tk error: %s\n", s);
}
return s;
}
stderr(): ref Sys->FD
{
return sys->fildes(2);
}
rf(path: string) : string
{
fd := sys->open(path, sys->OREAD);
if(fd == nil)
return nil;
buf := array[512] of byte;
n := sys->read(fd, buf, len buf);
if(n <= 0)
return nil;
return string buf[0:n];
}