diff options
Diffstat (limited to 'appl/collab/connect.b')
| -rw-r--r-- | appl/collab/connect.b | 156 |
1 files changed, 156 insertions, 0 deletions
diff --git a/appl/collab/connect.b b/appl/collab/connect.b new file mode 100644 index 00000000..a8dbffde --- /dev/null +++ b/appl/collab/connect.b @@ -0,0 +1,156 @@ +implement Connect; + +include "sys.m"; + sys: Sys; + +include "draw.m"; +include "keyring.m"; +include "security.m"; +include "arg.m"; + +Connect: module +{ + init: fn(ctxt: ref Draw->Context, argv: list of string); +}; + +remotedir := "/n/remote"; +localdir := "/n/ftree/collab"; +COLLABPORT: con "9999"; # TO DO: needs symbolic name in services + +usage() +{ + sys->fprint(sys->fildes(2), "Usage: connect [-v] [-C cryptoalg] [-k keyring] [net!addr [localdir]]\n"); + raise "fail:usage"; +} + +init(nil: ref Draw->Context, args: list of string) +{ + sys = load Sys Sys->PATH; + + vflag := 0; + alg := "none"; + keyfile := ""; + netaddr := "$collab"; + + arg := load Arg Arg->PATH; + if(arg != nil){ + arg->init(args); + while((c := arg->opt()) != 0) + case c { + 'C' => + alg = arg->arg(); + 'k' => + keyfile = arg->arg(); + 'v' => + vflag++; + * => + usage(); + } + } + args = arg->argv(); + arg = nil; + + if(args != nil){ + netaddr = hd args; + args = tl args; + if(args != nil) + localdir = hd args; + } + + if(vflag) + sys->print("connect: dial %s\n", netaddr); + (fd, user) := authdial(netaddr, keyfile, alg); + if(vflag) + sys->print("remote username is %s\n", user); + if(sys->mount(fd, nil, remotedir, Sys->MREPL, nil) < 0) + error(sys->sprint("can't mount %s on %s: %r", netaddr, remotedir)); + fd = nil; + + connectdir := remotedir+"/collab"; + if (sys->bind(connectdir, localdir, Sys->MCREATE|Sys->MREPL) < 0){ + error(sys->sprint("cannot bind %s onto %s: %r\n", connectdir, localdir)); + raise "fail:error"; + } + + # if something such as ftree is running and watching for changes, tell it about this one + fd = sys->open("/chan/nsupdate", Sys->OWRITE); + if(fd != nil) + sys->fprint(fd, "/n/ftree/collab"); + if(vflag) + sys->print("collab connected\n"); +} + +authdial(addr, keyfile, alg: string): (ref Sys->FD, string) +{ + cert : string; + + kr := load Keyring Keyring->PATH; + if(kr == nil) + error(sys->sprint("cannot load %s: %r", Keyring->PATH)); + + kd := "/usr/" + user() + "/keyring/"; + if (keyfile == nil) { + cert = kd + netmkaddr(addr, "tcp", ""); + (ok, nil) := sys->stat(cert); + if (ok < 0) + cert = kd + "default"; + } + else if (len keyfile > 0 && keyfile[0] != '/') + cert = kd + keyfile; + else + cert = keyfile; + ai := kr->readauthinfo(cert); + if (ai == nil) + error(sys->sprint("cannot read authentication data from %s: %r", cert)); + + au := load Auth Auth->PATH; + if(au == nil) + error(sys->sprint("cannot load %s: %r", Auth->PATH)); + err := au->init(); + if(err != nil) + error(sys->sprint("cannot init Auth: %s", err)); + + (ok, c) := sys->dial(netmkaddr(addr, "tcp", COLLABPORT), nil); + if(ok < 0) + error(sys->sprint("can't dial %s: %r", addr)); + (fd, id_or_err) := au->client(alg, ai, c.dfd); + if(fd == nil) + error(sys->sprint("authentication failed: %s", id_or_err)); + + return (fd, id_or_err); +} + +user(): string +{ + fd := sys->open("/dev/user", sys->OREAD); + if(fd == nil) + return ""; + + buf := array[Sys->NAMEMAX] of byte; + n := sys->read(fd, buf, len buf); + if(n < 0) + return ""; + + return string buf[0:n]; +} + +netmkaddr(addr, net, svc: string): string +{ + if(net == nil) + net = "net"; + (n, l) := sys->tokenize(addr, "!"); + if(n <= 1){ + if(svc== nil) + return sys->sprint("%s!%s", net, addr); + return sys->sprint("%s!%s!%s", net, addr, svc); + } + if(svc == nil || n > 2) + return addr; + return sys->sprint("%s!%s", addr, svc); +} + +error(m: string) +{ + sys->fprint(sys->fildes(2), "connect: %s\n", m); + raise "fail:error"; +} |