123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935 |
- #include <u.h>
- #include <libc.h>
- #include <bio.h>
- #include <String.h>
- #include <thread.h>
- #include "wiki.h"
- #include <auth.h>
- #include <fcall.h>
- #include <9p.h>
- enum {
- Qindexhtml,
- Qindextxt,
- Qraw,
- Qhistoryhtml,
- Qhistorytxt,
- Qdiffhtml,
- Qedithtml,
- Qwerrorhtml,
- Qwerrortxt,
- Nfile,
- };
- static char *filelist[] = {
- "index.html",
- "index.txt",
- "current",
- "history.html",
- "history.txt",
- "diff.html",
- "edit.html",
- "werror.html",
- "werror.txt",
- };
- static int needhist[Nfile] = {
- [Qhistoryhtml] 1,
- [Qhistorytxt] 1,
- [Qdiffhtml] 1,
- };
- /*
- * The qids are <8-bit type><16-bit page number><16-bit page version><8-bit file index>.
- */
- enum { /* <8-bit type> */
- Droot = 1,
- D1st,
- D2nd,
- Fnew,
- Fmap,
- F1st,
- F2nd,
- };
- uvlong
- mkqid(int type, int num, int vers, int file)
- {
- return ((uvlong)type<<40) | ((uvlong)num<<24) | (vers<<8) | file;
- }
- int
- qidtype(uvlong path)
- {
- return (path>>40)&0xFF;
- }
- int
- qidnum(uvlong path)
- {
- return (path>>24)&0xFFFF;
- }
- int
- qidvers(uvlong path)
- {
- return (path>>8)&0xFFFF;
- }
- int
- qidfile(uvlong path)
- {
- return path&0xFF;
- }
- typedef struct Aux Aux;
- struct Aux {
- char *name;
- Whist *w;
- int n;
- ulong t;
- String *s;
- Map *map;
- };
- static void
- fsattach(Req *r)
- {
- Aux *a;
- if(r->ifcall.aname && r->ifcall.aname[0]){
- respond(r, "invalid attach specifier");
- return;
- }
- a = emalloc(sizeof(Aux));
- r->fid->aux = a;
- a->name = estrdup(r->ifcall.uname);
- r->ofcall.qid = (Qid){mkqid(Droot, 0, 0, 42), 0, QTDIR};
- r->fid->qid = r->ofcall.qid;
- respond(r, nil);
- }
- static char*
- fswalk1(Fid *fid, char *name, Qid *qid)
- {
- char *q;
- int i, isdotdot, n, t;
- uvlong path;
- Aux *a;
- Whist *wh;
- String *s;
- isdotdot = strcmp(name, "..")==0;
- n = strtoul(name, &q, 10);
- path = fid->qid.path;
- a = fid->aux;
- switch(qidtype(path)){
- case 0:
- return "wikifs: bad path in server (bug)";
- case Droot:
- if(isdotdot){
- *qid = fid->qid;
- return nil;
- }
- if(strcmp(name, "new")==0){
- *qid = (Qid){mkqid(Fnew, 0, 0, 0), 0, 0};
- return nil;
- }
- if(strcmp(name, "map")==0){
- *qid = (Qid){mkqid(Fmap, 0, 0, 0), 0, 0};
- return nil;
- }
- if((*q!='\0' || (wh=getcurrent(n))==nil)
- && (wh=getcurrentbyname(name))==nil)
- return "file does not exist";
- *qid = (Qid){mkqid(D1st, wh->n, 0, 0), wh->doc->time, QTDIR};
- a->w = wh;
- return nil;
- case D1st:
- if(isdotdot){
- *qid = (Qid){mkqid(Droot, 0, 0, 0), 0, QTDIR};
- return nil;
- }
- /* handle history directories */
- if(*q == '\0'){
- if((wh = gethistory(qidnum(path))) == nil)
- return "file does not exist";
- for(i=0; i<wh->ndoc; i++)
- if(wh->doc[i].time == n)
- break;
- if(i==wh->ndoc){
- closewhist(wh);
- return "file does not exist";
- }
- closewhist(a->w);
- a->w = wh;
- a->n = i;
- *qid = (Qid){mkqid(D2nd, qidnum(path), i, 0), wh->doc[i].time, QTDIR};
- return nil;
- }
- /* handle files other than index */
- for(i=0; i<nelem(filelist); i++){
- if(strcmp(name, filelist[i])==0){
- if(needhist[i]){
- if((wh = gethistory(qidnum(path))) == nil)
- return "file does not exist";
- closewhist(a->w);
- a->w = wh;
- }
- *qid = (Qid){mkqid(F1st, qidnum(path), 0, i), a->w->doc->time, 0};
- goto Gotfile;
- }
- }
- return "file does not exist";
- case D2nd:
- if(isdotdot){
- /*
- * Can't use a->w[a->ndoc-1] because that
- * might be a failed write rather than the real one.
- */
- *qid = (Qid){mkqid(D1st, qidnum(path), 0, 0), 0, QTDIR};
- if((wh = getcurrent(qidnum(path))) == nil)
- return "file does not exist";
- closewhist(a->w);
- a->w = wh;
- a->n = 0;
- return nil;
- }
- for(i=0; i<=Qraw; i++){
- if(strcmp(name, filelist[i])==0){
- *qid = (Qid){mkqid(F2nd, qidnum(path), qidvers(path), i), a->w->doc->time, 0};
- goto Gotfile;
- }
- }
- return "file does not exist";
- default:
- return "bad programming";
- }
- assert(0);
- Gotfile:
- t = qidtype(qid->path);
- switch(qidfile(qid->path)){
- case Qindexhtml:
- s = tohtml(a->w, a->w->doc+a->n,
- t==F1st? Tpage : Toldpage);
- break;
- case Qindextxt:
- s = totext(a->w, a->w->doc+a->n,
- t==F1st? Tpage : Toldpage);
- break;
- case Qraw:
- s = s_copy(a->w->title);
- s = s_append(s, "\n");
- s = doctext(s, &a->w->doc[a->n]);
- break;
- case Qhistoryhtml:
- s = tohtml(a->w, a->w->doc+a->n, Thistory);
- break;
- case Qhistorytxt:
- s = totext(a->w, a->w->doc+a->n, Thistory);
- break;
- case Qdiffhtml:
- s = tohtml(a->w, a->w->doc+a->n, Tdiff);
- break;
- case Qedithtml:
- s = tohtml(a->w, a->w->doc+a->n, Tedit);
- break;
- case Qwerrorhtml:
- s = tohtml(a->w, a->w->doc+a->n, Twerror);
- break;
- case Qwerrortxt:
- s = totext(a->w, a->w->doc+a->n, Twerror);
- break;
- default:
- return "internal error";
- }
- a->s = s;
- return nil;
- }
- static void
- fsopen(Req *r)
- {
- int t;
- uvlong path;
- Aux *a;
- Fid *fid;
- Whist *wh;
- fid = r->fid;
- path = fid->qid.path;
- t = qidtype(fid->qid.path);
- if((r->ifcall.mode != OREAD && t != Fnew && t != Fmap)
- || (r->ifcall.mode&ORCLOSE)){
- respond(r, "permission denied");
- return;
- }
- a = fid->aux;
- switch(t){
- case Droot:
- currentmap(0);
- rlock(&maplock);
- a->map = map;
- incref(map);
- runlock(&maplock);
- respond(r, nil);
- break;
-
- case D1st:
- if((wh = gethistory(qidnum(path))) == nil){
- respond(r, "file does not exist");
- return;
- }
- closewhist(a->w);
- a->w = wh;
- a->n = a->w->ndoc-1;
- r->ofcall.qid.vers = wh->doc[a->n].time;
- r->fid->qid = r->ofcall.qid;
- respond(r, nil);
- break;
-
- case D2nd:
- respond(r, nil);
- break;
- case Fnew:
- a->s = s_copy("");
- respond(r, nil);
- break;
- case Fmap:
- case F1st:
- case F2nd:
- respond(r, nil);
- break;
- default:
- respond(r, "programmer error");
- break;
- }
- }
- static char*
- fsclone(Fid *old, Fid *new)
- {
- Aux *a;
- a = emalloc(sizeof(*a));
- *a = *(Aux*)old->aux;
- if(a->s)
- s_incref(a->s);
- if(a->w)
- incref(a->w);
- if(a->map)
- incref(a->map);
- new->aux = a;
- new->qid = old->qid;
- return nil;
- }
- static void
- fsdestroyfid(Fid *fid)
- {
- Aux *a;
- a = fid->aux;
- if(a==nil)
- return;
- if(a->map)
- closemap(a->map);
- if(a->s)
- s_free(a->s);
- if(a->w)
- closewhist(a->w);
- free(a);
- fid->aux = nil;
- }
- static void
- fillstat(Dir *d, uvlong path, ulong tm, ulong length)
- {
- char tmp[32], *p;
- int type;
- memset(d, 0, sizeof(Dir));
- d->uid = estrdup9p("wiki");
- d->gid = estrdup9p("wiki");
- switch(qidtype(path)){
- case Droot:
- case D1st:
- case D2nd:
- type = QTDIR;
- break;
- default:
- type = 0;
- break;
- }
- d->qid = (Qid){path, tm, type};
- d->atime = d->mtime = tm;
- d->length = length;
- if(qidfile(path) == Qedithtml)
- d->atime = d->mtime = time(0);
- switch(qidtype(path)){
- case Droot:
- d->name = estrdup("/");
- d->mode = DMDIR|0555;
- break;
- case D1st:
- d->name = numtoname(qidnum(path));
- if(d->name == nil)
- d->name = estrdup("<dead>");
- for(p=d->name; *p; p++)
- if(*p==' ')
- *p = '_';
- d->mode = DMDIR|0555;
- break;
- case D2nd:
- snprint(tmp, sizeof tmp, "%lud", tm);
- d->name = estrdup(tmp);
- d->mode = DMDIR|0555;
- break;
- case Fmap:
- d->name = estrdup("map");
- d->mode = 0666;
- break;
- case Fnew:
- d->name = estrdup("new");
- d->mode = 0666;
- break;
- case F1st:
- d->name = estrdup(filelist[qidfile(path)]);
- d->mode = 0444;
- break;
- case F2nd:
- d->name = estrdup(filelist[qidfile(path)]);
- d->mode = 0444;
- break;
- default:
- print("bad qid path 0x%.8llux\n", path);
- break;
- }
- }
- static void
- fsstat(Req *r)
- {
- Aux *a;
- Fid *fid;
- ulong t;
- t = 0;
- fid = r->fid;
- if((a = fid->aux) && a->w)
- t = a->w->doc[a->n].time;
- fillstat(&r->d, fid->qid.path, t, a->s ? s_len(a->s) : 0);
- respond(r, nil);
- }
- typedef struct Bogus Bogus;
- struct Bogus {
- uvlong path;
- Aux *a;
- };
- static int
- rootgen(int i, Dir *d, void *aux)
- {
- Aux *a;
- Bogus *b;
- b = aux;
- a = b->a;
- switch(i){
- case 0: /* new */
- fillstat(d, mkqid(Fnew, 0, 0, 0), a->map->t, 0);
- return 0;
- case 1: /* map */
- fillstat(d, mkqid(Fmap, 0, 0, 0), a->map->t, 0);
- return 0;
- default: /* first-level directory */
- i -= 2;
- if(i >= a->map->nel)
- return -1;
- fillstat(d, mkqid(D1st, a->map->el[i].n, 0, 0), a->map->t, 0);
- return 0;
- }
- }
- static int
- firstgen(int i, Dir *d, void *aux)
- {
- ulong t;
- Bogus *b;
- int num;
- Aux *a;
- b = aux;
- num = qidnum(b->path);
- a = b->a;
- t = a->w->doc[a->n].time;
- if(i < Nfile){ /* file in first-level directory */
- fillstat(d, mkqid(F1st, num, 0, i), t, 0);
- return 0;
- }
- i -= Nfile;
- if(i < a->w->ndoc){ /* second-level (history) directory */
- fillstat(d, mkqid(D2nd, num, i, 0), a->w->doc[i].time, 0);
- return 0;
- }
- //i -= a->w->ndoc;
- return -1;
- }
- static int
- secondgen(int i, Dir *d, void *aux)
- {
- Bogus *b;
- uvlong path;
- Aux *a;
- b = aux;
- path = b->path;
- a = b->a;
- if(i <= Qraw){ /* index.html, index.txt, raw */
- fillstat(d, mkqid(F2nd, qidnum(path), qidvers(path), i), a->w->doc[a->n].time, 0);
- return 0;
- }
- //i -= Qraw;
- return -1;
- }
- static void
- fsread(Req *r)
- {
- char *t, *s;
- uvlong path;
- Aux *a;
- Bogus b;
- a = r->fid->aux;
- path = r->fid->qid.path;
- b.a = a;
- b.path = path;
- switch(qidtype(path)){
- default:
- respond(r, "cannot happen (bad qid)");
- return;
- case Droot:
- if(a == nil || a->map == nil){
- respond(r, "cannot happen (no map)");
- return;
- }
- dirread9p(r, rootgen, &b);
- respond(r, nil);
- return;
-
- case D1st:
- if(a == nil || a->w == nil){
- respond(r, "cannot happen (no wh)");
- return;
- }
- dirread9p(r, firstgen, &b);
- respond(r, nil);
- return;
-
- case D2nd:
- dirread9p(r, secondgen, &b);
- respond(r, nil);
- return;
- case Fnew:
- if(a->s){
- respond(r, "protocol botch");
- return;
- }
- /* fall through */
- case Fmap:
- t = numtoname(a->n);
- if(t == nil){
- respond(r, "unknown name");
- return;
- }
- for(s=t; *s; s++)
- if(*s == ' ')
- *s = '_';
- readstr(r, t);
- free(t);
- respond(r, nil);
- return;
- case F1st:
- case F2nd:
- if(a == nil || a->s == nil){
- respond(r, "cannot happen (no s)");
- return;
- }
- readbuf(r, s_to_c(a->s), s_len(a->s));
- respond(r, nil);
- return;
- }
- }
- typedef struct Sread Sread;
- struct Sread {
- char *rp;
- };
- static char*
- Srdline(void *v, int c)
- {
- char *p, *rv;
- Sread *s;
- s = v;
- if(s->rp == nil)
- rv = nil;
- else if(p = strchr(s->rp, c)){
- *p = '\0';
- rv = s->rp;
- s->rp = p+1;
- }else{
- rv = s->rp;
- s->rp = nil;
- }
- return rv;
- }
- static void
- responderrstr(Req *r)
- {
- char buf[ERRMAX];
- rerrstr(buf, sizeof buf);
- if(buf[0] == '\0')
- strcpy(buf, "unknown error");
- respond(r, buf);
- }
- static void
- fswrite(Req *r)
- {
- char *author, *comment, *net, *err, *p, *title, tmp[40];
- int rv, n;
- ulong t;
- Aux *a;
- Fid *fid;
- Sread s;
- String *stmp;
- Whist *w;
- fid = r->fid;
- a = fid->aux;
- switch(qidtype(fid->qid.path)){
- case Fmap:
- stmp = s_nappend(s_reset(nil), r->ifcall.data, r->ifcall.count);
- a->n = nametonum(s_to_c(stmp));
- s_free(stmp);
- if(a->n < 0)
- respond(r, "name not found");
- else
- respond(r, nil);
- return;
- case Fnew:
- break;
- default:
- respond(r, "cannot happen");
- return;
- }
- if(a->s == nil){
- respond(r, "protocol botch");
- return;
- }
- if(r->ifcall.count==0){ /* do final processing */
- s.rp = s_to_c(a->s);
- w = nil;
- err = "bad format";
- if((title = Srdline(&s, '\n')) == nil){
- Error:
- if(w)
- closewhist(w);
- s_free(a->s);
- a->s = nil;
- respond(r, err);
- return;
- }
- w = emalloc(sizeof(*w));
- incref(w);
- w->title = estrdup(title);
- t = 0;
- author = estrdup(a->name);
- comment = nil;
- while(s.rp && *s.rp && *s.rp != '\n'){
- p = Srdline(&s, '\n');
- assert(p != nil);
- switch(p[0]){
- case 'A':
- free(author);
- author = estrdup(p+1);
- break;
- case 'D':
- t = strtoul(p+1, &p, 10);
- if(*p != '\0')
- goto Error;
- break;
- case 'C':
- free(comment);
- comment = estrdup(p+1);
- break;
- }
- }
- w->doc = emalloc(sizeof(w->doc[0]));
- w->doc->time = time(0);
- w->doc->comment = comment;
- if(net = r->pool->srv->aux){
- p = emalloc(strlen(author)+10+strlen(net));
- strcpy(p, author);
- strcat(p, " (");
- strcat(p, net);
- strcat(p, ")");
- free(author);
- author = p;
- }
- w->doc->author = author;
- if((w->doc->wtxt = Brdpage(Srdline, &s)) == nil){
- err = "empty document";
- goto Error;
- }
- w->ndoc = 1;
- if((n = allocnum(w->title, 0)) < 0)
- goto Error;
- sprint(tmp, "D%lud\n", w->doc->time);
- a->s = s_reset(a->s);
- a->s = doctext(a->s, w->doc);
- rv = writepage(n, t, a->s, w->title);
- s_free(a->s);
- a->s = nil;
- a->n = n;
- closewhist(w);
- if(rv < 0)
- responderrstr(r);
- else
- respond(r, nil);
- return;
- }
- if(s_len(a->s)+r->ifcall.count > Maxfile){
- respond(r, "file too large");
- s_free(a->s);
- a->s = nil;
- return;
- }
- a->s = s_nappend(a->s, r->ifcall.data, r->ifcall.count);
- r->ofcall.count = r->ifcall.count;
- respond(r, nil);
- }
- Srv wikisrv = {
- .attach= fsattach,
- .destroyfid= fsdestroyfid,
- .clone= fsclone,
- .walk1= fswalk1,
- .open= fsopen,
- .read= fsread,
- .write= fswrite,
- .stat= fsstat,
- };
- void
- usage(void)
- {
- fprint(2, "usage: wikifs [-D] [-a addr]... [-m mtpt] [-p perm] [-s service] dir\n");
- exits("usage");
- }
- char*
- getremotesys(char *ndir)
- {
- char buf[128], *serv, *sys;
- int fd, n;
- snprint(buf, sizeof buf, "%s/remote", ndir);
- sys = nil;
- fd = open(buf, OREAD);
- if(fd >= 0){
- n = read(fd, buf, sizeof(buf)-1);
- if(n>0){
- buf[n-1] = 0;
- serv = strchr(buf, '!');
- if(serv)
- *serv = 0;
- sys = estrdup(buf);
- }
- close(fd);
- }
- if(sys == nil)
- sys = estrdup("unknown");
- return sys;
- }
- void
- listensrv(Srv *os, char *addr)
- {
- int ctl, data, nctl;
- char ndir[NETPATHLEN], dir[NETPATHLEN];
- Srv *s;
- switch(rfork(RFPROC|RFMEM|RFNOWAIT)){
- case -1:
- sysfatal("rfork: %r");
- default:
- return;
- case 0:
- break;
- }
- ctl = announce(addr, dir);
- if(ctl < 0){
- fprint(2, "wikifs announce %s: %r", addr);
- _exits(0);
- }
- for(;;){
- nctl = listen(dir, ndir);
- if(nctl < 0){
- fprint(2, "wikifs listen %s: %r", addr);
- _exits(0);
- }
-
- switch(rfork(RFPROC|RFMEM|RFNOWAIT)){
- case -1:
- fprint(2, "rfork answerer: %r\n");
- continue;
- default:
- continue;
- case 0:
- break;
- }
- data = accept(ctl, ndir);
- if(data < 0){
- fprint(2, "wikifs accept %s: %r\n", ndir);
- _exits(nil);
- }
- s = emalloc(sizeof *s);
- *s = *os;
- s->aux = getremotesys(ndir);
- s->infd = s->outfd = data;
- srv(s);
- free(s);
- free(s->aux);
- _exits(nil);
- }
- }
- void
- main(int argc, char **argv)
- {
- char **addr;
- int i, naddr;
- char *buf;
- char *service, *mtpt;
- ulong perm;
- Dir d, *dp;
- Srv *s;
- naddr = 0;
- addr = nil;
- perm = 0;
- service = nil;
- mtpt = "/mnt/wiki";
- ARGBEGIN{
- case 'D':
- chatty9p++;
- break;
- case 'a':
- if(naddr%8 == 0)
- addr = erealloc(addr, (naddr+8)*sizeof(addr[0]));
- addr[naddr++] = EARGF(usage());
- break;
- case 'm':
- mtpt = EARGF(usage());
- break;
- case 'M':
- mtpt = nil;
- break;
- case 'p':
- perm = strtoul(EARGF(usage()), nil, 8);
- break;
- case 's':
- service = EARGF(usage());
- break;
- default:
- usage();
- break;
- }ARGEND
- if(argc != 1)
- usage();
- if((dp = dirstat(argv[0])) == nil)
- sysfatal("dirstat %s: %r", argv[0]);
- if((dp->mode&DMDIR) == 0)
- sysfatal("%s: not a directory", argv[0]);
- free(dp);
- wikidir = argv[0];
- currentmap(0);
- for(i=0; i<naddr; i++)
- listensrv(&wikisrv, addr[i]);
- s = emalloc(sizeof *s);
- *s = wikisrv;
- postmountsrv(s, service, mtpt, MREPL|MCREATE);
- if(perm){
- buf = emalloc9p(5+strlen(service)+1);
- strcpy(buf, "/srv/");
- strcat(buf, service);
- nulldir(&d);
- d.mode = perm;
- if(dirwstat(buf, &d) < 0)
- fprint(2, "wstat: %r\n");
- }
- exits(nil);
- }
|