8 typedef struct Args Args;
15 typedef struct Dfile Dfile;
16 typedef struct Fid Fid;
17 typedef struct File File;
19 typedef struct Request Request;
20 typedef struct Symbol Symbol;
21 typedef struct Tardir Tardir;
23 extern int threadrforkflag = RFNAMEG;
26 Nstat = 1024, /* plenty for this application */
27 MAXSIZE = 8192+IOHDRSZ,
30 int messagesize = MAXSIZE;
38 write(2, "depend: ", 8);
40 vseprint(buf, buf+1024, fmt, arg);
42 write(2, buf, strlen(buf));
55 Symbol *next; /* hash list chaining */
57 int fno; /* file symbol is defined in */
67 uchar *refvec; /* files resolving the references */
68 uint len; /* length of file */
69 uint tarlen; /* length of tar file */
81 int use; /* use count */
82 int old; /* true if this is an superceded dfile */
84 File *file; /* files */
85 int nfile; /* number of files */
86 int flen; /* length of file table */
88 Symbol **dhash; /* hash table of symbols */
89 int hlen; /* length of hash table */
91 Dfile *next; /* hash chain */
92 char *path; /* path name of dependency file */
93 Qid qid; /* qid of the dependency file */
124 Tblocksize= 512, /* tar block size */
125 Tnamesize= 100, /* tar name size */
130 char name[Tnamesize];
138 char linkname[Tnamesize];
145 int fd; /* to kernel mount point */
160 extern void fsrun(void*);
161 extern Fid* fsgetfid(Fs*, int);
162 extern void fsputfid(Fs*, Fid*);
163 extern void fsreply(Fs*, Request*, char*);
164 extern void fsversion(Fs*, Request*, Fid*);
165 extern void fsauth(Fs*, Request*, Fid*);
166 extern void fsflush(Fs*, Request*, Fid*);
167 extern void fsattach(Fs*, Request*, Fid*);
168 extern void fswalk(Fs*, Request*, Fid*);
169 extern void fsopen(Fs*, Request*, Fid*);
170 extern void fscreate(Fs*, Request*, Fid*);
171 extern void fsread(Fs*, Request*, Fid*);
172 extern void fswrite(Fs*, Request*, Fid*);
173 extern void fsclunk(Fs*, Request*, Fid*);
174 extern void fsremove(Fs*, Request*, Fid*);
175 extern void fsstat(Fs*, Request*, Fid*);
176 extern void fswstat(Fs*, Request*, Fid*);
178 void (*fcall[])(Fs*, Request*, Fid*) =
181 [Tversion] fsversion,
195 char Eperm[] = "permission denied";
196 char Eexist[] = "file does not exist";
197 char Enotdir[] = "not a directory";
198 char Eisopen[] = "file already open";
199 char Enofid[] = "no such fid";
200 char mallocerr[] = "malloc: %r";
201 char Etoolong[] = "name too long";
203 char *dependlog = "depend";
206 Dfile *dfhash[Ndfhash]; /* dependency file hash */
207 QLock dfhlock[Ndfhash];
210 Request* allocreq(int);
212 void releasedf(Dfile*);
213 Symbol* dfsearch(Dfile*, char*);
214 void dfresolve(Dfile*, int);
215 char* mkpath(char*, char*);
216 int mktar(Dfile*, Symbol*, uchar*, uint, int);
217 void closetar(Dfile*, Symbol*);
232 erealloc(void *ReallocP, int ReallocN)
237 ReallocP = emalloc(ReallocN);
238 else if(!(ReallocP = realloc(ReallocP, ReallocN)))
239 fatal("unable to allocate %d bytes",ReallocN);
250 d = d0 = emalloc(strlen(s)+1);
257 * mount the user interface and start one request processor
276 fmtinstall('F', fcallfmt);
284 fprint(2, "usage: %s [-d] svc-name directory\n", argv0);
287 snprint(service, sizeof service, "#s/%s", argv[0]);
288 if(argv[1][0] != '/')
289 fatal("directory must be rooted");
292 fatal("opening pipe: %r");
294 /* Typically mounted before /srv exists */
295 srv = create(service, OWRITE, 0666);
298 fprint(srv, "%d", pfd[1]);
302 time(nil); /* open fd for time before losing / */
303 if(bind(argv[1], "/", MREPL) == 0)
304 fatal("can't bind %s to /", argv[1]);
306 fs = emalloc(sizeof(Fs));
309 fsarg.root = argv[1];
310 proccreate(fsrun, &fsarg, 16*1024);
311 proccreate(fsrun, &fsarg, 16*1024);
317 threadmain(int argc, char *argv[])
324 proccreate(realmain, &args, 16*1024);
328 mkpath(char *dir, char *file)
333 len = strlen(dir) + 1;
335 len += strlen(file) + 1;
338 sprint(path, "%s/%s", dir, file);
340 sprint(path, "%s", dir);
361 fatal("root %s inaccessible: %r", root);
362 fs->rootqid = d->qid;
366 r = allocreq(messagesize);
368 n = read9pmsg(fs->fd, r->buf, messagesize);
370 threadexitsall("unmounted");
372 fatal("mount read: %r");
373 if(convM2S(r->buf, n, &r->f) != n)
374 fatal("convM2S format error: %r");
377 f = fsgetfid(fs, r->f.fid);
380 fprint(2, "%F path %llux\n", &r->f, f->qid.path);
384 (*fcall[t])(fs, r, f);
390 * any request that can get queued for a delayed reply
393 allocreq(int bufsize)
397 r = emalloc(sizeof(Request)+bufsize);
403 fsgetfid(Fs *fs, int fid)
408 for(f = fs->hash[fid%Nfidhash]; f; f = f->next){
416 nf = emalloc(sizeof(Fid));
417 nf->next = fs->hash[fid%Nfidhash];
418 fs->hash[fid%Nfidhash] = nf;
427 fsputfid(Fs *fs, Fid *f)
436 for(l = &fs->hash[f->fid%Nfidhash]; nf = *l; l = &nf->next)
446 fsreply(Fs *fs, Request *r, char *err)
456 fprint(2, "%F path %llux\n", &r->f, r->fid->qid.path);
457 n = convS2M(&r->f, buf, messagesize);
459 fatal("bad convS2M");
460 if(write(fs->fd, buf, n) != n)
466 fsversion(Fs *fs, Request *r, Fid*)
468 if(r->f.msize < 256){
469 fsreply(fs, r, "version: bad message size");
472 if(messagesize > r->f.msize)
473 messagesize = r->f.msize;
474 r->f.msize = messagesize;
475 r->f.version = "9P2000";
480 fsauth(Fs *fs, Request *r, Fid*)
482 fsreply(fs, r, "depend: authentication not required");
486 fsflush(Fs*, Request*, Fid*)
491 fsattach(Fs *fs, Request *r, Fid *f)
493 f->qid = fs->rootqid;
494 f->path = strdup("/");
495 f->df = getdf(mkpath(f->path, ".depend"));
497 /* hold down the fid till the clunk */
508 fswalk(Fs *fs, Request *r, Fid *f)
513 char errbuf[ERRMAX], *err;
520 if(f->attached == 0){
521 fsreply(fs, r, Eexist);
525 if(f->fd >= 0 || f->open)
526 fatal("walk of an open file");
529 if(r->f.newfid != r->f.fid){
530 nf = fsgetfid(fs, r->f.newfid);
533 nf->path = strdup(f->path);
543 if(r->f.nwname == 0){
552 path = strdup(f->path);
556 nwname = r->f.nwname;
560 for(; nqid<nwname; nqid++){
561 name = r->f.wname[nqid];
563 if(strcmp(name, ".") == 0){
568 qid[nqid] = qid[nqid-1];
572 if(strcmp(name, "..") == 0){
573 name = strrchr(path, '/');
575 if(name == path) /* at root */
582 errstr(errbuf, sizeof errbuf);
590 lastdf = getdf(mkpath(path, ".depend"));
594 npath = mkpath(path, name);
599 if(d !=nil && (d->qid.type & QTDIR))
603 qid[nqid].type = QTFILE;
607 dp = dfsearch(lastdf, name);
613 if(i > 4 && strcmp(&tmp[i-4], ".tar") == 0){
615 dp = dfsearch(lastdf, tmp);
624 qid[nqid].path = (uvlong)dp;
627 if(nqid == 0 && err == nil)
628 err = "file does not exist";
631 /* for error or partial success, put the cloned fid back*/
632 if(nf!=nil && (err != nil || nqid < nwname)){
639 /* return (possibly short) list of qids */
640 for(i=0; i<nqid; i++)
641 r->f.wqid[i] = qid[i];
644 /* for full success, advance f */
645 if(nqid > 0 && nqid == nwname){
650 f->qid = qid[nqid-1];
651 f->dp = (Symbol*)f->qid.path;
670 fsclone(Fs *fs, Request *r, Fid *f)
674 if(f->attached == 0){
675 fsreply(fs, r, Eexist);
678 nf = fsgetfid(fs, r->f.newfid);
682 nf->path = strdup(f->path);
696 fswalk(Fs *fs, Request *r, Fid *f)
705 if(f->attached == 0){
706 fsreply(fs, r, Enofid);
710 if(f->fd >= 0 || f->open)
711 fatal("walk of an open file");
714 if(strcmp(name, ".") == 0){
718 if(strcmp(name, "..") == 0){
719 name = strrchr(f->path, '/');
727 if(dirstat(f->path, &d) < 0){
730 fsreply(fs, r, errbuf);
733 r->f.qid = f->qid = d.qid;
736 f->df = getdf(mkpath(f->path, ".depend"));
742 path = mkpath(f->path, name);
743 if(dirstat(path, &d) < 0 || (d.qid.path & CHDIR) == 0){
744 dp = dfsearch(f->df, name);
747 if(i > 4 && strcmp(&name[i-4], ".tar") == 0){
749 dp = dfsearch(f->df, name);
753 fsreply(fs, r, Eexist);
758 d.qid.path = (uint)dp;
765 if(d.qid.path & CHDIR){
767 f->df = getdf(mkpath(f->path, ".depend"));
770 r->f.qid = f->qid = d.qid;
775 fsopen(Fs *fs, Request *r, Fid *f)
780 if(f->attached == 0){
781 fsreply(fs, r, Enofid);
785 fsreply(fs, r, Eisopen);
789 mode = r->f.mode & 3;
791 fsreply(fs, r, Eperm);
795 if(f->qid.type & QTDIR){
796 f->fd = open(f->path, OREAD);
798 errstr(errbuf, sizeof errbuf);
799 fsreply(fs, r, errbuf);
810 fscreate(Fs *fs, Request *r, Fid*)
812 fsreply(fs, r, Eperm);
816 fsread(Fs *fs, Request *r, Fid *f)
823 if(f->attached == 0){
824 fsreply(fs, r, Enofid);
827 if((int)r->f.count < 0){
828 fsreply(fs, r, "bad read count");
832 if(f->qid.type & QTDIR){
835 f->ndir = dirreadall(f->fd, &f->dir);
840 if(r->f.offset == 0) /* seeking to zero is permitted */
842 for(; f->dirindex < f->ndir; f->dirindex++){
843 if((f->dir[f->dirindex].qid.type & QTDIR) == 0)
845 len = convD2M(&f->dir[f->dirindex], r->buf+n, r->f.count-n);
851 skip = f->dirindex - f->ndir; /* # depend records already read */
854 for(i = 0; i < f->df->hlen; i++)
855 for(dp = f->df->dhash[i]; dp; dp = dp->next){
858 snprint(buf, sizeof buf, "%s.tar", dp->sym);
864 d.qid.path = (uvlong)dp;
866 d.length = f->df->file[dp->fno].tarlen;
870 len = convD2M(&d, r->buf + n, r->f.count - n);
878 n = mktar(f->df, f->dp, r->buf, r->f.offset, r->f.count);
881 r->f.data = (char*)r->buf;
887 fswrite(Fs *fs, Request *r, Fid*)
889 fsreply(fs, r, Eperm);
893 fsclunk(Fs *fs, Request *r, Fid *f)
895 if(f->attached == 0){
896 fsreply(fs, r, Enofid);
904 if((f->qid.type & QTDIR) == 0)
905 closetar(f->df, f->dp);
917 fsremove(Fs *fs, Request *r, Fid*)
919 fsreply(fs, r, Eperm);
923 fsstat(Fs *fs, Request *r, Fid *f)
929 uchar statbuf[Nstat];
931 if(f->qid.type & QTDIR)
932 n = stat(f->path, statbuf, sizeof statbuf);
940 d.qid.path = (uvlong)dp;
942 d.length = f->df->file[dp->fno].tarlen;
946 n = convD2M(&d, statbuf, sizeof statbuf);
949 errstr(err, sizeof err);
959 fswstat(Fs *fs, Request *r, Fid*)
961 fsreply(fs, r, Eperm);
968 shash(char *str, int len)
974 for(val = str; *val; val++)
975 hash = (hash*13) + *val-'a';
980 * free info about a dependency file
995 unlock(df); /* we're no longer referenced */
996 for(i = 0; i < df->nfile; i++)
997 free(df->file[i].name);
998 free(df->file[i].refvec);
1002 for(i = 0; i < df->hlen; i++)
1003 for(dp = df->dhash[i]; dp != nil; dp = next){
1015 * crack a dependency file
1018 newsym(char *name, int fno, Symbol **l)
1022 dp = emalloc(sizeof(Symbol));
1023 dp->sym = strdup(name);
1031 awk(Biobuf *b, char **field, int n)
1036 while(line = Brdline(b, '\n')){
1037 line[Blinelen(b)-1] = 0;
1038 while(*line == ' ' || *line == '\t')
1040 for(i = 0; i < n; i++){
1041 if(*line == 0 || *line == '#')
1044 while(*line && *line != ' ' && *line != '\t')
1046 while(*line == ' ' || *line == '\t')
1057 crackdf(Dfile *df, Biobuf *b, uint len, char *dpath)
1062 int n, inc, ok, npath;
1063 Symbol **l, *dp, *next;
1069 df->file = emalloc(df->flen*sizeof(File));
1072 df->hlen = 1 + len/8;
1073 df->dhash = emalloc(df->hlen*sizeof(Symbol*));
1076 while((n = awk(b, field, 3)) > 0){
1083 if(df->flen == df->nfile){
1085 df->file = realloc(df->file, df->flen*sizeof(File));
1088 memset(&df->file[df->nfile], 0, inc*sizeof(File));
1090 f = &df->file[df->nfile++];
1091 f->name = strdup(name);
1093 /* fall through and define as a symbol */
1097 newsym(name, df->nfile-1, &(df->dhash[shash(name, df->hlen)]));
1107 ef = &df->file[df->nfile];
1109 /* stat the files to get sizes */
1110 npath = strlen(dpath);
1111 if(npath+1+1 >= sizeof path)
1113 memmove(path, dpath, npath+1); /* include NUL */
1114 name = strrchr(path, '/') + 1;
1115 for(f = df->file; f < ef; f++){
1116 n = strlen(f->name);
1117 if(npath+1+n+3+1 > sizeof path)
1119 memmove(name, f->name, n+1); /* include NUL */
1120 ok = access(path, AEXIST);
1122 strcpy(name+n, ".Z");
1123 ok = access(path, AEXIST);
1125 strcpy(name+n, ".gz");
1126 ok = access(path, AEXIST);
1131 f->name = strdup(name);
1139 f->mtime = d->mtime;
1149 /* resolve all file references */
1150 for(f = df->file; f < ef; f++)
1151 dfresolve(df, f-df->file);
1153 /* free the referenced symbols, don't need them anymore */
1154 for(f = df->file; f < ef; f++){
1155 f->tarlen += 2*Tblocksize; /* tars trailing 0 blocks */
1156 for(dp = f->ref; dp != nil; dp = next){
1165 * get a cracked dependency file
1176 i = shash(path, Ndfhash);
1180 for(df = *l; df; df = *l){
1181 if(strcmp(path, df->path) == 0)
1188 if(d!=nil && d->qid.type == df->qid.type && d->qid.vers == df->qid.vers && d->qid.vers == df->qid.vers){
1199 fd = open(path, OREAD);
1200 if(d == nil || fd < 0){
1205 df = emalloc(sizeof(*df));
1206 b = emalloc(sizeof(Biobuf));
1208 Binit(b, fd, OREAD);
1211 crackdf(df, b, d->length, path);
1226 * stop using a dependency file. Free it if it is no longer linked in.
1229 releasedf(Dfile *df)
1238 /* remove from hash chain */
1239 i = shash(df->path, Ndfhash);
1245 if(df->old == 0 || df->use > 0){
1250 for(d = *l; d; d = *l){
1260 /* now we know it is unreferenced, remove it */
1265 * search a dependency file for a symbol
1268 dfsearch(Dfile *df, char *name)
1274 for(dp = df->dhash[shash(name, df->hlen)]; dp; dp = dp->next)
1275 if(strcmp(dp->sym, name) == 0)
1281 * resolve a single file into a vector of referenced files and the sum of their
1284 /* set a bit in the referenced file vector */
1286 set(uchar *vec, int fno)
1288 if(vec[fno/8] & (1<<(fno&7)))
1290 vec[fno/8] |= 1<<(fno&7);
1293 /* merge two referenced file vectors */
1295 merge(uchar *vec, uchar *ovec, int nfile)
1297 nfile = (nfile+7)/8;
1302 res(Dfile *df, uchar *vec, int fno)
1310 return 0; /* already set */
1311 if(f->refvec != nil){
1312 merge(vec, f->refvec, df->nfile); /* we've descended here before */
1317 for(rp = f->ref; rp; rp = rp->next){
1318 dp = dfsearch(df, rp->sym);
1321 len += res(df, vec, dp->fno);
1323 return len + Tblocksize + ((f->len + Tblocksize - 1)/Tblocksize)*Tblocksize;
1326 dfresolve(Dfile *df, int fno)
1332 vec = emalloc((df->nfile+7)/8);
1333 f->tarlen = res(df, vec, fno);
1338 * make the tar directory block for a file
1348 p = emalloc(Tblocksize);
1351 strcpy(tp->name, f->name);
1352 sprint(tp->mode, "%6o ", f->mode & 0777);
1353 sprint(tp->uid, "%6o ", 0);
1354 sprint(tp->gid, "%6o ", 0);
1355 sprint(tp->size, "%11o ", f->len);
1356 sprint(tp->mtime, "%11o ", f->mtime);
1358 /* calculate checksum */
1359 memset(tp->chksum, ' ', sizeof(tp->chksum));
1361 ep = p + Tblocksize;
1362 for (cp = p; cp < ep; cp++)
1364 sprint(tp->chksum, "%6o", sum);
1373 getfile(Dfile *df, File *f)
1376 char path[512], *name;
1381 name = strrchr(df->path, '/') + 1;
1382 n = snprint(path, sizeof path, "%.*s/%s", (int)(name-df->path), df->path, f->name);
1383 if(n >= sizeof path - UTFmax){
1384 syslog(0, dependlog, "path name too long: %.20s.../%.20s...", df->path, f->name);
1387 f->fd = open(path, OREAD);
1389 syslog(0, dependlog, "can't open %s: %r", path);
1395 releasefile(File *f)
1412 * return a block of a tar file
1415 mktar(Dfile *df, Symbol *dp, uchar *area, uint offset, int len)
1417 int fd, i, j, n, off;
1422 f = &df->file[dp->fno];
1427 for(i = 0; i < df->nfile && len > 0; i++){
1428 if((vec[i/8] & (1<<(i&7))) == 0)
1432 n = Tblocksize + ((f->len + Tblocksize - 1)/Tblocksize)*Tblocksize;
1438 if(offset < Tblocksize){
1440 if(offset + len > Tblocksize)
1441 j = Tblocksize - offset;
1444 //if(debug)fprint(2, "reading %d bytes dir of %s\n", j, f->name);
1445 memmove(p, buf+offset, j);
1453 off = offset - Tblocksize;
1454 if(off >= 0 && off < f->len){
1455 if(off + len > f->len)
1459 fd = getfile(df, f);
1461 //if(debug)fprint(2, "reading %d bytes from offset %d of %s\n", j, off, f->name);
1462 if(pread(fd, p, j, off) != j)
1463 syslog(0, dependlog, "%r reading %d bytes from offset %d of %s", j, off, f->name);
1473 if(offset + len > n)
1477 //if(debug)fprint(2, "filling %d bytes after %s\n", j, f->name);
1485 /* null blocks at end of tar file */
1486 if(offset < 2*Tblocksize && len > 0){
1487 if(offset + len > 2*Tblocksize)
1488 j = 2*Tblocksize - offset;
1491 //if(debug)fprint(2, "filling %d bytes at end\n", j);
1500 * close the files making up a tar file
1503 closetar(Dfile *df, Symbol *dp)
1509 f = &df->file[dp->fno];
1513 for(i = 0; i < df->nfile; i++){
1514 if((vec[i/8] & (1<<(i&7))) == 0)
1516 closefile(&df->file[i]);