xref: /plan9/sys/src/cmd/fossil/source.c (revision e569ccb51977f66f2b762edac90ad8d00e69e6cb)
15e96a66cSDavid du Colombier #include "stdinc.h"
25e96a66cSDavid du Colombier #include "dat.h"
35e96a66cSDavid du Colombier #include "fns.h"
45e96a66cSDavid du Colombier #include "error.h"
55e96a66cSDavid du Colombier 
65e96a66cSDavid du Colombier static int	sizeToDepth(uvlong s, int psize, int dsize);
75e96a66cSDavid du Colombier static u32int 	tagGen(void);
85e96a66cSDavid du Colombier static Block 	*sourceLoad(Source *r, Entry *e);
95e96a66cSDavid du Colombier static Block	*sourceLoadUnlocked(Source *r, Entry *e);
105e96a66cSDavid du Colombier static int	sourceShrinkDepth(Source*, Block*, Entry*, int);
115e96a66cSDavid du Colombier static int	sourceShrinkSize(Source*, Entry*, uvlong);
125e96a66cSDavid du Colombier static int	sourceGrowDepth(Source*, Block*, Entry*, int);
135e96a66cSDavid du Colombier 
145e96a66cSDavid du Colombier #define sourceIsLocked(r)	((r)->b != nil)
155e96a66cSDavid du Colombier 
165e96a66cSDavid du Colombier static Source *
175e96a66cSDavid du Colombier sourceAlloc(Fs *fs, Block *b, Source *p, u32int offset, int mode)
185e96a66cSDavid du Colombier {
195e96a66cSDavid du Colombier 	Source *r;
205e96a66cSDavid du Colombier 	int epb;
215e96a66cSDavid du Colombier 	u32int epoch;
225e96a66cSDavid du Colombier 	Entry e;
235e96a66cSDavid du Colombier 
245e96a66cSDavid du Colombier 	assert(p==nil || sourceIsLocked(p));
255e96a66cSDavid du Colombier 
265e96a66cSDavid du Colombier 	if(p == nil){
275e96a66cSDavid du Colombier 		assert(offset == 0);
285e96a66cSDavid du Colombier 		epb = 1;
295e96a66cSDavid du Colombier 	}else
305e96a66cSDavid du Colombier 		epb = p->dsize / VtEntrySize;
315e96a66cSDavid du Colombier 
325e96a66cSDavid du Colombier 	if(b->l.type != BtDir)
335e96a66cSDavid du Colombier 		goto Bad;
345e96a66cSDavid du Colombier 
355e96a66cSDavid du Colombier 	/*
365e96a66cSDavid du Colombier 	 * a non-active entry is the only thing that
375e96a66cSDavid du Colombier 	 * can legitimately happen here. all the others
385e96a66cSDavid du Colombier 	 * get prints.
395e96a66cSDavid du Colombier 	 */
405e96a66cSDavid du Colombier 	if(!entryUnpack(&e, b->data, offset % epb)){
415e96a66cSDavid du Colombier 		fprint(2, "entryUnpack failed\n");
425e96a66cSDavid du Colombier 		goto Bad;
435e96a66cSDavid du Colombier 	}
445e96a66cSDavid du Colombier 	if(!(e.flags & VtEntryActive)){
455e96a66cSDavid du Colombier 		if(0)fprint(2, "not active\n");
465e96a66cSDavid du Colombier 		goto Bad;
475e96a66cSDavid du Colombier 	}
485e96a66cSDavid du Colombier 	if(e.psize < 256 || e.dsize < 256){
495e96a66cSDavid du Colombier 		fprint(2, "psize %ud dsize %ud\n", e.psize, e.dsize);
505e96a66cSDavid du Colombier 		goto Bad;
515e96a66cSDavid du Colombier 	}
525e96a66cSDavid du Colombier 
535e96a66cSDavid du Colombier 	if(e.depth < sizeToDepth(e.size, e.psize, e.dsize)){
545e96a66cSDavid du Colombier 		fprint(2, "depth %ud size %llud psize %ud dsize %ud\n", e.depth, e.size, e.psize, e.dsize);
555e96a66cSDavid du Colombier 		goto Bad;
565e96a66cSDavid du Colombier 	}
575e96a66cSDavid du Colombier 
585e96a66cSDavid du Colombier 	if((e.flags & VtEntryLocal) && e.tag == 0){
595e96a66cSDavid du Colombier 		fprint(2, "flags %#ux tag %#ux\n", e.flags, e.tag);
605e96a66cSDavid du Colombier 		goto Bad;
615e96a66cSDavid du Colombier 	}
625e96a66cSDavid du Colombier 
635e96a66cSDavid du Colombier 	if(e.dsize > fs->blockSize || e.psize > fs->blockSize){
645e96a66cSDavid du Colombier 		fprint(2, "psize %ud dsize %ud blocksize %ud\n", e.psize, e.dsize, fs->blockSize);
655e96a66cSDavid du Colombier 		goto Bad;
665e96a66cSDavid du Colombier 	}
675e96a66cSDavid du Colombier 
685e96a66cSDavid du Colombier 	epoch = b->l.epoch;
695e96a66cSDavid du Colombier 	if(mode == OReadWrite){
705e96a66cSDavid du Colombier 		if(e.snap != 0){
715e96a66cSDavid du Colombier 			vtSetError(ESnapRO);
725e96a66cSDavid du Colombier 			return nil;
735e96a66cSDavid du Colombier 		}
745e96a66cSDavid du Colombier 	}else if(e.snap != 0){
755e96a66cSDavid du Colombier 		if(e.snap < fs->elo){
765e96a66cSDavid du Colombier 			vtSetError(ESnapOld);
775e96a66cSDavid du Colombier 			return nil;
785e96a66cSDavid du Colombier 		}
795e96a66cSDavid du Colombier 		if(e.snap >= fs->ehi)
805e96a66cSDavid du Colombier 			goto Bad;
815e96a66cSDavid du Colombier 		epoch = e.snap;
825e96a66cSDavid du Colombier 	}
835e96a66cSDavid du Colombier 
845e96a66cSDavid du Colombier 	r = vtMemAllocZ(sizeof(Source));
855e96a66cSDavid du Colombier 	r->fs = fs;
865e96a66cSDavid du Colombier 	r->mode = mode;
875e96a66cSDavid du Colombier 	r->dsize = e.dsize;
885e96a66cSDavid du Colombier 	r->gen = e.gen;
895e96a66cSDavid du Colombier 	r->dir = (e.flags & VtEntryDir) != 0;
905e96a66cSDavid du Colombier 	r->lk = vtLockAlloc();
915e96a66cSDavid du Colombier 	r->ref = 1;
925e96a66cSDavid du Colombier 	r->parent = p;
935e96a66cSDavid du Colombier 	if(p){
945e96a66cSDavid du Colombier 		vtLock(p->lk);
955e96a66cSDavid du Colombier 		assert(mode == OReadOnly || p->mode == OReadWrite);
965e96a66cSDavid du Colombier 		p->ref++;
975e96a66cSDavid du Colombier 		vtUnlock(p->lk);
985e96a66cSDavid du Colombier 	}
995e96a66cSDavid du Colombier 	r->epoch = epoch;
1005e96a66cSDavid du Colombier //fprint(2, "sourceAlloc have %V be.%d fse.%d %s\n", b->score, b->l.epoch, r->fs->ehi, mode==OReadWrite ? "rw" : "ro");
1015e96a66cSDavid du Colombier 	memmove(r->score, b->score, VtScoreSize);
1025e96a66cSDavid du Colombier 	r->scoreEpoch = b->l.epoch;
1035e96a66cSDavid du Colombier 	r->offset = offset;
1045e96a66cSDavid du Colombier 	r->epb = epb;
1055e96a66cSDavid du Colombier 	r->tag = b->l.tag;
1065e96a66cSDavid du Colombier 
1075e96a66cSDavid du Colombier //fprint(2, "sourceAlloc: %p -> %V %d\n", r, r->score, r->offset);
1085e96a66cSDavid du Colombier 
1095e96a66cSDavid du Colombier 	return r;
1105e96a66cSDavid du Colombier Bad:
1115e96a66cSDavid du Colombier 	vtSetError(EBadEntry);
1125e96a66cSDavid du Colombier 	return nil;
1135e96a66cSDavid du Colombier 
1145e96a66cSDavid du Colombier }
1155e96a66cSDavid du Colombier 
1165e96a66cSDavid du Colombier Source *
1175e96a66cSDavid du Colombier sourceRoot(Fs *fs, u32int addr, int mode)
1185e96a66cSDavid du Colombier {
1195e96a66cSDavid du Colombier 	Source *r;
1205e96a66cSDavid du Colombier 	Block *b;
1215e96a66cSDavid du Colombier 
1225e96a66cSDavid du Colombier 	b = cacheLocalData(fs->cache, addr, BtDir, RootTag, mode, 0);
1235e96a66cSDavid du Colombier 	if(b == nil)
1245e96a66cSDavid du Colombier 		return nil;
1255e96a66cSDavid du Colombier 
126*e569ccb5SDavid du Colombier 	if(mode == OReadWrite && b->l.epoch != fs->ehi){
1275e96a66cSDavid du Colombier 		fprint(2, "sourceRoot: fs->ehi = %ud, b->l = %L\n", fs->ehi, &b->l);
1285e96a66cSDavid du Colombier 		blockPut(b);
1295e96a66cSDavid du Colombier 		vtSetError(EBadRoot);
1305e96a66cSDavid du Colombier 		return nil;
1315e96a66cSDavid du Colombier 	}
1325e96a66cSDavid du Colombier 
1335e96a66cSDavid du Colombier 	r = sourceAlloc(fs, b, nil, 0, mode);
1345e96a66cSDavid du Colombier 	blockPut(b);
1355e96a66cSDavid du Colombier 	return r;
1365e96a66cSDavid du Colombier }
1375e96a66cSDavid du Colombier 
1385e96a66cSDavid du Colombier Source *
1395e96a66cSDavid du Colombier sourceOpen(Source *r, ulong offset, int mode)
1405e96a66cSDavid du Colombier {
1415e96a66cSDavid du Colombier 	ulong bn;
1425e96a66cSDavid du Colombier 	Block *b;
1435e96a66cSDavid du Colombier 
1445e96a66cSDavid du Colombier 	assert(sourceIsLocked(r));
1455e96a66cSDavid du Colombier 	if(r->mode == OReadWrite)
1465e96a66cSDavid du Colombier 		assert(r->epoch == r->b->l.epoch);
1475e96a66cSDavid du Colombier 	if(!r->dir){
1485e96a66cSDavid du Colombier 		vtSetError(ENotDir);
1495e96a66cSDavid du Colombier 		return nil;
1505e96a66cSDavid du Colombier 	}
1515e96a66cSDavid du Colombier 
1525e96a66cSDavid du Colombier 	bn = offset/(r->dsize/VtEntrySize);
1535e96a66cSDavid du Colombier 
1545e96a66cSDavid du Colombier 	b = sourceBlock(r, bn, mode);
1555e96a66cSDavid du Colombier 	if(b == nil)
1565e96a66cSDavid du Colombier 		return nil;
1575e96a66cSDavid du Colombier 	r = sourceAlloc(r->fs, b, r, offset, mode);
1585e96a66cSDavid du Colombier 	blockPut(b);
1595e96a66cSDavid du Colombier 	return r;
1605e96a66cSDavid du Colombier }
1615e96a66cSDavid du Colombier 
1625e96a66cSDavid du Colombier Source *
1635e96a66cSDavid du Colombier sourceCreate(Source *r, int dsize, int dir, u32int offset)
1645e96a66cSDavid du Colombier {
1655e96a66cSDavid du Colombier 	int i;
1665e96a66cSDavid du Colombier 	Block *b;
1675e96a66cSDavid du Colombier 	u32int bn, size;
1685e96a66cSDavid du Colombier 	Entry e;
1695e96a66cSDavid du Colombier 	int epb;
1705e96a66cSDavid du Colombier 	int psize;
1715e96a66cSDavid du Colombier 	Source *rr;
1725e96a66cSDavid du Colombier 
1735e96a66cSDavid du Colombier 	assert(sourceIsLocked(r));
1745e96a66cSDavid du Colombier 
1755e96a66cSDavid du Colombier 	if(!r->dir){
1765e96a66cSDavid du Colombier 		vtSetError(ENotDir);
1775e96a66cSDavid du Colombier 		return nil;
1785e96a66cSDavid du Colombier 	}
1795e96a66cSDavid du Colombier 
1805e96a66cSDavid du Colombier 	epb = r->dsize/VtEntrySize;
1815e96a66cSDavid du Colombier 	psize = (dsize/VtScoreSize)*VtScoreSize;
1825e96a66cSDavid du Colombier 
1835e96a66cSDavid du Colombier 	size = sourceGetDirSize(r);
1845e96a66cSDavid du Colombier 	if(offset == 0){
1855e96a66cSDavid du Colombier 		/*
1865e96a66cSDavid du Colombier 		 * look at a random block to see if we can find an empty entry
1875e96a66cSDavid du Colombier 		 */
1885e96a66cSDavid du Colombier 		offset = lnrand(size+1);
1895e96a66cSDavid du Colombier 		offset -= offset % epb;
1905e96a66cSDavid du Colombier 	}
1915e96a66cSDavid du Colombier 
1925e96a66cSDavid du Colombier 	/* try the given block and then try the last block */
1935e96a66cSDavid du Colombier 	for(;;){
1945e96a66cSDavid du Colombier 		bn = offset/epb;
1955e96a66cSDavid du Colombier 		b = sourceBlock(r, bn, OReadWrite);
1965e96a66cSDavid du Colombier 		if(b == nil)
1975e96a66cSDavid du Colombier 			return nil;
1985e96a66cSDavid du Colombier 		for(i=offset%r->epb; i<epb; i++){
1995e96a66cSDavid du Colombier 			entryUnpack(&e, b->data, i);
2005e96a66cSDavid du Colombier 			if((e.flags&VtEntryActive) == 0 && e.gen != ~0)
2015e96a66cSDavid du Colombier 				goto Found;
2025e96a66cSDavid du Colombier 		}
2035e96a66cSDavid du Colombier 		blockPut(b);
2045e96a66cSDavid du Colombier 		if(offset == size){
2055e96a66cSDavid du Colombier 			fprint(2, "sourceCreate: cannot happen\n");
2065e96a66cSDavid du Colombier 			vtSetError("sourceCreate: cannot happen");
2075e96a66cSDavid du Colombier 			return nil;
2085e96a66cSDavid du Colombier 		}
2095e96a66cSDavid du Colombier 		offset = size;
2105e96a66cSDavid du Colombier 	}
2115e96a66cSDavid du Colombier 
2125e96a66cSDavid du Colombier Found:
2135e96a66cSDavid du Colombier 	/* found an entry - gen already set */
2145e96a66cSDavid du Colombier 	e.psize = psize;
2155e96a66cSDavid du Colombier 	e.dsize = dsize;
216fe853e23SDavid du Colombier 	assert(psize && dsize);
2175e96a66cSDavid du Colombier 	e.flags = VtEntryActive;
2185e96a66cSDavid du Colombier 	if(dir)
2195e96a66cSDavid du Colombier 		e.flags |= VtEntryDir;
2205e96a66cSDavid du Colombier 	e.depth = 0;
2215e96a66cSDavid du Colombier 	e.size = 0;
2225e96a66cSDavid du Colombier 	memmove(e.score, vtZeroScore, VtScoreSize);
2235e96a66cSDavid du Colombier 	e.tag = 0;
2245e96a66cSDavid du Colombier 	e.snap = 0;
2255e96a66cSDavid du Colombier 	e.archive = 0;
2265e96a66cSDavid du Colombier 	entryPack(&e, b->data, i);
2275e96a66cSDavid du Colombier 	blockDirty(b);
2285e96a66cSDavid du Colombier 
2295e96a66cSDavid du Colombier 	offset = bn*epb + i;
2305e96a66cSDavid du Colombier 	if(offset+1 > size){
2315e96a66cSDavid du Colombier 		if(!sourceSetDirSize(r, offset+1)){
2325e96a66cSDavid du Colombier 			blockPut(b);
2335e96a66cSDavid du Colombier 			return nil;
2345e96a66cSDavid du Colombier 		}
2355e96a66cSDavid du Colombier 	}
2365e96a66cSDavid du Colombier 
2375e96a66cSDavid du Colombier 	rr = sourceAlloc(r->fs, b, r, offset, OReadWrite);
2385e96a66cSDavid du Colombier 	blockPut(b);
2395e96a66cSDavid du Colombier 	return rr;
2405e96a66cSDavid du Colombier }
2415e96a66cSDavid du Colombier 
2425e96a66cSDavid du Colombier static int
2435e96a66cSDavid du Colombier sourceKill(Source *r, int doremove)
2445e96a66cSDavid du Colombier {
2455e96a66cSDavid du Colombier 	Entry e;
2465e96a66cSDavid du Colombier 	Block *b;
2475e96a66cSDavid du Colombier 	u32int addr;
2485e96a66cSDavid du Colombier 	u32int tag;
2495e96a66cSDavid du Colombier 	int type;
2505e96a66cSDavid du Colombier 
2515e96a66cSDavid du Colombier 	assert(sourceIsLocked(r));
2525e96a66cSDavid du Colombier 	b = sourceLoad(r, &e);
2535e96a66cSDavid du Colombier 	if(b == nil)
2545e96a66cSDavid du Colombier 		return 0;
2555e96a66cSDavid du Colombier 
2565e96a66cSDavid du Colombier 	assert(b->l.epoch == r->fs->ehi);
2575e96a66cSDavid du Colombier 
2585e96a66cSDavid du Colombier 	if(doremove==0 && e.size == 0){
2595e96a66cSDavid du Colombier 		/* already truncated */
2605e96a66cSDavid du Colombier 		blockPut(b);
2615e96a66cSDavid du Colombier 		return 1;
2625e96a66cSDavid du Colombier 	}
2635e96a66cSDavid du Colombier 
2645e96a66cSDavid du Colombier 	/* remember info on link we are removing */
2655e96a66cSDavid du Colombier 	addr = globalToLocal(e.score);
2665e96a66cSDavid du Colombier 	type = entryType(&e);
2675e96a66cSDavid du Colombier 	tag = e.tag;
2685e96a66cSDavid du Colombier 
2695e96a66cSDavid du Colombier 	if(doremove){
2705e96a66cSDavid du Colombier 		if(e.gen != ~0)
2715e96a66cSDavid du Colombier 			e.gen++;
2725e96a66cSDavid du Colombier 		e.dsize = 0;
2735e96a66cSDavid du Colombier 		e.psize = 0;
2745e96a66cSDavid du Colombier 		e.flags = 0;
2755e96a66cSDavid du Colombier 	}else{
2765e96a66cSDavid du Colombier 		e.flags &= ~VtEntryLocal;
2775e96a66cSDavid du Colombier 	}
2785e96a66cSDavid du Colombier 	e.depth = 0;
2795e96a66cSDavid du Colombier 	e.size = 0;
2805e96a66cSDavid du Colombier 	e.tag = 0;
2815e96a66cSDavid du Colombier 	memmove(e.score, vtZeroScore, VtScoreSize);
2825e96a66cSDavid du Colombier 	entryPack(&e, b->data, r->offset % r->epb);
2835e96a66cSDavid du Colombier 	blockDirty(b);
2845e96a66cSDavid du Colombier 	if(addr != NilBlock)
285*e569ccb5SDavid du Colombier 		blockRemoveLink(b, addr, type, tag, 1);
2865e96a66cSDavid du Colombier 	blockPut(b);
2875e96a66cSDavid du Colombier 
2885e96a66cSDavid du Colombier 	if(doremove){
2895e96a66cSDavid du Colombier 		sourceUnlock(r);
2905e96a66cSDavid du Colombier 		sourceClose(r);
2915e96a66cSDavid du Colombier 	}
2925e96a66cSDavid du Colombier 
2935e96a66cSDavid du Colombier 	return 1;
2945e96a66cSDavid du Colombier }
2955e96a66cSDavid du Colombier 
2965e96a66cSDavid du Colombier int
2975e96a66cSDavid du Colombier sourceRemove(Source *r)
2985e96a66cSDavid du Colombier {
2995e96a66cSDavid du Colombier 	return sourceKill(r, 1);
3005e96a66cSDavid du Colombier }
3015e96a66cSDavid du Colombier 
3025e96a66cSDavid du Colombier int
3035e96a66cSDavid du Colombier sourceTruncate(Source *r)
3045e96a66cSDavid du Colombier {
3055e96a66cSDavid du Colombier 	return sourceKill(r, 0);
3065e96a66cSDavid du Colombier }
3075e96a66cSDavid du Colombier 
3085e96a66cSDavid du Colombier uvlong
3095e96a66cSDavid du Colombier sourceGetSize(Source *r)
3105e96a66cSDavid du Colombier {
3115e96a66cSDavid du Colombier 	Entry e;
3125e96a66cSDavid du Colombier 	Block *b;
3135e96a66cSDavid du Colombier 
3145e96a66cSDavid du Colombier 	assert(sourceIsLocked(r));
3155e96a66cSDavid du Colombier 	b = sourceLoad(r, &e);
3165e96a66cSDavid du Colombier 	if(b == nil)
3175e96a66cSDavid du Colombier 		return 0;
3185e96a66cSDavid du Colombier 	blockPut(b);
3195e96a66cSDavid du Colombier 
3205e96a66cSDavid du Colombier 	return e.size;
3215e96a66cSDavid du Colombier }
3225e96a66cSDavid du Colombier 
3235e96a66cSDavid du Colombier static int
3245e96a66cSDavid du Colombier sourceShrinkSize(Source *r, Entry *e, uvlong size)
3255e96a66cSDavid du Colombier {
3265e96a66cSDavid du Colombier 	int i, type, ppb;
3275e96a66cSDavid du Colombier 	uvlong ptrsz;
3285e96a66cSDavid du Colombier 	u32int addr;
3295e96a66cSDavid du Colombier 	uchar score[VtScoreSize];
3305e96a66cSDavid du Colombier 	Block *b;
3315e96a66cSDavid du Colombier 
3325e96a66cSDavid du Colombier 	type = entryType(e);
3335e96a66cSDavid du Colombier 	b = cacheGlobal(r->fs->cache, e->score, type, e->tag, OReadWrite);
3345e96a66cSDavid du Colombier 	if(b == nil)
3355e96a66cSDavid du Colombier 		return 0;
3365e96a66cSDavid du Colombier 
3375e96a66cSDavid du Colombier 	ptrsz = e->dsize;
3385e96a66cSDavid du Colombier 	ppb = e->psize/VtScoreSize;
3395e96a66cSDavid du Colombier 	for(i=0; i+1<e->depth; i++)
3405e96a66cSDavid du Colombier 		ptrsz *= ppb;
3415e96a66cSDavid du Colombier 
3425e96a66cSDavid du Colombier 	while(type&BtLevelMask){
343*e569ccb5SDavid du Colombier 		if(b->addr == NilBlock || b->l.epoch != r->fs->ehi){
3445e96a66cSDavid du Colombier 			/* not worth copying the block just so we can zero some of it */
3455e96a66cSDavid du Colombier 			blockPut(b);
3465e96a66cSDavid du Colombier 			return 0;
3475e96a66cSDavid du Colombier 		}
3485e96a66cSDavid du Colombier 
3495e96a66cSDavid du Colombier 		/*
3505e96a66cSDavid du Colombier 		 * invariant: each pointer in the tree rooted at b accounts for ptrsz bytes
3515e96a66cSDavid du Colombier 		 */
3525e96a66cSDavid du Colombier 
3535e96a66cSDavid du Colombier 		/* zero the pointers to unnecessary blocks */
3545e96a66cSDavid du Colombier 		i = (size+ptrsz-1)/ptrsz;
3555e96a66cSDavid du Colombier 		for(; i<ppb; i++){
3565e96a66cSDavid du Colombier 			addr = globalToLocal(b->data+i*VtScoreSize);
3575e96a66cSDavid du Colombier 			memmove(b->data+i*VtScoreSize, vtZeroScore, VtScoreSize);
3585e96a66cSDavid du Colombier 			blockDirty(b);
3595e96a66cSDavid du Colombier 			if(addr != NilBlock)
360*e569ccb5SDavid du Colombier 				blockRemoveLink(b, addr, type-1, e->tag, 1);
3615e96a66cSDavid du Colombier 		}
3625e96a66cSDavid du Colombier 
3635e96a66cSDavid du Colombier 		/* recurse (go around again) on the partially necessary block */
3645e96a66cSDavid du Colombier 		i = size/ptrsz;
3655e96a66cSDavid du Colombier 		size = size%ptrsz;
3665e96a66cSDavid du Colombier 		if(size == 0){
3675e96a66cSDavid du Colombier 			blockPut(b);
3685e96a66cSDavid du Colombier 			return 1;
3695e96a66cSDavid du Colombier 		}
3705e96a66cSDavid du Colombier 		ptrsz /= ppb;
3715e96a66cSDavid du Colombier 		type--;
3725e96a66cSDavid du Colombier 		memmove(score, b->data+i*VtScoreSize, VtScoreSize);
3735e96a66cSDavid du Colombier 		blockPut(b);
3745e96a66cSDavid du Colombier 		b = cacheGlobal(r->fs->cache, score, type, e->tag, OReadWrite);
3755e96a66cSDavid du Colombier 		if(b == nil)
3765e96a66cSDavid du Colombier 			return 0;
3775e96a66cSDavid du Colombier 	}
3785e96a66cSDavid du Colombier 
379*e569ccb5SDavid du Colombier 	if(b->addr == NilBlock || b->l.epoch != r->fs->ehi){
3805e96a66cSDavid du Colombier 		blockPut(b);
3815e96a66cSDavid du Colombier 		return 0;
3825e96a66cSDavid du Colombier 	}
3835e96a66cSDavid du Colombier 
3845e96a66cSDavid du Colombier 	/*
3855e96a66cSDavid du Colombier 	 * No one ever truncates BtDir blocks.
3865e96a66cSDavid du Colombier 	 */
3875e96a66cSDavid du Colombier 	if(type == BtData && e->dsize > size){
3885e96a66cSDavid du Colombier 		memset(b->data+size, 0, e->dsize-size);
3895e96a66cSDavid du Colombier 		blockDirty(b);
3905e96a66cSDavid du Colombier 	}
3915e96a66cSDavid du Colombier 	blockPut(b);
3925e96a66cSDavid du Colombier 	return 1;
3935e96a66cSDavid du Colombier }
3945e96a66cSDavid du Colombier 
3955e96a66cSDavid du Colombier int
3965e96a66cSDavid du Colombier sourceSetSize(Source *r, uvlong size)
3975e96a66cSDavid du Colombier {
3985e96a66cSDavid du Colombier 	int depth;
3995e96a66cSDavid du Colombier 	Entry e;
4005e96a66cSDavid du Colombier 	Block *b;
4015e96a66cSDavid du Colombier 
4025e96a66cSDavid du Colombier 	assert(sourceIsLocked(r));
4035e96a66cSDavid du Colombier 	if(size == 0)
4045e96a66cSDavid du Colombier 		return sourceTruncate(r);
4055e96a66cSDavid du Colombier 
4065e96a66cSDavid du Colombier 	if(size > VtMaxFileSize || size > ((uvlong)MaxBlock)*r->dsize){
4075e96a66cSDavid du Colombier 		vtSetError(ETooBig);
4085e96a66cSDavid du Colombier 		return 0;
4095e96a66cSDavid du Colombier 	}
4105e96a66cSDavid du Colombier 
4115e96a66cSDavid du Colombier 	b = sourceLoad(r, &e);
4125e96a66cSDavid du Colombier 	if(b == nil)
4135e96a66cSDavid du Colombier 		return 0;
4145e96a66cSDavid du Colombier 
4155e96a66cSDavid du Colombier 	/* quick out */
4165e96a66cSDavid du Colombier 	if(e.size == size){
4175e96a66cSDavid du Colombier 		blockPut(b);
4185e96a66cSDavid du Colombier 		return 1;
4195e96a66cSDavid du Colombier 	}
4205e96a66cSDavid du Colombier 
4215e96a66cSDavid du Colombier 	depth = sizeToDepth(size, e.psize, e.dsize);
4225e96a66cSDavid du Colombier 
4235e96a66cSDavid du Colombier 	if(depth < e.depth){
4245e96a66cSDavid du Colombier 		if(!sourceShrinkDepth(r, b, &e, depth)){
4255e96a66cSDavid du Colombier 			blockPut(b);
4265e96a66cSDavid du Colombier 			return 0;
4275e96a66cSDavid du Colombier 		}
4285e96a66cSDavid du Colombier 	}else if(depth > e.depth){
4295e96a66cSDavid du Colombier 		if(!sourceGrowDepth(r, b, &e, depth)){
4305e96a66cSDavid du Colombier 			blockPut(b);
4315e96a66cSDavid du Colombier 			return 0;
4325e96a66cSDavid du Colombier 		}
4335e96a66cSDavid du Colombier 	}
4345e96a66cSDavid du Colombier 
4355e96a66cSDavid du Colombier 	if(size < e.size)
4365e96a66cSDavid du Colombier 		sourceShrinkSize(r, &e, size);
4375e96a66cSDavid du Colombier 
4385e96a66cSDavid du Colombier 	e.size = size;
4395e96a66cSDavid du Colombier 	entryPack(&e, b->data, r->offset % r->epb);
4405e96a66cSDavid du Colombier 	blockDirty(b);
4415e96a66cSDavid du Colombier 	blockPut(b);
4425e96a66cSDavid du Colombier 
4435e96a66cSDavid du Colombier 	return 1;
4445e96a66cSDavid du Colombier }
4455e96a66cSDavid du Colombier 
4465e96a66cSDavid du Colombier int
4475e96a66cSDavid du Colombier sourceSetDirSize(Source *r, ulong ds)
4485e96a66cSDavid du Colombier {
4495e96a66cSDavid du Colombier 	uvlong size;
4505e96a66cSDavid du Colombier 	int epb;
4515e96a66cSDavid du Colombier 
4525e96a66cSDavid du Colombier 	assert(sourceIsLocked(r));
4535e96a66cSDavid du Colombier 	epb = r->dsize/VtEntrySize;
4545e96a66cSDavid du Colombier 
4555e96a66cSDavid du Colombier 	size = (uvlong)r->dsize*(ds/epb);
4565e96a66cSDavid du Colombier 	size += VtEntrySize*(ds%epb);
4575e96a66cSDavid du Colombier 	return sourceSetSize(r, size);
4585e96a66cSDavid du Colombier }
4595e96a66cSDavid du Colombier 
4605e96a66cSDavid du Colombier ulong
4615e96a66cSDavid du Colombier sourceGetDirSize(Source *r)
4625e96a66cSDavid du Colombier {
4635e96a66cSDavid du Colombier 	ulong ds;
4645e96a66cSDavid du Colombier 	uvlong size;
4655e96a66cSDavid du Colombier 	int epb;
4665e96a66cSDavid du Colombier 
4675e96a66cSDavid du Colombier 	assert(sourceIsLocked(r));
4685e96a66cSDavid du Colombier 	epb = r->dsize/VtEntrySize;
4695e96a66cSDavid du Colombier 
4705e96a66cSDavid du Colombier 	size = sourceGetSize(r);
4715e96a66cSDavid du Colombier 	ds = epb*(size/r->dsize);
4725e96a66cSDavid du Colombier 	ds += (size%r->dsize)/VtEntrySize;
4735e96a66cSDavid du Colombier 	return ds;
4745e96a66cSDavid du Colombier }
4755e96a66cSDavid du Colombier 
4765e96a66cSDavid du Colombier int
4775e96a66cSDavid du Colombier sourceGetEntry(Source *r, Entry *e)
4785e96a66cSDavid du Colombier {
4795e96a66cSDavid du Colombier 	Block *b;
4805e96a66cSDavid du Colombier 
4815e96a66cSDavid du Colombier 	assert(sourceIsLocked(r));
4825e96a66cSDavid du Colombier 	b = sourceLoad(r, e);
4835e96a66cSDavid du Colombier 	if(b == nil)
4845e96a66cSDavid du Colombier 		return 0;
4855e96a66cSDavid du Colombier 	blockPut(b);
4865e96a66cSDavid du Colombier 
4875e96a66cSDavid du Colombier 	return 1;
4885e96a66cSDavid du Colombier }
4895e96a66cSDavid du Colombier 
490fe853e23SDavid du Colombier /*
491fe853e23SDavid du Colombier  * Must be careful with this.  Doesn't record
492fe853e23SDavid du Colombier  * dependencies, so don't introduce any!
493fe853e23SDavid du Colombier  */
494fe853e23SDavid du Colombier int
495fe853e23SDavid du Colombier sourceSetEntry(Source *r, Entry *e)
496fe853e23SDavid du Colombier {
497fe853e23SDavid du Colombier 	Block *b;
498fe853e23SDavid du Colombier 	Entry oe;
499fe853e23SDavid du Colombier 
500fe853e23SDavid du Colombier 	assert(sourceIsLocked(r));
501fe853e23SDavid du Colombier 	b = sourceLoad(r, &oe);
502fe853e23SDavid du Colombier 	if(b == nil)
503fe853e23SDavid du Colombier 		return 0;
504fe853e23SDavid du Colombier 	entryPack(e, b->data, r->offset%r->epb);
505fe853e23SDavid du Colombier 	blockDirty(b);
506fe853e23SDavid du Colombier 	blockPut(b);
507fe853e23SDavid du Colombier 
508fe853e23SDavid du Colombier 	return 1;
509fe853e23SDavid du Colombier }
510fe853e23SDavid du Colombier 
5115e96a66cSDavid du Colombier static Block *
5125e96a66cSDavid du Colombier blockWalk(Block *p, int index, int mode, Fs *fs, Entry *e)
5135e96a66cSDavid du Colombier {
5145e96a66cSDavid du Colombier 	Block *b;
5155e96a66cSDavid du Colombier 	Cache *c;
5165e96a66cSDavid du Colombier 	u32int addr;
5175e96a66cSDavid du Colombier 	int type;
518fe853e23SDavid du Colombier 	uchar oscore[VtScoreSize], score[VtScoreSize];
51961201b97SDavid du Colombier 	Entry oe;
5205e96a66cSDavid du Colombier 
5215e96a66cSDavid du Colombier 	c = fs->cache;
5225e96a66cSDavid du Colombier 
5235e96a66cSDavid du Colombier 	if((p->l.type & BtLevelMask) == 0){
5245e96a66cSDavid du Colombier 		assert(p->l.type == BtDir);
5255e96a66cSDavid du Colombier 		type = entryType(e);
5265e96a66cSDavid du Colombier 		b = cacheGlobal(c, e->score, type, e->tag, mode);
5275e96a66cSDavid du Colombier 	}else{
5285e96a66cSDavid du Colombier 		type = p->l.type - 1;
5295e96a66cSDavid du Colombier 		b = cacheGlobal(c, p->data + index*VtScoreSize, type, e->tag, mode);
5305e96a66cSDavid du Colombier 	}
5315e96a66cSDavid du Colombier 
532fe853e23SDavid du Colombier 	if(b)
533fe853e23SDavid du Colombier 		b->pc = getcallerpc(&p);
534fe853e23SDavid du Colombier 
5355e96a66cSDavid du Colombier 	if(b == nil || mode == OReadOnly)
5365e96a66cSDavid du Colombier 		return b;
5375e96a66cSDavid du Colombier 
538*e569ccb5SDavid du Colombier 	if(p->l.epoch != fs->ehi){
539*e569ccb5SDavid du Colombier 		fprint(2, "blockWalk: parent not writable\n");
540*e569ccb5SDavid du Colombier 		abort();
541*e569ccb5SDavid du Colombier 	}
542*e569ccb5SDavid du Colombier 	if(b->l.epoch == fs->ehi)
5435e96a66cSDavid du Colombier 		return b;
5445e96a66cSDavid du Colombier 
54561201b97SDavid du Colombier 	oe = *e;
54661201b97SDavid du Colombier 
5475e96a66cSDavid du Colombier 	/*
5485e96a66cSDavid du Colombier 	 * Copy on write.
5495e96a66cSDavid du Colombier 	 */
5505e96a66cSDavid du Colombier 	if(e->tag == 0){
5515e96a66cSDavid du Colombier 		assert(p->l.type == BtDir);
5525e96a66cSDavid du Colombier 		e->tag = tagGen();
5535e96a66cSDavid du Colombier 		e->flags |= VtEntryLocal;
5545e96a66cSDavid du Colombier 	}
5555e96a66cSDavid du Colombier 
5565e96a66cSDavid du Colombier 	addr = b->addr;
5575e96a66cSDavid du Colombier 	b = blockCopy(b, e->tag, fs->ehi, fs->elo);
5585e96a66cSDavid du Colombier 	if(b == nil)
5595e96a66cSDavid du Colombier 		return nil;
5605e96a66cSDavid du Colombier 
561fe853e23SDavid du Colombier 	b->pc = getcallerpc(&p);
5625e96a66cSDavid du Colombier 	assert(b->l.epoch == fs->ehi);
5635e96a66cSDavid du Colombier 
56461201b97SDavid du Colombier 	blockDirty(b);
565fe853e23SDavid du Colombier 	memmove(score, b->score, VtScoreSize);
5665e96a66cSDavid du Colombier 	if(p->l.type == BtDir){
5675e96a66cSDavid du Colombier 		memmove(e->score, b->score, VtScoreSize);
5685e96a66cSDavid du Colombier 		entryPack(e, p->data, index);
56961201b97SDavid du Colombier 		blockDependency(p, b, index, nil, &oe);
5705e96a66cSDavid du Colombier 	}else{
57161201b97SDavid du Colombier 		memmove(oscore, p->data+index*VtScoreSize, VtScoreSize);
5725e96a66cSDavid du Colombier 		memmove(p->data+index*VtScoreSize, b->score, VtScoreSize);
57361201b97SDavid du Colombier 		blockDependency(p, b, index, oscore, nil);
5745e96a66cSDavid du Colombier 	}
5755e96a66cSDavid du Colombier 	blockDirty(p);
5765e96a66cSDavid du Colombier 
5775e96a66cSDavid du Colombier 	if(addr != NilBlock)
578*e569ccb5SDavid du Colombier 		blockRemoveLink(p, addr, type, e->tag, 0);
5795e96a66cSDavid du Colombier 
5805e96a66cSDavid du Colombier 	return b;
5815e96a66cSDavid du Colombier }
5825e96a66cSDavid du Colombier 
5835e96a66cSDavid du Colombier /*
5845e96a66cSDavid du Colombier  * Change the depth of the source r.
5855e96a66cSDavid du Colombier  * The entry e for r is contained in block p.
5865e96a66cSDavid du Colombier  */
5875e96a66cSDavid du Colombier static int
5885e96a66cSDavid du Colombier sourceGrowDepth(Source *r, Block *p, Entry *e, int depth)
5895e96a66cSDavid du Colombier {
5905e96a66cSDavid du Colombier 	Block *b, *bb;
5915e96a66cSDavid du Colombier 	u32int tag;
5925e96a66cSDavid du Colombier 	int type;
59361201b97SDavid du Colombier 	Entry oe;
5945e96a66cSDavid du Colombier 
5955e96a66cSDavid du Colombier 	assert(sourceIsLocked(r));
5965e96a66cSDavid du Colombier 	assert(depth <= VtPointerDepth);
5975e96a66cSDavid du Colombier 
5985e96a66cSDavid du Colombier 	type = entryType(e);
5995e96a66cSDavid du Colombier 	b = cacheGlobal(r->fs->cache, e->score, type, e->tag, OReadWrite);
6005e96a66cSDavid du Colombier 	if(b == nil)
6015e96a66cSDavid du Colombier 		return 0;
6025e96a66cSDavid du Colombier 
6035e96a66cSDavid du Colombier 	tag = e->tag;
6045e96a66cSDavid du Colombier 	if(tag == 0)
6055e96a66cSDavid du Colombier 		tag = tagGen();
6065e96a66cSDavid du Colombier 
60761201b97SDavid du Colombier 	oe = *e;
60861201b97SDavid du Colombier 
6095e96a66cSDavid du Colombier 	/*
6105e96a66cSDavid du Colombier 	 * Keep adding layers until we get to the right depth
6115e96a66cSDavid du Colombier 	 * or an error occurs.
6125e96a66cSDavid du Colombier 	 */
6135e96a66cSDavid du Colombier 	while(e->depth < depth){
6145e96a66cSDavid du Colombier 		bb = cacheAllocBlock(r->fs->cache, type+1, tag, r->fs->ehi, r->fs->elo);
6155e96a66cSDavid du Colombier 		if(bb == nil)
6165e96a66cSDavid du Colombier 			break;
617dc5a79c1SDavid du Colombier //fprint(2, "alloc %lux grow %V\n", bb->addr, b->score);
6185e96a66cSDavid du Colombier 		memmove(bb->data, b->score, VtScoreSize);
6195e96a66cSDavid du Colombier 		memmove(e->score, bb->score, VtScoreSize);
6205e96a66cSDavid du Colombier 		e->depth++;
6215e96a66cSDavid du Colombier 		type++;
6225e96a66cSDavid du Colombier 		e->tag = tag;
6235e96a66cSDavid du Colombier 		e->flags |= VtEntryLocal;
62461201b97SDavid du Colombier 		blockDependency(bb, b, 0, vtZeroScore, nil);
6255e96a66cSDavid du Colombier 		blockPut(b);
6265e96a66cSDavid du Colombier 		b = bb;
627fe853e23SDavid du Colombier 		blockDirty(b);
6285e96a66cSDavid du Colombier 	}
6295e96a66cSDavid du Colombier 
6305e96a66cSDavid du Colombier 	entryPack(e, p->data, r->offset % r->epb);
63161201b97SDavid du Colombier 	blockDependency(p, b, r->offset % r->epb, nil, &oe);
6325e96a66cSDavid du Colombier 	blockPut(b);
6335e96a66cSDavid du Colombier 	blockDirty(p);
6345e96a66cSDavid du Colombier 
6355e96a66cSDavid du Colombier 	return e->depth == depth;
6365e96a66cSDavid du Colombier }
6375e96a66cSDavid du Colombier 
6385e96a66cSDavid du Colombier static int
6395e96a66cSDavid du Colombier sourceShrinkDepth(Source *r, Block *p, Entry *e, int depth)
6405e96a66cSDavid du Colombier {
6415e96a66cSDavid du Colombier 	Block *b, *nb, *ob, *rb;
6425e96a66cSDavid du Colombier 	u32int tag;
6435e96a66cSDavid du Colombier 	int type, d;
64461201b97SDavid du Colombier 	Entry oe;
6455e96a66cSDavid du Colombier 
6465e96a66cSDavid du Colombier 	assert(sourceIsLocked(r));
6475e96a66cSDavid du Colombier 	assert(depth <= VtPointerDepth);
6485e96a66cSDavid du Colombier 
6495e96a66cSDavid du Colombier 	type = entryType(e);
6505e96a66cSDavid du Colombier 	rb = cacheGlobal(r->fs->cache, e->score, type, e->tag, OReadWrite);
6515e96a66cSDavid du Colombier 	if(rb == nil)
6525e96a66cSDavid du Colombier 		return 0;
6535e96a66cSDavid du Colombier 
6545e96a66cSDavid du Colombier 	tag = e->tag;
6555e96a66cSDavid du Colombier 	if(tag == 0)
6565e96a66cSDavid du Colombier 		tag = tagGen();
6575e96a66cSDavid du Colombier 
6585e96a66cSDavid du Colombier 	/*
6595e96a66cSDavid du Colombier 	 * Walk down to the new root block.
6605e96a66cSDavid du Colombier 	 * We may stop early, but something is better than nothing.
6615e96a66cSDavid du Colombier 	 */
66261201b97SDavid du Colombier 	oe = *e;
66361201b97SDavid du Colombier 
6645e96a66cSDavid du Colombier 	ob = nil;
6655e96a66cSDavid du Colombier 	b = rb;
666b5e190f4SDavid du Colombier /* BUG: explain type++.  i think it is a real bug */
6675e96a66cSDavid du Colombier 	for(d=e->depth; d > depth; d--, type++){
6685e96a66cSDavid du Colombier 		nb = cacheGlobal(r->fs->cache, b->data, type-1, tag, OReadWrite);
6695e96a66cSDavid du Colombier 		if(nb == nil)
6705e96a66cSDavid du Colombier 			break;
6715e96a66cSDavid du Colombier 		if(ob!=nil && ob!=rb)
6725e96a66cSDavid du Colombier 			blockPut(ob);
6735e96a66cSDavid du Colombier 		ob = b;
6745e96a66cSDavid du Colombier 		b = nb;
6755e96a66cSDavid du Colombier 	}
6765e96a66cSDavid du Colombier 
6775e96a66cSDavid du Colombier 	if(b == rb){
6785e96a66cSDavid du Colombier 		blockPut(rb);
6795e96a66cSDavid du Colombier 		return 0;
6805e96a66cSDavid du Colombier 	}
6815e96a66cSDavid du Colombier 
6825e96a66cSDavid du Colombier 	/*
6835e96a66cSDavid du Colombier 	 * Right now, e points at the root block rb, b is the new root block,
6845e96a66cSDavid du Colombier 	 * and ob points at b.  To update:
6855e96a66cSDavid du Colombier 	 *
6865e96a66cSDavid du Colombier 	 *	(i) change e to point at b
6875e96a66cSDavid du Colombier 	 *	(ii) zero the pointer ob -> b
6885e96a66cSDavid du Colombier 	 *	(iii) free the root block
6895e96a66cSDavid du Colombier 	 *
6905e96a66cSDavid du Colombier 	 * p (the block containing e) must be written before
6915e96a66cSDavid du Colombier 	 * anything else.
6925e96a66cSDavid du Colombier  	 */
6935e96a66cSDavid du Colombier 
6945e96a66cSDavid du Colombier 	/* (i) */
6955e96a66cSDavid du Colombier 	e->depth = d;
6965e96a66cSDavid du Colombier 	memmove(e->score, b->score, VtScoreSize);
6975e96a66cSDavid du Colombier 	entryPack(e, p->data, r->offset % r->epb);
69861201b97SDavid du Colombier 	blockDependency(p, b, r->offset % r->epb, nil, &oe);
6995e96a66cSDavid du Colombier 	blockDirty(p);
7005e96a66cSDavid du Colombier 
7015e96a66cSDavid du Colombier 	/* (ii) */
7025e96a66cSDavid du Colombier 	memmove(ob->data, vtZeroScore, VtScoreSize);
70361201b97SDavid du Colombier 	blockDependency(ob, p, 0, b->score, nil);
7045e96a66cSDavid du Colombier 	blockDirty(ob);
7055e96a66cSDavid du Colombier 
7065e96a66cSDavid du Colombier 	/* (iii) */
7075e96a66cSDavid du Colombier 	if(rb->addr != NilBlock)
708*e569ccb5SDavid du Colombier 		blockRemoveLink(p, rb->addr, rb->l.type, rb->l.tag, 1);
7095e96a66cSDavid du Colombier 
7105e96a66cSDavid du Colombier 	blockPut(rb);
7115e96a66cSDavid du Colombier 	if(ob!=nil && ob!=rb)
7125e96a66cSDavid du Colombier 		blockPut(ob);
7135e96a66cSDavid du Colombier 	blockPut(b);
7145e96a66cSDavid du Colombier 
7155e96a66cSDavid du Colombier 	return d == depth;
7165e96a66cSDavid du Colombier }
7175e96a66cSDavid du Colombier 
718fe853e23SDavid du Colombier /*
719fe853e23SDavid du Colombier  * Normally we return the block at the given number.
720fe853e23SDavid du Colombier  * If early is set, we stop earlier in the tree.  Setting early
721fe853e23SDavid du Colombier  * to 1 gives us the block that contains the pointer to bn.
722fe853e23SDavid du Colombier  */
7235e96a66cSDavid du Colombier Block *
724fe853e23SDavid du Colombier _sourceBlock(Source *r, ulong bn, int mode, int early, ulong tag)
7255e96a66cSDavid du Colombier {
7265e96a66cSDavid du Colombier 	Block *b, *bb;
7275e96a66cSDavid du Colombier 	int index[VtPointerDepth+1];
7285e96a66cSDavid du Colombier 	Entry e;
7295e96a66cSDavid du Colombier 	int i, np;
7305e96a66cSDavid du Colombier 	int m;
7315e96a66cSDavid du Colombier 
7325e96a66cSDavid du Colombier 	assert(sourceIsLocked(r));
7335e96a66cSDavid du Colombier 	assert(bn != NilBlock);
7345e96a66cSDavid du Colombier 
7355e96a66cSDavid du Colombier 	/* mode for intermediate block */
7365e96a66cSDavid du Colombier 	m = mode;
7375e96a66cSDavid du Colombier 	if(m == OOverWrite)
7385e96a66cSDavid du Colombier 		m = OReadWrite;
7395e96a66cSDavid du Colombier 
7405e96a66cSDavid du Colombier 	b = sourceLoad(r, &e);
7415e96a66cSDavid du Colombier 	if(b == nil)
7425e96a66cSDavid du Colombier 		return nil;
743fe853e23SDavid du Colombier 	if(r->mode == OReadOnly && (e.flags & VtEntryNoArchive)){
744fe853e23SDavid du Colombier 		blockPut(b);
745fe853e23SDavid du Colombier 		vtSetError(ENotArchived);
746fe853e23SDavid du Colombier 		return nil;
747fe853e23SDavid du Colombier 	}
748fe853e23SDavid du Colombier 
749fe853e23SDavid du Colombier 	if(tag){
750fe853e23SDavid du Colombier 		if(e.tag == 0)
751fe853e23SDavid du Colombier 			e.tag = tag;
752fe853e23SDavid du Colombier 		else if(e.tag != tag){
753fe853e23SDavid du Colombier 			fprint(2, "tag mismatch\n");
754fe853e23SDavid du Colombier 			vtSetError("tag mismatch");
755fe853e23SDavid du Colombier 			goto Err;
756fe853e23SDavid du Colombier 		}
757fe853e23SDavid du Colombier 	}
7585e96a66cSDavid du Colombier 
7595e96a66cSDavid du Colombier 	np = e.psize/VtScoreSize;
7605e96a66cSDavid du Colombier 	memset(index, 0, sizeof(index));
7615e96a66cSDavid du Colombier 	for(i=0; bn > 0; i++){
7625e96a66cSDavid du Colombier 		if(i >= VtPointerDepth){
7635e96a66cSDavid du Colombier 			vtSetError(EBadAddr);
7645e96a66cSDavid du Colombier 			goto Err;
7655e96a66cSDavid du Colombier 		}
7665e96a66cSDavid du Colombier 		index[i] = bn % np;
7675e96a66cSDavid du Colombier 		bn /= np;
7685e96a66cSDavid du Colombier 	}
7695e96a66cSDavid du Colombier 
7705e96a66cSDavid du Colombier 	if(i > e.depth){
7715e96a66cSDavid du Colombier 		if(mode == OReadOnly){
7725e96a66cSDavid du Colombier 			vtSetError(EBadAddr);
7735e96a66cSDavid du Colombier 			goto Err;
7745e96a66cSDavid du Colombier 		}
7755e96a66cSDavid du Colombier 		if(!sourceGrowDepth(r, b, &e, i))
7765e96a66cSDavid du Colombier 			goto Err;
7775e96a66cSDavid du Colombier 	}
7785e96a66cSDavid du Colombier 
7795e96a66cSDavid du Colombier 	index[e.depth] = r->offset % r->epb;
7805e96a66cSDavid du Colombier 
781fe853e23SDavid du Colombier 	for(i=e.depth; i>=early; i--){
7825e96a66cSDavid du Colombier 		bb = blockWalk(b, index[i], m, r->fs, &e);
7835e96a66cSDavid du Colombier 		if(bb == nil)
7845e96a66cSDavid du Colombier 			goto Err;
7855e96a66cSDavid du Colombier 		blockPut(b);
7865e96a66cSDavid du Colombier 		b = bb;
7875e96a66cSDavid du Colombier 	}
788fe853e23SDavid du Colombier 	b->pc = getcallerpc(&r);
7895e96a66cSDavid du Colombier 	return b;
7905e96a66cSDavid du Colombier Err:
7915e96a66cSDavid du Colombier 	blockPut(b);
7925e96a66cSDavid du Colombier 	return nil;
7935e96a66cSDavid du Colombier }
7945e96a66cSDavid du Colombier 
795fe853e23SDavid du Colombier Block*
796fe853e23SDavid du Colombier sourceBlock(Source *r, ulong bn, int mode)
797fe853e23SDavid du Colombier {
798fe853e23SDavid du Colombier 	Block *b;
799fe853e23SDavid du Colombier 
800fe853e23SDavid du Colombier 	b = _sourceBlock(r, bn, mode, 0, 0);
801fe853e23SDavid du Colombier 	if(b)
802fe853e23SDavid du Colombier 		b->pc = getcallerpc(&r);
803fe853e23SDavid du Colombier 	return b;
804fe853e23SDavid du Colombier }
805fe853e23SDavid du Colombier 
8065e96a66cSDavid du Colombier void
8075e96a66cSDavid du Colombier sourceClose(Source *r)
8085e96a66cSDavid du Colombier {
8095e96a66cSDavid du Colombier 	if(r == nil)
8105e96a66cSDavid du Colombier 		return;
8115e96a66cSDavid du Colombier 	vtLock(r->lk);
8125e96a66cSDavid du Colombier 	r->ref--;
8135e96a66cSDavid du Colombier 	if(r->ref){
8145e96a66cSDavid du Colombier 		vtUnlock(r->lk);
8155e96a66cSDavid du Colombier 		return;
8165e96a66cSDavid du Colombier 	}
8175e96a66cSDavid du Colombier 	assert(r->ref == 0);
8185e96a66cSDavid du Colombier 	vtUnlock(r->lk);
8195e96a66cSDavid du Colombier 	if(r->parent)
8205e96a66cSDavid du Colombier 		sourceClose(r->parent);
8215e96a66cSDavid du Colombier 	vtLockFree(r->lk);
8225e96a66cSDavid du Colombier 	memset(r, ~0, sizeof(*r));
8235e96a66cSDavid du Colombier 	vtMemFree(r);
8245e96a66cSDavid du Colombier }
8255e96a66cSDavid du Colombier 
8265e96a66cSDavid du Colombier /*
8275e96a66cSDavid du Colombier  * Retrieve the block containing the entry for r.
8285e96a66cSDavid du Colombier  * If a snapshot has happened, we might need
8295e96a66cSDavid du Colombier  * to get a new copy of the block.  We avoid this
8305e96a66cSDavid du Colombier  * in the common case by caching the score for
8315e96a66cSDavid du Colombier  * the block and the last epoch in which it was valid.
8325e96a66cSDavid du Colombier  *
8335e96a66cSDavid du Colombier  * We use r->mode to tell the difference between active
8345e96a66cSDavid du Colombier  * file system sources (OReadWrite) and sources for the
8355e96a66cSDavid du Colombier  * snapshot file system (OReadOnly).
8365e96a66cSDavid du Colombier  */
8375e96a66cSDavid du Colombier static Block*
8385e96a66cSDavid du Colombier sourceLoadBlock(Source *r, int mode)
8395e96a66cSDavid du Colombier {
8405e96a66cSDavid du Colombier 	u32int addr;
8415e96a66cSDavid du Colombier 	Block *b;
8425e96a66cSDavid du Colombier 
8435e96a66cSDavid du Colombier 	switch(r->mode){
8445e96a66cSDavid du Colombier 	default:
8455e96a66cSDavid du Colombier 		assert(0);
8465e96a66cSDavid du Colombier 	case OReadWrite:
8475e96a66cSDavid du Colombier 		assert(r->mode == OReadWrite);
84834e04225SDavid du Colombier 		/*
84934e04225SDavid du Colombier 		 * This needn't be true -- we might bump the low epoch
85034e04225SDavid du Colombier 		 * to reclaim some old blocks, but since this score is
85134e04225SDavid du Colombier 		 * OReadWrite, the blocks must all still be open, so none
85234e04225SDavid du Colombier 		 * are reclaimed.  Thus it's okay that the epoch is so low.
85334e04225SDavid du Colombier 		 * Proceed.
8545e96a66cSDavid du Colombier 		assert(r->epoch >= r->fs->elo);
85534e04225SDavid du Colombier 		 */
8565e96a66cSDavid du Colombier 		if(r->epoch == r->fs->ehi){
8575e96a66cSDavid du Colombier 			b = cacheGlobal(r->fs->cache, r->score, BtDir, r->tag, OReadWrite);
8585e96a66cSDavid du Colombier 			if(b == nil)
8595e96a66cSDavid du Colombier 				return nil;
8607611b47fSDavid du Colombier 			assert(r->epoch == b->l.epoch);
8615e96a66cSDavid du Colombier 			return b;
8625e96a66cSDavid du Colombier 		}
8635e96a66cSDavid du Colombier 		assert(r->parent != nil);
8645e96a66cSDavid du Colombier 		if(!sourceLock(r->parent, OReadWrite))
8655e96a66cSDavid du Colombier 			return nil;
8665e96a66cSDavid du Colombier 		b = sourceBlock(r->parent, r->offset/r->epb, OReadWrite);
8675e96a66cSDavid du Colombier 		sourceUnlock(r->parent);
8685e96a66cSDavid du Colombier 		if(b == nil)
8695e96a66cSDavid du Colombier 			return nil;
8705e96a66cSDavid du Colombier 		assert(b->l.epoch == r->fs->ehi);
871*e569ccb5SDavid du Colombier 	//	fprint(2, "sourceLoadBlock %p %V => %V\n", r, r->score, b->score);
8725e96a66cSDavid du Colombier 		memmove(r->score, b->score, VtScoreSize);
8735e96a66cSDavid du Colombier 		r->scoreEpoch = b->l.epoch;
8745e96a66cSDavid du Colombier 		r->tag = b->l.tag;
8755e96a66cSDavid du Colombier 		r->epoch = r->fs->ehi;
8765e96a66cSDavid du Colombier 		return b;
8775e96a66cSDavid du Colombier 
8785e96a66cSDavid du Colombier 	case OReadOnly:
8795e96a66cSDavid du Colombier 		addr = globalToLocal(r->score);
8805e96a66cSDavid du Colombier 		if(addr == NilBlock)
8815e96a66cSDavid du Colombier 			return cacheGlobal(r->fs->cache, r->score, BtDir, r->tag, mode);
8825e96a66cSDavid du Colombier 
8835e96a66cSDavid du Colombier 		b = cacheLocalData(r->fs->cache, addr, BtDir, r->tag, mode, r->scoreEpoch);
8845e96a66cSDavid du Colombier 		if(b)
8855e96a66cSDavid du Colombier 			return b;
8865e96a66cSDavid du Colombier 
8875e96a66cSDavid du Colombier 		/*
8885e96a66cSDavid du Colombier 		 * If it failed because the epochs don't match, the block has been
8895e96a66cSDavid du Colombier 		 * archived and reclaimed.  Rewalk from the parent and get the
8905e96a66cSDavid du Colombier 		 * new pointer.  This can't happen in the OReadWrite case
8915e96a66cSDavid du Colombier 		 * above because blocks in the current epoch don't get
8925e96a66cSDavid du Colombier 		 * reclaimed.  The fact that we're OReadOnly means we're
8935e96a66cSDavid du Colombier 		 * a snapshot.  (Or else the file system is read-only, but then
8945e96a66cSDavid du Colombier 		 * the archiver isn't going around deleting blocks.)
8955e96a66cSDavid du Colombier 		 */
8965e96a66cSDavid du Colombier 		if(strcmp(vtGetError(), ELabelMismatch) == 0){
8975e96a66cSDavid du Colombier 			if(!sourceLock(r->parent, OReadOnly))
8985e96a66cSDavid du Colombier 				return nil;
8995e96a66cSDavid du Colombier 			b = sourceBlock(r->parent, r->offset/r->epb, OReadOnly);
9005e96a66cSDavid du Colombier 			sourceUnlock(r->parent);
9015e96a66cSDavid du Colombier 			if(b){
9025e96a66cSDavid du Colombier 				fprint(2, "sourceAlloc: lost %V found %V\n",
9035e96a66cSDavid du Colombier 					r->score, b->score);
9045e96a66cSDavid du Colombier 				memmove(r->score, b->score, VtScoreSize);
9055e96a66cSDavid du Colombier 				r->scoreEpoch = b->l.epoch;
9065e96a66cSDavid du Colombier 				return b;
9075e96a66cSDavid du Colombier 			}
9085e96a66cSDavid du Colombier 		}
9095e96a66cSDavid du Colombier 		return nil;
9105e96a66cSDavid du Colombier 	}
9115e96a66cSDavid du Colombier }
9125e96a66cSDavid du Colombier 
9135e96a66cSDavid du Colombier int
9145e96a66cSDavid du Colombier sourceLock(Source *r, int mode)
9155e96a66cSDavid du Colombier {
9165e96a66cSDavid du Colombier 	Block *b;
9175e96a66cSDavid du Colombier 
9185e96a66cSDavid du Colombier 	if(mode == -1)
9195e96a66cSDavid du Colombier 		mode = r->mode;
9205e96a66cSDavid du Colombier 
9215e96a66cSDavid du Colombier 	b = sourceLoadBlock(r, mode);
9225e96a66cSDavid du Colombier 	if(b == nil)
9235e96a66cSDavid du Colombier 		return 0;
9245e96a66cSDavid du Colombier 	/*
9255e96a66cSDavid du Colombier 	 * The fact that we are holding b serves as the
9265e96a66cSDavid du Colombier 	 * lock entitling us to write to r->b.
9275e96a66cSDavid du Colombier 	 */
9285e96a66cSDavid du Colombier 	assert(r->b == nil);
9295e96a66cSDavid du Colombier 	r->b = b;
9305e96a66cSDavid du Colombier 	if(r->mode == OReadWrite)
9315e96a66cSDavid du Colombier 		assert(r->epoch == r->b->l.epoch);
9325e96a66cSDavid du Colombier 	return 1;
9335e96a66cSDavid du Colombier }
9345e96a66cSDavid du Colombier 
9355e96a66cSDavid du Colombier /*
9365e96a66cSDavid du Colombier  * Lock two (usually sibling) sources.  This needs special care
9375e96a66cSDavid du Colombier  * because the Entries for both sources might be in the same block.
9385e96a66cSDavid du Colombier  * We also try to lock blocks in left-to-right order within the tree.
9395e96a66cSDavid du Colombier  */
9405e96a66cSDavid du Colombier int
9415e96a66cSDavid du Colombier sourceLock2(Source *r, Source *rr, int mode)
9425e96a66cSDavid du Colombier {
9435e96a66cSDavid du Colombier 	Block *b, *bb;
9445e96a66cSDavid du Colombier 
9455e96a66cSDavid du Colombier 	if(rr == nil)
9465e96a66cSDavid du Colombier 		return sourceLock(r, mode);
9475e96a66cSDavid du Colombier 
9485e96a66cSDavid du Colombier 	if(mode == -1)
9495e96a66cSDavid du Colombier 		mode = r->mode;
9505e96a66cSDavid du Colombier 
9515e96a66cSDavid du Colombier 	if(r->parent==rr->parent && r->offset/r->epb == rr->offset/rr->epb){
9525e96a66cSDavid du Colombier 		b = sourceLoadBlock(r, mode);
9535e96a66cSDavid du Colombier 		if(b == nil)
9545e96a66cSDavid du Colombier 			return 0;
955*e569ccb5SDavid du Colombier 		if(memcmp(r->score, rr->score, VtScoreSize) != 0){
956*e569ccb5SDavid du Colombier 			memmove(rr->score, b->score, VtScoreSize);
957*e569ccb5SDavid du Colombier 			rr->scoreEpoch = b->l.epoch;
958*e569ccb5SDavid du Colombier 			rr->tag = b->l.tag;
959*e569ccb5SDavid du Colombier 			rr->epoch = rr->fs->ehi;
960*e569ccb5SDavid du Colombier 		}
9615e96a66cSDavid du Colombier 		blockDupLock(b);
9625e96a66cSDavid du Colombier 		bb = b;
9635e96a66cSDavid du Colombier 	}else if(r->parent==rr->parent || r->offset > rr->offset){
9645e96a66cSDavid du Colombier 		bb = sourceLoadBlock(rr, mode);
9655e96a66cSDavid du Colombier 		b = sourceLoadBlock(r, mode);
9665e96a66cSDavid du Colombier 	}else{
9675e96a66cSDavid du Colombier 		b = sourceLoadBlock(r, mode);
9685e96a66cSDavid du Colombier 		bb = sourceLoadBlock(rr, mode);
9695e96a66cSDavid du Colombier 	}
9705e96a66cSDavid du Colombier 	if(b == nil || bb == nil){
9715e96a66cSDavid du Colombier 		if(b)
9725e96a66cSDavid du Colombier 			blockPut(b);
9735e96a66cSDavid du Colombier 		if(bb)
9745e96a66cSDavid du Colombier 			blockPut(bb);
9755e96a66cSDavid du Colombier 		return 0;
9765e96a66cSDavid du Colombier 	}
9775e96a66cSDavid du Colombier 
9785e96a66cSDavid du Colombier 	/*
9795e96a66cSDavid du Colombier 	 * The fact that we are holding b and bb serves
9805e96a66cSDavid du Colombier 	 * as the lock entitling us to write to r->b and rr->b.
9815e96a66cSDavid du Colombier 	 */
9825e96a66cSDavid du Colombier 	r->b = b;
9835e96a66cSDavid du Colombier 	rr->b = bb;
9845e96a66cSDavid du Colombier 	return 1;
9855e96a66cSDavid du Colombier }
9865e96a66cSDavid du Colombier 
9875e96a66cSDavid du Colombier void
9885e96a66cSDavid du Colombier sourceUnlock(Source *r)
9895e96a66cSDavid du Colombier {
9905e96a66cSDavid du Colombier 	Block *b;
9915e96a66cSDavid du Colombier 
9925e96a66cSDavid du Colombier 	if(r->b == nil){
9935e96a66cSDavid du Colombier 		fprint(2, "sourceUnlock: already unlocked\n");
9945e96a66cSDavid du Colombier 		abort();
9955e96a66cSDavid du Colombier 	}
9965e96a66cSDavid du Colombier 	b = r->b;
9975e96a66cSDavid du Colombier 	r->b = nil;
9985e96a66cSDavid du Colombier 	blockPut(b);
9995e96a66cSDavid du Colombier }
10005e96a66cSDavid du Colombier 
10015e96a66cSDavid du Colombier static Block*
10025e96a66cSDavid du Colombier sourceLoad(Source *r, Entry *e)
10035e96a66cSDavid du Colombier {
10045e96a66cSDavid du Colombier 	Block *b;
10055e96a66cSDavid du Colombier 
10065e96a66cSDavid du Colombier 	assert(sourceIsLocked(r));
10075e96a66cSDavid du Colombier 	b = r->b;
10085e96a66cSDavid du Colombier 	if(!entryUnpack(e, b->data, r->offset % r->epb))
10095e96a66cSDavid du Colombier 		return nil;
10105e96a66cSDavid du Colombier 	if(e->gen != r->gen){
10115e96a66cSDavid du Colombier 		vtSetError(ERemoved);
10125e96a66cSDavid du Colombier 		return nil;
10135e96a66cSDavid du Colombier 	}
10145e96a66cSDavid du Colombier 	blockDupLock(b);
10155e96a66cSDavid du Colombier 	return b;
10165e96a66cSDavid du Colombier }
10175e96a66cSDavid du Colombier 
10185e96a66cSDavid du Colombier static int
10195e96a66cSDavid du Colombier sizeToDepth(uvlong s, int psize, int dsize)
10205e96a66cSDavid du Colombier {
10215e96a66cSDavid du Colombier 	int np;
10225e96a66cSDavid du Colombier 	int d;
10235e96a66cSDavid du Colombier 
10245e96a66cSDavid du Colombier 	/* determine pointer depth */
10255e96a66cSDavid du Colombier 	np = psize/VtScoreSize;
10265e96a66cSDavid du Colombier 	s = (s + dsize - 1)/dsize;
10275e96a66cSDavid du Colombier 	for(d = 0; s > 1; d++)
10285e96a66cSDavid du Colombier 		s = (s + np - 1)/np;
10295e96a66cSDavid du Colombier 	return d;
10305e96a66cSDavid du Colombier }
10315e96a66cSDavid du Colombier 
10325e96a66cSDavid du Colombier static u32int
10335e96a66cSDavid du Colombier tagGen(void)
10345e96a66cSDavid du Colombier {
10355e96a66cSDavid du Colombier 	u32int tag;
10365e96a66cSDavid du Colombier 
10375e96a66cSDavid du Colombier 	for(;;){
10385e96a66cSDavid du Colombier 		tag = lrand();
10395e96a66cSDavid du Colombier 		if(tag >= UserTag)
10405e96a66cSDavid du Colombier 			break;
10415e96a66cSDavid du Colombier 	}
10425e96a66cSDavid du Colombier 	return tag;
10435e96a66cSDavid du Colombier }
1044