X-Git-Url: http://mj.ucw.cz/gitweb/?a=blobdiff_plain;f=lib%2Fbuckettool.c;h=179e3d98ea67e10844be6dfe4b33a8f94ed7d844;hb=728e8c27a9fad3acb359d587e48298c88c18aa15;hp=bc8ce32d10ceca1cca491ecea404c6807c264ffe;hpb=98bc6b8a48b5cc006dd650cfdc5cf46966d501a6;p=libucw.git

diff --git a/lib/buckettool.c b/lib/buckettool.c
index bc8ce32d..179e3d98 100644
--- a/lib/buckettool.c
+++ b/lib/buckettool.c
@@ -34,7 +34,7 @@ CF_USAGE
 -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\
@@ -67,7 +67,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();
@@ -96,30 +96,41 @@ extract(char *id)
   b = obuck_fetch();
   while ((l = bread(b, buf, sizeof(buf))))
     fwrite(buf, 1, l, stdout);
-  obuck_fetch_end(b);
+  bclose(b);
   obuck_cleanup();
 }
 
 static void
-insert(void)
+insert(byte *arg)
 {
   struct fastbuf *b, *in;
   byte buf[4096];
   struct obuck_header h;
   byte *e;
+  u32 type;
+
+  if (!arg)
+    type = BUCKET_TYPE_PLAIN;
+  else if (sscanf(arg, "%x", &type) != 1)
+    die("Type `%s' is not a hexadecimal number");
 
   in = bfdopen_shared(0, 4096);
   obuck_init(1);
   do
     {
-      b = obuck_create();
+      b = NULL;
       while ((e = bgets(in, buf, sizeof(buf))) && buf[0])
 	{
 	  *e++ = '\n';
+	  if (!b)
+	    b = obuck_create(type);
 	  bwrite(b, buf, e-buf);
 	}
-      obuck_create_end(b, &h);
-      printf("%08x %d %d\n", h.oid, h.length, h.orig_length);
+      if (b)
+	{
+	  obuck_create_end(b, &h);
+	  printf("%08x %d %08x\n", h.oid, h.length, h.type);
+	}
     }
   while (e);
   obuck_cleanup();
@@ -130,23 +141,25 @@ static void
 cat(void)
 {
   struct obuck_header h;
-  struct fastbuf *b;
+  struct fastbuf *b, *out;
   byte buf[1024];
   int l, lf;
 
   obuck_init(0);
+  out = bfdopen_shared(1, 65536);
   while (b = obuck_slurp_pool(&h))
     {
-      printf("### %08x %6d %6d\n", h.oid, h.length, h.orig_length);
+      bprintf(out, "### %08x %6d %08x\n", h.oid, h.length, h.type);
       lf = 1;
       while ((l = bread(b, buf, sizeof(buf))))
 	{
-	  fwrite(buf, 1, l, stdout);
+	  bwrite(out, buf, l);
 	  lf = (buf[l-1] == '\n');
 	}
       if (!lf)
-	printf("\n# <missing EOL>\n");
+	bprintf(out, "\n# <missing EOL>\n");
     }
+  bclose(out);
   obuck_cleanup();
 }
 
@@ -224,7 +237,7 @@ 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);
@@ -277,7 +290,7 @@ main(int argc, char **argv)
 
   log_init(NULL);
   op = 0;
-  while ((i = cf_getopt(argc, argv, CF_SHORT_OPTS "lLd:x:icfFqsv", CF_NO_LONG_OPTS, NULL)) != -1)
+  while ((i = cf_getopt(argc, argv, CF_SHORT_OPTS "lLd:x:i::cfFqsv", CF_NO_LONG_OPTS, NULL)) != -1)
     if (i == '?' || op)
       help();
     else if (i == 'v')
@@ -305,7 +318,7 @@ main(int argc, char **argv)
       extract(arg);
       break;
     case 'i':
-      insert();
+      insert(arg);
       break;
     case 'c':
       cat();