annotate zlib/test/minigzip.c @ 144:8f4e72ab4e11

fix segmentation fault caused by nothing next cur_op to end
author Takahiro SHIMIZU <anatofuz@cr.ie.u-ryukyu.ac.jp>
date Sun, 23 Dec 2018 21:23:56 +0900
parents 04ced10e8804
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
111
kono
parents:
diff changeset
1 /* minigzip.c -- simulate gzip using the zlib compression library
kono
parents:
diff changeset
2 * Copyright (C) 1995-2006, 2010, 2011, 2016 Jean-loup Gailly
kono
parents:
diff changeset
3 * For conditions of distribution and use, see copyright notice in zlib.h
kono
parents:
diff changeset
4 */
kono
parents:
diff changeset
5
kono
parents:
diff changeset
6 /*
kono
parents:
diff changeset
7 * minigzip is a minimal implementation of the gzip utility. This is
kono
parents:
diff changeset
8 * only an example of using zlib and isn't meant to replace the
kono
parents:
diff changeset
9 * full-featured gzip. No attempt is made to deal with file systems
kono
parents:
diff changeset
10 * limiting names to 14 or 8+3 characters, etc... Error checking is
kono
parents:
diff changeset
11 * very limited. So use minigzip only for testing; use gzip for the
kono
parents:
diff changeset
12 * real thing. On MSDOS, use only on file names without extension
kono
parents:
diff changeset
13 * or in pipe mode.
kono
parents:
diff changeset
14 */
kono
parents:
diff changeset
15
kono
parents:
diff changeset
16 /* @(#) $Id$ */
kono
parents:
diff changeset
17
kono
parents:
diff changeset
18 #include "zlib.h"
kono
parents:
diff changeset
19 #include <stdio.h>
kono
parents:
diff changeset
20
kono
parents:
diff changeset
21 #ifdef STDC
kono
parents:
diff changeset
22 # include <string.h>
kono
parents:
diff changeset
23 # include <stdlib.h>
kono
parents:
diff changeset
24 #endif
kono
parents:
diff changeset
25
kono
parents:
diff changeset
26 #ifdef USE_MMAP
kono
parents:
diff changeset
27 # include <sys/types.h>
kono
parents:
diff changeset
28 # include <sys/mman.h>
kono
parents:
diff changeset
29 # include <sys/stat.h>
kono
parents:
diff changeset
30 #endif
kono
parents:
diff changeset
31
kono
parents:
diff changeset
32 #if defined(MSDOS) || defined(OS2) || defined(WIN32) || defined(__CYGWIN__)
kono
parents:
diff changeset
33 # include <fcntl.h>
kono
parents:
diff changeset
34 # include <io.h>
kono
parents:
diff changeset
35 # ifdef UNDER_CE
kono
parents:
diff changeset
36 # include <stdlib.h>
kono
parents:
diff changeset
37 # endif
kono
parents:
diff changeset
38 # define SET_BINARY_MODE(file) setmode(fileno(file), O_BINARY)
kono
parents:
diff changeset
39 #else
kono
parents:
diff changeset
40 # define SET_BINARY_MODE(file)
kono
parents:
diff changeset
41 #endif
kono
parents:
diff changeset
42
kono
parents:
diff changeset
43 #if defined(_MSC_VER) && _MSC_VER < 1900
kono
parents:
diff changeset
44 # define snprintf _snprintf
kono
parents:
diff changeset
45 #endif
kono
parents:
diff changeset
46
kono
parents:
diff changeset
47 #ifdef VMS
kono
parents:
diff changeset
48 # define unlink delete
kono
parents:
diff changeset
49 # define GZ_SUFFIX "-gz"
kono
parents:
diff changeset
50 #endif
kono
parents:
diff changeset
51 #ifdef RISCOS
kono
parents:
diff changeset
52 # define unlink remove
kono
parents:
diff changeset
53 # define GZ_SUFFIX "-gz"
kono
parents:
diff changeset
54 # define fileno(file) file->__file
kono
parents:
diff changeset
55 #endif
kono
parents:
diff changeset
56 #if defined(__MWERKS__) && __dest_os != __be_os && __dest_os != __win32_os
kono
parents:
diff changeset
57 # include <unix.h> /* for fileno */
kono
parents:
diff changeset
58 #endif
kono
parents:
diff changeset
59
kono
parents:
diff changeset
60 #if !defined(Z_HAVE_UNISTD_H) && !defined(_LARGEFILE64_SOURCE)
kono
parents:
diff changeset
61 #ifndef WIN32 /* unlink already in stdio.h for WIN32 */
kono
parents:
diff changeset
62 extern int unlink OF((const char *));
kono
parents:
diff changeset
63 #endif
kono
parents:
diff changeset
64 #endif
kono
parents:
diff changeset
65
kono
parents:
diff changeset
66 #if defined(UNDER_CE)
kono
parents:
diff changeset
67 # include <windows.h>
kono
parents:
diff changeset
68 # define perror(s) pwinerror(s)
kono
parents:
diff changeset
69
kono
parents:
diff changeset
70 /* Map the Windows error number in ERROR to a locale-dependent error
kono
parents:
diff changeset
71 message string and return a pointer to it. Typically, the values
kono
parents:
diff changeset
72 for ERROR come from GetLastError.
kono
parents:
diff changeset
73
kono
parents:
diff changeset
74 The string pointed to shall not be modified by the application,
kono
parents:
diff changeset
75 but may be overwritten by a subsequent call to strwinerror
kono
parents:
diff changeset
76
kono
parents:
diff changeset
77 The strwinerror function does not change the current setting
kono
parents:
diff changeset
78 of GetLastError. */
kono
parents:
diff changeset
79
kono
parents:
diff changeset
80 static char *strwinerror (error)
kono
parents:
diff changeset
81 DWORD error;
kono
parents:
diff changeset
82 {
kono
parents:
diff changeset
83 static char buf[1024];
kono
parents:
diff changeset
84
kono
parents:
diff changeset
85 wchar_t *msgbuf;
kono
parents:
diff changeset
86 DWORD lasterr = GetLastError();
kono
parents:
diff changeset
87 DWORD chars = FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM
kono
parents:
diff changeset
88 | FORMAT_MESSAGE_ALLOCATE_BUFFER,
kono
parents:
diff changeset
89 NULL,
kono
parents:
diff changeset
90 error,
kono
parents:
diff changeset
91 0, /* Default language */
kono
parents:
diff changeset
92 (LPVOID)&msgbuf,
kono
parents:
diff changeset
93 0,
kono
parents:
diff changeset
94 NULL);
kono
parents:
diff changeset
95 if (chars != 0) {
kono
parents:
diff changeset
96 /* If there is an \r\n appended, zap it. */
kono
parents:
diff changeset
97 if (chars >= 2
kono
parents:
diff changeset
98 && msgbuf[chars - 2] == '\r' && msgbuf[chars - 1] == '\n') {
kono
parents:
diff changeset
99 chars -= 2;
kono
parents:
diff changeset
100 msgbuf[chars] = 0;
kono
parents:
diff changeset
101 }
kono
parents:
diff changeset
102
kono
parents:
diff changeset
103 if (chars > sizeof (buf) - 1) {
kono
parents:
diff changeset
104 chars = sizeof (buf) - 1;
kono
parents:
diff changeset
105 msgbuf[chars] = 0;
kono
parents:
diff changeset
106 }
kono
parents:
diff changeset
107
kono
parents:
diff changeset
108 wcstombs(buf, msgbuf, chars + 1);
kono
parents:
diff changeset
109 LocalFree(msgbuf);
kono
parents:
diff changeset
110 }
kono
parents:
diff changeset
111 else {
kono
parents:
diff changeset
112 sprintf(buf, "unknown win32 error (%ld)", error);
kono
parents:
diff changeset
113 }
kono
parents:
diff changeset
114
kono
parents:
diff changeset
115 SetLastError(lasterr);
kono
parents:
diff changeset
116 return buf;
kono
parents:
diff changeset
117 }
kono
parents:
diff changeset
118
kono
parents:
diff changeset
119 static void pwinerror (s)
kono
parents:
diff changeset
120 const char *s;
kono
parents:
diff changeset
121 {
kono
parents:
diff changeset
122 if (s && *s)
kono
parents:
diff changeset
123 fprintf(stderr, "%s: %s\n", s, strwinerror(GetLastError ()));
kono
parents:
diff changeset
124 else
kono
parents:
diff changeset
125 fprintf(stderr, "%s\n", strwinerror(GetLastError ()));
kono
parents:
diff changeset
126 }
kono
parents:
diff changeset
127
kono
parents:
diff changeset
128 #endif /* UNDER_CE */
kono
parents:
diff changeset
129
kono
parents:
diff changeset
130 #ifndef GZ_SUFFIX
kono
parents:
diff changeset
131 # define GZ_SUFFIX ".gz"
kono
parents:
diff changeset
132 #endif
kono
parents:
diff changeset
133 #define SUFFIX_LEN (sizeof(GZ_SUFFIX)-1)
kono
parents:
diff changeset
134
kono
parents:
diff changeset
135 #define BUFLEN 16384
kono
parents:
diff changeset
136 #define MAX_NAME_LEN 1024
kono
parents:
diff changeset
137
kono
parents:
diff changeset
138 #ifdef MAXSEG_64K
kono
parents:
diff changeset
139 # define local static
kono
parents:
diff changeset
140 /* Needed for systems with limitation on stack size. */
kono
parents:
diff changeset
141 #else
kono
parents:
diff changeset
142 # define local
kono
parents:
diff changeset
143 #endif
kono
parents:
diff changeset
144
kono
parents:
diff changeset
145 #ifdef Z_SOLO
kono
parents:
diff changeset
146 /* for Z_SOLO, create simplified gz* functions using deflate and inflate */
kono
parents:
diff changeset
147
kono
parents:
diff changeset
148 #if defined(Z_HAVE_UNISTD_H) || defined(Z_LARGE)
kono
parents:
diff changeset
149 # include <unistd.h> /* for unlink() */
kono
parents:
diff changeset
150 #endif
kono
parents:
diff changeset
151
kono
parents:
diff changeset
152 void *myalloc OF((void *, unsigned, unsigned));
kono
parents:
diff changeset
153 void myfree OF((void *, void *));
kono
parents:
diff changeset
154
kono
parents:
diff changeset
155 void *myalloc(q, n, m)
kono
parents:
diff changeset
156 void *q;
kono
parents:
diff changeset
157 unsigned n, m;
kono
parents:
diff changeset
158 {
kono
parents:
diff changeset
159 (void)q;
kono
parents:
diff changeset
160 return calloc(n, m);
kono
parents:
diff changeset
161 }
kono
parents:
diff changeset
162
kono
parents:
diff changeset
163 void myfree(q, p)
kono
parents:
diff changeset
164 void *q, *p;
kono
parents:
diff changeset
165 {
kono
parents:
diff changeset
166 (void)q;
kono
parents:
diff changeset
167 free(p);
kono
parents:
diff changeset
168 }
kono
parents:
diff changeset
169
kono
parents:
diff changeset
170 typedef struct gzFile_s {
kono
parents:
diff changeset
171 FILE *file;
kono
parents:
diff changeset
172 int write;
kono
parents:
diff changeset
173 int err;
kono
parents:
diff changeset
174 char *msg;
kono
parents:
diff changeset
175 z_stream strm;
kono
parents:
diff changeset
176 } *gzFile;
kono
parents:
diff changeset
177
kono
parents:
diff changeset
178 gzFile gzopen OF((const char *, const char *));
kono
parents:
diff changeset
179 gzFile gzdopen OF((int, const char *));
kono
parents:
diff changeset
180 gzFile gz_open OF((const char *, int, const char *));
kono
parents:
diff changeset
181
kono
parents:
diff changeset
182 gzFile gzopen(path, mode)
kono
parents:
diff changeset
183 const char *path;
kono
parents:
diff changeset
184 const char *mode;
kono
parents:
diff changeset
185 {
kono
parents:
diff changeset
186 return gz_open(path, -1, mode);
kono
parents:
diff changeset
187 }
kono
parents:
diff changeset
188
kono
parents:
diff changeset
189 gzFile gzdopen(fd, mode)
kono
parents:
diff changeset
190 int fd;
kono
parents:
diff changeset
191 const char *mode;
kono
parents:
diff changeset
192 {
kono
parents:
diff changeset
193 return gz_open(NULL, fd, mode);
kono
parents:
diff changeset
194 }
kono
parents:
diff changeset
195
kono
parents:
diff changeset
196 gzFile gz_open(path, fd, mode)
kono
parents:
diff changeset
197 const char *path;
kono
parents:
diff changeset
198 int fd;
kono
parents:
diff changeset
199 const char *mode;
kono
parents:
diff changeset
200 {
kono
parents:
diff changeset
201 gzFile gz;
kono
parents:
diff changeset
202 int ret;
kono
parents:
diff changeset
203
kono
parents:
diff changeset
204 gz = malloc(sizeof(struct gzFile_s));
kono
parents:
diff changeset
205 if (gz == NULL)
kono
parents:
diff changeset
206 return NULL;
kono
parents:
diff changeset
207 gz->write = strchr(mode, 'w') != NULL;
kono
parents:
diff changeset
208 gz->strm.zalloc = myalloc;
kono
parents:
diff changeset
209 gz->strm.zfree = myfree;
kono
parents:
diff changeset
210 gz->strm.opaque = Z_NULL;
kono
parents:
diff changeset
211 if (gz->write)
kono
parents:
diff changeset
212 ret = deflateInit2(&(gz->strm), -1, 8, 15 + 16, 8, 0);
kono
parents:
diff changeset
213 else {
kono
parents:
diff changeset
214 gz->strm.next_in = 0;
kono
parents:
diff changeset
215 gz->strm.avail_in = Z_NULL;
kono
parents:
diff changeset
216 ret = inflateInit2(&(gz->strm), 15 + 16);
kono
parents:
diff changeset
217 }
kono
parents:
diff changeset
218 if (ret != Z_OK) {
kono
parents:
diff changeset
219 free(gz);
kono
parents:
diff changeset
220 return NULL;
kono
parents:
diff changeset
221 }
kono
parents:
diff changeset
222 gz->file = path == NULL ? fdopen(fd, gz->write ? "wb" : "rb") :
kono
parents:
diff changeset
223 fopen(path, gz->write ? "wb" : "rb");
kono
parents:
diff changeset
224 if (gz->file == NULL) {
kono
parents:
diff changeset
225 gz->write ? deflateEnd(&(gz->strm)) : inflateEnd(&(gz->strm));
kono
parents:
diff changeset
226 free(gz);
kono
parents:
diff changeset
227 return NULL;
kono
parents:
diff changeset
228 }
kono
parents:
diff changeset
229 gz->err = 0;
kono
parents:
diff changeset
230 gz->msg = "";
kono
parents:
diff changeset
231 return gz;
kono
parents:
diff changeset
232 }
kono
parents:
diff changeset
233
kono
parents:
diff changeset
234 int gzwrite OF((gzFile, const void *, unsigned));
kono
parents:
diff changeset
235
kono
parents:
diff changeset
236 int gzwrite(gz, buf, len)
kono
parents:
diff changeset
237 gzFile gz;
kono
parents:
diff changeset
238 const void *buf;
kono
parents:
diff changeset
239 unsigned len;
kono
parents:
diff changeset
240 {
kono
parents:
diff changeset
241 z_stream *strm;
kono
parents:
diff changeset
242 unsigned char out[BUFLEN];
kono
parents:
diff changeset
243
kono
parents:
diff changeset
244 if (gz == NULL || !gz->write)
kono
parents:
diff changeset
245 return 0;
kono
parents:
diff changeset
246 strm = &(gz->strm);
kono
parents:
diff changeset
247 strm->next_in = (void *)buf;
kono
parents:
diff changeset
248 strm->avail_in = len;
kono
parents:
diff changeset
249 do {
kono
parents:
diff changeset
250 strm->next_out = out;
kono
parents:
diff changeset
251 strm->avail_out = BUFLEN;
kono
parents:
diff changeset
252 (void)deflate(strm, Z_NO_FLUSH);
kono
parents:
diff changeset
253 fwrite(out, 1, BUFLEN - strm->avail_out, gz->file);
kono
parents:
diff changeset
254 } while (strm->avail_out == 0);
kono
parents:
diff changeset
255 return len;
kono
parents:
diff changeset
256 }
kono
parents:
diff changeset
257
kono
parents:
diff changeset
258 int gzread OF((gzFile, void *, unsigned));
kono
parents:
diff changeset
259
kono
parents:
diff changeset
260 int gzread(gz, buf, len)
kono
parents:
diff changeset
261 gzFile gz;
kono
parents:
diff changeset
262 void *buf;
kono
parents:
diff changeset
263 unsigned len;
kono
parents:
diff changeset
264 {
kono
parents:
diff changeset
265 int ret;
kono
parents:
diff changeset
266 unsigned got;
kono
parents:
diff changeset
267 unsigned char in[1];
kono
parents:
diff changeset
268 z_stream *strm;
kono
parents:
diff changeset
269
kono
parents:
diff changeset
270 if (gz == NULL || gz->write)
kono
parents:
diff changeset
271 return 0;
kono
parents:
diff changeset
272 if (gz->err)
kono
parents:
diff changeset
273 return 0;
kono
parents:
diff changeset
274 strm = &(gz->strm);
kono
parents:
diff changeset
275 strm->next_out = (void *)buf;
kono
parents:
diff changeset
276 strm->avail_out = len;
kono
parents:
diff changeset
277 do {
kono
parents:
diff changeset
278 got = fread(in, 1, 1, gz->file);
kono
parents:
diff changeset
279 if (got == 0)
kono
parents:
diff changeset
280 break;
kono
parents:
diff changeset
281 strm->next_in = in;
kono
parents:
diff changeset
282 strm->avail_in = 1;
kono
parents:
diff changeset
283 ret = inflate(strm, Z_NO_FLUSH);
kono
parents:
diff changeset
284 if (ret == Z_DATA_ERROR) {
kono
parents:
diff changeset
285 gz->err = Z_DATA_ERROR;
kono
parents:
diff changeset
286 gz->msg = strm->msg;
kono
parents:
diff changeset
287 return 0;
kono
parents:
diff changeset
288 }
kono
parents:
diff changeset
289 if (ret == Z_STREAM_END)
kono
parents:
diff changeset
290 inflateReset(strm);
kono
parents:
diff changeset
291 } while (strm->avail_out);
kono
parents:
diff changeset
292 return len - strm->avail_out;
kono
parents:
diff changeset
293 }
kono
parents:
diff changeset
294
kono
parents:
diff changeset
295 int gzclose OF((gzFile));
kono
parents:
diff changeset
296
kono
parents:
diff changeset
297 int gzclose(gz)
kono
parents:
diff changeset
298 gzFile gz;
kono
parents:
diff changeset
299 {
kono
parents:
diff changeset
300 z_stream *strm;
kono
parents:
diff changeset
301 unsigned char out[BUFLEN];
kono
parents:
diff changeset
302
kono
parents:
diff changeset
303 if (gz == NULL)
kono
parents:
diff changeset
304 return Z_STREAM_ERROR;
kono
parents:
diff changeset
305 strm = &(gz->strm);
kono
parents:
diff changeset
306 if (gz->write) {
kono
parents:
diff changeset
307 strm->next_in = Z_NULL;
kono
parents:
diff changeset
308 strm->avail_in = 0;
kono
parents:
diff changeset
309 do {
kono
parents:
diff changeset
310 strm->next_out = out;
kono
parents:
diff changeset
311 strm->avail_out = BUFLEN;
kono
parents:
diff changeset
312 (void)deflate(strm, Z_FINISH);
kono
parents:
diff changeset
313 fwrite(out, 1, BUFLEN - strm->avail_out, gz->file);
kono
parents:
diff changeset
314 } while (strm->avail_out == 0);
kono
parents:
diff changeset
315 deflateEnd(strm);
kono
parents:
diff changeset
316 }
kono
parents:
diff changeset
317 else
kono
parents:
diff changeset
318 inflateEnd(strm);
kono
parents:
diff changeset
319 fclose(gz->file);
kono
parents:
diff changeset
320 free(gz);
kono
parents:
diff changeset
321 return Z_OK;
kono
parents:
diff changeset
322 }
kono
parents:
diff changeset
323
kono
parents:
diff changeset
324 const char *gzerror OF((gzFile, int *));
kono
parents:
diff changeset
325
kono
parents:
diff changeset
326 const char *gzerror(gz, err)
kono
parents:
diff changeset
327 gzFile gz;
kono
parents:
diff changeset
328 int *err;
kono
parents:
diff changeset
329 {
kono
parents:
diff changeset
330 *err = gz->err;
kono
parents:
diff changeset
331 return gz->msg;
kono
parents:
diff changeset
332 }
kono
parents:
diff changeset
333
kono
parents:
diff changeset
334 #endif
kono
parents:
diff changeset
335
kono
parents:
diff changeset
336 static char *prog;
kono
parents:
diff changeset
337
kono
parents:
diff changeset
338 void error OF((const char *msg));
kono
parents:
diff changeset
339 void gz_compress OF((FILE *in, gzFile out));
kono
parents:
diff changeset
340 #ifdef USE_MMAP
kono
parents:
diff changeset
341 int gz_compress_mmap OF((FILE *in, gzFile out));
kono
parents:
diff changeset
342 #endif
kono
parents:
diff changeset
343 void gz_uncompress OF((gzFile in, FILE *out));
kono
parents:
diff changeset
344 void file_compress OF((char *file, char *mode));
kono
parents:
diff changeset
345 void file_uncompress OF((char *file));
kono
parents:
diff changeset
346 int main OF((int argc, char *argv[]));
kono
parents:
diff changeset
347
kono
parents:
diff changeset
348 /* ===========================================================================
kono
parents:
diff changeset
349 * Display error message and exit
kono
parents:
diff changeset
350 */
kono
parents:
diff changeset
351 void error(msg)
kono
parents:
diff changeset
352 const char *msg;
kono
parents:
diff changeset
353 {
kono
parents:
diff changeset
354 fprintf(stderr, "%s: %s\n", prog, msg);
kono
parents:
diff changeset
355 exit(1);
kono
parents:
diff changeset
356 }
kono
parents:
diff changeset
357
kono
parents:
diff changeset
358 /* ===========================================================================
kono
parents:
diff changeset
359 * Compress input to output then close both files.
kono
parents:
diff changeset
360 */
kono
parents:
diff changeset
361
kono
parents:
diff changeset
362 void gz_compress(in, out)
kono
parents:
diff changeset
363 FILE *in;
kono
parents:
diff changeset
364 gzFile out;
kono
parents:
diff changeset
365 {
kono
parents:
diff changeset
366 local char buf[BUFLEN];
kono
parents:
diff changeset
367 int len;
kono
parents:
diff changeset
368 int err;
kono
parents:
diff changeset
369
kono
parents:
diff changeset
370 #ifdef USE_MMAP
kono
parents:
diff changeset
371 /* Try first compressing with mmap. If mmap fails (minigzip used in a
kono
parents:
diff changeset
372 * pipe), use the normal fread loop.
kono
parents:
diff changeset
373 */
kono
parents:
diff changeset
374 if (gz_compress_mmap(in, out) == Z_OK) return;
kono
parents:
diff changeset
375 #endif
kono
parents:
diff changeset
376 for (;;) {
kono
parents:
diff changeset
377 len = (int)fread(buf, 1, sizeof(buf), in);
kono
parents:
diff changeset
378 if (ferror(in)) {
kono
parents:
diff changeset
379 perror("fread");
kono
parents:
diff changeset
380 exit(1);
kono
parents:
diff changeset
381 }
kono
parents:
diff changeset
382 if (len == 0) break;
kono
parents:
diff changeset
383
kono
parents:
diff changeset
384 if (gzwrite(out, buf, (unsigned)len) != len) error(gzerror(out, &err));
kono
parents:
diff changeset
385 }
kono
parents:
diff changeset
386 fclose(in);
kono
parents:
diff changeset
387 if (gzclose(out) != Z_OK) error("failed gzclose");
kono
parents:
diff changeset
388 }
kono
parents:
diff changeset
389
kono
parents:
diff changeset
390 #ifdef USE_MMAP /* MMAP version, Miguel Albrecht <malbrech@eso.org> */
kono
parents:
diff changeset
391
kono
parents:
diff changeset
392 /* Try compressing the input file at once using mmap. Return Z_OK if
kono
parents:
diff changeset
393 * if success, Z_ERRNO otherwise.
kono
parents:
diff changeset
394 */
kono
parents:
diff changeset
395 int gz_compress_mmap(in, out)
kono
parents:
diff changeset
396 FILE *in;
kono
parents:
diff changeset
397 gzFile out;
kono
parents:
diff changeset
398 {
kono
parents:
diff changeset
399 int len;
kono
parents:
diff changeset
400 int err;
kono
parents:
diff changeset
401 int ifd = fileno(in);
kono
parents:
diff changeset
402 caddr_t buf; /* mmap'ed buffer for the entire input file */
kono
parents:
diff changeset
403 off_t buf_len; /* length of the input file */
kono
parents:
diff changeset
404 struct stat sb;
kono
parents:
diff changeset
405
kono
parents:
diff changeset
406 /* Determine the size of the file, needed for mmap: */
kono
parents:
diff changeset
407 if (fstat(ifd, &sb) < 0) return Z_ERRNO;
kono
parents:
diff changeset
408 buf_len = sb.st_size;
kono
parents:
diff changeset
409 if (buf_len <= 0) return Z_ERRNO;
kono
parents:
diff changeset
410
kono
parents:
diff changeset
411 /* Now do the actual mmap: */
kono
parents:
diff changeset
412 buf = mmap((caddr_t) 0, buf_len, PROT_READ, MAP_SHARED, ifd, (off_t)0);
kono
parents:
diff changeset
413 if (buf == (caddr_t)(-1)) return Z_ERRNO;
kono
parents:
diff changeset
414
kono
parents:
diff changeset
415 /* Compress the whole file at once: */
kono
parents:
diff changeset
416 len = gzwrite(out, (char *)buf, (unsigned)buf_len);
kono
parents:
diff changeset
417
kono
parents:
diff changeset
418 if (len != (int)buf_len) error(gzerror(out, &err));
kono
parents:
diff changeset
419
kono
parents:
diff changeset
420 munmap(buf, buf_len);
kono
parents:
diff changeset
421 fclose(in);
kono
parents:
diff changeset
422 if (gzclose(out) != Z_OK) error("failed gzclose");
kono
parents:
diff changeset
423 return Z_OK;
kono
parents:
diff changeset
424 }
kono
parents:
diff changeset
425 #endif /* USE_MMAP */
kono
parents:
diff changeset
426
kono
parents:
diff changeset
427 /* ===========================================================================
kono
parents:
diff changeset
428 * Uncompress input to output then close both files.
kono
parents:
diff changeset
429 */
kono
parents:
diff changeset
430 void gz_uncompress(in, out)
kono
parents:
diff changeset
431 gzFile in;
kono
parents:
diff changeset
432 FILE *out;
kono
parents:
diff changeset
433 {
kono
parents:
diff changeset
434 local char buf[BUFLEN];
kono
parents:
diff changeset
435 int len;
kono
parents:
diff changeset
436 int err;
kono
parents:
diff changeset
437
kono
parents:
diff changeset
438 for (;;) {
kono
parents:
diff changeset
439 len = gzread(in, buf, sizeof(buf));
kono
parents:
diff changeset
440 if (len < 0) error (gzerror(in, &err));
kono
parents:
diff changeset
441 if (len == 0) break;
kono
parents:
diff changeset
442
kono
parents:
diff changeset
443 if ((int)fwrite(buf, 1, (unsigned)len, out) != len) {
kono
parents:
diff changeset
444 error("failed fwrite");
kono
parents:
diff changeset
445 }
kono
parents:
diff changeset
446 }
kono
parents:
diff changeset
447 if (fclose(out)) error("failed fclose");
kono
parents:
diff changeset
448
kono
parents:
diff changeset
449 if (gzclose(in) != Z_OK) error("failed gzclose");
kono
parents:
diff changeset
450 }
kono
parents:
diff changeset
451
kono
parents:
diff changeset
452
kono
parents:
diff changeset
453 /* ===========================================================================
kono
parents:
diff changeset
454 * Compress the given file: create a corresponding .gz file and remove the
kono
parents:
diff changeset
455 * original.
kono
parents:
diff changeset
456 */
kono
parents:
diff changeset
457 void file_compress(file, mode)
kono
parents:
diff changeset
458 char *file;
kono
parents:
diff changeset
459 char *mode;
kono
parents:
diff changeset
460 {
kono
parents:
diff changeset
461 local char outfile[MAX_NAME_LEN];
kono
parents:
diff changeset
462 FILE *in;
kono
parents:
diff changeset
463 gzFile out;
kono
parents:
diff changeset
464
kono
parents:
diff changeset
465 if (strlen(file) + strlen(GZ_SUFFIX) >= sizeof(outfile)) {
kono
parents:
diff changeset
466 fprintf(stderr, "%s: filename too long\n", prog);
kono
parents:
diff changeset
467 exit(1);
kono
parents:
diff changeset
468 }
kono
parents:
diff changeset
469
kono
parents:
diff changeset
470 #if !defined(NO_snprintf) && !defined(NO_vsnprintf)
kono
parents:
diff changeset
471 snprintf(outfile, sizeof(outfile), "%s%s", file, GZ_SUFFIX);
kono
parents:
diff changeset
472 #else
kono
parents:
diff changeset
473 strcpy(outfile, file);
kono
parents:
diff changeset
474 strcat(outfile, GZ_SUFFIX);
kono
parents:
diff changeset
475 #endif
kono
parents:
diff changeset
476
kono
parents:
diff changeset
477 in = fopen(file, "rb");
kono
parents:
diff changeset
478 if (in == NULL) {
kono
parents:
diff changeset
479 perror(file);
kono
parents:
diff changeset
480 exit(1);
kono
parents:
diff changeset
481 }
kono
parents:
diff changeset
482 out = gzopen(outfile, mode);
kono
parents:
diff changeset
483 if (out == NULL) {
kono
parents:
diff changeset
484 fprintf(stderr, "%s: can't gzopen %s\n", prog, outfile);
kono
parents:
diff changeset
485 exit(1);
kono
parents:
diff changeset
486 }
kono
parents:
diff changeset
487 gz_compress(in, out);
kono
parents:
diff changeset
488
kono
parents:
diff changeset
489 unlink(file);
kono
parents:
diff changeset
490 }
kono
parents:
diff changeset
491
kono
parents:
diff changeset
492
kono
parents:
diff changeset
493 /* ===========================================================================
kono
parents:
diff changeset
494 * Uncompress the given file and remove the original.
kono
parents:
diff changeset
495 */
kono
parents:
diff changeset
496 void file_uncompress(file)
kono
parents:
diff changeset
497 char *file;
kono
parents:
diff changeset
498 {
kono
parents:
diff changeset
499 local char buf[MAX_NAME_LEN];
kono
parents:
diff changeset
500 char *infile, *outfile;
kono
parents:
diff changeset
501 FILE *out;
kono
parents:
diff changeset
502 gzFile in;
kono
parents:
diff changeset
503 unsigned len = strlen(file);
kono
parents:
diff changeset
504
kono
parents:
diff changeset
505 if (len + strlen(GZ_SUFFIX) >= sizeof(buf)) {
kono
parents:
diff changeset
506 fprintf(stderr, "%s: filename too long\n", prog);
kono
parents:
diff changeset
507 exit(1);
kono
parents:
diff changeset
508 }
kono
parents:
diff changeset
509
kono
parents:
diff changeset
510 #if !defined(NO_snprintf) && !defined(NO_vsnprintf)
kono
parents:
diff changeset
511 snprintf(buf, sizeof(buf), "%s", file);
kono
parents:
diff changeset
512 #else
kono
parents:
diff changeset
513 strcpy(buf, file);
kono
parents:
diff changeset
514 #endif
kono
parents:
diff changeset
515
kono
parents:
diff changeset
516 if (len > SUFFIX_LEN && strcmp(file+len-SUFFIX_LEN, GZ_SUFFIX) == 0) {
kono
parents:
diff changeset
517 infile = file;
kono
parents:
diff changeset
518 outfile = buf;
kono
parents:
diff changeset
519 outfile[len-3] = '\0';
kono
parents:
diff changeset
520 } else {
kono
parents:
diff changeset
521 outfile = file;
kono
parents:
diff changeset
522 infile = buf;
kono
parents:
diff changeset
523 #if !defined(NO_snprintf) && !defined(NO_vsnprintf)
kono
parents:
diff changeset
524 snprintf(buf + len, sizeof(buf) - len, "%s", GZ_SUFFIX);
kono
parents:
diff changeset
525 #else
kono
parents:
diff changeset
526 strcat(infile, GZ_SUFFIX);
kono
parents:
diff changeset
527 #endif
kono
parents:
diff changeset
528 }
kono
parents:
diff changeset
529 in = gzopen(infile, "rb");
kono
parents:
diff changeset
530 if (in == NULL) {
kono
parents:
diff changeset
531 fprintf(stderr, "%s: can't gzopen %s\n", prog, infile);
kono
parents:
diff changeset
532 exit(1);
kono
parents:
diff changeset
533 }
kono
parents:
diff changeset
534 out = fopen(outfile, "wb");
kono
parents:
diff changeset
535 if (out == NULL) {
kono
parents:
diff changeset
536 perror(file);
kono
parents:
diff changeset
537 exit(1);
kono
parents:
diff changeset
538 }
kono
parents:
diff changeset
539
kono
parents:
diff changeset
540 gz_uncompress(in, out);
kono
parents:
diff changeset
541
kono
parents:
diff changeset
542 unlink(infile);
kono
parents:
diff changeset
543 }
kono
parents:
diff changeset
544
kono
parents:
diff changeset
545
kono
parents:
diff changeset
546 /* ===========================================================================
kono
parents:
diff changeset
547 * Usage: minigzip [-c] [-d] [-f] [-h] [-r] [-1 to -9] [files...]
kono
parents:
diff changeset
548 * -c : write to standard output
kono
parents:
diff changeset
549 * -d : decompress
kono
parents:
diff changeset
550 * -f : compress with Z_FILTERED
kono
parents:
diff changeset
551 * -h : compress with Z_HUFFMAN_ONLY
kono
parents:
diff changeset
552 * -r : compress with Z_RLE
kono
parents:
diff changeset
553 * -1 to -9 : compression level
kono
parents:
diff changeset
554 */
kono
parents:
diff changeset
555
kono
parents:
diff changeset
556 int main(argc, argv)
kono
parents:
diff changeset
557 int argc;
kono
parents:
diff changeset
558 char *argv[];
kono
parents:
diff changeset
559 {
kono
parents:
diff changeset
560 int copyout = 0;
kono
parents:
diff changeset
561 int uncompr = 0;
kono
parents:
diff changeset
562 gzFile file;
kono
parents:
diff changeset
563 char *bname, outmode[20];
kono
parents:
diff changeset
564
kono
parents:
diff changeset
565 #if !defined(NO_snprintf) && !defined(NO_vsnprintf)
kono
parents:
diff changeset
566 snprintf(outmode, sizeof(outmode), "%s", "wb6 ");
kono
parents:
diff changeset
567 #else
kono
parents:
diff changeset
568 strcpy(outmode, "wb6 ");
kono
parents:
diff changeset
569 #endif
kono
parents:
diff changeset
570
kono
parents:
diff changeset
571 prog = argv[0];
kono
parents:
diff changeset
572 bname = strrchr(argv[0], '/');
kono
parents:
diff changeset
573 if (bname)
kono
parents:
diff changeset
574 bname++;
kono
parents:
diff changeset
575 else
kono
parents:
diff changeset
576 bname = argv[0];
kono
parents:
diff changeset
577 argc--, argv++;
kono
parents:
diff changeset
578
kono
parents:
diff changeset
579 if (!strcmp(bname, "gunzip"))
kono
parents:
diff changeset
580 uncompr = 1;
kono
parents:
diff changeset
581 else if (!strcmp(bname, "zcat"))
kono
parents:
diff changeset
582 copyout = uncompr = 1;
kono
parents:
diff changeset
583
kono
parents:
diff changeset
584 while (argc > 0) {
kono
parents:
diff changeset
585 if (strcmp(*argv, "-c") == 0)
kono
parents:
diff changeset
586 copyout = 1;
kono
parents:
diff changeset
587 else if (strcmp(*argv, "-d") == 0)
kono
parents:
diff changeset
588 uncompr = 1;
kono
parents:
diff changeset
589 else if (strcmp(*argv, "-f") == 0)
kono
parents:
diff changeset
590 outmode[3] = 'f';
kono
parents:
diff changeset
591 else if (strcmp(*argv, "-h") == 0)
kono
parents:
diff changeset
592 outmode[3] = 'h';
kono
parents:
diff changeset
593 else if (strcmp(*argv, "-r") == 0)
kono
parents:
diff changeset
594 outmode[3] = 'R';
kono
parents:
diff changeset
595 else if ((*argv)[0] == '-' && (*argv)[1] >= '1' && (*argv)[1] <= '9' &&
kono
parents:
diff changeset
596 (*argv)[2] == 0)
kono
parents:
diff changeset
597 outmode[2] = (*argv)[1];
kono
parents:
diff changeset
598 else
kono
parents:
diff changeset
599 break;
kono
parents:
diff changeset
600 argc--, argv++;
kono
parents:
diff changeset
601 }
kono
parents:
diff changeset
602 if (outmode[3] == ' ')
kono
parents:
diff changeset
603 outmode[3] = 0;
kono
parents:
diff changeset
604 if (argc == 0) {
kono
parents:
diff changeset
605 SET_BINARY_MODE(stdin);
kono
parents:
diff changeset
606 SET_BINARY_MODE(stdout);
kono
parents:
diff changeset
607 if (uncompr) {
kono
parents:
diff changeset
608 file = gzdopen(fileno(stdin), "rb");
kono
parents:
diff changeset
609 if (file == NULL) error("can't gzdopen stdin");
kono
parents:
diff changeset
610 gz_uncompress(file, stdout);
kono
parents:
diff changeset
611 } else {
kono
parents:
diff changeset
612 file = gzdopen(fileno(stdout), outmode);
kono
parents:
diff changeset
613 if (file == NULL) error("can't gzdopen stdout");
kono
parents:
diff changeset
614 gz_compress(stdin, file);
kono
parents:
diff changeset
615 }
kono
parents:
diff changeset
616 } else {
kono
parents:
diff changeset
617 if (copyout) {
kono
parents:
diff changeset
618 SET_BINARY_MODE(stdout);
kono
parents:
diff changeset
619 }
kono
parents:
diff changeset
620 do {
kono
parents:
diff changeset
621 if (uncompr) {
kono
parents:
diff changeset
622 if (copyout) {
kono
parents:
diff changeset
623 file = gzopen(*argv, "rb");
kono
parents:
diff changeset
624 if (file == NULL)
kono
parents:
diff changeset
625 fprintf(stderr, "%s: can't gzopen %s\n", prog, *argv);
kono
parents:
diff changeset
626 else
kono
parents:
diff changeset
627 gz_uncompress(file, stdout);
kono
parents:
diff changeset
628 } else {
kono
parents:
diff changeset
629 file_uncompress(*argv);
kono
parents:
diff changeset
630 }
kono
parents:
diff changeset
631 } else {
kono
parents:
diff changeset
632 if (copyout) {
kono
parents:
diff changeset
633 FILE * in = fopen(*argv, "rb");
kono
parents:
diff changeset
634
kono
parents:
diff changeset
635 if (in == NULL) {
kono
parents:
diff changeset
636 perror(*argv);
kono
parents:
diff changeset
637 } else {
kono
parents:
diff changeset
638 file = gzdopen(fileno(stdout), outmode);
kono
parents:
diff changeset
639 if (file == NULL) error("can't gzdopen stdout");
kono
parents:
diff changeset
640
kono
parents:
diff changeset
641 gz_compress(in, file);
kono
parents:
diff changeset
642 }
kono
parents:
diff changeset
643
kono
parents:
diff changeset
644 } else {
kono
parents:
diff changeset
645 file_compress(*argv, outmode);
kono
parents:
diff changeset
646 }
kono
parents:
diff changeset
647 }
kono
parents:
diff changeset
648 } while (argv++, --argc);
kono
parents:
diff changeset
649 }
kono
parents:
diff changeset
650 return 0;
kono
parents:
diff changeset
651 }