blob: e275fa274210f725a8f457d93f216d52cf3989d7 [file] [log] [blame]
Erik Andersene49d5ec2000-02-08 19:58:47 +00001/* vi: set sw=4 ts=4: */
Eric Andersencc8ed391999-10-05 16:24:54 +00002/* gzip.c -- this is a stripped down version of gzip I put into busybox, it does
3 * only standard in to standard out with -9 compression. It also requires the
4 * zcat module for some important functions.
5 *
6 * Charles P. Wright <cpw@unix.asb.com>
7 */
8#include "internal.h"
9#ifdef BB_GZIP
10
Eric Andersen0dfac6b1999-11-11 05:46:32 +000011//#ifndef BB_ZCAT
12//#error you need zcat to have gzip support!
13//#endif
Eric Andersencc8ed391999-10-05 16:24:54 +000014
Eric Andersenc296b541999-11-11 01:36:55 +000015static const char gzip_usage[] =
Erik Andersene49d5ec2000-02-08 19:58:47 +000016 "gzip [OPTION]... FILE\n\n"
17 "Compress FILE with maximum compression.\n"
18 "When FILE is -, reads standard input. Implies -c.\n\n"
19
20 "Options:\n"
21 "\t-c\tWrite output to standard output instead of FILE.gz\n";
Eric Andersenc296b541999-11-11 01:36:55 +000022
Eric Andersencc8ed391999-10-05 16:24:54 +000023
24/* gzip.h -- common declarations for all gzip modules
25 * Copyright (C) 1992-1993 Jean-loup Gailly.
26 * This is free software; you can redistribute it and/or modify it under the
27 * terms of the GNU General Public License, see the file COPYING.
28 */
29
30#if defined(__STDC__) || defined(PROTO)
31# define OF(args) args
32#else
33# define OF(args) ()
34#endif
35
36#ifdef __STDC__
Erik Andersene49d5ec2000-02-08 19:58:47 +000037typedef void *voidp;
Eric Andersencc8ed391999-10-05 16:24:54 +000038#else
Erik Andersene49d5ec2000-02-08 19:58:47 +000039typedef char *voidp;
Eric Andersencc8ed391999-10-05 16:24:54 +000040#endif
41
42/* I don't like nested includes, but the string and io functions are used
43 * too often
44 */
45#include <stdio.h>
46#if !defined(NO_STRING_H) || defined(STDC_HEADERS)
47# include <string.h>
48# if !defined(STDC_HEADERS) && !defined(NO_MEMORY_H) && !defined(__GNUC__)
49# include <memory.h>
50# endif
51# define memzero(s, n) memset ((voidp)(s), 0, (n))
52#else
53# include <strings.h>
Erik Andersene49d5ec2000-02-08 19:58:47 +000054# define strchr index
Eric Andersencc8ed391999-10-05 16:24:54 +000055# define strrchr rindex
Erik Andersene49d5ec2000-02-08 19:58:47 +000056# define memcpy(d, s, n) bcopy((s), (d), (n))
57# define memcmp(s1, s2, n) bcmp((s1), (s2), (n))
Eric Andersencc8ed391999-10-05 16:24:54 +000058# define memzero(s, n) bzero((s), (n))
59#endif
60
61#ifndef RETSIGTYPE
62# define RETSIGTYPE void
63#endif
64
65#define local static
66
Erik Andersene49d5ec2000-02-08 19:58:47 +000067typedef unsigned char uch;
Eric Andersencc8ed391999-10-05 16:24:54 +000068typedef unsigned short ush;
Erik Andersene49d5ec2000-02-08 19:58:47 +000069typedef unsigned long ulg;
Eric Andersencc8ed391999-10-05 16:24:54 +000070
71/* Return codes from gzip */
72#define OK 0
73#define ERROR 1
74#define WARNING 2
75
76/* Compression methods (see algorithm.doc) */
77#define STORED 0
78#define COMPRESSED 1
79#define PACKED 2
80#define LZHED 3
81/* methods 4 to 7 reserved */
82#define DEFLATED 8
83#define MAX_METHODS 9
Erik Andersene49d5ec2000-02-08 19:58:47 +000084extern int method; /* compression method */
Eric Andersencc8ed391999-10-05 16:24:54 +000085
86/* To save memory for 16 bit systems, some arrays are overlaid between
87 * the various modules:
88 * deflate: prev+head window d_buf l_buf outbuf
89 * unlzw: tab_prefix tab_suffix stack inbuf outbuf
90 * inflate: window inbuf
91 * unpack: window inbuf prefix_len
92 * unlzh: left+right window c_table inbuf c_len
93 * For compression, input is done in window[]. For decompression, output
94 * is done in window except for unlzw.
95 */
96
97#ifndef INBUFSIZ
98# ifdef SMALL_MEM
Erik Andersene49d5ec2000-02-08 19:58:47 +000099# define INBUFSIZ 0x2000 /* input buffer size */
Eric Andersencc8ed391999-10-05 16:24:54 +0000100# else
Erik Andersene49d5ec2000-02-08 19:58:47 +0000101# define INBUFSIZ 0x8000 /* input buffer size */
Eric Andersencc8ed391999-10-05 16:24:54 +0000102# endif
103#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +0000104#define INBUF_EXTRA 64 /* required by unlzw() */
Eric Andersencc8ed391999-10-05 16:24:54 +0000105
106#ifndef OUTBUFSIZ
107# ifdef SMALL_MEM
Erik Andersene49d5ec2000-02-08 19:58:47 +0000108# define OUTBUFSIZ 8192 /* output buffer size */
Eric Andersencc8ed391999-10-05 16:24:54 +0000109# else
Erik Andersene49d5ec2000-02-08 19:58:47 +0000110# define OUTBUFSIZ 16384 /* output buffer size */
Eric Andersencc8ed391999-10-05 16:24:54 +0000111# endif
112#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +0000113#define OUTBUF_EXTRA 2048 /* required by unlzw() */
Eric Andersencc8ed391999-10-05 16:24:54 +0000114
115#ifndef DIST_BUFSIZE
116# ifdef SMALL_MEM
Erik Andersene49d5ec2000-02-08 19:58:47 +0000117# define DIST_BUFSIZE 0x2000 /* buffer for distances, see trees.c */
Eric Andersencc8ed391999-10-05 16:24:54 +0000118# else
Erik Andersene49d5ec2000-02-08 19:58:47 +0000119# define DIST_BUFSIZE 0x8000 /* buffer for distances, see trees.c */
Eric Andersencc8ed391999-10-05 16:24:54 +0000120# endif
121#endif
122
123#ifdef DYN_ALLOC
124# define EXTERN(type, array) extern type * near array
125# define DECLARE(type, array, size) type * near array
126# define ALLOC(type, array, size) { \
127 array = (type*)fcalloc((size_t)(((size)+1L)/2), 2*sizeof(type)); \
Erik Andersen9ffdaa62000-02-11 21:55:04 +0000128 if (array == NULL) errorMsg("insufficient memory"); \
Eric Andersencc8ed391999-10-05 16:24:54 +0000129 }
130# define FREE(array) {if (array != NULL) fcfree(array), array=NULL;}
131#else
132# define EXTERN(type, array) extern type array[]
133# define DECLARE(type, array, size) type array[size]
134# define ALLOC(type, array, size)
135# define FREE(array)
136#endif
137
Erik Andersene49d5ec2000-02-08 19:58:47 +0000138EXTERN(uch, inbuf); /* input buffer */
139EXTERN(uch, outbuf); /* output buffer */
140EXTERN(ush, d_buf); /* buffer for distances, see trees.c */
141EXTERN(uch, window); /* Sliding window and suffix table (unlzw) */
Eric Andersencc8ed391999-10-05 16:24:54 +0000142#define tab_suffix window
143#ifndef MAXSEG_64K
Erik Andersene49d5ec2000-02-08 19:58:47 +0000144# define tab_prefix prev /* hash link (see deflate.c) */
145# define head (prev+WSIZE) /* hash head (see deflate.c) */
146EXTERN(ush, tab_prefix); /* prefix code (see unlzw.c) */
Eric Andersencc8ed391999-10-05 16:24:54 +0000147#else
148# define tab_prefix0 prev
149# define head tab_prefix1
Erik Andersene49d5ec2000-02-08 19:58:47 +0000150EXTERN(ush, tab_prefix0); /* prefix for even codes */
151EXTERN(ush, tab_prefix1); /* prefix for odd codes */
Eric Andersencc8ed391999-10-05 16:24:54 +0000152#endif
153
Erik Andersene49d5ec2000-02-08 19:58:47 +0000154extern unsigned insize; /* valid bytes in inbuf */
155extern unsigned inptr; /* index of next byte to be processed in inbuf */
156extern unsigned outcnt; /* bytes in output buffer */
Eric Andersencc8ed391999-10-05 16:24:54 +0000157
Erik Andersene49d5ec2000-02-08 19:58:47 +0000158extern long bytes_in; /* number of input bytes */
159extern long bytes_out; /* number of output bytes */
160extern long header_bytes; /* number of bytes in gzip header */
Eric Andersencc8ed391999-10-05 16:24:54 +0000161
162#define isize bytes_in
163/* for compatibility with old zip sources (to be cleaned) */
164
Erik Andersene49d5ec2000-02-08 19:58:47 +0000165extern int ifd; /* input file descriptor */
166extern int ofd; /* output file descriptor */
167extern char ifname[]; /* input file name or "stdin" */
168extern char ofname[]; /* output file name or "stdout" */
169extern char *progname; /* program name */
Eric Andersencc8ed391999-10-05 16:24:54 +0000170
Erik Andersene49d5ec2000-02-08 19:58:47 +0000171extern long time_stamp; /* original time stamp (modification time) */
172extern long ifile_size; /* input file size, -1 for devices (debug only) */
Eric Andersencc8ed391999-10-05 16:24:54 +0000173
Erik Andersene49d5ec2000-02-08 19:58:47 +0000174typedef int file_t; /* Do not use stdio */
175
176#define NO_FILE (-1) /* in memory compression */
Eric Andersencc8ed391999-10-05 16:24:54 +0000177
178
Erik Andersene49d5ec2000-02-08 19:58:47 +0000179#define PACK_MAGIC "\037\036" /* Magic header for packed files */
180#define GZIP_MAGIC "\037\213" /* Magic header for gzip files, 1F 8B */
181#define OLD_GZIP_MAGIC "\037\236" /* Magic header for gzip 0.5 = freeze 1.x */
182#define LZH_MAGIC "\037\240" /* Magic header for SCO LZH Compress files */
183#define PKZIP_MAGIC "\120\113\003\004" /* Magic header for pkzip files */
Eric Andersencc8ed391999-10-05 16:24:54 +0000184
185/* gzip flag byte */
Erik Andersene49d5ec2000-02-08 19:58:47 +0000186#define ASCII_FLAG 0x01 /* bit 0 set: file probably ascii text */
187#define CONTINUATION 0x02 /* bit 1 set: continuation of multi-part gzip file */
188#define EXTRA_FIELD 0x04 /* bit 2 set: extra field present */
189#define ORIG_NAME 0x08 /* bit 3 set: original file name present */
190#define COMMENT 0x10 /* bit 4 set: file comment present */
191#define ENCRYPTED 0x20 /* bit 5 set: file is encrypted */
192#define RESERVED 0xC0 /* bit 6,7: reserved */
Eric Andersencc8ed391999-10-05 16:24:54 +0000193
194/* internal file attribute */
195#define UNKNOWN 0xffff
196#define BINARY 0
197#define ASCII 1
198
199#ifndef WSIZE
Erik Andersene49d5ec2000-02-08 19:58:47 +0000200# define WSIZE 0x8000 /* window size--must be a power of two, and */
201#endif /* at least 32K for zip's deflate method */
Eric Andersencc8ed391999-10-05 16:24:54 +0000202
203#define MIN_MATCH 3
204#define MAX_MATCH 258
205/* The minimum and maximum match lengths */
206
207#define MIN_LOOKAHEAD (MAX_MATCH+MIN_MATCH+1)
208/* Minimum amount of lookahead, except at the end of the input file.
209 * See deflate.c for comments about the MIN_MATCH+1.
210 */
211
212#define MAX_DIST (WSIZE-MIN_LOOKAHEAD)
213/* In order to simplify the code, particularly on 16 bit machines, match
214 * distances are limited to MAX_DIST instead of WSIZE.
215 */
216
Erik Andersene49d5ec2000-02-08 19:58:47 +0000217extern int decrypt; /* flag to turn on decryption */
218extern int exit_code; /* program exit code */
219extern int verbose; /* be verbose (-v) */
220extern int quiet; /* be quiet (-q) */
221extern int test; /* check .z file integrity */
222extern int save_orig_name; /* set if original name must be saved */
Eric Andersencc8ed391999-10-05 16:24:54 +0000223
224#define get_byte() (inptr < insize ? inbuf[inptr++] : fill_inbuf(0))
225#define try_byte() (inptr < insize ? inbuf[inptr++] : fill_inbuf(1))
226
227/* put_byte is used for the compressed output, put_ubyte for the
228 * uncompressed output. However unlzw() uses window for its
229 * suffix table instead of its output buffer, so it does not use put_ubyte
230 * (to be cleaned up).
231 */
232#define put_byte(c) {outbuf[outcnt++]=(uch)(c); if (outcnt==OUTBUFSIZ)\
233 flush_outbuf();}
234#define put_ubyte(c) {window[outcnt++]=(uch)(c); if (outcnt==WSIZE)\
235 flush_window();}
236
237/* Output a 16 bit value, lsb first */
238#define put_short(w) \
239{ if (outcnt < OUTBUFSIZ-2) { \
240 outbuf[outcnt++] = (uch) ((w) & 0xff); \
241 outbuf[outcnt++] = (uch) ((ush)(w) >> 8); \
242 } else { \
243 put_byte((uch)((w) & 0xff)); \
244 put_byte((uch)((ush)(w) >> 8)); \
245 } \
246}
247
248/* Output a 32 bit value to the bit stream, lsb first */
249#define put_long(n) { \
250 put_short((n) & 0xffff); \
251 put_short(((ulg)(n)) >> 16); \
252}
253
Erik Andersene49d5ec2000-02-08 19:58:47 +0000254#define seekable() 0 /* force sequential output */
255#define translate_eol 0 /* no option -a yet */
Eric Andersencc8ed391999-10-05 16:24:54 +0000256
Erik Andersene49d5ec2000-02-08 19:58:47 +0000257#define tolow(c) (isupper(c) ? (c)-'A'+'a' : (c)) /* force to lower case */
Eric Andersencc8ed391999-10-05 16:24:54 +0000258
259/* Macros for getting two-byte and four-byte header values */
260#define SH(p) ((ush)(uch)((p)[0]) | ((ush)(uch)((p)[1]) << 8))
261#define LG(p) ((ulg)(SH(p)) | ((ulg)(SH((p)+2)) << 16))
262
263/* Diagnostic functions */
264#ifdef DEBUG
Erik Andersen9ffdaa62000-02-11 21:55:04 +0000265# define Assert(cond,msg) {if(!(cond)) errorMsg(msg);}
Eric Andersencc8ed391999-10-05 16:24:54 +0000266# define Trace(x) fprintf x
267# define Tracev(x) {if (verbose) fprintf x ;}
268# define Tracevv(x) {if (verbose>1) fprintf x ;}
269# define Tracec(c,x) {if (verbose && (c)) fprintf x ;}
270# define Tracecv(c,x) {if (verbose>1 && (c)) fprintf x ;}
271#else
272# define Assert(cond,msg)
273# define Trace(x)
274# define Tracev(x)
275# define Tracevv(x)
276# define Tracec(c,x)
277# define Tracecv(c,x)
278#endif
279
280#define WARN(msg) {if (!quiet) fprintf msg ; \
281 if (exit_code == OK) exit_code = WARNING;}
282
Erik Andersen3fe39dc2000-01-25 18:13:53 +0000283#define do_exit(c) exit(c)
284
Eric Andersencc8ed391999-10-05 16:24:54 +0000285
286 /* in zip.c: */
Erik Andersene49d5ec2000-02-08 19:58:47 +0000287extern int zip OF((int in, int out));
288extern int file_read OF((char *buf, unsigned size));
Eric Andersencc8ed391999-10-05 16:24:54 +0000289
290 /* in unzip.c */
Erik Andersene49d5ec2000-02-08 19:58:47 +0000291extern int unzip OF((int in, int out));
Eric Andersencc8ed391999-10-05 16:24:54 +0000292extern int check_zipfile OF((int in));
293
294 /* in unpack.c */
Erik Andersene49d5ec2000-02-08 19:58:47 +0000295extern int unpack OF((int in, int out));
Eric Andersencc8ed391999-10-05 16:24:54 +0000296
297 /* in unlzh.c */
Erik Andersene49d5ec2000-02-08 19:58:47 +0000298extern int unlzh OF((int in, int out));
Eric Andersencc8ed391999-10-05 16:24:54 +0000299
300 /* in gzip.c */
301RETSIGTYPE abort_gzip OF((void));
302
Erik Andersene49d5ec2000-02-08 19:58:47 +0000303 /* in deflate.c */
304void lm_init OF((ush * flags));
305ulg deflate OF((void));
Eric Andersencc8ed391999-10-05 16:24:54 +0000306
Erik Andersene49d5ec2000-02-08 19:58:47 +0000307 /* in trees.c */
308void ct_init OF((ush * attr, int *method));
309int ct_tally OF((int dist, int lc));
310ulg flush_block OF((char *buf, ulg stored_len, int eof));
Eric Andersencc8ed391999-10-05 16:24:54 +0000311
Erik Andersene49d5ec2000-02-08 19:58:47 +0000312 /* in bits.c */
313void bi_init OF((file_t zipfile));
314void send_bits OF((int value, int length));
Eric Andersencc8ed391999-10-05 16:24:54 +0000315unsigned bi_reverse OF((unsigned value, int length));
Erik Andersene49d5ec2000-02-08 19:58:47 +0000316void bi_windup OF((void));
317void copy_block OF((char *buf, unsigned len, int header));
318extern int (*read_buf) OF((char *buf, unsigned size));
Eric Andersencc8ed391999-10-05 16:24:54 +0000319
320 /* in util.c: */
Erik Andersene49d5ec2000-02-08 19:58:47 +0000321extern int copy OF((int in, int out));
322extern ulg updcrc OF((uch * s, unsigned n));
323extern void clear_bufs OF((void));
324extern int fill_inbuf OF((int eof_ok));
325extern void flush_outbuf OF((void));
326extern void flush_window OF((void));
327extern void write_buf OF((int fd, voidp buf, unsigned cnt));
328extern char *strlwr OF((char *s));
329extern char *add_envopt OF((int *argcp, char ***argvp, char *env));
Erik Andersene49d5ec2000-02-08 19:58:47 +0000330extern void read_error OF((void));
331extern void write_error OF((void));
332extern void display_ratio OF((long num, long den, FILE * file));
Eric Andersencc8ed391999-10-05 16:24:54 +0000333
334 /* in inflate.c */
335extern int inflate OF((void));
Erik Andersene49d5ec2000-02-08 19:58:47 +0000336
Eric Andersencc8ed391999-10-05 16:24:54 +0000337/* lzw.h -- define the lzw functions.
338 * Copyright (C) 1992-1993 Jean-loup Gailly.
339 * This is free software; you can redistribute it and/or modify it under the
340 * terms of the GNU General Public License, see the file COPYING.
341 */
342
343#if !defined(OF) && defined(lint)
344# include "gzip.h"
345#endif
346
347#ifndef BITS
348# define BITS 16
349#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +0000350#define INIT_BITS 9 /* Initial number of bits per code */
Eric Andersencc8ed391999-10-05 16:24:54 +0000351
Erik Andersene49d5ec2000-02-08 19:58:47 +0000352#define BIT_MASK 0x1f /* Mask for 'number of compression bits' */
Eric Andersencc8ed391999-10-05 16:24:54 +0000353/* Mask 0x20 is reserved to mean a fourth header byte, and 0x40 is free.
354 * It's a pity that old uncompress does not check bit 0x20. That makes
355 * extension of the format actually undesirable because old compress
356 * would just crash on the new format instead of giving a meaningful
357 * error message. It does check the number of bits, but it's more
358 * helpful to say "unsupported format, get a new version" than
359 * "can only handle 16 bits".
360 */
361
362#define BLOCK_MODE 0x80
363/* Block compression: if table is full and compression rate is dropping,
364 * clear the dictionary.
365 */
366
Erik Andersene49d5ec2000-02-08 19:58:47 +0000367#define LZW_RESERVED 0x60 /* reserved bits */
Eric Andersencc8ed391999-10-05 16:24:54 +0000368
Erik Andersene49d5ec2000-02-08 19:58:47 +0000369#define CLEAR 256 /* flush the dictionary */
370#define FIRST (CLEAR+1) /* first free entry */
Eric Andersencc8ed391999-10-05 16:24:54 +0000371
Erik Andersene49d5ec2000-02-08 19:58:47 +0000372extern int maxbits; /* max bits per code for LZW */
373extern int block_mode; /* block compress mode -C compatible with 2.0 */
Eric Andersencc8ed391999-10-05 16:24:54 +0000374
375/* revision.h -- define the version number
376 * Copyright (C) 1992-1993 Jean-loup Gailly.
377 * This is free software; you can redistribute it and/or modify it under the
378 * terms of the GNU General Public License, see the file COPYING.
379 */
380
381#define VERSION "1.2.4"
382#define PATCHLEVEL 0
383#define REVDATE "18 Aug 93"
384
385/* This version does not support compression into old compress format: */
386#ifdef LZW
387# undef LZW
388#endif
389
Eric Andersencc8ed391999-10-05 16:24:54 +0000390/* tailor.h -- target dependent definitions
391 * Copyright (C) 1992-1993 Jean-loup Gailly.
392 * This is free software; you can redistribute it and/or modify it under the
393 * terms of the GNU General Public License, see the file COPYING.
394 */
395
396/* The target dependent definitions should be defined here only.
397 * The target dependent functions should be defined in tailor.c.
398 */
399
Eric Andersencc8ed391999-10-05 16:24:54 +0000400
401#if defined(__MSDOS__) && !defined(MSDOS)
402# define MSDOS
403#endif
404
405#if defined(__OS2__) && !defined(OS2)
406# define OS2
407#endif
408
Erik Andersene49d5ec2000-02-08 19:58:47 +0000409#if defined(OS2) && defined(MSDOS) /* MS C under OS/2 */
Eric Andersencc8ed391999-10-05 16:24:54 +0000410# undef MSDOS
411#endif
412
413#ifdef MSDOS
414# ifdef __GNUC__
Erik Andersene49d5ec2000-02-08 19:58:47 +0000415 /* DJGPP version 1.09+ on MS-DOS.
416 * The DJGPP 1.09 stat() function must be upgraded before gzip will
417 * fully work.
418 * No need for DIRENT, since <unistd.h> defines POSIX_SOURCE which
419 * implies DIRENT.
420 */
Eric Andersencc8ed391999-10-05 16:24:54 +0000421# define near
422# else
423# define MAXSEG_64K
424# ifdef __TURBOC__
425# define NO_OFF_T
426# ifdef __BORLANDC__
427# define DIRENT
428# else
429# define NO_UTIME
430# endif
Erik Andersene49d5ec2000-02-08 19:58:47 +0000431# else /* MSC */
Eric Andersencc8ed391999-10-05 16:24:54 +0000432# define HAVE_SYS_UTIME_H
433# define NO_UTIME_H
434# endif
435# endif
436# define PATH_SEP2 '\\'
437# define PATH_SEP3 ':'
438# define MAX_PATH_LEN 128
439# define NO_MULTIPLE_DOTS
440# define MAX_EXT_CHARS 3
441# define Z_SUFFIX "z"
442# define NO_CHOWN
443# define PROTO
444# define STDC_HEADERS
445# define NO_SIZE_CHECK
Erik Andersene49d5ec2000-02-08 19:58:47 +0000446# define casemap(c) tolow(c) /* Force file names to lower case */
Eric Andersencc8ed391999-10-05 16:24:54 +0000447# include <io.h>
448# define OS_CODE 0x00
449# define SET_BINARY_MODE(fd) setmode(fd, O_BINARY)
450# if !defined(NO_ASM) && !defined(ASMV)
451# define ASMV
452# endif
453#else
454# define near
455#endif
456
457#ifdef OS2
458# define PATH_SEP2 '\\'
459# define PATH_SEP3 ':'
460# define MAX_PATH_LEN 260
461# ifdef OS2FAT
462# define NO_MULTIPLE_DOTS
463# define MAX_EXT_CHARS 3
464# define Z_SUFFIX "z"
465# define casemap(c) tolow(c)
466# endif
467# define NO_CHOWN
468# define PROTO
469# define STDC_HEADERS
470# include <io.h>
471# define OS_CODE 0x06
472# define SET_BINARY_MODE(fd) setmode(fd, O_BINARY)
473# ifdef _MSC_VER
474# define HAVE_SYS_UTIME_H
475# define NO_UTIME_H
476# define MAXSEG_64K
477# undef near
478# define near _near
479# endif
480# ifdef __EMX__
481# define HAVE_SYS_UTIME_H
482# define NO_UTIME_H
483# define DIRENT
484# define EXPAND(argc,argv) \
485 {_response(&argc, &argv); _wildcard(&argc, &argv);}
486# endif
487# ifdef __BORLANDC__
488# define DIRENT
489# endif
490# ifdef __ZTC__
491# define NO_DIR
492# define NO_UTIME_H
493# include <dos.h>
494# define EXPAND(argc,argv) \
495 {response_expand(&argc, &argv);}
496# endif
497#endif
498
Erik Andersene49d5ec2000-02-08 19:58:47 +0000499#ifdef WIN32 /* Windows NT */
Eric Andersencc8ed391999-10-05 16:24:54 +0000500# define HAVE_SYS_UTIME_H
501# define NO_UTIME_H
502# define PATH_SEP2 '\\'
503# define PATH_SEP3 ':'
504# define MAX_PATH_LEN 260
505# define NO_CHOWN
506# define PROTO
507# define STDC_HEADERS
508# define SET_BINARY_MODE(fd) setmode(fd, O_BINARY)
509# include <io.h>
510# include <malloc.h>
511# ifdef NTFAT
512# define NO_MULTIPLE_DOTS
513# define MAX_EXT_CHARS 3
514# define Z_SUFFIX "z"
Erik Andersene49d5ec2000-02-08 19:58:47 +0000515# define casemap(c) tolow(c) /* Force file names to lower case */
Eric Andersencc8ed391999-10-05 16:24:54 +0000516# endif
517# define OS_CODE 0x0b
518#endif
519
520#ifdef MSDOS
521# ifdef __TURBOC__
522# include <alloc.h>
523# define DYN_ALLOC
Erik Andersene49d5ec2000-02-08 19:58:47 +0000524 /* Turbo C 2.0 does not accept static allocations of large arrays */
525void *fcalloc(unsigned items, unsigned size);
526void fcfree(void *ptr);
527# else /* MSC */
Eric Andersencc8ed391999-10-05 16:24:54 +0000528# include <malloc.h>
529# define fcalloc(nitems,itemsize) halloc((long)(nitems),(itemsize))
530# define fcfree(ptr) hfree(ptr)
531# endif
532#else
533# ifdef MAXSEG_64K
534# define fcalloc(items,size) calloc((items),(size))
535# else
536# define fcalloc(items,size) malloc((size_t)(items)*(size_t)(size))
537# endif
538# define fcfree(ptr) free(ptr)
539#endif
540
541#if defined(VAXC) || defined(VMS)
542# define PATH_SEP ']'
543# define PATH_SEP2 ':'
544# define SUFFIX_SEP ';'
545# define NO_MULTIPLE_DOTS
546# define Z_SUFFIX "-gz"
547# define RECORD_IO 1
548# define casemap(c) tolow(c)
549# define OS_CODE 0x02
550# define OPTIONS_VAR "GZIP_OPT"
551# define STDC_HEADERS
552# define NO_UTIME
553# define EXPAND(argc,argv) vms_expand_args(&argc,&argv);
554# include <file.h>
555# define unlink delete
556# ifdef VAXC
557# define NO_FCNTL_H
558# include <unixio.h>
559# endif
560#endif
561
562#ifdef AMIGA
563# define PATH_SEP2 ':'
564# define STDC_HEADERS
565# define OS_CODE 0x01
566# define ASMV
567# ifdef __GNUC__
568# define DIRENT
569# define HAVE_UNISTD_H
Erik Andersene49d5ec2000-02-08 19:58:47 +0000570# else /* SASC */
Eric Andersencc8ed391999-10-05 16:24:54 +0000571# define NO_STDIN_FSTAT
572# define SYSDIR
573# define NO_SYMLINK
574# define NO_CHOWN
575# define NO_FCNTL_H
Erik Andersene49d5ec2000-02-08 19:58:47 +0000576# include <fcntl.h> /* for read() and write() */
Eric Andersencc8ed391999-10-05 16:24:54 +0000577# define direct dirent
Erik Andersene49d5ec2000-02-08 19:58:47 +0000578extern void _expand_args(int *argc, char ***argv);
579
Eric Andersencc8ed391999-10-05 16:24:54 +0000580# define EXPAND(argc,argv) _expand_args(&argc,&argv);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000581# undef O_BINARY /* disable useless --ascii option */
Eric Andersencc8ed391999-10-05 16:24:54 +0000582# endif
583#endif
584
585#if defined(ATARI) || defined(atarist)
586# ifndef STDC_HEADERS
587# define STDC_HEADERS
588# define HAVE_UNISTD_H
589# define DIRENT
590# endif
591# define ASMV
592# define OS_CODE 0x05
593# ifdef TOSFS
594# define PATH_SEP2 '\\'
595# define PATH_SEP3 ':'
596# define MAX_PATH_LEN 128
597# define NO_MULTIPLE_DOTS
598# define MAX_EXT_CHARS 3
599# define Z_SUFFIX "z"
600# define NO_CHOWN
Erik Andersene49d5ec2000-02-08 19:58:47 +0000601# define casemap(c) tolow(c) /* Force file names to lower case */
Eric Andersencc8ed391999-10-05 16:24:54 +0000602# define NO_SYMLINK
603# endif
604#endif
605
606#ifdef MACOS
607# define PATH_SEP ':'
608# define DYN_ALLOC
609# define PROTO
610# define NO_STDIN_FSTAT
611# define NO_CHOWN
612# define NO_UTIME
613# define chmod(file, mode) (0)
614# define OPEN(name, flags, mode) open(name, flags)
615# define OS_CODE 0x07
616# ifdef MPW
617# define isatty(fd) ((fd) <= 2)
618# endif
619#endif
620
Erik Andersene49d5ec2000-02-08 19:58:47 +0000621#ifdef __50SERIES /* Prime/PRIMOS */
Eric Andersencc8ed391999-10-05 16:24:54 +0000622# define PATH_SEP '>'
623# define STDC_HEADERS
624# define NO_MEMORY_H
625# define NO_UTIME_H
626# define NO_UTIME
Erik Andersene49d5ec2000-02-08 19:58:47 +0000627# define NO_CHOWN
628# define NO_STDIN_FSTAT
629# define NO_SIZE_CHECK
Eric Andersencc8ed391999-10-05 16:24:54 +0000630# define NO_SYMLINK
631# define RECORD_IO 1
Erik Andersene49d5ec2000-02-08 19:58:47 +0000632# define casemap(c) tolow(c) /* Force file names to lower case */
Eric Andersencc8ed391999-10-05 16:24:54 +0000633# define put_char(c) put_byte((c) & 0x7F)
634# define get_char(c) ascii2pascii(get_byte())
Erik Andersene49d5ec2000-02-08 19:58:47 +0000635# define OS_CODE 0x0F /* temporary, subject to change */
Eric Andersencc8ed391999-10-05 16:24:54 +0000636# ifdef SIGTERM
Erik Andersene49d5ec2000-02-08 19:58:47 +0000637# undef SIGTERM /* We don't want a signal handler for SIGTERM */
Eric Andersencc8ed391999-10-05 16:24:54 +0000638# endif
639#endif
640
Erik Andersene49d5ec2000-02-08 19:58:47 +0000641#if defined(pyr) && !defined(NOMEMCPY) /* Pyramid */
642# define NOMEMCPY /* problem with overlapping copies */
Eric Andersencc8ed391999-10-05 16:24:54 +0000643#endif
644
645#ifdef TOPS20
646# define OS_CODE 0x0a
647#endif
648
649#ifndef unix
Erik Andersene49d5ec2000-02-08 19:58:47 +0000650# define NO_ST_INO /* don't rely on inode numbers */
Eric Andersencc8ed391999-10-05 16:24:54 +0000651#endif
652
653
654 /* Common defaults */
655
656#ifndef OS_CODE
Erik Andersene49d5ec2000-02-08 19:58:47 +0000657# define OS_CODE 0x03 /* assume Unix */
Eric Andersencc8ed391999-10-05 16:24:54 +0000658#endif
659
660#ifndef PATH_SEP
661# define PATH_SEP '/'
662#endif
663
664#ifndef casemap
665# define casemap(c) (c)
666#endif
667
668#ifndef OPTIONS_VAR
669# define OPTIONS_VAR "GZIP"
670#endif
671
672#ifndef Z_SUFFIX
673# define Z_SUFFIX ".gz"
674#endif
675
676#ifdef MAX_EXT_CHARS
677# define MAX_SUFFIX MAX_EXT_CHARS
678#else
679# define MAX_SUFFIX 30
680#endif
681
682#ifndef MAKE_LEGAL_NAME
683# ifdef NO_MULTIPLE_DOTS
684# define MAKE_LEGAL_NAME(name) make_simple_name(name)
685# else
686# define MAKE_LEGAL_NAME(name)
687# endif
688#endif
689
690#ifndef MIN_PART
691# define MIN_PART 3
692 /* keep at least MIN_PART chars between dots in a file name. */
693#endif
694
695#ifndef EXPAND
696# define EXPAND(argc,argv)
697#endif
698
699#ifndef RECORD_IO
700# define RECORD_IO 0
701#endif
702
703#ifndef SET_BINARY_MODE
704# define SET_BINARY_MODE(fd)
705#endif
706
707#ifndef OPEN
708# define OPEN(name, flags, mode) open(name, flags, mode)
709#endif
710
711#ifndef get_char
712# define get_char() get_byte()
713#endif
714
715#ifndef put_char
716# define put_char(c) put_byte(c)
717#endif
718/* bits.c -- output variable-length bit strings
719 * Copyright (C) 1992-1993 Jean-loup Gailly
720 * This is free software; you can redistribute it and/or modify it under the
721 * terms of the GNU General Public License, see the file COPYING.
722 */
723
724
725/*
726 * PURPOSE
727 *
728 * Output variable-length bit strings. Compression can be done
729 * to a file or to memory. (The latter is not supported in this version.)
730 *
731 * DISCUSSION
732 *
733 * The PKZIP "deflate" file format interprets compressed file data
734 * as a sequence of bits. Multi-bit strings in the file may cross
735 * byte boundaries without restriction.
736 *
737 * The first bit of each byte is the low-order bit.
738 *
739 * The routines in this file allow a variable-length bit value to
740 * be output right-to-left (useful for literal values). For
741 * left-to-right output (useful for code strings from the tree routines),
742 * the bits must have been reversed first with bi_reverse().
743 *
744 * For in-memory compression, the compressed bit stream goes directly
745 * into the requested output buffer. The input data is read in blocks
746 * by the mem_read() function. The buffer is limited to 64K on 16 bit
747 * machines.
748 *
749 * INTERFACE
750 *
751 * void bi_init (FILE *zipfile)
752 * Initialize the bit string routines.
753 *
754 * void send_bits (int value, int length)
755 * Write out a bit string, taking the source bits right to
756 * left.
757 *
758 * int bi_reverse (int value, int length)
759 * Reverse the bits of a bit string, taking the source bits left to
760 * right and emitting them right to left.
761 *
762 * void bi_windup (void)
763 * Write out any remaining bits in an incomplete byte.
764 *
765 * void copy_block(char *buf, unsigned len, int header)
766 * Copy a stored block to the zip file, storing first the length and
767 * its one's complement if requested.
768 *
769 */
770
771#ifdef DEBUG
772# include <stdio.h>
773#endif
774
Eric Andersencc8ed391999-10-05 16:24:54 +0000775/* ===========================================================================
776 * Local data used by the "bit string" routines.
777 */
778
Erik Andersene49d5ec2000-02-08 19:58:47 +0000779local file_t zfile; /* output gzip file */
Eric Andersencc8ed391999-10-05 16:24:54 +0000780
781local unsigned short bi_buf;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000782
Eric Andersencc8ed391999-10-05 16:24:54 +0000783/* Output buffer. bits are inserted starting at the bottom (least significant
784 * bits).
785 */
786
787#define Buf_size (8 * 2*sizeof(char))
788/* Number of bits used within bi_buf. (bi_buf might be implemented on
789 * more than 16 bits on some systems.)
790 */
791
792local int bi_valid;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000793
Eric Andersencc8ed391999-10-05 16:24:54 +0000794/* Number of valid bits in bi_buf. All bits above the last valid bit
795 * are always zero.
796 */
797
798int (*read_buf) OF((char *buf, unsigned size));
Erik Andersene49d5ec2000-02-08 19:58:47 +0000799
Eric Andersencc8ed391999-10-05 16:24:54 +0000800/* Current input function. Set to mem_read for in-memory compression */
801
802#ifdef DEBUG
Erik Andersene49d5ec2000-02-08 19:58:47 +0000803ulg bits_sent; /* bit length of the compressed data */
Eric Andersencc8ed391999-10-05 16:24:54 +0000804#endif
805
806/* ===========================================================================
807 * Initialize the bit string routines.
808 */
Erik Andersene49d5ec2000-02-08 19:58:47 +0000809void bi_init(zipfile)
810file_t zipfile; /* output zip file, NO_FILE for in-memory compression */
Eric Andersencc8ed391999-10-05 16:24:54 +0000811{
Erik Andersene49d5ec2000-02-08 19:58:47 +0000812 zfile = zipfile;
813 bi_buf = 0;
814 bi_valid = 0;
Eric Andersencc8ed391999-10-05 16:24:54 +0000815#ifdef DEBUG
Erik Andersene49d5ec2000-02-08 19:58:47 +0000816 bits_sent = 0L;
Eric Andersencc8ed391999-10-05 16:24:54 +0000817#endif
818
Erik Andersene49d5ec2000-02-08 19:58:47 +0000819 /* Set the defaults for file compression. They are set by memcompress
820 * for in-memory compression.
821 */
822 if (zfile != NO_FILE) {
823 read_buf = file_read;
824 }
Eric Andersencc8ed391999-10-05 16:24:54 +0000825}
826
827/* ===========================================================================
828 * Send a value on a given number of bits.
829 * IN assertion: length <= 16 and value fits in length bits.
830 */
831void send_bits(value, length)
Erik Andersene49d5ec2000-02-08 19:58:47 +0000832int value; /* value to send */
833int length; /* number of bits */
Eric Andersencc8ed391999-10-05 16:24:54 +0000834{
835#ifdef DEBUG
Erik Andersene49d5ec2000-02-08 19:58:47 +0000836 Tracev((stderr, " l %2d v %4x ", length, value));
837 Assert(length > 0 && length <= 15, "invalid length");
838 bits_sent += (ulg) length;
Eric Andersencc8ed391999-10-05 16:24:54 +0000839#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +0000840 /* If not enough room in bi_buf, use (valid) bits from bi_buf and
841 * (16 - bi_valid) bits from value, leaving (width - (16-bi_valid))
842 * unused bits in value.
843 */
844 if (bi_valid > (int) Buf_size - length) {
845 bi_buf |= (value << bi_valid);
846 put_short(bi_buf);
847 bi_buf = (ush) value >> (Buf_size - bi_valid);
848 bi_valid += length - Buf_size;
849 } else {
850 bi_buf |= value << bi_valid;
851 bi_valid += length;
852 }
Eric Andersencc8ed391999-10-05 16:24:54 +0000853}
854
855/* ===========================================================================
856 * Reverse the first len bits of a code, using straightforward code (a faster
857 * method would use a table)
858 * IN assertion: 1 <= len <= 15
859 */
860unsigned bi_reverse(code, len)
Erik Andersene49d5ec2000-02-08 19:58:47 +0000861unsigned code; /* the value to invert */
862int len; /* its bit length */
Eric Andersencc8ed391999-10-05 16:24:54 +0000863{
Erik Andersene49d5ec2000-02-08 19:58:47 +0000864 register unsigned res = 0;
865
866 do {
867 res |= code & 1;
868 code >>= 1, res <<= 1;
869 } while (--len > 0);
870 return res >> 1;
Eric Andersencc8ed391999-10-05 16:24:54 +0000871}
872
873/* ===========================================================================
874 * Write out any remaining bits in an incomplete byte.
875 */
876void bi_windup()
877{
Erik Andersene49d5ec2000-02-08 19:58:47 +0000878 if (bi_valid > 8) {
879 put_short(bi_buf);
880 } else if (bi_valid > 0) {
881 put_byte(bi_buf);
882 }
883 bi_buf = 0;
884 bi_valid = 0;
Eric Andersencc8ed391999-10-05 16:24:54 +0000885#ifdef DEBUG
Erik Andersene49d5ec2000-02-08 19:58:47 +0000886 bits_sent = (bits_sent + 7) & ~7;
Eric Andersencc8ed391999-10-05 16:24:54 +0000887#endif
888}
889
890/* ===========================================================================
891 * Copy a stored block to the zip file, storing first the length and its
892 * one's complement if requested.
893 */
894void copy_block(buf, len, header)
Erik Andersene49d5ec2000-02-08 19:58:47 +0000895char *buf; /* the input data */
896unsigned len; /* its length */
897int header; /* true if block header must be written */
Eric Andersencc8ed391999-10-05 16:24:54 +0000898{
Erik Andersene49d5ec2000-02-08 19:58:47 +0000899 bi_windup(); /* align on byte boundary */
Eric Andersencc8ed391999-10-05 16:24:54 +0000900
Erik Andersene49d5ec2000-02-08 19:58:47 +0000901 if (header) {
902 put_short((ush) len);
903 put_short((ush) ~ len);
Eric Andersencc8ed391999-10-05 16:24:54 +0000904#ifdef DEBUG
Erik Andersene49d5ec2000-02-08 19:58:47 +0000905 bits_sent += 2 * 16;
Eric Andersencc8ed391999-10-05 16:24:54 +0000906#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +0000907 }
Eric Andersencc8ed391999-10-05 16:24:54 +0000908#ifdef DEBUG
Erik Andersene49d5ec2000-02-08 19:58:47 +0000909 bits_sent += (ulg) len << 3;
Eric Andersencc8ed391999-10-05 16:24:54 +0000910#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +0000911 while (len--) {
Eric Andersencc8ed391999-10-05 16:24:54 +0000912#ifdef CRYPT
Erik Andersene49d5ec2000-02-08 19:58:47 +0000913 int t;
914
915 if (key)
916 zencode(*buf, t);
Eric Andersencc8ed391999-10-05 16:24:54 +0000917#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +0000918 put_byte(*buf++);
919 }
Eric Andersencc8ed391999-10-05 16:24:54 +0000920}
Erik Andersene49d5ec2000-02-08 19:58:47 +0000921
Eric Andersencc8ed391999-10-05 16:24:54 +0000922/* deflate.c -- compress data using the deflation algorithm
923 * Copyright (C) 1992-1993 Jean-loup Gailly
924 * This is free software; you can redistribute it and/or modify it under the
925 * terms of the GNU General Public License, see the file COPYING.
926 */
927
928/*
929 * PURPOSE
930 *
931 * Identify new text as repetitions of old text within a fixed-
932 * length sliding window trailing behind the new text.
933 *
934 * DISCUSSION
935 *
936 * The "deflation" process depends on being able to identify portions
937 * of the input text which are identical to earlier input (within a
938 * sliding window trailing behind the input currently being processed).
939 *
940 * The most straightforward technique turns out to be the fastest for
941 * most input files: try all possible matches and select the longest.
942 * The key feature of this algorithm is that insertions into the string
943 * dictionary are very simple and thus fast, and deletions are avoided
944 * completely. Insertions are performed at each input character, whereas
945 * string matches are performed only when the previous match ends. So it
946 * is preferable to spend more time in matches to allow very fast string
947 * insertions and avoid deletions. The matching algorithm for small
948 * strings is inspired from that of Rabin & Karp. A brute force approach
949 * is used to find longer strings when a small match has been found.
950 * A similar algorithm is used in comic (by Jan-Mark Wams) and freeze
951 * (by Leonid Broukhis).
952 * A previous version of this file used a more sophisticated algorithm
953 * (by Fiala and Greene) which is guaranteed to run in linear amortized
954 * time, but has a larger average cost, uses more memory and is patented.
955 * However the F&G algorithm may be faster for some highly redundant
956 * files if the parameter max_chain_length (described below) is too large.
957 *
958 * ACKNOWLEDGEMENTS
959 *
960 * The idea of lazy evaluation of matches is due to Jan-Mark Wams, and
961 * I found it in 'freeze' written by Leonid Broukhis.
962 * Thanks to many info-zippers for bug reports and testing.
963 *
964 * REFERENCES
965 *
966 * APPNOTE.TXT documentation file in PKZIP 1.93a distribution.
967 *
968 * A description of the Rabin and Karp algorithm is given in the book
969 * "Algorithms" by R. Sedgewick, Addison-Wesley, p252.
970 *
971 * Fiala,E.R., and Greene,D.H.
972 * Data Compression with Finite Windows, Comm.ACM, 32,4 (1989) 490-595
973 *
974 * INTERFACE
975 *
976 * void lm_init (int pack_level, ush *flags)
977 * Initialize the "longest match" routines for a new file
978 *
979 * ulg deflate (void)
980 * Processes a new input file and return its compressed length. Sets
981 * the compressed length, crc, deflate flags and internal file
982 * attributes.
983 */
984
985#include <stdio.h>
986
Eric Andersencc8ed391999-10-05 16:24:54 +0000987/* ===========================================================================
988 * Configuration parameters
989 */
990
991/* Compile with MEDIUM_MEM to reduce the memory requirements or
992 * with SMALL_MEM to use as little memory as possible. Use BIG_MEM if the
993 * entire input file can be held in memory (not possible on 16 bit systems).
994 * Warning: defining these symbols affects HASH_BITS (see below) and thus
995 * affects the compression ratio. The compressed output
996 * is still correct, and might even be smaller in some cases.
997 */
998
999#ifdef SMALL_MEM
Erik Andersene49d5ec2000-02-08 19:58:47 +00001000# define HASH_BITS 13 /* Number of bits used to hash strings */
Eric Andersencc8ed391999-10-05 16:24:54 +00001001#endif
1002#ifdef MEDIUM_MEM
1003# define HASH_BITS 14
1004#endif
1005#ifndef HASH_BITS
1006# define HASH_BITS 15
1007 /* For portability to 16 bit machines, do not use values above 15. */
1008#endif
1009
1010/* To save space (see unlzw.c), we overlay prev+head with tab_prefix and
1011 * window with tab_suffix. Check that we can do this:
1012 */
1013#if (WSIZE<<1) > (1<<BITS)
Erik Andersene49d5ec2000-02-08 19:58:47 +00001014error:cannot overlay window with tab_suffix and prev with tab_prefix0
Eric Andersencc8ed391999-10-05 16:24:54 +00001015#endif
1016#if HASH_BITS > BITS-1
Erik Andersene49d5ec2000-02-08 19:58:47 +00001017error:cannot overlay head with tab_prefix1
Eric Andersencc8ed391999-10-05 16:24:54 +00001018#endif
Eric Andersencc8ed391999-10-05 16:24:54 +00001019#define HASH_SIZE (unsigned)(1<<HASH_BITS)
1020#define HASH_MASK (HASH_SIZE-1)
1021#define WMASK (WSIZE-1)
1022/* HASH_SIZE and WSIZE must be powers of two */
Eric Andersencc8ed391999-10-05 16:24:54 +00001023#define NIL 0
1024/* Tail of hash chains */
Eric Andersencc8ed391999-10-05 16:24:54 +00001025#define FAST 4
1026#define SLOW 2
1027/* speed options for the general purpose bit flag */
Eric Andersencc8ed391999-10-05 16:24:54 +00001028#ifndef TOO_FAR
1029# define TOO_FAR 4096
1030#endif
1031/* Matches of length 3 are discarded if their distance exceeds TOO_FAR */
Eric Andersencc8ed391999-10-05 16:24:54 +00001032/* ===========================================================================
1033 * Local data used by the "longest match" routines.
1034 */
Eric Andersencc8ed391999-10-05 16:24:54 +00001035typedef ush Pos;
1036typedef unsigned IPos;
Erik Andersene49d5ec2000-02-08 19:58:47 +00001037
Eric Andersencc8ed391999-10-05 16:24:54 +00001038/* A Pos is an index in the character window. We use short instead of int to
1039 * save space in the various tables. IPos is used only for parameter passing.
1040 */
1041
1042/* DECLARE(uch, window, 2L*WSIZE); */
1043/* Sliding window. Input bytes are read into the second half of the window,
1044 * and move to the first half later to keep a dictionary of at least WSIZE
1045 * bytes. With this organization, matches are limited to a distance of
1046 * WSIZE-MAX_MATCH bytes, but this ensures that IO is always
1047 * performed with a length multiple of the block size. Also, it limits
1048 * the window size to 64K, which is quite useful on MSDOS.
1049 * To do: limit the window size to WSIZE+BSZ if SMALL_MEM (the code would
1050 * be less efficient).
1051 */
1052
1053/* DECLARE(Pos, prev, WSIZE); */
1054/* Link to older string with same hash index. To limit the size of this
1055 * array to 64K, this link is maintained only for the last 32K strings.
1056 * An index in this array is thus a window index modulo 32K.
1057 */
1058
1059/* DECLARE(Pos, head, 1<<HASH_BITS); */
1060/* Heads of the hash chains or NIL. */
1061
Erik Andersene49d5ec2000-02-08 19:58:47 +00001062ulg window_size = (ulg) 2 * WSIZE;
1063
Eric Andersencc8ed391999-10-05 16:24:54 +00001064/* window size, 2*WSIZE except for MMAP or BIG_MEM, where it is the
1065 * input file length plus MIN_LOOKAHEAD.
1066 */
1067
1068long block_start;
Erik Andersene49d5ec2000-02-08 19:58:47 +00001069
Eric Andersencc8ed391999-10-05 16:24:54 +00001070/* window position at the beginning of the current output block. Gets
1071 * negative when the window is moved backwards.
1072 */
1073
Erik Andersene49d5ec2000-02-08 19:58:47 +00001074local unsigned ins_h; /* hash index of string to be inserted */
Eric Andersencc8ed391999-10-05 16:24:54 +00001075
1076#define H_SHIFT ((HASH_BITS+MIN_MATCH-1)/MIN_MATCH)
1077/* Number of bits by which ins_h and del_h must be shifted at each
1078 * input step. It must be such that after MIN_MATCH steps, the oldest
1079 * byte no longer takes part in the hash key, that is:
1080 * H_SHIFT * MIN_MATCH >= HASH_BITS
1081 */
1082
1083unsigned int near prev_length;
Erik Andersene49d5ec2000-02-08 19:58:47 +00001084
Eric Andersencc8ed391999-10-05 16:24:54 +00001085/* Length of the best match at previous step. Matches not greater than this
1086 * are discarded. This is used in the lazy match evaluation.
1087 */
1088
Erik Andersene49d5ec2000-02-08 19:58:47 +00001089unsigned near strstart; /* start of string to insert */
1090unsigned near match_start; /* start of matching string */
1091local int eofile; /* flag set at end of input file */
1092local unsigned lookahead; /* number of valid bytes ahead in window */
Eric Andersencc8ed391999-10-05 16:24:54 +00001093
1094unsigned near max_chain_length;
Erik Andersene49d5ec2000-02-08 19:58:47 +00001095
Eric Andersencc8ed391999-10-05 16:24:54 +00001096/* To speed up deflation, hash chains are never searched beyond this length.
1097 * A higher limit improves compression ratio but degrades the speed.
1098 */
1099
1100local unsigned int max_lazy_match;
Erik Andersene49d5ec2000-02-08 19:58:47 +00001101
Eric Andersencc8ed391999-10-05 16:24:54 +00001102/* Attempt to find a better match only when the current match is strictly
1103 * smaller than this value. This mechanism is used only for compression
1104 * levels >= 4.
1105 */
1106#define max_insert_length max_lazy_match
1107/* Insert new strings in the hash table only if the match length
1108 * is not greater than this length. This saves time but degrades compression.
1109 * max_insert_length is used only for compression levels <= 3.
1110 */
1111
1112unsigned near good_match;
Erik Andersene49d5ec2000-02-08 19:58:47 +00001113
Eric Andersencc8ed391999-10-05 16:24:54 +00001114/* Use a faster search when the previous match is longer than this */
1115
1116
1117/* Values for max_lazy_match, good_match and max_chain_length, depending on
1118 * the desired pack level (0..9). The values given below have been tuned to
1119 * exclude worst case performance for pathological files. Better values may be
1120 * found for specific files.
1121 */
1122
1123typedef struct config {
Erik Andersene49d5ec2000-02-08 19:58:47 +00001124 ush good_length; /* reduce lazy search above this match length */
1125 ush max_lazy; /* do not perform lazy search above this match length */
1126 ush nice_length; /* quit search above this match length */
1127 ush max_chain;
Eric Andersencc8ed391999-10-05 16:24:54 +00001128} config;
1129
1130#ifdef FULL_SEARCH
1131# define nice_match MAX_MATCH
1132#else
Erik Andersene49d5ec2000-02-08 19:58:47 +00001133int near nice_match; /* Stop searching when current match exceeds this */
Eric Andersencc8ed391999-10-05 16:24:54 +00001134#endif
1135
Erik Andersene49d5ec2000-02-08 19:58:47 +00001136local config configuration_table =
1137 /* 9 */ { 32, 258, 258, 4096 };
1138 /* maximum compression */
Eric Andersencc8ed391999-10-05 16:24:54 +00001139
1140/* Note: the deflate() code requires max_lazy >= MIN_MATCH and max_chain >= 4
1141 * For deflate_fast() (levels <= 3) good is ignored and lazy has a different
1142 * meaning.
1143 */
1144
1145#define EQUAL 0
1146/* result of memcmp for equal strings */
1147
1148/* ===========================================================================
1149 * Prototypes for local functions.
1150 */
Erik Andersene49d5ec2000-02-08 19:58:47 +00001151local void fill_window OF((void));
Eric Andersencc8ed391999-10-05 16:24:54 +00001152
Erik Andersene49d5ec2000-02-08 19:58:47 +00001153int longest_match OF((IPos cur_match));
1154
Eric Andersencc8ed391999-10-05 16:24:54 +00001155#ifdef ASMV
Erik Andersene49d5ec2000-02-08 19:58:47 +00001156void match_init OF((void)); /* asm code initialization */
Eric Andersencc8ed391999-10-05 16:24:54 +00001157#endif
1158
1159#ifdef DEBUG
Erik Andersene49d5ec2000-02-08 19:58:47 +00001160local void check_match OF((IPos start, IPos match, int length));
Eric Andersencc8ed391999-10-05 16:24:54 +00001161#endif
1162
1163/* ===========================================================================
1164 * Update a hash value with the given input byte
1165 * IN assertion: all calls to to UPDATE_HASH are made with consecutive
1166 * input characters, so that a running hash key can be computed from the
1167 * previous key instead of complete recalculation each time.
1168 */
1169#define UPDATE_HASH(h,c) (h = (((h)<<H_SHIFT) ^ (c)) & HASH_MASK)
1170
1171/* ===========================================================================
1172 * Insert string s in the dictionary and set match_head to the previous head
1173 * of the hash chain (the most recent string with same hash key). Return
1174 * the previous length of the hash chain.
1175 * IN assertion: all calls to to INSERT_STRING are made with consecutive
1176 * input characters and the first MIN_MATCH bytes of s are valid
1177 * (except for the last MIN_MATCH-1 bytes of the input file).
1178 */
1179#define INSERT_STRING(s, match_head) \
1180 (UPDATE_HASH(ins_h, window[(s) + MIN_MATCH-1]), \
1181 prev[(s) & WMASK] = match_head = head[ins_h], \
1182 head[ins_h] = (s))
1183
1184/* ===========================================================================
1185 * Initialize the "longest match" routines for a new file
1186 */
Erik Andersene49d5ec2000-02-08 19:58:47 +00001187void lm_init(flags)
1188ush *flags; /* general purpose bit flag */
Eric Andersencc8ed391999-10-05 16:24:54 +00001189{
Erik Andersene49d5ec2000-02-08 19:58:47 +00001190 register unsigned j;
Eric Andersencc8ed391999-10-05 16:24:54 +00001191
Erik Andersene49d5ec2000-02-08 19:58:47 +00001192 /* Initialize the hash table. */
Eric Andersencc8ed391999-10-05 16:24:54 +00001193#if defined(MAXSEG_64K) && HASH_BITS == 15
Erik Andersene49d5ec2000-02-08 19:58:47 +00001194 for (j = 0; j < HASH_SIZE; j++)
1195 head[j] = NIL;
Eric Andersencc8ed391999-10-05 16:24:54 +00001196#else
Erik Andersene49d5ec2000-02-08 19:58:47 +00001197 memzero((char *) head, HASH_SIZE * sizeof(*head));
Eric Andersencc8ed391999-10-05 16:24:54 +00001198#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +00001199 /* prev will be initialized on the fly */
Eric Andersencc8ed391999-10-05 16:24:54 +00001200
Erik Andersene49d5ec2000-02-08 19:58:47 +00001201 /* Set the default configuration parameters:
1202 */
1203 max_lazy_match = configuration_table.max_lazy;
1204 good_match = configuration_table.good_length;
Eric Andersencc8ed391999-10-05 16:24:54 +00001205#ifndef FULL_SEARCH
Erik Andersene49d5ec2000-02-08 19:58:47 +00001206 nice_match = configuration_table.nice_length;
Eric Andersencc8ed391999-10-05 16:24:54 +00001207#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +00001208 max_chain_length = configuration_table.max_chain;
1209 *flags |= SLOW;
1210 /* ??? reduce max_chain_length for binary files */
Eric Andersencc8ed391999-10-05 16:24:54 +00001211
Erik Andersene49d5ec2000-02-08 19:58:47 +00001212 strstart = 0;
1213 block_start = 0L;
Eric Andersencc8ed391999-10-05 16:24:54 +00001214#ifdef ASMV
Erik Andersene49d5ec2000-02-08 19:58:47 +00001215 match_init(); /* initialize the asm code */
Eric Andersencc8ed391999-10-05 16:24:54 +00001216#endif
1217
Erik Andersene49d5ec2000-02-08 19:58:47 +00001218 lookahead = read_buf((char *) window,
1219 sizeof(int) <= 2 ? (unsigned) WSIZE : 2 * WSIZE);
Eric Andersencc8ed391999-10-05 16:24:54 +00001220
Erik Andersene49d5ec2000-02-08 19:58:47 +00001221 if (lookahead == 0 || lookahead == (unsigned) EOF) {
1222 eofile = 1, lookahead = 0;
1223 return;
1224 }
1225 eofile = 0;
1226 /* Make sure that we always have enough lookahead. This is important
1227 * if input comes from a device such as a tty.
1228 */
1229 while (lookahead < MIN_LOOKAHEAD && !eofile)
1230 fill_window();
Eric Andersencc8ed391999-10-05 16:24:54 +00001231
Erik Andersene49d5ec2000-02-08 19:58:47 +00001232 ins_h = 0;
1233 for (j = 0; j < MIN_MATCH - 1; j++)
1234 UPDATE_HASH(ins_h, window[j]);
1235 /* If lookahead < MIN_MATCH, ins_h is garbage, but this is
1236 * not important since only literal bytes will be emitted.
1237 */
Eric Andersencc8ed391999-10-05 16:24:54 +00001238}
1239
1240/* ===========================================================================
1241 * Set match_start to the longest match starting at the given string and
1242 * return its length. Matches shorter or equal to prev_length are discarded,
1243 * in which case the result is equal to prev_length and match_start is
1244 * garbage.
1245 * IN assertions: cur_match is the head of the hash chain for the current
1246 * string (strstart) and its distance is <= MAX_DIST, and prev_length >= 1
1247 */
1248#ifndef ASMV
1249/* For MSDOS, OS/2 and 386 Unix, an optimized version is in match.asm or
1250 * match.s. The code is functionally equivalent, so you can use the C version
1251 * if desired.
1252 */
1253int longest_match(cur_match)
Erik Andersene49d5ec2000-02-08 19:58:47 +00001254IPos cur_match; /* current match */
Eric Andersencc8ed391999-10-05 16:24:54 +00001255{
Erik Andersene49d5ec2000-02-08 19:58:47 +00001256 unsigned chain_length = max_chain_length; /* max hash chain length */
1257 register uch *scan = window + strstart; /* current string */
1258 register uch *match; /* matched string */
1259 register int len; /* length of current match */
1260 int best_len = prev_length; /* best match length so far */
1261 IPos limit =
1262
1263 strstart > (IPos) MAX_DIST ? strstart - (IPos) MAX_DIST : NIL;
1264 /* Stop when cur_match becomes <= limit. To simplify the code,
1265 * we prevent matches with the string of window index 0.
1266 */
Eric Andersencc8ed391999-10-05 16:24:54 +00001267
1268/* The code is optimized for HASH_BITS >= 8 and MAX_MATCH-2 multiple of 16.
1269 * It is easy to get rid of this optimization if necessary.
1270 */
1271#if HASH_BITS < 8 || MAX_MATCH != 258
Erik Andersene49d5ec2000-02-08 19:58:47 +00001272 error:Code too clever
Eric Andersencc8ed391999-10-05 16:24:54 +00001273#endif
Eric Andersencc8ed391999-10-05 16:24:54 +00001274#ifdef UNALIGNED_OK
Erik Andersene49d5ec2000-02-08 19:58:47 +00001275 /* Compare two bytes at a time. Note: this is not always beneficial.
1276 * Try with and without -DUNALIGNED_OK to check.
1277 */
1278 register uch *strend = window + strstart + MAX_MATCH - 1;
1279 register ush scan_start = *(ush *) scan;
1280 register ush scan_end = *(ush *) (scan + best_len - 1);
Eric Andersencc8ed391999-10-05 16:24:54 +00001281#else
Erik Andersene49d5ec2000-02-08 19:58:47 +00001282 register uch *strend = window + strstart + MAX_MATCH;
1283 register uch scan_end1 = scan[best_len - 1];
1284 register uch scan_end = scan[best_len];
Eric Andersencc8ed391999-10-05 16:24:54 +00001285#endif
1286
Erik Andersene49d5ec2000-02-08 19:58:47 +00001287 /* Do not waste too much time if we already have a good match: */
1288 if (prev_length >= good_match) {
1289 chain_length >>= 2;
1290 }
1291 Assert(strstart <= window_size - MIN_LOOKAHEAD,
1292 "insufficient lookahead");
Eric Andersencc8ed391999-10-05 16:24:54 +00001293
Erik Andersene49d5ec2000-02-08 19:58:47 +00001294 do {
1295 Assert(cur_match < strstart, "no future");
1296 match = window + cur_match;
Eric Andersencc8ed391999-10-05 16:24:54 +00001297
Erik Andersene49d5ec2000-02-08 19:58:47 +00001298 /* Skip to next match if the match length cannot increase
1299 * or if the match length is less than 2:
1300 */
Eric Andersencc8ed391999-10-05 16:24:54 +00001301#if (defined(UNALIGNED_OK) && MAX_MATCH == 258)
Erik Andersene49d5ec2000-02-08 19:58:47 +00001302 /* This code assumes sizeof(unsigned short) == 2. Do not use
1303 * UNALIGNED_OK if your compiler uses a different size.
1304 */
1305 if (*(ush *) (match + best_len - 1) != scan_end ||
1306 *(ush *) match != scan_start)
1307 continue;
Eric Andersencc8ed391999-10-05 16:24:54 +00001308
Erik Andersene49d5ec2000-02-08 19:58:47 +00001309 /* It is not necessary to compare scan[2] and match[2] since they are
1310 * always equal when the other bytes match, given that the hash keys
1311 * are equal and that HASH_BITS >= 8. Compare 2 bytes at a time at
1312 * strstart+3, +5, ... up to strstart+257. We check for insufficient
1313 * lookahead only every 4th comparison; the 128th check will be made
1314 * at strstart+257. If MAX_MATCH-2 is not a multiple of 8, it is
1315 * necessary to put more guard bytes at the end of the window, or
1316 * to check more often for insufficient lookahead.
1317 */
1318 scan++, match++;
1319 do {
1320 } while (*(ush *) (scan += 2) == *(ush *) (match += 2) &&
1321 *(ush *) (scan += 2) == *(ush *) (match += 2) &&
1322 *(ush *) (scan += 2) == *(ush *) (match += 2) &&
1323 *(ush *) (scan += 2) == *(ush *) (match += 2) &&
1324 scan < strend);
1325 /* The funny "do {}" generates better code on most compilers */
Eric Andersencc8ed391999-10-05 16:24:54 +00001326
Erik Andersene49d5ec2000-02-08 19:58:47 +00001327 /* Here, scan <= window+strstart+257 */
1328 Assert(scan <= window + (unsigned) (window_size - 1), "wild scan");
1329 if (*scan == *match)
1330 scan++;
Eric Andersencc8ed391999-10-05 16:24:54 +00001331
Erik Andersene49d5ec2000-02-08 19:58:47 +00001332 len = (MAX_MATCH - 1) - (int) (strend - scan);
1333 scan = strend - (MAX_MATCH - 1);
Eric Andersencc8ed391999-10-05 16:24:54 +00001334
Erik Andersene49d5ec2000-02-08 19:58:47 +00001335#else /* UNALIGNED_OK */
Eric Andersencc8ed391999-10-05 16:24:54 +00001336
Erik Andersene49d5ec2000-02-08 19:58:47 +00001337 if (match[best_len] != scan_end ||
1338 match[best_len - 1] != scan_end1 ||
1339 *match != *scan || *++match != scan[1])
1340 continue;
Eric Andersencc8ed391999-10-05 16:24:54 +00001341
Erik Andersene49d5ec2000-02-08 19:58:47 +00001342 /* The check at best_len-1 can be removed because it will be made
1343 * again later. (This heuristic is not always a win.)
1344 * It is not necessary to compare scan[2] and match[2] since they
1345 * are always equal when the other bytes match, given that
1346 * the hash keys are equal and that HASH_BITS >= 8.
1347 */
1348 scan += 2, match++;
Eric Andersencc8ed391999-10-05 16:24:54 +00001349
Erik Andersene49d5ec2000-02-08 19:58:47 +00001350 /* We check for insufficient lookahead only every 8th comparison;
1351 * the 256th check will be made at strstart+258.
1352 */
1353 do {
1354 } while (*++scan == *++match && *++scan == *++match &&
1355 *++scan == *++match && *++scan == *++match &&
1356 *++scan == *++match && *++scan == *++match &&
1357 *++scan == *++match && *++scan == *++match &&
1358 scan < strend);
Eric Andersencc8ed391999-10-05 16:24:54 +00001359
Erik Andersene49d5ec2000-02-08 19:58:47 +00001360 len = MAX_MATCH - (int) (strend - scan);
1361 scan = strend - MAX_MATCH;
Eric Andersencc8ed391999-10-05 16:24:54 +00001362
Erik Andersene49d5ec2000-02-08 19:58:47 +00001363#endif /* UNALIGNED_OK */
Eric Andersencc8ed391999-10-05 16:24:54 +00001364
Erik Andersene49d5ec2000-02-08 19:58:47 +00001365 if (len > best_len) {
1366 match_start = cur_match;
1367 best_len = len;
1368 if (len >= nice_match)
1369 break;
Eric Andersencc8ed391999-10-05 16:24:54 +00001370#ifdef UNALIGNED_OK
Erik Andersene49d5ec2000-02-08 19:58:47 +00001371 scan_end = *(ush *) (scan + best_len - 1);
Eric Andersencc8ed391999-10-05 16:24:54 +00001372#else
Erik Andersene49d5ec2000-02-08 19:58:47 +00001373 scan_end1 = scan[best_len - 1];
1374 scan_end = scan[best_len];
Eric Andersencc8ed391999-10-05 16:24:54 +00001375#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +00001376 }
1377 } while ((cur_match = prev[cur_match & WMASK]) > limit
1378 && --chain_length != 0);
Eric Andersencc8ed391999-10-05 16:24:54 +00001379
Erik Andersene49d5ec2000-02-08 19:58:47 +00001380 return best_len;
Eric Andersencc8ed391999-10-05 16:24:54 +00001381}
Erik Andersene49d5ec2000-02-08 19:58:47 +00001382#endif /* ASMV */
Eric Andersencc8ed391999-10-05 16:24:54 +00001383
1384#ifdef DEBUG
1385/* ===========================================================================
1386 * Check that the match at match_start is indeed a match.
1387 */
1388local void check_match(start, match, length)
Erik Andersene49d5ec2000-02-08 19:58:47 +00001389IPos start, match;
1390int length;
Eric Andersencc8ed391999-10-05 16:24:54 +00001391{
Erik Andersene49d5ec2000-02-08 19:58:47 +00001392 /* check that the match is indeed a match */
1393 if (memcmp((char *) window + match,
1394 (char *) window + start, length) != EQUAL) {
1395 fprintf(stderr,
1396 " start %d, match %d, length %d\n", start, match, length);
Erik Andersen9ffdaa62000-02-11 21:55:04 +00001397 errorMsg("invalid match");
Erik Andersene49d5ec2000-02-08 19:58:47 +00001398 }
1399 if (verbose > 1) {
1400 fprintf(stderr, "\\[%d,%d]", start - match, length);
1401 do {
1402 putc(window[start++], stderr);
1403 } while (--length != 0);
1404 }
Eric Andersencc8ed391999-10-05 16:24:54 +00001405}
1406#else
1407# define check_match(start, match, length)
1408#endif
1409
1410/* ===========================================================================
1411 * Fill the window when the lookahead becomes insufficient.
1412 * Updates strstart and lookahead, and sets eofile if end of input file.
1413 * IN assertion: lookahead < MIN_LOOKAHEAD && strstart + lookahead > 0
1414 * OUT assertions: at least one byte has been read, or eofile is set;
1415 * file reads are performed for at least two bytes (required for the
1416 * translate_eol option).
1417 */
1418local void fill_window()
1419{
Erik Andersene49d5ec2000-02-08 19:58:47 +00001420 register unsigned n, m;
1421 unsigned more =
Eric Andersencc8ed391999-10-05 16:24:54 +00001422
Erik Andersene49d5ec2000-02-08 19:58:47 +00001423 (unsigned) (window_size - (ulg) lookahead - (ulg) strstart);
1424 /* Amount of free space at the end of the window. */
Eric Andersencc8ed391999-10-05 16:24:54 +00001425
Erik Andersene49d5ec2000-02-08 19:58:47 +00001426 /* If the window is almost full and there is insufficient lookahead,
1427 * move the upper half to the lower one to make room in the upper half.
1428 */
1429 if (more == (unsigned) EOF) {
1430 /* Very unlikely, but possible on 16 bit machine if strstart == 0
1431 * and lookahead == 1 (input done one byte at time)
1432 */
1433 more--;
1434 } else if (strstart >= WSIZE + MAX_DIST) {
1435 /* By the IN assertion, the window is not empty so we can't confuse
1436 * more == 0 with more == 64K on a 16 bit machine.
1437 */
1438 Assert(window_size == (ulg) 2 * WSIZE, "no sliding with BIG_MEM");
Eric Andersencc8ed391999-10-05 16:24:54 +00001439
Erik Andersene49d5ec2000-02-08 19:58:47 +00001440 memcpy((char *) window, (char *) window + WSIZE, (unsigned) WSIZE);
1441 match_start -= WSIZE;
1442 strstart -= WSIZE; /* we now have strstart >= MAX_DIST: */
Eric Andersencc8ed391999-10-05 16:24:54 +00001443
Erik Andersene49d5ec2000-02-08 19:58:47 +00001444 block_start -= (long) WSIZE;
1445
1446 for (n = 0; n < HASH_SIZE; n++) {
1447 m = head[n];
1448 head[n] = (Pos) (m >= WSIZE ? m - WSIZE : NIL);
1449 }
1450 for (n = 0; n < WSIZE; n++) {
1451 m = prev[n];
1452 prev[n] = (Pos) (m >= WSIZE ? m - WSIZE : NIL);
1453 /* If n is not on any hash chain, prev[n] is garbage but
1454 * its value will never be used.
1455 */
1456 }
1457 more += WSIZE;
1458 }
1459 /* At this point, more >= 2 */
1460 if (!eofile) {
1461 n = read_buf((char *) window + strstart + lookahead, more);
1462 if (n == 0 || n == (unsigned) EOF) {
1463 eofile = 1;
1464 } else {
1465 lookahead += n;
1466 }
1467 }
Eric Andersencc8ed391999-10-05 16:24:54 +00001468}
1469
1470/* ===========================================================================
1471 * Flush the current block, with given end-of-file flag.
1472 * IN assertion: strstart is set to the end of the current match.
1473 */
1474#define FLUSH_BLOCK(eof) \
1475 flush_block(block_start >= 0L ? (char*)&window[(unsigned)block_start] : \
1476 (char*)NULL, (long)strstart - block_start, (eof))
1477
1478/* ===========================================================================
1479 * Same as above, but achieves better compression. We use a lazy
1480 * evaluation for matches: a match is finally adopted only if there is
1481 * no better match at the next window position.
1482 */
1483ulg deflate()
1484{
Erik Andersene49d5ec2000-02-08 19:58:47 +00001485 IPos hash_head; /* head of hash chain */
1486 IPos prev_match; /* previous match */
1487 int flush; /* set if current block must be flushed */
1488 int match_available = 0; /* set if previous match exists */
1489 register unsigned match_length = MIN_MATCH - 1; /* length of best match */
1490
Eric Andersencc8ed391999-10-05 16:24:54 +00001491#ifdef DEBUG
Erik Andersene49d5ec2000-02-08 19:58:47 +00001492 extern long isize; /* byte length of input file, for debug only */
Eric Andersencc8ed391999-10-05 16:24:54 +00001493#endif
1494
Erik Andersene49d5ec2000-02-08 19:58:47 +00001495 /* Process the input block. */
1496 while (lookahead != 0) {
1497 /* Insert the string window[strstart .. strstart+2] in the
1498 * dictionary, and set hash_head to the head of the hash chain:
1499 */
1500 INSERT_STRING(strstart, hash_head);
Eric Andersencc8ed391999-10-05 16:24:54 +00001501
Erik Andersene49d5ec2000-02-08 19:58:47 +00001502 /* Find the longest match, discarding those <= prev_length.
1503 */
1504 prev_length = match_length, prev_match = match_start;
1505 match_length = MIN_MATCH - 1;
Eric Andersencc8ed391999-10-05 16:24:54 +00001506
Erik Andersene49d5ec2000-02-08 19:58:47 +00001507 if (hash_head != NIL && prev_length < max_lazy_match &&
1508 strstart - hash_head <= MAX_DIST) {
1509 /* To simplify the code, we prevent matches with the string
1510 * of window index 0 (in particular we have to avoid a match
1511 * of the string with itself at the start of the input file).
1512 */
1513 match_length = longest_match(hash_head);
1514 /* longest_match() sets match_start */
1515 if (match_length > lookahead)
1516 match_length = lookahead;
Eric Andersencc8ed391999-10-05 16:24:54 +00001517
Erik Andersene49d5ec2000-02-08 19:58:47 +00001518 /* Ignore a length 3 match if it is too distant: */
1519 if (match_length == MIN_MATCH
1520 && strstart - match_start > TOO_FAR) {
1521 /* If prev_match is also MIN_MATCH, match_start is garbage
1522 * but we will ignore the current match anyway.
1523 */
1524 match_length--;
1525 }
1526 }
1527 /* If there was a match at the previous step and the current
1528 * match is not better, output the previous match:
1529 */
1530 if (prev_length >= MIN_MATCH && match_length <= prev_length) {
Eric Andersencc8ed391999-10-05 16:24:54 +00001531
Erik Andersene49d5ec2000-02-08 19:58:47 +00001532 check_match(strstart - 1, prev_match, prev_length);
Eric Andersencc8ed391999-10-05 16:24:54 +00001533
Erik Andersene49d5ec2000-02-08 19:58:47 +00001534 flush =
1535 ct_tally(strstart - 1 - prev_match,
1536 prev_length - MIN_MATCH);
Eric Andersencc8ed391999-10-05 16:24:54 +00001537
Erik Andersene49d5ec2000-02-08 19:58:47 +00001538 /* Insert in hash table all strings up to the end of the match.
1539 * strstart-1 and strstart are already inserted.
1540 */
1541 lookahead -= prev_length - 1;
1542 prev_length -= 2;
1543 do {
1544 strstart++;
1545 INSERT_STRING(strstart, hash_head);
1546 /* strstart never exceeds WSIZE-MAX_MATCH, so there are
1547 * always MIN_MATCH bytes ahead. If lookahead < MIN_MATCH
1548 * these bytes are garbage, but it does not matter since the
1549 * next lookahead bytes will always be emitted as literals.
1550 */
1551 } while (--prev_length != 0);
1552 match_available = 0;
1553 match_length = MIN_MATCH - 1;
1554 strstart++;
1555 if (flush)
1556 FLUSH_BLOCK(0), block_start = strstart;
Eric Andersencc8ed391999-10-05 16:24:54 +00001557
Erik Andersene49d5ec2000-02-08 19:58:47 +00001558 } else if (match_available) {
1559 /* If there was no match at the previous position, output a
1560 * single literal. If there was a match but the current match
1561 * is longer, truncate the previous match to a single literal.
1562 */
1563 Tracevv((stderr, "%c", window[strstart - 1]));
1564 if (ct_tally(0, window[strstart - 1])) {
1565 FLUSH_BLOCK(0), block_start = strstart;
1566 }
1567 strstart++;
1568 lookahead--;
1569 } else {
1570 /* There is no previous match to compare with, wait for
1571 * the next step to decide.
1572 */
1573 match_available = 1;
1574 strstart++;
1575 lookahead--;
1576 }
1577 Assert(strstart <= isize && lookahead <= isize, "a bit too far");
Eric Andersencc8ed391999-10-05 16:24:54 +00001578
Erik Andersene49d5ec2000-02-08 19:58:47 +00001579 /* Make sure that we always have enough lookahead, except
1580 * at the end of the input file. We need MAX_MATCH bytes
1581 * for the next match, plus MIN_MATCH bytes to insert the
1582 * string following the next match.
1583 */
1584 while (lookahead < MIN_LOOKAHEAD && !eofile)
1585 fill_window();
1586 }
1587 if (match_available)
1588 ct_tally(0, window[strstart - 1]);
Eric Andersencc8ed391999-10-05 16:24:54 +00001589
Erik Andersene49d5ec2000-02-08 19:58:47 +00001590 return FLUSH_BLOCK(1); /* eof */
Eric Andersencc8ed391999-10-05 16:24:54 +00001591}
Erik Andersene49d5ec2000-02-08 19:58:47 +00001592
Eric Andersencc8ed391999-10-05 16:24:54 +00001593/* gzip (GNU zip) -- compress files with zip algorithm and 'compress' interface
1594 * Copyright (C) 1992-1993 Jean-loup Gailly
1595 * The unzip code was written and put in the public domain by Mark Adler.
1596 * Portions of the lzw code are derived from the public domain 'compress'
1597 * written by Spencer Thomas, Joe Orost, James Woods, Jim McKie, Steve Davies,
1598 * Ken Turkowski, Dave Mack and Peter Jannesen.
1599 *
1600 * See the license_msg below and the file COPYING for the software license.
1601 * See the file algorithm.doc for the compression algorithms and file formats.
1602 */
1603
1604/* Compress files with zip algorithm and 'compress' interface.
1605 * See usage() and help() functions below for all options.
1606 * Outputs:
1607 * file.gz: compressed file with same mode, owner, and utimes
1608 * or stdout with -c option or if stdin used as input.
1609 * If the output file name had to be truncated, the original name is kept
1610 * in the compressed file.
1611 * On MSDOS, file.tmp -> file.tmz. On VMS, file.tmp -> file.tmp-gz.
1612 *
1613 * Using gz on MSDOS would create too many file name conflicts. For
1614 * example, foo.txt -> foo.tgz (.tgz must be reserved as shorthand for
1615 * tar.gz). Similarly, foo.dir and foo.doc would both be mapped to foo.dgz.
1616 * I also considered 12345678.txt -> 12345txt.gz but this truncates the name
1617 * too heavily. There is no ideal solution given the MSDOS 8+3 limitation.
1618 *
1619 * For the meaning of all compilation flags, see comments in Makefile.in.
1620 */
1621
Eric Andersencc8ed391999-10-05 16:24:54 +00001622#include <ctype.h>
1623#include <sys/types.h>
1624#include <signal.h>
1625#include <sys/stat.h>
1626#include <errno.h>
1627
1628 /* configuration */
1629
1630#ifdef NO_TIME_H
1631# include <sys/time.h>
1632#else
1633# include <time.h>
1634#endif
1635
1636#ifndef NO_FCNTL_H
1637# include <fcntl.h>
1638#endif
1639
1640#ifdef HAVE_UNISTD_H
1641# include <unistd.h>
1642#endif
1643
1644#if defined(STDC_HEADERS) || !defined(NO_STDLIB_H)
1645# include <stdlib.h>
1646#else
Erik Andersene49d5ec2000-02-08 19:58:47 +00001647extern int errno;
Eric Andersencc8ed391999-10-05 16:24:54 +00001648#endif
1649
1650#if defined(DIRENT)
1651# include <dirent.h>
Erik Andersene49d5ec2000-02-08 19:58:47 +00001652typedef struct dirent dir_type;
1653
Eric Andersencc8ed391999-10-05 16:24:54 +00001654# define NLENGTH(dirent) ((int)strlen((dirent)->d_name))
1655# define DIR_OPT "DIRENT"
1656#else
1657# define NLENGTH(dirent) ((dirent)->d_namlen)
1658# ifdef SYSDIR
1659# include <sys/dir.h>
Erik Andersene49d5ec2000-02-08 19:58:47 +00001660typedef struct direct dir_type;
1661
Eric Andersencc8ed391999-10-05 16:24:54 +00001662# define DIR_OPT "SYSDIR"
1663# else
1664# ifdef SYSNDIR
1665# include <sys/ndir.h>
Erik Andersene49d5ec2000-02-08 19:58:47 +00001666typedef struct direct dir_type;
1667
Eric Andersencc8ed391999-10-05 16:24:54 +00001668# define DIR_OPT "SYSNDIR"
1669# else
1670# ifdef NDIR
1671# include <ndir.h>
Erik Andersene49d5ec2000-02-08 19:58:47 +00001672typedef struct direct dir_type;
1673
Eric Andersencc8ed391999-10-05 16:24:54 +00001674# define DIR_OPT "NDIR"
1675# else
1676# define NO_DIR
1677# define DIR_OPT "NO_DIR"
1678# endif
1679# endif
1680# endif
1681#endif
1682
1683#ifndef NO_UTIME
1684# ifndef NO_UTIME_H
1685# include <utime.h>
1686# define TIME_OPT "UTIME"
1687# else
1688# ifdef HAVE_SYS_UTIME_H
1689# include <sys/utime.h>
1690# define TIME_OPT "SYS_UTIME"
1691# else
Erik Andersene49d5ec2000-02-08 19:58:47 +00001692struct utimbuf {
1693 time_t actime;
1694 time_t modtime;
1695};
1696
Eric Andersencc8ed391999-10-05 16:24:54 +00001697# define TIME_OPT ""
1698# endif
1699# endif
1700#else
1701# define TIME_OPT "NO_UTIME"
1702#endif
1703
1704#if !defined(S_ISDIR) && defined(S_IFDIR)
1705# define S_ISDIR(m) (((m) & S_IFMT) == S_IFDIR)
1706#endif
1707#if !defined(S_ISREG) && defined(S_IFREG)
1708# define S_ISREG(m) (((m) & S_IFMT) == S_IFREG)
1709#endif
1710
Erik Andersene49d5ec2000-02-08 19:58:47 +00001711typedef RETSIGTYPE(*sig_type) OF((int));
Eric Andersencc8ed391999-10-05 16:24:54 +00001712
1713#ifndef O_BINARY
Erik Andersene49d5ec2000-02-08 19:58:47 +00001714# define O_BINARY 0 /* creation mode for open() */
Eric Andersencc8ed391999-10-05 16:24:54 +00001715#endif
1716
1717#ifndef O_CREAT
1718 /* Pure BSD system? */
1719# include <sys/file.h>
1720# ifndef O_CREAT
1721# define O_CREAT FCREAT
1722# endif
1723# ifndef O_EXCL
1724# define O_EXCL FEXCL
1725# endif
1726#endif
1727
1728#ifndef S_IRUSR
1729# define S_IRUSR 0400
1730#endif
1731#ifndef S_IWUSR
1732# define S_IWUSR 0200
1733#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +00001734#define RW_USER (S_IRUSR | S_IWUSR) /* creation mode for open() */
Eric Andersencc8ed391999-10-05 16:24:54 +00001735
1736#ifndef MAX_PATH_LEN
Erik Andersene49d5ec2000-02-08 19:58:47 +00001737# define MAX_PATH_LEN 1024 /* max pathname length */
Eric Andersencc8ed391999-10-05 16:24:54 +00001738#endif
1739
1740#ifndef SEEK_END
1741# define SEEK_END 2
1742#endif
1743
1744#ifdef NO_OFF_T
Erik Andersene49d5ec2000-02-08 19:58:47 +00001745typedef long off_t;
1746off_t lseek OF((int fd, off_t offset, int whence));
Eric Andersencc8ed391999-10-05 16:24:54 +00001747#endif
1748
1749/* Separator for file name parts (see shorten_name()) */
1750#ifdef NO_MULTIPLE_DOTS
1751# define PART_SEP "-"
1752#else
1753# define PART_SEP "."
1754#endif
1755
1756 /* global buffers */
1757
Erik Andersene49d5ec2000-02-08 19:58:47 +00001758DECLARE(uch, inbuf, INBUFSIZ + INBUF_EXTRA);
1759DECLARE(uch, outbuf, OUTBUFSIZ + OUTBUF_EXTRA);
1760DECLARE(ush, d_buf, DIST_BUFSIZE);
1761DECLARE(uch, window, 2L * WSIZE);
Eric Andersencc8ed391999-10-05 16:24:54 +00001762#ifndef MAXSEG_64K
Erik Andersene49d5ec2000-02-08 19:58:47 +00001763DECLARE(ush, tab_prefix, 1L << BITS);
Eric Andersencc8ed391999-10-05 16:24:54 +00001764#else
Erik Andersene49d5ec2000-02-08 19:58:47 +00001765DECLARE(ush, tab_prefix0, 1L << (BITS - 1));
1766DECLARE(ush, tab_prefix1, 1L << (BITS - 1));
Eric Andersencc8ed391999-10-05 16:24:54 +00001767#endif
1768
1769 /* local variables */
1770
Erik Andersene49d5ec2000-02-08 19:58:47 +00001771int ascii = 0; /* convert end-of-lines to local OS conventions */
1772int decompress = 0; /* decompress (-d) */
1773int no_name = -1; /* don't save or restore the original file name */
1774int no_time = -1; /* don't save or restore the original file time */
1775int foreground; /* set if program run in foreground */
1776char *progname; /* program name */
1777static int method = DEFLATED; /* compression method */
1778static int exit_code = OK; /* program exit code */
1779int save_orig_name; /* set if original name must be saved */
1780int last_member; /* set for .zip and .Z files */
1781int part_nb; /* number of parts in .gz file */
1782long time_stamp; /* original time stamp (modification time) */
1783long ifile_size; /* input file size, -1 for devices (debug only) */
1784char *env; /* contents of GZIP env variable */
1785char **args = NULL; /* argv pointer if GZIP env variable defined */
1786char z_suffix[MAX_SUFFIX + 1]; /* default suffix (can be set with --suffix) */
1787int z_len; /* strlen(z_suffix) */
Eric Andersencc8ed391999-10-05 16:24:54 +00001788
Erik Andersene49d5ec2000-02-08 19:58:47 +00001789long bytes_in; /* number of input bytes */
1790long bytes_out; /* number of output bytes */
1791char ifname[MAX_PATH_LEN]; /* input file name */
1792char ofname[MAX_PATH_LEN]; /* output file name */
1793int remove_ofname = 0; /* remove output file on error */
1794struct stat istat; /* status for input file */
1795int ifd; /* input file descriptor */
1796int ofd; /* output file descriptor */
1797unsigned insize; /* valid bytes in inbuf */
1798unsigned inptr; /* index of next byte to be processed in inbuf */
1799unsigned outcnt; /* bytes in output buffer */
Eric Andersencc8ed391999-10-05 16:24:54 +00001800
1801/* local functions */
1802
Eric Andersencc8ed391999-10-05 16:24:54 +00001803#define strequ(s1, s2) (strcmp((s1),(s2)) == 0)
1804
1805/* ======================================================================== */
1806// int main (argc, argv)
1807// int argc;
1808// char **argv;
Erik Andersene49d5ec2000-02-08 19:58:47 +00001809int gzip_main(int argc, char **argv)
Eric Andersencc8ed391999-10-05 16:24:54 +00001810{
Erik Andersene49d5ec2000-02-08 19:58:47 +00001811 int result;
1812 int inFileNum;
1813 int outFileNum;
1814 struct stat statBuf;
1815 char *delFileName;
1816 int tostdout = 0;
1817 int fromstdin = 0;
Eric Andersen96bcfd31999-11-12 01:30:18 +00001818
Erik Andersene49d5ec2000-02-08 19:58:47 +00001819 if (argc == 1)
Eric Andersenc296b541999-11-11 01:36:55 +00001820 usage(gzip_usage);
Eric Andersenc296b541999-11-11 01:36:55 +00001821
Erik Andersene49d5ec2000-02-08 19:58:47 +00001822 /* Parse any options */
1823 while (--argc > 0 && **(++argv) == '-') {
1824 if (*((*argv) + 1) == '\0') {
1825 fromstdin = 1;
1826 tostdout = 1;
1827 }
1828 while (*(++(*argv))) {
1829 switch (**argv) {
1830 case 'c':
1831 tostdout = 1;
1832 break;
1833 default:
1834 usage(gzip_usage);
1835 }
1836 }
1837 }
1838
1839 foreground = signal(SIGINT, SIG_IGN) != SIG_IGN;
1840 if (foreground) {
1841 (void) signal(SIGINT, (sig_type) abort_gzip);
1842 }
Eric Andersencc8ed391999-10-05 16:24:54 +00001843#ifdef SIGTERM
Erik Andersene49d5ec2000-02-08 19:58:47 +00001844 if (signal(SIGTERM, SIG_IGN) != SIG_IGN) {
1845 (void) signal(SIGTERM, (sig_type) abort_gzip);
1846 }
Eric Andersencc8ed391999-10-05 16:24:54 +00001847#endif
1848#ifdef SIGHUP
Erik Andersene49d5ec2000-02-08 19:58:47 +00001849 if (signal(SIGHUP, SIG_IGN) != SIG_IGN) {
1850 (void) signal(SIGHUP, (sig_type) abort_gzip);
1851 }
Eric Andersencc8ed391999-10-05 16:24:54 +00001852#endif
1853
Erik Andersene49d5ec2000-02-08 19:58:47 +00001854 strncpy(z_suffix, Z_SUFFIX, sizeof(z_suffix) - 1);
1855 z_len = strlen(z_suffix);
Eric Andersencc8ed391999-10-05 16:24:54 +00001856
Erik Andersene49d5ec2000-02-08 19:58:47 +00001857 /* Allocate all global buffers (for DYN_ALLOC option) */
1858 ALLOC(uch, inbuf, INBUFSIZ + INBUF_EXTRA);
1859 ALLOC(uch, outbuf, OUTBUFSIZ + OUTBUF_EXTRA);
1860 ALLOC(ush, d_buf, DIST_BUFSIZE);
1861 ALLOC(uch, window, 2L * WSIZE);
Eric Andersencc8ed391999-10-05 16:24:54 +00001862#ifndef MAXSEG_64K
Erik Andersene49d5ec2000-02-08 19:58:47 +00001863 ALLOC(ush, tab_prefix, 1L << BITS);
Eric Andersencc8ed391999-10-05 16:24:54 +00001864#else
Erik Andersene49d5ec2000-02-08 19:58:47 +00001865 ALLOC(ush, tab_prefix0, 1L << (BITS - 1));
1866 ALLOC(ush, tab_prefix1, 1L << (BITS - 1));
Eric Andersencc8ed391999-10-05 16:24:54 +00001867#endif
1868
Erik Andersene49d5ec2000-02-08 19:58:47 +00001869 if (fromstdin == 1) {
1870 strcpy(ofname, "stdin");
Eric Andersen96bcfd31999-11-12 01:30:18 +00001871
Erik Andersene49d5ec2000-02-08 19:58:47 +00001872 inFileNum = fileno(stdin);
1873 time_stamp = 0; /* time unknown by default */
1874 ifile_size = -1L; /* convention for unknown size */
1875 } else {
1876 /* Open up the input file */
1877 if (*argv == '\0')
1878 usage(gzip_usage);
1879 strncpy(ifname, *argv, MAX_PATH_LEN);
Eric Andersen96bcfd31999-11-12 01:30:18 +00001880
Erik Andersene49d5ec2000-02-08 19:58:47 +00001881 /* Open input fille */
1882 inFileNum = open(ifname, O_RDONLY);
1883 if (inFileNum < 0) {
1884 perror(ifname);
1885 do_exit(WARNING);
1886 }
1887 /* Get the time stamp on the input file. */
1888 result = stat(ifname, &statBuf);
1889 if (result < 0) {
1890 perror(ifname);
1891 do_exit(WARNING);
1892 }
1893 time_stamp = statBuf.st_ctime;
1894 ifile_size = statBuf.st_size;
Eric Andersen96bcfd31999-11-12 01:30:18 +00001895 }
Eric Andersen96bcfd31999-11-12 01:30:18 +00001896
1897
Erik Andersene49d5ec2000-02-08 19:58:47 +00001898 if (tostdout == 1) {
1899 /* And get to work */
1900 strcpy(ofname, "stdout");
1901 outFileNum = fileno(stdout);
1902 SET_BINARY_MODE(fileno(stdout));
Eric Andersen0dfac6b1999-11-11 05:46:32 +00001903
Erik Andersene49d5ec2000-02-08 19:58:47 +00001904 clear_bufs(); /* clear input and output buffers */
1905 part_nb = 0;
Eric Andersen0dfac6b1999-11-11 05:46:32 +00001906
Erik Andersene49d5ec2000-02-08 19:58:47 +00001907 /* Actually do the compression/decompression. */
1908 zip(inFileNum, outFileNum);
Eric Andersen0dfac6b1999-11-11 05:46:32 +00001909
Erik Andersene49d5ec2000-02-08 19:58:47 +00001910 } else {
Eric Andersen0dfac6b1999-11-11 05:46:32 +00001911
Erik Andersene49d5ec2000-02-08 19:58:47 +00001912 /* And get to work */
1913 strncpy(ofname, ifname, MAX_PATH_LEN - 4);
1914 strcat(ofname, ".gz");
Eric Andersen0dfac6b1999-11-11 05:46:32 +00001915
Eric Andersen0dfac6b1999-11-11 05:46:32 +00001916
Erik Andersene49d5ec2000-02-08 19:58:47 +00001917 /* Open output fille */
Erik Andersen4d1d0111999-12-17 18:44:15 +00001918#if (__GLIBC__ >= 2) && (__GLIBC_MINOR__ >= 1)
Erik Andersene49d5ec2000-02-08 19:58:47 +00001919 outFileNum = open(ofname, O_RDWR | O_CREAT | O_EXCL | O_NOFOLLOW);
Erik Andersen4d1d0111999-12-17 18:44:15 +00001920#else
Erik Andersene49d5ec2000-02-08 19:58:47 +00001921 outFileNum = open(ofname, O_RDWR | O_CREAT | O_EXCL);
Erik Andersen4d1d0111999-12-17 18:44:15 +00001922#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +00001923 if (outFileNum < 0) {
1924 perror(ofname);
1925 do_exit(WARNING);
1926 }
1927 SET_BINARY_MODE(outFileNum);
1928 /* Set permissions on the file */
1929 fchmod(outFileNum, statBuf.st_mode);
1930
1931 clear_bufs(); /* clear input and output buffers */
1932 part_nb = 0;
1933
1934 /* Actually do the compression/decompression. */
1935 result = zip(inFileNum, outFileNum);
1936 close(outFileNum);
1937 close(inFileNum);
1938 /* Delete the original file */
1939 if (result == OK)
1940 delFileName = ifname;
1941 else
1942 delFileName = ofname;
1943
1944 if (unlink(delFileName) < 0) {
1945 perror(delFileName);
1946 exit(FALSE);
1947 }
Eric Andersen0dfac6b1999-11-11 05:46:32 +00001948 }
Eric Andersen0dfac6b1999-11-11 05:46:32 +00001949
Erik Andersene49d5ec2000-02-08 19:58:47 +00001950 do_exit(exit_code);
Eric Andersencc8ed391999-10-05 16:24:54 +00001951}
1952
Eric Andersencc8ed391999-10-05 16:24:54 +00001953/* trees.c -- output deflated data using Huffman coding
1954 * Copyright (C) 1992-1993 Jean-loup Gailly
1955 * This is free software; you can redistribute it and/or modify it under the
1956 * terms of the GNU General Public License, see the file COPYING.
1957 */
1958
1959/*
1960 * PURPOSE
1961 *
1962 * Encode various sets of source values using variable-length
1963 * binary code trees.
1964 *
1965 * DISCUSSION
1966 *
1967 * The PKZIP "deflation" process uses several Huffman trees. The more
1968 * common source values are represented by shorter bit sequences.
1969 *
1970 * Each code tree is stored in the ZIP file in a compressed form
1971 * which is itself a Huffman encoding of the lengths of
1972 * all the code strings (in ascending order by source values).
1973 * The actual code strings are reconstructed from the lengths in
1974 * the UNZIP process, as described in the "application note"
1975 * (APPNOTE.TXT) distributed as part of PKWARE's PKZIP program.
1976 *
1977 * REFERENCES
1978 *
1979 * Lynch, Thomas J.
1980 * Data Compression: Techniques and Applications, pp. 53-55.
1981 * Lifetime Learning Publications, 1985. ISBN 0-534-03418-7.
1982 *
1983 * Storer, James A.
1984 * Data Compression: Methods and Theory, pp. 49-50.
1985 * Computer Science Press, 1988. ISBN 0-7167-8156-5.
1986 *
1987 * Sedgewick, R.
1988 * Algorithms, p290.
1989 * Addison-Wesley, 1983. ISBN 0-201-06672-6.
1990 *
1991 * INTERFACE
1992 *
1993 * void ct_init (ush *attr, int *methodp)
1994 * Allocate the match buffer, initialize the various tables and save
1995 * the location of the internal file attribute (ascii/binary) and
1996 * method (DEFLATE/STORE)
1997 *
1998 * void ct_tally (int dist, int lc);
1999 * Save the match info and tally the frequency counts.
2000 *
2001 * long flush_block (char *buf, ulg stored_len, int eof)
2002 * Determine the best encoding for the current block: dynamic trees,
2003 * static trees or store, and output the encoded block to the zip
2004 * file. Returns the total compressed length for the file so far.
2005 *
2006 */
2007
2008#include <ctype.h>
2009
Eric Andersencc8ed391999-10-05 16:24:54 +00002010/* ===========================================================================
2011 * Constants
2012 */
2013
2014#define MAX_BITS 15
2015/* All codes must not exceed MAX_BITS bits */
2016
2017#define MAX_BL_BITS 7
2018/* Bit length codes must not exceed MAX_BL_BITS bits */
2019
2020#define LENGTH_CODES 29
2021/* number of length codes, not counting the special END_BLOCK code */
2022
2023#define LITERALS 256
2024/* number of literal bytes 0..255 */
2025
2026#define END_BLOCK 256
2027/* end of block literal code */
2028
2029#define L_CODES (LITERALS+1+LENGTH_CODES)
2030/* number of Literal or Length codes, including the END_BLOCK code */
2031
2032#define D_CODES 30
2033/* number of distance codes */
2034
2035#define BL_CODES 19
2036/* number of codes used to transfer the bit lengths */
2037
2038
Erik Andersene49d5ec2000-02-08 19:58:47 +00002039local int near extra_lbits[LENGTH_CODES] /* extra bits for each length code */
2040 = { 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3, 4, 4,
2041 4, 4, 5, 5, 5, 5, 0 };
Eric Andersencc8ed391999-10-05 16:24:54 +00002042
Erik Andersene49d5ec2000-02-08 19:58:47 +00002043local int near extra_dbits[D_CODES] /* extra bits for each distance code */
2044 = { 0, 0, 0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9,
2045 10, 10, 11, 11, 12, 12, 13, 13 };
Eric Andersencc8ed391999-10-05 16:24:54 +00002046
Erik Andersene49d5ec2000-02-08 19:58:47 +00002047local int near extra_blbits[BL_CODES] /* extra bits for each bit length code */
2048= { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2, 3, 7 };
Eric Andersencc8ed391999-10-05 16:24:54 +00002049
2050#define STORED_BLOCK 0
2051#define STATIC_TREES 1
2052#define DYN_TREES 2
2053/* The three kinds of block type */
2054
2055#ifndef LIT_BUFSIZE
2056# ifdef SMALL_MEM
2057# define LIT_BUFSIZE 0x2000
2058# else
2059# ifdef MEDIUM_MEM
2060# define LIT_BUFSIZE 0x4000
2061# else
2062# define LIT_BUFSIZE 0x8000
2063# endif
2064# endif
2065#endif
2066#ifndef DIST_BUFSIZE
2067# define DIST_BUFSIZE LIT_BUFSIZE
2068#endif
2069/* Sizes of match buffers for literals/lengths and distances. There are
2070 * 4 reasons for limiting LIT_BUFSIZE to 64K:
2071 * - frequencies can be kept in 16 bit counters
2072 * - if compression is not successful for the first block, all input data is
2073 * still in the window so we can still emit a stored block even when input
2074 * comes from standard input. (This can also be done for all blocks if
2075 * LIT_BUFSIZE is not greater than 32K.)
2076 * - if compression is not successful for a file smaller than 64K, we can
2077 * even emit a stored file instead of a stored block (saving 5 bytes).
2078 * - creating new Huffman trees less frequently may not provide fast
2079 * adaptation to changes in the input data statistics. (Take for
2080 * example a binary file with poorly compressible code followed by
2081 * a highly compressible string table.) Smaller buffer sizes give
2082 * fast adaptation but have of course the overhead of transmitting trees
2083 * more frequently.
2084 * - I can't count above 4
2085 * The current code is general and allows DIST_BUFSIZE < LIT_BUFSIZE (to save
2086 * memory at the expense of compression). Some optimizations would be possible
2087 * if we rely on DIST_BUFSIZE == LIT_BUFSIZE.
2088 */
2089#if LIT_BUFSIZE > INBUFSIZ
Erik Andersene49d5ec2000-02-08 19:58:47 +00002090error cannot overlay l_buf and inbuf
Eric Andersencc8ed391999-10-05 16:24:54 +00002091#endif
Eric Andersencc8ed391999-10-05 16:24:54 +00002092#define REP_3_6 16
2093/* repeat previous bit length 3-6 times (2 bits of repeat count) */
Eric Andersencc8ed391999-10-05 16:24:54 +00002094#define REPZ_3_10 17
2095/* repeat a zero length 3-10 times (3 bits of repeat count) */
Eric Andersencc8ed391999-10-05 16:24:54 +00002096#define REPZ_11_138 18
Erik Andersene49d5ec2000-02-08 19:58:47 +00002097/* repeat a zero length 11-138 times (7 bits of repeat count) *//* ===========================================================================
Eric Andersencc8ed391999-10-05 16:24:54 +00002098 * Local data
Erik Andersene49d5ec2000-02-08 19:58:47 +00002099 *//* Data structure describing a single value and its code string. */ typedef struct ct_data {
2100 union {
2101 ush freq; /* frequency count */
2102 ush code; /* bit string */
2103 } fc;
2104 union {
2105 ush dad; /* father node in Huffman tree */
2106 ush len; /* length of bit string */
2107 } dl;
Eric Andersencc8ed391999-10-05 16:24:54 +00002108} ct_data;
2109
2110#define Freq fc.freq
2111#define Code fc.code
2112#define Dad dl.dad
2113#define Len dl.len
2114
2115#define HEAP_SIZE (2*L_CODES+1)
2116/* maximum heap size */
2117
Erik Andersene49d5ec2000-02-08 19:58:47 +00002118local ct_data near dyn_ltree[HEAP_SIZE]; /* literal and length tree */
2119local ct_data near dyn_dtree[2 * D_CODES + 1]; /* distance tree */
Eric Andersencc8ed391999-10-05 16:24:54 +00002120
Erik Andersene49d5ec2000-02-08 19:58:47 +00002121local ct_data near static_ltree[L_CODES + 2];
2122
Eric Andersencc8ed391999-10-05 16:24:54 +00002123/* The static literal tree. Since the bit lengths are imposed, there is no
2124 * need for the L_CODES extra codes used during heap construction. However
2125 * The codes 286 and 287 are needed to build a canonical tree (see ct_init
2126 * below).
2127 */
2128
2129local ct_data near static_dtree[D_CODES];
Erik Andersene49d5ec2000-02-08 19:58:47 +00002130
Eric Andersencc8ed391999-10-05 16:24:54 +00002131/* The static distance tree. (Actually a trivial tree since all codes use
2132 * 5 bits.)
2133 */
2134
Erik Andersene49d5ec2000-02-08 19:58:47 +00002135local ct_data near bl_tree[2 * BL_CODES + 1];
2136
Eric Andersencc8ed391999-10-05 16:24:54 +00002137/* Huffman tree for the bit lengths */
2138
2139typedef struct tree_desc {
Erik Andersene49d5ec2000-02-08 19:58:47 +00002140 ct_data near *dyn_tree; /* the dynamic tree */
2141 ct_data near *static_tree; /* corresponding static tree or NULL */
2142 int near *extra_bits; /* extra bits for each code or NULL */
2143 int extra_base; /* base index for extra_bits */
2144 int elems; /* max number of elements in the tree */
2145 int max_length; /* max bit length for the codes */
2146 int max_code; /* largest code with non zero frequency */
Eric Andersencc8ed391999-10-05 16:24:54 +00002147} tree_desc;
2148
2149local tree_desc near l_desc =
Erik Andersene49d5ec2000-02-08 19:58:47 +00002150 { dyn_ltree, static_ltree, extra_lbits, LITERALS + 1, L_CODES,
2151 MAX_BITS, 0 };
Eric Andersencc8ed391999-10-05 16:24:54 +00002152
2153local tree_desc near d_desc =
Erik Andersene49d5ec2000-02-08 19:58:47 +00002154 { dyn_dtree, static_dtree, extra_dbits, 0, D_CODES, MAX_BITS, 0 };
Eric Andersencc8ed391999-10-05 16:24:54 +00002155
2156local tree_desc near bl_desc =
Erik Andersene49d5ec2000-02-08 19:58:47 +00002157 { bl_tree, (ct_data near *) 0, extra_blbits, 0, BL_CODES, MAX_BL_BITS,
2158 0 };
Eric Andersencc8ed391999-10-05 16:24:54 +00002159
2160
Erik Andersene49d5ec2000-02-08 19:58:47 +00002161local ush near bl_count[MAX_BITS + 1];
2162
Eric Andersencc8ed391999-10-05 16:24:54 +00002163/* number of codes at each bit length for an optimal tree */
2164
2165local uch near bl_order[BL_CODES]
Erik Andersene49d5ec2000-02-08 19:58:47 +00002166= { 16, 17, 18, 0, 8, 7, 9, 6, 10, 5, 11, 4, 12, 3, 13, 2, 14, 1, 15 };
2167
Eric Andersencc8ed391999-10-05 16:24:54 +00002168/* The lengths of the bit length codes are sent in order of decreasing
2169 * probability, to avoid transmitting the lengths for unused bit length codes.
2170 */
2171
Erik Andersene49d5ec2000-02-08 19:58:47 +00002172local int near heap[2 * L_CODES + 1]; /* heap used to build the Huffman trees */
2173local int heap_len; /* number of elements in the heap */
2174local int heap_max; /* element of largest frequency */
2175
Eric Andersencc8ed391999-10-05 16:24:54 +00002176/* The sons of heap[n] are heap[2*n] and heap[2*n+1]. heap[0] is not used.
2177 * The same heap array is used to build all trees.
2178 */
2179
Erik Andersene49d5ec2000-02-08 19:58:47 +00002180local uch near depth[2 * L_CODES + 1];
2181
Eric Andersencc8ed391999-10-05 16:24:54 +00002182/* Depth of each subtree used as tie breaker for trees of equal frequency */
2183
Erik Andersene49d5ec2000-02-08 19:58:47 +00002184local uch length_code[MAX_MATCH - MIN_MATCH + 1];
2185
Eric Andersencc8ed391999-10-05 16:24:54 +00002186/* length code for each normalized match length (0 == MIN_MATCH) */
2187
2188local uch dist_code[512];
Erik Andersene49d5ec2000-02-08 19:58:47 +00002189
Eric Andersencc8ed391999-10-05 16:24:54 +00002190/* distance codes. The first 256 values correspond to the distances
2191 * 3 .. 258, the last 256 values correspond to the top 8 bits of
2192 * the 15 bit distances.
2193 */
2194
2195local int near base_length[LENGTH_CODES];
Erik Andersene49d5ec2000-02-08 19:58:47 +00002196
Eric Andersencc8ed391999-10-05 16:24:54 +00002197/* First normalized length for each code (0 = MIN_MATCH) */
2198
2199local int near base_dist[D_CODES];
Erik Andersene49d5ec2000-02-08 19:58:47 +00002200
Eric Andersencc8ed391999-10-05 16:24:54 +00002201/* First normalized distance for each code (0 = distance of 1) */
2202
2203#define l_buf inbuf
2204/* DECLARE(uch, l_buf, LIT_BUFSIZE); buffer for literals or lengths */
2205
2206/* DECLARE(ush, d_buf, DIST_BUFSIZE); buffer for distances */
2207
Erik Andersene49d5ec2000-02-08 19:58:47 +00002208local uch near flag_buf[(LIT_BUFSIZE / 8)];
2209
Eric Andersencc8ed391999-10-05 16:24:54 +00002210/* flag_buf is a bit array distinguishing literals from lengths in
2211 * l_buf, thus indicating the presence or absence of a distance.
2212 */
2213
Erik Andersene49d5ec2000-02-08 19:58:47 +00002214local unsigned last_lit; /* running index in l_buf */
2215local unsigned last_dist; /* running index in d_buf */
2216local unsigned last_flags; /* running index in flag_buf */
2217local uch flags; /* current flags not yet saved in flag_buf */
2218local uch flag_bit; /* current bit used in flags */
2219
Eric Andersencc8ed391999-10-05 16:24:54 +00002220/* bits are filled in flags starting at bit 0 (least significant).
2221 * Note: these flags are overkill in the current code since we don't
2222 * take advantage of DIST_BUFSIZE == LIT_BUFSIZE.
2223 */
2224
Erik Andersene49d5ec2000-02-08 19:58:47 +00002225local ulg opt_len; /* bit length of current block with optimal trees */
2226local ulg static_len; /* bit length of current block with static trees */
Eric Andersencc8ed391999-10-05 16:24:54 +00002227
Erik Andersene49d5ec2000-02-08 19:58:47 +00002228local ulg compressed_len; /* total bit length of compressed file */
Eric Andersencc8ed391999-10-05 16:24:54 +00002229
Erik Andersene49d5ec2000-02-08 19:58:47 +00002230local ulg input_len; /* total byte length of input file */
2231
Eric Andersencc8ed391999-10-05 16:24:54 +00002232/* input_len is for debugging only since we can get it by other means. */
2233
Erik Andersene49d5ec2000-02-08 19:58:47 +00002234ush *file_type; /* pointer to UNKNOWN, BINARY or ASCII */
2235int *file_method; /* pointer to DEFLATE or STORE */
Eric Andersencc8ed391999-10-05 16:24:54 +00002236
2237#ifdef DEBUG
Erik Andersene49d5ec2000-02-08 19:58:47 +00002238extern ulg bits_sent; /* bit length of the compressed data */
2239extern long isize; /* byte length of input file */
Eric Andersencc8ed391999-10-05 16:24:54 +00002240#endif
2241
Erik Andersene49d5ec2000-02-08 19:58:47 +00002242extern long block_start; /* window offset of current block */
2243extern unsigned near strstart; /* window offset of current string */
Eric Andersencc8ed391999-10-05 16:24:54 +00002244
2245/* ===========================================================================
2246 * Local (static) routines in this file.
2247 */
2248
Erik Andersene49d5ec2000-02-08 19:58:47 +00002249local void init_block OF((void));
2250local void pqdownheap OF((ct_data near * tree, int k));
2251local void gen_bitlen OF((tree_desc near * desc));
2252local void gen_codes OF((ct_data near * tree, int max_code));
2253local void build_tree OF((tree_desc near * desc));
2254local void scan_tree OF((ct_data near * tree, int max_code));
2255local void send_tree OF((ct_data near * tree, int max_code));
2256local int build_bl_tree OF((void));
Eric Andersencc8ed391999-10-05 16:24:54 +00002257local void send_all_trees OF((int lcodes, int dcodes, int blcodes));
Erik Andersene49d5ec2000-02-08 19:58:47 +00002258local void compress_block OF((ct_data near * ltree, ct_data near * dtree));
2259local void set_file_type OF((void));
Eric Andersencc8ed391999-10-05 16:24:54 +00002260
2261
2262#ifndef DEBUG
2263# define send_code(c, tree) send_bits(tree[c].Code, tree[c].Len)
2264 /* Send a code of the given tree. c and tree must not have side effects */
2265
Erik Andersene49d5ec2000-02-08 19:58:47 +00002266#else /* DEBUG */
Eric Andersencc8ed391999-10-05 16:24:54 +00002267# define send_code(c, tree) \
2268 { if (verbose>1) fprintf(stderr,"\ncd %3d ",(c)); \
2269 send_bits(tree[c].Code, tree[c].Len); }
2270#endif
2271
2272#define d_code(dist) \
2273 ((dist) < 256 ? dist_code[dist] : dist_code[256+((dist)>>7)])
2274/* Mapping from a distance to a distance code. dist is the distance - 1 and
2275 * must not have side effects. dist_code[256] and dist_code[257] are never
2276 * used.
2277 */
2278
2279#define MAX(a,b) (a >= b ? a : b)
2280/* the arguments must not have side effects */
2281
2282/* ===========================================================================
2283 * Allocate the match buffer, initialize the various tables and save the
2284 * location of the internal file attribute (ascii/binary) and method
2285 * (DEFLATE/STORE).
2286 */
2287void ct_init(attr, methodp)
Erik Andersene49d5ec2000-02-08 19:58:47 +00002288ush *attr; /* pointer to internal file attribute */
2289int *methodp; /* pointer to compression method */
Eric Andersencc8ed391999-10-05 16:24:54 +00002290{
Erik Andersene49d5ec2000-02-08 19:58:47 +00002291 int n; /* iterates over tree elements */
2292 int bits; /* bit counter */
2293 int length; /* length value */
2294 int code; /* code value */
2295 int dist; /* distance index */
Eric Andersencc8ed391999-10-05 16:24:54 +00002296
Erik Andersene49d5ec2000-02-08 19:58:47 +00002297 file_type = attr;
2298 file_method = methodp;
2299 compressed_len = input_len = 0L;
Eric Andersencc8ed391999-10-05 16:24:54 +00002300
Erik Andersene49d5ec2000-02-08 19:58:47 +00002301 if (static_dtree[0].Len != 0)
2302 return; /* ct_init already called */
Eric Andersencc8ed391999-10-05 16:24:54 +00002303
Erik Andersene49d5ec2000-02-08 19:58:47 +00002304 /* Initialize the mapping length (0..255) -> length code (0..28) */
2305 length = 0;
2306 for (code = 0; code < LENGTH_CODES - 1; code++) {
2307 base_length[code] = length;
2308 for (n = 0; n < (1 << extra_lbits[code]); n++) {
2309 length_code[length++] = (uch) code;
2310 }
2311 }
2312 Assert(length == 256, "ct_init: length != 256");
2313 /* Note that the length 255 (match length 258) can be represented
2314 * in two different ways: code 284 + 5 bits or code 285, so we
2315 * overwrite length_code[255] to use the best encoding:
2316 */
2317 length_code[length - 1] = (uch) code;
Eric Andersencc8ed391999-10-05 16:24:54 +00002318
Erik Andersene49d5ec2000-02-08 19:58:47 +00002319 /* Initialize the mapping dist (0..32K) -> dist code (0..29) */
2320 dist = 0;
2321 for (code = 0; code < 16; code++) {
2322 base_dist[code] = dist;
2323 for (n = 0; n < (1 << extra_dbits[code]); n++) {
2324 dist_code[dist++] = (uch) code;
2325 }
2326 }
2327 Assert(dist == 256, "ct_init: dist != 256");
2328 dist >>= 7; /* from now on, all distances are divided by 128 */
2329 for (; code < D_CODES; code++) {
2330 base_dist[code] = dist << 7;
2331 for (n = 0; n < (1 << (extra_dbits[code] - 7)); n++) {
2332 dist_code[256 + dist++] = (uch) code;
2333 }
2334 }
2335 Assert(dist == 256, "ct_init: 256+dist != 512");
Eric Andersencc8ed391999-10-05 16:24:54 +00002336
Erik Andersene49d5ec2000-02-08 19:58:47 +00002337 /* Construct the codes of the static literal tree */
2338 for (bits = 0; bits <= MAX_BITS; bits++)
2339 bl_count[bits] = 0;
2340 n = 0;
2341 while (n <= 143)
2342 static_ltree[n++].Len = 8, bl_count[8]++;
2343 while (n <= 255)
2344 static_ltree[n++].Len = 9, bl_count[9]++;
2345 while (n <= 279)
2346 static_ltree[n++].Len = 7, bl_count[7]++;
2347 while (n <= 287)
2348 static_ltree[n++].Len = 8, bl_count[8]++;
2349 /* Codes 286 and 287 do not exist, but we must include them in the
2350 * tree construction to get a canonical Huffman tree (longest code
2351 * all ones)
2352 */
2353 gen_codes((ct_data near *) static_ltree, L_CODES + 1);
Eric Andersencc8ed391999-10-05 16:24:54 +00002354
Erik Andersene49d5ec2000-02-08 19:58:47 +00002355 /* The static distance tree is trivial: */
2356 for (n = 0; n < D_CODES; n++) {
2357 static_dtree[n].Len = 5;
2358 static_dtree[n].Code = bi_reverse(n, 5);
2359 }
2360
2361 /* Initialize the first block of the first file: */
2362 init_block();
Eric Andersencc8ed391999-10-05 16:24:54 +00002363}
2364
2365/* ===========================================================================
2366 * Initialize a new block.
2367 */
2368local void init_block()
2369{
Erik Andersene49d5ec2000-02-08 19:58:47 +00002370 int n; /* iterates over tree elements */
Eric Andersencc8ed391999-10-05 16:24:54 +00002371
Erik Andersene49d5ec2000-02-08 19:58:47 +00002372 /* Initialize the trees. */
2373 for (n = 0; n < L_CODES; n++)
2374 dyn_ltree[n].Freq = 0;
2375 for (n = 0; n < D_CODES; n++)
2376 dyn_dtree[n].Freq = 0;
2377 for (n = 0; n < BL_CODES; n++)
2378 bl_tree[n].Freq = 0;
Eric Andersencc8ed391999-10-05 16:24:54 +00002379
Erik Andersene49d5ec2000-02-08 19:58:47 +00002380 dyn_ltree[END_BLOCK].Freq = 1;
2381 opt_len = static_len = 0L;
2382 last_lit = last_dist = last_flags = 0;
2383 flags = 0;
2384 flag_bit = 1;
Eric Andersencc8ed391999-10-05 16:24:54 +00002385}
2386
2387#define SMALLEST 1
2388/* Index within the heap array of least frequent node in the Huffman tree */
2389
2390
2391/* ===========================================================================
2392 * Remove the smallest element from the heap and recreate the heap with
2393 * one less element. Updates heap and heap_len.
2394 */
2395#define pqremove(tree, top) \
2396{\
2397 top = heap[SMALLEST]; \
2398 heap[SMALLEST] = heap[heap_len--]; \
2399 pqdownheap(tree, SMALLEST); \
2400}
2401
2402/* ===========================================================================
2403 * Compares to subtrees, using the tree depth as tie breaker when
2404 * the subtrees have equal frequency. This minimizes the worst case length.
2405 */
2406#define smaller(tree, n, m) \
2407 (tree[n].Freq < tree[m].Freq || \
2408 (tree[n].Freq == tree[m].Freq && depth[n] <= depth[m]))
2409
2410/* ===========================================================================
2411 * Restore the heap property by moving down the tree starting at node k,
2412 * exchanging a node with the smallest of its two sons if necessary, stopping
2413 * when the heap property is re-established (each father smaller than its
2414 * two sons).
2415 */
2416local void pqdownheap(tree, k)
Erik Andersene49d5ec2000-02-08 19:58:47 +00002417ct_data near *tree; /* the tree to restore */
2418int k; /* node to move down */
Eric Andersencc8ed391999-10-05 16:24:54 +00002419{
Erik Andersene49d5ec2000-02-08 19:58:47 +00002420 int v = heap[k];
2421 int j = k << 1; /* left son of k */
Eric Andersencc8ed391999-10-05 16:24:54 +00002422
Erik Andersene49d5ec2000-02-08 19:58:47 +00002423 while (j <= heap_len) {
2424 /* Set j to the smallest of the two sons: */
2425 if (j < heap_len && smaller(tree, heap[j + 1], heap[j]))
2426 j++;
Eric Andersencc8ed391999-10-05 16:24:54 +00002427
Erik Andersene49d5ec2000-02-08 19:58:47 +00002428 /* Exit if v is smaller than both sons */
2429 if (smaller(tree, v, heap[j]))
2430 break;
Eric Andersencc8ed391999-10-05 16:24:54 +00002431
Erik Andersene49d5ec2000-02-08 19:58:47 +00002432 /* Exchange v with the smallest son */
2433 heap[k] = heap[j];
2434 k = j;
2435
2436 /* And continue down the tree, setting j to the left son of k */
2437 j <<= 1;
2438 }
2439 heap[k] = v;
Eric Andersencc8ed391999-10-05 16:24:54 +00002440}
2441
2442/* ===========================================================================
2443 * Compute the optimal bit lengths for a tree and update the total bit length
2444 * for the current block.
2445 * IN assertion: the fields freq and dad are set, heap[heap_max] and
2446 * above are the tree nodes sorted by increasing frequency.
2447 * OUT assertions: the field len is set to the optimal bit length, the
2448 * array bl_count contains the frequencies for each bit length.
2449 * The length opt_len is updated; static_len is also updated if stree is
2450 * not null.
2451 */
2452local void gen_bitlen(desc)
Erik Andersene49d5ec2000-02-08 19:58:47 +00002453tree_desc near *desc; /* the tree descriptor */
Eric Andersencc8ed391999-10-05 16:24:54 +00002454{
Erik Andersene49d5ec2000-02-08 19:58:47 +00002455 ct_data near *tree = desc->dyn_tree;
2456 int near *extra = desc->extra_bits;
2457 int base = desc->extra_base;
2458 int max_code = desc->max_code;
2459 int max_length = desc->max_length;
2460 ct_data near *stree = desc->static_tree;
2461 int h; /* heap index */
2462 int n, m; /* iterate over the tree elements */
2463 int bits; /* bit length */
2464 int xbits; /* extra bits */
2465 ush f; /* frequency */
2466 int overflow = 0; /* number of elements with bit length too large */
Eric Andersencc8ed391999-10-05 16:24:54 +00002467
Erik Andersene49d5ec2000-02-08 19:58:47 +00002468 for (bits = 0; bits <= MAX_BITS; bits++)
2469 bl_count[bits] = 0;
Eric Andersencc8ed391999-10-05 16:24:54 +00002470
Erik Andersene49d5ec2000-02-08 19:58:47 +00002471 /* In a first pass, compute the optimal bit lengths (which may
2472 * overflow in the case of the bit length tree).
2473 */
2474 tree[heap[heap_max]].Len = 0; /* root of the heap */
Eric Andersencc8ed391999-10-05 16:24:54 +00002475
Erik Andersene49d5ec2000-02-08 19:58:47 +00002476 for (h = heap_max + 1; h < HEAP_SIZE; h++) {
2477 n = heap[h];
2478 bits = tree[tree[n].Dad].Len + 1;
2479 if (bits > max_length)
2480 bits = max_length, overflow++;
2481 tree[n].Len = (ush) bits;
2482 /* We overwrite tree[n].Dad which is no longer needed */
Eric Andersencc8ed391999-10-05 16:24:54 +00002483
Erik Andersene49d5ec2000-02-08 19:58:47 +00002484 if (n > max_code)
2485 continue; /* not a leaf node */
Eric Andersencc8ed391999-10-05 16:24:54 +00002486
Erik Andersene49d5ec2000-02-08 19:58:47 +00002487 bl_count[bits]++;
2488 xbits = 0;
2489 if (n >= base)
2490 xbits = extra[n - base];
2491 f = tree[n].Freq;
2492 opt_len += (ulg) f *(bits + xbits);
Eric Andersencc8ed391999-10-05 16:24:54 +00002493
Erik Andersene49d5ec2000-02-08 19:58:47 +00002494 if (stree)
2495 static_len += (ulg) f *(stree[n].Len + xbits);
2496 }
2497 if (overflow == 0)
2498 return;
Eric Andersencc8ed391999-10-05 16:24:54 +00002499
Erik Andersene49d5ec2000-02-08 19:58:47 +00002500 Trace((stderr, "\nbit length overflow\n"));
2501 /* This happens for example on obj2 and pic of the Calgary corpus */
Eric Andersencc8ed391999-10-05 16:24:54 +00002502
Erik Andersene49d5ec2000-02-08 19:58:47 +00002503 /* Find the first bit length which could increase: */
2504 do {
2505 bits = max_length - 1;
2506 while (bl_count[bits] == 0)
2507 bits--;
2508 bl_count[bits]--; /* move one leaf down the tree */
2509 bl_count[bits + 1] += 2; /* move one overflow item as its brother */
2510 bl_count[max_length]--;
2511 /* The brother of the overflow item also moves one step up,
2512 * but this does not affect bl_count[max_length]
2513 */
2514 overflow -= 2;
2515 } while (overflow > 0);
2516
2517 /* Now recompute all bit lengths, scanning in increasing frequency.
2518 * h is still equal to HEAP_SIZE. (It is simpler to reconstruct all
2519 * lengths instead of fixing only the wrong ones. This idea is taken
2520 * from 'ar' written by Haruhiko Okumura.)
2521 */
2522 for (bits = max_length; bits != 0; bits--) {
2523 n = bl_count[bits];
2524 while (n != 0) {
2525 m = heap[--h];
2526 if (m > max_code)
2527 continue;
2528 if (tree[m].Len != (unsigned) bits) {
2529 Trace(
2530 (stderr, "code %d bits %d->%d\n", m, tree[m].Len,
2531 bits));
2532 opt_len +=
2533 ((long) bits -
2534 (long) tree[m].Len) * (long) tree[m].Freq;
2535 tree[m].Len = (ush) bits;
2536 }
2537 n--;
2538 }
2539 }
Eric Andersencc8ed391999-10-05 16:24:54 +00002540}
2541
2542/* ===========================================================================
2543 * Generate the codes for a given tree and bit counts (which need not be
2544 * optimal).
2545 * IN assertion: the array bl_count contains the bit length statistics for
2546 * the given tree and the field len is set for all tree elements.
2547 * OUT assertion: the field code is set for all tree elements of non
2548 * zero code length.
2549 */
Erik Andersene49d5ec2000-02-08 19:58:47 +00002550local void gen_codes(tree, max_code)
2551ct_data near *tree; /* the tree to decorate */
2552int max_code; /* largest code with non zero frequency */
Eric Andersencc8ed391999-10-05 16:24:54 +00002553{
Erik Andersene49d5ec2000-02-08 19:58:47 +00002554 ush next_code[MAX_BITS + 1]; /* next code value for each bit length */
2555 ush code = 0; /* running code value */
2556 int bits; /* bit index */
2557 int n; /* code index */
Eric Andersencc8ed391999-10-05 16:24:54 +00002558
Erik Andersene49d5ec2000-02-08 19:58:47 +00002559 /* The distribution counts are first used to generate the code values
2560 * without bit reversal.
2561 */
2562 for (bits = 1; bits <= MAX_BITS; bits++) {
2563 next_code[bits] = code = (code + bl_count[bits - 1]) << 1;
2564 }
2565 /* Check that the bit counts in bl_count are consistent. The last code
2566 * must be all ones.
2567 */
2568 Assert(code + bl_count[MAX_BITS] - 1 == (1 << MAX_BITS) - 1,
2569 "inconsistent bit counts");
2570 Tracev((stderr, "\ngen_codes: max_code %d ", max_code));
Eric Andersencc8ed391999-10-05 16:24:54 +00002571
Erik Andersene49d5ec2000-02-08 19:58:47 +00002572 for (n = 0; n <= max_code; n++) {
2573 int len = tree[n].Len;
Eric Andersencc8ed391999-10-05 16:24:54 +00002574
Erik Andersene49d5ec2000-02-08 19:58:47 +00002575 if (len == 0)
2576 continue;
2577 /* Now reverse the bits */
2578 tree[n].Code = bi_reverse(next_code[len]++, len);
2579
2580 Tracec(tree != static_ltree,
2581 (stderr, "\nn %3d %c l %2d c %4x (%x) ", n,
2582 (isgraph(n) ? n : ' '), len, tree[n].Code,
2583 next_code[len] - 1));
2584 }
Eric Andersencc8ed391999-10-05 16:24:54 +00002585}
2586
2587/* ===========================================================================
2588 * Construct one Huffman tree and assigns the code bit strings and lengths.
2589 * Update the total bit length for the current block.
2590 * IN assertion: the field freq is set for all tree elements.
2591 * OUT assertions: the fields len and code are set to the optimal bit length
2592 * and corresponding code. The length opt_len is updated; static_len is
2593 * also updated if stree is not null. The field max_code is set.
2594 */
2595local void build_tree(desc)
Erik Andersene49d5ec2000-02-08 19:58:47 +00002596tree_desc near *desc; /* the tree descriptor */
Eric Andersencc8ed391999-10-05 16:24:54 +00002597{
Erik Andersene49d5ec2000-02-08 19:58:47 +00002598 ct_data near *tree = desc->dyn_tree;
2599 ct_data near *stree = desc->static_tree;
2600 int elems = desc->elems;
2601 int n, m; /* iterate over heap elements */
2602 int max_code = -1; /* largest code with non zero frequency */
2603 int node = elems; /* next internal node of the tree */
Eric Andersencc8ed391999-10-05 16:24:54 +00002604
Erik Andersene49d5ec2000-02-08 19:58:47 +00002605 /* Construct the initial heap, with least frequent element in
2606 * heap[SMALLEST]. The sons of heap[n] are heap[2*n] and heap[2*n+1].
2607 * heap[0] is not used.
2608 */
2609 heap_len = 0, heap_max = HEAP_SIZE;
Eric Andersencc8ed391999-10-05 16:24:54 +00002610
Erik Andersene49d5ec2000-02-08 19:58:47 +00002611 for (n = 0; n < elems; n++) {
2612 if (tree[n].Freq != 0) {
2613 heap[++heap_len] = max_code = n;
2614 depth[n] = 0;
2615 } else {
2616 tree[n].Len = 0;
2617 }
2618 }
Eric Andersencc8ed391999-10-05 16:24:54 +00002619
Erik Andersene49d5ec2000-02-08 19:58:47 +00002620 /* The pkzip format requires that at least one distance code exists,
2621 * and that at least one bit should be sent even if there is only one
2622 * possible code. So to avoid special checks later on we force at least
2623 * two codes of non zero frequency.
2624 */
2625 while (heap_len < 2) {
2626 int new = heap[++heap_len] = (max_code < 2 ? ++max_code : 0);
Eric Andersencc8ed391999-10-05 16:24:54 +00002627
Erik Andersene49d5ec2000-02-08 19:58:47 +00002628 tree[new].Freq = 1;
2629 depth[new] = 0;
2630 opt_len--;
2631 if (stree)
2632 static_len -= stree[new].Len;
2633 /* new is 0 or 1 so it does not have extra bits */
2634 }
2635 desc->max_code = max_code;
Eric Andersencc8ed391999-10-05 16:24:54 +00002636
Erik Andersene49d5ec2000-02-08 19:58:47 +00002637 /* The elements heap[heap_len/2+1 .. heap_len] are leaves of the tree,
2638 * establish sub-heaps of increasing lengths:
2639 */
2640 for (n = heap_len / 2; n >= 1; n--)
2641 pqdownheap(tree, n);
Eric Andersencc8ed391999-10-05 16:24:54 +00002642
Erik Andersene49d5ec2000-02-08 19:58:47 +00002643 /* Construct the Huffman tree by repeatedly combining the least two
2644 * frequent nodes.
2645 */
2646 do {
2647 pqremove(tree, n); /* n = node of least frequency */
2648 m = heap[SMALLEST]; /* m = node of next least frequency */
Eric Andersencc8ed391999-10-05 16:24:54 +00002649
Erik Andersene49d5ec2000-02-08 19:58:47 +00002650 heap[--heap_max] = n; /* keep the nodes sorted by frequency */
2651 heap[--heap_max] = m;
2652
2653 /* Create a new node father of n and m */
2654 tree[node].Freq = tree[n].Freq + tree[m].Freq;
2655 depth[node] = (uch) (MAX(depth[n], depth[m]) + 1);
2656 tree[n].Dad = tree[m].Dad = (ush) node;
Eric Andersencc8ed391999-10-05 16:24:54 +00002657#ifdef DUMP_BL_TREE
Erik Andersene49d5ec2000-02-08 19:58:47 +00002658 if (tree == bl_tree) {
2659 fprintf(stderr, "\nnode %d(%d), sons %d(%d) %d(%d)",
2660 node, tree[node].Freq, n, tree[n].Freq, m,
2661 tree[m].Freq);
2662 }
Eric Andersencc8ed391999-10-05 16:24:54 +00002663#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +00002664 /* and insert the new node in the heap */
2665 heap[SMALLEST] = node++;
2666 pqdownheap(tree, SMALLEST);
Eric Andersencc8ed391999-10-05 16:24:54 +00002667
Erik Andersene49d5ec2000-02-08 19:58:47 +00002668 } while (heap_len >= 2);
Eric Andersencc8ed391999-10-05 16:24:54 +00002669
Erik Andersene49d5ec2000-02-08 19:58:47 +00002670 heap[--heap_max] = heap[SMALLEST];
Eric Andersencc8ed391999-10-05 16:24:54 +00002671
Erik Andersene49d5ec2000-02-08 19:58:47 +00002672 /* At this point, the fields freq and dad are set. We can now
2673 * generate the bit lengths.
2674 */
2675 gen_bitlen((tree_desc near *) desc);
Eric Andersencc8ed391999-10-05 16:24:54 +00002676
Erik Andersene49d5ec2000-02-08 19:58:47 +00002677 /* The field len is now set, we can generate the bit codes */
2678 gen_codes((ct_data near *) tree, max_code);
Eric Andersencc8ed391999-10-05 16:24:54 +00002679}
2680
2681/* ===========================================================================
2682 * Scan a literal or distance tree to determine the frequencies of the codes
2683 * in the bit length tree. Updates opt_len to take into account the repeat
2684 * counts. (The contribution of the bit length codes will be added later
2685 * during the construction of bl_tree.)
2686 */
Erik Andersene49d5ec2000-02-08 19:58:47 +00002687local void scan_tree(tree, max_code)
2688ct_data near *tree; /* the tree to be scanned */
2689int max_code; /* and its largest code of non zero frequency */
Eric Andersencc8ed391999-10-05 16:24:54 +00002690{
Erik Andersene49d5ec2000-02-08 19:58:47 +00002691 int n; /* iterates over all tree elements */
2692 int prevlen = -1; /* last emitted length */
2693 int curlen; /* length of current code */
2694 int nextlen = tree[0].Len; /* length of next code */
2695 int count = 0; /* repeat count of the current code */
2696 int max_count = 7; /* max repeat count */
2697 int min_count = 4; /* min repeat count */
Eric Andersencc8ed391999-10-05 16:24:54 +00002698
Erik Andersene49d5ec2000-02-08 19:58:47 +00002699 if (nextlen == 0)
2700 max_count = 138, min_count = 3;
2701 tree[max_code + 1].Len = (ush) 0xffff; /* guard */
Eric Andersencc8ed391999-10-05 16:24:54 +00002702
Erik Andersene49d5ec2000-02-08 19:58:47 +00002703 for (n = 0; n <= max_code; n++) {
2704 curlen = nextlen;
2705 nextlen = tree[n + 1].Len;
2706 if (++count < max_count && curlen == nextlen) {
2707 continue;
2708 } else if (count < min_count) {
2709 bl_tree[curlen].Freq += count;
2710 } else if (curlen != 0) {
2711 if (curlen != prevlen)
2712 bl_tree[curlen].Freq++;
2713 bl_tree[REP_3_6].Freq++;
2714 } else if (count <= 10) {
2715 bl_tree[REPZ_3_10].Freq++;
2716 } else {
2717 bl_tree[REPZ_11_138].Freq++;
2718 }
2719 count = 0;
2720 prevlen = curlen;
2721 if (nextlen == 0) {
2722 max_count = 138, min_count = 3;
2723 } else if (curlen == nextlen) {
2724 max_count = 6, min_count = 3;
2725 } else {
2726 max_count = 7, min_count = 4;
2727 }
2728 }
Eric Andersencc8ed391999-10-05 16:24:54 +00002729}
2730
2731/* ===========================================================================
2732 * Send a literal or distance tree in compressed form, using the codes in
2733 * bl_tree.
2734 */
Erik Andersene49d5ec2000-02-08 19:58:47 +00002735local void send_tree(tree, max_code)
2736ct_data near *tree; /* the tree to be scanned */
2737int max_code; /* and its largest code of non zero frequency */
Eric Andersencc8ed391999-10-05 16:24:54 +00002738{
Erik Andersene49d5ec2000-02-08 19:58:47 +00002739 int n; /* iterates over all tree elements */
2740 int prevlen = -1; /* last emitted length */
2741 int curlen; /* length of current code */
2742 int nextlen = tree[0].Len; /* length of next code */
2743 int count = 0; /* repeat count of the current code */
2744 int max_count = 7; /* max repeat count */
2745 int min_count = 4; /* min repeat count */
Eric Andersencc8ed391999-10-05 16:24:54 +00002746
Erik Andersene49d5ec2000-02-08 19:58:47 +00002747/* tree[max_code+1].Len = -1; *//* guard already set */
2748 if (nextlen == 0)
2749 max_count = 138, min_count = 3;
Eric Andersencc8ed391999-10-05 16:24:54 +00002750
Erik Andersene49d5ec2000-02-08 19:58:47 +00002751 for (n = 0; n <= max_code; n++) {
2752 curlen = nextlen;
2753 nextlen = tree[n + 1].Len;
2754 if (++count < max_count && curlen == nextlen) {
2755 continue;
2756 } else if (count < min_count) {
2757 do {
2758 send_code(curlen, bl_tree);
2759 } while (--count != 0);
Eric Andersencc8ed391999-10-05 16:24:54 +00002760
Erik Andersene49d5ec2000-02-08 19:58:47 +00002761 } else if (curlen != 0) {
2762 if (curlen != prevlen) {
2763 send_code(curlen, bl_tree);
2764 count--;
2765 }
2766 Assert(count >= 3 && count <= 6, " 3_6?");
2767 send_code(REP_3_6, bl_tree);
2768 send_bits(count - 3, 2);
Eric Andersencc8ed391999-10-05 16:24:54 +00002769
Erik Andersene49d5ec2000-02-08 19:58:47 +00002770 } else if (count <= 10) {
2771 send_code(REPZ_3_10, bl_tree);
2772 send_bits(count - 3, 3);
Eric Andersencc8ed391999-10-05 16:24:54 +00002773
Erik Andersene49d5ec2000-02-08 19:58:47 +00002774 } else {
2775 send_code(REPZ_11_138, bl_tree);
2776 send_bits(count - 11, 7);
2777 }
2778 count = 0;
2779 prevlen = curlen;
2780 if (nextlen == 0) {
2781 max_count = 138, min_count = 3;
2782 } else if (curlen == nextlen) {
2783 max_count = 6, min_count = 3;
2784 } else {
2785 max_count = 7, min_count = 4;
2786 }
2787 }
Eric Andersencc8ed391999-10-05 16:24:54 +00002788}
2789
2790/* ===========================================================================
2791 * Construct the Huffman tree for the bit lengths and return the index in
2792 * bl_order of the last bit length code to send.
2793 */
2794local int build_bl_tree()
2795{
Erik Andersene49d5ec2000-02-08 19:58:47 +00002796 int max_blindex; /* index of last bit length code of non zero freq */
Eric Andersencc8ed391999-10-05 16:24:54 +00002797
Erik Andersene49d5ec2000-02-08 19:58:47 +00002798 /* Determine the bit length frequencies for literal and distance trees */
2799 scan_tree((ct_data near *) dyn_ltree, l_desc.max_code);
2800 scan_tree((ct_data near *) dyn_dtree, d_desc.max_code);
Eric Andersencc8ed391999-10-05 16:24:54 +00002801
Erik Andersene49d5ec2000-02-08 19:58:47 +00002802 /* Build the bit length tree: */
2803 build_tree((tree_desc near *) (&bl_desc));
2804 /* opt_len now includes the length of the tree representations, except
2805 * the lengths of the bit lengths codes and the 5+5+4 bits for the counts.
2806 */
Eric Andersencc8ed391999-10-05 16:24:54 +00002807
Erik Andersene49d5ec2000-02-08 19:58:47 +00002808 /* Determine the number of bit length codes to send. The pkzip format
2809 * requires that at least 4 bit length codes be sent. (appnote.txt says
2810 * 3 but the actual value used is 4.)
2811 */
2812 for (max_blindex = BL_CODES - 1; max_blindex >= 3; max_blindex--) {
2813 if (bl_tree[bl_order[max_blindex]].Len != 0)
2814 break;
2815 }
2816 /* Update opt_len to include the bit length tree and counts */
2817 opt_len += 3 * (max_blindex + 1) + 5 + 5 + 4;
2818 Tracev(
2819 (stderr, "\ndyn trees: dyn %ld, stat %ld", opt_len,
2820 static_len));
Eric Andersencc8ed391999-10-05 16:24:54 +00002821
Erik Andersene49d5ec2000-02-08 19:58:47 +00002822 return max_blindex;
Eric Andersencc8ed391999-10-05 16:24:54 +00002823}
2824
2825/* ===========================================================================
2826 * Send the header for a block using dynamic Huffman trees: the counts, the
2827 * lengths of the bit length codes, the literal tree and the distance tree.
2828 * IN assertion: lcodes >= 257, dcodes >= 1, blcodes >= 4.
2829 */
2830local void send_all_trees(lcodes, dcodes, blcodes)
Erik Andersene49d5ec2000-02-08 19:58:47 +00002831int lcodes, dcodes, blcodes; /* number of codes for each tree */
Eric Andersencc8ed391999-10-05 16:24:54 +00002832{
Erik Andersene49d5ec2000-02-08 19:58:47 +00002833 int rank; /* index in bl_order */
Eric Andersencc8ed391999-10-05 16:24:54 +00002834
Erik Andersene49d5ec2000-02-08 19:58:47 +00002835 Assert(lcodes >= 257 && dcodes >= 1
2836 && blcodes >= 4, "not enough codes");
2837 Assert(lcodes <= L_CODES && dcodes <= D_CODES
2838 && blcodes <= BL_CODES, "too many codes");
2839 Tracev((stderr, "\nbl counts: "));
2840 send_bits(lcodes - 257, 5); /* not +255 as stated in appnote.txt */
2841 send_bits(dcodes - 1, 5);
2842 send_bits(blcodes - 4, 4); /* not -3 as stated in appnote.txt */
2843 for (rank = 0; rank < blcodes; rank++) {
2844 Tracev((stderr, "\nbl code %2d ", bl_order[rank]));
2845 send_bits(bl_tree[bl_order[rank]].Len, 3);
2846 }
2847 Tracev((stderr, "\nbl tree: sent %ld", bits_sent));
Eric Andersencc8ed391999-10-05 16:24:54 +00002848
Erik Andersene49d5ec2000-02-08 19:58:47 +00002849 send_tree((ct_data near *) dyn_ltree, lcodes - 1); /* send the literal tree */
2850 Tracev((stderr, "\nlit tree: sent %ld", bits_sent));
Eric Andersencc8ed391999-10-05 16:24:54 +00002851
Erik Andersene49d5ec2000-02-08 19:58:47 +00002852 send_tree((ct_data near *) dyn_dtree, dcodes - 1); /* send the distance tree */
2853 Tracev((stderr, "\ndist tree: sent %ld", bits_sent));
Eric Andersencc8ed391999-10-05 16:24:54 +00002854}
2855
2856/* ===========================================================================
2857 * Determine the best encoding for the current block: dynamic trees, static
2858 * trees or store, and output the encoded block to the zip file. This function
2859 * returns the total compressed length for the file so far.
2860 */
2861ulg flush_block(buf, stored_len, eof)
Erik Andersene49d5ec2000-02-08 19:58:47 +00002862char *buf; /* input block, or NULL if too old */
2863ulg stored_len; /* length of input block */
2864int eof; /* true if this is the last block for a file */
Eric Andersencc8ed391999-10-05 16:24:54 +00002865{
Erik Andersene49d5ec2000-02-08 19:58:47 +00002866 ulg opt_lenb, static_lenb; /* opt_len and static_len in bytes */
2867 int max_blindex; /* index of last bit length code of non zero freq */
Eric Andersencc8ed391999-10-05 16:24:54 +00002868
Erik Andersene49d5ec2000-02-08 19:58:47 +00002869 flag_buf[last_flags] = flags; /* Save the flags for the last 8 items */
Eric Andersencc8ed391999-10-05 16:24:54 +00002870
Erik Andersene49d5ec2000-02-08 19:58:47 +00002871 /* Check if the file is ascii or binary */
2872 if (*file_type == (ush) UNKNOWN)
2873 set_file_type();
Eric Andersencc8ed391999-10-05 16:24:54 +00002874
Erik Andersene49d5ec2000-02-08 19:58:47 +00002875 /* Construct the literal and distance trees */
2876 build_tree((tree_desc near *) (&l_desc));
2877 Tracev((stderr, "\nlit data: dyn %ld, stat %ld", opt_len, static_len));
Eric Andersencc8ed391999-10-05 16:24:54 +00002878
Erik Andersene49d5ec2000-02-08 19:58:47 +00002879 build_tree((tree_desc near *) (&d_desc));
2880 Tracev(
2881 (stderr, "\ndist data: dyn %ld, stat %ld", opt_len,
2882 static_len));
2883 /* At this point, opt_len and static_len are the total bit lengths of
2884 * the compressed block data, excluding the tree representations.
2885 */
Eric Andersencc8ed391999-10-05 16:24:54 +00002886
Erik Andersene49d5ec2000-02-08 19:58:47 +00002887 /* Build the bit length tree for the above two trees, and get the index
2888 * in bl_order of the last bit length code to send.
2889 */
2890 max_blindex = build_bl_tree();
Eric Andersencc8ed391999-10-05 16:24:54 +00002891
Erik Andersene49d5ec2000-02-08 19:58:47 +00002892 /* Determine the best encoding. Compute first the block length in bytes */
2893 opt_lenb = (opt_len + 3 + 7) >> 3;
2894 static_lenb = (static_len + 3 + 7) >> 3;
2895 input_len += stored_len; /* for debugging only */
Eric Andersencc8ed391999-10-05 16:24:54 +00002896
Erik Andersene49d5ec2000-02-08 19:58:47 +00002897 Trace(
2898 (stderr,
2899 "\nopt %lu(%lu) stat %lu(%lu) stored %lu lit %u dist %u ",
2900 opt_lenb, opt_len, static_lenb, static_len, stored_len,
2901 last_lit, last_dist));
Eric Andersencc8ed391999-10-05 16:24:54 +00002902
Erik Andersene49d5ec2000-02-08 19:58:47 +00002903 if (static_lenb <= opt_lenb)
2904 opt_lenb = static_lenb;
Eric Andersencc8ed391999-10-05 16:24:54 +00002905
Erik Andersene49d5ec2000-02-08 19:58:47 +00002906 /* If compression failed and this is the first and last block,
2907 * and if the zip file can be seeked (to rewrite the local header),
2908 * the whole file is transformed into a stored file:
2909 */
Eric Andersencc8ed391999-10-05 16:24:54 +00002910#ifdef FORCE_METHOD
2911#else
Erik Andersene49d5ec2000-02-08 19:58:47 +00002912 if (stored_len <= opt_lenb && eof && compressed_len == 0L
2913 && seekable()) {
Eric Andersencc8ed391999-10-05 16:24:54 +00002914#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +00002915 /* Since LIT_BUFSIZE <= 2*WSIZE, the input data must be there: */
2916 if (buf == (char *) 0)
Erik Andersen9ffdaa62000-02-11 21:55:04 +00002917 errorMsg("block vanished");
Eric Andersencc8ed391999-10-05 16:24:54 +00002918
Erik Andersene49d5ec2000-02-08 19:58:47 +00002919 copy_block(buf, (unsigned) stored_len, 0); /* without header */
2920 compressed_len = stored_len << 3;
2921 *file_method = STORED;
Eric Andersencc8ed391999-10-05 16:24:54 +00002922
2923#ifdef FORCE_METHOD
2924#else
Erik Andersene49d5ec2000-02-08 19:58:47 +00002925 } else if (stored_len + 4 <= opt_lenb && buf != (char *) 0) {
2926 /* 4: two words for the lengths */
Eric Andersencc8ed391999-10-05 16:24:54 +00002927#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +00002928 /* The test buf != NULL is only necessary if LIT_BUFSIZE > WSIZE.
2929 * Otherwise we can't have processed more than WSIZE input bytes since
2930 * the last block flush, because compression would have been
2931 * successful. If LIT_BUFSIZE <= WSIZE, it is never too late to
2932 * transform a block into a stored block.
2933 */
2934 send_bits((STORED_BLOCK << 1) + eof, 3); /* send block type */
2935 compressed_len = (compressed_len + 3 + 7) & ~7L;
2936 compressed_len += (stored_len + 4) << 3;
Eric Andersencc8ed391999-10-05 16:24:54 +00002937
Erik Andersene49d5ec2000-02-08 19:58:47 +00002938 copy_block(buf, (unsigned) stored_len, 1); /* with header */
Eric Andersencc8ed391999-10-05 16:24:54 +00002939
2940#ifdef FORCE_METHOD
2941#else
Erik Andersene49d5ec2000-02-08 19:58:47 +00002942 } else if (static_lenb == opt_lenb) {
Eric Andersencc8ed391999-10-05 16:24:54 +00002943#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +00002944 send_bits((STATIC_TREES << 1) + eof, 3);
2945 compress_block((ct_data near *) static_ltree,
2946 (ct_data near *) static_dtree);
2947 compressed_len += 3 + static_len;
2948 } else {
2949 send_bits((DYN_TREES << 1) + eof, 3);
2950 send_all_trees(l_desc.max_code + 1, d_desc.max_code + 1,
2951 max_blindex + 1);
2952 compress_block((ct_data near *) dyn_ltree,
2953 (ct_data near *) dyn_dtree);
2954 compressed_len += 3 + opt_len;
2955 }
2956 Assert(compressed_len == bits_sent, "bad compressed size");
2957 init_block();
Eric Andersencc8ed391999-10-05 16:24:54 +00002958
Erik Andersene49d5ec2000-02-08 19:58:47 +00002959 if (eof) {
2960 Assert(input_len == isize, "bad input size");
2961 bi_windup();
2962 compressed_len += 7; /* align on byte boundary */
2963 }
2964 Tracev((stderr, "\ncomprlen %lu(%lu) ", compressed_len >> 3,
2965 compressed_len - 7 * eof));
Eric Andersencc8ed391999-10-05 16:24:54 +00002966
Erik Andersene49d5ec2000-02-08 19:58:47 +00002967 return compressed_len >> 3;
Eric Andersencc8ed391999-10-05 16:24:54 +00002968}
2969
2970/* ===========================================================================
2971 * Save the match info and tally the frequency counts. Return true if
2972 * the current block must be flushed.
2973 */
Erik Andersene49d5ec2000-02-08 19:58:47 +00002974int ct_tally(dist, lc)
2975int dist; /* distance of matched string */
2976int lc; /* match length-MIN_MATCH or unmatched char (if dist==0) */
Eric Andersencc8ed391999-10-05 16:24:54 +00002977{
Erik Andersene49d5ec2000-02-08 19:58:47 +00002978 l_buf[last_lit++] = (uch) lc;
2979 if (dist == 0) {
2980 /* lc is the unmatched char */
2981 dyn_ltree[lc].Freq++;
2982 } else {
2983 /* Here, lc is the match length - MIN_MATCH */
2984 dist--; /* dist = match distance - 1 */
2985 Assert((ush) dist < (ush) MAX_DIST &&
2986 (ush) lc <= (ush) (MAX_MATCH - MIN_MATCH) &&
2987 (ush) d_code(dist) < (ush) D_CODES, "ct_tally: bad match");
Eric Andersencc8ed391999-10-05 16:24:54 +00002988
Erik Andersene49d5ec2000-02-08 19:58:47 +00002989 dyn_ltree[length_code[lc] + LITERALS + 1].Freq++;
2990 dyn_dtree[d_code(dist)].Freq++;
Eric Andersencc8ed391999-10-05 16:24:54 +00002991
Erik Andersene49d5ec2000-02-08 19:58:47 +00002992 d_buf[last_dist++] = (ush) dist;
2993 flags |= flag_bit;
2994 }
2995 flag_bit <<= 1;
Eric Andersencc8ed391999-10-05 16:24:54 +00002996
Erik Andersene49d5ec2000-02-08 19:58:47 +00002997 /* Output the flags if they fill a byte: */
2998 if ((last_lit & 7) == 0) {
2999 flag_buf[last_flags++] = flags;
3000 flags = 0, flag_bit = 1;
3001 }
3002 /* Try to guess if it is profitable to stop the current block here */
3003 if ((last_lit & 0xfff) == 0) {
3004 /* Compute an upper bound for the compressed length */
3005 ulg out_length = (ulg) last_lit * 8L;
3006 ulg in_length = (ulg) strstart - block_start;
3007 int dcode;
3008
3009 for (dcode = 0; dcode < D_CODES; dcode++) {
3010 out_length +=
3011 (ulg) dyn_dtree[dcode].Freq * (5L + extra_dbits[dcode]);
3012 }
3013 out_length >>= 3;
3014 Trace(
3015 (stderr,
3016 "\nlast_lit %u, last_dist %u, in %ld, out ~%ld(%ld%%) ",
3017 last_lit, last_dist, in_length, out_length,
3018 100L - out_length * 100L / in_length));
3019 if (last_dist < last_lit / 2 && out_length < in_length / 2)
3020 return 1;
3021 }
3022 return (last_lit == LIT_BUFSIZE - 1 || last_dist == DIST_BUFSIZE);
3023 /* We avoid equality with LIT_BUFSIZE because of wraparound at 64K
3024 * on 16 bit machines and because stored blocks are restricted to
3025 * 64K-1 bytes.
3026 */
Eric Andersencc8ed391999-10-05 16:24:54 +00003027}
3028
3029/* ===========================================================================
3030 * Send the block data compressed using the given Huffman trees
3031 */
3032local void compress_block(ltree, dtree)
Erik Andersene49d5ec2000-02-08 19:58:47 +00003033ct_data near *ltree; /* literal tree */
3034ct_data near *dtree; /* distance tree */
Eric Andersencc8ed391999-10-05 16:24:54 +00003035{
Erik Andersene49d5ec2000-02-08 19:58:47 +00003036 unsigned dist; /* distance of matched string */
3037 int lc; /* match length or unmatched char (if dist == 0) */
3038 unsigned lx = 0; /* running index in l_buf */
3039 unsigned dx = 0; /* running index in d_buf */
3040 unsigned fx = 0; /* running index in flag_buf */
3041 uch flag = 0; /* current flags */
3042 unsigned code; /* the code to send */
3043 int extra; /* number of extra bits to send */
Eric Andersencc8ed391999-10-05 16:24:54 +00003044
Erik Andersene49d5ec2000-02-08 19:58:47 +00003045 if (last_lit != 0)
3046 do {
3047 if ((lx & 7) == 0)
3048 flag = flag_buf[fx++];
3049 lc = l_buf[lx++];
3050 if ((flag & 1) == 0) {
3051 send_code(lc, ltree); /* send a literal byte */
3052 Tracecv(isgraph(lc), (stderr, " '%c' ", lc));
3053 } else {
3054 /* Here, lc is the match length - MIN_MATCH */
3055 code = length_code[lc];
3056 send_code(code + LITERALS + 1, ltree); /* send the length code */
3057 extra = extra_lbits[code];
3058 if (extra != 0) {
3059 lc -= base_length[code];
3060 send_bits(lc, extra); /* send the extra length bits */
3061 }
3062 dist = d_buf[dx++];
3063 /* Here, dist is the match distance - 1 */
3064 code = d_code(dist);
3065 Assert(code < D_CODES, "bad d_code");
Eric Andersencc8ed391999-10-05 16:24:54 +00003066
Erik Andersene49d5ec2000-02-08 19:58:47 +00003067 send_code(code, dtree); /* send the distance code */
3068 extra = extra_dbits[code];
3069 if (extra != 0) {
3070 dist -= base_dist[code];
3071 send_bits(dist, extra); /* send the extra distance bits */
3072 }
3073 } /* literal or match pair ? */
3074 flag >>= 1;
3075 } while (lx < last_lit);
Eric Andersencc8ed391999-10-05 16:24:54 +00003076
Erik Andersene49d5ec2000-02-08 19:58:47 +00003077 send_code(END_BLOCK, ltree);
Eric Andersencc8ed391999-10-05 16:24:54 +00003078}
3079
3080/* ===========================================================================
3081 * Set the file type to ASCII or BINARY, using a crude approximation:
3082 * binary if more than 20% of the bytes are <= 6 or >= 128, ascii otherwise.
3083 * IN assertion: the fields freq of dyn_ltree are set and the total of all
3084 * frequencies does not exceed 64K (to fit in an int on 16 bit machines).
3085 */
3086local void set_file_type()
3087{
Erik Andersene49d5ec2000-02-08 19:58:47 +00003088 int n = 0;
3089 unsigned ascii_freq = 0;
3090 unsigned bin_freq = 0;
3091
3092 while (n < 7)
3093 bin_freq += dyn_ltree[n++].Freq;
3094 while (n < 128)
3095 ascii_freq += dyn_ltree[n++].Freq;
3096 while (n < LITERALS)
3097 bin_freq += dyn_ltree[n++].Freq;
3098 *file_type = bin_freq > (ascii_freq >> 2) ? BINARY : ASCII;
3099 if (*file_type == BINARY && translate_eol) {
Erik Andersen9ffdaa62000-02-11 21:55:04 +00003100 errorMsg("-l used on binary file", "");
Erik Andersene49d5ec2000-02-08 19:58:47 +00003101 }
Eric Andersencc8ed391999-10-05 16:24:54 +00003102}
Erik Andersene49d5ec2000-02-08 19:58:47 +00003103
Eric Andersencc8ed391999-10-05 16:24:54 +00003104/* util.c -- utility functions for gzip support
3105 * Copyright (C) 1992-1993 Jean-loup Gailly
3106 * This is free software; you can redistribute it and/or modify it under the
3107 * terms of the GNU General Public License, see the file COPYING.
3108 */
3109
Eric Andersencc8ed391999-10-05 16:24:54 +00003110#include <ctype.h>
3111#include <errno.h>
3112#include <sys/types.h>
3113
3114#ifdef HAVE_UNISTD_H
3115# include <unistd.h>
3116#endif
3117#ifndef NO_FCNTL_H
3118# include <fcntl.h>
3119#endif
3120
3121#if defined(STDC_HEADERS) || !defined(NO_STDLIB_H)
3122# include <stdlib.h>
3123#else
Erik Andersene49d5ec2000-02-08 19:58:47 +00003124extern int errno;
Eric Andersencc8ed391999-10-05 16:24:54 +00003125#endif
3126
Eric Andersencc8ed391999-10-05 16:24:54 +00003127/* ===========================================================================
3128 * Copy input to output unchanged: zcat == cat with --force.
3129 * IN assertion: insize bytes have already been read in inbuf.
3130 */
3131int copy(in, out)
Erik Andersene49d5ec2000-02-08 19:58:47 +00003132int in, out; /* input and output file descriptors */
Eric Andersencc8ed391999-10-05 16:24:54 +00003133{
Erik Andersene49d5ec2000-02-08 19:58:47 +00003134 errno = 0;
3135 while (insize != 0 && (int) insize != EOF) {
3136 write_buf(out, (char *) inbuf, insize);
3137 bytes_out += insize;
3138 insize = read(in, (char *) inbuf, INBUFSIZ);
3139 }
3140 if ((int) insize == EOF && errno != 0) {
3141 read_error();
3142 }
3143 bytes_in = bytes_out;
3144 return OK;
Eric Andersencc8ed391999-10-05 16:24:54 +00003145}
3146
3147/* ========================================================================
3148 * Put string s in lower case, return s.
3149 */
3150char *strlwr(s)
Erik Andersene49d5ec2000-02-08 19:58:47 +00003151char *s;
Eric Andersencc8ed391999-10-05 16:24:54 +00003152{
Erik Andersene49d5ec2000-02-08 19:58:47 +00003153 char *t;
3154
3155 for (t = s; *t; t++)
3156 *t = tolow(*t);
3157 return s;
Eric Andersencc8ed391999-10-05 16:24:54 +00003158}
3159
3160#if defined(NO_STRING_H) && !defined(STDC_HEADERS)
3161
3162/* Provide missing strspn and strcspn functions. */
3163
3164# ifndef __STDC__
3165# define const
3166# endif
3167
Erik Andersene49d5ec2000-02-08 19:58:47 +00003168int strspn OF((const char *s, const char *accept));
Eric Andersencc8ed391999-10-05 16:24:54 +00003169int strcspn OF((const char *s, const char *reject));
3170
3171/* ========================================================================
3172 * Return the length of the maximum initial segment
3173 * of s which contains only characters in accept.
3174 */
3175int strspn(s, accept)
Erik Andersene49d5ec2000-02-08 19:58:47 +00003176const char *s;
3177const char *accept;
Eric Andersencc8ed391999-10-05 16:24:54 +00003178{
Erik Andersene49d5ec2000-02-08 19:58:47 +00003179 register const char *p;
3180 register const char *a;
3181 register int count = 0;
Eric Andersencc8ed391999-10-05 16:24:54 +00003182
Erik Andersene49d5ec2000-02-08 19:58:47 +00003183 for (p = s; *p != '\0'; ++p) {
3184 for (a = accept; *a != '\0'; ++a) {
3185 if (*p == *a)
3186 break;
3187 }
3188 if (*a == '\0')
3189 return count;
3190 ++count;
Eric Andersencc8ed391999-10-05 16:24:54 +00003191 }
Erik Andersene49d5ec2000-02-08 19:58:47 +00003192 return count;
Eric Andersencc8ed391999-10-05 16:24:54 +00003193}
3194
3195/* ========================================================================
3196 * Return the length of the maximum inital segment of s
3197 * which contains no characters from reject.
3198 */
3199int strcspn(s, reject)
Erik Andersene49d5ec2000-02-08 19:58:47 +00003200const char *s;
3201const char *reject;
Eric Andersencc8ed391999-10-05 16:24:54 +00003202{
Erik Andersene49d5ec2000-02-08 19:58:47 +00003203 register int count = 0;
Eric Andersencc8ed391999-10-05 16:24:54 +00003204
Erik Andersene49d5ec2000-02-08 19:58:47 +00003205 while (*s != '\0') {
3206 if (strchr(reject, *s++) != NULL)
3207 return count;
3208 ++count;
3209 }
3210 return count;
Eric Andersencc8ed391999-10-05 16:24:54 +00003211}
3212
Erik Andersene49d5ec2000-02-08 19:58:47 +00003213#endif /* NO_STRING_H */
Eric Andersencc8ed391999-10-05 16:24:54 +00003214
3215/* ========================================================================
3216 * Add an environment variable (if any) before argv, and update argc.
3217 * Return the expanded environment variable to be freed later, or NULL
3218 * if no options were added to argv.
3219 */
Erik Andersene49d5ec2000-02-08 19:58:47 +00003220#define SEPARATOR " \t" /* separators in env variable */
Eric Andersencc8ed391999-10-05 16:24:54 +00003221
3222char *add_envopt(argcp, argvp, env)
Erik Andersene49d5ec2000-02-08 19:58:47 +00003223int *argcp; /* pointer to argc */
3224char ***argvp; /* pointer to argv */
3225char *env; /* name of environment variable */
Eric Andersencc8ed391999-10-05 16:24:54 +00003226{
Erik Andersene49d5ec2000-02-08 19:58:47 +00003227 char *p; /* running pointer through env variable */
3228 char **oargv; /* runs through old argv array */
3229 char **nargv; /* runs through new argv array */
3230 int oargc = *argcp; /* old argc */
3231 int nargc = 0; /* number of arguments in env variable */
Eric Andersencc8ed391999-10-05 16:24:54 +00003232
Erik Andersene49d5ec2000-02-08 19:58:47 +00003233 env = (char *) getenv(env);
3234 if (env == NULL)
3235 return NULL;
Eric Andersencc8ed391999-10-05 16:24:54 +00003236
Erik Andersene49d5ec2000-02-08 19:58:47 +00003237 p = (char *) xmalloc(strlen(env) + 1);
3238 env = strcpy(p, env); /* keep env variable intact */
Eric Andersencc8ed391999-10-05 16:24:54 +00003239
Erik Andersene49d5ec2000-02-08 19:58:47 +00003240 for (p = env; *p; nargc++) { /* move through env */
3241 p += strspn(p, SEPARATOR); /* skip leading separators */
3242 if (*p == '\0')
3243 break;
Eric Andersencc8ed391999-10-05 16:24:54 +00003244
Erik Andersene49d5ec2000-02-08 19:58:47 +00003245 p += strcspn(p, SEPARATOR); /* find end of word */
3246 if (*p)
3247 *p++ = '\0'; /* mark it */
3248 }
3249 if (nargc == 0) {
3250 free(env);
3251 return NULL;
3252 }
3253 *argcp += nargc;
3254 /* Allocate the new argv array, with an extra element just in case
3255 * the original arg list did not end with a NULL.
3256 */
3257 nargv = (char **) calloc(*argcp + 1, sizeof(char *));
Eric Andersencc8ed391999-10-05 16:24:54 +00003258
Erik Andersene49d5ec2000-02-08 19:58:47 +00003259 if (nargv == NULL)
Erik Andersen9ffdaa62000-02-11 21:55:04 +00003260 errorMsg("out of memory");
Erik Andersene49d5ec2000-02-08 19:58:47 +00003261 oargv = *argvp;
3262 *argvp = nargv;
Eric Andersencc8ed391999-10-05 16:24:54 +00003263
Erik Andersene49d5ec2000-02-08 19:58:47 +00003264 /* Copy the program name first */
3265 if (oargc-- < 0)
Erik Andersen9ffdaa62000-02-11 21:55:04 +00003266 errorMsg("argc<=0");
Erik Andersene49d5ec2000-02-08 19:58:47 +00003267 *(nargv++) = *(oargv++);
Eric Andersencc8ed391999-10-05 16:24:54 +00003268
Erik Andersene49d5ec2000-02-08 19:58:47 +00003269 /* Then copy the environment args */
3270 for (p = env; nargc > 0; nargc--) {
3271 p += strspn(p, SEPARATOR); /* skip separators */
3272 *(nargv++) = p; /* store start */
3273 while (*p++); /* skip over word */
3274 }
3275
3276 /* Finally copy the old args and add a NULL (usual convention) */
3277 while (oargc--)
3278 *(nargv++) = *(oargv++);
3279 *nargv = NULL;
3280 return env;
Eric Andersencc8ed391999-10-05 16:24:54 +00003281}
Erik Andersene49d5ec2000-02-08 19:58:47 +00003282
Eric Andersencc8ed391999-10-05 16:24:54 +00003283/* ========================================================================
3284 * Display compression ratio on the given stream on 6 characters.
3285 */
3286void display_ratio(num, den, file)
Erik Andersene49d5ec2000-02-08 19:58:47 +00003287long num;
3288long den;
3289FILE *file;
Eric Andersencc8ed391999-10-05 16:24:54 +00003290{
Erik Andersene49d5ec2000-02-08 19:58:47 +00003291 long ratio; /* 1000 times the compression ratio */
Eric Andersencc8ed391999-10-05 16:24:54 +00003292
Erik Andersene49d5ec2000-02-08 19:58:47 +00003293 if (den == 0) {
3294 ratio = 0; /* no compression */
3295 } else if (den < 2147483L) { /* (2**31 -1)/1000 */
3296 ratio = 1000L * num / den;
3297 } else {
3298 ratio = num / (den / 1000L);
3299 }
3300 if (ratio < 0) {
3301 putc('-', file);
3302 ratio = -ratio;
3303 } else {
3304 putc(' ', file);
3305 }
3306 fprintf(file, "%2ld.%1ld%%", ratio / 10L, ratio % 10L);
Eric Andersencc8ed391999-10-05 16:24:54 +00003307}
3308
3309
3310/* zip.c -- compress files to the gzip or pkzip format
3311 * Copyright (C) 1992-1993 Jean-loup Gailly
3312 * This is free software; you can redistribute it and/or modify it under the
3313 * terms of the GNU General Public License, see the file COPYING.
3314 */
3315
Eric Andersencc8ed391999-10-05 16:24:54 +00003316#include <ctype.h>
3317#include <sys/types.h>
3318
3319#ifdef HAVE_UNISTD_H
3320# include <unistd.h>
3321#endif
3322#ifndef NO_FCNTL_H
3323# include <fcntl.h>
3324#endif
3325
Erik Andersene49d5ec2000-02-08 19:58:47 +00003326local ulg crc; /* crc on uncompressed file data */
3327long header_bytes; /* number of bytes in gzip header */
Eric Andersencc8ed391999-10-05 16:24:54 +00003328
3329/* ===========================================================================
3330 * Deflate in to out.
3331 * IN assertions: the input and output buffers are cleared.
3332 * The variables time_stamp and save_orig_name are initialized.
3333 */
3334int zip(in, out)
Erik Andersene49d5ec2000-02-08 19:58:47 +00003335int in, out; /* input and output file descriptors */
Eric Andersencc8ed391999-10-05 16:24:54 +00003336{
Erik Andersene49d5ec2000-02-08 19:58:47 +00003337 uch flags = 0; /* general purpose bit flags */
3338 ush attr = 0; /* ascii/binary flag */
3339 ush deflate_flags = 0; /* pkzip -es, -en or -ex equivalent */
Eric Andersencc8ed391999-10-05 16:24:54 +00003340
Erik Andersene49d5ec2000-02-08 19:58:47 +00003341 ifd = in;
3342 ofd = out;
3343 outcnt = 0;
Eric Andersencc8ed391999-10-05 16:24:54 +00003344
Erik Andersene49d5ec2000-02-08 19:58:47 +00003345 /* Write the header to the gzip file. See algorithm.doc for the format */
Eric Andersencc8ed391999-10-05 16:24:54 +00003346
Eric Andersen96bcfd31999-11-12 01:30:18 +00003347
Erik Andersene49d5ec2000-02-08 19:58:47 +00003348 method = DEFLATED;
3349 put_byte(GZIP_MAGIC[0]); /* magic header */
3350 put_byte(GZIP_MAGIC[1]);
3351 put_byte(DEFLATED); /* compression method */
Eric Andersencc8ed391999-10-05 16:24:54 +00003352
Erik Andersene49d5ec2000-02-08 19:58:47 +00003353 put_byte(flags); /* general flags */
3354 put_long(time_stamp);
Eric Andersencc8ed391999-10-05 16:24:54 +00003355
Erik Andersene49d5ec2000-02-08 19:58:47 +00003356 /* Write deflated file to zip file */
3357 crc = updcrc(0, 0);
Eric Andersencc8ed391999-10-05 16:24:54 +00003358
Erik Andersene49d5ec2000-02-08 19:58:47 +00003359 bi_init(out);
3360 ct_init(&attr, &method);
3361 lm_init(&deflate_flags);
Eric Andersencc8ed391999-10-05 16:24:54 +00003362
Erik Andersene49d5ec2000-02-08 19:58:47 +00003363 put_byte((uch) deflate_flags); /* extra flags */
3364 put_byte(OS_CODE); /* OS identifier */
Eric Andersencc8ed391999-10-05 16:24:54 +00003365
Erik Andersene49d5ec2000-02-08 19:58:47 +00003366 header_bytes = (long) outcnt;
Eric Andersencc8ed391999-10-05 16:24:54 +00003367
Erik Andersene49d5ec2000-02-08 19:58:47 +00003368 (void) deflate();
Eric Andersencc8ed391999-10-05 16:24:54 +00003369
Erik Andersene49d5ec2000-02-08 19:58:47 +00003370 /* Write the crc and uncompressed size */
3371 put_long(crc);
3372 put_long(isize);
3373 header_bytes += 2 * sizeof(long);
Eric Andersencc8ed391999-10-05 16:24:54 +00003374
Erik Andersene49d5ec2000-02-08 19:58:47 +00003375 flush_outbuf();
3376 return OK;
Eric Andersencc8ed391999-10-05 16:24:54 +00003377}
3378
3379
3380/* ===========================================================================
3381 * Read a new buffer from the current input file, perform end-of-line
3382 * translation, and update the crc and input file size.
3383 * IN assertion: size >= 2 (for end-of-line translation)
3384 */
3385int file_read(buf, size)
Erik Andersene49d5ec2000-02-08 19:58:47 +00003386char *buf;
3387unsigned size;
Eric Andersencc8ed391999-10-05 16:24:54 +00003388{
Erik Andersene49d5ec2000-02-08 19:58:47 +00003389 unsigned len;
Eric Andersencc8ed391999-10-05 16:24:54 +00003390
Erik Andersene49d5ec2000-02-08 19:58:47 +00003391 Assert(insize == 0, "inbuf not empty");
Eric Andersencc8ed391999-10-05 16:24:54 +00003392
Erik Andersene49d5ec2000-02-08 19:58:47 +00003393 len = read(ifd, buf, size);
3394 if (len == (unsigned) (-1) || len == 0)
3395 return (int) len;
Eric Andersencc8ed391999-10-05 16:24:54 +00003396
Erik Andersene49d5ec2000-02-08 19:58:47 +00003397 crc = updcrc((uch *) buf, len);
3398 isize += (ulg) len;
3399 return (int) len;
Eric Andersencc8ed391999-10-05 16:24:54 +00003400}
3401#endif