9e052883 |
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 | } |