ref: 3071c38b16cda2e1b5947ea28f64e4fd182250db
dir: /main.c/
#include <u.h> #include <libc.h> #include <bio.h> #include <avl.h> #include <fcall.h> #include "dat.h" #include "fns.h" Gefs *fs; int ream; int debug; int noauth; int nproc; char *forceuser; char *srvname = "gefs"; int cachesz = 512*MiB; vlong inc64(vlong *v, vlong dv) { vlong ov, nv; while(1){ ov = *v; nv = ov + dv; if(cas64((u64int*)v, ov, nv)) return ov; } } static void initfs(vlong cachesz) { if((fs = mallocz(sizeof(Gefs), 1)) == nil) sysfatal("malloc: %r"); fs->noauth = noauth; fs->cmax = cachesz/Blksz; if(fs->cmax >= (2ULL*GiB)/sizeof(Bucket)) sysfatal("cache too big"); if((fs->cache = mallocz(fs->cmax*sizeof(Bucket), 1)) == nil) sysfatal("malloc: %r"); } static void launch(void (*f)(int, void *), int wid, void *arg, char *text) { int pid; assert(wid == -1 || wid < nelem(fs->active)); pid = rfork(RFPROC|RFMEM|RFNOWAIT); if (pid < 0) sysfatal("can't fork: %r"); if (pid == 0) { procsetname("%s", text); (*f)(wid, arg); exits("child returned"); } } static int postfd(char *name, char *suff) { char buf[80]; int fd[2]; int cfd; if(pipe(fd) < 0) sysfatal("can't make a pipe"); snprint(buf, sizeof buf, "/srv/%s%s", name, suff); if((cfd = create(buf, OWRITE|ORCLOSE|OCEXEC, 0600)) == -1) sysfatal("create %s: %r", buf); if(fprint(cfd, "%d", fd[0]) == -1) sysfatal("write %s: %r", buf); close(fd[0]); return fd[1]; } static void usage(void) { fprint(2, "usage: %s [-rA] [-m mem] [-s srv] [-u usr] dev\n", argv0); exits("usage"); } void main(int argc, char **argv) { int i, srvfd, ctlfd; char *s; ARGBEGIN{ case 'r': ream = 1; break; case 'm': cachesz = strtoll(EARGF(usage()), nil, 0)*MiB; break; case 'd': debug++; break; case 's': srvname = EARGF(usage()); break; case 'A': noauth = 1; break; case 'u': forceuser = EARGF(usage()); break; default: usage(); break; }ARGEND; if(argc == 0) usage(); /* * sanity checks -- I've tuned these to stupid * values in the past. */ assert(4*Kpmax < Pivspc); assert(2*Msgmax < Bufspc); assert(Treesz < Inlmax); initfs(cachesz); initshow(); quotefmtinstall(); fmtinstall('H', encodefmt); fmtinstall('B', Bconv); fmtinstall('M', Mconv); fmtinstall('P', Pconv); fmtinstall('K', Kconv); fmtinstall('R', Rconv); fmtinstall('F', fcallfmt); fmtinstall('Q', Qconv); if((s = getenv("NPROC")) != nil) nproc = atoi(s); if(nproc == 0) nproc = 2; if(nproc > nelem(fs->active)) nproc = nelem(fs->active); if(ream){ reamfs(argv[0]); exits(nil); } loadfs(argv[0]); fs->syncrz.l = &fs->synclk; fs->rdchan = mkchan(32); fs->wrchan = mkchan(32); fs->nsyncers = nproc; if(fs->nsyncers > fs->narena) fs->nsyncers = fs->narena; for(i = 0; i < fs->nsyncers; i++) fs->chsync[i] = mkchan(128); for(i = 0; i < fs->narena; i++) fs->arenas[i].sync = fs->chsync[i%nproc]; srvfd = postfd(srvname, ""); ctlfd = postfd(srvname, ".cmd"); launch(runtasks, -1, nil, "tasks"); launch(runcons, fs->nquiesce++, (void*)ctlfd, "ctl"); launch(runwrite, fs->nquiesce++, nil, "mutate"); for(i = 0; i < nproc; i++) launch(runread, fs->nquiesce++, nil, "readio"); for(i = 0; i < fs->nsyncers; i++) launch(runsync, -1, fs->chsync[i], "syncio"); if(srvfd != -1) launch(runfs, fs->nquiesce++, (void*)srvfd, "srvio"); exits(nil); }