paqfs.c 19 KB

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