* Sherlock Library -- Bucket Manipulation Tool
*
* (c) 2001 Martin Mares <mj@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/bucket.h"
#include "lib/fastbuf.h"
#include "lib/lfs.h"
+#include "lib/conf.h"
#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <unistd.h>
+static int verbose;
+
static void
help(void)
{
fprintf(stderr, "\
-Usage: buckettool <commands>\n\
+Usage: buckettool [<options>] <command>\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 <obj>\tdelete bucket\n\
-x <obj>\textract bucket\n\
--i\t\tinsert bucket\n\
+-i\t\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\
+-s\t\tshake down bucket file (without updating other structures!!!)\n\
+-v\t\tbe verbose\n\
");
exit(1);
}
static void
insert(void)
{
- struct fastbuf *b;
- byte buf[1024];
- int l;
+ struct fastbuf *b, *in;
+ byte buf[4096];
struct obuck_header h;
+ byte *e;
+ 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
+ {
+ b = obuck_create();
+ while ((e = bgets(in, buf, sizeof(buf))) && buf[0])
+ {
+ *e++ = '\n';
+ bwrite(b, buf, e-buf);
+ }
+ obuck_create_end(b, &h);
+ printf("%08x %d %d\n", h.oid, h.length, h.orig_length);
+ }
+ while (e);
obuck_cleanup();
- printf("%08x %d %d\n", h.oid, h.length, h.orig_length);
+ bclose(in);
}
static void
int l, lf;
obuck_init(0);
- if (obuck_find_first(&h, 0))
- do
- {
- printf("### %08x %6d %6d\n", h.oid, h.length, h.orig_length);
- b = obuck_fetch();
- lf = 1;
- while ((l = bread(b, buf, sizeof(buf))))
- {
- fwrite(buf, 1, l, stdout);
- lf = (buf[l-1] == '\n');
- }
- obuck_fetch_end(b);
- if (!lf)
- printf("\n# <missing EOL>\n");
- }
- while (obuck_find_next(&h, 0));
+ while (b = obuck_slurp_pool(&h))
+ {
+ printf("### %08x %6d %6d\n", h.oid, h.length, h.orig_length);
+ lf = 1;
+ while ((l = bread(b, buf, sizeof(buf))))
+ {
+ fwrite(buf, 1, l, stdout);
+ lf = (buf[l-1] == '\n');
+ }
+ if (!lf)
+ printf("\n# <missing EOL>\n");
+ }
obuck_cleanup();
}
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;
if (fix)
{
printf("*** truncating file\n");
- ftruncate(fd, pos);
+ sh_ftruncate(fd, pos);
}
else
printf("*** would truncate the file here\n");
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:
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;
+ log_init(NULL);
op = 0;
- while ((i = getopt(argc, argv, "lLd:x:icfF")) != -1)
- switch (i)
+ while ((i = cf_getopt(argc, argv, CF_SHORT_OPTS "lLd:x:icfFqsv", CF_NO_LONG_OPTS, NULL)) != -1)
+ if (i == '?' || op)
+ help();
+ else if (i == 'v')
+ verbose++;
+ else
{
- case 'l':
- list(0);
- op++;
- break;
- case 'L':
- list(1);
- op++;
- break;
- case 'd':
- delete(optarg);
- op++;
- break;
- case 'x':
- extract(optarg);
- op++;
- break;
- case 'i':
- insert();
- op++;
- break;
- case 'c':
- cat();
- op++;
- break;
- case 'f':
- fsck(0);
- op++;
- break;
- case 'F':
- fsck(1);
- op++;
- break;
- default:
- help();
+ op = i;
+ arg = optarg;
}
- if (optind < argc || !op)
+ if (optind < argc)
help();
+ switch (op)
+ {
+ case 'l':
+ list(0);
+ break;
+ case 'L':
+ list(1);
+ break;
+ case 'd':
+ delete(arg);
+ break;
+ case 'x':
+ extract(arg);
+ break;
+ case 'i':
+ insert();
+ break;
+ case 'c':
+ cat();
+ break;
+ case 'f':
+ fsck(0);
+ break;
+ case 'F':
+ fsck(1);
+ break;
+ case 'q':
+ quickcheck();
+ break;
+ case 's':
+ shake();
+ break;
+ default:
+ help();
+ }
+
return 0;
}