bytetest.c

Version: - FREEBSD - FREEBSD-13-STABLE - FREEBSD-13-0 - FREEBSD-12-STABLE - FREEBSD-12-0 - FREEBSD-11-STABLE - FREEBSD-11-0 - FREEBSD-10-STABLE - FREEBSD-10-0 - FREEBSD-9-STABLE - FREEBSD-9-0 - FREEBSD-8-STABLE - FREEBSD-8-0 - FREEBSD-7-STABLE - FREEBSD-7-0 - FREEBSD-6-STABLE - FREEBSD-6-0 - FREEBSD-5-STABLE - FREEBSD-5-0 - FREEBSD-4-STABLE - FREEBSD-3-STABLE - FREEBSD22 - l41 - OPENBSD - linux-2.6 - MK84 - PLAN9 - xnu-8792
SearchContext: - none - 3 - 10

    1 /*
    2  * Lazy test for the case when the output size is known
    3  *
    4  * Author: Lasse Collin <lasse.collin@tukaani.org>
    5  *
    6  * This file has been put into the public domain.
    7  * You can do whatever you want with this file.
    8  */
    9 
   10 #include <stdbool.h>
   11 #include <stdio.h>
   12 #include <string.h>
   13 #include <stdlib.h>
   14 #include "xz.h"
   15 
   16 static uint8_t in[1];
   17 static uint8_t out[BUFSIZ];
   18 
   19 int main(int argc, char **argv)
   20 {
   21         struct xz_buf b;
   22         struct xz_dec *s;
   23         enum xz_ret ret;
   24         const char *msg;
   25         size_t uncomp_size;
   26 
   27         if (argc != 2) {
   28                 fputs("Give uncompressed size as the argument\n", stderr);
   29                 return 1;
   30         }
   31 
   32         uncomp_size = atoi(argv[1]);
   33 
   34         xz_crc32_init();
   35 
   36         /*
   37          * Support up to 64 MiB dictionary. The actually needed memory
   38          * is allocated once the headers have been parsed.
   39          */
   40         s = xz_dec_init(XZ_DYNALLOC, 1 << 26);
   41         if (s == NULL) {
   42                 msg = "Memory allocation failed\n";
   43                 goto error;
   44         }
   45 
   46         b.in = in;
   47         b.in_pos = 0;
   48         b.in_size = 0;
   49         b.out = out;
   50         b.out_pos = 0;
   51         b.out_size = uncomp_size < BUFSIZ ? uncomp_size : BUFSIZ;
   52 
   53         while (true) {
   54                 if (b.in_pos == b.in_size) {
   55                         b.in_size = fread(in, 1, sizeof(in), stdin);
   56                         b.in_pos = 0;
   57                 }
   58 
   59                 ret = xz_dec_run(s, &b);
   60 
   61                 if (b.out_pos == sizeof(out)) {
   62                         if (fwrite(out, 1, b.out_pos, stdout) != b.out_pos) {
   63                                 msg = "Write error\n";
   64                                 goto error;
   65                         }
   66 
   67                         uncomp_size -= b.out_pos;
   68                         b.out_pos = 0;
   69                         b.out_size = uncomp_size < BUFSIZ
   70                                         ? uncomp_size : BUFSIZ;
   71                 }
   72 
   73                 if (ret == XZ_OK)
   74                         continue;
   75 
   76 #ifdef XZ_DEC_ANY_CHECK
   77                 if (ret == XZ_UNSUPPORTED_CHECK) {
   78                         fputs(argv[0], stderr);
   79                         fputs(": ", stderr);
   80                         fputs("Unsupported check; not verifying "
   81                                         "file integrity\n", stderr);
   82                         continue;
   83                 }
   84 #endif
   85 
   86                 if (uncomp_size != b.out_pos) {
   87                         msg = "Uncompressed size doesn't match\n";
   88                         goto error;
   89                 }
   90 
   91                 if (fwrite(out, 1, b.out_pos, stdout) != b.out_pos
   92                                 || fclose(stdout)) {
   93                         msg = "Write error\n";
   94                         goto error;
   95                 }
   96 
   97                 switch (ret) {
   98                 case XZ_STREAM_END:
   99                         xz_dec_end(s);
  100                         return 0;
  101 
  102                 case XZ_MEM_ERROR:
  103                         msg = "Memory allocation failed\n";
  104                         goto error;
  105 
  106                 case XZ_MEMLIMIT_ERROR:
  107                         msg = "Memory usage limit reached\n";
  108                         goto error;
  109 
  110                 case XZ_FORMAT_ERROR:
  111                         msg = "Not a .xz file\n";
  112                         goto error;
  113 
  114                 case XZ_OPTIONS_ERROR:
  115                         msg = "Unsupported options in the .xz headers\n";
  116                         goto error;
  117 
  118                 case XZ_DATA_ERROR:
  119                 case XZ_BUF_ERROR:
  120                         msg = "File is corrupt\n";
  121                         goto error;
  122 
  123                 default:
  124                         msg = "Bug!\n";
  125                         goto error;
  126                 }
  127         }
  128 
  129 error:
  130         xz_dec_end(s);
  131         fputs(argv[0], stderr);
  132         fputs(": ", stderr);
  133         fputs(msg, stderr);
  134         return 1;
  135 }

Cache object: 2b6308941a0fce1ba8cf078ee0bd67c9

[ source navigation ] [ diff markup ] [ identifier search ] [ freetext search ] [ file search ] [ list types ] [ track identifier ]

This page is part of the FreeBSD/Linux Linux Kernel Cross-Reference, and was automatically generated using a modified version of the LXR engine.

FreeBSD/Linux Kernel Cross Reference sys/contrib/xz-embedded/userspace/bytetest.c

FreeBSD/Linux Kernel Cross Reference
sys/contrib/xz-embedded/userspace/bytetest.c