1 6f4d00ee 2013-09-23 0intro #include "stdinc.h"
2 6f4d00ee 2013-09-23 0intro #include "dat.h"
3 6f4d00ee 2013-09-23 0intro #include "fns.h"
4 6f4d00ee 2013-09-23 0intro #include "error.h"
5 6f4d00ee 2013-09-23 0intro #include "9.h"
7 6f4d00ee 2013-09-23 0intro static int sizeToDepth(uvlong s, int psize, int dsize);
8 6f4d00ee 2013-09-23 0intro static u32int tagGen(void);
9 6f4d00ee 2013-09-23 0intro static Block *sourceLoad(Source *r, Entry *e);
10 6f4d00ee 2013-09-23 0intro static int sourceShrinkDepth(Source*, Block*, Entry*, int);
11 6f4d00ee 2013-09-23 0intro static int sourceShrinkSize(Source*, Entry*, uvlong);
12 6f4d00ee 2013-09-23 0intro static int sourceGrowDepth(Source*, Block*, Entry*, int);
14 6f4d00ee 2013-09-23 0intro #define sourceIsLocked(r) ((r)->b != nil)
16 6f4d00ee 2013-09-23 0intro static Source *
17 6f4d00ee 2013-09-23 0intro sourceAlloc(Fs *fs, Block *b, Source *p, u32int offset, int mode, int issnapshot)
20 6f4d00ee 2013-09-23 0intro u32int epoch;
21 6f4d00ee 2013-09-23 0intro char *pname = nil;
22 6f4d00ee 2013-09-23 0intro Source *r;
25 6f4d00ee 2013-09-23 0intro assert(p==nil || sourceIsLocked(p));
27 6f4d00ee 2013-09-23 0intro if(p == nil){
28 6f4d00ee 2013-09-23 0intro assert(offset == 0);
31 6f4d00ee 2013-09-23 0intro epb = p->dsize / VtEntrySize;
33 6f4d00ee 2013-09-23 0intro if(b->l.type != BtDir)
37 6f4d00ee 2013-09-23 0intro * a non-active entry is the only thing that
38 6f4d00ee 2013-09-23 0intro * can legitimately happen here. all the others
39 6f4d00ee 2013-09-23 0intro * get prints.
41 6f4d00ee 2013-09-23 0intro if(!entryUnpack(&e, b->data, offset % epb)){
42 6f4d00ee 2013-09-23 0intro pname = sourceName(p);
43 6f4d00ee 2013-09-23 0intro consPrint("%s: %s %V: sourceAlloc: entryUnpack failed\n",
44 6f4d00ee 2013-09-23 0intro fs->name, pname, b->score);
47 6f4d00ee 2013-09-23 0intro if(!(e.flags & VtEntryActive)){
48 6f4d00ee 2013-09-23 0intro pname = sourceName(p);
49 6f4d00ee 2013-09-23 0intro if(0) consPrint("%s: %s %V: sourceAlloc: not active\n",
50 6f4d00ee 2013-09-23 0intro fs->name, pname, e.score);
53 6f4d00ee 2013-09-23 0intro if(e.psize < 256 || e.dsize < 256){
54 6f4d00ee 2013-09-23 0intro pname = sourceName(p);
55 6f4d00ee 2013-09-23 0intro consPrint("%s: %s %V: sourceAlloc: psize %ud or dsize %ud < 256\n",
56 6f4d00ee 2013-09-23 0intro fs->name, pname, e.score, e.psize, e.dsize);
60 6f4d00ee 2013-09-23 0intro if(e.depth < sizeToDepth(e.size, e.psize, e.dsize)){
61 6f4d00ee 2013-09-23 0intro pname = sourceName(p);
62 6f4d00ee 2013-09-23 0intro consPrint("%s: %s %V: sourceAlloc: depth %ud size %llud "
63 6f4d00ee 2013-09-23 0intro "psize %ud dsize %ud\n", fs->name, pname,
64 6f4d00ee 2013-09-23 0intro e.score, e.depth, e.size, e.psize, e.dsize);
68 6f4d00ee 2013-09-23 0intro if((e.flags & VtEntryLocal) && e.tag == 0){
69 6f4d00ee 2013-09-23 0intro pname = sourceName(p);
70 6f4d00ee 2013-09-23 0intro consPrint("%s: %s %V: sourceAlloc: flags %#ux tag %#ux\n",
71 6f4d00ee 2013-09-23 0intro fs->name, pname, e.score, e.flags, e.tag);
75 6f4d00ee 2013-09-23 0intro if(e.dsize > fs->blockSize || e.psize > fs->blockSize){
76 6f4d00ee 2013-09-23 0intro pname = sourceName(p);
77 6f4d00ee 2013-09-23 0intro consPrint("%s: %s %V: sourceAlloc: psize %ud or dsize %ud "
78 6f4d00ee 2013-09-23 0intro "> blocksize %ud\n", fs->name, pname, e.score,
79 6f4d00ee 2013-09-23 0intro e.psize, e.dsize, fs->blockSize);
83 6f4d00ee 2013-09-23 0intro epoch = b->l.epoch;
84 6f4d00ee 2013-09-23 0intro if(mode == OReadWrite){
85 6f4d00ee 2013-09-23 0intro if(e.snap != 0){
86 4b576658 2013-09-23 0intro werrstr(ESnapRO);
87 6f4d00ee 2013-09-23 0intro return nil;
89 6f4d00ee 2013-09-23 0intro }else if(e.snap != 0){
90 6f4d00ee 2013-09-23 0intro if(e.snap < fs->elo){
91 4b576658 2013-09-23 0intro werrstr(ESnapOld);
92 6f4d00ee 2013-09-23 0intro return nil;
94 6f4d00ee 2013-09-23 0intro if(e.snap >= fs->ehi)
96 6f4d00ee 2013-09-23 0intro epoch = e.snap;
99 4b576658 2013-09-23 0intro r = vtmallocz(sizeof(Source));
100 6f4d00ee 2013-09-23 0intro r->fs = fs;
101 6f4d00ee 2013-09-23 0intro r->mode = mode;
102 6f4d00ee 2013-09-23 0intro r->issnapshot = issnapshot;
103 6f4d00ee 2013-09-23 0intro r->dsize = e.dsize;
104 6f4d00ee 2013-09-23 0intro r->gen = e.gen;
105 4b576658 2013-09-23 0intro r->dir = (e.flags & _VtEntryDir) != 0;
106 6f4d00ee 2013-09-23 0intro r->ref = 1;
107 6f4d00ee 2013-09-23 0intro r->parent = p;
109 4b576658 2013-09-23 0intro qlock(&p->lk);
110 6f4d00ee 2013-09-23 0intro assert(mode == OReadOnly || p->mode == OReadWrite);
111 6f4d00ee 2013-09-23 0intro p->ref++;
112 4b576658 2013-09-23 0intro qunlock(&p->lk);
114 6f4d00ee 2013-09-23 0intro r->epoch = epoch;
115 6f4d00ee 2013-09-23 0intro // consPrint("sourceAlloc: have %V be.%d fse.%d %s\n", b->score,
116 6f4d00ee 2013-09-23 0intro // b->l.epoch, r->fs->ehi, mode == OReadWrite? "rw": "ro");
117 6f4d00ee 2013-09-23 0intro memmove(r->score, b->score, VtScoreSize);
118 6f4d00ee 2013-09-23 0intro r->scoreEpoch = b->l.epoch;
119 6f4d00ee 2013-09-23 0intro r->offset = offset;
120 6f4d00ee 2013-09-23 0intro r->epb = epb;
121 6f4d00ee 2013-09-23 0intro r->tag = b->l.tag;
123 6f4d00ee 2013-09-23 0intro // consPrint("%s: sourceAlloc: %p -> %V %d\n", r, r->score, r->offset);
125 6f4d00ee 2013-09-23 0intro return r;
127 6f4d00ee 2013-09-23 0intro free(pname);
128 4b576658 2013-09-23 0intro werrstr(EBadEntry);
129 6f4d00ee 2013-09-23 0intro return nil;
133 6f4d00ee 2013-09-23 0intro sourceRoot(Fs *fs, u32int addr, int mode)
135 6f4d00ee 2013-09-23 0intro Source *r;
136 6f4d00ee 2013-09-23 0intro Block *b;
138 6f4d00ee 2013-09-23 0intro b = cacheLocalData(fs->cache, addr, BtDir, RootTag, mode, 0);
139 6f4d00ee 2013-09-23 0intro if(b == nil)
140 6f4d00ee 2013-09-23 0intro return nil;
142 6f4d00ee 2013-09-23 0intro if(mode == OReadWrite && b->l.epoch != fs->ehi){
143 6f4d00ee 2013-09-23 0intro consPrint("sourceRoot: fs->ehi = %ud, b->l = %L\n",
144 6f4d00ee 2013-09-23 0intro fs->ehi, &b->l);
145 6f4d00ee 2013-09-23 0intro blockPut(b);
146 4b576658 2013-09-23 0intro werrstr(EBadRoot);
147 6f4d00ee 2013-09-23 0intro return nil;
150 6f4d00ee 2013-09-23 0intro r = sourceAlloc(fs, b, nil, 0, mode, 0);
151 6f4d00ee 2013-09-23 0intro blockPut(b);
152 6f4d00ee 2013-09-23 0intro return r;
156 6f4d00ee 2013-09-23 0intro sourceOpen(Source *r, ulong offset, int mode, int issnapshot)
158 6f4d00ee 2013-09-23 0intro ulong bn;
159 6f4d00ee 2013-09-23 0intro Block *b;
161 6f4d00ee 2013-09-23 0intro assert(sourceIsLocked(r));
162 6f4d00ee 2013-09-23 0intro if(r->mode == OReadWrite)
163 6f4d00ee 2013-09-23 0intro assert(r->epoch == r->b->l.epoch);
164 6f4d00ee 2013-09-23 0intro if(!r->dir){
165 4b576658 2013-09-23 0intro werrstr(ENotDir);
166 6f4d00ee 2013-09-23 0intro return nil;
169 6f4d00ee 2013-09-23 0intro bn = offset/(r->dsize/VtEntrySize);
171 6f4d00ee 2013-09-23 0intro b = sourceBlock(r, bn, mode);
172 6f4d00ee 2013-09-23 0intro if(b == nil)
173 6f4d00ee 2013-09-23 0intro return nil;
174 6f4d00ee 2013-09-23 0intro r = sourceAlloc(r->fs, b, r, offset, mode, issnapshot);
175 6f4d00ee 2013-09-23 0intro blockPut(b);
176 6f4d00ee 2013-09-23 0intro return r;
180 6f4d00ee 2013-09-23 0intro sourceCreate(Source *r, int dsize, int dir, u32int offset)
182 6f4d00ee 2013-09-23 0intro int i, epb, psize;
183 6f4d00ee 2013-09-23 0intro u32int bn, size;
184 6f4d00ee 2013-09-23 0intro Block *b;
186 6f4d00ee 2013-09-23 0intro Source *rr;
188 6f4d00ee 2013-09-23 0intro assert(sourceIsLocked(r));
190 6f4d00ee 2013-09-23 0intro if(!r->dir){
191 4b576658 2013-09-23 0intro werrstr(ENotDir);
192 6f4d00ee 2013-09-23 0intro return nil;
195 6f4d00ee 2013-09-23 0intro epb = r->dsize/VtEntrySize;
196 6f4d00ee 2013-09-23 0intro psize = (dsize/VtScoreSize)*VtScoreSize;
198 6f4d00ee 2013-09-23 0intro size = sourceGetDirSize(r);
199 6f4d00ee 2013-09-23 0intro if(offset == 0){
201 6f4d00ee 2013-09-23 0intro * look at a random block to see if we can find an empty entry
203 6f4d00ee 2013-09-23 0intro offset = lnrand(size+1);
204 6f4d00ee 2013-09-23 0intro offset -= offset % epb;
207 6f4d00ee 2013-09-23 0intro /* try the given block and then try the last block */
209 6f4d00ee 2013-09-23 0intro bn = offset/epb;
210 6f4d00ee 2013-09-23 0intro b = sourceBlock(r, bn, OReadWrite);
211 6f4d00ee 2013-09-23 0intro if(b == nil)
212 6f4d00ee 2013-09-23 0intro return nil;
213 6f4d00ee 2013-09-23 0intro for(i=offset%r->epb; i<epb; i++){
214 6f4d00ee 2013-09-23 0intro entryUnpack(&e, b->data, i);
215 6f4d00ee 2013-09-23 0intro if((e.flags&VtEntryActive) == 0 && e.gen != ~0)
216 6f4d00ee 2013-09-23 0intro goto Found;
218 6f4d00ee 2013-09-23 0intro blockPut(b);
219 6f4d00ee 2013-09-23 0intro if(offset == size){
220 6f4d00ee 2013-09-23 0intro fprint(2, "sourceCreate: cannot happen\n");
221 4b576658 2013-09-23 0intro werrstr("sourceCreate: cannot happen");
222 6f4d00ee 2013-09-23 0intro return nil;
224 6f4d00ee 2013-09-23 0intro offset = size;
228 6f4d00ee 2013-09-23 0intro /* found an entry - gen already set */
229 6f4d00ee 2013-09-23 0intro e.psize = psize;
230 6f4d00ee 2013-09-23 0intro e.dsize = dsize;
231 6f4d00ee 2013-09-23 0intro assert(psize && dsize);
232 6f4d00ee 2013-09-23 0intro e.flags = VtEntryActive;
234 4b576658 2013-09-23 0intro e.flags |= _VtEntryDir;
235 6f4d00ee 2013-09-23 0intro e.depth = 0;
236 6f4d00ee 2013-09-23 0intro e.size = 0;
237 4b576658 2013-09-23 0intro memmove(e.score, vtzeroscore, VtScoreSize);
238 6f4d00ee 2013-09-23 0intro e.tag = 0;
239 6f4d00ee 2013-09-23 0intro e.snap = 0;
240 6f4d00ee 2013-09-23 0intro e.archive = 0;
241 6f4d00ee 2013-09-23 0intro entryPack(&e, b->data, i);
242 6f4d00ee 2013-09-23 0intro blockDirty(b);
244 6f4d00ee 2013-09-23 0intro offset = bn*epb + i;
245 6f4d00ee 2013-09-23 0intro if(offset+1 > size){
246 6f4d00ee 2013-09-23 0intro if(!sourceSetDirSize(r, offset+1)){
247 6f4d00ee 2013-09-23 0intro blockPut(b);
248 6f4d00ee 2013-09-23 0intro return nil;
252 6f4d00ee 2013-09-23 0intro rr = sourceAlloc(r->fs, b, r, offset, OReadWrite, 0);
253 6f4d00ee 2013-09-23 0intro blockPut(b);
254 6f4d00ee 2013-09-23 0intro return rr;
257 6f4d00ee 2013-09-23 0intro static int
258 6f4d00ee 2013-09-23 0intro sourceKill(Source *r, int doremove)
261 6f4d00ee 2013-09-23 0intro Block *b;
262 6f4d00ee 2013-09-23 0intro u32int addr;
263 6f4d00ee 2013-09-23 0intro u32int tag;
264 6f4d00ee 2013-09-23 0intro int type;
266 6f4d00ee 2013-09-23 0intro assert(sourceIsLocked(r));
267 6f4d00ee 2013-09-23 0intro b = sourceLoad(r, &e);
268 6f4d00ee 2013-09-23 0intro if(b == nil)
269 6f4d00ee 2013-09-23 0intro return 0;
271 6f4d00ee 2013-09-23 0intro assert(b->l.epoch == r->fs->ehi);
273 6f4d00ee 2013-09-23 0intro if(doremove==0 && e.size == 0){
274 6f4d00ee 2013-09-23 0intro /* already truncated */
275 6f4d00ee 2013-09-23 0intro blockPut(b);
276 6f4d00ee 2013-09-23 0intro return 1;
279 6f4d00ee 2013-09-23 0intro /* remember info on link we are removing */
280 6f4d00ee 2013-09-23 0intro addr = globalToLocal(e.score);
281 6f4d00ee 2013-09-23 0intro type = entryType(&e);
282 6f4d00ee 2013-09-23 0intro tag = e.tag;
284 6f4d00ee 2013-09-23 0intro if(doremove){
285 6f4d00ee 2013-09-23 0intro if(e.gen != ~0)
287 6f4d00ee 2013-09-23 0intro e.dsize = 0;
288 6f4d00ee 2013-09-23 0intro e.psize = 0;
289 6f4d00ee 2013-09-23 0intro e.flags = 0;
291 6f4d00ee 2013-09-23 0intro e.flags &= ~VtEntryLocal;
293 6f4d00ee 2013-09-23 0intro e.depth = 0;
294 6f4d00ee 2013-09-23 0intro e.size = 0;
295 6f4d00ee 2013-09-23 0intro e.tag = 0;
296 4b576658 2013-09-23 0intro memmove(e.score, vtzeroscore, VtScoreSize);
297 6f4d00ee 2013-09-23 0intro entryPack(&e, b->data, r->offset % r->epb);
298 6f4d00ee 2013-09-23 0intro blockDirty(b);
299 6f4d00ee 2013-09-23 0intro if(addr != NilBlock)
300 6f4d00ee 2013-09-23 0intro blockRemoveLink(b, addr, type, tag, 1);
301 6f4d00ee 2013-09-23 0intro blockPut(b);
303 6f4d00ee 2013-09-23 0intro if(doremove){
304 6f4d00ee 2013-09-23 0intro sourceUnlock(r);
305 6f4d00ee 2013-09-23 0intro sourceClose(r);
308 6f4d00ee 2013-09-23 0intro return 1;
312 6f4d00ee 2013-09-23 0intro sourceRemove(Source *r)
314 6f4d00ee 2013-09-23 0intro return sourceKill(r, 1);
318 6f4d00ee 2013-09-23 0intro sourceTruncate(Source *r)
320 6f4d00ee 2013-09-23 0intro return sourceKill(r, 0);
324 6f4d00ee 2013-09-23 0intro sourceGetSize(Source *r)
327 6f4d00ee 2013-09-23 0intro Block *b;
329 6f4d00ee 2013-09-23 0intro assert(sourceIsLocked(r));
330 6f4d00ee 2013-09-23 0intro b = sourceLoad(r, &e);
331 6f4d00ee 2013-09-23 0intro if(b == nil)
332 6f4d00ee 2013-09-23 0intro return 0;
333 6f4d00ee 2013-09-23 0intro blockPut(b);
335 6f4d00ee 2013-09-23 0intro return e.size;
338 6f4d00ee 2013-09-23 0intro static int
339 6f4d00ee 2013-09-23 0intro sourceShrinkSize(Source *r, Entry *e, uvlong size)
341 6f4d00ee 2013-09-23 0intro int i, type, ppb;
342 6f4d00ee 2013-09-23 0intro uvlong ptrsz;
343 6f4d00ee 2013-09-23 0intro u32int addr;
344 6f4d00ee 2013-09-23 0intro uchar score[VtScoreSize];
345 6f4d00ee 2013-09-23 0intro Block *b;
347 6f4d00ee 2013-09-23 0intro type = entryType(e);
348 6f4d00ee 2013-09-23 0intro b = cacheGlobal(r->fs->cache, e->score, type, e->tag, OReadWrite);
349 6f4d00ee 2013-09-23 0intro if(b == nil)
350 6f4d00ee 2013-09-23 0intro return 0;
352 6f4d00ee 2013-09-23 0intro ptrsz = e->dsize;
353 6f4d00ee 2013-09-23 0intro ppb = e->psize/VtScoreSize;
354 6f4d00ee 2013-09-23 0intro for(i=0; i+1<e->depth; i++)
355 6f4d00ee 2013-09-23 0intro ptrsz *= ppb;
357 6f4d00ee 2013-09-23 0intro while(type&BtLevelMask){
358 6f4d00ee 2013-09-23 0intro if(b->addr == NilBlock || b->l.epoch != r->fs->ehi){
359 6f4d00ee 2013-09-23 0intro /* not worth copying the block just so we can zero some of it */
360 6f4d00ee 2013-09-23 0intro blockPut(b);
361 6f4d00ee 2013-09-23 0intro return 0;
365 6f4d00ee 2013-09-23 0intro * invariant: each pointer in the tree rooted at b accounts for ptrsz bytes
368 6f4d00ee 2013-09-23 0intro /* zero the pointers to unnecessary blocks */
369 6f4d00ee 2013-09-23 0intro i = (size+ptrsz-1)/ptrsz;
370 6f4d00ee 2013-09-23 0intro for(; i<ppb; i++){
371 6f4d00ee 2013-09-23 0intro addr = globalToLocal(b->data+i*VtScoreSize);
372 4b576658 2013-09-23 0intro memmove(b->data+i*VtScoreSize, vtzeroscore, VtScoreSize);
373 6f4d00ee 2013-09-23 0intro blockDirty(b);
374 6f4d00ee 2013-09-23 0intro if(addr != NilBlock)
375 6f4d00ee 2013-09-23 0intro blockRemoveLink(b, addr, type-1, e->tag, 1);
378 6f4d00ee 2013-09-23 0intro /* recurse (go around again) on the partially necessary block */
379 6f4d00ee 2013-09-23 0intro i = size/ptrsz;
380 6f4d00ee 2013-09-23 0intro size = size%ptrsz;
381 6f4d00ee 2013-09-23 0intro if(size == 0){
382 6f4d00ee 2013-09-23 0intro blockPut(b);
383 6f4d00ee 2013-09-23 0intro return 1;
385 6f4d00ee 2013-09-23 0intro ptrsz /= ppb;
387 6f4d00ee 2013-09-23 0intro memmove(score, b->data+i*VtScoreSize, VtScoreSize);
388 6f4d00ee 2013-09-23 0intro blockPut(b);
389 6f4d00ee 2013-09-23 0intro b = cacheGlobal(r->fs->cache, score, type, e->tag, OReadWrite);
390 6f4d00ee 2013-09-23 0intro if(b == nil)
391 6f4d00ee 2013-09-23 0intro return 0;
394 6f4d00ee 2013-09-23 0intro if(b->addr == NilBlock || b->l.epoch != r->fs->ehi){
395 6f4d00ee 2013-09-23 0intro blockPut(b);
396 6f4d00ee 2013-09-23 0intro return 0;
400 6f4d00ee 2013-09-23 0intro * No one ever truncates BtDir blocks.
402 6f4d00ee 2013-09-23 0intro if(type == BtData && e->dsize > size){
403 6f4d00ee 2013-09-23 0intro memset(b->data+size, 0, e->dsize-size);
404 6f4d00ee 2013-09-23 0intro blockDirty(b);
406 6f4d00ee 2013-09-23 0intro blockPut(b);
407 6f4d00ee 2013-09-23 0intro return 1;
411 6f4d00ee 2013-09-23 0intro sourceSetSize(Source *r, uvlong size)
413 6f4d00ee 2013-09-23 0intro int depth;
415 6f4d00ee 2013-09-23 0intro Block *b;
417 6f4d00ee 2013-09-23 0intro assert(sourceIsLocked(r));
418 6f4d00ee 2013-09-23 0intro if(size == 0)
419 6f4d00ee 2013-09-23 0intro return sourceTruncate(r);
421 6f4d00ee 2013-09-23 0intro if(size > VtMaxFileSize || size > ((uvlong)MaxBlock)*r->dsize){
422 4b576658 2013-09-23 0intro werrstr(ETooBig);
423 6f4d00ee 2013-09-23 0intro return 0;
426 6f4d00ee 2013-09-23 0intro b = sourceLoad(r, &e);
427 6f4d00ee 2013-09-23 0intro if(b == nil)
428 6f4d00ee 2013-09-23 0intro return 0;
430 6f4d00ee 2013-09-23 0intro /* quick out */
431 6f4d00ee 2013-09-23 0intro if(e.size == size){
432 6f4d00ee 2013-09-23 0intro blockPut(b);
433 6f4d00ee 2013-09-23 0intro return 1;
436 6f4d00ee 2013-09-23 0intro depth = sizeToDepth(size, e.psize, e.dsize);
438 6f4d00ee 2013-09-23 0intro if(depth < e.depth){
439 6f4d00ee 2013-09-23 0intro if(!sourceShrinkDepth(r, b, &e, depth)){
440 6f4d00ee 2013-09-23 0intro blockPut(b);
441 6f4d00ee 2013-09-23 0intro return 0;
443 6f4d00ee 2013-09-23 0intro }else if(depth > e.depth){
444 6f4d00ee 2013-09-23 0intro if(!sourceGrowDepth(r, b, &e, depth)){
445 6f4d00ee 2013-09-23 0intro blockPut(b);
446 6f4d00ee 2013-09-23 0intro return 0;
450 6f4d00ee 2013-09-23 0intro if(size < e.size)
451 6f4d00ee 2013-09-23 0intro sourceShrinkSize(r, &e, size);
453 6f4d00ee 2013-09-23 0intro e.size = size;
454 6f4d00ee 2013-09-23 0intro entryPack(&e, b->data, r->offset % r->epb);
455 6f4d00ee 2013-09-23 0intro blockDirty(b);
456 6f4d00ee 2013-09-23 0intro blockPut(b);
458 6f4d00ee 2013-09-23 0intro return 1;
462 6f4d00ee 2013-09-23 0intro sourceSetDirSize(Source *r, ulong ds)
464 6f4d00ee 2013-09-23 0intro uvlong size;
467 6f4d00ee 2013-09-23 0intro assert(sourceIsLocked(r));
468 6f4d00ee 2013-09-23 0intro epb = r->dsize/VtEntrySize;
470 6f4d00ee 2013-09-23 0intro size = (uvlong)r->dsize*(ds/epb);
471 6f4d00ee 2013-09-23 0intro size += VtEntrySize*(ds%epb);
472 6f4d00ee 2013-09-23 0intro return sourceSetSize(r, size);
476 6f4d00ee 2013-09-23 0intro sourceGetDirSize(Source *r)
478 6f4d00ee 2013-09-23 0intro ulong ds;
479 6f4d00ee 2013-09-23 0intro uvlong size;
482 6f4d00ee 2013-09-23 0intro assert(sourceIsLocked(r));
483 6f4d00ee 2013-09-23 0intro epb = r->dsize/VtEntrySize;
485 6f4d00ee 2013-09-23 0intro size = sourceGetSize(r);
486 6f4d00ee 2013-09-23 0intro ds = epb*(size/r->dsize);
487 6f4d00ee 2013-09-23 0intro ds += (size%r->dsize)/VtEntrySize;
488 6f4d00ee 2013-09-23 0intro return ds;
492 6f4d00ee 2013-09-23 0intro sourceGetEntry(Source *r, Entry *e)
494 6f4d00ee 2013-09-23 0intro Block *b;
496 6f4d00ee 2013-09-23 0intro assert(sourceIsLocked(r));
497 6f4d00ee 2013-09-23 0intro b = sourceLoad(r, e);
498 6f4d00ee 2013-09-23 0intro if(b == nil)
499 6f4d00ee 2013-09-23 0intro return 0;
500 6f4d00ee 2013-09-23 0intro blockPut(b);
502 6f4d00ee 2013-09-23 0intro return 1;
506 6f4d00ee 2013-09-23 0intro * Must be careful with this. Doesn't record
507 6f4d00ee 2013-09-23 0intro * dependencies, so don't introduce any!
510 6f4d00ee 2013-09-23 0intro sourceSetEntry(Source *r, Entry *e)
512 6f4d00ee 2013-09-23 0intro Block *b;
513 6f4d00ee 2013-09-23 0intro Entry oe;
515 6f4d00ee 2013-09-23 0intro assert(sourceIsLocked(r));
516 6f4d00ee 2013-09-23 0intro b = sourceLoad(r, &oe);
517 6f4d00ee 2013-09-23 0intro if(b == nil)
518 6f4d00ee 2013-09-23 0intro return 0;
519 6f4d00ee 2013-09-23 0intro entryPack(e, b->data, r->offset%r->epb);
520 6f4d00ee 2013-09-23 0intro blockDirty(b);
521 6f4d00ee 2013-09-23 0intro blockPut(b);
523 6f4d00ee 2013-09-23 0intro return 1;
526 6f4d00ee 2013-09-23 0intro static Block *
527 6f4d00ee 2013-09-23 0intro blockWalk(Block *p, int index, int mode, Fs *fs, Entry *e)
529 6f4d00ee 2013-09-23 0intro Block *b;
530 6f4d00ee 2013-09-23 0intro Cache *c;
531 6f4d00ee 2013-09-23 0intro u32int addr;
532 6f4d00ee 2013-09-23 0intro int type;
533 6f4d00ee 2013-09-23 0intro uchar oscore[VtScoreSize], score[VtScoreSize];
534 6f4d00ee 2013-09-23 0intro Entry oe;
536 6f4d00ee 2013-09-23 0intro c = fs->cache;
538 6f4d00ee 2013-09-23 0intro if((p->l.type & BtLevelMask) == 0){
539 6f4d00ee 2013-09-23 0intro assert(p->l.type == BtDir);
540 6f4d00ee 2013-09-23 0intro type = entryType(e);
541 6f4d00ee 2013-09-23 0intro b = cacheGlobal(c, e->score, type, e->tag, mode);
543 6f4d00ee 2013-09-23 0intro type = p->l.type - 1;
544 6f4d00ee 2013-09-23 0intro b = cacheGlobal(c, p->data + index*VtScoreSize, type, e->tag, mode);
548 6f4d00ee 2013-09-23 0intro b->pc = getcallerpc(&p);
550 6f4d00ee 2013-09-23 0intro if(b == nil || mode == OReadOnly)
551 6f4d00ee 2013-09-23 0intro return b;
553 6f4d00ee 2013-09-23 0intro if(p->l.epoch != fs->ehi){
554 6f4d00ee 2013-09-23 0intro fprint(2, "blockWalk: parent not writable\n");
557 6f4d00ee 2013-09-23 0intro if(b->l.epoch == fs->ehi)
558 6f4d00ee 2013-09-23 0intro return b;
563 6f4d00ee 2013-09-23 0intro * Copy on write.
565 6f4d00ee 2013-09-23 0intro if(e->tag == 0){
566 6f4d00ee 2013-09-23 0intro assert(p->l.type == BtDir);
567 6f4d00ee 2013-09-23 0intro e->tag = tagGen();
568 6f4d00ee 2013-09-23 0intro e->flags |= VtEntryLocal;
571 6f4d00ee 2013-09-23 0intro addr = b->addr;
572 6f4d00ee 2013-09-23 0intro b = blockCopy(b, e->tag, fs->ehi, fs->elo);
573 6f4d00ee 2013-09-23 0intro if(b == nil)
574 6f4d00ee 2013-09-23 0intro return nil;
576 6f4d00ee 2013-09-23 0intro b->pc = getcallerpc(&p);
577 6f4d00ee 2013-09-23 0intro assert(b->l.epoch == fs->ehi);
579 6f4d00ee 2013-09-23 0intro blockDirty(b);
580 6f4d00ee 2013-09-23 0intro memmove(score, b->score, VtScoreSize);
581 6f4d00ee 2013-09-23 0intro if(p->l.type == BtDir){
582 6f4d00ee 2013-09-23 0intro memmove(e->score, b->score, VtScoreSize);
583 6f4d00ee 2013-09-23 0intro entryPack(e, p->data, index);
584 6f4d00ee 2013-09-23 0intro blockDependency(p, b, index, nil, &oe);
586 6f4d00ee 2013-09-23 0intro memmove(oscore, p->data+index*VtScoreSize, VtScoreSize);
587 6f4d00ee 2013-09-23 0intro memmove(p->data+index*VtScoreSize, b->score, VtScoreSize);
588 6f4d00ee 2013-09-23 0intro blockDependency(p, b, index, oscore, nil);
590 6f4d00ee 2013-09-23 0intro blockDirty(p);
592 6f4d00ee 2013-09-23 0intro if(addr != NilBlock)
593 6f4d00ee 2013-09-23 0intro blockRemoveLink(p, addr, type, e->tag, 0);
595 6f4d00ee 2013-09-23 0intro return b;
599 6f4d00ee 2013-09-23 0intro * Change the depth of the source r.
600 6f4d00ee 2013-09-23 0intro * The entry e for r is contained in block p.
602 6f4d00ee 2013-09-23 0intro static int
603 6f4d00ee 2013-09-23 0intro sourceGrowDepth(Source *r, Block *p, Entry *e, int depth)
605 6f4d00ee 2013-09-23 0intro Block *b, *bb;
606 6f4d00ee 2013-09-23 0intro u32int tag;
607 6f4d00ee 2013-09-23 0intro int type;
608 6f4d00ee 2013-09-23 0intro Entry oe;
610 6f4d00ee 2013-09-23 0intro assert(sourceIsLocked(r));
611 6f4d00ee 2013-09-23 0intro assert(depth <= VtPointerDepth);
613 6f4d00ee 2013-09-23 0intro type = entryType(e);
614 6f4d00ee 2013-09-23 0intro b = cacheGlobal(r->fs->cache, e->score, type, e->tag, OReadWrite);
615 6f4d00ee 2013-09-23 0intro if(b == nil)
616 6f4d00ee 2013-09-23 0intro return 0;
618 6f4d00ee 2013-09-23 0intro tag = e->tag;
619 6f4d00ee 2013-09-23 0intro if(tag == 0)
620 6f4d00ee 2013-09-23 0intro tag = tagGen();
625 6f4d00ee 2013-09-23 0intro * Keep adding layers until we get to the right depth
626 6f4d00ee 2013-09-23 0intro * or an error occurs.
628 6f4d00ee 2013-09-23 0intro while(e->depth < depth){
629 6f4d00ee 2013-09-23 0intro bb = cacheAllocBlock(r->fs->cache, type+1, tag, r->fs->ehi, r->fs->elo);
630 6f4d00ee 2013-09-23 0intro if(bb == nil)
632 6f4d00ee 2013-09-23 0intro //fprint(2, "alloc %lux grow %V\n", bb->addr, b->score);
633 6f4d00ee 2013-09-23 0intro memmove(bb->data, b->score, VtScoreSize);
634 6f4d00ee 2013-09-23 0intro memmove(e->score, bb->score, VtScoreSize);
635 6f4d00ee 2013-09-23 0intro e->depth++;
637 6f4d00ee 2013-09-23 0intro e->tag = tag;
638 6f4d00ee 2013-09-23 0intro e->flags |= VtEntryLocal;
639 4b576658 2013-09-23 0intro blockDependency(bb, b, 0, vtzeroscore, nil);
640 6f4d00ee 2013-09-23 0intro blockPut(b);
642 6f4d00ee 2013-09-23 0intro blockDirty(b);
645 6f4d00ee 2013-09-23 0intro entryPack(e, p->data, r->offset % r->epb);
646 6f4d00ee 2013-09-23 0intro blockDependency(p, b, r->offset % r->epb, nil, &oe);
647 6f4d00ee 2013-09-23 0intro blockPut(b);
648 6f4d00ee 2013-09-23 0intro blockDirty(p);
650 6f4d00ee 2013-09-23 0intro return e->depth == depth;
653 6f4d00ee 2013-09-23 0intro static int
654 6f4d00ee 2013-09-23 0intro sourceShrinkDepth(Source *r, Block *p, Entry *e, int depth)
656 6f4d00ee 2013-09-23 0intro Block *b, *nb, *ob, *rb;
657 6f4d00ee 2013-09-23 0intro u32int tag;
658 6f4d00ee 2013-09-23 0intro int type, d;
659 6f4d00ee 2013-09-23 0intro Entry oe;
661 6f4d00ee 2013-09-23 0intro assert(sourceIsLocked(r));
662 6f4d00ee 2013-09-23 0intro assert(depth <= VtPointerDepth);
664 6f4d00ee 2013-09-23 0intro type = entryType(e);
665 6f4d00ee 2013-09-23 0intro rb = cacheGlobal(r->fs->cache, e->score, type, e->tag, OReadWrite);
666 6f4d00ee 2013-09-23 0intro if(rb == nil)
667 6f4d00ee 2013-09-23 0intro return 0;
669 6f4d00ee 2013-09-23 0intro tag = e->tag;
670 6f4d00ee 2013-09-23 0intro if(tag == 0)
671 6f4d00ee 2013-09-23 0intro tag = tagGen();
674 6f4d00ee 2013-09-23 0intro * Walk down to the new root block.
675 6f4d00ee 2013-09-23 0intro * We may stop early, but something is better than nothing.
679 6f4d00ee 2013-09-23 0intro ob = nil;
681 6f4d00ee 2013-09-23 0intro /* BUG: explain type++. i think it is a real bug */
682 6f4d00ee 2013-09-23 0intro for(d=e->depth; d > depth; d--, type++){
683 6f4d00ee 2013-09-23 0intro nb = cacheGlobal(r->fs->cache, b->data, type-1, tag, OReadWrite);
684 6f4d00ee 2013-09-23 0intro if(nb == nil)
686 6f4d00ee 2013-09-23 0intro if(ob!=nil && ob!=rb)
687 6f4d00ee 2013-09-23 0intro blockPut(ob);
692 6f4d00ee 2013-09-23 0intro if(b == rb){
693 6f4d00ee 2013-09-23 0intro blockPut(rb);
694 6f4d00ee 2013-09-23 0intro return 0;
698 6f4d00ee 2013-09-23 0intro * Right now, e points at the root block rb, b is the new root block,
699 6f4d00ee 2013-09-23 0intro * and ob points at b. To update:
701 6f4d00ee 2013-09-23 0intro * (i) change e to point at b
702 6f4d00ee 2013-09-23 0intro * (ii) zero the pointer ob -> b
703 6f4d00ee 2013-09-23 0intro * (iii) free the root block
705 6f4d00ee 2013-09-23 0intro * p (the block containing e) must be written before
706 6f4d00ee 2013-09-23 0intro * anything else.
709 6f4d00ee 2013-09-23 0intro /* (i) */
710 6f4d00ee 2013-09-23 0intro e->depth = d;
711 6f4d00ee 2013-09-23 0intro /* might have been local and now global; reverse cannot happen */
712 6f4d00ee 2013-09-23 0intro if(globalToLocal(b->score) == NilBlock)
713 6f4d00ee 2013-09-23 0intro e->flags &= ~VtEntryLocal;
714 6f4d00ee 2013-09-23 0intro memmove(e->score, b->score, VtScoreSize);
715 6f4d00ee 2013-09-23 0intro entryPack(e, p->data, r->offset % r->epb);
716 6f4d00ee 2013-09-23 0intro blockDependency(p, b, r->offset % r->epb, nil, &oe);
717 6f4d00ee 2013-09-23 0intro blockDirty(p);
719 6f4d00ee 2013-09-23 0intro /* (ii) */
720 4b576658 2013-09-23 0intro memmove(ob->data, vtzeroscore, VtScoreSize);
721 6f4d00ee 2013-09-23 0intro blockDependency(ob, p, 0, b->score, nil);
722 6f4d00ee 2013-09-23 0intro blockDirty(ob);
724 6f4d00ee 2013-09-23 0intro /* (iii) */
725 6f4d00ee 2013-09-23 0intro if(rb->addr != NilBlock)
726 6f4d00ee 2013-09-23 0intro blockRemoveLink(p, rb->addr, rb->l.type, rb->l.tag, 1);
728 6f4d00ee 2013-09-23 0intro blockPut(rb);
729 6f4d00ee 2013-09-23 0intro if(ob!=nil && ob!=rb)
730 6f4d00ee 2013-09-23 0intro blockPut(ob);
731 6f4d00ee 2013-09-23 0intro blockPut(b);
733 6f4d00ee 2013-09-23 0intro return d == depth;
737 6f4d00ee 2013-09-23 0intro * Normally we return the block at the given number.
738 6f4d00ee 2013-09-23 0intro * If early is set, we stop earlier in the tree. Setting early
739 6f4d00ee 2013-09-23 0intro * to 1 gives us the block that contains the pointer to bn.
742 6f4d00ee 2013-09-23 0intro _sourceBlock(Source *r, ulong bn, int mode, int early, ulong tag)
744 6f4d00ee 2013-09-23 0intro Block *b, *bb;
745 6f4d00ee 2013-09-23 0intro int index[VtPointerDepth+1];
747 6f4d00ee 2013-09-23 0intro int i, np;
750 6f4d00ee 2013-09-23 0intro assert(sourceIsLocked(r));
751 6f4d00ee 2013-09-23 0intro assert(bn != NilBlock);
753 6f4d00ee 2013-09-23 0intro /* mode for intermediate block */
754 6f4d00ee 2013-09-23 0intro m = mode;
755 6f4d00ee 2013-09-23 0intro if(m == OOverWrite)
756 6f4d00ee 2013-09-23 0intro m = OReadWrite;
758 6f4d00ee 2013-09-23 0intro b = sourceLoad(r, &e);
759 6f4d00ee 2013-09-23 0intro if(b == nil)
760 6f4d00ee 2013-09-23 0intro return nil;
761 6f4d00ee 2013-09-23 0intro if(r->issnapshot && (e.flags & VtEntryNoArchive)){
762 6f4d00ee 2013-09-23 0intro blockPut(b);
763 4b576658 2013-09-23 0intro werrstr(ENotArchived);
764 6f4d00ee 2013-09-23 0intro return nil;
768 6f4d00ee 2013-09-23 0intro if(e.tag == 0)
769 6f4d00ee 2013-09-23 0intro e.tag = tag;
770 6f4d00ee 2013-09-23 0intro else if(e.tag != tag){
771 6f4d00ee 2013-09-23 0intro fprint(2, "tag mismatch\n");
772 4b576658 2013-09-23 0intro werrstr("tag mismatch");
773 6f4d00ee 2013-09-23 0intro goto Err;
777 6f4d00ee 2013-09-23 0intro np = e.psize/VtScoreSize;
778 6f4d00ee 2013-09-23 0intro memset(index, 0, sizeof(index));
779 6f4d00ee 2013-09-23 0intro for(i=0; bn > 0; i++){
780 6f4d00ee 2013-09-23 0intro if(i >= VtPointerDepth){
781 4b576658 2013-09-23 0intro werrstr(EBadAddr);
782 6f4d00ee 2013-09-23 0intro goto Err;
784 6f4d00ee 2013-09-23 0intro index[i] = bn % np;
785 6f4d00ee 2013-09-23 0intro bn /= np;
788 6f4d00ee 2013-09-23 0intro if(i > e.depth){
789 6f4d00ee 2013-09-23 0intro if(mode == OReadOnly){
790 4b576658 2013-09-23 0intro werrstr(EBadAddr);
791 6f4d00ee 2013-09-23 0intro goto Err;
793 6f4d00ee 2013-09-23 0intro if(!sourceGrowDepth(r, b, &e, i))
794 6f4d00ee 2013-09-23 0intro goto Err;
797 6f4d00ee 2013-09-23 0intro index[e.depth] = r->offset % r->epb;
799 6f4d00ee 2013-09-23 0intro for(i=e.depth; i>=early; i--){
800 6f4d00ee 2013-09-23 0intro bb = blockWalk(b, index[i], m, r->fs, &e);
801 6f4d00ee 2013-09-23 0intro if(bb == nil)
802 6f4d00ee 2013-09-23 0intro goto Err;
803 6f4d00ee 2013-09-23 0intro blockPut(b);
806 6f4d00ee 2013-09-23 0intro b->pc = getcallerpc(&r);
807 6f4d00ee 2013-09-23 0intro return b;
809 6f4d00ee 2013-09-23 0intro blockPut(b);
810 6f4d00ee 2013-09-23 0intro return nil;
814 6f4d00ee 2013-09-23 0intro sourceBlock(Source *r, ulong bn, int mode)
816 6f4d00ee 2013-09-23 0intro Block *b;
818 6f4d00ee 2013-09-23 0intro b = _sourceBlock(r, bn, mode, 0, 0);
820 6f4d00ee 2013-09-23 0intro b->pc = getcallerpc(&r);
821 6f4d00ee 2013-09-23 0intro return b;
825 6f4d00ee 2013-09-23 0intro sourceClose(Source *r)
827 6f4d00ee 2013-09-23 0intro if(r == nil)
829 4b576658 2013-09-23 0intro qlock(&r->lk);
830 6f4d00ee 2013-09-23 0intro r->ref--;
831 6f4d00ee 2013-09-23 0intro if(r->ref){
832 4b576658 2013-09-23 0intro qunlock(&r->lk);
835 6f4d00ee 2013-09-23 0intro assert(r->ref == 0);
836 4b576658 2013-09-23 0intro qunlock(&r->lk);
837 6f4d00ee 2013-09-23 0intro if(r->parent)
838 6f4d00ee 2013-09-23 0intro sourceClose(r->parent);
839 6f4d00ee 2013-09-23 0intro memset(r, ~0, sizeof(*r));
840 4b576658 2013-09-23 0intro vtfree(r);
844 6f4d00ee 2013-09-23 0intro * Retrieve the block containing the entry for r.
845 6f4d00ee 2013-09-23 0intro * If a snapshot has happened, we might need
846 6f4d00ee 2013-09-23 0intro * to get a new copy of the block. We avoid this
847 6f4d00ee 2013-09-23 0intro * in the common case by caching the score for
848 6f4d00ee 2013-09-23 0intro * the block and the last epoch in which it was valid.
850 6f4d00ee 2013-09-23 0intro * We use r->mode to tell the difference between active
851 6f4d00ee 2013-09-23 0intro * file system sources (OReadWrite) and sources for the
852 6f4d00ee 2013-09-23 0intro * snapshot file system (OReadOnly).
854 6f4d00ee 2013-09-23 0intro static Block*
855 6f4d00ee 2013-09-23 0intro sourceLoadBlock(Source *r, int mode)
857 6f4d00ee 2013-09-23 0intro u32int addr;
858 6f4d00ee 2013-09-23 0intro Block *b;
859 4b576658 2013-09-23 0intro char e[ERRMAX];
861 6f4d00ee 2013-09-23 0intro switch(r->mode){
863 6f4d00ee 2013-09-23 0intro assert(0);
864 6f4d00ee 2013-09-23 0intro case OReadWrite:
865 6f4d00ee 2013-09-23 0intro assert(r->mode == OReadWrite);
867 6f4d00ee 2013-09-23 0intro * This needn't be true -- we might bump the low epoch
868 6f4d00ee 2013-09-23 0intro * to reclaim some old blocks, but since this score is
869 6f4d00ee 2013-09-23 0intro * OReadWrite, the blocks must all still be open, so none
870 6f4d00ee 2013-09-23 0intro * are reclaimed. Thus it's okay that the epoch is so low.
871 6f4d00ee 2013-09-23 0intro * Proceed.
872 6f4d00ee 2013-09-23 0intro assert(r->epoch >= r->fs->elo);
874 6f4d00ee 2013-09-23 0intro if(r->epoch == r->fs->ehi){
875 6f4d00ee 2013-09-23 0intro b = cacheGlobal(r->fs->cache, r->score, BtDir, r->tag, OReadWrite);
876 6f4d00ee 2013-09-23 0intro if(b == nil)
877 6f4d00ee 2013-09-23 0intro return nil;
878 6f4d00ee 2013-09-23 0intro assert(r->epoch == b->l.epoch);
879 6f4d00ee 2013-09-23 0intro return b;
881 6f4d00ee 2013-09-23 0intro assert(r->parent != nil);
882 6f4d00ee 2013-09-23 0intro if(!sourceLock(r->parent, OReadWrite))
883 6f4d00ee 2013-09-23 0intro return nil;
884 6f4d00ee 2013-09-23 0intro b = sourceBlock(r->parent, r->offset/r->epb, OReadWrite);
885 6f4d00ee 2013-09-23 0intro sourceUnlock(r->parent);
886 6f4d00ee 2013-09-23 0intro if(b == nil)
887 6f4d00ee 2013-09-23 0intro return nil;
888 6f4d00ee 2013-09-23 0intro assert(b->l.epoch == r->fs->ehi);
889 6f4d00ee 2013-09-23 0intro // fprint(2, "sourceLoadBlock %p %V => %V\n", r, r->score, b->score);
890 6f4d00ee 2013-09-23 0intro memmove(r->score, b->score, VtScoreSize);
891 6f4d00ee 2013-09-23 0intro r->scoreEpoch = b->l.epoch;
892 6f4d00ee 2013-09-23 0intro r->tag = b->l.tag;
893 6f4d00ee 2013-09-23 0intro r->epoch = r->fs->ehi;
894 6f4d00ee 2013-09-23 0intro return b;
896 6f4d00ee 2013-09-23 0intro case OReadOnly:
897 6f4d00ee 2013-09-23 0intro addr = globalToLocal(r->score);
898 6f4d00ee 2013-09-23 0intro if(addr == NilBlock)
899 6f4d00ee 2013-09-23 0intro return cacheGlobal(r->fs->cache, r->score, BtDir, r->tag, mode);
901 6f4d00ee 2013-09-23 0intro b = cacheLocalData(r->fs->cache, addr, BtDir, r->tag, mode, r->scoreEpoch);
903 6f4d00ee 2013-09-23 0intro return b;
906 6f4d00ee 2013-09-23 0intro * If it failed because the epochs don't match, the block has been
907 6f4d00ee 2013-09-23 0intro * archived and reclaimed. Rewalk from the parent and get the
908 6f4d00ee 2013-09-23 0intro * new pointer. This can't happen in the OReadWrite case
909 6f4d00ee 2013-09-23 0intro * above because blocks in the current epoch don't get
910 6f4d00ee 2013-09-23 0intro * reclaimed. The fact that we're OReadOnly means we're
911 6f4d00ee 2013-09-23 0intro * a snapshot. (Or else the file system is read-only, but then
912 6f4d00ee 2013-09-23 0intro * the archiver isn't going around deleting blocks.)
914 4b576658 2013-09-23 0intro rerrstr(e, sizeof e);
915 4b576658 2013-09-23 0intro if(strcmp(e, ELabelMismatch) == 0){
916 6f4d00ee 2013-09-23 0intro if(!sourceLock(r->parent, OReadOnly))
917 6f4d00ee 2013-09-23 0intro return nil;
918 6f4d00ee 2013-09-23 0intro b = sourceBlock(r->parent, r->offset/r->epb, OReadOnly);
919 6f4d00ee 2013-09-23 0intro sourceUnlock(r->parent);
921 6f4d00ee 2013-09-23 0intro fprint(2, "sourceAlloc: lost %V found %V\n",
922 6f4d00ee 2013-09-23 0intro r->score, b->score);
923 6f4d00ee 2013-09-23 0intro memmove(r->score, b->score, VtScoreSize);
924 6f4d00ee 2013-09-23 0intro r->scoreEpoch = b->l.epoch;
925 6f4d00ee 2013-09-23 0intro return b;
928 6f4d00ee 2013-09-23 0intro return nil;
933 6f4d00ee 2013-09-23 0intro sourceLock(Source *r, int mode)
935 6f4d00ee 2013-09-23 0intro Block *b;
937 6f4d00ee 2013-09-23 0intro if(mode == -1)
938 6f4d00ee 2013-09-23 0intro mode = r->mode;
940 6f4d00ee 2013-09-23 0intro b = sourceLoadBlock(r, mode);
941 6f4d00ee 2013-09-23 0intro if(b == nil)
942 6f4d00ee 2013-09-23 0intro return 0;
944 6f4d00ee 2013-09-23 0intro * The fact that we are holding b serves as the
945 6f4d00ee 2013-09-23 0intro * lock entitling us to write to r->b.
947 6f4d00ee 2013-09-23 0intro assert(r->b == nil);
948 6f4d00ee 2013-09-23 0intro r->b = b;
949 6f4d00ee 2013-09-23 0intro if(r->mode == OReadWrite)
950 6f4d00ee 2013-09-23 0intro assert(r->epoch == r->b->l.epoch);
951 6f4d00ee 2013-09-23 0intro return 1;
955 6f4d00ee 2013-09-23 0intro * Lock two (usually sibling) sources. This needs special care
956 6f4d00ee 2013-09-23 0intro * because the Entries for both sources might be in the same block.
957 6f4d00ee 2013-09-23 0intro * We also try to lock blocks in left-to-right order within the tree.
960 6f4d00ee 2013-09-23 0intro sourceLock2(Source *r, Source *rr, int mode)
962 6f4d00ee 2013-09-23 0intro Block *b, *bb;
964 6f4d00ee 2013-09-23 0intro if(rr == nil)
965 6f4d00ee 2013-09-23 0intro return sourceLock(r, mode);
967 6f4d00ee 2013-09-23 0intro if(mode == -1)
968 6f4d00ee 2013-09-23 0intro mode = r->mode;
970 6f4d00ee 2013-09-23 0intro if(r->parent==rr->parent && r->offset/r->epb == rr->offset/rr->epb){
971 6f4d00ee 2013-09-23 0intro b = sourceLoadBlock(r, mode);
972 6f4d00ee 2013-09-23 0intro if(b == nil)
973 6f4d00ee 2013-09-23 0intro return 0;
974 6f4d00ee 2013-09-23 0intro if(memcmp(r->score, rr->score, VtScoreSize) != 0){
975 6f4d00ee 2013-09-23 0intro memmove(rr->score, b->score, VtScoreSize);
976 6f4d00ee 2013-09-23 0intro rr->scoreEpoch = b->l.epoch;
977 6f4d00ee 2013-09-23 0intro rr->tag = b->l.tag;
978 6f4d00ee 2013-09-23 0intro rr->epoch = rr->fs->ehi;
980 6f4d00ee 2013-09-23 0intro blockDupLock(b);
982 6f4d00ee 2013-09-23 0intro }else if(r->parent==rr->parent || r->offset > rr->offset){
983 6f4d00ee 2013-09-23 0intro bb = sourceLoadBlock(rr, mode);
984 6f4d00ee 2013-09-23 0intro b = sourceLoadBlock(r, mode);
986 6f4d00ee 2013-09-23 0intro b = sourceLoadBlock(r, mode);
987 6f4d00ee 2013-09-23 0intro bb = sourceLoadBlock(rr, mode);
989 6f4d00ee 2013-09-23 0intro if(b == nil || bb == nil){
991 6f4d00ee 2013-09-23 0intro blockPut(b);
993 6f4d00ee 2013-09-23 0intro blockPut(bb);
994 6f4d00ee 2013-09-23 0intro return 0;
998 6f4d00ee 2013-09-23 0intro * The fact that we are holding b and bb serves
999 6f4d00ee 2013-09-23 0intro * as the lock entitling us to write to r->b and rr->b.
1001 6f4d00ee 2013-09-23 0intro r->b = b;
1002 6f4d00ee 2013-09-23 0intro rr->b = bb;
1003 6f4d00ee 2013-09-23 0intro return 1;
1007 6f4d00ee 2013-09-23 0intro sourceUnlock(Source *r)
1009 6f4d00ee 2013-09-23 0intro Block *b;
1011 6f4d00ee 2013-09-23 0intro if(r->b == nil){
1012 6f4d00ee 2013-09-23 0intro fprint(2, "sourceUnlock: already unlocked\n");
1013 6f4d00ee 2013-09-23 0intro abort();
1015 6f4d00ee 2013-09-23 0intro b = r->b;
1016 6f4d00ee 2013-09-23 0intro r->b = nil;
1017 6f4d00ee 2013-09-23 0intro blockPut(b);
1020 6f4d00ee 2013-09-23 0intro static Block*
1021 6f4d00ee 2013-09-23 0intro sourceLoad(Source *r, Entry *e)
1023 6f4d00ee 2013-09-23 0intro Block *b;
1025 6f4d00ee 2013-09-23 0intro assert(sourceIsLocked(r));
1026 6f4d00ee 2013-09-23 0intro b = r->b;
1027 6f4d00ee 2013-09-23 0intro if(!entryUnpack(e, b->data, r->offset % r->epb))
1028 6f4d00ee 2013-09-23 0intro return nil;
1029 6f4d00ee 2013-09-23 0intro if(e->gen != r->gen){
1030 4b576658 2013-09-23 0intro werrstr(ERemoved);
1031 6f4d00ee 2013-09-23 0intro return nil;
1033 6f4d00ee 2013-09-23 0intro blockDupLock(b);
1034 6f4d00ee 2013-09-23 0intro return b;
1037 6f4d00ee 2013-09-23 0intro static int
1038 6f4d00ee 2013-09-23 0intro sizeToDepth(uvlong s, int psize, int dsize)
1043 6f4d00ee 2013-09-23 0intro /* determine pointer depth */
1044 6f4d00ee 2013-09-23 0intro np = psize/VtScoreSize;
1045 6f4d00ee 2013-09-23 0intro s = (s + dsize - 1)/dsize;
1046 6f4d00ee 2013-09-23 0intro for(d = 0; s > 1; d++)
1047 6f4d00ee 2013-09-23 0intro s = (s + np - 1)/np;
1048 6f4d00ee 2013-09-23 0intro return d;
1051 6f4d00ee 2013-09-23 0intro static u32int
1052 6f4d00ee 2013-09-23 0intro tagGen(void)
1054 6f4d00ee 2013-09-23 0intro u32int tag;
1056 6f4d00ee 2013-09-23 0intro for(;;){
1057 6f4d00ee 2013-09-23 0intro tag = lrand();
1058 6f4d00ee 2013-09-23 0intro if(tag >= UserTag)
1061 6f4d00ee 2013-09-23 0intro return tag;
1065 6f4d00ee 2013-09-23 0intro sourceName(Source *s)
1067 6f4d00ee 2013-09-23 0intro return fileName(s->file);