1 #include "stdinc.h"
2 #include "dat.h"
3 #include "fns.h"
4 
5 void
6 usage(void)
7 {
8 	fprint(2, "usage: fmtindex config\n");
9 	threadexitsall(0);
10 }
11 
12 void
13 threadmain(int argc, char *argv[])
14 {
15 	Config conf;
16 	Index *ix;
17 	ArenaPart *ap;
18 	Arena **arenas;
19 	AMap *amap;
20 	u64int addr;
21 	char *file;
22 	u32int i, j, n, narenas;
23 	int add;
24 
25 	fmtinstall('V', vtscorefmt);
26 	statsinit();
27 
28 	add = 0;
29 	ARGBEGIN{
30 	case 'a':
31 		add = 1;
32 		break;
33 	default:
34 		usage();
35 		break;
36 	}ARGEND
37 
38 	if(argc != 1)
39 		usage();
40 
41 	file = argv[0];
42 
43 	if(runconfig(file, &conf) < 0)
44 		sysfatal("can't initialize config %s: %r", file);
45 	if(conf.index == nil)
46 		sysfatal("no index specified in %s", file);
47 	if(nameok(conf.index) < 0)
48 		sysfatal("illegal index name %s", conf.index);
49 
50 	narenas = 0;
51 	for(i = 0; i < conf.naparts; i++){
52 		ap = conf.aparts[i];
53 		narenas += ap->narenas;
54 	}
55 
56 	if(add){
57 		ix = initindex(conf.index, conf.sects, conf.nsects);
58 		if(ix == nil)
59 			sysfatal("can't initialize index %s: %r", conf.index);
60 	}else{
61 		ix = newindex(conf.index, conf.sects, conf.nsects);
62 		if(ix == nil)
63 			sysfatal("can't create new index %s: %r", conf.index);
64 
65 		n = 0;
66 		for(i = 0; i < ix->nsects; i++)
67 			n += ix->sects[i]->blocks;
68 
69 		if(ix->div < 100)
70 			sysfatal("index divisor too coarse: use bigger block size");
71 
72 		fprint(2, "using %ud buckets of %ud; div=%d\n", ix->buckets, n, ix->div);
73 	}
74 	amap = MKNZ(AMap, narenas);
75 	arenas = MKNZ(Arena*, narenas);
76 
77 	addr = IndexBase;
78 	n = 0;
79 	for(i = 0; i < conf.naparts; i++){
80 		ap = conf.aparts[i];
81 		for(j = 0; j < ap->narenas; j++){
82 			if(n >= narenas)
83 				sysfatal("too few slots in index's arena set");
84 
85 			arenas[n] = ap->arenas[j];
86 			if(n < ix->narenas){
87 				if(arenas[n] != ix->arenas[n])
88 					sysfatal("mismatched arenas %s and %s at slot %d\n",
89 						arenas[n]->name, ix->arenas[n]->name, n);
90 				amap[n] = ix->amap[n];
91 				if(amap[n].start != addr)
92 					sysfatal("mis-located arena %s in index %s\n", arenas[n]->name, ix->name);
93 				addr = amap[n].stop;
94 			}else{
95 				amap[n].start = addr;
96 				addr += ap->arenas[j]->size;
97 				amap[n].stop = addr;
98 				namecp(amap[n].name, ap->arenas[j]->name);
99 				fprint(2, "add arena %s at [%lld,%lld)\n",
100 					amap[n].name, amap[n].start, amap[n].stop);
101 			}
102 
103 			n++;
104 		}
105 	}
106 	fprint(2, "configured index=%s with arenas=%d and storage=%lld\n",
107 		ix->name, n, addr - IndexBase);
108 
109 	ix->amap = amap;
110 	ix->arenas = arenas;
111 	ix->narenas = narenas;
112 
113 	if(wbindex(ix) < 0)
114 		fprint(2, "can't write back arena partition header for %s: %r\n", file);
115 
116 	threadexitsall(0);
117 }