9fsys.c 33 KB

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