paqfs.c 18 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135
  1. #include <u.h>
  2. #include <libc.h>
  3. #include <auth.h>
  4. #include <fcall.h>
  5. #include <bio.h>
  6. #include <mp.h>
  7. #include <libsec.h>
  8. #include <flate.h>
  9. #include "paqfs.h"
  10. enum
  11. {
  12. OPERM = 0x3, /* mask of all permission types in open mode */
  13. OffsetSize = 4, /* size in bytes of an offset */
  14. };
  15. typedef struct Fid Fid;
  16. typedef struct Paq Paq;
  17. typedef struct Block Block;
  18. struct Fid
  19. {
  20. short busy;
  21. short open;
  22. int fid;
  23. char *user;
  24. ulong offset; /* for directory reading */
  25. Paq *paq;
  26. Fid *next;
  27. };
  28. struct Paq
  29. {
  30. int ref;
  31. Paq *up;
  32. PaqDir *dir;
  33. Qid qid;
  34. };
  35. struct Block
  36. {
  37. int ref;
  38. ulong addr; /* block byte address */
  39. ulong age;
  40. uchar *data;
  41. };
  42. enum
  43. {
  44. Pexec = 1,
  45. Pwrite = 2,
  46. Pread = 4,
  47. Pother = 1,
  48. Pgroup = 8,
  49. Powner = 64,
  50. };
  51. Fid *fids;
  52. Fcall rhdr, thdr;
  53. int blocksize;
  54. int cachesize = 20;
  55. int mesgsize = 8*1024 + IOHDRSZ;
  56. Paq *root;
  57. Block *cache;
  58. ulong cacheage;
  59. Biobuf *bin;
  60. Fid * newfid(int);
  61. void paqstat(PaqDir*, char*);
  62. void io(int fd);
  63. void *erealloc(void*, ulong);
  64. void *emalloc(ulong);
  65. void *emallocz(ulong n);
  66. char *estrdup(char*);
  67. void usage(void);
  68. ulong getl(uchar *p);
  69. int gets(uchar *p);
  70. char *getstr(uchar *p);
  71. PaqDir *getDir(uchar*);
  72. void getHeader(uchar *p, PaqHeader *b);
  73. void getBlock(uchar *p, PaqBlock *b);
  74. void getTrailer(uchar *p, PaqTrailer *b);
  75. void init(char*, int);
  76. void paqDirFree(PaqDir*);
  77. Qid paqDirQid(PaqDir *d);
  78. Paq *paqCpy(Paq *s);
  79. Paq *paqLookup(Paq *s, char *name);
  80. void paqFree(Paq*);
  81. Paq *paqWalk(Paq *s, char *name);
  82. int perm(PaqDir *s, char *user, int p);
  83. int dirRead(Fid*, uchar*, int);
  84. Block *blockLoad(ulong addr, int type);
  85. void blockFree(Block*);
  86. int checkDirSize(uchar *p, uchar *ep);
  87. int packDir(PaqDir*, uchar*, int);
  88. int blockRead(uchar *data, ulong addr, int type);
  89. void readHeader(PaqHeader *hdr, char *name, DigestState *ds);
  90. void readBlocks(char *name, DigestState *ds);
  91. void readTrailer(PaqTrailer *tlr, char *name, DigestState *ds);
  92. char *rflush(Fid*), *rversion(Fid*),
  93. *rauth(Fid*), *rattach(Fid*), *rwalk(Fid*),
  94. *ropen(Fid*), *rcreate(Fid*),
  95. *rread(Fid*), *rwrite(Fid*), *rclunk(Fid*),
  96. *rremove(Fid*), *rstat(Fid*), *rwstat(Fid*);
  97. char *(*fcalls[])(Fid*) = {
  98. [Tflush] rflush,
  99. [Tversion] rversion,
  100. [Tattach] rattach,
  101. [Tauth] rauth,
  102. [Twalk] rwalk,
  103. [Topen] ropen,
  104. [Tcreate] rcreate,
  105. [Tread] rread,
  106. [Twrite] rwrite,
  107. [Tclunk] rclunk,
  108. [Tremove] rremove,
  109. [Tstat] rstat,
  110. [Twstat] rwstat,
  111. };
  112. char Eperm[] = "permission denied";
  113. char Enotdir[] = "not a directory";
  114. char Enoauth[] = "authentication not required";
  115. char Enotexist[] = "file does not exist";
  116. char Einuse[] = "file in use";
  117. char Eexist[] = "file exists";
  118. char Enotowner[] = "not owner";
  119. char Eisopen[] = "file already open for I/O";
  120. char Excl[] = "exclusive use file already open";
  121. char Ename[] = "illegal name";
  122. char Erdonly[] = "read only file system";
  123. char Ebadblock[] = "bad block";
  124. char Eversion[] = "bad version of P9";
  125. char Edirtoobig[] = "directory entry too big";
  126. int debug;
  127. void
  128. main(int argc, char *argv[])
  129. {
  130. char *defmnt, *p;
  131. int pfd[2];
  132. int fd;
  133. int stdio = 0;
  134. int verify = 0;
  135. defmnt = "/n/paq";
  136. ARGBEGIN{
  137. case 'c':
  138. p = ARGF();
  139. if(p == nil)
  140. usage();
  141. cachesize = atoi(p);
  142. break;
  143. case 'v':
  144. verify = 1;
  145. break;
  146. case 'd':
  147. debug = 1;
  148. break;
  149. case 'i':
  150. defmnt = nil;
  151. stdio = 1;
  152. pfd[0] = 0;
  153. pfd[1] = 1;
  154. break;
  155. case 's':
  156. defmnt = nil;
  157. break;
  158. case 'm':
  159. defmnt = ARGF();
  160. if(defmnt == nil)
  161. usage();
  162. break;
  163. case 'M':
  164. p = ARGF();
  165. if(p == nil)
  166. usage();
  167. mesgsize = atoi(p);
  168. if(mesgsize < 512)
  169. mesgsize = 512;
  170. if(mesgsize > 128*1024)
  171. mesgsize = 128*1024;
  172. break;
  173. default:
  174. usage();
  175. }ARGEND
  176. if(argc != 1)
  177. usage();
  178. init(argv[0], verify);
  179. if(!stdio){
  180. if(pipe(pfd) < 0)
  181. sysfatal("pipe failed");
  182. if(defmnt == 0){
  183. fd = create("#s/paqfs", OWRITE, 0666);
  184. if(fd < 0)
  185. sysfatal("create of /srv/paqfs failed");
  186. if(fprint(fd, "%d", pfd[0]) < 0)
  187. sysfatal("writing /srv/paqfs");
  188. }
  189. }
  190. if(debug)
  191. fmtinstall('F', fcallfmt);
  192. switch(rfork(RFFDG|RFPROC|RFNAMEG|RFNOTEG)){
  193. case -1:
  194. sysfatal("fork");
  195. case 0:
  196. close(pfd[0]);
  197. io(pfd[1]);
  198. break;
  199. default:
  200. close(pfd[1]); /* don't deadlock if child fails */
  201. if(defmnt && mount(pfd[0], -1, defmnt, MREPL|MCREATE, "") < 0)
  202. sysfatal("mount failed");
  203. }
  204. exits(0);
  205. }
  206. char*
  207. rversion(Fid*)
  208. {
  209. Fid *f;
  210. for(f = fids; f; f = f->next)
  211. if(f->busy)
  212. rclunk(f);
  213. if(rhdr.msize > mesgsize)
  214. thdr.msize = mesgsize;
  215. else
  216. thdr.msize = rhdr.msize;
  217. if(strcmp(rhdr.version, "9P2000") != 0)
  218. return Eversion;
  219. thdr.version = "9P2000";
  220. return 0;
  221. }
  222. char*
  223. rauth(Fid*)
  224. {
  225. return Enoauth;
  226. }
  227. char*
  228. rflush(Fid *f)
  229. {
  230. USED(f);
  231. return 0;
  232. }
  233. char*
  234. rattach(Fid *f)
  235. {
  236. /* no authentication! */
  237. f->busy = 1;
  238. f->paq = paqCpy(root);
  239. thdr.qid = f->paq->qid;
  240. if(rhdr.uname[0])
  241. f->user = estrdup(rhdr.uname);
  242. else
  243. f->user = "none";
  244. return 0;
  245. }
  246. char*
  247. clone(Fid *f, Fid **res)
  248. {
  249. Fid *nf;
  250. if(f->open)
  251. return Eisopen;
  252. if(f->busy == 0)
  253. return Enotexist;
  254. nf = newfid(rhdr.newfid);
  255. nf->busy = 1;
  256. nf->open = 0;
  257. nf->paq = paqCpy(f->paq);
  258. nf->user = strdup(f->user);
  259. *res = nf;
  260. return 0;
  261. }
  262. char*
  263. rwalk(Fid *f)
  264. {
  265. Paq *paq, *npaq;
  266. Fid *nf;
  267. int nqid, nwname;
  268. Qid qid;
  269. char *err;
  270. if(f->busy == 0)
  271. return Enotexist;
  272. nf = nil;
  273. if(rhdr.fid != rhdr.newfid){
  274. err = clone(f, &nf);
  275. if(err)
  276. return err;
  277. f = nf; /* walk the new fid */
  278. }
  279. nwname = rhdr.nwname;
  280. /* easy case */
  281. if(nwname == 0) {
  282. thdr.nwqid = 0;
  283. return 0;
  284. }
  285. paq = paqCpy(f->paq);
  286. qid = paq->qid;
  287. err = nil;
  288. for(nqid = 0; nqid < nwname; nqid++){
  289. if((qid.type & QTDIR) == 0){
  290. err = Enotdir;
  291. break;
  292. }
  293. if(!perm(paq->dir, f->user, Pexec)) {
  294. err = Eperm;
  295. break;
  296. }
  297. npaq = paqWalk(paq, rhdr.wname[nqid]);
  298. if(npaq == nil) {
  299. err = Enotexist;
  300. break;
  301. }
  302. paqFree(paq);
  303. paq = npaq;
  304. qid = paq->qid;
  305. thdr.wqid[nqid] = qid;
  306. }
  307. thdr.nwqid = nqid;
  308. if(nqid == nwname){
  309. /* success */
  310. paqFree(f->paq);
  311. f->paq = paq;
  312. return 0;
  313. }
  314. paqFree(paq);
  315. if(nf != nil)
  316. rclunk(nf);
  317. /* only error on the first element */
  318. if(nqid == 0)
  319. return err;
  320. return 0;
  321. }
  322. char *
  323. ropen(Fid *f)
  324. {
  325. int mode, trunc;
  326. if(f->open)
  327. return Eisopen;
  328. if(f->busy == 0)
  329. return Enotexist;
  330. mode = rhdr.mode;
  331. if(f->paq->qid.type & QTDIR){
  332. if(mode != OREAD)
  333. return Eperm;
  334. thdr.qid = f->paq->qid;
  335. return 0;
  336. }
  337. if(mode & ORCLOSE)
  338. return Erdonly;
  339. trunc = mode & OTRUNC;
  340. mode &= OPERM;
  341. if(mode==OWRITE || mode==ORDWR || trunc)
  342. return Erdonly;
  343. if(mode==OREAD)
  344. if(!perm(f->paq->dir, f->user, Pread))
  345. return Eperm;
  346. if(mode==OEXEC)
  347. if(!perm(f->paq->dir, f->user, Pexec))
  348. return Eperm;
  349. thdr.qid = f->paq->qid;
  350. f->open = 1;
  351. return 0;
  352. }
  353. char *
  354. rcreate(Fid *f)
  355. {
  356. if(f->open)
  357. return Eisopen;
  358. if(f->busy == 0)
  359. return Enotexist;
  360. return Erdonly;
  361. }
  362. char *
  363. readdir(Fid *f)
  364. {
  365. PaqDir *pd;
  366. uchar *p, *ep;
  367. ulong off;
  368. int n, cnt, i;
  369. uchar *buf;
  370. Block *ptr, *b;
  371. buf = (uchar*)thdr.data;
  372. cnt = rhdr.count;
  373. if(rhdr.offset == 0)
  374. f->offset = 0;
  375. off = f->offset;
  376. ptr = blockLoad(f->paq->dir->offset, PointerBlock);
  377. if(ptr == nil)
  378. return Ebadblock;
  379. i = off/blocksize;
  380. off -= i*blocksize;
  381. thdr.count = 0;
  382. b = blockLoad(getl(ptr->data + i*4), DirBlock);
  383. while(b != nil) {
  384. p = b->data + off;
  385. ep = b->data + blocksize;
  386. if(checkDirSize(p, ep)) {
  387. pd = getDir(p);
  388. n = packDir(pd, buf, cnt);
  389. paqDirFree(pd);
  390. if(n == 0) {
  391. blockFree(b);
  392. if(thdr.count == 0) {
  393. blockFree(ptr);
  394. return Edirtoobig;
  395. }
  396. break;
  397. }
  398. off += gets(p);
  399. cnt -= n;
  400. buf += n;
  401. thdr.count += n;
  402. } else {
  403. off = 0;
  404. i++;
  405. blockFree(b);
  406. b = blockLoad(getl(ptr->data + i*4), DataBlock);
  407. }
  408. }
  409. f->offset = i*blocksize + off;
  410. blockFree(ptr);
  411. return 0;
  412. }
  413. char*
  414. rread(Fid *f)
  415. {
  416. PaqDir *pd;
  417. uchar *buf;
  418. ulong off;
  419. int n, cnt, i;
  420. Block *ptr, *b;
  421. if(f->busy == 0)
  422. return Enotexist;
  423. if(f->paq->qid.type & QTDIR)
  424. return readdir(f);
  425. pd = f->paq->dir;
  426. off = rhdr.offset;
  427. buf = (uchar*)thdr.data;
  428. cnt = rhdr.count;
  429. thdr.count = 0;
  430. if(off >= pd->length || cnt == 0)
  431. return 0;
  432. if(cnt > pd->length - off)
  433. cnt = pd->length - off;
  434. ptr = blockLoad(pd->offset, PointerBlock);
  435. if(ptr == nil)
  436. return Ebadblock;
  437. i = off/blocksize;
  438. off -= i*blocksize;
  439. while(cnt > 0) {
  440. b = blockLoad(getl(ptr->data + i*4), DataBlock);
  441. if(b == nil) {
  442. blockFree(ptr);
  443. return Ebadblock;
  444. }
  445. n = blocksize - off;
  446. if(n > cnt)
  447. n = cnt;
  448. memmove(buf, b->data + off, n);
  449. cnt -= n;
  450. thdr.count += n;
  451. buf += n;
  452. off = 0;
  453. i++;
  454. blockFree(b);
  455. }
  456. blockFree(ptr);
  457. return 0;
  458. }
  459. char*
  460. rwrite(Fid *f)
  461. {
  462. if(f->busy == 0)
  463. return Enotexist;
  464. return Erdonly;
  465. }
  466. char *
  467. rclunk(Fid *f)
  468. {
  469. f->busy = 0;
  470. f->open = 0;
  471. free(f->user);
  472. paqFree(f->paq);
  473. return 0;
  474. }
  475. char *
  476. rremove(Fid *f)
  477. {
  478. rclunk(f);
  479. return Erdonly;
  480. }
  481. char *
  482. rstat(Fid *f)
  483. {
  484. if(f->busy == 0)
  485. return Enotexist;
  486. thdr.stat = (uchar*)thdr.data;
  487. thdr.nstat = packDir(f->paq->dir, thdr.stat, mesgsize);
  488. if(thdr.nstat == 0)
  489. return Edirtoobig;
  490. return 0;
  491. }
  492. char *
  493. rwstat(Fid *f)
  494. {
  495. if(f->busy == 0)
  496. return Enotexist;
  497. return Erdonly;
  498. }
  499. Paq*
  500. paqCpy(Paq *s)
  501. {
  502. s->ref++;
  503. return s;
  504. }
  505. void
  506. paqFree(Paq *p)
  507. {
  508. if(p == nil)
  509. return;
  510. p->ref--;
  511. if(p->ref > 0)
  512. return;
  513. assert(p != root);
  514. paqFree(p->up);
  515. paqDirFree(p->dir);
  516. free(p);
  517. }
  518. void
  519. paqDirFree(PaqDir *pd)
  520. {
  521. if(pd == nil)
  522. return;
  523. free(pd->name);
  524. free(pd->uid);
  525. free(pd->gid);
  526. free(pd);
  527. }
  528. Qid
  529. paqDirQid(PaqDir *d)
  530. {
  531. Qid q;
  532. q.path = d->qid;
  533. q.vers = 0;
  534. q.type = d->mode >> 24;
  535. return q;
  536. }
  537. int
  538. packDir(PaqDir *s, uchar *buf, int n)
  539. {
  540. Dir dir;
  541. memset(&dir, 0, sizeof(dir));
  542. dir.qid = paqDirQid(s);
  543. dir.mode = s->mode;
  544. dir.atime = s->mtime;
  545. dir.mtime = s->mtime;
  546. dir.length = s->length;
  547. dir.name = s->name;
  548. dir.uid = s->uid;
  549. dir.gid = s->gid;
  550. dir.muid = s->uid;
  551. n = convD2M(&dir, buf, n);
  552. if(n < STATFIXLEN)
  553. return 0;
  554. return n;
  555. }
  556. Block *
  557. blockLoad(ulong addr, int type)
  558. {
  559. ulong age;
  560. int i, j;
  561. Block *b;
  562. if(addr == 0)
  563. return nil;
  564. cacheage++;
  565. /* age has wraped */
  566. if(cacheage == 0) {
  567. for(i=0; i<cachesize; i++)
  568. cache[i].age = 0;
  569. }
  570. j = -1;
  571. age = ~0;
  572. for(i=0; i<cachesize; i++) {
  573. b = &cache[i];
  574. if(b->age < age && b->ref == 0) {
  575. age = b->age;
  576. j = i;
  577. }
  578. if(b->addr != addr)
  579. continue;
  580. b->age = cacheage;
  581. b->ref++;
  582. return b;
  583. }
  584. if(j < 0)
  585. sysfatal("no empty spots in cache!\n");
  586. b = &cache[j];
  587. assert(b->ref == 0);
  588. if(!blockRead(b->data, addr, type)) {
  589. b->addr = 0;
  590. b->age = 0;
  591. return nil;
  592. }
  593. b->age = cacheage;
  594. b->addr = addr;
  595. b->ref = 1;
  596. return b;
  597. }
  598. void
  599. blockFree(Block *b)
  600. {
  601. if(b == nil)
  602. return;
  603. if(--b->ref > 0)
  604. return;
  605. assert(b->ref == 0);
  606. }
  607. Paq*
  608. paqWalk(Paq *s, char *name)
  609. {
  610. Block *ptr, *b;
  611. uchar *p, *ep;
  612. PaqDir *pd;
  613. int i, n;
  614. Paq *ss;
  615. if(strcmp(name, "..") == 0)
  616. return paqCpy(s->up);
  617. ptr = blockLoad(s->dir->offset, PointerBlock);
  618. if(ptr == nil)
  619. return nil;
  620. for(i=0; i<blocksize/4; i++) {
  621. b = blockLoad(getl(ptr->data+i*4), DirBlock);
  622. if(b == nil)
  623. break;
  624. p = b->data;
  625. ep = p + blocksize;
  626. while(checkDirSize(p, ep)) {
  627. n = gets(p);
  628. pd = getDir(p);
  629. if(strcmp(pd->name, name) == 0) {
  630. ss = emallocz(sizeof(Paq));
  631. ss->ref = 1;
  632. ss->up = paqCpy(s);
  633. ss->dir = pd;
  634. ss->qid = paqDirQid(pd);
  635. blockFree(b);
  636. blockFree(ptr);
  637. return ss;
  638. }
  639. paqDirFree(pd);
  640. p += n;
  641. }
  642. blockFree(b);
  643. }
  644. blockFree(ptr);
  645. return nil;
  646. }
  647. Fid *
  648. newfid(int fid)
  649. {
  650. Fid *f, *ff;
  651. ff = 0;
  652. for(f = fids; f; f = f->next)
  653. if(f->fid == fid)
  654. return f;
  655. else if(!ff && !f->busy)
  656. ff = f;
  657. if(ff){
  658. ff->fid = fid;
  659. return ff;
  660. }
  661. f = emallocz(sizeof *f);
  662. f->fid = fid;
  663. f->next = fids;
  664. fids = f;
  665. return f;
  666. }
  667. void
  668. io(int fd)
  669. {
  670. char *err;
  671. int n, pid;
  672. uchar *mdata;
  673. mdata = emalloc(mesgsize);
  674. pid = getpid();
  675. for(;;){
  676. n = read9pmsg(fd, mdata, mesgsize);
  677. if(n < 0)
  678. sysfatal("mount read");
  679. if(n == 0)
  680. break;
  681. if(convM2S(mdata, n, &rhdr) == 0)
  682. continue;
  683. if(debug)
  684. fprint(2, "paqfs %d:<-%F\n", pid, &rhdr);
  685. thdr.data = (char*)mdata + IOHDRSZ;
  686. if(!fcalls[rhdr.type])
  687. err = "bad fcall type";
  688. else
  689. err = (*fcalls[rhdr.type])(newfid(rhdr.fid));
  690. if(err){
  691. thdr.type = Rerror;
  692. thdr.ename = err;
  693. }else{
  694. thdr.type = rhdr.type + 1;
  695. thdr.fid = rhdr.fid;
  696. }
  697. thdr.tag = rhdr.tag;
  698. if(debug)
  699. fprint(2, "paqfs %d:->%F\n", pid, &thdr);/**/
  700. n = convS2M(&thdr, mdata, mesgsize);
  701. if(n == 0)
  702. sysfatal("convS2M sysfatal on write");
  703. if(write(fd, mdata, n) != n)
  704. sysfatal("mount write");
  705. }
  706. }
  707. int
  708. perm(PaqDir *s, char *user, int p)
  709. {
  710. ulong perm = s->mode;
  711. if((p*Pother) & perm)
  712. return 1;
  713. if(strcmp(user, s->gid)==0 && ((p*Pgroup) & perm))
  714. return 1;
  715. if(strcmp(user, s->uid)==0 && ((p*Powner) & perm))
  716. return 1;
  717. return 0;
  718. }
  719. void
  720. init(char *file, int verify)
  721. {
  722. PaqHeader hdr;
  723. PaqTrailer tlr;
  724. Dir *dir;
  725. int i;
  726. uchar *p;
  727. DigestState *ds = nil;
  728. PaqDir *r;
  729. Block *b;
  730. ulong offset;
  731. inflateinit();
  732. bin = Bopen(file, OREAD);
  733. if(bin == nil)
  734. sysfatal("could not open file: %s: %r", file);
  735. if(verify)
  736. ds = sha1(0, 0, 0, 0);
  737. readHeader(&hdr, file, ds);
  738. blocksize = hdr.blocksize;
  739. if(verify) {
  740. readBlocks(file, ds);
  741. } else {
  742. dir = dirstat(file);
  743. if(dir == nil)
  744. sysfatal("could not stat file: %s: %r", file);
  745. offset = dir->length - TrailerSize;
  746. free(dir);
  747. if(Bseek(bin, offset, 0) != offset)
  748. sysfatal("could not seek to trailer: %s", file);
  749. }
  750. readTrailer(&tlr, file, ds);
  751. /* asctime includes a newline - yuk */
  752. fprint(2, "%s: %s", hdr.label, asctime(gmtime(hdr.time)));
  753. fprint(2, "fingerprint: ");
  754. for(i=0; i<20; i++)
  755. fprint(2, "%.2x", tlr.sha1[i]);
  756. fprint(2, "\n");
  757. cache = emallocz(cachesize*sizeof(Block));
  758. p = emalloc(cachesize*blocksize);
  759. for(i=0; i<cachesize; i++) {
  760. cache[i].data = p;
  761. p += blocksize;
  762. }
  763. /* hand craft root */
  764. b = blockLoad(tlr.root, DirBlock);
  765. if(b == nil || !checkDirSize(b->data, b->data+blocksize))
  766. sysfatal("could not read root block: %s", file);
  767. r = getDir(b->data);
  768. blockFree(b);
  769. root = emallocz(sizeof(Paq));
  770. root->qid = paqDirQid(r);
  771. root->ref = 1;
  772. root->dir = r;
  773. root->up = root; /* parent of root is root */
  774. }
  775. int
  776. blockRead(uchar *data, ulong addr, int type)
  777. {
  778. uchar buf[BlockSize];
  779. PaqBlock b;
  780. uchar *cdat;
  781. if(Bseek(bin, addr, 0) != addr)
  782. return 0;
  783. if(Bread(bin, buf, BlockSize) != BlockSize)
  784. return 0;
  785. getBlock(buf, &b);
  786. if(b.magic != BlockMagic || b.size > blocksize || b.type != type)
  787. return 0;
  788. switch(b.encoding) {
  789. default:
  790. return 0;
  791. case NoEnc:
  792. if(Bread(bin, data, blocksize) < blocksize)
  793. return 0;
  794. break;
  795. case DeflateEnc:
  796. cdat = emalloc(b.size);
  797. if(Bread(bin, cdat, b.size) < b.size) {
  798. free(cdat);
  799. return 0;
  800. }
  801. if(inflateblock(data, blocksize, cdat, b.size) < 0) {
  802. free(cdat);
  803. return 0;
  804. }
  805. free(cdat);
  806. break;
  807. }
  808. if(adler32(0, data, blocksize) != b.adler32)
  809. return 0;
  810. return 1;
  811. }
  812. void
  813. readHeader(PaqHeader *hdr, char *name, DigestState *ds)
  814. {
  815. uchar buf[HeaderSize];
  816. if(Bread(bin, buf, HeaderSize) < HeaderSize)
  817. sysfatal("could not read header: %s: %r", name);
  818. if(ds)
  819. sha1(buf, HeaderSize, 0, ds);
  820. getHeader(buf, hdr);
  821. if(hdr->magic != HeaderMagic)
  822. sysfatal("bad header magic 0x%lux: %s", hdr->magic, name);
  823. if(hdr->version != Version)
  824. sysfatal("unknown file version: %s", name);
  825. }
  826. void
  827. readBlocks(char *name, DigestState *ds)
  828. {
  829. uchar *buf;
  830. PaqBlock b;
  831. buf = emalloc(BlockSize+blocksize);
  832. for(;;) {
  833. if(Bread(bin, buf, 4) < 4)
  834. sysfatal("could not read block: %s: %r", name);
  835. Bseek(bin, -4, 1);
  836. /* check if it is a data block */
  837. if(getl(buf) != BlockMagic)
  838. break;
  839. if(Bread(bin, buf, BlockSize) < BlockSize)
  840. sysfatal("could not read block: %s: %r", name);
  841. if(ds)
  842. sha1(buf, BlockSize, 0, ds);
  843. getBlock(buf, &b);
  844. if(b.size > blocksize)
  845. sysfatal("bad block size: %d: %s", b.size, name);
  846. if(ds) {
  847. if(Bread(bin, buf, b.size) < b.size)
  848. sysfatal("sysfatal reading block: %s: %r", name);
  849. sha1(buf, b.size, 0, ds);
  850. } else
  851. Bseek(bin, b.size, 1);
  852. }
  853. free(buf);
  854. }
  855. void
  856. readTrailer(PaqTrailer *tlr, char *name, DigestState *ds)
  857. {
  858. uchar buf[TrailerSize];
  859. uchar digest[20];
  860. if(Bread(bin, buf, TrailerSize) < TrailerSize)
  861. sysfatal("could not read trailer: %s: %r", name);
  862. getTrailer(buf, tlr);
  863. if(tlr->magic != TrailerMagic)
  864. sysfatal("bad trailer magic: %s", name);
  865. if(ds) {
  866. sha1(buf, TrailerSize-20, digest, ds);
  867. if(memcmp(digest, tlr->sha1, 20) != 0)
  868. sysfatal("bad sha1 digest: %s", name);
  869. }
  870. }
  871. void *
  872. emalloc(ulong n)
  873. {
  874. void *p;
  875. p = malloc(n);
  876. if(!p)
  877. sysfatal("out of memory");
  878. return p;
  879. }
  880. void *
  881. emallocz(ulong n)
  882. {
  883. void *p;
  884. p = emalloc(n);
  885. memset(p, 0, n);
  886. return p;
  887. }
  888. void *
  889. erealloc(void *p, ulong n)
  890. {
  891. p = realloc(p, n);
  892. if(!p)
  893. sysfatal("out of memory");
  894. return p;
  895. }
  896. char *
  897. estrdup(char *s)
  898. {
  899. s = strdup(s);
  900. if(s == nil)
  901. sysfatal("out of memory");
  902. return s;
  903. }
  904. ulong
  905. getl(uchar *p)
  906. {
  907. return (p[0]<<24) | (p[1]<<16) | (p[2]<<8) | p[3];
  908. }
  909. int
  910. gets(uchar *p)
  911. {
  912. return (p[0]<<8) | p[1];
  913. }
  914. int
  915. checkDirSize(uchar *p, uchar *ep)
  916. {
  917. int n;
  918. int i;
  919. if(ep-p < 2)
  920. return 0;
  921. n = gets(p);
  922. if(p+n > ep)
  923. return 0;
  924. ep = p+n;
  925. p += 22;
  926. for(i=0; i<3; i++) {
  927. if(p+2 > ep)
  928. return 0;
  929. n = gets(p);
  930. if(p+n > ep)
  931. return 0;
  932. p += n;
  933. }
  934. return 1;
  935. }
  936. void
  937. getHeader(uchar *p, PaqHeader *h)
  938. {
  939. h->magic = getl(p);
  940. h->version = gets(p+4);
  941. h->blocksize = gets(p+6);
  942. h->time = getl(p+8);
  943. memmove(h->label, p+12, sizeof(h->label));
  944. h->label[sizeof(h->label)-1] = 0;
  945. }
  946. void
  947. getTrailer(uchar *p, PaqTrailer *t)
  948. {
  949. t->magic = getl(p);
  950. t->root = getl(p+4);
  951. memmove(t->sha1, p+8, 20);
  952. }
  953. void
  954. getBlock(uchar *p, PaqBlock *b)
  955. {
  956. b->magic = getl(p);
  957. b->size = gets(p+4);
  958. b->type = p[6];
  959. b->encoding = p[7];
  960. b->adler32 = getl(p+8);
  961. }
  962. PaqDir *
  963. getDir(uchar *p)
  964. {
  965. PaqDir *pd;
  966. pd = emallocz(sizeof(PaqDir));
  967. pd->qid = getl(p+2);
  968. pd->mode = getl(p+6);
  969. pd->mtime = getl(p+10);
  970. pd->length = getl(p+14);
  971. pd->offset = getl(p+18);
  972. p += 22;
  973. pd->name = getstr(p);
  974. p += gets(p);
  975. pd->uid = getstr(p);
  976. p += gets(p);
  977. pd->gid = getstr(p);
  978. return pd;
  979. }
  980. char *
  981. getstr(uchar *p)
  982. {
  983. char *s;
  984. int n;
  985. n = gets(p);
  986. s = emalloc(n+1);
  987. memmove(s, p+2, n);
  988. s[n] = 0;
  989. return s;
  990. }
  991. void
  992. usage(void)
  993. {
  994. fprint(2, "usage: %s [-disv] [-c cachesize] [-m mountpoint] [-M mesgsize] paqfile\n", argv0);
  995. exits("usage");
  996. }