X-Git-Url: http://mj.ucw.cz/gitweb/?a=blobdiff_plain;ds=sidebyside;f=lib%2Fbuckettool.c;h=ae55b3ca4d4f92c784c25bae194ac3c0fae308fd;hb=2a5686e35c9d6be6cefe22d49cfff6d23ec2e842;hp=609d4e03b933b71ee25f1560d4fda731e7cd4b18;hpb=f2b94f13bdba84efb9bd56a1a57e3d0b33145a9a;p=libucw.git

diff --git a/lib/buckettool.c b/lib/buckettool.c
index 609d4e03..ae55b3ca 100644
--- a/lib/buckettool.c
+++ b/lib/buckettool.c
@@ -2,12 +2,16 @@
  *	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>
@@ -15,21 +19,28 @@
 #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);
 }
@@ -92,18 +103,27 @@ extract(char *id)
 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
@@ -170,9 +190,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;
@@ -191,7 +214,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");
@@ -200,7 +223,7 @@ fsck(int fix)
 	}
       while (nh.magic != OBUCK_MAGIC ||
 	     (nh.oid != (oid_t)(end >> OBUCK_SHIFT) && nh.oid != OBUCK_OID_DELETED));
-      printf("*** match at oid %08x\n", end >> OBUCK_SHIFT);
+      printf("*** match at oid %08x\n", (uns)(end >> OBUCK_SHIFT));
       if (fix)
 	{
 	  h.magic = OBUCK_MAGIC;
@@ -209,8 +232,10 @@ fsck(int fix)
 	  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", end - pos);
+	  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:
@@ -219,52 +244,91 @@ 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;
 
+  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;
 }