| 1 | /* |
| 2 | * pufftest.c |
| 3 | * Copyright (C) 2002-2013 Mark Adler |
| 4 | * For conditions of distribution and use, see copyright notice in puff.h |
| 5 | * version 2.3, 21 Jan 2013 |
| 6 | */ |
| 7 | |
| 8 | /* Example of how to use puff(). |
| 9 | |
| 10 | Usage: puff [-w] [-f] [-nnn] file |
| 11 | ... | puff [-w] [-f] [-nnn] |
| 12 | |
| 13 | where file is the input file with deflate data, nnn is the number of bytes |
| 14 | of input to skip before inflating (e.g. to skip a zlib or gzip header), and |
| 15 | -w is used to write the decompressed data to stdout. -f is for coverage |
| 16 | testing, and causes pufftest to fail with not enough output space (-f does |
| 17 | a write like -w, so -w is not required). */ |
| 18 | |
| 19 | #include <stdio.h> |
| 20 | #include <stdlib.h> |
| 21 | #include "puff.h" |
| 22 | |
| 23 | #if defined(MSDOS) || defined(OS2) || defined(WIN32) || defined(__CYGWIN__) |
| 24 | # include <fcntl.h> |
| 25 | # include <io.h> |
| 26 | # define SET_BINARY_MODE(file) setmode(fileno(file), O_BINARY) |
| 27 | #else |
| 28 | # define SET_BINARY_MODE(file) |
| 29 | #endif |
| 30 | |
| 31 | #define local static |
| 32 | |
| 33 | /* Return size times approximately the cube root of 2, keeping the result as 1, |
| 34 | 3, or 5 times a power of 2 -- the result is always > size, until the result |
| 35 | is the maximum value of an unsigned long, where it remains. This is useful |
| 36 | to keep reallocations less than ~33% over the actual data. */ |
| 37 | local size_t bythirds(size_t size) |
| 38 | { |
| 39 | int n; |
| 40 | size_t m; |
| 41 | |
| 42 | m = size; |
| 43 | for (n = 0; m; n++) |
| 44 | m >>= 1; |
| 45 | if (n < 3) |
| 46 | return size + 1; |
| 47 | n -= 3; |
| 48 | m = size >> n; |
| 49 | m += m == 6 ? 2 : 1; |
| 50 | m <<= n; |
| 51 | return m > size ? m : (size_t)(-1); |
| 52 | } |
| 53 | |
| 54 | /* Read the input file *name, or stdin if name is NULL, into allocated memory. |
| 55 | Reallocate to larger buffers until the entire file is read in. Return a |
| 56 | pointer to the allocated data, or NULL if there was a memory allocation |
| 57 | failure. *len is the number of bytes of data read from the input file (even |
| 58 | if load() returns NULL). If the input file was empty or could not be opened |
| 59 | or read, *len is zero. */ |
| 60 | local void *load(const char *name, size_t *len) |
| 61 | { |
| 62 | size_t size; |
| 63 | void *buf, *swap; |
| 64 | FILE *in; |
| 65 | |
| 66 | *len = 0; |
| 67 | buf = malloc(size = 4096); |
| 68 | if (buf == NULL) |
| 69 | return NULL; |
| 70 | in = name == NULL ? stdin : fopen(name, "rb"); |
| 71 | if (in != NULL) { |
| 72 | for (;;) { |
| 73 | *len += fread((char *)buf + *len, 1, size - *len, in); |
| 74 | if (*len < size) break; |
| 75 | size = bythirds(size); |
| 76 | if (size == *len || (swap = realloc(buf, size)) == NULL) { |
| 77 | free(buf); |
| 78 | buf = NULL; |
| 79 | break; |
| 80 | } |
| 81 | buf = swap; |
| 82 | } |
| 83 | fclose(in); |
| 84 | } |
| 85 | return buf; |
| 86 | } |
| 87 | |
| 88 | int main(int argc, char **argv) |
| 89 | { |
| 90 | int ret, put = 0, fail = 0; |
| 91 | unsigned skip = 0; |
| 92 | char *arg, *name = NULL; |
| 93 | unsigned char *source = NULL, *dest; |
| 94 | size_t len = 0; |
| 95 | unsigned long sourcelen, destlen; |
| 96 | |
| 97 | /* process arguments */ |
| 98 | while (arg = *++argv, --argc) |
| 99 | if (arg[0] == '-') { |
| 100 | if (arg[1] == 'w' && arg[2] == 0) |
| 101 | put = 1; |
| 102 | else if (arg[1] == 'f' && arg[2] == 0) |
| 103 | fail = 1, put = 1; |
| 104 | else if (arg[1] >= '0' && arg[1] <= '9') |
| 105 | skip = (unsigned)atoi(arg + 1); |
| 106 | else { |
| 107 | fprintf(stderr, "invalid option %s\n", arg); |
| 108 | return 3; |
| 109 | } |
| 110 | } |
| 111 | else if (name != NULL) { |
| 112 | fprintf(stderr, "only one file name allowed\n"); |
| 113 | return 3; |
| 114 | } |
| 115 | else |
| 116 | name = arg; |
| 117 | source = load(name, &len); |
| 118 | if (source == NULL) { |
| 119 | fprintf(stderr, "memory allocation failure\n"); |
| 120 | return 4; |
| 121 | } |
| 122 | if (len == 0) { |
| 123 | fprintf(stderr, "could not read %s, or it was empty\n", |
| 124 | name == NULL ? "<stdin>" : name); |
| 125 | free(source); |
| 126 | return 3; |
| 127 | } |
| 128 | if (skip >= len) { |
| 129 | fprintf(stderr, "skip request of %d leaves no input\n", skip); |
| 130 | free(source); |
| 131 | return 3; |
| 132 | } |
| 133 | |
| 134 | /* test inflate data with offset skip */ |
| 135 | len -= skip; |
| 136 | sourcelen = (unsigned long)len; |
| 137 | ret = puff(NIL, &destlen, source + skip, &sourcelen); |
| 138 | if (ret) |
| 139 | fprintf(stderr, "puff() failed with return code %d\n", ret); |
| 140 | else { |
| 141 | fprintf(stderr, "puff() succeeded uncompressing %lu bytes\n", destlen); |
| 142 | if (sourcelen < len) fprintf(stderr, "%lu compressed bytes unused\n", |
| 143 | len - sourcelen); |
| 144 | } |
| 145 | |
| 146 | /* if requested, inflate again and write decompressed data to stdout */ |
| 147 | if (put && ret == 0) { |
| 148 | if (fail) |
| 149 | destlen >>= 1; |
| 150 | dest = malloc(destlen); |
| 151 | if (dest == NULL) { |
| 152 | fprintf(stderr, "memory allocation failure\n"); |
| 153 | free(source); |
| 154 | return 4; |
| 155 | } |
| 156 | puff(dest, &destlen, source + skip, &sourcelen); |
| 157 | SET_BINARY_MODE(stdout); |
| 158 | fwrite(dest, 1, destlen, stdout); |
| 159 | free(dest); |
| 160 | } |
| 161 | |
| 162 | /* clean up */ |
| 163 | free(source); |
| 164 | return ret; |
| 165 | } |