Merge with git+ssh://git.ucw.cz/projects/sherlock/GIT/sherlock.git#v3.10

[libucw.git] / lib / lizard-test.c

diff --git a/lib/lizard-test.c b/lib/lizard-test.c
index 38a76c56647f4265ca358914a7dfa3668ccd4e1c..137cdc79249a5b7d725ca9b70978f46f3fdaa06e 100644 (file)
--- a/lib/lizard-test.c
+++ b/lib/lizard-test.c
@@ -1,21 +1,23 @@
 #include "lib/lib.h"
 #include "lib/lib.h"

-#include "lib/conf.h"
+#include "lib/getopt.h"

 #include "lib/fastbuf.h"
 #include "lib/fastbuf.h"

+#include "lib/ff-binary.h"

 #include "lib/lizard.h"
 #include <stdio.h>
 #include <stdlib.h>
 #include <fcntl.h>
 #include <unistd.h>
 
 #include "lib/lizard.h"
 #include <stdio.h>
 #include <stdlib.h>
 #include <fcntl.h>
 #include <unistd.h>
 

-static char *options = CF_SHORT_OPTS "cdt";
+static char *options = CF_SHORT_OPTS "cdtx";

 static char *help = "\
 Usage: lizard-test <options> input-file [output-file]\n\
 \n\
 Options:\n"
 CF_USAGE
 static char *help = "\
 Usage: lizard-test <options> input-file [output-file]\n\
 \n\
 Options:\n"
 CF_USAGE

-"-c\t\tCompress (default)\n\
+"-c\t\tCompress\n\

 -d\t\tDecompress\n\
 -d\t\tDecompress\n\

--t\t\tCompress, decompress, and compare (in memory only)\n\
+-t\t\tCompress, decompress, and compare (in memory only, default)\n\
+-x\t\tLet the test crash by shrinking the output buffer\n\

 ";
 
 static void NONRET
 ";
 
 static void NONRET


@@ -29,7 +31,8 @@ int
 main(int argc, char **argv)
 {
   int opt;
 main(int argc, char **argv)
 {
   int opt;

-  uns action = 'c';
+  uns action = 't';
+  uns crash = 0;

   log_init(argv[0]);
   while ((opt = cf_getopt(argc, argv, options, CF_NO_LONG_OPTS, NULL)) >= 0)
     switch (opt)
   log_init(argv[0]);
   while ((opt = cf_getopt(argc, argv, options, CF_NO_LONG_OPTS, NULL)) >= 0)
     switch (opt)


@@ -39,6 +42,9 @@ main(int argc, char **argv)
       case 't':
        action = opt;
        break;
       case 't':
        action = opt;
        break;

+      case 'x':
+       crash++;
+       break;

       default:
        usage();
     }
       default:
        usage();
     }


@@ -47,7 +53,8 @@ main(int argc, char **argv)
     usage();
 
   void *mi, *mo;
     usage();
 
   void *mi, *mo;

-  uns li, lo;
+  int li, lo;
+  uns adler = 0;

 
   struct stat st;
   stat(argv[optind], &st);
 
   struct stat st;
   stat(argv[optind], &st);


@@ -61,7 +68,8 @@ main(int argc, char **argv)
   else
   {
     lo = bgetl(fi);
   else
   {
     lo = bgetl(fi);

-    li -= 4;
+    adler = bgetl(fi);
+    li -= 8;

   }
   mi = xmalloc(li);
   mo = xmalloc(lo);
   }
   mi = xmalloc(li);
   mo = xmalloc(lo);


@@ -70,12 +78,16 @@ main(int argc, char **argv)
 
   printf("%d ", li);
   if (action == 'd')
 
   printf("%d ", li);
   if (action == 'd')

-    printf("->expected %d ", lo);
+    printf("->expected %d (%08x) ", lo, adler);

   fflush(stdout);
   if (action != 'd')
     lo = lizard_compress(mi, li, mo);
   else
   fflush(stdout);
   if (action != 'd')
     lo = lizard_compress(mi, li, mo);
   else

+  {

     lo = lizard_decompress(mi, mo);
     lo = lizard_decompress(mi, mo);

+    if (adler32(mo, lo) != adler)
+      printf("wrong Adler32 ");
+  }

   printf("-> %d ", lo);
   fflush(stdout);
 
   printf("-> %d ", lo);
   fflush(stdout);
 


@@ -83,22 +95,29 @@ main(int argc, char **argv)
   {
     struct fastbuf *fo = bopen(argv[optind+1], O_CREAT | O_TRUNC | O_WRONLY, 1<<16);
     if (action == 'c')
   {
     struct fastbuf *fo = bopen(argv[optind+1], O_CREAT | O_TRUNC | O_WRONLY, 1<<16);
     if (action == 'c')

+    {

       bputl(fo, li);
       bputl(fo, li);

+      bputl(fo, adler32(mi, li));
+    }

     bwrite(fo, mo, lo);
     bclose(fo);
   }
   else
   {
     bwrite(fo, mo, lo);
     bclose(fo);
   }
   else
   {

-    void *mv;
-    uns lv;
-    mv = xmalloc(li);
-    lv = lizard_decompress(mo, mv);
-    printf("-> %d ", lv);
-    fflush(stdout);
-    if (lv != li || memcmp(mi, mv, lv))
+    int smaller_li;
+    if (li >= (int) CPU_PAGE_SIZE)
+      smaller_li = li - CPU_PAGE_SIZE;
+    else
+      smaller_li = 0;
+    struct lizard_buffer *buf = lizard_alloc();
+    byte *ptr = lizard_decompress_safe(mo, buf, crash ? smaller_li : li);
+    if (!ptr)
+      printf("err: %m");
+    else if (memcmp(mi, ptr, li))

       printf("WRONG");
     else
       printf("OK");
       printf("WRONG");
     else
       printf("OK");

+    lizard_free(buf);

   }
   printf("\n");
 }
   }
   printf("\n");
 }