#include "blocktree.h"
#define min(a, b) (((b) < (a))?(b):(a))
+#define ISDELOP(op) (((op).buf == NULL) && ((op).fillfn == NULL))
-ssize_t btget(struct store *st, struct btnode *tree, block_t bl, void *buf, size_t len)
+ssize_t btget(struct store *st, struct btnode *tree, block_t bl, void *buf, size_t len, size_t blsize)
{
int d;
block_t c, sel;
- struct btnode indir[BT_INDSZ];
+ struct btnode indir[1 << blsize];
ssize_t sz;
if(tree->d == 0) {
/* This check should really only be necessary on the first
* iteration, but I felt it was easier to put it in the
* loop. */
- if((bl >> (d * BT_INDBITS)) > 0) {
+ if((bl >> (d * blsize)) > 0) {
errno = ERANGE;
return(-1);
}
return(storeget(st, buf, len, &tree->a));
/* Luckily, this is tail recursive */
- if((sz = storeget(st, indir, BT_INDBSZ, &tree->a)) < 0)
+ if((sz = storeget(st, indir, sizeof(indir), &tree->a)) < 0)
return(-1);
c = sz / sizeof(struct btnode);
- sel = bl >> ((d - 1) * BT_INDBITS);
+ sel = bl >> ((d - 1) * blsize);
if(sel >= c) {
errno = ERANGE;
return(-1);
}
tree = &indir[sel];
- bl &= (1LL << ((d - 1) * BT_INDBITS)) - 1;
+ bl &= (1LL << ((d - 1) * blsize)) - 1;
}
return(0);
}
struct addr na;
int ret;
+ if(ISDELOP(*op)) {
+ leaf->d = 0;
+ return(0);
+ }
buf = NULL;
if(op->buf == NULL) {
buf = op->buf = malloc(op->len);
* blputmany() in many ways makes the code uglier, but it saves a
* *lot* of space, since it doesn't need to store intermediary blocks.
*/
-static int btputmany2(struct store *st, struct btnode *tree, struct btop *ops, int numops, block_t bloff)
+static int btputmany2(struct store *st, struct btnode *tree, struct btop *ops, int numops, size_t blsize, block_t bloff)
{
int i, subops, d, f, hasid;
block_t c, sel, bl, nextsz;
struct addr na;
- struct btnode indir[BT_INDSZ];
+ struct btnode indir[1 << blsize];
ssize_t sz;
d = tree->d & 0x7f;
hasid = 0;
for(i = 0; i < numops; ) {
+ if(ops[i].blk < bloff) {
+ errno = ERANGE;
+ return(-1);
+ }
bl = ops[i].blk - bloff;
if((d == 0) && (bl == 0)) {
continue;
}
- if(f && (bl == (1LL << (d * BT_INDBITS)))) {
+ if(f && (bl == (1LL << (d * blsize)))) {
/* New level of indirection */
if(hasid) {
if(storeput(st, indir, c * sizeof(struct btnode), &na))
}
/* Assume that numops == largest block number + 1 -- gaps
* will be detected as errors later */
- for(bl = numops - 1; bl > 0; d++, bl <<= BT_INDBITS);
+ for(bl = numops - 1; bl > 0; d++, bl <<= blsize);
tree->d = d;
c = 0;
hasid = 1;
} else {
/* Get indirect block */
if(!hasid) {
- if((sz = storeget(st, indir, BT_INDBSZ, &tree->a)) < 0)
+ if((sz = storeget(st, indir, sizeof(indir), &tree->a)) < 0)
return(-1);
c = sz / sizeof(struct btnode);
hasid = 1;
}
}
- sel = bl >> ((d - 1) * BT_INDBITS);
+ sel = bl >> ((d - 1) * blsize);
if(sel > c) {
errno = ERANGE;
return(-1);
indir[c].d = 0;
c++;
}
- nextsz = 1LL << ((d - 1) * BT_INDBITS);
+ nextsz = 1LL << ((d - 1) * blsize);
subops = countops(ops + i, numops - i, bloff + (sel * nextsz), nextsz);
- if(btputmany2(st, &indir[sel], ops + i, subops, bloff + (sel * nextsz)))
+ if(btputmany2(st, &indir[sel], ops + i, subops, blsize, bloff + (sel * nextsz)))
return(-1);
i += subops;
- if((sel == BT_INDSZ - 1) && (indir[sel].d == ((d - 1) | 0x80))) {
+ if((sel == (1 << blsize) - 1) && (indir[sel].d == ((d - 1) | 0x80))) {
+ /* Filled up */
tree->d |= 0x80;
f = 1;
+ } else if(indir[sel].d == 0) {
+ /* Erased */
+ if(--c == 1) {
+ tree->d = indir[0].d;
+ tree->a = indir[0].a;
+ }
}
}
if(hasid) {
return(0);
}
-int btputmany(struct store *st, struct btnode *tree, struct btop *ops, int numops)
+int btputmany(struct store *st, struct btnode *tree, struct btop *ops, int numops, size_t blsize)
{
- return(btputmany2(st, tree, ops, numops, 0));
+ return(btputmany2(st, tree, ops, numops, blsize, 0));
}
-int btput(struct store *st, struct btnode *tree, block_t bl, void *buf, size_t len)
+int btput(struct store *st, struct btnode *tree, block_t bl, void *buf, size_t len, size_t blsize)
{
- struct btop ops[1];
+ struct btop ops;
- ops[0].blk = bl;
- ops[0].buf = buf;
- ops[0].len = len;
- return(btputmany(st, tree, ops, 1));
+ memset(&ops, 0, sizeof(ops));
+ ops.blk = bl;
+ ops.buf = buf;
+ ops.len = len;
+ return(btputmany(st, tree, &ops, 1, blsize));
}
void btmkop(struct btop *op, block_t bl, void *buf, size_t len)
static int opcmp(const struct btop **op1, const struct btop **op2)
{
- return((*op1)->blk - (*op2)->blk);
+ if(ISDELOP(**op1) && ISDELOP(**op2))
+ return((*op2)->blk - (*op1)->blk);
+ else if(!ISDELOP(**op1) && ISDELOP(**op2))
+ return(-1);
+ else if(ISDELOP(**op1) && !ISDELOP(**op2))
+ return(1);
+ else
+ return((*op1)->blk - (*op2)->blk);
}
void btsortops(struct btop *ops, int numops)
qsort(ops, numops, sizeof(*ops), (int (*)(const void *, const void *))opcmp);
}
-/*
-Obsoleted
-
-int btappend(struct store *st, struct btnode *tree, void *buf, size_t len)
-{
- int d, f;
- struct btnode indir[BT_INDSZ];
- struct addr na;
- block_t c;
- ssize_t sz;
-
- d = tree->d & 0x7f;
- f = tree->d & 0x80;
-
- if(f) {
- if(storeput(st, buf, len, &na))
- return(-1);
- indir[0] = *tree;
- indir[1].d = 0x80;
- indir[1].a = na;
- if(storeput(st, indir, 2 * sizeof(*indir), &na))
- return(-1);
- tree->d = d + 1;
- tree->a = na;
- return(0);
- }
-
- if(d == 0) {
- if(storeput(st, buf, len, &na))
- return(-1);
- tree->d |= 0x80;
- tree->a = na;
- return(0);
- }
-
- if((sz = storeget(st, indir, BT_INDBSZ, &tree->a)) < 0)
- return(-1);
- c = sz / sizeof(struct btnode);
- if(!(indir[c - 1].d & 0x80) || ((indir[c - 1].d & 0x7f) < (d - 1))) {
- if(btappend(st, &indir[c - 1], buf, len))
- return(-1);
- if(storeput(st, indir, sz, &na))
- return(-1);
- tree->a = na;
- if((indir[c - 1].d & 0x80) && ((indir[c - 1].d & 0x7f) == (d - 1)))
- tree->d |= 0x80;
- return(0);
- }
-
- if(storeput(st, buf, len, &na))
- return(-1);
- indir[c].d = 0x80;
- indir[c].a = na;
- if(storeput(st, indir, sz + sizeof(struct btnode), &na))
- return(-1);
- tree->a = na;
- return(0);
-}
-*/
-
-block_t btcount(struct store *st, struct btnode *tree)
+block_t btcount(struct store *st, struct btnode *tree, size_t blsize)
{
int d, f;
- struct btnode indir[BT_INDSZ];
+ struct btnode indir[1 << blsize];
block_t c, ret;
ssize_t sz;
f = tree->d & 0x80;
if(f)
- return(1LL << (d * BT_INDBITS));
+ return(1LL << (d * blsize));
if(d == 0)
return(0);
ret = 0;
while(1) {
- if((sz = storeget(st, indir, BT_INDBSZ, &tree->a)) < 0)
+ if((sz = storeget(st, indir, sizeof(indir), &tree->a)) < 0)
return(-1);
c = sz / sizeof(struct btnode);
- ret += (c - 1) * (1LL << ((d - 1) * BT_INDBITS));
+ ret += (c - 1) * (1LL << ((d - 1) * blsize));
d = indir[c - 1].d & 0x7f;
f = indir[c - 1].d & 0x80;
if(f)
- return(ret + (1LL << (d * BT_INDBITS)));
+ return(ret + (1LL << (d * blsize)));
tree = &indir[c - 1];
}
}