void
wclean(Win *w)
{
if(w->body)
Bflush(w->body);
ctlwrite(w, "clean\n");
}
void
wname(Win *w, char *s)
{
char buf[128];
sprint(buf, "name %s\n", s);
ctlwrite(w, buf);
}
void
wselect(Win *w, char *addr)
{
if(w->addr < 0)
w->addr = openfile(w, "addr");
if(write(w->addr, addr, strlen(addr)) < 0)
fprint(2,"writing addr");
ctlwrite(w, "dot=addr\n");
}
void
wnew(Win *w)
{
char buf[12];
w->ctl = open("/mnt/acme/new/ctl", ORDWR);
if(w->ctl<0 || read(w->ctl, buf, 12)!=12)
fprint (2, "can't open window ctl file: %r");
ctlwrite(w, "noscroll\n");
w->winid = atoi(buf);
w->event = openfile(w, "event");
w->addr = -1; /* will be opened when needed */
w->body = nil;
w->data = -1;
}
void
fswrite(Req *r)
{
switch((int)r->fid->qid.path){
case Qroot:
case Qdir:
respond(r, "write to a directory?");
break;
case Qctl:
ctlwrite(r);
break;
default:
rdwrpart(r);
break;
}
}
static void
fswrite(Req* r)
{
Query* q;
File* f;
long count;
char* ntext;
int l;
if(r->fid->qid.type&QTDIR){
respond(r, "bug: write on dir");
return;
}
f = r->fid->file;
if(f == ctlf){
ctlwrite(r);
return;
}
count = r->ifcall.count;
r->ofcall.count = count;
q = f->aux;
if(q->expr != nil){
// a previous query was made. start another.
freeexpr(q->expr);
free(q->text);
q->expr = nil;
q->text = nil;
}
/* append text to query text, ignore offset.
*/
l = strlen(q->text);
ntext = emalloc9p(l + count + 1);
strcpy(ntext, q->text);
memmove(ntext+l, r->ifcall.data, count);
ntext[l+count]=0;
free(q->text);
q->text = ntext;
respond(r, nil);
}
static void
fswrite(Req *r)
{
char e[ERRMAX];
uint32_t path;
path = r->fid->qid.path;
switch(TYPE(path)){
default:
snprint(e, sizeof e, "bug in execnet path=%lx", path);
respond(r, e);
break;
case Qctl:
ctlwrite(r, client[NUM(path)]);
break;
case Qdata:
datawrite(r, client[NUM(path)]);
break;
}
}
void
threadmain(int argc, char *argv[])
{
char *mtpt, *s;
char *secstorepw;
char err[ERRMAX];
Dir d;
rfork(RFNOTEG);
mtpt = "/mnt";
extrafactotumdir = 1;
secstorepw = nil;
quotefmtinstall();
fmtinstall('A', attrfmt);
fmtinstall('H', encodefmt);
fmtinstall('N', attrnamefmt);
if(argc == 3 && strcmp(argv[1], "-g") == 0){
gflag(argv[2]);
threadexitsall(nil);
}
ARGBEGIN{
default:
usage();
case 'D':
chatty9p++;
break;
case 'S': /* server: read nvram, no prompting for keys */
askforkeys = 0;
trysecstore = 0;
sflag = 1;
break;
case 'a':
authaddr = EARGF(usage());
break;
case 'd':
debug = 1;
doprivate = 0;
break;
case 'g':
usage();
case 'k': /* reinitialize nvram */
kflag = 1;
break;
case 'm':
mtpt = EARGF(usage());
break;
case 'n':
trysecstore = 0;
break;
case 'p':
doprivate = 0;
break;
case 's':
service = EARGF(usage());
break;
case 'u': /* user: set hostowner */
uflag = 1;
break;
case 'x':
extrafactotumdir = 0;
break;
}ARGEND
if(argc != 0)
usage();
if(doprivate)
private();
initcap();
if(sflag){
s = getnvramkey(kflag ? NVwrite : NVwriteonerr, &secstorepw);
if(s == nil)
fprint(2, "factotum warning: cannot read nvram: %r\n");
else if(ctlwrite(s) < 0)
fprint(2, "factotum warning: cannot add nvram key: %r\n");
if(secstorepw != nil)
trysecstore = 1;
if (s != nil) {
memset(s, 0, strlen(s));
free(s);
}
} else if(uflag)
promptforhostowner();
owner = getuser();
if(trysecstore && havesecstore()){
while(secstorefetch(secstorepw) < 0){
rerrstr(err, sizeof err);
if(strcmp(err, "cancel") == 0)
break;
fprint(2, "secstorefetch: %r\n");
fprint(2, "Enter an empty password to quit.\n");
free(secstorepw);
secstorepw = nil; /* just try nvram pw once */
}
}
fsinit0();
threadpostmountsrv(&fs, service, mtpt, MBEFORE);
if(service){
nulldir(&d);
d.mode = 0666;
s = emalloc(10+strlen(service));
strcpy(s, "/srv/");
strcat(s, service);
if(dirwstat(s, &d) < 0)
fprint(2, "factotum warning: cannot chmod 666 %s: %r\n", s);
free(s);
}
threadexits(nil);
}
void
main(int argc, char **argv)
{
int i, trysecstore;
char err[ERRMAX], *s;
Dir d;
Proto *p;
char *secstorepw;
trysecstore = 1;
secstorepw = nil;
ARGBEGIN{
case 'D':
chatty9p++;
break;
case 'S': /* server: read nvram, no prompting for keys */
askforkeys = 0;
trysecstore = 0;
sflag = 1;
break;
case 'a':
authaddr = EARGF(usage());
break;
case 'd':
debug = 1;
doprivate = 0;
break;
case 'g': /* get: prompt for key for name and domain */
gflag = 1;
break;
case 'k': /* reinitialize nvram */
kflag = 1;
break;
case 'm': /* set default mount point */
mtpt = EARGF(usage());
break;
case 'n':
trysecstore = 0;
break;
case 'p':
doprivate = 0;
break;
case 's': /* set service name */
service = EARGF(usage());
break;
case 'u': /* user: set hostowner */
uflag = 1;
break;
default:
usage();
}ARGEND
if(argc != 0 && !gflag)
usage();
if(doprivate)
private();
initcap();
quotefmtinstall();
fmtinstall('A', _attrfmt);
fmtinstall('N', attrnamefmt);
fmtinstall('H', encodefmt);
ring = emalloc(sizeof(*ring));
notify(notifyf);
if(gflag){
if(argc != 1)
usage();
askuser(argv[0]);
exits(nil);
}
for(i=0; prototab[i]; i++){
p = prototab[i];
if(p->name == nil)
sysfatal("protocol %d has no name", i);
if(p->init == nil)
sysfatal("protocol %s has no init", p->name);
if(p->write == nil)
sysfatal("protocol %s has no write", p->name);
if(p->read == nil)
sysfatal("protocol %s has no read", p->name);
if(p->close == nil)
sysfatal("protocol %s has no close", p->name);
if(p->keyprompt == nil)
p->keyprompt = "";
}
if(sflag){
s = getnvramkey(kflag ? NVwrite : NVwriteonerr, &secstorepw);
if(s == nil)
fprint(2, "factotum warning: cannot read nvram: %r\n");
else if(ctlwrite(s, 0) < 0)
fprint(2, "factotum warning: cannot add nvram key: %r\n");
if(secstorepw != nil)
trysecstore = 1;
if (s != nil) {
memset(s, 0, strlen(s));
free(s);
}
} else if(uflag)
promptforhostowner();
owner = getuser();
if(trysecstore){
if(havesecstore() == 1){
while(secstorefetch(secstorepw) < 0){
rerrstr(err, sizeof err);
if(strcmp(err, "cancel") == 0)
break;
fprint(2, "factotum: secstorefetch: %r\n");
fprint(2, "Enter an empty password to quit.\n");
free(secstorepw);
secstorepw = nil; /* just try nvram pw once */
}
}else{
/*
rerrstr(err, sizeof err);
if(*err)
fprint(2, "factotum: havesecstore: %r\n");
*/
}
}
postmountsrv(&fs, service, mtpt, MBEFORE);
if(service){
nulldir(&d);
d.mode = 0666;
s = emalloc(10+strlen(service));
strcpy(s, "/srv/");
strcat(s, service);
if(dirwstat(s, &d) < 0)
fprint(2, "factotum warning: cannot chmod 666 %s: %r\n", s);
free(s);
}
exits(nil);
}
