]>
Commit | Line | Data |
---|---|---|
9acb8ddf MK |
1 | /* |
2 | * pufftest.c | |
cd9ec142 | 3 | * Copyright (C) 2002-2013 Mark Adler |
9acb8ddf | 4 | * For conditions of distribution and use, see copyright notice in puff.h |
cd9ec142 | 5 | * version 2.3, 21 Jan 2013 |
9acb8ddf MK |
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 decompressd 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 | } |