1 #include "stdinc.h"
2 #include "dat.h"
3 #include "fns.h"
4
5 void
usage(void)6 usage(void)
7 {
8 fprint(2, "usage: fmtarenas [-Z] [-b blocksize] [-a arenasize] name file\n");
9 threadexitsall(0);
10 }
11
12 void
threadmain(int argc,char * argv[])13 threadmain(int argc, char *argv[])
14 {
15 int vers;
16 ArenaPart *ap;
17 Part *part;
18 Arena *arena;
19 u64int addr, limit, asize, apsize;
20 char *file, *name, aname[ANameSize];
21 int i, n, blocksize, tabsize, zero;
22
23 ventifmtinstall();
24 statsinit();
25
26 blocksize = 8 * 1024;
27 asize = 512 * 1024 *1024;
28 tabsize = 512 * 1024; /* BUG: should be determine from number of arenas */
29 zero = -1;
30 vers = ArenaVersion5;
31 ARGBEGIN{
32 case 'D':
33 settrace(EARGF(usage()));
34 break;
35 case 'a':
36 asize = unittoull(EARGF(usage()));
37 if(asize == TWID64)
38 usage();
39 break;
40 case 'b':
41 blocksize = unittoull(EARGF(usage()));
42 if(blocksize == ~0)
43 usage();
44 if(blocksize > MaxDiskBlock){
45 fprint(2, "block size too large, max %d\n", MaxDiskBlock);
46 threadexitsall("usage");
47 }
48 break;
49 case '4':
50 vers = ArenaVersion4;
51 break;
52 case 'Z':
53 zero = 0;
54 break;
55 default:
56 usage();
57 break;
58 }ARGEND
59
60 if(zero == -1){
61 if(vers == ArenaVersion4)
62 zero = 1;
63 else
64 zero = 0;
65 }
66
67 if(argc != 2)
68 usage();
69
70 name = argv[0];
71 file = argv[1];
72
73 if(nameok(name) < 0)
74 sysfatal("illegal name template %s", name);
75
76 part = initpart(file, ORDWR|ODIRECT);
77 if(part == nil)
78 sysfatal("can't open partition %s: %r", file);
79
80 if(zero)
81 zeropart(part, blocksize);
82
83 maxblocksize = blocksize;
84 initdcache(20*blocksize);
85
86 ap = newarenapart(part, blocksize, tabsize);
87 if(ap == nil)
88 sysfatal("can't initialize arena: %r");
89
90 apsize = ap->size - ap->arenabase;
91 n = apsize / asize;
92 if(apsize - (n * asize) >= MinArenaSize)
93 n++;
94
95 fprint(2, "fmtarenas %s: %,d arenas, %,lld bytes storage, %,d bytes for index map\n",
96 file, n, apsize, ap->tabsize);
97
98 ap->narenas = n;
99 ap->map = MKNZ(AMap, n);
100 ap->arenas = MKNZ(Arena*, n);
101
102 addr = ap->arenabase;
103 for(i = 0; i < n; i++){
104 limit = addr + asize;
105 if(limit >= ap->size || ap->size - limit < MinArenaSize){
106 limit = ap->size;
107 if(limit - addr < MinArenaSize)
108 sysfatal("bad arena set math: runt arena at %lld,%lld %lld", addr, limit, ap->size);
109 }
110
111 snprint(aname, ANameSize, "%s%d", name, i);
112
113 if(0) fprint(2, "adding arena %s at [%lld,%lld)\n", aname, addr, limit);
114
115 arena = newarena(part, vers, aname, addr, limit - addr, blocksize);
116 if(!arena)
117 fprint(2, "can't make new arena %s: %r", aname);
118 freearena(arena);
119
120 ap->map[i].start = addr;
121 ap->map[i].stop = limit;
122 namecp(ap->map[i].name, aname);
123
124 addr = limit;
125 }
126
127 if(wbarenapart(ap) < 0)
128 fprint(2, "can't write back arena partition header for %s: %r\n", file);
129
130 flushdcache();
131 threadexitsall(0);
132 }
133