9fsys.c 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800
  1. #include "stdinc.h"
  2. #include "dat.h"
  3. #include "fns.h"
  4. #include "9.h"
  5. typedef struct Fsys Fsys;
  6. struct Fsys {
  7. VtLock* lock;
  8. char* name;
  9. char* dev;
  10. char* venti;
  11. Fs* fs;
  12. VtSession* session;
  13. int ref;
  14. int noauth;
  15. int noperm;
  16. int wstatallow;
  17. Fsys* next;
  18. };
  19. static struct {
  20. VtLock* lock;
  21. Fsys* head;
  22. Fsys* tail;
  23. char* curfsys;
  24. } sbox;
  25. static char *_argv0;
  26. #define argv0 _argv0
  27. static char FsysAll[] = "all";
  28. static char EFsysBusy[] = "fsys: '%s' busy";
  29. static char EFsysExists[] = "fsys: '%s' already exists";
  30. static char EFsysNoCurrent[] = "fsys: no current fsys";
  31. static char EFsysNotFound[] = "fsys: '%s' not found";
  32. static char EFsysNotOpen[] = "fsys: '%s' not open";
  33. static Fsys*
  34. _fsysGet(char* name)
  35. {
  36. Fsys *fsys;
  37. if(name == nil || name[0] == '\0')
  38. name = "main";
  39. vtRLock(sbox.lock);
  40. for(fsys = sbox.head; fsys != nil; fsys = fsys->next){
  41. if(strcmp(name, fsys->name) == 0){
  42. fsys->ref++;
  43. break;
  44. }
  45. }
  46. vtRUnlock(sbox.lock);
  47. if(fsys == nil)
  48. vtSetError(EFsysNotFound, name);
  49. return fsys;
  50. }
  51. static int
  52. cmdPrintConfig(int argc, char* argv[])
  53. {
  54. Fsys *fsys;
  55. char *usage = "usage: printconfig";
  56. ARGBEGIN{
  57. default:
  58. return cliError(usage);
  59. }ARGEND
  60. if(argc)
  61. return cliError(usage);
  62. vtRLock(sbox.lock);
  63. for(fsys = sbox.head; fsys != nil; fsys = fsys->next){
  64. consPrint("\tfsys %s config %s\n", fsys->name, fsys->dev);
  65. if(fsys->venti && fsys->venti[0])
  66. consPrint("\tfsys %s venti %q\n", fsys->name, fsys->venti);
  67. }
  68. vtRUnlock(sbox.lock);
  69. return 1;
  70. }
  71. Fsys*
  72. fsysGet(char* name)
  73. {
  74. Fsys *fsys;
  75. if((fsys = _fsysGet(name)) == nil)
  76. return nil;
  77. vtLock(fsys->lock);
  78. if(fsys->fs == nil){
  79. vtSetError(EFsysNotOpen, fsys->name);
  80. vtUnlock(fsys->lock);
  81. fsysPut(fsys);
  82. return nil;
  83. }
  84. vtUnlock(fsys->lock);
  85. return fsys;
  86. }
  87. char*
  88. fsysGetName(Fsys* fsys)
  89. {
  90. return fsys->name;
  91. }
  92. Fsys*
  93. fsysIncRef(Fsys* fsys)
  94. {
  95. vtLock(sbox.lock);
  96. fsys->ref++;
  97. vtUnlock(sbox.lock);
  98. return fsys;
  99. }
  100. void
  101. fsysPut(Fsys* fsys)
  102. {
  103. vtLock(sbox.lock);
  104. assert(fsys->ref > 0);
  105. fsys->ref--;
  106. vtUnlock(sbox.lock);
  107. }
  108. Fs*
  109. fsysGetFs(Fsys* fsys)
  110. {
  111. assert(fsys != nil && fsys->fs != nil);
  112. return fsys->fs;
  113. }
  114. void
  115. fsysFsRlock(Fsys* fsys)
  116. {
  117. vtRLock(fsys->fs->elk);
  118. }
  119. void
  120. fsysFsRUnlock(Fsys* fsys)
  121. {
  122. vtRUnlock(fsys->fs->elk);
  123. }
  124. int
  125. fsysNoAuthCheck(Fsys* fsys)
  126. {
  127. return fsys->noauth;
  128. }
  129. int
  130. fsysNoPermCheck(Fsys* fsys)
  131. {
  132. return fsys->noperm;
  133. }
  134. int
  135. fsysWstatAllow(Fsys* fsys)
  136. {
  137. return fsys->wstatallow;
  138. }
  139. static char modechars[] = "YUGalLdHSATs";
  140. static ulong modebits[] = {
  141. ModeSticky,
  142. ModeSetUid,
  143. ModeSetGid,
  144. ModeAppend,
  145. ModeExclusive,
  146. ModeLink,
  147. ModeDir,
  148. ModeHidden,
  149. ModeSystem,
  150. ModeArchive,
  151. ModeTemporary,
  152. ModeSnapshot,
  153. 0
  154. };
  155. char*
  156. fsysModeString(ulong mode, char *buf)
  157. {
  158. int i;
  159. char *p;
  160. p = buf;
  161. for(i=0; modebits[i]; i++)
  162. if(mode & modebits[i])
  163. *p++ = modechars[i];
  164. sprint(p, "%luo", mode&0777);
  165. return buf;
  166. }
  167. int
  168. fsysParseMode(char* s, ulong* mode)
  169. {
  170. ulong x, y;
  171. char *p;
  172. x = 0;
  173. for(; *s < '0' || *s > '9'; s++){
  174. if(*s == 0)
  175. return 0;
  176. p = strchr(modechars, *s);
  177. if(p == nil)
  178. return 0;
  179. x |= modebits[p-modechars];
  180. }
  181. y = strtoul(s, &p, 8);
  182. if(*p != '\0' || y > 0777)
  183. return 0;
  184. *mode = x|y;
  185. return 1;
  186. }
  187. File*
  188. fsysGetRoot(Fsys* fsys, char* name)
  189. {
  190. File *root, *sub;
  191. assert(fsys != nil && fsys->fs != nil);
  192. root = fsGetRoot(fsys->fs);
  193. if(name == nil || strcmp(name, "") == 0)
  194. return root;
  195. sub = fileWalk(root, name);
  196. fileDecRef(root);
  197. return sub;
  198. }
  199. static Fsys*
  200. fsysAlloc(char* name, char* dev)
  201. {
  202. Fsys *fsys;
  203. vtLock(sbox.lock);
  204. for(fsys = sbox.head; fsys != nil; fsys = fsys->next){
  205. if(strcmp(fsys->name, name) != 0)
  206. continue;
  207. vtSetError(EFsysExists, name);
  208. vtUnlock(sbox.lock);
  209. return nil;
  210. }
  211. fsys = vtMemAllocZ(sizeof(Fsys));
  212. fsys->lock = vtLockAlloc();
  213. fsys->name = vtStrDup(name);
  214. fsys->dev = vtStrDup(dev);
  215. fsys->ref = 1;
  216. if(sbox.tail != nil)
  217. sbox.tail->next = fsys;
  218. else
  219. sbox.head = fsys;
  220. sbox.tail = fsys;
  221. vtUnlock(sbox.lock);
  222. return fsys;
  223. }
  224. static int
  225. fsysClose(Fsys* fsys, int argc, char* argv[])
  226. {
  227. char *usage = "usage: [fsys name] close";
  228. ARGBEGIN{
  229. default:
  230. return cliError(usage);
  231. }ARGEND
  232. if(argc)
  233. return cliError(usage);
  234. return cliError("close isn't working yet; halt %s and then kill fossil",
  235. fsys->name);
  236. /*
  237. * Oooh. This could be hard. What if fsys->ref != 1?
  238. * Also, fsClose() either does the job or panics, can we
  239. * gracefully detect it's still busy?
  240. *
  241. * More thought and care needed here.
  242. fsClose(fsys->fs);
  243. fsys->fs = nil;
  244. vtClose(fsys->session);
  245. fsys->session = nil;
  246. if(sbox.curfsys != nil && strcmp(fsys->name, sbox.curfsys) == 0){
  247. sbox.curfsys = nil;
  248. consPrompt(nil);
  249. }
  250. return 1;
  251. */
  252. }
  253. static int
  254. fsysVac(Fsys* fsys, int argc, char* argv[])
  255. {
  256. uchar score[VtScoreSize];
  257. char *usage = "usage: [fsys name] vac path";
  258. ARGBEGIN{
  259. default:
  260. return cliError(usage);
  261. }ARGEND
  262. if(argc != 1)
  263. return cliError(usage);
  264. if(!fsVac(fsys->fs, argv[0], score))
  265. return 0;
  266. consPrint("vac:%V\n", score);
  267. return 1;
  268. }
  269. static int
  270. fsysSnap(Fsys* fsys, int argc, char* argv[])
  271. {
  272. int doarchive;
  273. char *usage = "usage: [fsys name] snap [-a] [-s /active] [-d /archive/yyyy/mmmm]";
  274. char *src, *dst;
  275. src = nil;
  276. dst = nil;
  277. doarchive = 0;
  278. ARGBEGIN{
  279. default:
  280. return cliError(usage);
  281. case 'a':
  282. doarchive = 1;
  283. break;
  284. case 'd':
  285. if((dst = ARGF()) == nil)
  286. return cliError(usage);
  287. break;
  288. case 's':
  289. if((src = ARGF()) == nil)
  290. return cliError(usage);
  291. break;
  292. }ARGEND
  293. if(argc)
  294. return cliError(usage);
  295. if(!fsSnapshot(fsys->fs, src, dst, doarchive))
  296. return 0;
  297. return 1;
  298. }
  299. static int
  300. fsysSnapClean(Fsys *fsys, int argc, char* argv[])
  301. {
  302. u32int arch, snap, life;
  303. char *usage = "usage: [fsys name] snapclean [maxminutes]\n";
  304. ARGBEGIN{
  305. default:
  306. return cliError(usage);
  307. }ARGEND
  308. if(argc > 1)
  309. return cliError(usage);
  310. if(argc == 1)
  311. life = atoi(argv[0]);
  312. else
  313. snapGetTimes(fsys->fs->snap, &arch, &snap, &life);
  314. fsSnapshotCleanup(fsys->fs, life);
  315. return 1;
  316. }
  317. static int
  318. fsysSnapTime(Fsys* fsys, int argc, char* argv[])
  319. {
  320. char buf[128], *x;
  321. int hh, mm, changed;
  322. u32int arch, snap, life;
  323. char *usage = "usage: [fsys name] snaptime [-a hhmm] [-s snapminutes] [-t maxminutes]";
  324. changed = 0;
  325. snapGetTimes(fsys->fs->snap, &arch, &snap, &life);
  326. ARGBEGIN{
  327. case 'a':
  328. changed = 1;
  329. x = ARGF();
  330. if(x == nil)
  331. return cliError(usage);
  332. if(strcmp(x, "none") == 0){
  333. arch = ~(u32int)0;
  334. break;
  335. }
  336. if(strlen(x) != 4 || strspn(x, "0123456789") != 4)
  337. return cliError(usage);
  338. hh = (x[0]-'0')*10 + x[1]-'0';
  339. mm = (x[2]-'0')*10 + x[3]-'0';
  340. if(hh >= 24 || mm >= 60)
  341. return cliError(usage);
  342. arch = hh*60+mm;
  343. break;
  344. case 's':
  345. changed = 1;
  346. x = ARGF();
  347. if(x == nil)
  348. return cliError(usage);
  349. if(strcmp(x, "none") == 0){
  350. snap = ~(u32int)0;
  351. break;
  352. }
  353. snap = atoi(x);
  354. break;
  355. case 't':
  356. changed = 1;
  357. x = ARGF();
  358. if(x == nil)
  359. return cliError(usage);
  360. if(strcmp(x, "none") == 0){
  361. life = ~(u32int)0;
  362. break;
  363. }
  364. life = atoi(x);
  365. break;
  366. default:
  367. return cliError(usage);
  368. }ARGEND
  369. if(argc > 0)
  370. return cliError(usage);
  371. if(changed){
  372. snapSetTimes(fsys->fs->snap, arch, snap, life);
  373. return 1;
  374. }
  375. snapGetTimes(fsys->fs->snap, &arch, &snap, &life);
  376. if(arch != ~(u32int)0)
  377. sprint(buf, "-a %02d%02d", arch/60, arch%60);
  378. else
  379. sprint(buf, "-a none");
  380. if(snap != ~(u32int)0)
  381. sprint(buf+strlen(buf), " -s %d", snap);
  382. else
  383. sprint(buf+strlen(buf), " -s none");
  384. if(life != ~(u32int)0)
  385. sprint(buf+strlen(buf), " -t %ud", life);
  386. else
  387. sprint(buf+strlen(buf), " -t none");
  388. consPrint("\tsnaptime %s\n", buf);
  389. return 1;
  390. }
  391. static int
  392. fsysSync(Fsys* fsys, int argc, char* argv[])
  393. {
  394. char *usage = "usage: [fsys name] sync";
  395. int n;
  396. ARGBEGIN{
  397. default:
  398. return cliError(usage);
  399. }ARGEND
  400. if(argc > 0)
  401. return cliError(usage);
  402. n = cacheDirty(fsys->fs->cache);
  403. fsSync(fsys->fs);
  404. consPrint("\t%s sync: wrote %d blocks\n", fsys->name, n);
  405. return 1;
  406. }
  407. static int
  408. fsysHalt(Fsys *fsys, int argc, char* argv[])
  409. {
  410. char *usage = "usage: [fsys name] halt";
  411. ARGBEGIN{
  412. default:
  413. return cliError(usage);
  414. }ARGEND
  415. if(argc > 0)
  416. return cliError(usage);
  417. fsHalt(fsys->fs);
  418. return 1;
  419. }
  420. static int
  421. fsysUnhalt(Fsys *fsys, int argc, char* argv[])
  422. {
  423. char *usage = "usage: [fsys name] unhalt";
  424. ARGBEGIN{
  425. default:
  426. return cliError(usage);
  427. }ARGEND
  428. if(argc > 0)
  429. return cliError(usage);
  430. if(!fsys->fs->halted)
  431. return cliError("file system %s not halted", fsys->name);
  432. fsUnhalt(fsys->fs);
  433. return 1;
  434. }
  435. static int
  436. fsysRemove(Fsys* fsys, int argc, char* argv[])
  437. {
  438. File *file;
  439. char *usage = "usage: [fsys name] remove path ...";
  440. ARGBEGIN{
  441. default:
  442. return cliError(usage);
  443. }ARGEND
  444. if(argc == 0)
  445. return cliError(usage);
  446. vtRLock(fsys->fs->elk);
  447. while(argc > 0){
  448. if((file = fileOpen(fsys->fs, argv[0])) == nil)
  449. consPrint("%s: %R\n", argv[0]);
  450. else{
  451. if(!fileRemove(file, uidadm))
  452. consPrint("%s: %R\n", argv[0]);
  453. fileDecRef(file);
  454. }
  455. argc--;
  456. argv++;
  457. }
  458. vtRUnlock(fsys->fs->elk);
  459. return 1;
  460. }
  461. static int
  462. fsysClri(Fsys* fsys, int argc, char* argv[])
  463. {
  464. char *usage = "usage: [fsys name] clri path ...";
  465. ARGBEGIN{
  466. default:
  467. return cliError(usage);
  468. }ARGEND
  469. if(argc == 0)
  470. return cliError(usage);
  471. vtRLock(fsys->fs->elk);
  472. while(argc > 0){
  473. if(!fileClriPath(fsys->fs, argv[0], uidadm))
  474. consPrint("clri %s: %R\n", argv[0]);
  475. argc--;
  476. argv++;
  477. }
  478. vtRUnlock(fsys->fs->elk);
  479. return 1;
  480. }
  481. /*
  482. * Inspect and edit the labels for blocks on disk.
  483. */
  484. static int
  485. fsysLabel(Fsys* fsys, int argc, char* argv[])
  486. {
  487. Fs *fs;
  488. Label l;
  489. int n, r;
  490. u32int addr;
  491. Block *b, *bb;
  492. char *usage = "usage: [fsys name] label addr [type state epoch epochClose tag]";
  493. ARGBEGIN{
  494. default:
  495. return cliError(usage);
  496. }ARGEND
  497. if(argc != 1 && argc != 6)
  498. return cliError(usage);
  499. r = 0;
  500. vtRLock(fsys->fs->elk);
  501. fs = fsys->fs;
  502. addr = strtoul(argv[0], 0, 0);
  503. b = cacheLocal(fs->cache, PartData, addr, OReadOnly);
  504. if(b == nil)
  505. goto Out0;
  506. l = b->l;
  507. consPrint("%slabel %#ux %ud %ud %ud %ud %#x\n",
  508. argc==6 ? "old: " : "", addr, l.type, l.state,
  509. l.epoch, l.epochClose, l.tag);
  510. if(argc == 6){
  511. if(strcmp(argv[1], "-") != 0)
  512. l.type = atoi(argv[1]);
  513. if(strcmp(argv[2], "-") != 0)
  514. l.state = atoi(argv[2]);
  515. if(strcmp(argv[3], "-") != 0)
  516. l.epoch = strtoul(argv[3], 0, 0);
  517. if(strcmp(argv[4], "-") != 0)
  518. l.epochClose = strtoul(argv[4], 0, 0);
  519. if(strcmp(argv[5], "-") != 0)
  520. l.tag = strtoul(argv[5], 0, 0);
  521. consPrint("new: label %#ux %ud %ud %ud %ud %#x\n",
  522. addr, l.type, l.state, l.epoch, l.epochClose, l.tag);
  523. bb = _blockSetLabel(b, &l);
  524. if(bb == nil)
  525. goto Out1;
  526. n = 0;
  527. for(;;){
  528. if(blockWrite(bb)){
  529. while(bb->iostate != BioClean){
  530. assert(bb->iostate == BioWriting);
  531. vtSleep(bb->ioready);
  532. }
  533. break;
  534. }
  535. consPrint("blockWrite: %R\n");
  536. if(n++ >= 5){
  537. consPrint("giving up\n");
  538. break;
  539. }
  540. sleep(5*1000);
  541. }
  542. blockPut(bb);
  543. }
  544. r = 1;
  545. Out1:
  546. blockPut(b);
  547. Out0:
  548. vtRUnlock(fs->elk);
  549. return r;
  550. }
  551. /*
  552. * Inspect and edit the blocks on disk.
  553. */
  554. static int
  555. fsysBlock(Fsys* fsys, int argc, char* argv[])
  556. {
  557. Fs *fs;
  558. char *s;
  559. Block *b;
  560. uchar *buf;
  561. u32int addr;
  562. int c, count, i, offset;
  563. char *usage = "usage: [fsys name] block addr offset [count [data]]";
  564. ARGBEGIN{
  565. default:
  566. return cliError(usage);
  567. }ARGEND
  568. if(argc < 2 || argc > 4)
  569. return cliError(usage);
  570. fs = fsys->fs;
  571. addr = strtoul(argv[0], 0, 0);
  572. offset = strtoul(argv[1], 0, 0);
  573. if(offset < 0 || offset >= fs->blockSize){
  574. vtSetError("bad offset");
  575. return 0;
  576. }
  577. if(argc > 2)
  578. count = strtoul(argv[2], 0, 0);
  579. else
  580. count = 100000000;
  581. if(offset+count > fs->blockSize)
  582. count = fs->blockSize - count;
  583. vtRLock(fs->elk);
  584. b = cacheLocal(fs->cache, PartData, addr, argc==4 ? OReadWrite : OReadOnly);
  585. if(b == nil){
  586. vtSetError("cacheLocal %#ux: %R", addr);
  587. vtRUnlock(fs->elk);
  588. return 0;
  589. }
  590. consPrint("\t%sblock %#ux %ud %ud %.*H\n",
  591. argc==4 ? "old: " : "", addr, offset, count, count, b->data+offset);
  592. if(argc == 4){
  593. s = argv[3];
  594. if(strlen(s) != 2*count){
  595. vtSetError("bad data count");
  596. goto Out;
  597. }
  598. buf = vtMemAllocZ(count);
  599. for(i = 0; i < count*2; i++){
  600. if(s[i] >= '0' && s[i] <= '9')
  601. c = s[i] - '0';
  602. else if(s[i] >= 'a' && s[i] <= 'f')
  603. c = s[i] - 'a' + 10;
  604. else if(s[i] >= 'A' && s[i] <= 'F')
  605. c = s[i] - 'A' + 10;
  606. else{
  607. vtSetError("bad hex");
  608. vtMemFree(buf);
  609. goto Out;
  610. }
  611. if((i & 1) == 0)
  612. c <<= 4;
  613. buf[i>>1] |= c;
  614. }
  615. memmove(b->data+offset, buf, count);
  616. consPrint("\tnew: block %#ux %ud %ud %.*H\n",
  617. addr, offset, count, count, b->data+offset);
  618. blockDirty(b);
  619. }
  620. Out:
  621. blockPut(b);
  622. vtRUnlock(fs->elk);
  623. return 1;
  624. }
  625. /*
  626. * Free a disk block.
  627. */
  628. static int
  629. fsysBfree(Fsys* fsys, int argc, char* argv[])
  630. {
  631. Fs *fs;
  632. Label l;
  633. char *p;
  634. Block *b;
  635. u32int addr;
  636. char *usage = "usage: [fsys name] bfree addr ...";
  637. ARGBEGIN{
  638. default:
  639. return cliError(usage);
  640. }ARGEND
  641. if(argc == 0)
  642. return cliError(usage);
  643. fs = fsys->fs;
  644. vtRLock(fs->elk);
  645. while(argc > 0){
  646. addr = strtoul(argv[0], &p, 0);
  647. if(*p != '\0'){
  648. consPrint("bad address - '%s'\n", addr);
  649. /* syntax error; let's stop */
  650. vtRUnlock(fs->elk);
  651. return 0;
  652. }
  653. b = cacheLocal(fs->cache, PartData, addr, OReadOnly);
  654. if(b == nil){
  655. consPrint("loading %#ux: %R\n", addr);
  656. continue;
  657. }
  658. l = b->l;
  659. if(l.state == BsFree)
  660. consPrint("%#ux is already free\n", addr);
  661. else{
  662. consPrint("label %#ux %ud %ud %ud %ud %#x\n",
  663. addr, l.type, l.state, l.epoch, l.epochClose, l.tag);
  664. l.state = BsFree;
  665. l.type = BtMax;
  666. l.tag = 0;
  667. l.epoch = 0;
  668. l.epochClose = 0;
  669. if(!blockSetLabel(b, &l, 0))
  670. consPrint("freeing %#ux: %R\n", addr);
  671. }
  672. blockPut(b);
  673. argc--;
  674. argv++;
  675. }
  676. vtRUnlock(fs->elk);
  677. return 1;
  678. }
  679. static int
  680. fsysDf(Fsys *fsys, int argc, char* argv[])
  681. {
  682. char *usage = "usage: [fsys name] df";
  683. u32int used, tot, bsize;
  684. Fs *fs;
  685. ARGBEGIN{
  686. default:
  687. return cliError(usage);
  688. }ARGEND
  689. if(argc != 0)
  690. return cliError(usage);
  691. fs = fsys->fs;
  692. cacheCountUsed(fs->cache, fs->elo, &used, &tot, &bsize);
  693. consPrint("\t%s: %,llud used + %,llud free = %,llud (%ud%% used)\n",
  694. fsys->name, used*(vlong)bsize, (tot-used)*(vlong)bsize,
  695. tot*(vlong)bsize, used*100LL/tot);
  696. return 1;
  697. }
  698. /*
  699. * Zero an entry or a pointer.
  700. */
  701. static int
  702. fsysClrep(Fsys* fsys, int argc, char* argv[], int ch)
  703. {
  704. Fs *fs;
  705. Entry e;
  706. Block *b;
  707. u32int addr;
  708. int i, max, offset, sz;
  709. uchar zero[VtEntrySize];
  710. char *usage = "usage: [fsys name] clr%c addr offset ...";
  711. ARGBEGIN{
  712. default:
  713. return cliError(usage, ch);
  714. }ARGEND
  715. if(argc < 2)
  716. return cliError(usage, ch);
  717. fs = fsys->fs;
  718. vtRLock(fsys->fs->elk);
  719. addr = strtoul(argv[0], 0, 0);
  720. b = cacheLocal(fs->cache, PartData, addr, argc==4 ? OReadWrite : OReadOnly);
  721. if(b == nil){
  722. vtSetError("cacheLocal %#ux: %R", addr);
  723. Err:
  724. vtRUnlock(fsys->fs->elk);
  725. return 0;
  726. }
  727. switch(ch){
  728. default:
  729. vtSetError("clrep");
  730. goto Err;
  731. case 'e':
  732. if(b->l.type != BtDir){
  733. vtSetError("wrong block type");
  734. goto Err;
  735. }
  736. sz = VtEntrySize;
  737. memset(&e, 0, sizeof e);
  738. entryPack(&e, zero, 0);
  739. break;
  740. case 'p':
  741. if(b->l.type == BtDir || b->l.type == BtData){
  742. vtSetError("wrong block type");
  743. goto Err;
  744. }
  745. sz = VtScoreSize;
  746. memmove(zero, vtZeroScore, VtScoreSize);
  747. break;
  748. }
  749. max = fs->blockSize/sz;
  750. for(i = 1; i < argc; i++){
  751. offset = atoi(argv[i]);
  752. if(offset >= max){
  753. consPrint("\toffset %d too large (>= %d)\n", i, max);
  754. continue;
  755. }
  756. consPrint("\tblock %#ux %d %d %.*H\n", addr, offset*sz, sz, sz, b->data+offset*sz);
  757. memmove(b->data+offset*sz, zero, sz);
  758. }
  759. blockDirty(b);
  760. blockPut(b);
  761. vtRUnlock(fsys->fs->elk);
  762. return 1;
  763. }
  764. static int
  765. fsysClre(Fsys* fsys, int argc, char* argv[])
  766. {
  767. return fsysClrep(fsys, argc, argv, 'e');
  768. }
  769. static int
  770. fsysClrp(Fsys* fsys, int argc, char* argv[])
  771. {
  772. return fsysClrep(fsys, argc, argv, 'p');
  773. }
  774. static int
  775. fsysEsearch1(File* f, char* s, u32int elo)
  776. {
  777. int n, r;
  778. DirEntry de;
  779. DirEntryEnum *dee;
  780. File *ff;
  781. Entry e, ee;
  782. char *t;
  783. dee = deeOpen(f);
  784. if(dee == nil)
  785. return 0;
  786. n = 0;
  787. for(;;){
  788. r = deeRead(dee, &de);
  789. if(r < 0){
  790. consPrint("\tdeeRead %s/%s: %R\n", s, de.elem);
  791. break;
  792. }
  793. if(r == 0)
  794. break;
  795. if(de.mode & ModeSnapshot){
  796. if((ff = fileWalk(f, de.elem)) == nil)
  797. consPrint("\tcannot walk %s/%s: %R\n", s, de.elem);
  798. else{
  799. if(!fileGetSources(ff, &e, &ee))
  800. consPrint("\tcannot get sources for %s/%s: %R\n", s, de.elem);
  801. else if(e.snap != 0 && e.snap < elo){
  802. consPrint("\t%ud\tclri %s/%s\n", e.snap, s, de.elem);
  803. n++;
  804. }
  805. fileDecRef(ff);
  806. }
  807. }
  808. else if(de.mode & ModeDir){
  809. if((ff = fileWalk(f, de.elem)) == nil)
  810. consPrint("\tcannot walk %s/%s: %R\n", s, de.elem);
  811. else{
  812. t = smprint("%s/%s", s, de.elem);
  813. n += fsysEsearch1(ff, t, elo);
  814. vtMemFree(t);
  815. fileDecRef(ff);
  816. }
  817. }
  818. deCleanup(&de);
  819. if(r < 0)
  820. break;
  821. }
  822. deeClose(dee);
  823. return n;
  824. }
  825. static int
  826. fsysEsearch(Fs* fs, char* path, u32int elo)
  827. {
  828. int n;
  829. File *f;
  830. DirEntry de;
  831. f = fileOpen(fs, path);
  832. if(f == nil)
  833. return 0;
  834. if(!fileGetDir(f, &de)){
  835. consPrint("\tfileGetDir %s failed: %R\n", path);
  836. fileDecRef(f);
  837. return 0;
  838. }
  839. if((de.mode & ModeDir) == 0){
  840. fileDecRef(f);
  841. deCleanup(&de);
  842. return 0;
  843. }
  844. deCleanup(&de);
  845. n = fsysEsearch1(f, path, elo);
  846. fileDecRef(f);
  847. return n;
  848. }
  849. static int
  850. fsysEpoch(Fsys* fsys, int argc, char* argv[])
  851. {
  852. Fs *fs;
  853. int force, n, remove;
  854. u32int low, old;
  855. char *usage = "usage: [fsys name] epoch [[-ry] low]";
  856. force = 0;
  857. remove = 0;
  858. ARGBEGIN{
  859. case 'y':
  860. force = 1;
  861. break;
  862. case 'r':
  863. remove = 1;
  864. break;
  865. default:
  866. return cliError(usage);
  867. }ARGEND
  868. if(argc > 1)
  869. return cliError(usage);
  870. if(argc > 0)
  871. low = strtoul(argv[0], 0, 0);
  872. else
  873. low = ~(u32int)0;
  874. if(low == 0)
  875. return cliError("low epoch cannot be zero");
  876. fs = fsys->fs;
  877. vtRLock(fs->elk);
  878. consPrint("\tlow %ud hi %ud\n", fs->elo, fs->ehi);
  879. if(low == ~(u32int)0){
  880. vtRUnlock(fs->elk);
  881. return 1;
  882. }
  883. n = fsysEsearch(fsys->fs, "/archive", low);
  884. n += fsysEsearch(fsys->fs, "/snapshot", low);
  885. consPrint("\t%d snapshot%s found with epoch < %ud\n", n, n==1 ? "" : "s", low);
  886. vtRUnlock(fs->elk);
  887. /*
  888. * There's a small race here -- a new snapshot with epoch < low might
  889. * get introduced now that we unlocked fs->elk. Low has to
  890. * be <= fs->ehi. Of course, in order for this to happen low has
  891. * to be equal to the current fs->ehi _and_ a snapshot has to
  892. * run right now. This is a small enough window that I don't care.
  893. */
  894. if(n != 0 && !force){
  895. consPrint("\tnot setting low epoch\n");
  896. return 1;
  897. }
  898. old = fs->elo;
  899. if(!fsEpochLow(fs, low))
  900. consPrint("\tfsEpochLow: %R\n");
  901. else{
  902. consPrint("\told: epoch%s %ud\n", force ? " -y" : "", old);
  903. consPrint("\tnew: epoch%s %ud\n", force ? " -y" : "", fs->elo);
  904. if(fs->elo < low)
  905. consPrint("\twarning: new low epoch < old low epoch\n");
  906. if(force && remove)
  907. fsSnapshotRemove(fs);
  908. }
  909. return 1;
  910. }
  911. static int
  912. fsysCreate(Fsys* fsys, int argc, char* argv[])
  913. {
  914. int r;
  915. ulong mode;
  916. char *elem, *p, *path;
  917. char *usage = "usage: [fsys name] create path uid gid perm";
  918. DirEntry de;
  919. File *file, *parent;
  920. ARGBEGIN{
  921. default:
  922. return cliError(usage);
  923. }ARGEND
  924. if(argc != 4)
  925. return cliError(usage);
  926. if(!fsysParseMode(argv[3], &mode))
  927. return cliError(usage);
  928. if(mode&ModeSnapshot)
  929. return cliError("create - cannot create with snapshot bit set");
  930. if(strcmp(argv[1], uidnoworld) == 0)
  931. return cliError("permission denied");
  932. vtRLock(fsys->fs->elk);
  933. path = vtStrDup(argv[0]);
  934. if((p = strrchr(path, '/')) != nil){
  935. *p++ = '\0';
  936. elem = p;
  937. p = path;
  938. if(*p == '\0')
  939. p = "/";
  940. }
  941. else{
  942. p = "/";
  943. elem = path;
  944. }
  945. r = 0;
  946. if((parent = fileOpen(fsys->fs, p)) == nil)
  947. goto out;
  948. file = fileCreate(parent, elem, mode, argv[1]);
  949. fileDecRef(parent);
  950. if(file == nil){
  951. vtSetError("create %s/%s: %R", p, elem);
  952. goto out;
  953. }
  954. if(!fileGetDir(file, &de)){
  955. vtSetError("stat failed after create: %R");
  956. goto out1;
  957. }
  958. if(strcmp(de.gid, argv[2]) != 0){
  959. vtMemFree(de.gid);
  960. de.gid = vtStrDup(argv[2]);
  961. if(!fileSetDir(file, &de, argv[1])){
  962. vtSetError("wstat failed after create: %R");
  963. goto out2;
  964. }
  965. }
  966. r = 1;
  967. out2:
  968. deCleanup(&de);
  969. out1:
  970. fileDecRef(file);
  971. out:
  972. vtMemFree(path);
  973. vtRUnlock(fsys->fs->elk);
  974. return r;
  975. }
  976. static void
  977. fsysPrintStat(char *prefix, char *file, DirEntry *de)
  978. {
  979. char buf[64];
  980. if(prefix == nil)
  981. prefix = "";
  982. consPrint("%sstat %q %q %q %q %s %llud\n", prefix,
  983. file, de->elem, de->uid, de->gid, fsysModeString(de->mode, buf), de->size);
  984. }
  985. static int
  986. fsysStat(Fsys* fsys, int argc, char* argv[])
  987. {
  988. int i;
  989. File *f;
  990. DirEntry de;
  991. char *usage = "usage: [fsys name] stat files...";
  992. ARGBEGIN{
  993. default:
  994. return cliError(usage);
  995. }ARGEND
  996. if(argc == 0)
  997. return cliError(usage);
  998. vtRLock(fsys->fs->elk);
  999. for(i=0; i<argc; i++){
  1000. if((f = fileOpen(fsys->fs, argv[i])) == nil){
  1001. consPrint("%s: %R\n");
  1002. continue;
  1003. }
  1004. if(!fileGetDir(f, &de)){
  1005. consPrint("%s: %R\n");
  1006. fileDecRef(f);
  1007. continue;
  1008. }
  1009. fsysPrintStat("\t", argv[i], &de);
  1010. deCleanup(&de);
  1011. fileDecRef(f);
  1012. }
  1013. vtRUnlock(fsys->fs->elk);
  1014. return 1;
  1015. }
  1016. static int
  1017. fsysWstat(Fsys *fsys, int argc, char* argv[])
  1018. {
  1019. File *f;
  1020. char *p;
  1021. DirEntry de;
  1022. char *usage = "usage: [fsys name] wstat file elem uid gid mode length\n"
  1023. "\tuse - for any field to mean don't change";
  1024. ARGBEGIN{
  1025. default:
  1026. return cliError(usage);
  1027. }ARGEND
  1028. if(argc != 6)
  1029. return cliError(usage);
  1030. vtRLock(fsys->fs->elk);
  1031. if((f = fileOpen(fsys->fs, argv[0])) == nil){
  1032. vtSetError("console wstat - walk - %R");
  1033. vtRUnlock(fsys->fs->elk);
  1034. return 0;
  1035. }
  1036. if(!fileGetDir(f, &de)){
  1037. vtSetError("console wstat - stat - %R");
  1038. fileDecRef(f);
  1039. vtRUnlock(fsys->fs->elk);
  1040. return 0;
  1041. }
  1042. fsysPrintStat("\told: w", argv[0], &de);
  1043. if(strcmp(argv[1], "-") != 0){
  1044. if(!validFileName(argv[1])){
  1045. vtSetError("console wstat - bad elem");
  1046. goto error;
  1047. }
  1048. vtMemFree(de.elem);
  1049. de.elem = vtStrDup(argv[1]);
  1050. }
  1051. if(strcmp(argv[2], "-") != 0){
  1052. if(!validUserName(argv[2])){
  1053. vtSetError("console wstat - bad uid");
  1054. goto error;
  1055. }
  1056. vtMemFree(de.uid);
  1057. de.uid = vtStrDup(argv[2]);
  1058. }
  1059. if(strcmp(argv[3], "-") != 0){
  1060. if(!validUserName(argv[3])){
  1061. vtSetError("console wstat - bad gid");
  1062. goto error;
  1063. }
  1064. vtMemFree(de.gid);
  1065. de.gid = vtStrDup(argv[3]);
  1066. }
  1067. if(strcmp(argv[4], "-") != 0){
  1068. if(!fsysParseMode(argv[4], &de.mode)){
  1069. vtSetError("console wstat - bad mode");
  1070. goto error;
  1071. }
  1072. }
  1073. if(strcmp(argv[5], "-") != 0){
  1074. de.size = strtoull(argv[5], &p, 0);
  1075. if(argv[5][0] == '\0' || *p != '\0' || (vlong)de.size < 0){
  1076. vtSetError("console wstat - bad length");
  1077. goto error;
  1078. }
  1079. }
  1080. if(!fileSetDir(f, &de, uidadm)){
  1081. vtSetError("console wstat - %R");
  1082. goto error;
  1083. }
  1084. deCleanup(&de);
  1085. if(!fileGetDir(f, &de)){
  1086. vtSetError("console wstat - stat2 - %R");
  1087. goto error;
  1088. }
  1089. fsysPrintStat("\tnew: w", argv[0], &de);
  1090. deCleanup(&de);
  1091. fileDecRef(f);
  1092. vtRUnlock(fsys->fs->elk);
  1093. return 1;
  1094. error:
  1095. deCleanup(&de); /* okay to do this twice */
  1096. fileDecRef(f);
  1097. vtRUnlock(fsys->fs->elk);
  1098. return 0;
  1099. }
  1100. static void
  1101. fsckClri(Fsck *fsck, char *name, MetaBlock *mb, int i, Block *b)
  1102. {
  1103. USED(name);
  1104. if((fsck->flags&DoClri) == 0)
  1105. return;
  1106. mbDelete(mb, i);
  1107. mbPack(mb);
  1108. blockDirty(b);
  1109. }
  1110. static void
  1111. fsckClose(Fsck *fsck, Block *b, u32int epoch)
  1112. {
  1113. Label l;
  1114. if((fsck->flags&DoClose) == 0)
  1115. return;
  1116. l = b->l;
  1117. if(l.state == BsFree || (l.state&BsClosed)){
  1118. consPrint("%#ux is already closed\n", b->addr);
  1119. return;
  1120. }
  1121. if(epoch){
  1122. l.state |= BsClosed;
  1123. l.epochClose = epoch;
  1124. }else
  1125. l.state = BsFree;
  1126. if(!blockSetLabel(b, &l, 0))
  1127. consPrint("%#ux setlabel: %R\n", b->addr);
  1128. }
  1129. static void
  1130. fsckClre(Fsck *fsck, Block *b, int offset)
  1131. {
  1132. Entry e;
  1133. if((fsck->flags&DoClre) == 0)
  1134. return;
  1135. if(offset<0 || offset*VtEntrySize >= fsck->bsize){
  1136. consPrint("bad clre\n");
  1137. return;
  1138. }
  1139. memset(&e, 0, sizeof e);
  1140. entryPack(&e, b->data, offset);
  1141. blockDirty(b);
  1142. }
  1143. static void
  1144. fsckClrp(Fsck *fsck, Block *b, int offset)
  1145. {
  1146. if((fsck->flags&DoClrp) == 0)
  1147. return;
  1148. if(offset<0 || offset*VtScoreSize >= fsck->bsize){
  1149. consPrint("bad clre\n");
  1150. return;
  1151. }
  1152. memmove(b->data+offset*VtScoreSize, vtZeroScore, VtScoreSize);
  1153. blockDirty(b);
  1154. }
  1155. static int
  1156. fsysCheck(Fsys *fsys, int argc, char *argv[])
  1157. {
  1158. int i, halting;
  1159. char *usage = "usage: [fsys name] check [-v] [options]";
  1160. Fsck fsck;
  1161. Block *b;
  1162. Super super;
  1163. memset(&fsck, 0, sizeof fsck);
  1164. fsck.fs = fsys->fs;
  1165. fsck.clri = fsckClri;
  1166. fsck.clre = fsckClre;
  1167. fsck.clrp = fsckClrp;
  1168. fsck.close = fsckClose;
  1169. fsck.print = consPrint;
  1170. ARGBEGIN{
  1171. default:
  1172. return cliError(usage);
  1173. }ARGEND
  1174. for(i=0; i<argc; i++){
  1175. if(strcmp(argv[i], "pblock") == 0)
  1176. fsck.printblocks = 1;
  1177. else if(strcmp(argv[i], "pdir") == 0)
  1178. fsck.printdirs = 1;
  1179. else if(strcmp(argv[i], "pfile") == 0)
  1180. fsck.printfiles = 1;
  1181. else if(strcmp(argv[i], "bclose") == 0)
  1182. fsck.flags |= DoClose;
  1183. else if(strcmp(argv[i], "clri") == 0)
  1184. fsck.flags |= DoClri;
  1185. else if(strcmp(argv[i], "clre") == 0)
  1186. fsck.flags |= DoClre;
  1187. else if(strcmp(argv[i], "clrp") == 0)
  1188. fsck.flags |= DoClrp;
  1189. else if(strcmp(argv[i], "fix") == 0)
  1190. fsck.flags |= DoClose|DoClri|DoClre|DoClrp;
  1191. else if(strcmp(argv[i], "venti") == 0)
  1192. fsck.useventi = 1;
  1193. else if(strcmp(argv[i], "snapshot") == 0)
  1194. fsck.walksnapshots = 1;
  1195. else{
  1196. consPrint("unknown option '%s'\n", argv[i]);
  1197. return cliError(usage);
  1198. }
  1199. }
  1200. halting = fsys->fs->halted==0;
  1201. if(halting)
  1202. fsHalt(fsys->fs);
  1203. if(fsys->fs->arch){
  1204. b = superGet(fsys->fs->cache, &super);
  1205. if(b == nil){
  1206. consPrint("could not load super block\n");
  1207. goto Out;
  1208. }
  1209. blockPut(b);
  1210. if(super.current != NilBlock){
  1211. consPrint("cannot check fs while archiver is running; wait for it to finish\n");
  1212. goto Out;
  1213. }
  1214. }
  1215. fsCheck(&fsck);
  1216. consPrint("fsck: %d clri, %d clre, %d clrp, %d bclose\n",
  1217. fsck.nclri, fsck.nclre, fsck.nclrp, fsck.nclose);
  1218. Out:
  1219. if(halting)
  1220. fsUnhalt(fsys->fs);
  1221. return 1;
  1222. }
  1223. static int
  1224. fsysVenti(char* name, int argc, char* argv[])
  1225. {
  1226. int r;
  1227. char *host;
  1228. char *usage = "usage: [fsys name] venti [address]";
  1229. Fsys *fsys;
  1230. ARGBEGIN{
  1231. default:
  1232. return cliError(usage);
  1233. }ARGEND
  1234. if(argc == 0)
  1235. host = nil;
  1236. else if(argc == 1)
  1237. host = argv[0];
  1238. else
  1239. return cliError(usage);
  1240. if((fsys = _fsysGet(name)) == nil)
  1241. return 0;
  1242. vtLock(fsys->lock);
  1243. if(host == nil)
  1244. host = fsys->venti;
  1245. else{
  1246. vtMemFree(fsys->venti);
  1247. if(host[0])
  1248. fsys->venti = vtStrDup(host);
  1249. else{
  1250. host = nil;
  1251. fsys->venti = nil;
  1252. }
  1253. }
  1254. /* already open: do a redial */
  1255. if(fsys->fs != nil){
  1256. if(fsys->session == nil){
  1257. vtSetError("file system was opened with -V");
  1258. r = 0;
  1259. goto out;
  1260. }
  1261. r = 1;
  1262. if(!vtRedial(fsys->session, host)
  1263. || !vtConnect(fsys->session, 0))
  1264. r = 0;
  1265. goto out;
  1266. }
  1267. /* not yet open: try to dial */
  1268. if(fsys->session)
  1269. vtClose(fsys->session);
  1270. r = 1;
  1271. if((fsys->session = vtDial(host, 0)) == nil
  1272. || !vtConnect(fsys->session, 0))
  1273. r = 0;
  1274. out:
  1275. vtUnlock(fsys->lock);
  1276. fsysPut(fsys);
  1277. return r;
  1278. }
  1279. static int
  1280. fsysOpen(char* name, int argc, char* argv[])
  1281. {
  1282. char *p, *host;
  1283. Fsys *fsys;
  1284. long ncache;
  1285. int noauth, noventi, noperm, rflag, wstatallow;
  1286. char *usage = "usage: fsys name open [-APVWr] [-c ncache]";
  1287. ncache = 1000;
  1288. noauth = noperm = wstatallow = noventi = 0;
  1289. rflag = OReadWrite;
  1290. ARGBEGIN{
  1291. default:
  1292. return cliError(usage);
  1293. case 'A':
  1294. noauth = 1;
  1295. break;
  1296. case 'P':
  1297. noperm = 1;
  1298. break;
  1299. case 'V':
  1300. noventi = 1;
  1301. break;
  1302. case 'W':
  1303. wstatallow = 1;
  1304. break;
  1305. case 'c':
  1306. p = ARGF();
  1307. if(p == nil)
  1308. return cliError(usage);
  1309. ncache = strtol(argv[0], &p, 0);
  1310. if(ncache <= 0 || p == argv[0] || *p != '\0')
  1311. return cliError(usage);
  1312. break;
  1313. case 'r':
  1314. rflag = OReadOnly;
  1315. break;
  1316. }ARGEND
  1317. if(argc)
  1318. return cliError(usage);
  1319. if((fsys = _fsysGet(name)) == nil)
  1320. return 0;
  1321. vtLock(fsys->lock);
  1322. if(fsys->fs != nil){
  1323. vtSetError(EFsysBusy, fsys->name);
  1324. vtUnlock(fsys->lock);
  1325. fsysPut(fsys);
  1326. return 0;
  1327. }
  1328. if(noventi){
  1329. if(fsys->session){
  1330. vtClose(fsys->session);
  1331. fsys->session = nil;
  1332. }
  1333. }
  1334. else if(fsys->session == nil){
  1335. if(fsys->venti && fsys->venti[0])
  1336. host = fsys->venti;
  1337. else
  1338. host = nil;
  1339. fsys->session = vtDial(host, 1);
  1340. if(!vtConnect(fsys->session, nil) && !noventi)
  1341. fprint(2, "warning: connecting to venti: %R\n");
  1342. }
  1343. if((fsys->fs = fsOpen(fsys->dev, fsys->session, ncache, rflag)) == nil){
  1344. vtSetError("fsOpen: %R");
  1345. vtUnlock(fsys->lock);
  1346. fsysPut(fsys);
  1347. return 0;
  1348. }
  1349. fsys->noauth = noauth;
  1350. fsys->noperm = noperm;
  1351. fsys->wstatallow = wstatallow;
  1352. vtUnlock(fsys->lock);
  1353. fsysPut(fsys);
  1354. if(strcmp(name, "main") == 0)
  1355. usersFileRead(nil);
  1356. return 1;
  1357. }
  1358. static int
  1359. fsysUnconfig(char* name, int argc, char* argv[])
  1360. {
  1361. Fsys *fsys, **fp;
  1362. char *usage = "usage: fsys name unconfig";
  1363. ARGBEGIN{
  1364. default:
  1365. return cliError(usage);
  1366. }ARGEND
  1367. if(argc)
  1368. return cliError(usage);
  1369. vtLock(sbox.lock);
  1370. fp = &sbox.head;
  1371. for(fsys = *fp; fsys != nil; fsys = fsys->next){
  1372. if(strcmp(fsys->name, name) == 0)
  1373. break;
  1374. fp = &fsys->next;
  1375. }
  1376. if(fsys == nil){
  1377. vtSetError(EFsysNotFound, name);
  1378. vtUnlock(sbox.lock);
  1379. return 0;
  1380. }
  1381. if(fsys->ref != 0 || fsys->fs != nil){
  1382. vtSetError(EFsysBusy, fsys->name);
  1383. vtUnlock(sbox.lock);
  1384. return 0;
  1385. }
  1386. *fp = fsys->next;
  1387. vtUnlock(sbox.lock);
  1388. if(fsys->session != nil){
  1389. vtClose(fsys->session);
  1390. vtFree(fsys->session);
  1391. }
  1392. if(fsys->venti != nil)
  1393. vtMemFree(fsys->venti);
  1394. if(fsys->dev != nil)
  1395. vtMemFree(fsys->dev);
  1396. if(fsys->name != nil)
  1397. vtMemFree(fsys->name);
  1398. vtMemFree(fsys);
  1399. return 1;
  1400. }
  1401. static int
  1402. fsysConfig(char* name, int argc, char* argv[])
  1403. {
  1404. Fsys *fsys;
  1405. char *usage = "usage: fsys name config dev";
  1406. ARGBEGIN{
  1407. default:
  1408. return cliError(usage);
  1409. }ARGEND
  1410. if(argc != 1)
  1411. return cliError(usage);
  1412. if((fsys = _fsysGet(argv[0])) != nil){
  1413. vtLock(fsys->lock);
  1414. if(fsys->fs != nil){
  1415. vtSetError(EFsysBusy, fsys->name);
  1416. vtUnlock(fsys->lock);
  1417. fsysPut(fsys);
  1418. return 0;
  1419. }
  1420. vtMemFree(fsys->dev);
  1421. fsys->dev = vtStrDup(argv[0]);
  1422. vtUnlock(fsys->lock);
  1423. }
  1424. else if((fsys = fsysAlloc(name, argv[0])) == nil)
  1425. return 0;
  1426. fsysPut(fsys);
  1427. return 1;
  1428. }
  1429. static struct {
  1430. char* cmd;
  1431. int (*f)(Fsys*, int, char**);
  1432. int (*f1)(char*, int, char**);
  1433. } fsyscmd[] = {
  1434. { "close", fsysClose, },
  1435. { "config", nil, fsysConfig, },
  1436. { "open", nil, fsysOpen, },
  1437. { "unconfig", nil, fsysUnconfig, },
  1438. { "venti", nil, fsysVenti, },
  1439. { "bfree", fsysBfree, },
  1440. { "block", fsysBlock, },
  1441. { "check", fsysCheck, },
  1442. { "clre", fsysClre, },
  1443. { "clri", fsysClri, },
  1444. { "clrp", fsysClrp, },
  1445. { "create", fsysCreate, },
  1446. { "df", fsysDf, },
  1447. { "epoch", fsysEpoch, },
  1448. { "halt", fsysHalt, },
  1449. { "label", fsysLabel, },
  1450. { "remove", fsysRemove, },
  1451. { "snap", fsysSnap, },
  1452. { "snaptime", fsysSnapTime, },
  1453. { "snapclean", fsysSnapClean, },
  1454. { "stat", fsysStat, },
  1455. { "sync", fsysSync, },
  1456. { "unhalt", fsysUnhalt, },
  1457. { "wstat", fsysWstat, },
  1458. { "vac", fsysVac, },
  1459. { nil, nil, },
  1460. };
  1461. static int
  1462. fsysXXX1(Fsys *fsys, int i, int argc, char* argv[])
  1463. {
  1464. int r;
  1465. vtLock(fsys->lock);
  1466. if(fsys->fs == nil){
  1467. vtUnlock(fsys->lock);
  1468. vtSetError(EFsysNotOpen, fsys->name);
  1469. return 0;
  1470. }
  1471. if(fsys->fs->halted
  1472. && fsyscmd[i].f != fsysUnhalt && fsyscmd[i].f != fsysCheck){
  1473. vtSetError("file system %s is halted", fsys->name);
  1474. vtUnlock(fsys->lock);
  1475. return 0;
  1476. }
  1477. r = (*fsyscmd[i].f)(fsys, argc, argv);
  1478. vtUnlock(fsys->lock);
  1479. return r;
  1480. }
  1481. static int
  1482. fsysXXX(char* name, int argc, char* argv[])
  1483. {
  1484. int i, r;
  1485. Fsys *fsys;
  1486. for(i = 0; fsyscmd[i].cmd != nil; i++){
  1487. if(strcmp(fsyscmd[i].cmd, argv[0]) == 0)
  1488. break;
  1489. }
  1490. if(fsyscmd[i].cmd == nil){
  1491. vtSetError("unknown command - '%s'", argv[0]);
  1492. return 0;
  1493. }
  1494. /* some commands want the name... */
  1495. if(fsyscmd[i].f1 != nil){
  1496. if(strcmp(name, FsysAll) == 0){
  1497. vtSetError("cannot use fsys %#q with %#q command", FsysAll, argv[0]);
  1498. return 0;
  1499. }
  1500. return (*fsyscmd[i].f1)(name, argc, argv);
  1501. }
  1502. /* ... but most commands want the Fsys */
  1503. if(strcmp(name, FsysAll) == 0){
  1504. r = 1;
  1505. vtRLock(sbox.lock);
  1506. for(fsys = sbox.head; fsys != nil; fsys = fsys->next){
  1507. fsys->ref++;
  1508. r = fsysXXX1(fsys, i, argc, argv) && r;
  1509. fsys->ref--;
  1510. }
  1511. vtRUnlock(sbox.lock);
  1512. }else{
  1513. if((fsys = _fsysGet(name)) == nil)
  1514. return 0;
  1515. r = fsysXXX1(fsys, i, argc, argv);
  1516. fsysPut(fsys);
  1517. }
  1518. return r;
  1519. }
  1520. static int
  1521. cmdFsysXXX(int argc, char* argv[])
  1522. {
  1523. char *name;
  1524. if((name = sbox.curfsys) == nil){
  1525. vtSetError(EFsysNoCurrent, argv[0]);
  1526. return 0;
  1527. }
  1528. return fsysXXX(name, argc, argv);
  1529. }
  1530. static int
  1531. cmdFsys(int argc, char* argv[])
  1532. {
  1533. Fsys *fsys;
  1534. char *usage = "usage: fsys [name ...]";
  1535. ARGBEGIN{
  1536. default:
  1537. return cliError(usage);
  1538. }ARGEND
  1539. if(argc == 0){
  1540. vtRLock(sbox.lock);
  1541. for(fsys = sbox.head; fsys != nil; fsys = fsys->next)
  1542. consPrint("\t%s\n", fsys->name);
  1543. vtRUnlock(sbox.lock);
  1544. return 1;
  1545. }
  1546. if(argc == 1){
  1547. fsys = nil;
  1548. if(strcmp(argv[0], FsysAll) != 0 && (fsys = fsysGet(argv[0])) == nil)
  1549. return 0;
  1550. sbox.curfsys = vtStrDup(argv[0]);
  1551. consPrompt(sbox.curfsys);
  1552. if(fsys)
  1553. fsysPut(fsys);
  1554. return 1;
  1555. }
  1556. return fsysXXX(argv[0], argc-1, argv+1);
  1557. }
  1558. int
  1559. fsysInit(void)
  1560. {
  1561. int i;
  1562. fmtinstall('H', encodefmt);
  1563. fmtinstall('V', scoreFmt);
  1564. fmtinstall('R', vtErrFmt);
  1565. fmtinstall('L', labelFmt);
  1566. sbox.lock = vtLockAlloc();
  1567. cliAddCmd("fsys", cmdFsys);
  1568. for(i = 0; fsyscmd[i].cmd != nil; i++){
  1569. if(fsyscmd[i].f != nil)
  1570. cliAddCmd(fsyscmd[i].cmd, cmdFsysXXX);
  1571. }
  1572. /* the venti cmd is special: the fs can be either open or closed */
  1573. cliAddCmd("venti", cmdFsysXXX);
  1574. cliAddCmd("printconfig", cmdPrintConfig);
  1575. return 1;
  1576. }