1 a0d146ed 2005-07-12 devnull /*
  2 a0d146ed 2005-07-12 devnull  * Index, mapping scores to log positions. 
  3 a0d146ed 2005-07-12 devnull  *
  4 a0d146ed 2005-07-12 devnull  * The index is made up of some number of index sections, each of
  5 a0d146ed 2005-07-12 devnull  * which is typically stored on a different disk.  The blocks in all the 
  6 a0d146ed 2005-07-12 devnull  * index sections are logically numbered, with each index section 
  7 a0d146ed 2005-07-12 devnull  * responsible for a range of blocks.  Blocks are typically 8kB.
  8 a0d146ed 2005-07-12 devnull  *
  9 a0d146ed 2005-07-12 devnull  * The N index blocks are treated as a giant hash table.  The top 32 bits
 10 a0d146ed 2005-07-12 devnull  * of score are used as the key for a lookup.  Each index block holds
 11 a0d146ed 2005-07-12 devnull  * one hash bucket, which is responsible for ceil(2^32 / N) of the key space.
 12 a0d146ed 2005-07-12 devnull  * 
 13 a0d146ed 2005-07-12 devnull  * The index is sized so that a particular bucket is extraordinarily 
 14 a0d146ed 2005-07-12 devnull  * unlikely to overflow: assuming compressed data blocks are 4kB 
 15 a0d146ed 2005-07-12 devnull  * on disk, and assuming each block has a 40 byte index entry,
 16 a0d146ed 2005-07-12 devnull  * the index data will be 1% of the total data.  Since scores are essentially
 17 a0d146ed 2005-07-12 devnull  * random, all buckets should be about the same fullness.
 18 a0d146ed 2005-07-12 devnull  * A factor of 5 gives us a wide comfort boundary to account for 
 19 a0d146ed 2005-07-12 devnull  * random variation.  So the index disk space should be 5% of the arena disk space.
 20 a0d146ed 2005-07-12 devnull  */
 21 a0d146ed 2005-07-12 devnull 
 22 a0d146ed 2005-07-12 devnull #include "stdinc.h"
 23 a0d146ed 2005-07-12 devnull #include "dat.h"
 24 a0d146ed 2005-07-12 devnull #include "fns.h"
 25 a0d146ed 2005-07-12 devnull 
 26 a0d146ed 2005-07-12 devnull static int	initindex1(Index*);
 27 a0d146ed 2005-07-12 devnull static ISect	*initisect1(ISect *is);
 28 a0d146ed 2005-07-12 devnull 
 29 a0d146ed 2005-07-12 devnull #define KEY(k,d)	((d) ? (k)>>(32-(d)) : 0)
 30 a0d146ed 2005-07-12 devnull 
 31 a0d146ed 2005-07-12 devnull static char IndexMagic[] = "venti index configuration";
 32 a0d146ed 2005-07-12 devnull 
 33 a0d146ed 2005-07-12 devnull Index*
 34 a0d146ed 2005-07-12 devnull initindex(char *name, ISect **sects, int n)
 35 a0d146ed 2005-07-12 devnull {
 36 a0d146ed 2005-07-12 devnull 	IFile f;
 37 a0d146ed 2005-07-12 devnull 	Index *ix;
 38 a0d146ed 2005-07-12 devnull 	ISect *is;
 39 a0d146ed 2005-07-12 devnull 	u32int last, blocksize, tabsize;
 40 a0d146ed 2005-07-12 devnull 	int i;
 41 a0d146ed 2005-07-12 devnull 
 42 a0d146ed 2005-07-12 devnull 	if(n <= 0){
 43 a0d146ed 2005-07-12 devnull fprint(2, "bad n\n");
 44 a0d146ed 2005-07-12 devnull 		seterr(EOk, "no index sections to initialize index");
 45 a0d146ed 2005-07-12 devnull 		return nil;
 46 a0d146ed 2005-07-12 devnull 	}
 47 a0d146ed 2005-07-12 devnull 	ix = MKZ(Index);
 48 a0d146ed 2005-07-12 devnull 	if(ix == nil){
 49 a0d146ed 2005-07-12 devnull fprint(2, "no mem\n");
 50 a0d146ed 2005-07-12 devnull 		seterr(EOk, "can't initialize index: out of memory");
 51 a0d146ed 2005-07-12 devnull 		freeindex(ix);
 52 a0d146ed 2005-07-12 devnull 		return nil;
 53 a0d146ed 2005-07-12 devnull 	}
 54 a0d146ed 2005-07-12 devnull 
 55 a0d146ed 2005-07-12 devnull 	tabsize = sects[0]->tabsize;
 56 a0d146ed 2005-07-12 devnull 	if(partifile(&f, sects[0]->part, sects[0]->tabbase, tabsize) < 0)
 57 a0d146ed 2005-07-12 devnull 		return nil;
 58 a0d146ed 2005-07-12 devnull 	if(parseindex(&f, ix) < 0){
 59 a0d146ed 2005-07-12 devnull 		freeifile(&f);
 60 a0d146ed 2005-07-12 devnull 		freeindex(ix);
 61 a0d146ed 2005-07-12 devnull 		return nil;
 62 a0d146ed 2005-07-12 devnull 	}
 63 a0d146ed 2005-07-12 devnull 	freeifile(&f);
 64 a0d146ed 2005-07-12 devnull 	if(namecmp(ix->name, name) != 0){
 65 a0d146ed 2005-07-12 devnull 		seterr(ECorrupt, "mismatched index name: found %s expected %s", ix->name, name);
 66 a0d146ed 2005-07-12 devnull 		return nil;
 67 a0d146ed 2005-07-12 devnull 	}
 68 a0d146ed 2005-07-12 devnull 	if(ix->nsects != n){
 69 a0d146ed 2005-07-12 devnull 		seterr(ECorrupt, "mismatched number index sections: found %d expected %d", n, ix->nsects);
 70 a0d146ed 2005-07-12 devnull 		freeindex(ix);
 71 a0d146ed 2005-07-12 devnull 		return nil;
 72 a0d146ed 2005-07-12 devnull 	}
 73 a0d146ed 2005-07-12 devnull 	ix->sects = sects;
 74 a0d146ed 2005-07-12 devnull 	last = 0;
 75 a0d146ed 2005-07-12 devnull 	blocksize = ix->blocksize;
 76 a0d146ed 2005-07-12 devnull 	for(i = 0; i < ix->nsects; i++){
 77 a0d146ed 2005-07-12 devnull 		is = sects[i];
 78 3b06b757 2008-12-23 rsc 		if(namecmp(is->index, ix->name) != 0) {
 79 3b06b757 2008-12-23 rsc 			seterr(ECorrupt, "%s: index name is %s, not %s",
 80 3b06b757 2008-12-23 rsc 				sects[i]->part->name, is->index, ix->name);
 81 3b06b757 2008-12-23 rsc 		bad:
 82 a0d146ed 2005-07-12 devnull 			freeindex(ix);
 83 a0d146ed 2005-07-12 devnull 			return nil;
 84 a0d146ed 2005-07-12 devnull 		}
 85 3b06b757 2008-12-23 rsc 		if(is->blocksize != blocksize) {
 86 3b06b757 2008-12-23 rsc 			seterr(ECorrupt, "%s: blocksize is %d, not %d",
 87 3b06b757 2008-12-23 rsc 				sects[i]->part->name, (int)is->blocksize, (int)blocksize);
 88 3b06b757 2008-12-23 rsc 			goto bad;
 89 3b06b757 2008-12-23 rsc 		}
 90 3b06b757 2008-12-23 rsc 		if(is->tabsize != tabsize) {
 91 3b06b757 2008-12-23 rsc 			seterr(ECorrupt, "%s: tabsize is %d, not %d",
 92 3b06b757 2008-12-23 rsc 				sects[i]->part->name, (int)is->tabsize, (int)tabsize);
 93 3b06b757 2008-12-23 rsc 			goto bad;
 94 3b06b757 2008-12-23 rsc 		}
 95 3b06b757 2008-12-23 rsc 		if(namecmp(is->name, ix->smap[i].name) != 0) {
 96 3b06b757 2008-12-23 rsc 			seterr(ECorrupt, "%s: name is %s, not %s",
 97 3b06b757 2008-12-23 rsc 				sects[i]->part->name, is->name, ix->smap[i].name);
 98 3b06b757 2008-12-23 rsc 			goto bad;
 99 3b06b757 2008-12-23 rsc 		}
100 3b06b757 2008-12-23 rsc 		if(is->start != ix->smap[i].start || is->stop != ix->smap[i].stop) {
101 3b06b757 2008-12-23 rsc 			seterr(ECorrupt, "%s: range is %lld,%lld, not %lld,%lld",
102 3b06b757 2008-12-23 rsc 				sects[i]->part->name, is->start, is->stop,
103 3b06b757 2008-12-23 rsc 				ix->smap[i].start, ix->smap[i].stop);
104 3b06b757 2008-12-23 rsc 			goto bad;
105 3b06b757 2008-12-23 rsc 		}
106 3b06b757 2008-12-23 rsc 		if(is->start > is->stop) {
107 3b06b757 2008-12-23 rsc 			seterr(ECorrupt, "%s: invalid range %lld,%lld",
108 3b06b757 2008-12-23 rsc 				sects[i]->part->name, is->start, is->stop);
109 3b06b757 2008-12-23 rsc 			goto bad;
110 3b06b757 2008-12-23 rsc 		}
111 3b06b757 2008-12-23 rsc 		if(is->start != last || is->start > is->stop) {
112 3b06b757 2008-12-23 rsc 			seterr(ECorrupt, "%s: range %lld-%lld, but last section ended at %lld",
113 3b06b757 2008-12-23 rsc 				sects[i]->part->name, is->start, is->stop, last);
114 3b06b757 2008-12-23 rsc 			goto bad;
115 3b06b757 2008-12-23 rsc 		}
116 a0d146ed 2005-07-12 devnull 		last = is->stop;
117 a0d146ed 2005-07-12 devnull 	}
118 a0d146ed 2005-07-12 devnull 	ix->tabsize = tabsize;
119 a0d146ed 2005-07-12 devnull 	ix->buckets = last;
120 a0d146ed 2005-07-12 devnull 
121 a0d146ed 2005-07-12 devnull 	if(initindex1(ix) < 0){
122 a0d146ed 2005-07-12 devnull 		freeindex(ix);
123 a0d146ed 2005-07-12 devnull 		return nil;
124 a0d146ed 2005-07-12 devnull 	}
125 a0d146ed 2005-07-12 devnull 
126 a0d146ed 2005-07-12 devnull 	ix->arenas = MKNZ(Arena*, ix->narenas);
127 a0d146ed 2005-07-12 devnull 	if(maparenas(ix->amap, ix->arenas, ix->narenas, ix->name) < 0){
128 a0d146ed 2005-07-12 devnull 		freeindex(ix);
129 a0d146ed 2005-07-12 devnull 		return nil;
130 a0d146ed 2005-07-12 devnull 	}
131 a0d146ed 2005-07-12 devnull 
132 a0d146ed 2005-07-12 devnull 	return ix;
133 a0d146ed 2005-07-12 devnull }
134 a0d146ed 2005-07-12 devnull 
135 a0d146ed 2005-07-12 devnull static int
136 a0d146ed 2005-07-12 devnull initindex1(Index *ix)
137 a0d146ed 2005-07-12 devnull {
138 a0d146ed 2005-07-12 devnull 	u32int buckets;
139 a0d146ed 2005-07-12 devnull 
140 a0d146ed 2005-07-12 devnull 	ix->div = (((u64int)1 << 32) + ix->buckets - 1) / ix->buckets;
141 a0d146ed 2005-07-12 devnull 	buckets = (((u64int)1 << 32) - 1) / ix->div + 1;
142 a0d146ed 2005-07-12 devnull 	if(buckets != ix->buckets){
143 a0d146ed 2005-07-12 devnull 		seterr(ECorrupt, "inconsistent math for divisor and buckets in %s", ix->name);
144 a0d146ed 2005-07-12 devnull 		return -1;
145 a0d146ed 2005-07-12 devnull 	}
146 a0d146ed 2005-07-12 devnull 
147 a0d146ed 2005-07-12 devnull 	return 0;
148 a0d146ed 2005-07-12 devnull }
149 a0d146ed 2005-07-12 devnull 
150 a0d146ed 2005-07-12 devnull int
151 a0d146ed 2005-07-12 devnull wbindex(Index *ix)
152 a0d146ed 2005-07-12 devnull {
153 a0d146ed 2005-07-12 devnull 	Fmt f;
154 a0d146ed 2005-07-12 devnull 	ZBlock *b;
155 a0d146ed 2005-07-12 devnull 	int i;
156 a0d146ed 2005-07-12 devnull 
157 a0d146ed 2005-07-12 devnull 	if(ix->nsects == 0){
158 a0d146ed 2005-07-12 devnull 		seterr(EOk, "no sections in index %s", ix->name);
159 a0d146ed 2005-07-12 devnull 		return -1;
160 a0d146ed 2005-07-12 devnull 	}
161 a0d146ed 2005-07-12 devnull 	b = alloczblock(ix->tabsize, 1, ix->blocksize);
162 a0d146ed 2005-07-12 devnull 	if(b == nil){
163 a0d146ed 2005-07-12 devnull 		seterr(EOk, "can't write index configuration: out of memory");
164 a0d146ed 2005-07-12 devnull 		return -1;
165 a0d146ed 2005-07-12 devnull 	}
166 a0d146ed 2005-07-12 devnull 	fmtzbinit(&f, b);
167 a0d146ed 2005-07-12 devnull 	if(outputindex(&f, ix) < 0){
168 a0d146ed 2005-07-12 devnull 		seterr(EOk, "can't make index configuration: table storage too small %d", ix->tabsize);
169 a0d146ed 2005-07-12 devnull 		freezblock(b);
170 a0d146ed 2005-07-12 devnull 		return -1;
171 a0d146ed 2005-07-12 devnull 	}
172 a0d146ed 2005-07-12 devnull 	for(i = 0; i < ix->nsects; i++){
173 e46cacb0 2007-04-27 devnull 		if(writepart(ix->sects[i]->part, ix->sects[i]->tabbase, b->data, ix->tabsize) < 0
174 e46cacb0 2007-04-27 devnull 		|| flushpart(ix->sects[i]->part) < 0){
175 a0d146ed 2005-07-12 devnull 			seterr(EOk, "can't write index: %r");
176 a0d146ed 2005-07-12 devnull 			freezblock(b);
177 a0d146ed 2005-07-12 devnull 			return -1;
178 a0d146ed 2005-07-12 devnull 		}
179 a0d146ed 2005-07-12 devnull 	}
180 a0d146ed 2005-07-12 devnull 	freezblock(b);
181 a0d146ed 2005-07-12 devnull 
182 a0d146ed 2005-07-12 devnull 	for(i = 0; i < ix->nsects; i++)
183 a0d146ed 2005-07-12 devnull 		if(wbisect(ix->sects[i]) < 0)
184 a0d146ed 2005-07-12 devnull 			return -1;
185 a0d146ed 2005-07-12 devnull 
186 a0d146ed 2005-07-12 devnull 	return 0;
187 a0d146ed 2005-07-12 devnull }
188 a0d146ed 2005-07-12 devnull 
189 a0d146ed 2005-07-12 devnull /*
190 a0d146ed 2005-07-12 devnull  * index: IndexMagic '\n' version '\n' name '\n' blocksize '\n' [V2: bitblocks '\n'] sections arenas
191 a0d146ed 2005-07-12 devnull  * version, blocksize: u32int
192 a0d146ed 2005-07-12 devnull  * name: max. ANameSize string
193 a0d146ed 2005-07-12 devnull  * sections, arenas: AMap
194 a0d146ed 2005-07-12 devnull  */
195 a0d146ed 2005-07-12 devnull int
196 a0d146ed 2005-07-12 devnull outputindex(Fmt *f, Index *ix)
197 a0d146ed 2005-07-12 devnull {
198 a0d146ed 2005-07-12 devnull 	if(fmtprint(f, "%s\n%ud\n%s\n%ud\n", IndexMagic, ix->version, ix->name, ix->blocksize) < 0
199 a0d146ed 2005-07-12 devnull 	|| outputamap(f, ix->smap, ix->nsects) < 0
200 a0d146ed 2005-07-12 devnull 	|| outputamap(f, ix->amap, ix->narenas) < 0)
201 a0d146ed 2005-07-12 devnull 		return -1;
202 a0d146ed 2005-07-12 devnull 	return 0;
203 a0d146ed 2005-07-12 devnull }
204 a0d146ed 2005-07-12 devnull 
205 a0d146ed 2005-07-12 devnull int
206 a0d146ed 2005-07-12 devnull parseindex(IFile *f, Index *ix)
207 a0d146ed 2005-07-12 devnull {
208 a0d146ed 2005-07-12 devnull 	AMapN amn;
209 a0d146ed 2005-07-12 devnull 	u32int v;
210 a0d146ed 2005-07-12 devnull 	char *s;
211 a0d146ed 2005-07-12 devnull 
212 a0d146ed 2005-07-12 devnull 	/*
213 a0d146ed 2005-07-12 devnull 	 * magic
214 a0d146ed 2005-07-12 devnull 	 */
215 a0d146ed 2005-07-12 devnull 	s = ifileline(f);
216 a0d146ed 2005-07-12 devnull 	if(s == nil || strcmp(s, IndexMagic) != 0){
217 a0d146ed 2005-07-12 devnull 		seterr(ECorrupt, "bad index magic for %s", f->name);
218 a0d146ed 2005-07-12 devnull 		return -1;
219 a0d146ed 2005-07-12 devnull 	}
220 a0d146ed 2005-07-12 devnull 
221 a0d146ed 2005-07-12 devnull 	/*
222 a0d146ed 2005-07-12 devnull 	 * version
223 a0d146ed 2005-07-12 devnull 	 */
224 a0d146ed 2005-07-12 devnull 	if(ifileu32int(f, &v) < 0){
225 a0d146ed 2005-07-12 devnull 		seterr(ECorrupt, "syntax error: bad version number in %s", f->name);
226 a0d146ed 2005-07-12 devnull 		return -1;
227 a0d146ed 2005-07-12 devnull 	}
228 a0d146ed 2005-07-12 devnull 	ix->version = v;
229 a0d146ed 2005-07-12 devnull 	if(ix->version != IndexVersion){
230 a0d146ed 2005-07-12 devnull 		seterr(ECorrupt, "bad version number in %s", f->name);
231 a0d146ed 2005-07-12 devnull 		return -1;
232 a0d146ed 2005-07-12 devnull 	}
233 a0d146ed 2005-07-12 devnull 
234 a0d146ed 2005-07-12 devnull 	/*
235 a0d146ed 2005-07-12 devnull 	 * name
236 a0d146ed 2005-07-12 devnull 	 */
237 a0d146ed 2005-07-12 devnull 	if(ifilename(f, ix->name) < 0){
238 a0d146ed 2005-07-12 devnull 		seterr(ECorrupt, "syntax error: bad index name in %s", f->name);
239 a0d146ed 2005-07-12 devnull 		return -1;
240 a0d146ed 2005-07-12 devnull 	}
241 a0d146ed 2005-07-12 devnull 
242 a0d146ed 2005-07-12 devnull 	/*
243 a0d146ed 2005-07-12 devnull 	 * block size
244 a0d146ed 2005-07-12 devnull 	 */
245 a0d146ed 2005-07-12 devnull 	if(ifileu32int(f, &v) < 0){
246 a0d146ed 2005-07-12 devnull 		seterr(ECorrupt, "syntax error: bad block size number in %s", f->name);
247 a0d146ed 2005-07-12 devnull 		return -1;
248 a0d146ed 2005-07-12 devnull 	}
249 a0d146ed 2005-07-12 devnull 	ix->blocksize = v;
250 a0d146ed 2005-07-12 devnull 
251 a0d146ed 2005-07-12 devnull 	if(parseamap(f, &amn) < 0)
252 a0d146ed 2005-07-12 devnull 		return -1;
253 a0d146ed 2005-07-12 devnull 	ix->nsects = amn.n;
254 a0d146ed 2005-07-12 devnull 	ix->smap = amn.map;
255 a0d146ed 2005-07-12 devnull 
256 a0d146ed 2005-07-12 devnull 	if(parseamap(f, &amn) < 0)
257 a0d146ed 2005-07-12 devnull 		return -1;
258 a0d146ed 2005-07-12 devnull 	ix->narenas = amn.n;
259 a0d146ed 2005-07-12 devnull 	ix->amap = amn.map;
260 a0d146ed 2005-07-12 devnull 
261 a0d146ed 2005-07-12 devnull 	return 0;
262 a0d146ed 2005-07-12 devnull }
263 a0d146ed 2005-07-12 devnull 
264 a0d146ed 2005-07-12 devnull /*
265 a0d146ed 2005-07-12 devnull  * initialize an entirely new index
266 a0d146ed 2005-07-12 devnull  */
267 a0d146ed 2005-07-12 devnull Index *
268 a0d146ed 2005-07-12 devnull newindex(char *name, ISect **sects, int n)
269 a0d146ed 2005-07-12 devnull {
270 a0d146ed 2005-07-12 devnull 	Index *ix;
271 a0d146ed 2005-07-12 devnull 	AMap *smap;
272 a0d146ed 2005-07-12 devnull 	u64int nb;
273 07029cdb 2007-04-23 devnull 	u32int div, ub, xb, start, stop, blocksize, tabsize;
274 a0d146ed 2005-07-12 devnull 	int i, j;
275 a0d146ed 2005-07-12 devnull 
276 a0d146ed 2005-07-12 devnull 	if(n < 1){
277 a0d146ed 2005-07-12 devnull 		seterr(EOk, "creating index with no index sections");
278 a0d146ed 2005-07-12 devnull 		return nil;
279 a0d146ed 2005-07-12 devnull 	}
280 a0d146ed 2005-07-12 devnull 
281 a0d146ed 2005-07-12 devnull 	/*
282 a0d146ed 2005-07-12 devnull 	 * compute the total buckets available in the index,
283 a0d146ed 2005-07-12 devnull 	 * and the total buckets which are used.
284 a0d146ed 2005-07-12 devnull 	 */
285 a0d146ed 2005-07-12 devnull 	nb = 0;
286 a0d146ed 2005-07-12 devnull 	blocksize = sects[0]->blocksize;
287 a0d146ed 2005-07-12 devnull 	tabsize = sects[0]->tabsize;
288 a0d146ed 2005-07-12 devnull 	for(i = 0; i < n; i++){
289 7e452401 2007-04-27 devnull 		/*
290 7e452401 2007-04-27 devnull 		 * allow index, start, and stop to be set if index is correct
291 7e452401 2007-04-27 devnull 		 * and start and stop are what we would have picked.
292 7e452401 2007-04-27 devnull 		 * this allows calling fmtindex to reformat the index after
293 7e452401 2007-04-27 devnull 		 * replacing a bad index section with a freshly formatted one.
294 7e452401 2007-04-27 devnull 		 * start and stop are checked below.
295 7e452401 2007-04-27 devnull 		 */
296 7e452401 2007-04-27 devnull 		if(sects[i]->index[0] != '\0' && strcmp(sects[i]->index, name) != 0){
297 a0d146ed 2005-07-12 devnull 			seterr(EOk, "creating new index using non-empty section %s", sects[i]->name);
298 a0d146ed 2005-07-12 devnull 			return nil;
299 a0d146ed 2005-07-12 devnull 		}
300 a0d146ed 2005-07-12 devnull 		if(blocksize != sects[i]->blocksize){
301 3b06b757 2008-12-23 rsc 			seterr(EOk, "%s has block size %d, but %s has %d",
302 3b06b757 2008-12-23 rsc 				sects[0]->part->name, (int)blocksize,
303 3b06b757 2008-12-23 rsc 				sects[i]->part->name, (int)sects[i]->blocksize);
304 a0d146ed 2005-07-12 devnull 			return nil;
305 a0d146ed 2005-07-12 devnull 		}
306 a0d146ed 2005-07-12 devnull 		if(tabsize != sects[i]->tabsize){
307 3b06b757 2008-12-23 rsc 			seterr(EOk, "%s has table size %d, but %s has %d",
308 3b06b757 2008-12-23 rsc 				sects[0]->part->name, (int)tabsize,
309 3b06b757 2008-12-23 rsc 				sects[i]->part->name, (int)sects[i]->tabsize);
310 a0d146ed 2005-07-12 devnull 			return nil;
311 a0d146ed 2005-07-12 devnull 		}
312 a0d146ed 2005-07-12 devnull 		nb += sects[i]->blocks;
313 a0d146ed 2005-07-12 devnull 	}
314 a0d146ed 2005-07-12 devnull 
315 a0d146ed 2005-07-12 devnull 	/*
316 a0d146ed 2005-07-12 devnull 	 * check for duplicate names
317 a0d146ed 2005-07-12 devnull 	 */
318 a0d146ed 2005-07-12 devnull 	for(i = 0; i < n; i++){
319 a0d146ed 2005-07-12 devnull 		for(j = i + 1; j < n; j++){
320 a0d146ed 2005-07-12 devnull 			if(namecmp(sects[i]->name, sects[j]->name) == 0){
321 3b06b757 2008-12-23 rsc 				seterr(EOk, "%s and %s both have section name %s",
322 3b06b757 2008-12-23 rsc 					sects[i]->part->name,
323 3b06b757 2008-12-23 rsc 					sects[j]->part->name,
324 3b06b757 2008-12-23 rsc 					sects[i]->name);
325 a0d146ed 2005-07-12 devnull 				return nil;
326 a0d146ed 2005-07-12 devnull 			}
327 a0d146ed 2005-07-12 devnull 		}
328 a0d146ed 2005-07-12 devnull 	}
329 a0d146ed 2005-07-12 devnull 
330 a0d146ed 2005-07-12 devnull 	if(nb >= ((u64int)1 << 32)){
331 a0d146ed 2005-07-12 devnull 		seterr(EBug, "index too large");
332 a0d146ed 2005-07-12 devnull 		return nil;
333 a0d146ed 2005-07-12 devnull 	}
334 a0d146ed 2005-07-12 devnull 
335 a0d146ed 2005-07-12 devnull 	div = (((u64int)1 << 32) + nb - 1) / nb;
336 a0d146ed 2005-07-12 devnull 	ub = (((u64int)1 << 32) - 1) / div + 1;
337 a0d146ed 2005-07-12 devnull 	if(div < 100){
338 cd87898f 2008-07-04 rsc 		seterr(EBug, "index divisor too coarse [%lld buckets]", nb);
339 a0d146ed 2005-07-12 devnull 		return nil;
340 a0d146ed 2005-07-12 devnull 	}
341 a0d146ed 2005-07-12 devnull 	if(ub > nb){
342 a0d146ed 2005-07-12 devnull 		seterr(EBug, "index initialization math wrong");
343 a0d146ed 2005-07-12 devnull 		return nil;
344 a0d146ed 2005-07-12 devnull 	}
345 a0d146ed 2005-07-12 devnull 	xb = nb - ub;
346 a0d146ed 2005-07-12 devnull 
347 a0d146ed 2005-07-12 devnull 	/*
348 a0d146ed 2005-07-12 devnull 	 * initialize each of the index sections
349 a0d146ed 2005-07-12 devnull 	 * and the section map table
350 a0d146ed 2005-07-12 devnull 	 */
351 a0d146ed 2005-07-12 devnull 	smap = MKNZ(AMap, n);
352 a0d146ed 2005-07-12 devnull 	if(smap == nil){
353 a0d146ed 2005-07-12 devnull 		seterr(EOk, "can't create new index: out of memory");
354 a0d146ed 2005-07-12 devnull 		return nil;
355 a0d146ed 2005-07-12 devnull 	}
356 a0d146ed 2005-07-12 devnull 	start = 0;
357 a0d146ed 2005-07-12 devnull 	for(i = 0; i < n; i++){
358 a0d146ed 2005-07-12 devnull 		stop = start + sects[i]->blocks - xb / n;
359 a0d146ed 2005-07-12 devnull 		if(i == n - 1)
360 a0d146ed 2005-07-12 devnull 			stop = ub;
361 7e452401 2007-04-27 devnull 
362 7e452401 2007-04-27 devnull 		if(sects[i]->start != 0 || sects[i]->stop != 0)
363 7e452401 2007-04-27 devnull 		if(sects[i]->start != start || sects[i]->stop != stop){
364 7e452401 2007-04-27 devnull 			seterr(EOk, "creating new index using non-empty section %s", sects[i]->name);
365 7e452401 2007-04-27 devnull 			return nil;
366 7e452401 2007-04-27 devnull 		}
367 7e452401 2007-04-27 devnull 
368 a0d146ed 2005-07-12 devnull 		sects[i]->start = start;
369 a0d146ed 2005-07-12 devnull 		sects[i]->stop = stop;
370 a0d146ed 2005-07-12 devnull 		namecp(sects[i]->index, name);
371 a0d146ed 2005-07-12 devnull 
372 a0d146ed 2005-07-12 devnull 		smap[i].start = start;
373 a0d146ed 2005-07-12 devnull 		smap[i].stop = stop;
374 a0d146ed 2005-07-12 devnull 		namecp(smap[i].name, sects[i]->name);
375 a0d146ed 2005-07-12 devnull 		start = stop;
376 a0d146ed 2005-07-12 devnull 	}
377 a0d146ed 2005-07-12 devnull 
378 a0d146ed 2005-07-12 devnull 	/*
379 a0d146ed 2005-07-12 devnull 	 * initialize the index itself
380 a0d146ed 2005-07-12 devnull 	 */
381 a0d146ed 2005-07-12 devnull 	ix = MKZ(Index);
382 a0d146ed 2005-07-12 devnull 	if(ix == nil){
383 a0d146ed 2005-07-12 devnull 		seterr(EOk, "can't create new index: out of memory");
384 a0d146ed 2005-07-12 devnull 		free(smap);
385 a0d146ed 2005-07-12 devnull 		return nil;
386 a0d146ed 2005-07-12 devnull 	}
387 a0d146ed 2005-07-12 devnull 	ix->version = IndexVersion;
388 a0d146ed 2005-07-12 devnull 	namecp(ix->name, name);
389 a0d146ed 2005-07-12 devnull 	ix->sects = sects;
390 a0d146ed 2005-07-12 devnull 	ix->smap = smap;
391 a0d146ed 2005-07-12 devnull 	ix->nsects = n;
392 a0d146ed 2005-07-12 devnull 	ix->blocksize = blocksize;
393 a0d146ed 2005-07-12 devnull 	ix->buckets = ub;
394 a0d146ed 2005-07-12 devnull 	ix->tabsize = tabsize;
395 a0d146ed 2005-07-12 devnull 	ix->div = div;
396 a0d146ed 2005-07-12 devnull 
397 a0d146ed 2005-07-12 devnull 	if(initindex1(ix) < 0){
398 a0d146ed 2005-07-12 devnull 		free(smap);
399 a0d146ed 2005-07-12 devnull 		return nil;
400 a0d146ed 2005-07-12 devnull 	}
401 a0d146ed 2005-07-12 devnull 
402 a0d146ed 2005-07-12 devnull 	return ix;
403 a0d146ed 2005-07-12 devnull }
404 a0d146ed 2005-07-12 devnull 
405 a0d146ed 2005-07-12 devnull ISect*
406 a0d146ed 2005-07-12 devnull initisect(Part *part)
407 a0d146ed 2005-07-12 devnull {
408 a0d146ed 2005-07-12 devnull 	ISect *is;
409 a0d146ed 2005-07-12 devnull 	ZBlock *b;
410 a0d146ed 2005-07-12 devnull 	int ok;
411 a0d146ed 2005-07-12 devnull 
412 a0d146ed 2005-07-12 devnull 	b = alloczblock(HeadSize, 0, 0);
413 a0d146ed 2005-07-12 devnull 	if(b == nil || readpart(part, PartBlank, b->data, HeadSize) < 0){
414 a0d146ed 2005-07-12 devnull 		seterr(EAdmin, "can't read index section header: %r");
415 a0d146ed 2005-07-12 devnull 		return nil;
416 a0d146ed 2005-07-12 devnull 	}
417 a0d146ed 2005-07-12 devnull 
418 a0d146ed 2005-07-12 devnull 	is = MKZ(ISect);
419 a0d146ed 2005-07-12 devnull 	if(is == nil){
420 a0d146ed 2005-07-12 devnull 		freezblock(b);
421 a0d146ed 2005-07-12 devnull 		return nil;
422 a0d146ed 2005-07-12 devnull 	}
423 a0d146ed 2005-07-12 devnull 	is->part = part;
424 a0d146ed 2005-07-12 devnull 	ok = unpackisect(is, b->data);
425 a0d146ed 2005-07-12 devnull 	freezblock(b);
426 a0d146ed 2005-07-12 devnull 	if(ok < 0){
427 a0d146ed 2005-07-12 devnull 		seterr(ECorrupt, "corrupted index section header: %r");
428 a0d146ed 2005-07-12 devnull 		freeisect(is);
429 a0d146ed 2005-07-12 devnull 		return nil;
430 a0d146ed 2005-07-12 devnull 	}
431 a0d146ed 2005-07-12 devnull 
432 a0d146ed 2005-07-12 devnull 	if(is->version != ISectVersion1 && is->version != ISectVersion2){
433 a0d146ed 2005-07-12 devnull 		seterr(EAdmin, "unknown index section version %d", is->version);
434 a0d146ed 2005-07-12 devnull 		freeisect(is);
435 a0d146ed 2005-07-12 devnull 		return nil;
436 a0d146ed 2005-07-12 devnull 	}
437 a0d146ed 2005-07-12 devnull 
438 a0d146ed 2005-07-12 devnull 	return initisect1(is);
439 a0d146ed 2005-07-12 devnull }
440 a0d146ed 2005-07-12 devnull 
441 a0d146ed 2005-07-12 devnull ISect*
442 a0d146ed 2005-07-12 devnull newisect(Part *part, u32int vers, char *name, u32int blocksize, u32int tabsize)
443 a0d146ed 2005-07-12 devnull {
444 a0d146ed 2005-07-12 devnull 	ISect *is;
445 a0d146ed 2005-07-12 devnull 	u32int tabbase;
446 a0d146ed 2005-07-12 devnull 
447 a0d146ed 2005-07-12 devnull 	is = MKZ(ISect);
448 a0d146ed 2005-07-12 devnull 	if(is == nil)
449 a0d146ed 2005-07-12 devnull 		return nil;
450 a0d146ed 2005-07-12 devnull 
451 a0d146ed 2005-07-12 devnull 	namecp(is->name, name);
452 a0d146ed 2005-07-12 devnull 	is->version = vers;
453 a0d146ed 2005-07-12 devnull 	is->part = part;
454 a0d146ed 2005-07-12 devnull 	is->blocksize = blocksize;
455 a0d146ed 2005-07-12 devnull 	is->start = 0;
456 a0d146ed 2005-07-12 devnull 	is->stop = 0;
457 a0d146ed 2005-07-12 devnull 	tabbase = (PartBlank + HeadSize + blocksize - 1) & ~(blocksize - 1);
458 a0d146ed 2005-07-12 devnull 	is->blockbase = (tabbase + tabsize + blocksize - 1) & ~(blocksize - 1);
459 a0d146ed 2005-07-12 devnull 	is->blocks = is->part->size / blocksize - is->blockbase / blocksize;
460 a0d146ed 2005-07-12 devnull 	is->bucketmagic = 0;
461 a0d146ed 2005-07-12 devnull 	if(is->version == ISectVersion2){
462 a0d146ed 2005-07-12 devnull 		do{
463 a0d146ed 2005-07-12 devnull 			is->bucketmagic = fastrand();
464 a0d146ed 2005-07-12 devnull 		}while(is->bucketmagic==0);
465 a0d146ed 2005-07-12 devnull 	}
466 a0d146ed 2005-07-12 devnull 	is = initisect1(is);
467 a0d146ed 2005-07-12 devnull 	if(is == nil)
468 a0d146ed 2005-07-12 devnull 		return nil;
469 a0d146ed 2005-07-12 devnull 
470 a0d146ed 2005-07-12 devnull 	return is;
471 a0d146ed 2005-07-12 devnull }
472 a0d146ed 2005-07-12 devnull 
473 a0d146ed 2005-07-12 devnull /*
474 a0d146ed 2005-07-12 devnull  * initialize the computed parameters for an index
475 a0d146ed 2005-07-12 devnull  */
476 a0d146ed 2005-07-12 devnull static ISect*
477 a0d146ed 2005-07-12 devnull initisect1(ISect *is)
478 a0d146ed 2005-07-12 devnull {
479 a0d146ed 2005-07-12 devnull 	u64int v;
480 a0d146ed 2005-07-12 devnull 
481 a0d146ed 2005-07-12 devnull 	is->buckmax = (is->blocksize - IBucketSize) / IEntrySize;
482 a0d146ed 2005-07-12 devnull 	is->blocklog = u64log2(is->blocksize);
483 a0d146ed 2005-07-12 devnull 	if(is->blocksize != (1 << is->blocklog)){
484 a0d146ed 2005-07-12 devnull 		seterr(ECorrupt, "illegal non-power-of-2 bucket size %d\n", is->blocksize);
485 a0d146ed 2005-07-12 devnull 		freeisect(is);
486 a0d146ed 2005-07-12 devnull 		return nil;
487 a0d146ed 2005-07-12 devnull 	}
488 a0d146ed 2005-07-12 devnull 	partblocksize(is->part, is->blocksize);
489 a0d146ed 2005-07-12 devnull 	is->tabbase = (PartBlank + HeadSize + is->blocksize - 1) & ~(is->blocksize - 1);
490 a0d146ed 2005-07-12 devnull 	if(is->tabbase >= is->blockbase){
491 a0d146ed 2005-07-12 devnull 		seterr(ECorrupt, "index section config table overlaps bucket storage");
492 a0d146ed 2005-07-12 devnull 		freeisect(is);
493 a0d146ed 2005-07-12 devnull 		return nil;
494 a0d146ed 2005-07-12 devnull 	}
495 a0d146ed 2005-07-12 devnull 	is->tabsize = is->blockbase - is->tabbase;
496 a0d146ed 2005-07-12 devnull 	v = is->part->size & ~(u64int)(is->blocksize - 1);
497 a0d146ed 2005-07-12 devnull 	if(is->blockbase + (u64int)is->blocks * is->blocksize != v){
498 a0d146ed 2005-07-12 devnull 		seterr(ECorrupt, "invalid blocks in index section %s", is->name);
499 28b49df3 2006-07-18 devnull 		/* ZZZ what to do? 
500 28b49df3 2006-07-18 devnull 		freeisect(is);
501 28b49df3 2006-07-18 devnull 		return nil;
502 28b49df3 2006-07-18 devnull 		*/
503 a0d146ed 2005-07-12 devnull 	}
504 a0d146ed 2005-07-12 devnull 
505 a0d146ed 2005-07-12 devnull 	if(is->stop - is->start > is->blocks){
506 a0d146ed 2005-07-12 devnull 		seterr(ECorrupt, "index section overflows available space");
507 a0d146ed 2005-07-12 devnull 		freeisect(is);
508 a0d146ed 2005-07-12 devnull 		return nil;
509 a0d146ed 2005-07-12 devnull 	}
510 a0d146ed 2005-07-12 devnull 	if(is->start > is->stop){
511 a0d146ed 2005-07-12 devnull 		seterr(ECorrupt, "invalid index section range");
512 a0d146ed 2005-07-12 devnull 		freeisect(is);
513 a0d146ed 2005-07-12 devnull 		return nil;
514 a0d146ed 2005-07-12 devnull 	}
515 a0d146ed 2005-07-12 devnull 
516 a0d146ed 2005-07-12 devnull 	return is;
517 a0d146ed 2005-07-12 devnull }
518 a0d146ed 2005-07-12 devnull 
519 a0d146ed 2005-07-12 devnull int
520 a0d146ed 2005-07-12 devnull wbisect(ISect *is)
521 a0d146ed 2005-07-12 devnull {
522 a0d146ed 2005-07-12 devnull 	ZBlock *b;
523 a0d146ed 2005-07-12 devnull 
524 a0d146ed 2005-07-12 devnull 	b = alloczblock(HeadSize, 1, 0);
525 28b49df3 2006-07-18 devnull 	if(b == nil){
526 28b49df3 2006-07-18 devnull 		/* ZZZ set error? */
527 a0d146ed 2005-07-12 devnull 		return -1;
528 28b49df3 2006-07-18 devnull 	}
529 a0d146ed 2005-07-12 devnull 
530 a0d146ed 2005-07-12 devnull 	if(packisect(is, b->data) < 0){
531 a0d146ed 2005-07-12 devnull 		seterr(ECorrupt, "can't make index section header: %r");
532 a0d146ed 2005-07-12 devnull 		freezblock(b);
533 a0d146ed 2005-07-12 devnull 		return -1;
534 a0d146ed 2005-07-12 devnull 	}
535 e46cacb0 2007-04-27 devnull 	if(writepart(is->part, PartBlank, b->data, HeadSize) < 0 || flushpart(is->part) < 0){
536 a0d146ed 2005-07-12 devnull 		seterr(EAdmin, "can't write index section header: %r");
537 a0d146ed 2005-07-12 devnull 		freezblock(b);
538 a0d146ed 2005-07-12 devnull 		return -1;
539 a0d146ed 2005-07-12 devnull 	}
540 a0d146ed 2005-07-12 devnull 	freezblock(b);
541 a0d146ed 2005-07-12 devnull 
542 a0d146ed 2005-07-12 devnull 	return 0;
543 a0d146ed 2005-07-12 devnull }
544 a0d146ed 2005-07-12 devnull 
545 a0d146ed 2005-07-12 devnull void
546 a0d146ed 2005-07-12 devnull freeisect(ISect *is)
547 a0d146ed 2005-07-12 devnull {
548 a0d146ed 2005-07-12 devnull 	if(is == nil)
549 a0d146ed 2005-07-12 devnull 		return;
550 a0d146ed 2005-07-12 devnull 	free(is);
551 a0d146ed 2005-07-12 devnull }
552 a0d146ed 2005-07-12 devnull 
553 a0d146ed 2005-07-12 devnull void
554 a0d146ed 2005-07-12 devnull freeindex(Index *ix)
555 a0d146ed 2005-07-12 devnull {
556 a0d146ed 2005-07-12 devnull 	int i;
557 a0d146ed 2005-07-12 devnull 
558 a0d146ed 2005-07-12 devnull 	if(ix == nil)
559 a0d146ed 2005-07-12 devnull 		return;
560 a0d146ed 2005-07-12 devnull 	free(ix->amap);
561 a0d146ed 2005-07-12 devnull 	free(ix->arenas);
562 a0d146ed 2005-07-12 devnull 	if(ix->sects)
563 a0d146ed 2005-07-12 devnull 		for(i = 0; i < ix->nsects; i++)
564 a0d146ed 2005-07-12 devnull 			freeisect(ix->sects[i]);
565 a0d146ed 2005-07-12 devnull 	free(ix->sects);
566 a0d146ed 2005-07-12 devnull 	free(ix->smap);
567 a0d146ed 2005-07-12 devnull 	free(ix);
568 a0d146ed 2005-07-12 devnull }
569 a0d146ed 2005-07-12 devnull 
570 a0d146ed 2005-07-12 devnull /*
571 a0d146ed 2005-07-12 devnull  * write a clump to an available arena in the index
572 a0d146ed 2005-07-12 devnull  * and return the address of the clump within the index.
573 a0d146ed 2005-07-12 devnull ZZZ question: should this distinguish between an arena
574 a0d146ed 2005-07-12 devnull filling up and real errors writing the clump?
575 a0d146ed 2005-07-12 devnull  */
576 a0d146ed 2005-07-12 devnull u64int
577 45ac814c 2007-10-29 rsc writeiclump(Index *ix, Clump *c, u8int *clbuf)
578 a0d146ed 2005-07-12 devnull {
579 a0d146ed 2005-07-12 devnull 	u64int a;
580 a0d146ed 2005-07-12 devnull 	int i;
581 45ac814c 2007-10-29 rsc 	IAddr ia;
582 45ac814c 2007-10-29 rsc 	AState as;
583 a0d146ed 2005-07-12 devnull 
584 a0d146ed 2005-07-12 devnull 	trace(TraceLump, "writeiclump enter");
585 45ac814c 2007-10-29 rsc 	qlock(&ix->writing);
586 45ac814c 2007-10-29 rsc 	for(i = ix->mapalloc; i < ix->narenas; i++){
587 45ac814c 2007-10-29 rsc 		a = writeaclump(ix->arenas[i], c, clbuf);
588 a0d146ed 2005-07-12 devnull 		if(a != TWID64){
589 45ac814c 2007-10-29 rsc 			ix->mapalloc = i;
590 45ac814c 2007-10-29 rsc 			ia.addr = ix->amap[i].start + a;
591 45ac814c 2007-10-29 rsc 			ia.type = c->info.type;
592 45ac814c 2007-10-29 rsc 			ia.size = c->info.uncsize;
593 45ac814c 2007-10-29 rsc 			ia.blocks = (c->info.size + ClumpSize + (1<<ABlockLog) - 1) >> ABlockLog;
594 45ac814c 2007-10-29 rsc 			as.arena = ix->arenas[i];
595 45ac814c 2007-10-29 rsc 			as.aa = ia.addr;
596 45ac814c 2007-10-29 rsc 			as.stats = as.arena->memstats;
597 45ac814c 2007-10-29 rsc 			insertscore(c->info.score, &ia, IEDirty, &as);
598 45ac814c 2007-10-29 rsc 			qunlock(&ix->writing);
599 a0d146ed 2005-07-12 devnull 			trace(TraceLump, "writeiclump exit");
600 45ac814c 2007-10-29 rsc 			return ia.addr;
601 a0d146ed 2005-07-12 devnull 		}
602 a0d146ed 2005-07-12 devnull 	}
603 45ac814c 2007-10-29 rsc 	qunlock(&ix->writing);
604 a0d146ed 2005-07-12 devnull 
605 a0d146ed 2005-07-12 devnull 	seterr(EAdmin, "no space left in arenas");
606 a0d146ed 2005-07-12 devnull 	trace(TraceLump, "writeiclump failed");
607 a0d146ed 2005-07-12 devnull 	return TWID64;
608 a0d146ed 2005-07-12 devnull }
609 a0d146ed 2005-07-12 devnull 
610 a0d146ed 2005-07-12 devnull /*
611 a0d146ed 2005-07-12 devnull  * convert an arena index to an relative arena address
612 a0d146ed 2005-07-12 devnull  */
613 a0d146ed 2005-07-12 devnull Arena*
614 a0d146ed 2005-07-12 devnull amapitoa(Index *ix, u64int a, u64int *aa)
615 a0d146ed 2005-07-12 devnull {
616 a0d146ed 2005-07-12 devnull 	int i, r, l, m;
617 a0d146ed 2005-07-12 devnull 
618 a0d146ed 2005-07-12 devnull 	l = 1;
619 a0d146ed 2005-07-12 devnull 	r = ix->narenas - 1;
620 a0d146ed 2005-07-12 devnull 	while(l <= r){
621 a0d146ed 2005-07-12 devnull 		m = (r + l) / 2;
622 a0d146ed 2005-07-12 devnull 		if(ix->amap[m].start <= a)
623 a0d146ed 2005-07-12 devnull 			l = m + 1;
624 a0d146ed 2005-07-12 devnull 		else
625 a0d146ed 2005-07-12 devnull 			r = m - 1;
626 a0d146ed 2005-07-12 devnull 	}
627 a0d146ed 2005-07-12 devnull 	l--;
628 a0d146ed 2005-07-12 devnull 
629 a0d146ed 2005-07-12 devnull 	if(a > ix->amap[l].stop){
630 a0d146ed 2005-07-12 devnull for(i=0; i<ix->narenas; i++)
631 a0d146ed 2005-07-12 devnull 	print("arena %d: %llux - %llux\n", i, ix->amap[i].start, ix->amap[i].stop);
632 a0d146ed 2005-07-12 devnull print("want arena %d for %llux\n", l, a);
633 a0d146ed 2005-07-12 devnull 		seterr(ECrash, "unmapped address passed to amapitoa");
634 a0d146ed 2005-07-12 devnull 		return nil;
635 a0d146ed 2005-07-12 devnull 	}
636 a0d146ed 2005-07-12 devnull 
637 a0d146ed 2005-07-12 devnull 	if(ix->arenas[l] == nil){
638 a0d146ed 2005-07-12 devnull 		seterr(ECrash, "unmapped arena selected in amapitoa");
639 a0d146ed 2005-07-12 devnull 		return nil;
640 a0d146ed 2005-07-12 devnull 	}
641 a0d146ed 2005-07-12 devnull 	*aa = a - ix->amap[l].start;
642 a0d146ed 2005-07-12 devnull 	return ix->arenas[l];
643 a0d146ed 2005-07-12 devnull }
644 a0d146ed 2005-07-12 devnull 
645 7a400ee9 2007-09-25 rsc /*
646 7a400ee9 2007-09-25 rsc  * convert an arena index to the bounds of the containing arena group.
647 7a400ee9 2007-09-25 rsc  */
648 7a400ee9 2007-09-25 rsc Arena*
649 7a400ee9 2007-09-25 rsc amapitoag(Index *ix, u64int a, u64int *gstart, u64int *glimit, int *g)
650 7a400ee9 2007-09-25 rsc {
651 7a400ee9 2007-09-25 rsc 	u64int aa;
652 7a400ee9 2007-09-25 rsc 	Arena *arena;
653 7a400ee9 2007-09-25 rsc 	
654 7a400ee9 2007-09-25 rsc 	arena = amapitoa(ix, a, &aa);
655 7a400ee9 2007-09-25 rsc 	if(arena == nil)
656 7a400ee9 2007-09-25 rsc 		return nil;
657 7a400ee9 2007-09-25 rsc 	if(arenatog(arena, aa, gstart, glimit, g) < 0)
658 7a400ee9 2007-09-25 rsc 		return nil;
659 7a400ee9 2007-09-25 rsc 	*gstart += a - aa;
660 7a400ee9 2007-09-25 rsc 	*glimit += a - aa;
661 7a400ee9 2007-09-25 rsc 	return arena;
662 7a400ee9 2007-09-25 rsc }
663 7a400ee9 2007-09-25 rsc 
664 a0d146ed 2005-07-12 devnull int
665 a0d146ed 2005-07-12 devnull iaddrcmp(IAddr *ia1, IAddr *ia2)
666 a0d146ed 2005-07-12 devnull {
667 a0d146ed 2005-07-12 devnull 	return ia1->type != ia2->type
668 a0d146ed 2005-07-12 devnull 		|| ia1->size != ia2->size
669 a0d146ed 2005-07-12 devnull 		|| ia1->blocks != ia2->blocks
670 a0d146ed 2005-07-12 devnull 		|| ia1->addr != ia2->addr;
671 a0d146ed 2005-07-12 devnull }
672 a0d146ed 2005-07-12 devnull 
673 a0d146ed 2005-07-12 devnull /*
674 a0d146ed 2005-07-12 devnull  * lookup the score in the partition
675 a0d146ed 2005-07-12 devnull  *
676 a0d146ed 2005-07-12 devnull  * nothing needs to be explicitly locked:
677 a0d146ed 2005-07-12 devnull  * only static parts of ix are used, and
678 a0d146ed 2005-07-12 devnull  * the bucket is locked by the DBlock lock.
679 a0d146ed 2005-07-12 devnull  */
680 a0d146ed 2005-07-12 devnull int
681 a0d146ed 2005-07-12 devnull loadientry(Index *ix, u8int *score, int type, IEntry *ie)
682 a0d146ed 2005-07-12 devnull {
683 a0d146ed 2005-07-12 devnull 	ISect *is;
684 a0d146ed 2005-07-12 devnull 	DBlock *b;
685 a0d146ed 2005-07-12 devnull 	IBucket ib;
686 a0d146ed 2005-07-12 devnull 	u32int buck;
687 a0d146ed 2005-07-12 devnull 	int h, ok;
688 a0d146ed 2005-07-12 devnull 
689 a0d146ed 2005-07-12 devnull 	ok = -1;
690 a0d146ed 2005-07-12 devnull 
691 a0d146ed 2005-07-12 devnull 	trace(TraceLump, "loadientry enter");
692 a0d146ed 2005-07-12 devnull 
693 a0d146ed 2005-07-12 devnull 	/*
694 a0d146ed 2005-07-12 devnull 	qlock(&stats.lock);
695 a0d146ed 2005-07-12 devnull 	stats.indexreads++;
696 a0d146ed 2005-07-12 devnull 	qunlock(&stats.lock);
697 a0d146ed 2005-07-12 devnull 	*/
698 a0d146ed 2005-07-12 devnull 
699 a0d146ed 2005-07-12 devnull 	if(!inbloomfilter(mainindex->bloom, score)){
700 a0d146ed 2005-07-12 devnull 		trace(TraceLump, "loadientry bloomhit");
701 a0d146ed 2005-07-12 devnull 		return -1;
702 a0d146ed 2005-07-12 devnull 	}
703 a0d146ed 2005-07-12 devnull 
704 a0d146ed 2005-07-12 devnull 	trace(TraceLump, "loadientry loadibucket");
705 a0d146ed 2005-07-12 devnull 	b = loadibucket(ix, score, &is, &buck, &ib);
706 a0d146ed 2005-07-12 devnull 	trace(TraceLump, "loadientry loadedibucket");
707 a0d146ed 2005-07-12 devnull 	if(b == nil)
708 a0d146ed 2005-07-12 devnull 		return -1;
709 a0d146ed 2005-07-12 devnull 
710 a0d146ed 2005-07-12 devnull 	if(okibucket(&ib, is) < 0){
711 a0d146ed 2005-07-12 devnull 		trace(TraceLump, "loadientry badbucket");
712 a0d146ed 2005-07-12 devnull 		goto out;
713 a0d146ed 2005-07-12 devnull 	}
714 a0d146ed 2005-07-12 devnull 
715 a0d146ed 2005-07-12 devnull 	h = bucklook(score, type, ib.data, ib.n);
716 a0d146ed 2005-07-12 devnull 	if(h & 1){
717 a0d146ed 2005-07-12 devnull 		h ^= 1;
718 a0d146ed 2005-07-12 devnull 		trace(TraceLump, "loadientry found");
719 a0d146ed 2005-07-12 devnull 		unpackientry(ie, &ib.data[h]);
720 a0d146ed 2005-07-12 devnull 		ok = 0;
721 a0d146ed 2005-07-12 devnull 		goto out;
722 a0d146ed 2005-07-12 devnull 	}
723 a0d146ed 2005-07-12 devnull 	trace(TraceLump, "loadientry notfound");
724 a0d146ed 2005-07-12 devnull 	addstat(StatBloomFalseMiss, 1);
725 a0d146ed 2005-07-12 devnull out:
726 a0d146ed 2005-07-12 devnull 	putdblock(b);
727 a0d146ed 2005-07-12 devnull 	trace(TraceLump, "loadientry exit");
728 a0d146ed 2005-07-12 devnull 	return ok;
729 a0d146ed 2005-07-12 devnull }
730 a0d146ed 2005-07-12 devnull 
731 a0d146ed 2005-07-12 devnull int
732 a0d146ed 2005-07-12 devnull okibucket(IBucket *ib, ISect *is)
733 a0d146ed 2005-07-12 devnull {
734 a0d146ed 2005-07-12 devnull 	if(ib->n <= is->buckmax)
735 a0d146ed 2005-07-12 devnull 		return 0;
736 a0d146ed 2005-07-12 devnull 
737 a0d146ed 2005-07-12 devnull 	seterr(EICorrupt, "corrupted disk index bucket: n=%ud max=%ud, range=[%lud,%lud)",
738 a0d146ed 2005-07-12 devnull 		ib->n, is->buckmax, is->start, is->stop);
739 a0d146ed 2005-07-12 devnull 	return -1;
740 a0d146ed 2005-07-12 devnull }
741 a0d146ed 2005-07-12 devnull 
742 a0d146ed 2005-07-12 devnull /*
743 a0d146ed 2005-07-12 devnull  * look for score within data;
744 a0d146ed 2005-07-12 devnull  * return 1 | byte index of matching index,
745 a0d146ed 2005-07-12 devnull  * or 0 | index of least element > score
746 a0d146ed 2005-07-12 devnull  */
747 a0d146ed 2005-07-12 devnull int
748 a0d146ed 2005-07-12 devnull bucklook(u8int *score, int otype, u8int *data, int n)
749 a0d146ed 2005-07-12 devnull {
750 a0d146ed 2005-07-12 devnull 	int i, r, l, m, h, c, cc, type;
751 a0d146ed 2005-07-12 devnull 
752 27d28098 2007-04-21 devnull 	if(otype == -1)
753 27d28098 2007-04-21 devnull 		type = -1;
754 27d28098 2007-04-21 devnull 	else
755 27d28098 2007-04-21 devnull 		type = vttodisktype(otype);
756 a0d146ed 2005-07-12 devnull 	l = 0;
757 a0d146ed 2005-07-12 devnull 	r = n - 1;
758 a0d146ed 2005-07-12 devnull 	while(l <= r){
759 a0d146ed 2005-07-12 devnull 		m = (r + l) >> 1;
760 a0d146ed 2005-07-12 devnull 		h = m * IEntrySize;
761 a0d146ed 2005-07-12 devnull 		for(i = 0; i < VtScoreSize; i++){
762 a0d146ed 2005-07-12 devnull 			c = score[i];
763 a0d146ed 2005-07-12 devnull 			cc = data[h + i];
764 a0d146ed 2005-07-12 devnull 			if(c != cc){
765 a0d146ed 2005-07-12 devnull 				if(c > cc)
766 a0d146ed 2005-07-12 devnull 					l = m + 1;
767 a0d146ed 2005-07-12 devnull 				else
768 a0d146ed 2005-07-12 devnull 					r = m - 1;
769 a0d146ed 2005-07-12 devnull 				goto cont;
770 a0d146ed 2005-07-12 devnull 			}
771 a0d146ed 2005-07-12 devnull 		}
772 a0d146ed 2005-07-12 devnull 		cc = data[h + IEntryTypeOff];
773 27d28098 2007-04-21 devnull 		if(type != cc && type != -1){
774 a0d146ed 2005-07-12 devnull 			if(type > cc)
775 a0d146ed 2005-07-12 devnull 				l = m + 1;
776 a0d146ed 2005-07-12 devnull 			else
777 a0d146ed 2005-07-12 devnull 				r = m - 1;
778 a0d146ed 2005-07-12 devnull 			goto cont;
779 a0d146ed 2005-07-12 devnull 		}
780 a0d146ed 2005-07-12 devnull 		return h | 1;
781 a0d146ed 2005-07-12 devnull 	cont:;
782 a0d146ed 2005-07-12 devnull 	}
783 a0d146ed 2005-07-12 devnull 
784 a0d146ed 2005-07-12 devnull 	return l * IEntrySize;
785 a0d146ed 2005-07-12 devnull }
786 a0d146ed 2005-07-12 devnull 
787 a0d146ed 2005-07-12 devnull /*
788 a0d146ed 2005-07-12 devnull  * compare two IEntries; consistent with bucklook
789 a0d146ed 2005-07-12 devnull  */
790 a0d146ed 2005-07-12 devnull int
791 a0d146ed 2005-07-12 devnull ientrycmp(const void *vie1, const void *vie2)
792 a0d146ed 2005-07-12 devnull {
793 a0d146ed 2005-07-12 devnull 	u8int *ie1, *ie2;
794 a0d146ed 2005-07-12 devnull 	int i, v1, v2;
795 a0d146ed 2005-07-12 devnull 
796 a0d146ed 2005-07-12 devnull 	ie1 = (u8int*)vie1;
797 a0d146ed 2005-07-12 devnull 	ie2 = (u8int*)vie2;
798 a0d146ed 2005-07-12 devnull 	for(i = 0; i < VtScoreSize; i++){
799 a0d146ed 2005-07-12 devnull 		v1 = ie1[i];
800 a0d146ed 2005-07-12 devnull 		v2 = ie2[i];
801 a0d146ed 2005-07-12 devnull 		if(v1 != v2){
802 a0d146ed 2005-07-12 devnull 			if(v1 < v2)
803 a0d146ed 2005-07-12 devnull 				return -1;
804 a0d146ed 2005-07-12 devnull 			return 1;
805 a0d146ed 2005-07-12 devnull 		}
806 a0d146ed 2005-07-12 devnull 	}
807 a0d146ed 2005-07-12 devnull 	v1 = ie1[IEntryTypeOff];
808 a0d146ed 2005-07-12 devnull 	v2 = ie2[IEntryTypeOff];
809 a0d146ed 2005-07-12 devnull 	if(v1 != v2){
810 a0d146ed 2005-07-12 devnull 		if(v1 < v2)
811 a0d146ed 2005-07-12 devnull 			return -1;
812 a0d146ed 2005-07-12 devnull 		return 1;
813 a0d146ed 2005-07-12 devnull 	}
814 a0d146ed 2005-07-12 devnull 	return 0;
815 a0d146ed 2005-07-12 devnull }
816 a0d146ed 2005-07-12 devnull 
817 a0d146ed 2005-07-12 devnull /*
818 a0d146ed 2005-07-12 devnull  * find the number of the index section holding bucket #buck
819 a0d146ed 2005-07-12 devnull  */
820 a0d146ed 2005-07-12 devnull int
821 a0d146ed 2005-07-12 devnull indexsect0(Index *ix, u32int buck)
822 a0d146ed 2005-07-12 devnull {
823 a0d146ed 2005-07-12 devnull 	int r, l, m;
824 a0d146ed 2005-07-12 devnull 
825 a0d146ed 2005-07-12 devnull 	l = 1;
826 a0d146ed 2005-07-12 devnull 	r = ix->nsects - 1;
827 a0d146ed 2005-07-12 devnull 	while(l <= r){
828 a0d146ed 2005-07-12 devnull 		m = (r + l) >> 1;
829 a0d146ed 2005-07-12 devnull 		if(ix->sects[m]->start <= buck)
830 a0d146ed 2005-07-12 devnull 			l = m + 1;
831 a0d146ed 2005-07-12 devnull 		else
832 a0d146ed 2005-07-12 devnull 			r = m - 1;
833 a0d146ed 2005-07-12 devnull 	}
834 a0d146ed 2005-07-12 devnull 	return l - 1;
835 a0d146ed 2005-07-12 devnull }
836 a0d146ed 2005-07-12 devnull 
837 a0d146ed 2005-07-12 devnull /*
838 a0d146ed 2005-07-12 devnull  * load the index block at bucket #buck
839 a0d146ed 2005-07-12 devnull  */
840 a0d146ed 2005-07-12 devnull static DBlock*
841 a0d146ed 2005-07-12 devnull loadibucket0(Index *ix, u32int buck, ISect **pis, u32int *pbuck, IBucket *ib, int mode)
842 a0d146ed 2005-07-12 devnull {
843 a0d146ed 2005-07-12 devnull 	ISect *is;
844 a0d146ed 2005-07-12 devnull 	DBlock *b;
845 a0d146ed 2005-07-12 devnull 
846 a0d146ed 2005-07-12 devnull 	is = ix->sects[indexsect0(ix, buck)];
847 a0d146ed 2005-07-12 devnull 	if(buck < is->start || is->stop <= buck){
848 a0d146ed 2005-07-12 devnull 		seterr(EAdmin, "index lookup out of range: %ud not found in index\n", buck);
849 a0d146ed 2005-07-12 devnull 		return nil;
850 a0d146ed 2005-07-12 devnull 	}
851 a0d146ed 2005-07-12 devnull 
852 a0d146ed 2005-07-12 devnull 	buck -= is->start;
853 a0d146ed 2005-07-12 devnull 	if((b = getdblock(is->part, is->blockbase + ((u64int)buck << is->blocklog), mode)) == nil)
854 a0d146ed 2005-07-12 devnull 		return nil;
855 a0d146ed 2005-07-12 devnull 
856 a0d146ed 2005-07-12 devnull 	if(pis)
857 a0d146ed 2005-07-12 devnull 		*pis = is;
858 a0d146ed 2005-07-12 devnull 	if(pbuck)
859 a0d146ed 2005-07-12 devnull 		*pbuck = buck;
860 a0d146ed 2005-07-12 devnull 	if(ib)
861 a0d146ed 2005-07-12 devnull 		unpackibucket(ib, b->data, is->bucketmagic);
862 a0d146ed 2005-07-12 devnull 	return b;
863 a0d146ed 2005-07-12 devnull }
864 a0d146ed 2005-07-12 devnull 
865 a0d146ed 2005-07-12 devnull /*
866 a0d146ed 2005-07-12 devnull  * find the number of the index section holding score
867 a0d146ed 2005-07-12 devnull  */
868 28b49df3 2006-07-18 devnull int
869 a0d146ed 2005-07-12 devnull indexsect1(Index *ix, u8int *score)
870 a0d146ed 2005-07-12 devnull {
871 a0d146ed 2005-07-12 devnull 	return indexsect0(ix, hashbits(score, 32) / ix->div);
872 a0d146ed 2005-07-12 devnull }
873 a0d146ed 2005-07-12 devnull 
874 a0d146ed 2005-07-12 devnull /*
875 a0d146ed 2005-07-12 devnull  * load the index block responsible for score.
876 a0d146ed 2005-07-12 devnull  */
877 a0d146ed 2005-07-12 devnull static DBlock*
878 a0d146ed 2005-07-12 devnull loadibucket1(Index *ix, u8int *score, ISect **pis, u32int *pbuck, IBucket *ib)
879 a0d146ed 2005-07-12 devnull {
880 a0d146ed 2005-07-12 devnull 	return loadibucket0(ix, hashbits(score, 32)/ix->div, pis, pbuck, ib, OREAD);
881 a0d146ed 2005-07-12 devnull }
882 a0d146ed 2005-07-12 devnull 
883 a0d146ed 2005-07-12 devnull int
884 a0d146ed 2005-07-12 devnull indexsect(Index *ix, u8int *score)
885 a0d146ed 2005-07-12 devnull {
886 a0d146ed 2005-07-12 devnull 	return indexsect1(ix, score);
887 a0d146ed 2005-07-12 devnull }
888 a0d146ed 2005-07-12 devnull 
889 a0d146ed 2005-07-12 devnull DBlock*
890 a0d146ed 2005-07-12 devnull loadibucket(Index *ix, u8int *score, ISect **pis, u32int *pbuck, IBucket *ib)
891 a0d146ed 2005-07-12 devnull {
892 a0d146ed 2005-07-12 devnull 	return loadibucket1(ix, score, pis, pbuck, ib);
893 a0d146ed 2005-07-12 devnull }
894 a0d146ed 2005-07-12 devnull 
895 a0d146ed 2005-07-12 devnull