1368c31abSDavid du Colombier #include "stdinc.h"
2368c31abSDavid du Colombier #include "dat.h"
3368c31abSDavid du Colombier #include "fns.h"
4368c31abSDavid du Colombier
5368c31abSDavid du Colombier void
usage(void)6368c31abSDavid du Colombier usage(void)
7368c31abSDavid du Colombier {
8368c31abSDavid du Colombier fprint(2, "usage: fmtindex [-a] config\n");
9368c31abSDavid du Colombier threadexitsall(0);
10368c31abSDavid du Colombier }
11368c31abSDavid du Colombier
12368c31abSDavid du Colombier void
threadmain(int argc,char * argv[])13368c31abSDavid du Colombier threadmain(int argc, char *argv[])
14368c31abSDavid du Colombier {
15368c31abSDavid du Colombier Config conf;
16368c31abSDavid du Colombier Index *ix;
17368c31abSDavid du Colombier ArenaPart *ap;
18368c31abSDavid du Colombier Arena **arenas;
19368c31abSDavid du Colombier AMap *amap;
20368c31abSDavid du Colombier u64int addr;
21368c31abSDavid du Colombier char *file;
22368c31abSDavid du Colombier u32int i, j, n, narenas;
23368c31abSDavid du Colombier int add;
24368c31abSDavid du Colombier
25368c31abSDavid du Colombier ventifmtinstall();
26368c31abSDavid du Colombier statsinit();
27368c31abSDavid du Colombier
28368c31abSDavid du Colombier add = 0;
29368c31abSDavid du Colombier ARGBEGIN{
30368c31abSDavid du Colombier case 'a':
31368c31abSDavid du Colombier add = 1;
32368c31abSDavid du Colombier break;
33368c31abSDavid du Colombier default:
34368c31abSDavid du Colombier usage();
35368c31abSDavid du Colombier break;
36368c31abSDavid du Colombier }ARGEND
37368c31abSDavid du Colombier
38368c31abSDavid du Colombier if(argc != 1)
39368c31abSDavid du Colombier usage();
40368c31abSDavid du Colombier
41368c31abSDavid du Colombier file = argv[0];
42368c31abSDavid du Colombier
43368c31abSDavid du Colombier if(runconfig(file, &conf) < 0)
44368c31abSDavid du Colombier sysfatal("can't initialize config %s: %r", file);
45368c31abSDavid du Colombier if(conf.index == nil)
46368c31abSDavid du Colombier sysfatal("no index specified in %s", file);
47368c31abSDavid du Colombier if(nameok(conf.index) < 0)
48368c31abSDavid du Colombier sysfatal("illegal index name %s", conf.index);
49368c31abSDavid du Colombier
50368c31abSDavid du Colombier narenas = 0;
51368c31abSDavid du Colombier for(i = 0; i < conf.naparts; i++){
52368c31abSDavid du Colombier ap = conf.aparts[i];
53368c31abSDavid du Colombier narenas += ap->narenas;
54368c31abSDavid du Colombier }
55368c31abSDavid du Colombier
56368c31abSDavid du Colombier if(add){
57368c31abSDavid du Colombier ix = initindex(conf.index, conf.sects, conf.nsects);
58368c31abSDavid du Colombier if(ix == nil)
59368c31abSDavid du Colombier sysfatal("can't initialize index %s: %r", conf.index);
60368c31abSDavid du Colombier }else{
61368c31abSDavid du Colombier ix = newindex(conf.index, conf.sects, conf.nsects);
62368c31abSDavid du Colombier if(ix == nil)
63368c31abSDavid du Colombier sysfatal("can't create new index %s: %r", conf.index);
64368c31abSDavid du Colombier
65368c31abSDavid du Colombier n = 0;
66368c31abSDavid du Colombier for(i = 0; i < ix->nsects; i++)
67368c31abSDavid du Colombier n += ix->sects[i]->blocks;
68368c31abSDavid du Colombier
69368c31abSDavid du Colombier if(0) fprint(2, "using %ud buckets of %ud; div=%d\n", ix->buckets, n, ix->div);
70368c31abSDavid du Colombier }
71368c31abSDavid du Colombier amap = MKNZ(AMap, narenas);
72368c31abSDavid du Colombier arenas = MKNZ(Arena*, narenas);
73368c31abSDavid du Colombier
74368c31abSDavid du Colombier addr = IndexBase;
75368c31abSDavid du Colombier n = 0;
76368c31abSDavid du Colombier for(i = 0; i < conf.naparts; i++){
77368c31abSDavid du Colombier ap = conf.aparts[i];
78368c31abSDavid du Colombier for(j = 0; j < ap->narenas; j++){
79368c31abSDavid du Colombier if(n >= narenas)
80368c31abSDavid du Colombier sysfatal("too few slots in index's arena set");
81368c31abSDavid du Colombier
82368c31abSDavid du Colombier arenas[n] = ap->arenas[j];
83368c31abSDavid du Colombier if(n < ix->narenas){
84368c31abSDavid du Colombier if(arenas[n] != ix->arenas[n])
85*14cc0f53SDavid du Colombier sysfatal("mismatched arenas %s and %s at slot %d",
86368c31abSDavid du Colombier arenas[n]->name, ix->arenas[n]->name, n);
87368c31abSDavid du Colombier amap[n] = ix->amap[n];
88368c31abSDavid du Colombier if(amap[n].start != addr)
89*14cc0f53SDavid du Colombier sysfatal("mis-located arena %s in index %s", arenas[n]->name, ix->name);
90368c31abSDavid du Colombier addr = amap[n].stop;
91368c31abSDavid du Colombier }else{
92368c31abSDavid du Colombier amap[n].start = addr;
93368c31abSDavid du Colombier addr += ap->arenas[j]->size;
94368c31abSDavid du Colombier amap[n].stop = addr;
95368c31abSDavid du Colombier namecp(amap[n].name, ap->arenas[j]->name);
96368c31abSDavid du Colombier if(0) fprint(2, "add arena %s at [%lld,%lld)\n",
97368c31abSDavid du Colombier amap[n].name, amap[n].start, amap[n].stop);
98368c31abSDavid du Colombier }
99368c31abSDavid du Colombier
100368c31abSDavid du Colombier n++;
101368c31abSDavid du Colombier }
102368c31abSDavid du Colombier }
103368c31abSDavid du Colombier if(0){
104368c31abSDavid du Colombier fprint(2, "configured index=%s with arenas=%d and storage=%lld\n",
105368c31abSDavid du Colombier ix->name, n, addr - IndexBase);
106368c31abSDavid du Colombier fprint(2, "\tbuckets=%d\n",
107368c31abSDavid du Colombier ix->buckets);
108368c31abSDavid du Colombier }
109368c31abSDavid du Colombier fprint(2, "fmtindex: %,d arenas, %,d index buckets, %,lld bytes storage\n",
110368c31abSDavid du Colombier n, ix->buckets, addr-IndexBase);
111368c31abSDavid du Colombier
112368c31abSDavid du Colombier ix->amap = amap;
113368c31abSDavid du Colombier ix->arenas = arenas;
114368c31abSDavid du Colombier ix->narenas = narenas;
115368c31abSDavid du Colombier
116368c31abSDavid du Colombier if(wbindex(ix) < 0)
117368c31abSDavid du Colombier fprint(2, "can't write back arena partition header for %s: %r\n", file);
118368c31abSDavid du Colombier
119368c31abSDavid du Colombier threadexitsall(0);
120368c31abSDavid du Colombier }
121