X-Git-Url: http://mj.ucw.cz/gitweb/?a=blobdiff_plain;f=lib%2Fbuckettool.c;h=6458c539a1268ad342cdbcabc76b6967f3738ca2;hb=5d26aed3953fe00d6d394f9767b4a029406eeeea;hp=c46cf425023a9e724352306459ff5381dc41aa22;hpb=358365e678cf9494ba55ef1fd1bf46a8e3b5cd44;p=libucw.git diff --git a/lib/buckettool.c b/lib/buckettool.c index c46cf425..6458c539 100644 --- a/lib/buckettool.c +++ b/lib/buckettool.c @@ -1,7 +1,11 @@ /* * Sherlock Library -- Bucket Manipulation Tool * - * (c) 2001 Martin Mares + * (c) 2001--2004 Martin Mares + * (c) 2004 Robert Spalek + * + * This software may be freely distributed and used according to the terms + * of the GNU Lesser General Public License. */ #include "lib/lib.h" @@ -9,6 +13,11 @@ #include "lib/fastbuf.h" #include "lib/lfs.h" #include "lib/conf.h" +#include "lib/mempool.h" +#include "lib/object.h" +#include "lib/lizard.h" +#include "lib/bbuf.h" +#include "lib/ff-utf8.h" #include #include @@ -16,21 +25,31 @@ #include #include +static int verbose; +static struct mempool *pool; +static struct buck2obj_buf *buck_buf; + static void help(void) { fprintf(stderr, "\ -Usage: buckettool \n\ +Usage: buckettool [] \n\ \n\ -Commands:\n\ +Options:\n" +CF_USAGE +"\nCommands:\n\ -l\t\tlist all buckets\n\ -L\t\tlist all buckets including deleted ones\n\ -d \tdelete bucket\n\ -x \textract bucket\n\ --i\t\tinsert bucket\n\ +-i[]\tinsert buckets separated by blank lines\n\ -c\t\tconcatenate and dump all buckets\n\ -f\t\taudit bucket file structure\n\ -F\t\taudit and fix bucket file structure\n\ +-q\t\tquick check of bucket file consistency\n\ +-r\t\tdo not parse V33 buckets, but print the raw content\n\ +-s\t\tshake down bucket file (without updating other structures!!!)\n\ +-v\t\tbe verbose\n\ "); exit(1); } @@ -57,7 +76,7 @@ list(int full) if (h.oid == OBUCK_OID_DELETED) printf("DELETED %6d\n", h.length); else - printf("%08x %6d %6d\n", h.oid, h.length, h.orig_length); + printf("%08x %6d %08x\n", h.oid, h.length, h.type); } while (obuck_find_next(&h, full)); obuck_cleanup(); @@ -72,66 +91,170 @@ delete(char *id) obuck_cleanup(); } +static inline void +dump_oattrs(struct fastbuf *out, struct oattr *oa) +{ + for (; oa; oa = oa->next) + for (struct oattr *a=oa; a; a = a->same) + bprintf(out, "%c%s\n", a->attr, a->val); +} + +static void +dump_parsed_bucket(struct fastbuf *out, struct obuck_header *h, struct fastbuf *b) +{ + struct odes *o_hdr, *o_body; + mp_flush(pool); + o_hdr = obj_new(pool); + o_body = obj_new(pool); + if (buck2obj_parse(buck_buf, h->type, h->length, b, o_hdr, NULL, o_body) < 0) + bprintf(out, ".Cannot parse bucket %x of type %x and length %d: %m\n", h->oid, h->type, h->length); + else + { + dump_oattrs(out, o_hdr->attrs); + bputc(out, '\n'); + dump_oattrs(out, o_body->attrs); + } +} + static void extract(char *id) { - struct fastbuf *b; - byte buf[1024]; - int l; + struct fastbuf *b, *out; struct obuck_header h; h.oid = parse_id(id); obuck_init(0); obuck_find_by_oid(&h); + out = bfdopen_shared(1, 65536); + if (verbose) + bprintf(out, "### %08x %6d %08x\n", h.oid, h.length, h.type); b = obuck_fetch(); - while ((l = bread(b, buf, sizeof(buf)))) - fwrite(buf, 1, l, stdout); - obuck_fetch_end(b); + if (h.type < BUCKET_TYPE_V33 || !buck_buf) + bbcopy_slow(b, out, ~0U); + else + dump_parsed_bucket(out, &h, b); + bclose(b); + bclose(out); obuck_cleanup(); } static void -insert(void) +insert(byte *arg) { - struct fastbuf *b; - byte buf[1024]; - int l; + struct fastbuf *b, *in; + byte buf[4096]; struct obuck_header h; + byte *e; + u32 type; + bb_t lizard_buf, compressed_buf; + + bb_init(&lizard_buf); + bb_init(&compressed_buf); + if (!arg) + type = BUCKET_TYPE_PLAIN; + else if (sscanf(arg, "%x", &type) != 1) + die("Type `%s' is not a hexadecimal number"); + if (type < 10) + type += BUCKET_TYPE_PLAIN; + put_attr_set_type(type); + in = bfdopen_shared(0, 4096); obuck_init(1); - b = obuck_create(); - while ((l = fread(buf, 1, sizeof(buf), stdin))) - bwrite(b, buf, l); - obuck_create_end(b, &h); + do + { + uns lizard_filled = 0; + uns in_body = 0; + b = NULL; + while ((e = bgets(in, buf, sizeof(buf)))) + { + if (!buf[0]) + { + if (in_body || type < BUCKET_TYPE_V30) + break; + in_body = 1; + } + if (!b) + b = obuck_create(type); + if (in_body == 1) + { + bputc(b, 0); + in_body = 2; + } + else if (type <= BUCKET_TYPE_V33 || !in_body) + { + bput_attr(b, buf[0], buf+1, e-buf-1); + } + else + { + ASSERT(BUCKET_TYPE_V33_LIZARD); + uns want_len = lizard_filled + (e-buf) + 6 + LIZARD_NEEDS_CHARS; // +6 is the maximum UTF-8 length + bb_grow(&lizard_buf, want_len); + byte *ptr = lizard_buf.ptr + lizard_filled; + ptr = put_attr(ptr, buf[0], buf+1, e-buf-1); + lizard_filled = ptr - lizard_buf.ptr; + } + } + if (in_body && type == BUCKET_TYPE_V33_LIZARD) + { + bputl(b, lizard_filled +#if 0 //TEST error resilience: write wrong length + +1 +#endif + ); + bputl(b, adler32(lizard_buf.ptr, lizard_filled) +#if 0 //TEST error resilience: write wrong checksum + +1 +#endif + ); + uns want_len = lizard_filled * LIZARD_MAX_MULTIPLY + LIZARD_MAX_ADD; + bb_grow(&compressed_buf, want_len); + want_len = lizard_compress(lizard_buf.ptr, lizard_filled, compressed_buf.ptr); +#if 0 //TEST error resilience: tamper the compressed data by removing EOF + compressed_buf[want_len-1] = 1; +#endif + bwrite(b, compressed_buf.ptr, want_len); + } + if (b) + { + obuck_create_end(b, &h); + printf("%08x %d %08x\n", h.oid, h.length, h.type); + } + } + while (e); + bb_done(&lizard_buf); + bb_done(&compressed_buf); obuck_cleanup(); - printf("%08x %d %d\n", h.oid, h.length, h.orig_length); + bclose(in); } static void cat(void) { struct obuck_header h; - struct fastbuf *b; + struct fastbuf *b, *out; byte buf[1024]; - int l, lf; obuck_init(0); - if (obuck_find_first(&h, 0)) - do + out = bfdopen_shared(1, 65536); + while (b = obuck_slurp_pool(&h, OBUCK_OID_ANY)) + { + bprintf(out, "### %08x %6d %08x\n", h.oid, h.length, h.type); + if (h.type < BUCKET_TYPE_V33 || !buck_buf) { - printf("### %08x %6d %6d\n", h.oid, h.length, h.orig_length); - b = obuck_fetch(); - lf = 1; + int lf = 1, l; while ((l = bread(b, buf, sizeof(buf)))) - { - fwrite(buf, 1, l, stdout); - lf = (buf[l-1] == '\n'); - } - obuck_fetch_end(b); + { + bwrite(out, buf, l); + lf = (buf[l-1] == '\n'); + } if (!lf) - printf("\n# \n"); + bprintf(out, "\n# \n"); } - while (obuck_find_next(&h, 0)); + else + dump_parsed_bucket(out, &h, b); + bputc(out, '\n'); + } + bclose(out); obuck_cleanup(); } @@ -171,9 +294,12 @@ fsck(int fix) printf("%08x missing trailer\n", oid); else if (chk != OBUCK_TRAILER) printf("%08x mismatched trailer\n", oid); - /* OK */ - pos = end; - continue; + else + { + /* OK */ + pos = end; + continue; + } } errors++; end = pos; @@ -192,7 +318,7 @@ fsck(int fix) if (fix) { printf("*** truncating file\n"); - ftruncate(fd, pos); + sh_ftruncate(fd, pos); } else printf("*** would truncate the file here\n"); @@ -206,12 +332,14 @@ fsck(int fix) { h.magic = OBUCK_MAGIC; h.oid = OBUCK_OID_DELETED; - h.length = h.orig_length = end - pos - sizeof(h) - 4; + h.length = end - pos - sizeof(h) - 4; sh_pwrite(fd, &h, sizeof(h), pos); chk = OBUCK_TRAILER; sh_pwrite(fd, &chk, 4, end-4); printf("*** replaced the invalid chunk by a DELETED bucket of size %d\n", (uns)(end - pos)); } + else + printf("*** would mark %d bytes as DELETED\n", (uns)(end - pos)); pos = end; } finish: @@ -220,17 +348,51 @@ fsck(int fix) exit(1); } +static int +shake_kibitz(struct obuck_header *old, oid_t new, byte *buck UNUSED) +{ + if (verbose) + { + printf("%08x -> ", old->oid); + if (new == OBUCK_OID_DELETED) + puts("DELETED"); + else + printf("%08x\n", new); + } + return 1; +} + +static void +shake(void) +{ + obuck_init(1); + obuck_shakedown(shake_kibitz); + obuck_cleanup(); +} + +static void +quickcheck(void) +{ + obuck_init(1); + obuck_cleanup(); +} + int main(int argc, char **argv) { int i, op; char *arg = NULL; + uns raw = 0; log_init(NULL); op = 0; - while ((i = cf_getopt(argc, argv, CF_SHORT_OPTS "lLd:x:icfF", CF_NO_LONG_OPTS, NULL)) != -1) + while ((i = cf_getopt(argc, argv, CF_SHORT_OPTS "lLd:x:i::cfFqrsv", CF_NO_LONG_OPTS, NULL)) != -1) if (i == '?' || op) help(); + else if (i == 'v') + verbose++; + else if (i == 'r') + raw++; else { op = i; @@ -239,6 +401,11 @@ main(int argc, char **argv) if (optind < argc) help(); + if (!raw) + { + pool = mp_new(1<<14); + buck_buf = buck2obj_alloc(); + } switch (op) { case 'l': @@ -254,7 +421,7 @@ main(int argc, char **argv) extract(arg); break; case 'i': - insert(); + insert(arg); break; case 'c': cat(); @@ -265,9 +432,20 @@ main(int argc, char **argv) case 'F': fsck(1); break; + case 'q': + quickcheck(); + break; + case 's': + shake(); + break; default: help(); } + if (buck_buf) + { + buck2obj_free(buck_buf); + mp_delete(pool); + } return 0; }