ref: c32c77ec90cb57b1a0cdfd3a94f5693475f160c7
dir: /blk.c/
#include <u.h> #include <libc.h> #include <fcall.h> #include <avl.h> typedef struct Range Range; struct Range { vlong off; vlong len; }; #include "dat.h" #include "fns.h" #include "atomic.h" static vlong blkalloc_lk(Arena*, int); static vlong blkalloc(int, uint); static int blkdealloc_lk(vlong); static Blk* initblk(Blk*, vlong, vlong, int); int checkflag(Blk *b, int f) { long v; v = agetl(&b->flag); return (v & f) == f; } void setflag(Blk *b, int f) { long ov, nv; while(1){ ov = agetl(&b->flag); nv = ov | f; if(acasl(&b->flag, ov, nv) == ov) break; } } void clrflag(Blk *b, int f) { long ov, nv; while(1){ ov = agetl(&b->flag); nv = ov & ~f; if(acasl(&b->flag, ov, nv) == ov) break; } } int syncblk(Blk *b) { assert(checkflag(b, Bfinal)); assert(b->bp.addr >= 0); clrflag(b, Bdirty); return pwrite(fs->fd, b->buf, Blksz, b->bp.addr); } static Blk* readblk(vlong bp, int flg) { Blk *b; vlong off, rem, n; assert(bp != -1); if((b = cachepluck()) == nil) return nil; b->alloced = getcallerpc(&bp); off = bp; rem = Blksz; while(rem != 0){ n = pread(fs->fd, b->buf, rem, off); if(n <= 0){ dropblk(b); return nil; } off += n; rem -= n; } b->cnext = nil; b->cprev = nil; b->hnext = nil; b->flag = 0; b->type = (flg&GBraw) ? Tdat : UNPACK16(b->buf+0); b->bp.addr = bp; b->bp.hash = -1; b->bp.gen = -1; b->fnext = nil; b->nval = 0; b->valsz = 0; b->nbuf = 0; b->bufsz = 0; b->logsz = 0; switch(b->type){ default: fprint(2, "invalid block @%llx\n", bp); abort(); break; case Tdat: case Tarena: b->data = b->buf; break; case Tdlist: b->deadsz = UNPACK16(b->buf+2); b->deadp = unpackbp(b->buf+4, Blksz-4); b->data = b->buf + Dlhdsz; break; case Tlog: b->data = b->buf + Loghdsz; break; case Tpivot: b->data = b->buf + Pivhdsz; b->nval = UNPACK16(b->buf+2); b->valsz = UNPACK16(b->buf+4); b->nbuf = UNPACK16(b->buf+6); b->bufsz = UNPACK16(b->buf+8); break; case Tleaf: b->data = b->buf + Leafhdsz; b->nval = UNPACK16(b->buf+2); b->valsz = UNPACK16(b->buf+4); break; } assert(b->magic == Magic); return b; } static Arena* pickarena(uint ty, uint hint, int tries) { uint n; n = hint + tries + ainc(&fs->roundrobin)/1024; if(ty == Tdat) n++; if(hint % fs->narena == 0) n++; return &fs->arenas[n%fs->narena]; } Arena* getarena(vlong b) { int i; i = b / fs->arenasz; if(i < 0 || i >= fs->narena){ werrstr("out of range block %lld", b); abort(); return nil; } return &fs->arenas[i]; } static int freerange(Avltree *t, vlong off, vlong len) { Arange *r, *s; assert(len % Blksz == 0); if((r = calloc(1, sizeof(Arange))) == nil) return -1; r->off = off; r->len = len; assert(avllookup(t, r, 0) == nil); avlinsert(t, r); Again: s = (Arange*)avlprev(r); if(s != nil && s->off+s->len == r->off){ avldelete(t, r); s->len = s->len + r->len; free(r); r = s; goto Again; } s = (Arange*)avlnext(r); if(s != nil && r->off+r->len == s->off){ avldelete(t, r); s->off = r->off; s->len = s->len + r->len; free(r); r = s; goto Again; } return 0; } static int grabrange(Avltree *t, vlong off, vlong len) { Arange *r, *s, q; vlong l; assert(len % Blksz == 0); q.off = off; q.len = len; r = (Arange*)avllookup(t, &q.Avl, -1); if(r == nil || off + len > r->off + r->len) abort(); if(off == r->off){ r->off += len; r->len -= len; }else if(off + len == r->off + r->len){ r->len -= len; }else if(off > r->off && off+len < r->off + r->len){ if((s = malloc(sizeof(Arange))) == nil) return -1; l = r->len; s->off = off + len; r->len = off - r->off; s->len = l - r->len - len; avlinsert(t, s); }else abort(); if(r->len == 0){ avldelete(t, r); free(r); } return 0; } static Blk* chainlog(Blk *pb, vlong o) { char *p; Blk *lb; if((lb = cachepluck()) == nil) return nil; initblk(lb, o, -1, Tlog); lb->logsz = Loghashsz; p = lb->data + lb->logsz; PACK64(p+0, o|LogAlloc1); PACK64(p+8, (uvlong)LogEnd); finalize(lb); if(syncblk(lb) == -1){ dropblk(lb); return nil; } if(pb != nil){ assert(pb->type == Tlog); p = pb->data + pb->logsz; PACK64(p, lb->bp.addr|LogChain); finalize(pb); if(syncblk(pb) == -1){ dropblk(pb); return nil; } dropblk(pb); } return lb; } /* * Logs an allocation. Must be called * with arena lock held. Duplicates some * of the work in allocblk to prevent * recursion. */ static int logappend(Arena *a, vlong off, vlong len, int op, Blk **tl) { vlong o, ao; Blk *lb; char *p; o = -1; lb = *tl; assert(off % Blksz == 0); assert(op == LogAlloc || op == LogFree); assert(lb == nil || lb->type == Tlog); dprint("logop %llx+%llx@%x: %s\n", off, len, lb?lb->logsz:-1, (op == LogAlloc) ? "Alloc" : "Free"); /* * move to the next block when we have * 40 bytes in the log: * We're appending up to 16 bytes as * part of the operation, followed by * 16 bytes of new log entry allocation * and chaining. */ if(lb == nil || lb->logsz >= Logspc - Logslop){ if((o = blkalloc_lk(a, 1)) == -1) return -1; if((lb = chainlog(lb, o)) == nil) return -1; *tl = lb; a->nlog++; } if(len == Blksz){ if(op == LogAlloc) op = LogAlloc1; else if(op == LogFree) op = LogFree1; } off |= op; p = lb->data + lb->logsz; PACK64(p, off); lb->logsz += 8; if(op >= Log2wide){ PACK64(p+8, len); lb->logsz += 8; } /* * The newly allocated log block needs * to be recorded. If we're compressing * a log, it needs to go to the tail of * the new log, rather than after the * current allocation. so that we don't * reorder allocs and frees. */ if(o != -1){ p = lb->data + lb->logsz; ao = o|LogAlloc1; PACK64(p, ao); lb->logsz += 8; } /* this gets overwritten by the next append */ p = lb->data + lb->logsz; PACK64(p, (uvlong)LogEnd); return 0; } int loadlog(Arena *a, Bptr bp) { vlong ent, off, len; int op, i, n; uvlong bh; char *d; Blk *b; Nextblk: if((b = getblk(bp, GBnochk)) == nil) return -1; bh = UNPACK64(b->data); /* the hash covers the log and offset */ if(bh != bufhash(b->data+Loghashsz, Logspc-Loghashsz)){ werrstr("corrupt log"); return -1; } for(i = Loghashsz; i < Logspc; i += n){ d = b->data + i; ent = UNPACK64(d); op = ent & 0xff; off = ent & ~0xff; n = (op >= Log2wide) ? 16 : 8; switch(op){ case LogEnd: dprint("log@%d: end\n", i); dropblk(b); return 0; case LogChain: bp.addr = off & ~0xff; bp.hash = -1; bp.gen = -1; dropblk(b); dprint("log@%d: chain %B\n", i, bp); goto Nextblk; break; case LogAlloc: case LogAlloc1: len = (op >= Log2wide) ? UNPACK64(d+8) : Blksz; dprint("log@%d alloc: %llx+%llx\n", i, off, len); if(grabrange(a->free, off & ~0xff, len) == -1) return -1; break; case LogFree: case LogFree1: len = (op >= Log2wide) ? UNPACK64(d+8) : Blksz; dprint("log@%d free: %llx+%llx\n", i, off, len); if(freerange(a->free, off & ~0xff, len) == -1) return -1; break; default: n = 0; dprint("log@%d: log op %d\n", i, op); abort(); break; } } return -1; } int compresslog(Arena *a) { vlong v, ba, na, nl, sz, graft, oldhd, *log; int i, n, nr; Blk *b, *hd, *tl; Range *rng; Arange *r; Bptr bp; char *p; /* * Sync the current log to disk, and * set up a new block log tail. While * compressing the log, nothing else is * using this arena, so any allocs come * from the log compression, and go into * this new log segment. * * A bit of copy paste from newblk, * because otherwise we have a deadlock * allocating the block. */ if((ba = blkalloc_lk(a, 1)) == -1) return -1; if((b = cachepluck()) == nil) return -1; initblk(b, ba, -1, Tlog); b->logsz = Loghashsz; p = b->data + b->logsz; PACK64(p, (uvlong)LogEnd); finalize(b); if(syncblk(b) == -1){ dropblk(b); return -1; } graft = b->bp.addr; if(a->logtl != nil){ finalize(a->logtl); if(syncblk(a->logtl) == -1){ dropblk(b); return -1; } } a->logtl = b; /* * Prepare what we're writing back. * Arenas must be sized so that we can * keep the merged log in memory for * a rewrite. */ sz = 0; nr = 0; lock(a); for(r = (Arange*)avlmin(a->free); r != nil; r = (Arange*)avlnext(r)){ sz += (r->len == Blksz) ? 8 : 16; nr++; } nl = (sz+Logspc-Logslop-1)/(Logspc - Logslop); if((log = malloc(nl*sizeof(vlong))) == nil){ unlock(a); return -1; } if((rng = malloc(nr*sizeof(Range))) == nil){ unlock(a); free(log); return -1; } for(i = 0; i < nl; i++){ if((log[i] = blkalloc_lk(a, 1)) == -1){ unlock(a); free(log); free(rng); return -1; } } nr = 0; for(r = (Arange*)avlmin(a->free); r != nil; r = (Arange*)avlnext(r)){ rng[nr].off = r->off; rng[nr].len = r->len; nr++; } unlock(a); n = 0; hd = nil; tl = nil; for(i = 0; i < nr; i++){ if(tl == nil || tl->logsz >= Logspc - Logslop){ if((tl = chainlog(tl, log[n++])) == nil){ free(rng); free(log); return -1; } if(hd == nil) hd = tl; p = tl->data + tl->logsz; } if(rng[i].len == Blksz){ PACK64(p+0, rng[i].off | LogFree1); tl->logsz += 8; p += 8; }else{ PACK64(p+0, rng[i].off | LogFree); PACK64(p+8, rng[i].len); tl->logsz += 16; p += 16; } } PACK64(p, LogChain|graft); free(log); free(rng); finalize(tl); if(syncblk(tl) == -1) return -1; lock(a); oldhd = a->loghd.addr; a->loghd.addr = hd->bp.addr; a->loghd.hash = hd->bp.hash; a->loghd.gen = -1; unlock(a); if(oldhd != -1){ for(ba = oldhd; ba != -1 && ba != graft; ba = na){ na = -1; bp.addr = ba; bp.hash = -1; bp.gen = -1; if((b = getblk(bp, GBnochk)) == nil) return -1; for(i = Loghashsz; i < Logspc; i += n){ p = b->data + i; v = UNPACK64(p); n = ((v&0xff) >= Log2wide) ? 16 : 8; if((v&0xff) == LogChain){ na = v & ~0xff; break; }else if((v&0xff) == LogEnd){ na = -1; break; } } lock(a); cachedel(b->bp.addr); if(blkdealloc_lk(ba) == -1){ unlock(a); return -1; } dropblk(b); unlock(a); } } lock(a); a->nlog = 0; unlock(a); finalize(a->logtl); if(syncblk(a->logtl) == -1) return -1; return 0; } /* * Allocate from an arena, with lock * held. May be called multiple times * per operation, to alloc space for * the alloc log. */ static vlong blkalloc_lk(Arena *a, int force) { Avltree *t; Arange *r; vlong b; t = a->free; r = (Arange*)t->root; if(!force && a->size - a->used <= a->reserve) return -1; if(r == nil){ fprint(2, "out of space"); abort(); } /* * A bit of sleight of hand here: * while we're changing the sorting * key, but we know it won't change * the sort order because the tree * covers disjoint ranges */ b = r->off; r->len -= Blksz; r->off += Blksz; if(r->len == 0){ avldelete(t, r); free(r); } a->used += Blksz; return b; } static int blkdealloc_lk(vlong b) { Arena *a; int r; r = -1; a = getarena(b); if(logappend(a, b, Blksz, LogFree, &a->logtl) == -1) return -1; if(a->loghd.addr == -1) a->loghd = a->logtl->bp; if(freerange(a->free, b, Blksz) == -1) goto out; a->used -= Blksz; r = 0; out: return r; } int blkrelease(vlong b) { Arena *a; int r; a = getarena(b); lock(a); r = -1; a = getarena(b); if(logappend(a, b, Blksz, LogFree, &a->defertl) == -1) return -1; if(a->loghd.addr == -1) a->loghd = a->logtl->bp; unlock(a); return r; } static vlong blkalloc(int ty, uint hint) { Arena *a; vlong b; int tries; tries = 0; Again: a = pickarena(ty, hint, tries); if(a == nil || tries == fs->narena){ werrstr("no empty arenas"); return -1; } /* * TODO: there's an extreme edge case * here. * * If the file system has room to alloc * a data block but no log block, then * we end up with it in a stuck state. * The fix is to reserve alloc blocks, * so that we're guaranteed to be able * to log an alloc if the disk is working * correctly. */ tries++; lock(a); if((b = blkalloc_lk(a, 0)) == -1){ unlock(a); goto Again; } if(logappend(a, b, Blksz, LogAlloc, &a->logtl) == -1){ unlock(a); return -1; } if(a->loghd.addr == -1) a->loghd = a->logtl->bp; unlock(a); return b; } static Blk* initblk(Blk *b, vlong bp, vlong gen, int ty) { Blk *ob; ob = cacheget(bp); if(ob != nil){ fprint(2, "dup block: %#p %B (alloced %#llx freed %#llx lasthold: %#llx, lastdrop: %#llx)\n", ob, ob->bp, ob->alloced, ob->freed, ob->lasthold, ob->lastdrop); abort(); } b->type = ty; b->bp.addr = bp; b->bp.hash = -1; b->bp.gen = gen; switch(ty){ case Tdat: case Tarena: b->data = b->buf; break; case Tdlist: b->deadsz = 0; b->data = b->buf + Dlhdsz; break; case Tlog: b->logsz = 0; b->data = b->buf + Loghdsz; break; case Tpivot: b->data = b->buf + Pivhdsz; break; case Tleaf: b->data = b->buf + Leafhdsz; break; } b->fnext = nil; setflag(b, Bdirty); b->nval = 0; b->valsz = 0; b->nbuf = 0; b->bufsz = 0; b->logsz = 0; b->alloced = getcallerpc(&b); return b; } Blk* newdblk(Tree *t, int ty, vlong hint) { vlong bp; Blk *b; if((bp = blkalloc(ty, hint)) == -1) return nil; if((b = cachepluck()) == nil) return nil; initblk(b, bp, t->gen, ty); b->alloced = getcallerpc(&t); return b; } Blk* newblk(Tree *t, int ty) { return newdblk(t, ty, 0); } Blk* dupblk(Tree *t, Blk *b) { Blk *r; if((r = newblk(t, b->type)) == nil) return nil; setflag(r, Bdirty); r->bp.hash = -1; r->nval = b->nval; r->valsz = b->valsz; r->nbuf = b->nbuf; r->bufsz = b->bufsz; r->logsz = b->logsz; r->alloced = getcallerpc(&t); memcpy(r->buf, b->buf, sizeof(r->buf)); return r; } void finalize(Blk *b) { uvlong h; if(b->type != Tdat) PACK16(b->buf, b->type); switch(b->type){ default: case Tpivot: PACK16(b->buf+2, b->nval); PACK16(b->buf+4, b->valsz); PACK16(b->buf+6, b->nbuf); PACK16(b->buf+8, b->bufsz); break; case Tleaf: PACK16(b->buf+2, b->nval); PACK16(b->buf+4, b->valsz); break; case Tdlist: PACK16(b->buf+2, b->deadsz); packbp(b->buf+4, Ptrsz, &b->deadp); break; case Tlog: h = bufhash(b->data + Loghashsz, Logspc-Loghashsz); PACK64(b->data, h); break; case Tdat: case Tmagic: case Tarena: break; } b->bp.hash = blkhash(b); setflag(b, Bfinal); cacheins(b); } Blk* getblk(Bptr bp, int flg) { uvlong h; Blk *b; int i; i = ihash(bp.addr) % nelem(fs->blklk); qlock(&fs->blklk[i]); if((b = cacheget(bp.addr)) != nil){ qunlock(&fs->blklk[i]); return b; } if((b = readblk(bp.addr, flg)) == nil){ qunlock(&fs->blklk[i]); return nil; } b->alloced = getcallerpc(&bp); h = blkhash(b); if((flg&GBnochk) == 0 && h != bp.hash){ fprint(2, "corrupt block %p %B: %.16llux != %.16llux\n", b, bp, h, bp.hash); qunlock(&fs->blklk[i]); return nil; } b->bp.hash = h; b->bp.gen = bp.gen; cacheins(b); qunlock(&fs->blklk[i]); return b; } Blk* holdblk(Blk *b) { ainc(&b->ref); b->lasthold1 = b->lasthold0; b->lasthold0 = b->lasthold; b->lasthold = getcallerpc(&b); return b; } void dropblk(Blk *b) { assert(b == nil || b->ref > 0); if(b == nil || adec(&b->ref) != 0) return; b->lastdrop1 = b->lastdrop0; b->lastdrop0 = b->lastdrop; b->lastdrop = getcallerpc(&b); /* * While a freed block can get resurrected * before quiescence, it's unlikely -- so * it goes into the bottom of the LRU to * get selected early for reuse. */ if(checkflag(b, Bfreed)) lrubot(b); else lrutop(b); } ushort blkfill(Blk *b) { switch(b->type){ case Tpivot: return 2*b->nbuf + b->bufsz + 2*b->nval + b->valsz; case Tleaf: return 2*b->nval + b->valsz; default: fprint(2, "invalid block @%lld\n", b->bp.addr); abort(); } return 0; // shut up kencc } void deferfree(Tree *t, Bptr bp, Blk *b) { Bfree *f; ulong ge; if(t != nil && t != &fs->snap && bp.gen <= t->prev){ killblk(t, bp); return; } if((f = malloc(sizeof(Bfree))) == nil) return; f->bp = bp; f->b = b; ge = agetl(&fs->epoch); f->next = fs->limbo[ge]; fs->limbo[ge] = f; } void freebp(Tree *t, Bptr bp) { deferfree(t, bp, nil); } void freeblk(Tree *t, Blk *b) { holdblk(b); b->freed = getcallerpc(&t); setflag(b, Bfreed); deferfree(t, b->bp, b); } void epochstart(int tid) { ulong ge; ge = agetl(&fs->epoch); asetl(&fs->lepoch[tid], ge | Eactive); } void epochend(int tid) { ulong le; le = agetl(&fs->lepoch[tid]); asetl(&fs->lepoch[tid], le &~ Eactive); } void epochclean(void) { ulong e, ge; Bfree *p, *n; int i; ge = agetl(&fs->epoch); for(i = 0; i < fs->nworker; i++){ e = agetl(&fs->lepoch[i]); if((e & Eactive) && e != (ge | Eactive)) return; } p = asetp(&fs->limbo[(ge+1)%3], nil); asetl(&fs->epoch, (ge+1) % 3); while(p != nil){ n = p->next; cachedel(p->bp.addr); blkrelease(p->bp.addr); if(p->b != nil) dropblk(p->b); free(p); p = n; } } int blkcmp(Blk *a, Blk *b) { if(a->qgen != b->qgen) return (a->qgen < b->qgen) ? -1 : 1; if(a->bp.addr != b->bp.addr) return (a->bp.addr < b->bp.addr) ? -1 : 1; return 0; } void enqueue(Blk *b) { Arena *a; b->qgen = agetv(&fs->qgen); a = getarena(b->bp.addr); assert(checkflag(b, Bdirty)); assert(b->bp.addr >= 0); holdblk(b); finalize(b); qput(a->sync, b); } void qinit(Syncq *q) { q->fullrz.l = &q->lk; q->emptyrz.l = &q->lk; q->nheap = 0; q->heapsz = fs->cmax; if((q->heap = malloc(q->heapsz*sizeof(Blk*))) == nil) sysfatal("alloc queue: %r"); } void qput(Syncq *q, Blk *b) { int i; qlock(&q->lk); while(q->nheap == q->heapsz) rsleep(&q->fullrz); for(i = q->nheap; i > 0; i = (i-1)/2){ if(blkcmp(b, q->heap[(i-1)/2]) == 1) break; q->heap[i] = q->heap[(i-1)/2]; } q->heap[i] = b; q->nheap++; rwakeup(&q->emptyrz); qunlock(&q->lk); } static Blk* qpop(Syncq *q) { int i, l, r, m; Blk *b, *t; qlock(&q->lk); while(q->nheap == 0) rsleep(&q->emptyrz); b = q->heap[0]; if(--q->nheap == 0) goto Out; i = 0; q->heap[0] = q->heap[q->nheap]; while(1){ m = i; l = 2*i+1; r = 2*i+2; if(l < q->nheap && blkcmp(q->heap[m], q->heap[l]) == 1) m = l; if(r < q->nheap && blkcmp(q->heap[m], q->heap[r]) == 1) m = r; if(m == i) break; t = q->heap[m]; q->heap[m] = q->heap[i]; q->heap[i] = t; i = m; } Out: rwakeup(&q->fullrz); qunlock(&q->lk); return b; } void runsync(int, void *p) { Syncq *q; Blk *b; q = p; while(1){ b = qpop(q); if(b->type == Tmagic){ qlock(&fs->synclk); if(--fs->syncing == 0) rwakeupall(&fs->syncrz); qunlock(&fs->synclk); }else if(!checkflag(b, Bfreed)){ if(syncblk(b) == -1){ ainc(&fs->broken); fprint(2, "write: %r\n"); abort(); } } dropblk(b); } } void sync(void) { Arena *a; Blk *b; int i; if(fs->rdonly) return; qlock(&fs->synclk); if(!fs->snap.dirty){ qunlock(&fs->synclk); return; } flushdlcache(0); aincv(&fs->qgen, 1); fs->syncing = fs->nsyncers; for(i = 0; i < fs->nsyncers; i++){ b = cachepluck(); b->type = Tmagic; qput(&fs->syncq[i], b); } while(fs->syncing != 0) rsleep(&fs->syncrz); for(i = 0; i < fs->narena; i++){ a = &fs->arenas[i]; lock(a); if(a->deferhd.addr != -1){ chainlog(a->logtl, a->deferhd.addr); loadlog(a, a->deferhd); a->logtl = a->defertl; a->deferhd = (Bptr){-1, -1, -1}; a->defertl = nil; } packarena(a->b->data, Blksz, a, fs); finalize(a->b); finalize(a->logtl); if(syncblk(a->logtl) == -1) sysfatal("sync arena: %r"); if(syncblk(a->b) == -1) sysfatal("sync arena: %r"); unlock(a); } qunlock(&fs->synclk); }