sub.c 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644
  1. #include "all.h"
  2. #include "io.h"
  3. #ifdef OLD
  4. #define swaboff swab4
  5. #else
  6. #define swaboff swab8
  7. #endif
  8. Filsys*
  9. fsstr(char *p)
  10. {
  11. Filsys *fs;
  12. for(fs=filsys; fs->name; fs++)
  13. if(strcmp(fs->name, p) == 0)
  14. return fs;
  15. return 0;
  16. }
  17. Filsys*
  18. dev2fs(Device *dev)
  19. {
  20. Filsys *fs;
  21. for(fs=filsys; fs->name; fs++)
  22. if(fs->dev == dev)
  23. return fs;
  24. return 0;
  25. }
  26. /*
  27. * allocate 'count' contiguous channels
  28. * of type 'type' and return pointer to base
  29. */
  30. Chan*
  31. chaninit(int type, int count, int data)
  32. {
  33. uchar *p;
  34. Chan *cp, *icp;
  35. int i;
  36. p = ialloc(count * (sizeof(Chan)+data), 0);
  37. icp = (Chan*)p;
  38. for(i=0; i<count; i++) {
  39. cp = (Chan*)p;
  40. cp->next = chans;
  41. chans = cp;
  42. cp->type = type;
  43. cp->chan = cons.chano;
  44. cons.chano++;
  45. strncpy(cp->whoname, "<none>", sizeof(cp->whoname));
  46. dofilter(&cp->work, C0a, C0b, 1);
  47. dofilter(&cp->rate, C0a, C0b, 1000);
  48. wlock(&cp->reflock);
  49. wunlock(&cp->reflock);
  50. rlock(&cp->reflock);
  51. runlock(&cp->reflock);
  52. p += sizeof(Chan);
  53. if(data){
  54. cp->pdata = p;
  55. p += data;
  56. }
  57. }
  58. return icp;
  59. }
  60. void
  61. fileinit(Chan *cp)
  62. {
  63. File *f, *prev;
  64. Tlock *t;
  65. int h;
  66. loop:
  67. lock(&flock);
  68. for (h=0; h<nelem(flist); h++)
  69. for (prev=0, f=flist[h]; f; prev=f, f=f->next) {
  70. if(f->cp != cp)
  71. continue;
  72. if(prev) {
  73. prev->next = f->next;
  74. f->next = flist[h];
  75. flist[h] = f;
  76. }
  77. flist[h] = f->next;
  78. unlock(&flock);
  79. qlock(f);
  80. if(t = f->tlock) {
  81. if(t->file == f)
  82. t->time = 0; /* free the lock */
  83. f->tlock = 0;
  84. }
  85. if(f->open & FREMOV)
  86. doremove(f, 0);
  87. freewp(f->wpath);
  88. f->open = 0;
  89. authfree(f->auth);
  90. f->auth = 0;
  91. f->cp = 0;
  92. qunlock(f);
  93. goto loop;
  94. }
  95. unlock(&flock);
  96. }
  97. #define NOFID (ulong)~0
  98. /*
  99. * returns a locked file structure
  100. */
  101. File*
  102. filep(Chan *cp, ulong fid, int flag)
  103. {
  104. File *f;
  105. int h;
  106. if(fid == NOFID)
  107. return 0;
  108. h = (long)cp + fid;
  109. if(h < 0)
  110. h = ~h;
  111. h %= nelem(flist);
  112. loop:
  113. lock(&flock);
  114. for(f=flist[h]; f; f=f->next)
  115. if(f->fid == fid && f->cp == cp){
  116. /*
  117. * Already in use is an error
  118. * when called from attach or clone (walk
  119. * in 9P2000). The console uses FID[12] and
  120. * never clunks them so catch that case.
  121. */
  122. if(flag == 0 || cp == cons.chan)
  123. goto out;
  124. unlock(&flock);
  125. return 0;
  126. }
  127. if(flag) {
  128. f = newfp();
  129. if(f) {
  130. f->fid = fid;
  131. f->cp = cp;
  132. f->wpath = 0;
  133. f->tlock = 0;
  134. f->doffset = 0;
  135. f->dslot = 0;
  136. f->auth = 0;
  137. f->next = flist[h];
  138. flist[h] = f;
  139. goto out;
  140. }
  141. }
  142. unlock(&flock);
  143. return 0;
  144. out:
  145. unlock(&flock);
  146. qlock(f);
  147. if(f->fid == fid && f->cp == cp)
  148. return f;
  149. qunlock(f);
  150. goto loop;
  151. }
  152. /*
  153. * always called with flock locked
  154. */
  155. File*
  156. newfp(void)
  157. {
  158. static int first;
  159. File *f;
  160. int start, i;
  161. i = first;
  162. start = i;
  163. do {
  164. f = &files[i];
  165. i++;
  166. if(i >= conf.nfile)
  167. i = 0;
  168. if(f->cp)
  169. continue;
  170. first = i;
  171. return f;
  172. } while(i != start);
  173. print("out of files\n");
  174. return 0;
  175. }
  176. void
  177. freefp(File *fp)
  178. {
  179. Chan *cp;
  180. File *f, *prev;
  181. int h;
  182. if(!fp || !(cp = fp->cp))
  183. return;
  184. h = (long)cp + fp->fid;
  185. if(h < 0)
  186. h = ~h;
  187. h %= nelem(flist);
  188. lock(&flock);
  189. for(prev=0,f=flist[h]; f; prev=f,f=f->next)
  190. if(f == fp) {
  191. if(prev)
  192. prev->next = f->next;
  193. else
  194. flist[h] = f->next;
  195. break;
  196. }
  197. fp->cp = 0;
  198. unlock(&flock);
  199. }
  200. int
  201. iaccess(File *f, Dentry *d, int m)
  202. {
  203. /* uid none gets only other permissions */
  204. if(f->uid != 0) {
  205. /*
  206. * owner
  207. */
  208. if(f->uid == d->uid)
  209. if((m<<6) & d->mode)
  210. return 0;
  211. /*
  212. * group membership
  213. */
  214. if(ingroup(f->uid, d->gid))
  215. if((m<<3) & d->mode)
  216. return 0;
  217. }
  218. /*
  219. * other
  220. */
  221. if(m & d->mode) {
  222. if((d->mode & DDIR) && (m == DEXEC))
  223. return 0;
  224. if(!ingroup(f->uid, 9999))
  225. return 0;
  226. }
  227. /*
  228. * various forms of superuser
  229. */
  230. if(wstatallow)
  231. return 0;
  232. if(duallow != 0 && duallow == f->uid)
  233. if((d->mode & DDIR) && (m == DREAD || m == DEXEC))
  234. return 0;
  235. return 1;
  236. }
  237. Tlock*
  238. tlocked(Iobuf *p, Dentry *d)
  239. {
  240. Tlock *t, *t1;
  241. Off qpath;
  242. Timet tim;
  243. Device *dev;
  244. tim = toytime();
  245. qpath = d->qid.path;
  246. dev = p->dev;
  247. again:
  248. t1 = 0;
  249. for(t=tlocks+NTLOCK-1; t>=tlocks; t--) {
  250. if(t->qpath == qpath)
  251. if(t->time >= tim)
  252. if(t->dev == dev)
  253. return nil; /* its locked */
  254. if(t1 != nil && t->time == 0)
  255. t1 = t; /* remember free lock */
  256. }
  257. if(t1 == 0) {
  258. // reclaim old locks
  259. lock(&tlocklock);
  260. for(t=tlocks+NTLOCK-1; t>=tlocks; t--)
  261. if(t->time < tim) {
  262. t->time = 0;
  263. t1 = t;
  264. }
  265. unlock(&tlocklock);
  266. }
  267. if(t1) {
  268. lock(&tlocklock);
  269. if(t1->time != 0) {
  270. unlock(&tlocklock);
  271. goto again;
  272. }
  273. t1->dev = dev;
  274. t1->qpath = qpath;
  275. t1->time = tim + TLOCK;
  276. unlock(&tlocklock);
  277. }
  278. /* botch
  279. * out of tlock nodes simulates
  280. * a locked file
  281. */
  282. return t1;
  283. }
  284. Wpath*
  285. newwp(void)
  286. {
  287. static int si = 0;
  288. int i;
  289. Wpath *w, *sw, *ew;
  290. i = si + 1;
  291. if(i < 0 || i >= conf.nwpath)
  292. i = 0;
  293. si = i;
  294. sw = &wpaths[i];
  295. ew = &wpaths[conf.nwpath];
  296. for(w=sw;;) {
  297. w++;
  298. if(w >= ew)
  299. w = &wpaths[0];
  300. if(w == sw) {
  301. print("out of wpaths\n");
  302. return 0;
  303. }
  304. if(w->refs)
  305. continue;
  306. lock(&wpathlock);
  307. if(w->refs) {
  308. unlock(&wpathlock);
  309. continue;
  310. }
  311. w->refs = 1;
  312. w->up = 0;
  313. unlock(&wpathlock);
  314. return w;
  315. }
  316. }
  317. void
  318. freewp(Wpath *w)
  319. {
  320. lock(&wpathlock);
  321. for(; w; w=w->up)
  322. w->refs--;
  323. unlock(&wpathlock);
  324. }
  325. Off
  326. qidpathgen(Device *dev)
  327. {
  328. Iobuf *p;
  329. Superb *sb;
  330. Off path;
  331. p = getbuf(dev, superaddr(dev), Bread|Bmod);
  332. if(!p || checktag(p, Tsuper, QPSUPER))
  333. panic("newqid: super block");
  334. sb = (Superb*)p->iobuf;
  335. sb->qidgen++;
  336. path = sb->qidgen;
  337. putbuf(p);
  338. return path;
  339. }
  340. /* truncating to length > 0 */
  341. static void
  342. truncfree(Truncstate *ts, Device *dev, int d, Iobuf *p, int i)
  343. {
  344. int pastlast;
  345. Off a;
  346. pastlast = ts->pastlast;
  347. a = ((Off *)p->iobuf)[i];
  348. if (d > 0 || pastlast)
  349. buffree(dev, a, d, ts);
  350. if (pastlast) {
  351. ((Off *)p->iobuf)[i] = 0;
  352. p->flags |= Bmod|Bimm;
  353. } else if (d == 0 && ts->relblk == ts->lastblk)
  354. ts->pastlast = 1;
  355. if (d == 0)
  356. ts->relblk++;
  357. }
  358. /*
  359. * free the block at `addr' on dev.
  360. * if it's an indirect block (d [depth] > 0),
  361. * first recursively free all the blocks it names.
  362. *
  363. * ts->relblk is the block number within the file of this
  364. * block (or the first data block eventually pointed to via
  365. * this indirect block).
  366. */
  367. void
  368. buffree(Device *dev, Off addr, int d, Truncstate *ts)
  369. {
  370. Iobuf *p;
  371. Off a;
  372. int i, pastlast;
  373. if(!addr)
  374. return;
  375. pastlast = (ts == nil? 1: ts->pastlast);
  376. /*
  377. * if this is an indirect block, recurse and free any
  378. * suitable blocks within it (possibly via further indirect blocks).
  379. */
  380. if(d > 0) {
  381. d--;
  382. p = getbuf(dev, addr, Bread);
  383. if(p) {
  384. if (ts == nil) /* common case: create */
  385. for(i=INDPERBUF-1; i>=0; i--) {
  386. a = ((Off *)p->iobuf)[i];
  387. buffree(dev, a, d, nil);
  388. }
  389. else /* wstat truncation */
  390. for (i = 0; i < INDPERBUF; i++)
  391. truncfree(ts, dev, d, p, i);
  392. putbuf(p);
  393. }
  394. }
  395. if (!pastlast)
  396. return;
  397. /*
  398. * having zeroed the pointer to this block, add it to the free list.
  399. * stop outstanding i/o
  400. */
  401. p = getbuf(dev, addr, Bprobe);
  402. if(p) {
  403. p->flags &= ~(Bmod|Bimm);
  404. putbuf(p);
  405. }
  406. /*
  407. * dont put written worm
  408. * blocks into free list
  409. */
  410. if(dev->type == Devcw) {
  411. i = cwfree(dev, addr);
  412. if(i)
  413. return;
  414. }
  415. p = getbuf(dev, superaddr(dev), Bread|Bmod);
  416. if(!p || checktag(p, Tsuper, QPSUPER))
  417. panic("buffree: super block");
  418. addfree(dev, addr, (Superb*)p->iobuf);
  419. putbuf(p);
  420. }
  421. Off
  422. bufalloc(Device *dev, int tag, long qid, int uid)
  423. {
  424. Iobuf *bp, *p;
  425. Superb *sb;
  426. Off a, n;
  427. p = getbuf(dev, superaddr(dev), Bread|Bmod);
  428. if(!p || checktag(p, Tsuper, QPSUPER)) {
  429. print("bufalloc: super block\n");
  430. if(p)
  431. putbuf(p);
  432. return 0;
  433. }
  434. sb = (Superb*)p->iobuf;
  435. loop:
  436. n = --sb->fbuf.nfree;
  437. sb->tfree--;
  438. if(n < 0 || n >= FEPERBUF) {
  439. print("bufalloc: %Z: bad freelist\n", dev);
  440. n = 0;
  441. sb->fbuf.free[0] = 0;
  442. }
  443. a = sb->fbuf.free[n];
  444. if(n <= 0) {
  445. if(a == 0) {
  446. sb->tfree = 0;
  447. sb->fbuf.nfree = 1;
  448. if(dev->type == Devcw) {
  449. n = uid;
  450. if(n < 0 || n >= nelem(growacct))
  451. n = 0;
  452. growacct[n]++;
  453. if(cwgrow(dev, sb, uid))
  454. goto loop;
  455. }
  456. putbuf(p);
  457. print("fs %Z full uid=%d\n", dev, uid);
  458. return 0;
  459. }
  460. bp = getbuf(dev, a, Bread);
  461. if(!bp || checktag(bp, Tfree, QPNONE)) {
  462. if(bp)
  463. putbuf(bp);
  464. putbuf(p);
  465. return 0;
  466. }
  467. sb->fbuf = *(Fbuf*)bp->iobuf;
  468. putbuf(bp);
  469. }
  470. bp = getbuf(dev, a, Bmod);
  471. memset(bp->iobuf, 0, RBUFSIZE);
  472. settag(bp, tag, qid);
  473. if(tag == Tind1 || tag == Tind2 || tag == Tdir)
  474. bp->flags |= Bimm;
  475. putbuf(bp);
  476. putbuf(p);
  477. return a;
  478. }
  479. /*
  480. * what are legal characters in a name?
  481. * only disallow control characters.
  482. * a) utf avoids control characters.
  483. * b) '/' may not be the separator
  484. */
  485. int
  486. checkname(char *n)
  487. {
  488. int i, c;
  489. for(i=0; i<NAMELEN; i++) {
  490. c = *n & 0xff;
  491. if(c == 0) {
  492. if(i == 0)
  493. return 1;
  494. memset(n, 0, NAMELEN-i);
  495. return 0;
  496. }
  497. if(c <= 040)
  498. return 1;
  499. n++;
  500. }
  501. return 1; /* too long */
  502. }
  503. void
  504. addfree(Device *dev, Off addr, Superb *sb)
  505. {
  506. int n;
  507. Iobuf *p;
  508. n = sb->fbuf.nfree;
  509. if(n < 0 || n > FEPERBUF)
  510. panic("addfree: bad freelist");
  511. if(n >= FEPERBUF) {
  512. p = getbuf(dev, addr, Bmod|Bimm);
  513. if(p == 0)
  514. panic("addfree: getbuf");
  515. *(Fbuf*)p->iobuf = sb->fbuf;
  516. settag(p, Tfree, QPNONE);
  517. putbuf(p);
  518. n = 0;
  519. }
  520. sb->fbuf.free[n++] = addr;
  521. sb->fbuf.nfree = n;
  522. sb->tfree++;
  523. if(addr >= sb->fsize)
  524. sb->fsize = addr+1;
  525. }
  526. static int
  527. Yfmt(Fmt* fmt)
  528. {
  529. Chan *cp;
  530. char s[20];
  531. cp = va_arg(fmt->args, Chan*);
  532. sprint(s, "C%d.%.3d", cp->type, cp->chan);
  533. return fmtstrcpy(fmt, s);
  534. }
  535. static int
  536. Zfmt(Fmt* fmt)
  537. {
  538. Device *d;
  539. int c, c1;
  540. char s[100];
  541. d = va_arg(fmt->args, Device*);
  542. if(d == 0) {
  543. sprint(s, "Z***");
  544. goto out;
  545. }
  546. switch(d->type) {
  547. default:
  548. sprint(s, "D%d", d->type);
  549. break;
  550. case Devwren:
  551. c = 'w';
  552. goto d1;
  553. case Devide:
  554. c = 'h';
  555. goto d1;
  556. case Devworm:
  557. c = 'r';
  558. goto d1;
  559. case Devlworm:
  560. c = 'l';
  561. goto d1;
  562. d1:
  563. if(d->wren.ctrl == 0 && d->wren.lun == 0)
  564. sprint(s, "%c%d", c, d->wren.targ);
  565. else
  566. sprint(s, "%c%d.%d.%d", c, d->wren.ctrl, d->wren.targ, d->wren.lun);
  567. break;
  568. case Devmcat:
  569. c = '(';
  570. c1 = ')';
  571. goto d2;
  572. case Devmlev:
  573. c = '[';
  574. c1 = ']';
  575. goto d2;
  576. case Devmirr:
  577. c = '{';
  578. c1 = '}';
  579. d2:
  580. if(d->cat.first == d->cat.last)
  581. sprint(s, "%c%Z%c", c, d->cat.first, c1);
  582. else
  583. if(d->cat.first->link == d->cat.last)
  584. sprint(s, "%c%Z%Z%c", c, d->cat.first, d->cat.last, c1);
  585. else
  586. sprint(s, "%c%Z-%Z%c", c, d->cat.first, d->cat.last, c1);
  587. break;
  588. case Devro:
  589. sprint(s, "o%Z%Z", d->ro.parent->cw.c, d->ro.parent->cw.w);
  590. break;
  591. case Devcw:
  592. sprint(s, "c%Z%Z", d->cw.c, d->cw.w);
  593. break;
  594. case Devjuke:
  595. sprint(s, "j%Z%Z", d->j.j, d->j.m);
  596. break;
  597. case Devfworm:
  598. sprint(s, "f%Z", d->fw.fw);
  599. break;
  600. case Devpart:
  601. sprint(s, "p(%Z)%ld.%ld",
  602. d->part.d, d->part.base, d->part.size);
  603. break;
  604. case Devswab:
  605. sprint(s, "x%Z", d->swab.d);
  606. break;
  607. case Devnone:
  608. sprint(s, "n");
  609. break;
  610. }
  611. out:
  612. return fmtstrcpy(fmt, s);
  613. }
  614. static int
  615. Wfmt(Fmt* fmt)
  616. {
  617. Filter* a;
  618. char s[30];
  619. a = va_arg(fmt->args, Filter*);
  620. sprint(s, "%lud", fdf(a->filter, a->c3*a->c1));
  621. return fmtstrcpy(fmt, s);
  622. }
  623. static int
  624. Gfmt(Fmt* fmt)
  625. {
  626. int t;
  627. char *s;
  628. t = va_arg(fmt->args, int);
  629. s = "<badtag>";
  630. if(t >= 0 && t < MAXTAG)
  631. s = tagnames[t];
  632. return fmtstrcpy(fmt, s);
  633. }
  634. static int
  635. Efmt(Fmt* fmt)
  636. {
  637. char s[64];
  638. uchar *p;
  639. p = va_arg(fmt->args, uchar*);
  640. sprint(s, "%.2ux%.2ux%.2ux%.2ux%.2ux%.2ux",
  641. p[0], p[1], p[2], p[3], p[4], p[5]);
  642. return fmtstrcpy(fmt, s);
  643. }
  644. static int
  645. Ifmt(Fmt* fmt)
  646. {
  647. char s[64];
  648. uchar *p;
  649. p = va_arg(fmt->args, uchar*);
  650. sprint(s, "%d.%d.%d.%d", p[0], p[1], p[2], p[3]);
  651. return fmtstrcpy(fmt, s);
  652. }
  653. void
  654. formatinit(void)
  655. {
  656. quotefmtinstall();
  657. fmtinstall('Y', Yfmt); /* print channels */
  658. fmtinstall('Z', Zfmt); /* print devices */
  659. fmtinstall('W', Wfmt); /* print filters */
  660. fmtinstall('G', Gfmt); /* print tags */
  661. fmtinstall('T', Tfmt); /* print times */
  662. fmtinstall('E', Efmt); /* print ether addresses */
  663. fmtinstall('I', Ifmt); /* print ip addresses */
  664. }
  665. void
  666. rootream(Device *dev, Off addr)
  667. {
  668. Iobuf *p;
  669. Dentry *d;
  670. p = getbuf(dev, addr, Bmod|Bimm);
  671. memset(p->iobuf, 0, RBUFSIZE);
  672. settag(p, Tdir, QPROOT);
  673. d = getdir(p, 0);
  674. strcpy(d->name, "/");
  675. d->uid = -1;
  676. d->gid = -1;
  677. d->mode = DALLOC | DDIR |
  678. ((DREAD|DEXEC) << 6) |
  679. ((DREAD|DEXEC) << 3) |
  680. ((DREAD|DEXEC) << 0);
  681. d->qid = QID9P1(QPROOT|QPDIR,0);
  682. d->atime = time();
  683. d->mtime = d->atime;
  684. d->muid = 0;
  685. putbuf(p);
  686. }
  687. void
  688. superream(Device *dev, Off addr)
  689. {
  690. Iobuf *p;
  691. Superb *s;
  692. Off i;
  693. p = getbuf(dev, addr, Bmod|Bimm);
  694. memset(p->iobuf, 0, RBUFSIZE);
  695. settag(p, Tsuper, QPSUPER);
  696. s = (Superb*)p->iobuf;
  697. s->fstart = 2;
  698. s->fsize = devsize(dev);
  699. s->fbuf.nfree = 1;
  700. s->qidgen = 10;
  701. #ifdef AUTOSWAB
  702. s->magic = 0x123456789abcdef0;
  703. #endif
  704. for(i=s->fsize-1; i>=addr+2; i--)
  705. addfree(dev, i, s);
  706. putbuf(p);
  707. }
  708. struct
  709. {
  710. Lock;
  711. Msgbuf *smsgbuf;
  712. Msgbuf *lmsgbuf;
  713. } msgalloc;
  714. /*
  715. * pre-allocate some message buffers at boot time.
  716. * if this supply is exhausted, more will be allocated as needed.
  717. */
  718. void
  719. mbinit(void)
  720. {
  721. Msgbuf *mb;
  722. Rabuf *rb;
  723. int i;
  724. lock(&msgalloc);
  725. unlock(&msgalloc);
  726. msgalloc.lmsgbuf = 0;
  727. msgalloc.smsgbuf = 0;
  728. for(i=0; i<conf.nlgmsg; i++) {
  729. mb = ialloc(sizeof(Msgbuf), 0);
  730. if(1)
  731. mb->xdata = ialloc(LARGEBUF+256, 256);
  732. else
  733. mb->xdata = ialloc(LARGEBUF+OFFMSG, LINESIZE);
  734. mb->flags = LARGE;
  735. mb->free = 0;
  736. mbfree(mb);
  737. cons.nlarge++;
  738. }
  739. for(i=0; i<conf.nsmmsg; i++) {
  740. mb = ialloc(sizeof(Msgbuf), 0);
  741. if(1)
  742. mb->xdata = ialloc(SMALLBUF+256, 256);
  743. else
  744. mb->xdata = ialloc(SMALLBUF+OFFMSG, LINESIZE);
  745. mb->flags = 0;
  746. mb->free = 0;
  747. mbfree(mb);
  748. cons.nsmall++;
  749. }
  750. memset(mballocs, 0, sizeof(mballocs));
  751. lock(&rabuflock);
  752. unlock(&rabuflock);
  753. rabuffree = 0;
  754. for(i=0; i<1000; i++) {
  755. rb = ialloc(sizeof(*rb), 0);
  756. rb->link = rabuffree;
  757. rabuffree = rb;
  758. }
  759. }
  760. Msgbuf*
  761. mballoc(int count, Chan *cp, int category)
  762. {
  763. Msgbuf *mb;
  764. ilock(&msgalloc);
  765. if(count > SMALLBUF) {
  766. if(count > LARGEBUF)
  767. panic("msgbuf count");
  768. mb = msgalloc.lmsgbuf;
  769. if(mb == 0) {
  770. mb = ialloc(sizeof(Msgbuf), 0);
  771. if(1)
  772. mb->xdata = ialloc(LARGEBUF+256, 256);
  773. else
  774. mb->xdata = ialloc(LARGEBUF+OFFMSG, LINESIZE);
  775. mb->free = 0;
  776. cons.nlarge++;
  777. } else
  778. msgalloc.lmsgbuf = mb->next;
  779. mb->flags = LARGE;
  780. } else {
  781. mb = msgalloc.smsgbuf;
  782. if(mb == 0) {
  783. mb = ialloc(sizeof(Msgbuf), 0);
  784. if(1)
  785. mb->xdata = ialloc(SMALLBUF+256, 256);
  786. else
  787. mb->xdata = ialloc(SMALLBUF+OFFMSG, LINESIZE);
  788. mb->free = 0;
  789. cons.nsmall++;
  790. } else
  791. msgalloc.smsgbuf = mb->next;
  792. mb->flags = 0;
  793. }
  794. mballocs[category]++;
  795. iunlock(&msgalloc);
  796. mb->count = count;
  797. mb->chan = cp;
  798. mb->next = 0;
  799. mb->param = 0;
  800. mb->category = category;
  801. if(1)
  802. mb->data = mb->xdata+256;
  803. else
  804. mb->data = mb->xdata+OFFMSG;
  805. mb->free = 0;
  806. return mb;
  807. }
  808. void
  809. mbfree(Msgbuf *mb)
  810. {
  811. if(mb == nil)
  812. return;
  813. if(mb->flags & BTRACE)
  814. print("mbfree: BTRACE cat=%d flags=%ux, caller 0x%lux\n",
  815. mb->category, mb->flags, getcallerpc(&mb));
  816. /*
  817. * drivers which perform non cache coherent DMA manage their own buffer
  818. * pool of uncached buffers and provide their own free routine.
  819. * this is provided mainly for ethernet drivers ported from cpu kernel.
  820. */
  821. if(mb->flags&Mbrcvbuf) {
  822. if (mb->free == nil)
  823. panic("freeb: nil mb->free");
  824. (*mb->free)(mb);
  825. return;
  826. }
  827. if(mb->flags & FREE)
  828. panic("mbfree already free");
  829. ilock(&msgalloc);
  830. mballocs[mb->category]--;
  831. mb->flags |= FREE;
  832. if(mb->flags & LARGE) {
  833. mb->next = msgalloc.lmsgbuf;
  834. msgalloc.lmsgbuf = mb;
  835. } else {
  836. mb->next = msgalloc.smsgbuf;
  837. msgalloc.smsgbuf = mb;
  838. }
  839. mb->data = 0;
  840. mb->free = 0;
  841. iunlock(&msgalloc);
  842. }
  843. /*
  844. * returns 1 if n is prime
  845. * used for adjusting lengths
  846. * of hashing things.
  847. * there is no need to be clever
  848. */
  849. int
  850. prime(vlong n)
  851. {
  852. long i;
  853. if((n%2) == 0)
  854. return 0;
  855. for(i=3;; i+=2) {
  856. if((n%i) == 0)
  857. return 0;
  858. if((vlong)i*i >= n)
  859. return 1;
  860. }
  861. }
  862. char*
  863. getwd(char *word, char *line)
  864. {
  865. int c, n;
  866. while(*line == ' ')
  867. line++;
  868. for(n=0; n<Maxword; n++) {
  869. c = *line;
  870. if(c == ' ' || c == 0 || c == '\n')
  871. break;
  872. line++;
  873. *word++ = c;
  874. }
  875. *word = 0;
  876. return line;
  877. }
  878. void
  879. hexdump(void *a, int n)
  880. {
  881. char s1[30], s2[4];
  882. uchar *p;
  883. int i;
  884. p = a;
  885. s1[0] = 0;
  886. for(i=0; i<n; i++) {
  887. sprint(s2, " %.2ux", p[i]);
  888. strcat(s1, s2);
  889. if((i&7) == 7) {
  890. print("%s\n", s1);
  891. s1[0] = 0;
  892. }
  893. }
  894. if(s1[0])
  895. print("%s\n", s1);
  896. }
  897. void*
  898. recv(Queue *q, int)
  899. {
  900. User *p;
  901. void *a;
  902. int i, c;
  903. long s;
  904. if(q == 0)
  905. panic("recv null q");
  906. for(;;) {
  907. lock(q);
  908. c = q->count;
  909. if(c > 0) {
  910. i = q->loc;
  911. a = q->args[i];
  912. i++;
  913. if(i >= q->size)
  914. i = 0;
  915. q->loc = i;
  916. q->count = c-1;
  917. p = q->whead;
  918. if(p) {
  919. q->whead = p->qnext;
  920. if(q->whead == 0)
  921. q->wtail = 0;
  922. ready(p);
  923. }
  924. unlock(q);
  925. return a;
  926. }
  927. p = q->rtail;
  928. if(p == 0)
  929. q->rhead = u;
  930. else
  931. p->qnext = u;
  932. q->rtail = u;
  933. u->qnext = 0;
  934. s = splhi();
  935. u->state = Recving;
  936. unlock(q);
  937. sched();
  938. splx(s);
  939. }
  940. }
  941. void
  942. send(Queue *q, void *a)
  943. {
  944. User *p;
  945. int i, c;
  946. long s;
  947. if(q == 0)
  948. panic("send null q");
  949. for(;;) {
  950. lock(q);
  951. c = q->count;
  952. if(c < q->size) {
  953. i = q->loc + c;
  954. if(i >= q->size)
  955. i -= q->size;
  956. q->args[i] = a;
  957. q->count = c+1;
  958. p = q->rhead;
  959. if(p) {
  960. q->rhead = p->qnext;
  961. if(q->rhead == 0)
  962. q->rtail = 0;
  963. ready(p);
  964. }
  965. unlock(q);
  966. return;
  967. }
  968. p = q->wtail;
  969. if(p == 0)
  970. q->whead = u;
  971. else
  972. p->qnext = u;
  973. q->wtail = u;
  974. u->qnext = 0;
  975. s = splhi();
  976. u->state = Sending;
  977. unlock(q);
  978. sched();
  979. splx(s);
  980. }
  981. }
  982. Queue*
  983. newqueue(int size)
  984. {
  985. Queue *q;
  986. q = ialloc(sizeof(Queue) + (size-1)*sizeof(void*), 0);
  987. q->size = size;
  988. lock(q);
  989. unlock(q);
  990. return q;
  991. }
  992. no(void*)
  993. {
  994. return 0;
  995. }
  996. int
  997. devread(Device *d, Off b, void *c)
  998. {
  999. int e;
  1000. for (;;)
  1001. switch(d->type)
  1002. {
  1003. case Devcw:
  1004. return cwread(d, b, c);
  1005. case Devjuke:
  1006. d = d->j.m;
  1007. break;
  1008. case Devro:
  1009. return roread(d, b, c);
  1010. case Devwren:
  1011. return wrenread(d, b, c);
  1012. case Devide:
  1013. return ideread(d, b, c);
  1014. case Devworm:
  1015. case Devlworm:
  1016. return wormread(d, b, c);
  1017. case Devfworm:
  1018. return fwormread(d, b, c);
  1019. case Devmcat:
  1020. return mcatread(d, b, c);
  1021. case Devmlev:
  1022. return mlevread(d, b, c);
  1023. case Devmirr:
  1024. return mirrread(d, b, c);
  1025. case Devpart:
  1026. return partread(d, b, c);
  1027. case Devswab:
  1028. e = devread(d->swab.d, b, c);
  1029. if(e == 0)
  1030. swab(c, 0);
  1031. return e;
  1032. case Devnone:
  1033. print("read from device none(%lld)\n", (Wideoff)b);
  1034. return 1;
  1035. default:
  1036. panic("illegal device in read: %Z %lld", d, (Wideoff)b);
  1037. return 1;
  1038. }
  1039. }
  1040. int
  1041. devwrite(Device *d, Off b, void *c)
  1042. {
  1043. int e;
  1044. /*
  1045. * set readonly to non-0 to prevent all writes;
  1046. * mainly for trying dangerous experiments.
  1047. */
  1048. if (readonly)
  1049. return 0;
  1050. for (;;)
  1051. switch(d->type)
  1052. {
  1053. case Devcw:
  1054. return cwwrite(d, b, c);
  1055. case Devjuke:
  1056. d = d->j.m;
  1057. break;
  1058. case Devro:
  1059. print("write to ro device %Z(%lld)\n", d, (Wideoff)b);
  1060. return 1;
  1061. case Devwren:
  1062. return wrenwrite(d, b, c);
  1063. case Devide:
  1064. return idewrite(d, b, c);
  1065. case Devworm:
  1066. case Devlworm:
  1067. return wormwrite(d, b, c);
  1068. case Devfworm:
  1069. return fwormwrite(d, b, c);
  1070. case Devmcat:
  1071. return mcatwrite(d, b, c);
  1072. case Devmlev:
  1073. return mlevwrite(d, b, c);
  1074. case Devmirr:
  1075. return mirrwrite(d, b, c);
  1076. case Devpart:
  1077. return partwrite(d, b, c);
  1078. case Devswab:
  1079. swab(c, 1);
  1080. e = devwrite(d->swab.d, b, c);
  1081. swab(c, 0);
  1082. return e;
  1083. case Devnone:
  1084. /* checktag() can generate blocks with type devnone */
  1085. return 0;
  1086. default:
  1087. panic("illegal device in write: %Z %ld", d, b);
  1088. return 1;
  1089. }
  1090. }
  1091. Devsize
  1092. devsize(Device *d)
  1093. {
  1094. for (;;)
  1095. switch(d->type)
  1096. {
  1097. case Devcw:
  1098. case Devro:
  1099. return cwsize(d);
  1100. case Devjuke:
  1101. d = d->j.m;
  1102. break;
  1103. case Devwren:
  1104. return wrensize(d);
  1105. case Devide:
  1106. return idesize(d);
  1107. case Devworm:
  1108. case Devlworm:
  1109. return wormsize(d);
  1110. case Devfworm:
  1111. return fwormsize(d);
  1112. case Devmcat:
  1113. return mcatsize(d);
  1114. case Devmlev:
  1115. return mlevsize(d);
  1116. case Devmirr:
  1117. return mirrsize(d);
  1118. case Devpart:
  1119. return partsize(d);
  1120. case Devswab:
  1121. d = d->swab.d;
  1122. break;
  1123. default:
  1124. panic("illegal device in dev_size: %Z", d);
  1125. return 0;
  1126. }
  1127. }
  1128. Off
  1129. superaddr(Device *d)
  1130. {
  1131. for (;;)
  1132. switch(d->type) {
  1133. default:
  1134. return SUPER_ADDR;
  1135. case Devcw:
  1136. case Devro:
  1137. return cwsaddr(d);
  1138. case Devswab:
  1139. d = d->swab.d;
  1140. break;
  1141. }
  1142. }
  1143. Off
  1144. getraddr(Device *d)
  1145. {
  1146. for (;;)
  1147. switch(d->type) {
  1148. default:
  1149. return ROOT_ADDR;
  1150. case Devcw:
  1151. case Devro:
  1152. return cwraddr(d);
  1153. case Devswab:
  1154. d = d->swab.d;
  1155. break;
  1156. }
  1157. }
  1158. void
  1159. devream(Device *d, int top)
  1160. {
  1161. Device *l;
  1162. loop:
  1163. print(" devream: %Z %d\n", d, top);
  1164. switch(d->type) {
  1165. default:
  1166. print("ream: unknown dev type %Z\n", d);
  1167. return;
  1168. case Devcw:
  1169. devream(d->cw.w, 0);
  1170. devream(d->cw.c, 0);
  1171. if(top) {
  1172. wlock(&mainlock);
  1173. cwream(d);
  1174. wunlock(&mainlock);
  1175. }
  1176. devinit(d);
  1177. return;
  1178. case Devfworm:
  1179. devream(d->fw.fw, 0);
  1180. fwormream(d);
  1181. break;
  1182. case Devpart:
  1183. devream(d->part.d, 0);
  1184. break;
  1185. case Devmlev:
  1186. case Devmcat:
  1187. case Devmirr:
  1188. for(l=d->cat.first; l; l=l->link)
  1189. devream(l, 0);
  1190. break;
  1191. case Devjuke:
  1192. case Devworm:
  1193. case Devlworm:
  1194. case Devwren:
  1195. case Devide:
  1196. break;
  1197. case Devswab:
  1198. d = d->swab.d;
  1199. goto loop;
  1200. }
  1201. devinit(d);
  1202. if(top) {
  1203. wlock(&mainlock);
  1204. rootream(d, ROOT_ADDR);
  1205. superream(d, SUPER_ADDR);
  1206. wunlock(&mainlock);
  1207. }
  1208. }
  1209. void
  1210. devrecover(Device *d)
  1211. {
  1212. for (;;) {
  1213. print("recover: %Z\n", d);
  1214. switch(d->type) {
  1215. default:
  1216. print("recover: unknown dev type %Z\n", d);
  1217. return;
  1218. case Devcw:
  1219. wlock(&mainlock); /* recover */
  1220. cwrecover(d);
  1221. wunlock(&mainlock);
  1222. return;
  1223. case Devswab:
  1224. d = d->swab.d;
  1225. break;
  1226. }
  1227. }
  1228. }
  1229. void
  1230. devinit(Device *d)
  1231. {
  1232. for (;;) {
  1233. if(d->init)
  1234. return;
  1235. d->init = 1;
  1236. print(" devinit %Z\n", d);
  1237. switch(d->type) {
  1238. default:
  1239. print("devinit unknown device %Z\n", d);
  1240. return;
  1241. case Devro:
  1242. cwinit(d->ro.parent);
  1243. return;
  1244. case Devcw:
  1245. cwinit(d);
  1246. return;
  1247. case Devjuke:
  1248. jukeinit(d);
  1249. return;
  1250. case Devwren:
  1251. wreninit(d);
  1252. return;
  1253. case Devide:
  1254. ideinit(d);
  1255. return;
  1256. case Devworm:
  1257. case Devlworm:
  1258. return;
  1259. case Devfworm:
  1260. fworminit(d);
  1261. return;
  1262. case Devmcat:
  1263. mcatinit(d);
  1264. return;
  1265. case Devmlev:
  1266. mlevinit(d);
  1267. return;
  1268. case Devmirr:
  1269. mirrinit(d);
  1270. return;
  1271. case Devpart:
  1272. partinit(d);
  1273. return;
  1274. case Devswab:
  1275. d = d->swab.d;
  1276. break;
  1277. case Devnone:
  1278. print("devinit of Devnone\n");
  1279. return;
  1280. }
  1281. }
  1282. }
  1283. void
  1284. swab2(void *c)
  1285. {
  1286. uchar *p;
  1287. int t;
  1288. p = c;
  1289. t = p[0];
  1290. p[0] = p[1];
  1291. p[1] = t;
  1292. }
  1293. void
  1294. swab4(void *c)
  1295. {
  1296. uchar *p;
  1297. int t;
  1298. p = c;
  1299. t = p[0];
  1300. p[0] = p[3];
  1301. p[3] = t;
  1302. t = p[1];
  1303. p[1] = p[2];
  1304. p[2] = t;
  1305. }
  1306. void
  1307. swab8(void *c)
  1308. {
  1309. uchar *p;
  1310. int t;
  1311. p = c;
  1312. t = p[0];
  1313. p[0] = p[7];
  1314. p[7] = t;
  1315. t = p[1];
  1316. p[1] = p[6];
  1317. p[6] = t;
  1318. t = p[2];
  1319. p[2] = p[5];
  1320. p[5] = t;
  1321. t = p[3];
  1322. p[3] = p[4];
  1323. p[4] = t;
  1324. }
  1325. /*
  1326. * swab a block
  1327. * flag = 0 -- convert from foreign to native
  1328. * flag = 1 -- convert from native to foreign
  1329. */
  1330. void
  1331. swab(void *c, int flag)
  1332. {
  1333. uchar *p;
  1334. Tag *t;
  1335. int i, j;
  1336. Dentry *d;
  1337. Cache *h;
  1338. Bucket *b;
  1339. Superb *s;
  1340. Fbuf *f;
  1341. Off *l;
  1342. /* swab the tag */
  1343. p = (uchar*)c;
  1344. t = (Tag*)(p + BUFSIZE);
  1345. if(!flag) {
  1346. swab2(&t->pad);
  1347. swab2(&t->tag);
  1348. swaboff(&t->path);
  1349. }
  1350. /* swab each block type */
  1351. switch(t->tag) {
  1352. default:
  1353. print("no swab for tag=%G rw=%d\n", t->tag, flag);
  1354. for(j=0; j<16; j++)
  1355. print(" %.2x", p[BUFSIZE+j]);
  1356. print("\n");
  1357. for(i=0; i<16; i++) {
  1358. print("%.4x", i*16);
  1359. for(j=0; j<16; j++)
  1360. print(" %.2x", p[i*16+j]);
  1361. print("\n");
  1362. }
  1363. panic("swab");
  1364. break;
  1365. case Tsuper:
  1366. s = (Superb*)p;
  1367. swaboff(&s->fbuf.nfree);
  1368. for(i=0; i<FEPERBUF; i++)
  1369. swaboff(&s->fbuf.free[i]);
  1370. #ifdef AUTOSWAB
  1371. swaboff(&s->magic);
  1372. #endif
  1373. swaboff(&s->fstart);
  1374. swaboff(&s->fsize);
  1375. swaboff(&s->tfree);
  1376. swaboff(&s->qidgen);
  1377. swaboff(&s->cwraddr);
  1378. swaboff(&s->roraddr);
  1379. swaboff(&s->last);
  1380. swaboff(&s->next);
  1381. break;
  1382. case Tdir:
  1383. for(i=0; i<DIRPERBUF; i++) {
  1384. d = (Dentry*)p + i;
  1385. swab2(&d->uid);
  1386. swab2(&d->gid);
  1387. swab2(&d->mode);
  1388. swab2(&d->muid);
  1389. swaboff(&d->qid.path);
  1390. swab4(&d->qid.version);
  1391. swaboff(&d->size);
  1392. for(j=0; j<NDBLOCK; j++)
  1393. swaboff(&d->dblock[j]);
  1394. for (j = 0; j < NIBLOCK; j++)
  1395. swaboff(&d->iblocks[j]);
  1396. swab4(&d->atime);
  1397. swab4(&d->mtime);
  1398. }
  1399. break;
  1400. case Tind1:
  1401. case Tind2:
  1402. #ifndef OLD
  1403. case Tind3:
  1404. case Tind4:
  1405. /* add more Tind tags here ... */
  1406. #endif
  1407. l = (Off *)p;
  1408. for(i=0; i<INDPERBUF; i++) {
  1409. swaboff(l);
  1410. l++;
  1411. }
  1412. break;
  1413. case Tfree:
  1414. f = (Fbuf*)p;
  1415. swaboff(&f->nfree);
  1416. for(i=0; i<FEPERBUF; i++)
  1417. swaboff(&f->free[i]);
  1418. break;
  1419. case Tbuck:
  1420. for(i=0; i<BKPERBLK; i++) {
  1421. b = (Bucket*)p + i;
  1422. swab4(&b->agegen);
  1423. for(j=0; j<CEPERBK; j++) {
  1424. swab2(&b->entry[j].age);
  1425. swab2(&b->entry[j].state);
  1426. swaboff(&b->entry[j].waddr);
  1427. }
  1428. }
  1429. break;
  1430. case Tcache:
  1431. h = (Cache*)p;
  1432. swaboff(&h->maddr);
  1433. swaboff(&h->msize);
  1434. swaboff(&h->caddr);
  1435. swaboff(&h->csize);
  1436. swaboff(&h->fsize);
  1437. swaboff(&h->wsize);
  1438. swaboff(&h->wmax);
  1439. swaboff(&h->sbaddr);
  1440. swaboff(&h->cwraddr);
  1441. swaboff(&h->roraddr);
  1442. swab4(&h->toytime);
  1443. swab4(&h->time);
  1444. break;
  1445. case Tnone: // unitialized
  1446. case Tfile: // someone elses problem
  1447. case Tvirgo: // bit map -- all bytes
  1448. case Tconfig: // configuration string -- all bytes
  1449. break;
  1450. }
  1451. /* swab the tag */
  1452. if(flag) {
  1453. swab2(&t->pad);
  1454. swab2(&t->tag);
  1455. swaboff(&t->path);
  1456. }
  1457. }