/*
* Sherlock Library -- Bucket Manipulation Tool
*
- * (c) 2001 Martin Mares <mj@ucw.cz>
+ * (c) 2001--2004 Martin Mares <mj@ucw.cz>
+ * (c) 2004 Robert Spalek <robert@ucw.cz>
+ *
+ * This software may be freely distributed and used according to the terms
+ * of the GNU Lesser General Public License.
*/
#include "lib/lib.h"
#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 <stdio.h>
#include <stdlib.h>
#include <unistd.h>
static int verbose;
+static struct mempool *pool;
+static struct buck2obj_buf *buck_buf;
static void
help(void)
-L\t\tlist all buckets including deleted ones\n\
-d <obj>\tdelete bucket\n\
-x <obj>\textract bucket\n\
--i\t\tinsert buckets separated by blank lines\n\
+-i[<type>]\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\
");
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();
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, *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(0, 4096);
+ in = bfdopen_shared(0, 4096);
obuck_init(1);
do
{
- b = obuck_create();
- while ((e = bgets(in, buf, sizeof(buf))) && buf[0])
+ uns lizard_filled = 0;
+ uns in_body = 0;
+ b = NULL;
+ while ((e = bgets(in, buf, sizeof(buf))))
{
- *e++ = '\n';
- bwrite(b, buf, e-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);
}
- obuck_create_end(b, &h);
- printf("%08x %d %d\n", h.oid, h.length, h.orig_length);
}
while (e);
+ bb_done(&lizard_buf);
+ bb_done(&compressed_buf);
obuck_cleanup();
- /* bclose(in) not done, we don't want fd 0 closed */
+ 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# <missing EOL>\n");
+ bprintf(out, "\n# <missing EOL>\n");
}
- while (obuck_find_next(&h, 0));
+ else
+ dump_parsed_bucket(out, &h, b);
+ bputc(out, '\n');
+ }
+ bclose(out);
obuck_cleanup();
}
{
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);
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:icfFsv", 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;
if (optind < argc)
help();
+ if (!raw)
+ {
+ pool = mp_new(1<<14);
+ buck_buf = buck2obj_alloc();
+ }
switch (op)
{
case 'l':
extract(arg);
break;
case 'i':
- insert();
+ insert(arg);
break;
case 'c':
cat();
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;
}