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