zlib.c revision 245102
1168404Spjd/*
2168404Spjd * This file is derived from various .h and .c files from the zlib-1.0.4
3168404Spjd * distribution by Jean-loup Gailly and Mark Adler, with some additions
4168404Spjd * by Paul Mackerras to aid in implementing Deflate compression and
5168404Spjd * decompression for PPP packets.  See zlib.h for conditions of
6168404Spjd * distribution and use.
7168404Spjd *
8168404Spjd * Changes that have been made include:
9168404Spjd * - added Z_PACKET_FLUSH (see zlib.h for details)
10168404Spjd * - added inflateIncomp and deflateOutputPending
11168404Spjd * - allow strm->next_out to be NULL, meaning discard the output
12168404Spjd *
13168404Spjd * $FreeBSD: head/sys/net/zlib.c 245102 2013-01-06 14:59:59Z peter $
14168404Spjd */
15168404Spjd
16168404Spjd/*
17168404Spjd *  ==FILEVERSION 971210==
18168404Spjd *
19168404Spjd * This marker is used by the Linux installation script to determine
20168404Spjd * whether an up-to-date version of this file is already installed.
21168404Spjd */
22219089Spjd
23168404Spjd#define NO_DUMMY_DECL
24168404Spjd#define NO_ZCFUNCS
25169195Spjd#define MY_ZCALLOC
26169195Spjd
27168404Spjd#if defined(__FreeBSD__) && defined(_KERNEL)
28168404Spjd#define	_tr_init		_zlib104_tr_init
29168404Spjd#define	_tr_align		_zlib104_tr_align
30168404Spjd#define	_tr_tally		_zlib104_tr_tally
31168404Spjd#define	_tr_flush_block		_zlib104_tr_flush_block
32168404Spjd#define	_tr_stored_block	_zlib104_tr_stored_block
33168404Spjd#define	inflate_fast		_zlib104_inflate_fast
34168404Spjd#define	inflate			_zlib104_inflate
35185029Spjd#define	zlibVersion		_zlib104_Version
36185029Spjd#endif
37168404Spjd
38168404Spjd
39168404Spjd/* +++ zutil.h */
40168404Spjd/*-
41168404Spjd * zutil.h -- internal interface and configuration of the compression library
42168404Spjd * Copyright (C) 1995-1996 Jean-loup Gailly.
43168404Spjd * For conditions of distribution and use, see copyright notice in zlib.h
44168404Spjd */
45168404Spjd
46168404Spjd/* WARNING: this file should *not* be used by applications. It is
47168404Spjd   part of the implementation of the compression library and is
48185029Spjd   subject to change. Applications should only use zlib.h.
49219089Spjd */
50168404Spjd
51185029Spjd/* From: zutil.h,v 1.16 1996/07/24 13:41:13 me Exp $ */
52168404Spjd
53168404Spjd#ifndef _Z_UTIL_H
54168404Spjd#define _Z_UTIL_H
55168404Spjd
56168404Spjd#ifdef _KERNEL
57168404Spjd#include <net/zlib.h>
58168404Spjd#else
59219089Spjd#include "zlib.h"
60219089Spjd#endif
61219089Spjd
62168404Spjd#ifdef _KERNEL
63168404Spjd/* Assume this is a *BSD or SVR4 kernel */
64185029Spjd#include <sys/types.h>
65219089Spjd#include <sys/time.h>
66185029Spjd#include <sys/systm.h>
67173268Slulf#include <sys/param.h>
68173268Slulf#include <sys/kernel.h>
69173268Slulf#include <sys/module.h>
70173268Slulf#  define HAVE_MEMCPY
71168404Spjd
72185029Spjd#else
73185029Spjd#if defined(__KERNEL__)
74185029Spjd/* Assume this is a Linux kernel */
75185029Spjd#include <linux/string.h>
76185029Spjd#define HAVE_MEMCPY
77185029Spjd
78185029Spjd#else /* not kernel */
79185029Spjd
80185029Spjd#if defined(MSDOS)||defined(VMS)||defined(CRAY)||defined(WIN32)||defined(RISCOS)
81185029Spjd#   include <stddef.h>
82185029Spjd#   include <errno.h>
83185029Spjd#else
84185029Spjd    extern int errno;
85185029Spjd#endif
86168404Spjd#ifdef STDC
87168404Spjd#  include <string.h>
88168404Spjd#  include <stdlib.h>
89168404Spjd#endif
90168404Spjd#endif /* __KERNEL__ */
91210470Smm#endif /* _KERNEL */
92210470Smm
93210470Smm#ifndef local
94210470Smm#  define local static
95210470Smm#endif
96210470Smm/* compile with -Dlocal if your debugger can't find static symbols */
97185029Spjd
98168404Spjdtypedef unsigned char  uch;
99168404Spjdtypedef uch FAR uchf;
100168404Spjdtypedef unsigned short ush;
101185029Spjdtypedef ush FAR ushf;
102168404Spjdtypedef unsigned long  ulg;
103185029Spjd
104185029Spjdstatic const char *z_errmsg[10]; /* indexed by 2-zlib_error */
105185029Spjd/* (size given to avoid silly warnings with Visual C++) */
106185029Spjd
107185029Spjd#define ERR_MSG(err) z_errmsg[Z_NEED_DICT-(err)]
108168404Spjd
109168404Spjd#define ERR_RETURN(strm,err) \
110168404Spjd  return (strm->msg = (const char*)ERR_MSG(err), (err))
111168404Spjd/* To be used only when the state is known to be valid */
112209962Smm
113168404Spjd        /* common constants */
114168404Spjd
115168404Spjd#ifndef DEF_WBITS
116168404Spjd#  define DEF_WBITS MAX_WBITS
117168404Spjd#endif
118168404Spjd/* default windowBits for decompression. MAX_WBITS is for compression only */
119168404Spjd
120219089Spjd#if MAX_MEM_LEVEL >= 8
121168404Spjd#  define DEF_MEM_LEVEL 8
122185029Spjd#else
123168404Spjd#  define DEF_MEM_LEVEL  MAX_MEM_LEVEL
124168404Spjd#endif
125169196Spjd/* default memLevel */
126185029Spjd
127168404Spjd#define STORED_BLOCK 0
128168404Spjd#define STATIC_TREES 1
129185029Spjd#define DYN_TREES    2
130185029Spjd/* The three kinds of block type */
131185029Spjd
132199156Spjd#define MIN_MATCH  3
133199156Spjd#define MAX_MATCH  258
134199156Spjd/* The minimum and maximum match lengths */
135199156Spjd
136199156Spjd#define PRESET_DICT 0x20 /* preset dictionary flag in zlib header */
137199156Spjd
138199156Spjd        /* target dependencies */
139199156Spjd
140199156Spjd#ifdef MSDOS
141219089Spjd#  define OS_CODE  0x00
142199156Spjd#  ifdef __TURBOC__
143199156Spjd#    include <alloc.h>
144199156Spjd#  else /* MSC or DJGPP */
145185029Spjd#    include <malloc.h>
146185029Spjd#  endif
147185029Spjd#endif
148168404Spjd
149168404Spjd#ifdef OS2
150168404Spjd#  define OS_CODE  0x06
151168404Spjd#endif
152168404Spjd
153168404Spjd#ifdef WIN32 /* Window 95 & Windows NT */
154168404Spjd#  define OS_CODE  0x0b
155168404Spjd#endif
156168404Spjd
157185029Spjd#if defined(VAXC) || defined(VMS)
158211932Smm#  define OS_CODE  0x02
159219089Spjd#  define FOPEN(name, mode) \
160168404Spjd     fopen((name), (mode), "mbc=60", "ctx=stm", "rfm=fix", "mrs=512")
161168404Spjd#endif
162168404Spjd
163168404Spjd#ifdef AMIGA
164168404Spjd#  define OS_CODE  0x01
165185029Spjd#endif
166168404Spjd
167168404Spjd#if defined(ATARI) || defined(atarist)
168168404Spjd#  define OS_CODE  0x05
169185029Spjd#endif
170185029Spjd
171185029Spjd#ifdef MACOS
172185029Spjd#  define OS_CODE  0x07
173168404Spjd#endif
174168404Spjd
175168404Spjd#ifdef __50SERIES /* Prime/PRIMOS */
176168404Spjd#  define OS_CODE  0x0F
177185029Spjd#endif
178168404Spjd
179168404Spjd#ifdef TOPS20
180185029Spjd#  define OS_CODE  0x0a
181211932Smm#endif
182168404Spjd
183168404Spjd#if defined(_BEOS_) || defined(RISCOS)
184185029Spjd#  define fdopen(fd,mode) NULL /* No fdopen() */
185185029Spjd#endif
186185029Spjd
187210470Smm        /* Common defaults */
188185029Spjd
189210470Smm#ifndef OS_CODE
190185029Spjd#  define OS_CODE  0x03  /* assume Unix */
191185029Spjd#endif
192185029Spjd
193185029Spjd#ifndef FOPEN
194185029Spjd#  define FOPEN(name, mode) fopen((name), (mode))
195185029Spjd#endif
196219089Spjd
197185029Spjd         /* functions */
198185029Spjd
199185029Spjd#ifdef HAVE_STRERROR
200185029Spjd   extern char *strerror OF((int));
201185029Spjd#  define zstrerror(errnum) strerror(errnum)
202185029Spjd#else
203185029Spjd#  define zstrerror(errnum) ""
204185029Spjd#endif
205185029Spjd
206185029Spjd#if defined(pyr)
207185029Spjd#  define NO_MEMCPY
208185029Spjd#endif
209185029Spjd#if (defined(M_I86SM) || defined(M_I86MM)) && !defined(_MSC_VER)
210185029Spjd /* Use our own functions for small and medium model with MSC <= 5.0.
211185029Spjd  * You may have to use the same strategy for Borland C (untested).
212185029Spjd  */
213185029Spjd#  define NO_MEMCPY
214185029Spjd#endif
215185029Spjd#if defined(STDC) && !defined(HAVE_MEMCPY) && !defined(NO_MEMCPY)
216185029Spjd#  define HAVE_MEMCPY
217185029Spjd#endif
218185029Spjd#ifdef HAVE_MEMCPY
219185029Spjd#  ifdef SMALL_MEDIUM /* MSDOS small or medium model */
220185029Spjd#    define zmemcpy _fmemcpy
221185029Spjd#    define zmemcmp _fmemcmp
222185029Spjd#    define zmemzero(dest, len) _fmemset(dest, 0, len)
223219089Spjd#  else
224219089Spjd#    define zmemcpy memcpy
225219089Spjd#    define zmemcmp memcmp
226219089Spjd#    define zmemzero(dest, len) memset(dest, 0, len)
227219089Spjd#  endif
228219089Spjd#else
229219089Spjd   extern void zmemcpy  OF((Bytef* dest, Bytef* source, uInt len));
230219089Spjd   extern int  zmemcmp  OF((Bytef* s1,   Bytef* s2, uInt len));
231219089Spjd   extern void zmemzero OF((Bytef* dest, uInt len));
232185029Spjd#endif
233211932Smm
234211932Smm/* Diagnostic functions */
235211932Smm#ifdef DEBUG_ZLIB
236211932Smm#  include <stdio.h>
237211932Smm#  ifndef verbose
238211932Smm#    define verbose 0
239211932Smm#  endif
240211932Smm   extern void z_error    OF((char *m));
241211932Smm#  define Assert(cond,msg) {if(!(cond)) z_error(msg);}
242219089Spjd#  define Trace(x) fprintf x
243185029Spjd#  define Tracev(x) {if (verbose) fprintf x ;}
244185029Spjd#  define Tracevv(x) {if (verbose>1) fprintf x ;}
245185029Spjd#  define Tracec(c,x) {if (verbose && (c)) fprintf x ;}
246185029Spjd#  define Tracecv(c,x) {if (verbose>1 && (c)) fprintf x ;}
247185029Spjd#else
248185029Spjd#  define Assert(cond,msg)
249185029Spjd#  define Trace(x)
250219089Spjd#  define Tracev(x)
251185029Spjd#  define Tracevv(x)
252219089Spjd#  define Tracec(c,x)
253219089Spjd#  define Tracecv(c,x)
254219089Spjd#endif
255219089Spjd
256219089Spjd
257219089Spjdtypedef uLong (*check_func) OF((uLong check, const Bytef *buf, uInt len));
258185029Spjd
259185029Spjdvoidpf zcalloc OF((voidpf opaque, unsigned items, unsigned size));
260185029Spjdvoid   zcfree  OF((voidpf opaque, voidpf ptr));
261185029Spjd
262185029Spjd#define ZALLOC(strm, items, size) \
263185029Spjd           (*((strm)->zalloc))((strm)->opaque, (items), (size))
264185029Spjd#define ZFREE(strm, addr)  (*((strm)->zfree))((strm)->opaque, (voidpf)(addr))
265185029Spjd#define TRY_FREE(s, p) {if (p) ZFREE(s, p);}
266185029Spjd
267185029Spjd#endif /* _Z_UTIL_H */
268185029Spjd/* --- zutil.h */
269185029Spjd
270185029Spjd/* +++ deflate.h */
271185029Spjd/* deflate.h -- internal compression state
272185029Spjd * Copyright (C) 1995-1996 Jean-loup Gailly
273185029Spjd * For conditions of distribution and use, see copyright notice in zlib.h
274185029Spjd */
275185029Spjd
276185029Spjd/* WARNING: this file should *not* be used by applications. It is
277185029Spjd   part of the implementation of the compression library and is
278185029Spjd   subject to change. Applications should only use zlib.h.
279185029Spjd */
280185029Spjd
281185029Spjd/* From: deflate.h,v 1.10 1996/07/02 12:41:00 me Exp $ */
282185029Spjd
283210470Smm#ifndef _DEFLATE_H
284210470Smm#define _DEFLATE_H
285210470Smm
286210470Smm/* #include "zutil.h" */
287185029Spjd
288210470Smm/* ===========================================================================
289210470Smm * Internal compression state.
290210470Smm */
291210470Smm
292210470Smm#define LENGTH_CODES 29
293210470Smm/* number of length codes, not counting the special END_BLOCK code */
294210470Smm
295210470Smm#define LITERALS  256
296210470Smm/* number of literal bytes 0..255 */
297210470Smm
298210470Smm#define L_CODES (LITERALS+1+LENGTH_CODES)
299210470Smm/* number of Literal or Length codes, including the END_BLOCK code */
300210470Smm
301209962Smm#define D_CODES   30
302209962Smm/* number of distance codes */
303209962Smm
304210470Smm#define BL_CODES  19
305185029Spjd/* number of codes used to transfer the bit lengths */
306185029Spjd
307185029Spjd#define HEAP_SIZE (2*L_CODES+1)
308210470Smm/* maximum heap size */
309185029Spjd
310185029Spjd#define MAX_BITS 15
311185029Spjd/* All codes must not exceed MAX_BITS bits */
312185029Spjd
313185029Spjd#define INIT_STATE    42
314185029Spjd#define BUSY_STATE   113
315185029Spjd#define FINISH_STATE 666
316185029Spjd/* Stream status */
317185029Spjd
318210470Smm
319185029Spjd/* Data structure describing a single value and its code string. */
320185029Spjdtypedef struct ct_data_s {
321185029Spjd    union {
322185029Spjd        ush  freq;       /* frequency count */
323185029Spjd        ush  code;       /* bit string */
324185029Spjd    } fc;
325185029Spjd    union {
326185029Spjd        ush  dad;        /* father node in Huffman tree */
327185029Spjd        ush  len;        /* length of bit string */
328185029Spjd    } dl;
329185029Spjd} FAR ct_data;
330185029Spjd
331185029Spjd#define Freq fc.freq
332185029Spjd#define Code fc.code
333185029Spjd#define Dad  dl.dad
334185029Spjd#define Len  dl.len
335185029Spjd
336185029Spjdtypedef struct static_tree_desc_s  static_tree_desc;
337185029Spjd
338185029Spjdtypedef struct tree_desc_s {
339185029Spjd    ct_data *dyn_tree;           /* the dynamic tree */
340185029Spjd    int     max_code;            /* largest code with non zero frequency */
341185029Spjd    static_tree_desc *stat_desc; /* the corresponding static tree */
342185029Spjd} FAR tree_desc;
343185029Spjd
344185029Spjdtypedef ush Pos;
345185029Spjdtypedef Pos FAR Posf;
346185029Spjdtypedef unsigned IPos;
347185029Spjd
348185029Spjd/* A Pos is an index in the character window. We use short instead of int to
349185029Spjd * save space in the various tables. IPos is used only for parameter passing.
350185029Spjd */
351185029Spjd
352185029Spjdtypedef struct deflate_state {
353185029Spjd    z_streamp strm;      /* pointer back to this zlib stream */
354185029Spjd    int   status;        /* as the name implies */
355185029Spjd    Bytef *pending_buf;  /* output still pending */
356185029Spjd    ulg   pending_buf_size; /* size of pending_buf */
357185029Spjd    Bytef *pending_out;  /* next pending byte to output to the stream */
358185029Spjd    int   pending;       /* nb of bytes in the pending buffer */
359185029Spjd    int   noheader;      /* suppress zlib header and adler32 */
360185029Spjd    Byte  data_type;     /* UNKNOWN, BINARY or ASCII */
361185029Spjd    Byte  method;        /* STORED (for zip only) or DEFLATED */
362185029Spjd    int   last_flush;    /* value of flush param for previous deflate call */
363185029Spjd
364185029Spjd                /* used by deflate.c: */
365185029Spjd
366185029Spjd    uInt  w_size;        /* LZ77 window size (32K by default) */
367185029Spjd    uInt  w_bits;        /* log2(w_size)  (8..16) */
368185029Spjd    uInt  w_mask;        /* w_size - 1 */
369168404Spjd
370168404Spjd    Bytef *window;
371168404Spjd    /* Sliding window. Input bytes are read into the second half of the window,
372168404Spjd     * and move to the first half later to keep a dictionary of at least wSize
373168404Spjd     * bytes. With this organization, matches are limited to a distance of
374168404Spjd     * wSize-MAX_MATCH bytes, but this ensures that IO is always
375210470Smm     * performed with a length multiple of the block size. Also, it limits
376168404Spjd     * the window size to 64K, which is quite useful on MSDOS.
377168404Spjd     * To do: use the user input buffer as sliding window.
378168404Spjd     */
379168404Spjd
380185029Spjd    ulg window_size;
381168404Spjd    /* Actual size of window: 2*wSize, except when the user input buffer
382168404Spjd     * is directly used as sliding window.
383168404Spjd     */
384168404Spjd
385168404Spjd    Posf *prev;
386219089Spjd    /* Link to older string with same hash index. To limit the size of this
387168404Spjd     * array to 64K, this link is maintained only for the last 32K strings.
388219089Spjd     * An index in this array is thus a window index modulo 32K.
389219089Spjd     */
390219089Spjd
391219089Spjd    Posf *head; /* Heads of the hash chains or NIL. */
392219089Spjd
393219089Spjd    uInt  ins_h;          /* hash index of string to be inserted */
394168404Spjd    uInt  hash_size;      /* number of elements in hash table */
395168404Spjd    uInt  hash_bits;      /* log2(hash_size) */
396168404Spjd    uInt  hash_mask;      /* hash_size-1 */
397168404Spjd
398168404Spjd    uInt  hash_shift;
399210470Smm    /* Number of bits by which ins_h must be shifted at each input
400168404Spjd     * step. It must be such that after MIN_MATCH steps, the oldest
401168404Spjd     * byte no longer takes part in the hash key, that is:
402219089Spjd     *   hash_shift * MIN_MATCH >= hash_bits
403219089Spjd     */
404219089Spjd
405219089Spjd    long block_start;
406219089Spjd    /* Window position at the beginning of the current output block. Gets
407219089Spjd     * negative when the window is moved backwards.
408219089Spjd     */
409219089Spjd
410219089Spjd    uInt match_length;           /* length of best match */
411219089Spjd    IPos prev_match;             /* previous match */
412219089Spjd    int match_available;         /* set if previous match exists */
413219089Spjd    uInt strstart;               /* start of string to insert */
414219089Spjd    uInt match_start;            /* start of matching string */
415219089Spjd    uInt lookahead;              /* number of valid bytes ahead in window */
416219089Spjd
417219089Spjd    uInt prev_length;
418219089Spjd    /* Length of the best match at previous step. Matches not greater than this
419219089Spjd     * are discarded. This is used in the lazy match evaluation.
420219089Spjd     */
421219089Spjd
422219089Spjd    uInt max_chain_length;
423219089Spjd    /* To speed up deflation, hash chains are never searched beyond this
424219089Spjd     * length.  A higher limit improves compression ratio but degrades the
425219089Spjd     * speed.
426219089Spjd     */
427219089Spjd
428219089Spjd    uInt max_lazy_match;
429219089Spjd    /* Attempt to find a better match only when the current match is strictly
430219089Spjd     * smaller than this value. This mechanism is used only for compression
431219089Spjd     * levels >= 4.
432219089Spjd     */
433219089Spjd#   define max_insert_length  max_lazy_match
434219089Spjd    /* Insert new strings in the hash table only if the match length is not
435219089Spjd     * greater than this length. This saves time but degrades compression.
436219089Spjd     * max_insert_length is used only for compression levels <= 3.
437219089Spjd     */
438219089Spjd
439219089Spjd    int level;    /* compression level (1..9) */
440219089Spjd    int strategy; /* favor or force Huffman coding*/
441219089Spjd
442219089Spjd    uInt good_match;
443219089Spjd    /* Use a faster search when the previous match is longer than this */
444219089Spjd
445219089Spjd    int nice_match; /* Stop searching when current match exceeds this */
446219089Spjd
447219089Spjd                /* used by trees.c: */
448219089Spjd    /* Didn't use ct_data typedef below to supress compiler warning */
449219089Spjd    struct ct_data_s dyn_ltree[HEAP_SIZE];   /* literal and length tree */
450219089Spjd    struct ct_data_s dyn_dtree[2*D_CODES+1]; /* distance tree */
451168404Spjd    struct ct_data_s bl_tree[2*BL_CODES+1];  /* Huffman tree for bit lengths */
452219089Spjd
453219089Spjd    struct tree_desc_s l_desc;               /* desc. for literal tree */
454219089Spjd    struct tree_desc_s d_desc;               /* desc. for distance tree */
455219089Spjd    struct tree_desc_s bl_desc;              /* desc. for bit length tree */
456219089Spjd
457219089Spjd    ush bl_count[MAX_BITS+1];
458219089Spjd    /* number of codes at each bit length for an optimal tree */
459219089Spjd
460219089Spjd    int heap[2*L_CODES+1];      /* heap used to build the Huffman trees */
461219089Spjd    int heap_len;               /* number of elements in the heap */
462219089Spjd    int heap_max;               /* element of largest frequency */
463219089Spjd    /* The sons of heap[n] are heap[2*n] and heap[2*n+1]. heap[0] is not used.
464219089Spjd     * The same heap array is used to build all trees.
465219089Spjd     */
466219089Spjd
467219089Spjd    uch depth[2*L_CODES+1];
468219089Spjd    /* Depth of each subtree used as tie breaker for trees of equal frequency
469219089Spjd     */
470219089Spjd
471219089Spjd    uchf *l_buf;          /* buffer for literals or lengths */
472219089Spjd
473219089Spjd    uInt  lit_bufsize;
474219089Spjd    /* Size of match buffer for literals/lengths.  There are 4 reasons for
475219089Spjd     * limiting lit_bufsize to 64K:
476219089Spjd     *   - frequencies can be kept in 16 bit counters
477219089Spjd     *   - if compression is not successful for the first block, all input
478219089Spjd     *     data is still in the window so we can still emit a stored block even
479219089Spjd     *     when input comes from standard input.  (This can also be done for
480219089Spjd     *     all blocks if lit_bufsize is not greater than 32K.)
481219089Spjd     *   - if compression is not successful for a file smaller than 64K, we can
482219089Spjd     *     even emit a stored file instead of a stored block (saving 5 bytes).
483219089Spjd     *     This is applicable only for zip (not gzip or zlib).
484219089Spjd     *   - creating new Huffman trees less frequently may not provide fast
485219089Spjd     *     adaptation to changes in the input data statistics. (Take for
486219089Spjd     *     example a binary file with poorly compressible code followed by
487219089Spjd     *     a highly compressible string table.) Smaller buffer sizes give
488219089Spjd     *     fast adaptation but have of course the overhead of transmitting
489219089Spjd     *     trees more frequently.
490219089Spjd     *   - I can't count above 4
491219089Spjd     */
492219089Spjd
493219089Spjd    uInt last_lit;      /* running index in l_buf */
494219089Spjd
495219089Spjd    ushf *d_buf;
496219089Spjd    /* Buffer for distances. To simplify the code, d_buf and l_buf have
497209962Smm     * the same number of elements. To use different lengths, an extra flag
498168404Spjd     * array would be necessary.
499209962Smm     */
500209962Smm
501209962Smm    ulg opt_len;        /* bit length of current block with optimal trees */
502209962Smm    ulg static_len;     /* bit length of current block with static trees */
503209962Smm    ulg compressed_len; /* total bit length of compressed file */
504209962Smm    uInt matches;       /* number of string matches in current block */
505168404Spjd    int last_eob_len;   /* bit length of EOB code for last block */
506209962Smm
507209962Smm#ifdef DEBUG_ZLIB
508209962Smm    ulg bits_sent;      /* bit length of the compressed data */
509209962Smm#endif
510209962Smm
511168404Spjd    ush bi_buf;
512209962Smm    /* Output buffer. bits are inserted starting at the bottom (least
513209962Smm     * significant bits).
514219089Spjd     */
515219089Spjd    int bi_valid;
516209962Smm    /* Number of valid bits in bi_buf.  All bits above the last valid bit
517209962Smm     * are always zero.
518209962Smm     */
519219089Spjd
520168404Spjd} FAR deflate_state;
521209962Smm
522209962Smm/* Output a byte on the stream.
523185029Spjd * IN assertion: there is enough room in pending_buf.
524209962Smm */
525209962Smm#define put_byte(s, c) {s->pending_buf[s->pending++] = (c);}
526168404Spjd
527209962Smm
528209962Smm#define MIN_LOOKAHEAD (MAX_MATCH+MIN_MATCH+1)
529219089Spjd/* Minimum amount of lookahead, except at the end of the input file.
530209962Smm * See deflate.c for comments about the MIN_MATCH+1.
531209962Smm */
532209962Smm
533209962Smm#define MAX_DIST(s)  ((s)->w_size-MIN_LOOKAHEAD)
534209962Smm/* In order to simplify the code, particularly on 16 bit machines, match
535168404Spjd * distances are limited to MAX_DIST instead of WSIZE.
536209962Smm */
537209962Smm
538209962Smm        /* in trees.c */
539209962Smmvoid _tr_init         OF((deflate_state *s));
540209962Smmint  _tr_tally        OF((deflate_state *s, unsigned dist, unsigned lc));
541219089Spjdulg  _tr_flush_block  OF((deflate_state *s, charf *buf, ulg stored_len,
542209962Smm			  int eof));
543209962Smmvoid _tr_align        OF((deflate_state *s));
544168404Spjdvoid _tr_stored_block OF((deflate_state *s, charf *buf, ulg stored_len,
545209962Smm                          int eof));
546168404Spjdvoid _tr_stored_type_only OF((deflate_state *));
547168404Spjd
548168404Spjd#endif
549168404Spjd/* --- deflate.h */
550168404Spjd
551168404Spjd/* +++ deflate.c */
552168404Spjd/* deflate.c -- compress data using the deflation algorithm
553168404Spjd * Copyright (C) 1995-1996 Jean-loup Gailly.
554168404Spjd * For conditions of distribution and use, see copyright notice in zlib.h
555168404Spjd */
556168404Spjd
557168404Spjd/*
558168404Spjd *  ALGORITHM
559168404Spjd *
560168404Spjd *      The "deflation" process depends on being able to identify portions
561168404Spjd *      of the input text which are identical to earlier input (within a
562168404Spjd *      sliding window trailing behind the input currently being processed).
563168404Spjd *
564168404Spjd *      The most straightforward technique turns out to be the fastest for
565168404Spjd *      most input files: try all possible matches and select the longest.
566168404Spjd *      The key feature of this algorithm is that insertions into the string
567168404Spjd *      dictionary are very simple and thus fast, and deletions are avoided
568168404Spjd *      completely. Insertions are performed at each input character, whereas
569168404Spjd *      string matches are performed only when the previous match ends. So it
570168404Spjd *      is preferable to spend more time in matches to allow very fast string
571168404Spjd *      insertions and avoid deletions. The matching algorithm for small
572168404Spjd *      strings is inspired from that of Rabin & Karp. A brute force approach
573187830Sed *      is used to find longer strings when a small match has been found.
574168404Spjd *      A similar algorithm is used in comic (by Jan-Mark Wams) and freeze
575168404Spjd *      (by Leonid Broukhis).
576168404Spjd *         A previous version of this file used a more sophisticated algorithm
577168404Spjd *      (by Fiala and Greene) which is guaranteed to run in linear amortized
578168404Spjd *      time, but has a larger average cost, uses more memory and is patented.
579168404Spjd *      However the F&G algorithm may be faster for some highly redundant
580168404Spjd *      files if the parameter max_chain_length (described below) is too large.
581168404Spjd *
582168404Spjd *  ACKNOWLEDGEMENTS
583168404Spjd *
584168404Spjd *      The idea of lazy evaluation of matches is due to Jan-Mark Wams, and
585168958Spjd *      I found it in 'freeze' written by Leonid Broukhis.
586168404Spjd *      Thanks to many people for bug reports and testing.
587168404Spjd *
588185029Spjd *  REFERENCES
589219089Spjd *
590219089Spjd *      Deutsch, L.P.,"DEFLATE Compressed Data Format Specification".
591185029Spjd *      Available in ftp://ds.internic.net/rfc/rfc1951.txt
592185029Spjd *
593185029Spjd *      A description of the Rabin and Karp algorithm is given in the book
594185029Spjd *         "Algorithms" by R. Sedgewick, Addison-Wesley, p252.
595185029Spjd *
596185029Spjd *      Fiala,E.R., and Greene,D.H.
597219089Spjd *         Data Compression with Finite Windows, Comm.ACM, 32,4 (1989) 490-595
598211932Smm *
599219089Spjd */
600219089Spjd
601219089Spjd/* From: deflate.c,v 1.15 1996/07/24 13:40:58 me Exp $ */
602219089Spjd
603219089Spjd/* #include "deflate.h" */
604219089Spjd
605219089Spjdchar deflate_copyright[] = " deflate 1.0.4 Copyright 1995-1996 Jean-loup Gailly ";
606185029Spjd/*
607219089Spjd  If you use the zlib library in a product, an acknowledgment is welcome
608185029Spjd  in the documentation of your product. If for some reason you cannot
609185029Spjd  include such an acknowledgment, I would appreciate that you keep this
610185029Spjd  copyright string in the executable of your product.
611185029Spjd */
612185029Spjd
613185029Spjd/* ===========================================================================
614185029Spjd *  Function prototypes.
615185029Spjd */
616185029Spjdtypedef enum {
617185029Spjd    need_more,      /* block not completed, need more input or more output */
618185029Spjd    block_done,     /* block flush performed */
619185029Spjd    finish_started, /* finish started, need only more output at next deflate */
620185029Spjd    finish_done     /* finish done, accept no more input or output */
621185029Spjd} block_state;
622185029Spjd
623185029Spjdtypedef block_state (*compress_func) OF((deflate_state *s, int flush));
624185029Spjd/* Compression function. Returns the block state after the call. */
625219089Spjd
626219089Spjdlocal void fill_window    OF((deflate_state *s));
627219089Spjdlocal block_state deflate_stored OF((deflate_state *s, int flush));
628185029Spjdlocal block_state deflate_fast   OF((deflate_state *s, int flush));
629185029Spjdlocal block_state deflate_slow   OF((deflate_state *s, int flush));
630168404Spjdlocal void lm_init        OF((deflate_state *s));
631168404Spjdlocal void putShortMSB    OF((deflate_state *s, uInt b));
632168404Spjdlocal void flush_pending  OF((z_streamp strm));
633168404Spjdlocal int read_buf        OF((z_streamp strm, charf *buf, unsigned size));
634168404Spjd#ifdef ASMV
635168404Spjd      void match_init OF((void)); /* asm code initialization */
636168404Spjd      uInt longest_match  OF((deflate_state *s, IPos cur_match));
637168404Spjd#else
638219089Spjdlocal uInt longest_match  OF((deflate_state *s, IPos cur_match));
639219089Spjd#endif
640168404Spjd
641168404Spjd#ifdef DEBUG_ZLIB
642168404Spjdlocal  void check_match OF((deflate_state *s, IPos start, IPos match,
643219089Spjd                            int length));
644219089Spjd#endif
645219089Spjd
646219089Spjd/* ===========================================================================
647168404Spjd * Local data
648168404Spjd */
649185029Spjd
650168404Spjd#define NIL 0
651168404Spjd/* Tail of hash chains */
652185029Spjd
653219089Spjd#ifndef TOO_FAR
654168404Spjd#  define TOO_FAR 4096
655185029Spjd#endif
656185029Spjd/* Matches of length 3 are discarded if their distance exceeds TOO_FAR */
657185029Spjd
658185029Spjd#define MIN_LOOKAHEAD (MAX_MATCH+MIN_MATCH+1)
659219089Spjd/* Minimum amount of lookahead, except at the end of the input file.
660168404Spjd * See deflate.c for comments about the MIN_MATCH+1.
661168404Spjd */
662168404Spjd
663185029Spjd/* Values for max_lazy_match, good_match and max_chain_length, depending on
664168404Spjd * the desired pack level (0..9). The values given below have been tuned to
665168404Spjd * exclude worst case performance for pathological files. Better values may be
666168404Spjd * found for specific files.
667168404Spjd */
668185029Spjdtypedef struct config_s {
669168404Spjd   ush good_length; /* reduce lazy search above this match length */
670219089Spjd   ush max_lazy;    /* do not perform lazy search above this match length */
671185029Spjd   ush nice_length; /* quit search above this match length */
672219089Spjd   ush max_chain;
673219089Spjd   compress_func func;
674219089Spjd} config;
675219089Spjd
676219089Spjdlocal config configuration_table[10] = {
677219089Spjd/*      good lazy nice chain */
678219089Spjd/* 0 */ {0,    0,  0,    0, deflate_stored},  /* store only */
679219089Spjd/* 1 */ {4,    4,  8,    4, deflate_fast}, /* maximum speed, no lazy matches */
680219089Spjd/* 2 */ {4,    5, 16,    8, deflate_fast},
681219089Spjd/* 3 */ {4,    6, 32,   32, deflate_fast},
682219089Spjd
683219089Spjd/* 4 */ {4,    4, 16,   16, deflate_slow},  /* lazy matches */
684219089Spjd/* 5 */ {8,   16, 32,   32, deflate_slow},
685219089Spjd/* 6 */ {8,   16, 128, 128, deflate_slow},
686219089Spjd/* 7 */ {8,   32, 128, 256, deflate_slow},
687185029Spjd/* 8 */ {32, 128, 258, 1024, deflate_slow},
688219089Spjd/* 9 */ {32, 258, 258, 4096, deflate_slow}}; /* maximum compression */
689219089Spjd
690219089Spjd/* Note: the deflate() code requires max_lazy >= MIN_MATCH and max_chain >= 4
691219089Spjd * For deflate_fast() (levels <= 3) good is ignored and lazy has a different
692219089Spjd * meaning.
693219089Spjd */
694168404Spjd
695219089Spjd#define EQUAL 0
696219089Spjd/* result of memcmp for equal strings */
697219089Spjd
698219089Spjd#ifndef NO_DUMMY_DECL
699168404Spjdstruct static_tree_desc_s {int dummy;}; /* for buggy compilers */
700168404Spjd#endif
701168404Spjd
702168404Spjd/* ===========================================================================
703224251Sdelphij * Update a hash value with the given input byte
704224231Sdelphij * IN  assertion: all calls to to UPDATE_HASH are made with consecutive
705224231Sdelphij *    input characters, so that a running hash key can be computed from the
706224231Sdelphij *    previous key instead of complete recalculation each time.
707224231Sdelphij */
708224231Sdelphij#define UPDATE_HASH(s,h,c) (h = (((h)<<s->hash_shift) ^ (c)) & s->hash_mask)
709224231Sdelphij
710224231Sdelphij
711224251Sdelphij/* ===========================================================================
712224231Sdelphij * Insert string str in the dictionary and set match_head to the previous head
713224231Sdelphij * of the hash chain (the most recent string with same hash key). Return
714224251Sdelphij * the previous length of the hash chain.
715168404Spjd * IN  assertion: all calls to to INSERT_STRING are made with consecutive
716168404Spjd *    input characters and the first MIN_MATCH bytes of str are valid
717168404Spjd *    (except for the last MIN_MATCH-1 bytes of the input file).
718219089Spjd */
719219089Spjd#define INSERT_STRING(s, str, match_head) \
720219089Spjd   (UPDATE_HASH(s, s->ins_h, s->window[(str) + (MIN_MATCH-1)]), \
721209962Smm    s->prev[(str) & s->w_mask] = match_head = s->head[s->ins_h], \
722209962Smm    s->head[s->ins_h] = (Pos)(str))
723209962Smm
724168404Spjd/* ===========================================================================
725210172Sjhb * Initialize the hash table (avoiding 64K overflow for 16 bit systems).
726189696Sjhb * prev[] will be initialized on the fly.
727168404Spjd */
728185029Spjd#define CLEAR_HASH(s) \
729185029Spjd    s->head[s->hash_size-1] = NIL; \
730185029Spjd    zmemzero((charf *)s->head, (unsigned)(s->hash_size-1)*sizeof(*s->head));
731168404Spjd
732185029Spjd/* ========================================================================= */
733185029Spjdint deflateInit_(strm, level, version, stream_size)
734168404Spjd    z_streamp strm;
735185029Spjd    int level;
736185029Spjd    const char *version;
737168404Spjd    int stream_size;
738168404Spjd{
739185029Spjd    return deflateInit2_(strm, level, Z_DEFLATED, MAX_WBITS, DEF_MEM_LEVEL,
740168404Spjd			 Z_DEFAULT_STRATEGY, version, stream_size);
741168404Spjd    /* To do: ignore strm->next_in if we use it as window */
742219089Spjd}
743219089Spjd
744219089Spjd/* ========================================================================= */
745168404Spjdint deflateInit2_(strm, level, method, windowBits, memLevel, strategy,
746168404Spjd		  version, stream_size)
747168404Spjd    z_streamp strm;
748168404Spjd    int  level;
749168404Spjd    int  method;
750168404Spjd    int  windowBits;
751168404Spjd    int  memLevel;
752168404Spjd    int  strategy;
753168404Spjd    const char *version;
754168404Spjd    int stream_size;
755185029Spjd{
756185029Spjd    deflate_state *s;
757185029Spjd    int noheader = 0;
758168404Spjd    static char* my_version = ZLIB_VERSION;
759185029Spjd
760168404Spjd    ushf *overlay;
761168404Spjd    /* We overlay pending_buf and d_buf+l_buf. This works since the average
762168404Spjd     * output size for (length,distance) codes is <= 24 bits.
763185029Spjd     */
764219089Spjd
765168404Spjd    if (version == Z_NULL || version[0] != my_version[0] ||
766219089Spjd        stream_size != sizeof(z_stream)) {
767219089Spjd	return Z_VERSION_ERROR;
768219089Spjd    }
769219089Spjd    if (strm == Z_NULL) return Z_STREAM_ERROR;
770219089Spjd
771185029Spjd    strm->msg = Z_NULL;
772168404Spjd#ifndef NO_ZCFUNCS
773185029Spjd    if (strm->zalloc == Z_NULL) {
774168404Spjd	strm->zalloc = zcalloc;
775219089Spjd	strm->opaque = (voidpf)0;
776219089Spjd    }
777219089Spjd    if (strm->zfree == Z_NULL) strm->zfree = zcfree;
778219089Spjd#endif
779219089Spjd
780219089Spjd    if (level == Z_DEFAULT_COMPRESSION) level = 6;
781168404Spjd
782168404Spjd    if (windowBits < 0) { /* undocumented feature: suppress zlib header */
783168404Spjd        noheader = 1;
784209962Smm        windowBits = -windowBits;
785185029Spjd    }
786168404Spjd    if (memLevel < 1 || memLevel > MAX_MEM_LEVEL || method != Z_DEFLATED ||
787168404Spjd        windowBits < 9 || windowBits > 15 || level < 0 || level > 9 ||
788168404Spjd	strategy < 0 || strategy > Z_HUFFMAN_ONLY) {
789185029Spjd        return Z_STREAM_ERROR;
790168404Spjd    }
791168404Spjd    s = (deflate_state *) ZALLOC(strm, 1, sizeof(deflate_state));
792168404Spjd    if (s == Z_NULL) return Z_MEM_ERROR;
793168404Spjd    strm->state = (struct internal_state FAR *)s;
794219089Spjd    s->strm = strm;
795219089Spjd
796219089Spjd    s->noheader = noheader;
797219089Spjd    s->w_bits = windowBits;
798168404Spjd    s->w_size = 1 << s->w_bits;
799168404Spjd    s->w_mask = s->w_size - 1;
800168404Spjd
801168404Spjd    s->hash_bits = memLevel + 7;
802168404Spjd    s->hash_size = 1 << s->hash_bits;
803219089Spjd    s->hash_mask = s->hash_size - 1;
804168404Spjd    s->hash_shift =  ((s->hash_bits+MIN_MATCH-1)/MIN_MATCH);
805168404Spjd
806168404Spjd    s->window = (Bytef *) ZALLOC(strm, s->w_size, 2*sizeof(Byte));
807168404Spjd    s->prev   = (Posf *)  ZALLOC(strm, s->w_size, sizeof(Pos));
808209962Smm    s->head   = (Posf *)  ZALLOC(strm, s->hash_size, sizeof(Pos));
809185029Spjd
810185029Spjd    s->lit_bufsize = 1 << (memLevel + 6); /* 16K elements by default */
811219089Spjd
812168404Spjd    overlay = (ushf *) ZALLOC(strm, s->lit_bufsize, sizeof(ush)+2);
813168404Spjd    s->pending_buf = (uchf *) overlay;
814185029Spjd    s->pending_buf_size = (ulg)s->lit_bufsize * (sizeof(ush)+2L);
815185029Spjd
816219089Spjd    if (s->window == Z_NULL || s->prev == Z_NULL || s->head == Z_NULL ||
817168404Spjd        s->pending_buf == Z_NULL) {
818168404Spjd        strm->msg = (const char*)ERR_MSG(Z_MEM_ERROR);
819209962Smm        deflateEnd (strm);
820185029Spjd        return Z_MEM_ERROR;
821168404Spjd    }
822219089Spjd    s->d_buf = overlay + s->lit_bufsize/sizeof(ush);
823168404Spjd    s->l_buf = s->pending_buf + (1+sizeof(ush))*s->lit_bufsize;
824168404Spjd
825185029Spjd    s->level = level;
826168404Spjd    s->strategy = strategy;
827219089Spjd    s->method = (Byte)method;
828168404Spjd
829168404Spjd    return deflateReset(strm);
830207334Spjd}
831207334Spjd
832219089Spjd/* ========================================================================= */
833168404Spjdint deflateSetDictionary (strm, dictionary, dictLength)
834168404Spjd    z_streamp strm;
835168404Spjd    const Bytef *dictionary;
836168404Spjd    uInt  dictLength;
837168404Spjd{
838168404Spjd    deflate_state *s;
839185029Spjd    uInt length = dictLength;
840219089Spjd    uInt n;
841219089Spjd    IPos hash_head = 0;
842168404Spjd
843168404Spjd    if (strm == Z_NULL || strm->state == Z_NULL || dictionary == Z_NULL)
844168404Spjd	return Z_STREAM_ERROR;
845168404Spjd
846168404Spjd    s = (deflate_state *) strm->state;
847219089Spjd    if (s->status != INIT_STATE) return Z_STREAM_ERROR;
848168404Spjd
849168404Spjd    strm->adler = adler32(strm->adler, dictionary, dictLength);
850168404Spjd
851185029Spjd    if (length < MIN_MATCH) return Z_OK;
852219089Spjd    if (length > MAX_DIST(s)) {
853219089Spjd	length = MAX_DIST(s);
854219089Spjd#ifndef USE_DICT_HEAD
855185029Spjd	dictionary += dictLength - length; /* use the tail of the dictionary */
856168404Spjd#endif
857219089Spjd    }
858219089Spjd    zmemcpy((charf *)s->window, dictionary, length);
859219089Spjd    s->strstart = length;
860219089Spjd    s->block_start = (long)length;
861168404Spjd
862168404Spjd    /* Insert all strings in the hash table (except for the last two bytes).
863219089Spjd     * s->lookahead stays null, so s->ins_h will be recomputed at the next
864219089Spjd     * call of fill_window.
865219089Spjd     */
866219089Spjd    s->ins_h = s->window[0];
867219089Spjd    UPDATE_HASH(s, s->ins_h, s->window[1]);
868219089Spjd    for (n = 0; n <= length - MIN_MATCH; n++) {
869168404Spjd	INSERT_STRING(s, n, hash_head);
870219089Spjd    }
871168404Spjd    if (hash_head) hash_head = 0;  /* to make compiler happy */
872219089Spjd    return Z_OK;
873219089Spjd}
874219089Spjd
875219089Spjd/* ========================================================================= */
876219089Spjdint deflateReset (strm)
877219089Spjd    z_streamp strm;
878168404Spjd{
879219089Spjd    deflate_state *s;
880219089Spjd
881168404Spjd    if (strm == Z_NULL || strm->state == Z_NULL ||
882168404Spjd        strm->zalloc == Z_NULL || strm->zfree == Z_NULL) return Z_STREAM_ERROR;
883219089Spjd
884168404Spjd    strm->total_in = strm->total_out = 0;
885219089Spjd    strm->msg = Z_NULL; /* use zfree if we ever allocate msg dynamically */
886168404Spjd    strm->data_type = Z_UNKNOWN;
887168404Spjd
888168404Spjd    s = (deflate_state *)strm->state;
889219089Spjd    s->pending = 0;
890168404Spjd    s->pending_out = s->pending_buf;
891219089Spjd
892168404Spjd    if (s->noheader < 0) {
893168404Spjd        s->noheader = 0; /* was set to -1 by deflate(..., Z_FINISH); */
894219089Spjd    }
895219089Spjd    s->status = s->noheader ? BUSY_STATE : INIT_STATE;
896219089Spjd    strm->adler = 1;
897219089Spjd    s->last_flush = Z_NO_FLUSH;
898219089Spjd
899219089Spjd    _tr_init(s);
900219089Spjd    lm_init(s);
901219089Spjd
902219089Spjd    return Z_OK;
903219089Spjd}
904219089Spjd
905219089Spjd/* ========================================================================= */
906219089Spjdint deflateParams(strm, level, strategy)
907219089Spjd    z_streamp strm;
908219089Spjd    int level;
909219089Spjd    int strategy;
910219089Spjd{
911219089Spjd    deflate_state *s;
912219089Spjd    compress_func func;
913219089Spjd    int err = Z_OK;
914219089Spjd
915219089Spjd    if (strm == Z_NULL || strm->state == Z_NULL) return Z_STREAM_ERROR;
916219089Spjd    s = (deflate_state *) strm->state;
917219089Spjd
918219089Spjd    if (level == Z_DEFAULT_COMPRESSION) {
919219089Spjd	level = 6;
920219089Spjd    }
921219089Spjd    if (level < 0 || level > 9 || strategy < 0 || strategy > Z_HUFFMAN_ONLY) {
922219089Spjd	return Z_STREAM_ERROR;
923219089Spjd    }
924219089Spjd    func = configuration_table[s->level].func;
925219089Spjd
926219089Spjd    if (func != configuration_table[level].func && strm->total_in != 0) {
927219089Spjd	/* Flush the last buffer: */
928219089Spjd	err = deflate(strm, Z_PARTIAL_FLUSH);
929219089Spjd    }
930219089Spjd    if (s->level != level) {
931219089Spjd	s->level = level;
932219089Spjd	s->max_lazy_match   = configuration_table[level].max_lazy;
933219089Spjd	s->good_match       = configuration_table[level].good_length;
934219089Spjd	s->nice_match       = configuration_table[level].nice_length;
935219089Spjd	s->max_chain_length = configuration_table[level].max_chain;
936219089Spjd    }
937219089Spjd    s->strategy = strategy;
938219089Spjd    return err;
939219089Spjd}
940219089Spjd
941219089Spjd/* =========================================================================
942219089Spjd * Put a short in the pending buffer. The 16-bit value is put in MSB order.
943219089Spjd * IN assertion: the stream state is correct and there is enough room in
944219089Spjd * pending_buf.
945219089Spjd */
946219089Spjdlocal void putShortMSB (s, b)
947219089Spjd    deflate_state *s;
948219089Spjd    uInt b;
949219089Spjd{
950219089Spjd    put_byte(s, (Byte)(b >> 8));
951219089Spjd    put_byte(s, (Byte)(b & 0xff));
952219089Spjd}
953219089Spjd
954219089Spjd/* =========================================================================
955219089Spjd * Flush as much pending output as possible. All deflate() output goes
956219089Spjd * through this function so some applications may wish to modify it
957219089Spjd * to avoid allocating a large strm->next_out buffer and copying into it.
958219089Spjd * (See also read_buf()).
959219089Spjd */
960219089Spjdlocal void flush_pending(strm)
961219089Spjd    z_streamp strm;
962219089Spjd{
963219089Spjd    deflate_state *s = (deflate_state *) strm->state;
964219089Spjd    unsigned len = s->pending;
965219089Spjd
966219089Spjd    if (len > strm->avail_out) len = strm->avail_out;
967219089Spjd    if (len == 0) return;
968219089Spjd
969219089Spjd    if (strm->next_out != Z_NULL) {
970219089Spjd	zmemcpy(strm->next_out, s->pending_out, len);
971219089Spjd	strm->next_out += len;
972219089Spjd    }
973219089Spjd    s->pending_out += len;
974219089Spjd    strm->total_out += len;
975219089Spjd    strm->avail_out  -= len;
976219089Spjd    s->pending -= len;
977219089Spjd    if (s->pending == 0) {
978219089Spjd        s->pending_out = s->pending_buf;
979219089Spjd    }
980219089Spjd}
981219089Spjd
982219089Spjd/* ========================================================================= */
983185029Spjdint deflate (strm, flush)
984219089Spjd    z_streamp strm;
985219089Spjd    int flush;
986185029Spjd{
987185029Spjd    int old_flush; /* value of flush param for previous deflate call */
988185029Spjd    deflate_state *s;
989185029Spjd
990185029Spjd    if (strm == Z_NULL || strm->state == Z_NULL ||
991185029Spjd	flush > Z_FINISH || flush < 0) {
992219089Spjd        return Z_STREAM_ERROR;
993219089Spjd    }
994185029Spjd    s = (deflate_state *) strm->state;
995219089Spjd
996219089Spjd    if ((strm->next_in == Z_NULL && strm->avail_in != 0) ||
997219089Spjd	(s->status == FINISH_STATE && flush != Z_FINISH)) {
998219089Spjd        ERR_RETURN(strm, Z_STREAM_ERROR);
999209962Smm    }
1000219089Spjd    if (strm->avail_out == 0) ERR_RETURN(strm, Z_BUF_ERROR);
1001219089Spjd
1002219089Spjd    s->strm = strm; /* just in case */
1003219089Spjd    old_flush = s->last_flush;
1004219089Spjd    s->last_flush = flush;
1005219089Spjd
1006219089Spjd    /* Write the zlib header */
1007185029Spjd    if (s->status == INIT_STATE) {
1008185029Spjd
1009168404Spjd        uInt header = (Z_DEFLATED + ((s->w_bits-8)<<4)) << 8;
1010185029Spjd        uInt level_flags = (s->level-1) >> 1;
1011185029Spjd
1012185029Spjd        if (level_flags > 3) level_flags = 3;
1013185029Spjd        header |= (level_flags << 6);
1014185029Spjd	if (s->strstart != 0) header |= PRESET_DICT;
1015185029Spjd        header += 31 - (header % 31);
1016219089Spjd
1017185029Spjd        s->status = BUSY_STATE;
1018168404Spjd        putShortMSB(s, header);
1019219089Spjd
1020219089Spjd	/* Save the adler32 of the preset dictionary: */
1021219089Spjd	if (s->strstart != 0) {
1022219089Spjd	    putShortMSB(s, (uInt)(strm->adler >> 16));
1023219089Spjd	    putShortMSB(s, (uInt)(strm->adler & 0xffff));
1024185029Spjd	}
1025219089Spjd	strm->adler = 1L;
1026185029Spjd    }
1027185029Spjd
1028168404Spjd    /* Flush as much pending output as possible */
1029185029Spjd    if (s->pending != 0) {
1030185029Spjd        flush_pending(strm);
1031168404Spjd        if (strm->avail_out == 0) {
1032185029Spjd	    /* Since avail_out is 0, deflate will be called again with
1033219089Spjd	     * more output space, but possibly with both pending and
1034219089Spjd	     * avail_in equal to zero. There won't be anything to do,
1035219089Spjd	     * but this is not an error situation so make sure we
1036219089Spjd	     * return OK instead of BUF_ERROR at next call of deflate:
1037185029Spjd             */
1038168404Spjd	    s->last_flush = -1;
1039185029Spjd	    return Z_OK;
1040219089Spjd	}
1041219089Spjd
1042185029Spjd    /* Make sure there is something to do and avoid duplicate consecutive
1043185029Spjd     * flushes. For repeated and useless calls with Z_FINISH, we keep
1044185029Spjd     * returning Z_STREAM_END instead of Z_BUFF_ERROR.
1045219089Spjd     */
1046219089Spjd    } else if (strm->avail_in == 0 && flush <= old_flush &&
1047185029Spjd	       flush != Z_FINISH) {
1048185029Spjd        ERR_RETURN(strm, Z_BUF_ERROR);
1049185029Spjd    }
1050219089Spjd
1051219089Spjd    /* User must not provide more input after the first FINISH: */
1052185029Spjd    if (s->status == FINISH_STATE && strm->avail_in != 0) {
1053185029Spjd        ERR_RETURN(strm, Z_BUF_ERROR);
1054185029Spjd    }
1055219089Spjd
1056219089Spjd    /* Start a new block or continue the current one.
1057185029Spjd     */
1058185029Spjd    if (strm->avail_in != 0 || s->lookahead != 0 ||
1059185029Spjd        (flush != Z_NO_FLUSH && s->status != FINISH_STATE)) {
1060219089Spjd        block_state bstate;
1061219089Spjd
1062185029Spjd	bstate = (*(configuration_table[s->level].func))(s, flush);
1063185029Spjd
1064185029Spjd        if (bstate == finish_started || bstate == finish_done) {
1065219089Spjd            s->status = FINISH_STATE;
1066219089Spjd        }
1067185029Spjd        if (bstate == need_more || bstate == finish_started) {
1068185029Spjd	    if (strm->avail_out == 0) {
1069185029Spjd	        s->last_flush = -1; /* avoid BUF_ERROR next call, see above */
1070219089Spjd	    }
1071219089Spjd	    return Z_OK;
1072185029Spjd	    /* If flush != Z_NO_FLUSH && avail_out == 0, the next call
1073185029Spjd	     * of deflate should use the same flush parameter to make sure
1074185029Spjd	     * that the flush is complete. So we don't have to output an
1075219089Spjd	     * empty block here, this will be done at next call. This also
1076219089Spjd	     * ensures that for a very small output buffer, we emit at most
1077185029Spjd	     * one empty block.
1078185029Spjd	     */
1079185029Spjd	}
1080219089Spjd        if (bstate == block_done) {
1081219089Spjd            if (flush == Z_PARTIAL_FLUSH) {
1082185029Spjd                _tr_align(s);
1083185029Spjd	    } else if (flush == Z_PACKET_FLUSH) {
1084185029Spjd		/* Output just the 3-bit `stored' block type value,
1085185029Spjd		   but not a zero length. */
1086219089Spjd		_tr_stored_type_only(s);
1087185029Spjd            } else { /* FULL_FLUSH or SYNC_FLUSH */
1088185029Spjd                _tr_stored_block(s, (char*)0, 0L, 0);
1089185029Spjd                /* For a full flush, this empty block will be recognized
1090219089Spjd                 * as a special marker by inflate_sync().
1091219089Spjd                 */
1092185029Spjd                if (flush == Z_FULL_FLUSH) {
1093185029Spjd                    CLEAR_HASH(s);             /* forget history */
1094185029Spjd                }
1095219089Spjd            }
1096185029Spjd            flush_pending(strm);
1097185029Spjd	    if (strm->avail_out == 0) {
1098219089Spjd	      s->last_flush = -1; /* avoid BUF_ERROR at next call, see above */
1099219089Spjd	      return Z_OK;
1100219089Spjd	    }
1101219089Spjd        }
1102219089Spjd    }
1103219089Spjd    Assert(strm->avail_out > 0, "bug2");
1104219089Spjd
1105219089Spjd    if (flush != Z_FINISH) return Z_OK;
1106219089Spjd    if (s->noheader) return Z_STREAM_END;
1107219089Spjd
1108219089Spjd    /* Write the zlib trailer (adler32) */
1109219089Spjd    putShortMSB(s, (uInt)(strm->adler >> 16));
1110219089Spjd    putShortMSB(s, (uInt)(strm->adler & 0xffff));
1111219089Spjd    flush_pending(strm);
1112219089Spjd    /* If avail_out is zero, the application will call deflate again
1113168404Spjd     * to flush the rest.
1114168404Spjd     */
1115168404Spjd    s->noheader = -1; /* write the trailer only once! */
1116168404Spjd    return s->pending != 0 ? Z_OK : Z_STREAM_END;
1117168404Spjd}
1118168404Spjd
1119168404Spjd/* ========================================================================= */
1120168404Spjdint deflateEnd (strm)
1121219089Spjd    z_streamp strm;
1122219089Spjd{
1123219089Spjd    int status;
1124168404Spjd    deflate_state *s;
1125168404Spjd
1126219089Spjd    if (strm == Z_NULL || strm->state == Z_NULL) return Z_STREAM_ERROR;
1127185029Spjd    s = (deflate_state *) strm->state;
1128168404Spjd
1129168404Spjd    status = s->status;
1130219089Spjd    if (status != INIT_STATE && status != BUSY_STATE &&
1131168404Spjd	status != FINISH_STATE) {
1132168404Spjd      return Z_STREAM_ERROR;
1133168404Spjd    }
1134168404Spjd
1135168404Spjd    /* Deallocate in reverse order of allocations: */
1136168404Spjd    TRY_FREE(strm, s->pending_buf);
1137219089Spjd    TRY_FREE(strm, s->head);
1138219089Spjd    TRY_FREE(strm, s->prev);
1139219089Spjd    TRY_FREE(strm, s->window);
1140219089Spjd
1141219089Spjd    ZFREE(strm, s);
1142168404Spjd    strm->state = Z_NULL;
1143168404Spjd
1144168404Spjd    return status == BUSY_STATE ? Z_DATA_ERROR : Z_OK;
1145168404Spjd}
1146219089Spjd
1147219089Spjd/* =========================================================================
1148219089Spjd * Copy the source state to the destination state.
1149168404Spjd */
1150219089Spjdint deflateCopy (dest, source)
1151185029Spjd    z_streamp dest;
1152219089Spjd    z_streamp source;
1153219089Spjd{
1154219089Spjd    deflate_state *ds;
1155185029Spjd    deflate_state *ss;
1156219089Spjd    ushf *overlay;
1157219089Spjd
1158219089Spjd    if (source == Z_NULL || dest == Z_NULL || source->state == Z_NULL)
1159219089Spjd        return Z_STREAM_ERROR;
1160168404Spjd    ss = (deflate_state *) source->state;
1161168404Spjd
1162185029Spjd    zmemcpy(dest, source, sizeof(*dest));
1163168404Spjd
1164219089Spjd    ds = (deflate_state *) ZALLOC(dest, 1, sizeof(deflate_state));
1165197458Spjd    if (ds == Z_NULL) return Z_MEM_ERROR;
1166197458Spjd    dest->state = (struct internal_state FAR *) ds;
1167197458Spjd    zmemcpy(ds, ss, sizeof(*ds));
1168197458Spjd    ds->strm = dest;
1169197458Spjd
1170197458Spjd    ds->window = (Bytef *) ZALLOC(dest, ds->w_size, 2*sizeof(Byte));
1171197458Spjd    ds->prev   = (Posf *)  ZALLOC(dest, ds->w_size, sizeof(Pos));
1172197131Spjd    ds->head   = (Posf *)  ZALLOC(dest, ds->hash_size, sizeof(Pos));
1173197458Spjd    overlay = (ushf *) ZALLOC(dest, ds->lit_bufsize, sizeof(ush)+2);
1174197458Spjd    ds->pending_buf = (uchf *) overlay;
1175197458Spjd
1176197458Spjd    if (ds->window == Z_NULL || ds->prev == Z_NULL || ds->head == Z_NULL ||
1177197458Spjd        ds->pending_buf == Z_NULL) {
1178197458Spjd        deflateEnd (dest);
1179197458Spjd        return Z_MEM_ERROR;
1180197458Spjd    }
1181197458Spjd    /* ??? following zmemcpy doesn't work for 16-bit MSDOS */
1182197131Spjd    zmemcpy(ds->window, ss->window, ds->w_size * 2 * sizeof(Byte));
1183197458Spjd    zmemcpy(ds->prev, ss->prev, ds->w_size * sizeof(Pos));
1184185029Spjd    zmemcpy(ds->head, ss->head, ds->hash_size * sizeof(Pos));
1185185029Spjd    zmemcpy(ds->pending_buf, ss->pending_buf, (uInt)ds->pending_buf_size);
1186185029Spjd
1187185029Spjd    ds->pending_out = ds->pending_buf + (ss->pending_out - ss->pending_buf);
1188185029Spjd    ds->d_buf = overlay + ds->lit_bufsize/sizeof(ush);
1189185029Spjd    ds->l_buf = ds->pending_buf + (1+sizeof(ush))*ds->lit_bufsize;
1190185029Spjd
1191185029Spjd    ds->l_desc.dyn_tree = ds->dyn_ltree;
1192219089Spjd    ds->d_desc.dyn_tree = ds->dyn_dtree;
1193185029Spjd    ds->bl_desc.dyn_tree = ds->bl_tree;
1194185029Spjd
1195197458Spjd    return Z_OK;
1196197458Spjd}
1197185029Spjd
1198185029Spjd/* ===========================================================================
1199185029Spjd * Return the number of bytes of output which are immediately available
1200185029Spjd * for output from the decompressor.
1201185029Spjd */
1202168404Spjdint deflateOutputPending (strm)
1203219089Spjd    z_streamp strm;
1204168404Spjd{
1205168404Spjd    if (strm == Z_NULL || strm->state == Z_NULL) return 0;
1206185029Spjd
1207168404Spjd    return ((deflate_state *)(strm->state))->pending;
1208168404Spjd}
1209168404Spjd
1210168404Spjd/* ===========================================================================
1211207334Spjd * Read a new buffer from the current input stream, update the adler32
1212207334Spjd * and total number of bytes read.  All deflate() input goes through
1213207334Spjd * this function so some applications may wish to modify it to avoid
1214207334Spjd * allocating a large strm->next_in buffer and copying from it.
1215219089Spjd * (See also flush_pending()).
1216219089Spjd */
1217219089Spjdlocal int read_buf(strm, buf, size)
1218219089Spjd    z_streamp strm;
1219168404Spjd    charf *buf;
1220219089Spjd    unsigned size;
1221219089Spjd{
1222219089Spjd    unsigned len = strm->avail_in;
1223219089Spjd
1224219089Spjd    if (len > size) len = size;
1225219089Spjd    if (len == 0) return 0;
1226219089Spjd
1227219089Spjd    strm->avail_in  -= len;
1228219089Spjd
1229185029Spjd    if (!((deflate_state *)(strm->state))->noheader) {
1230207334Spjd        strm->adler = adler32(strm->adler, strm->next_in, len);
1231219089Spjd    }
1232219089Spjd    zmemcpy(buf, strm->next_in, len);
1233219089Spjd    strm->next_in  += len;
1234219089Spjd    strm->total_in += len;
1235219089Spjd
1236219089Spjd    return (int)len;
1237219089Spjd}
1238219089Spjd
1239207334Spjd/* ===========================================================================
1240168404Spjd * Initialize the "longest match" routines for a new zlib stream
1241207334Spjd */
1242168404Spjdlocal void lm_init (s)
1243168404Spjd    deflate_state *s;
1244185029Spjd{
1245185029Spjd    s->window_size = (ulg)2L*s->w_size;
1246185029Spjd
1247185029Spjd    CLEAR_HASH(s);
1248185029Spjd
1249185029Spjd    /* Set the default configuration parameters:
1250185029Spjd     */
1251219089Spjd    s->max_lazy_match   = configuration_table[s->level].max_lazy;
1252219089Spjd    s->good_match       = configuration_table[s->level].good_length;
1253185029Spjd    s->nice_match       = configuration_table[s->level].nice_length;
1254219089Spjd    s->max_chain_length = configuration_table[s->level].max_chain;
1255219089Spjd
1256185029Spjd    s->strstart = 0;
1257185029Spjd    s->block_start = 0L;
1258185029Spjd    s->lookahead = 0;
1259219089Spjd    s->match_length = s->prev_length = MIN_MATCH-1;
1260219089Spjd    s->match_available = 0;
1261219089Spjd    s->ins_h = 0;
1262219089Spjd#ifdef ASMV
1263219089Spjd    match_init(); /* initialize the asm code */
1264219089Spjd#endif
1265219089Spjd}
1266219089Spjd
1267219089Spjd/* ===========================================================================
1268185029Spjd * Set match_start to the longest match starting at the given string and
1269185029Spjd * return its length. Matches shorter or equal to prev_length are discarded,
1270185029Spjd * in which case the result is equal to prev_length and match_start is
1271185029Spjd * garbage.
1272185029Spjd * IN assertions: cur_match is the head of the hash chain for the current
1273185029Spjd *   string (strstart) and its distance is <= MAX_DIST, and prev_length >= 1
1274219089Spjd * OUT assertion: the match length is not greater than s->lookahead.
1275219089Spjd */
1276219089Spjd#ifndef ASMV
1277219089Spjd/* For 80x86 and 680x0, an optimized version will be provided in match.asm or
1278219089Spjd * match.S. The code will be functionally equivalent.
1279185029Spjd */
1280185029Spjdlocal uInt longest_match(s, cur_match)
1281185029Spjd    deflate_state *s;
1282185029Spjd    IPos cur_match;                             /* current match */
1283219089Spjd{
1284219089Spjd    unsigned chain_length = s->max_chain_length;/* max hash chain length */
1285219089Spjd    register Bytef *scan = s->window + s->strstart; /* current string */
1286219089Spjd    register Bytef *match;                       /* matched string */
1287219089Spjd    register int len;                           /* length of current match */
1288219089Spjd    int best_len = s->prev_length;              /* best match length so far */
1289219089Spjd    int nice_match = s->nice_match;             /* stop if match long enough */
1290219089Spjd    IPos limit = s->strstart > (IPos)MAX_DIST(s) ?
1291219089Spjd        s->strstart - (IPos)MAX_DIST(s) : NIL;
1292219089Spjd    /* Stop when cur_match becomes <= limit. To simplify the code,
1293219089Spjd     * we prevent matches with the string of window index 0.
1294219089Spjd     */
1295219089Spjd    Posf *prev = s->prev;
1296219089Spjd    uInt wmask = s->w_mask;
1297219089Spjd
1298219089Spjd#ifdef UNALIGNED_OK
1299219089Spjd    /* Compare two bytes at a time. Note: this is not always beneficial.
1300219089Spjd     * Try with and without -DUNALIGNED_OK to check.
1301219089Spjd     */
1302219089Spjd    register Bytef *strend = s->window + s->strstart + MAX_MATCH - 1;
1303219089Spjd    register ush scan_start = *(ushf*)scan;
1304219089Spjd    register ush scan_end   = *(ushf*)(scan+best_len-1);
1305219089Spjd#else
1306185029Spjd    register Bytef *strend = s->window + s->strstart + MAX_MATCH;
1307185029Spjd    register Byte scan_end1  = scan[best_len-1];
1308185029Spjd    register Byte scan_end   = scan[best_len];
1309185029Spjd#endif
1310219089Spjd
1311219089Spjd    /* The code is optimized for HASH_BITS >= 8 and MAX_MATCH-2 multiple of 16.
1312219089Spjd     * It is easy to get rid of this optimization if necessary.
1313219089Spjd     */
1314219089Spjd    Assert(s->hash_bits >= 8 && MAX_MATCH == 258, "Code too clever");
1315219089Spjd
1316211932Smm    /* Do not waste too much time if we already have a good match: */
1317211932Smm    if (s->prev_length >= s->good_match) {
1318219089Spjd        chain_length >>= 2;
1319219089Spjd    }
1320219089Spjd    /* Do not look for matches beyond the end of the input. This is necessary
1321219089Spjd     * to make deflate deterministic.
1322219089Spjd     */
1323219089Spjd    if ((uInt)nice_match > s->lookahead) nice_match = s->lookahead;
1324219089Spjd
1325219089Spjd    Assert((ulg)s->strstart <= s->window_size-MIN_LOOKAHEAD, "need lookahead");
1326219089Spjd
1327219089Spjd    do {
1328219089Spjd        Assert(cur_match < s->strstart, "no future");
1329219089Spjd        match = s->window + cur_match;
1330219089Spjd
1331219089Spjd        /* Skip to next match if the match length cannot increase
1332219089Spjd         * or if the match length is less than 2:
1333185029Spjd         */
1334219089Spjd#if (defined(UNALIGNED_OK) && MAX_MATCH == 258)
1335219089Spjd        /* This code assumes sizeof(unsigned short) == 2. Do not use
1336185029Spjd         * UNALIGNED_OK if your compiler uses a different size.
1337185029Spjd         */
1338185029Spjd        if (*(ushf*)(match+best_len-1) != scan_end ||
1339185029Spjd            *(ushf*)match != scan_start) continue;
1340185029Spjd
1341185029Spjd        /* It is not necessary to compare scan[2] and match[2] since they are
1342168404Spjd         * always equal when the other bytes match, given that the hash keys
1343168404Spjd         * are equal and that HASH_BITS >= 8. Compare 2 bytes at a time at
1344168404Spjd         * strstart+3, +5, ... up to strstart+257. We check for insufficient
1345168404Spjd         * lookahead only every 4th comparison; the 128th check will be made
1346185029Spjd         * at strstart+257. If MAX_MATCH-2 is not a multiple of 8, it is
1347185029Spjd         * necessary to put more guard bytes at the end of the window, or
1348219089Spjd         * to check more often for insufficient lookahead.
1349168404Spjd         */
1350185029Spjd        Assert(scan[2] == match[2], "scan[2]?");
1351219089Spjd        scan++, match++;
1352219089Spjd        do {
1353185029Spjd        } while (*(ushf*)(scan+=2) == *(ushf*)(match+=2) &&
1354219089Spjd                 *(ushf*)(scan+=2) == *(ushf*)(match+=2) &&
1355185029Spjd                 *(ushf*)(scan+=2) == *(ushf*)(match+=2) &&
1356185029Spjd                 *(ushf*)(scan+=2) == *(ushf*)(match+=2) &&
1357185029Spjd                 scan < strend);
1358185029Spjd        /* The funny "do {}" generates better code on most compilers */
1359168404Spjd
1360168404Spjd        /* Here, scan <= window+strstart+257 */
1361168404Spjd        Assert(scan <= s->window+(unsigned)(s->window_size-1), "wild scan");
1362168404Spjd        if (*scan == *match) scan++;
1363168404Spjd
1364168404Spjd        len = (MAX_MATCH - 1) - (int)(strend-scan);
1365168404Spjd        scan = strend - (MAX_MATCH-1);
1366168404Spjd
1367201406Sdelphij#else /* UNALIGNED_OK */
1368168404Spjd
1369219089Spjd        if (match[best_len]   != scan_end  ||
1370168404Spjd            match[best_len-1] != scan_end1 ||
1371168404Spjd            *match            != *scan     ||
1372168404Spjd            *++match          != scan[1])      continue;
1373168404Spjd
1374168404Spjd        /* The check at best_len-1 can be removed because it will be made
1375168404Spjd         * again later. (This heuristic is not always a win.)
1376168404Spjd         * It is not necessary to compare scan[2] and match[2] since they
1377168404Spjd         * are always equal when the other bytes match, given that
1378168404Spjd         * the hash keys are equal and that HASH_BITS >= 8.
1379168404Spjd         */
1380168404Spjd        scan += 2, match++;
1381168404Spjd        Assert(*scan == *match, "match[2]?");
1382168404Spjd
1383168404Spjd        /* We check for insufficient lookahead only every 8th comparison;
1384168404Spjd         * the 256th check will be made at strstart+258.
1385168404Spjd         */
1386168404Spjd        do {
1387168404Spjd        } while (*++scan == *++match && *++scan == *++match &&
1388168404Spjd                 *++scan == *++match && *++scan == *++match &&
1389168404Spjd                 *++scan == *++match && *++scan == *++match &&
1390168404Spjd                 *++scan == *++match && *++scan == *++match &&
1391168404Spjd                 scan < strend);
1392168404Spjd
1393168404Spjd        Assert(scan <= s->window+(unsigned)(s->window_size-1), "wild scan");
1394168404Spjd
1395168404Spjd        len = MAX_MATCH - (int)(strend - scan);
1396168404Spjd        scan = strend - MAX_MATCH;
1397168404Spjd
1398168404Spjd#endif /* UNALIGNED_OK */
1399168404Spjd
1400201406Sdelphij        if (len > best_len) {
1401168404Spjd            s->match_start = cur_match;
1402201406Sdelphij            best_len = len;
1403168404Spjd            if (len >= nice_match) break;
1404168404Spjd#ifdef UNALIGNED_OK
1405219089Spjd            scan_end = *(ushf*)(scan+best_len-1);
1406168404Spjd#else
1407168404Spjd            scan_end1  = scan[best_len-1];
1408168404Spjd            scan_end   = scan[best_len];
1409168404Spjd#endif
1410168404Spjd        }
1411168404Spjd    } while ((cur_match = prev[cur_match & wmask]) > limit
1412168404Spjd             && --chain_length != 0);
1413168404Spjd
1414168404Spjd    if ((uInt)best_len <= s->lookahead) return best_len;
1415185029Spjd    return s->lookahead;
1416219089Spjd}
1417168404Spjd#endif /* ASMV */
1418185029Spjd
1419168404Spjd#ifdef DEBUG_ZLIB
1420168404Spjd/* ===========================================================================
1421168404Spjd * Check that the match at match_start is indeed a match.
1422211932Smm */
1423211932Smmlocal void check_match(s, start, match, length)
1424211932Smm    deflate_state *s;
1425211932Smm    IPos start, match;
1426211932Smm    int length;
1427168404Spjd{
1428185029Spjd    /* check that the match is indeed a match */
1429185029Spjd    if (zmemcmp((charf *)s->window + match,
1430168404Spjd                (charf *)s->window + start, length) != EQUAL) {
1431168404Spjd        fprintf(stderr, " start %u, match %u, length %d\n",
1432168404Spjd		start, match, length);
1433219089Spjd        do {
1434219089Spjd	    fprintf(stderr, "%c%c", s->window[match++], s->window[start++]);
1435168404Spjd	} while (--length != 0);
1436168404Spjd        z_error("invalid match");
1437168404Spjd    }
1438168404Spjd    if (z_verbose > 1) {
1439168404Spjd        fprintf(stderr,"\\[%d,%d]", start-match, length);
1440219089Spjd        do { putc(s->window[start++], stderr); } while (--length != 0);
1441168404Spjd    }
1442168404Spjd}
1443168404Spjd#else
1444168404Spjd#  define check_match(s, start, match, length)
1445168404Spjd#endif
1446168404Spjd
1447219089Spjd/* ===========================================================================
1448219089Spjd * Fill the window when the lookahead becomes insufficient.
1449219089Spjd * Updates strstart and lookahead.
1450168404Spjd *
1451185029Spjd * IN assertion: lookahead < MIN_LOOKAHEAD
1452219089Spjd * OUT assertions: strstart <= window_size-MIN_LOOKAHEAD
1453219089Spjd *    At least one byte has been read, or avail_in == 0; reads are
1454219089Spjd *    performed for at least two bytes (required for the zip translate_eol
1455219089Spjd *    option -- not supported here).
1456219089Spjd */
1457185029Spjdlocal void fill_window(s)
1458168404Spjd    deflate_state *s;
1459185029Spjd{
1460219089Spjd    register unsigned n, m;
1461185029Spjd    register Posf *p;
1462219089Spjd    unsigned more;    /* Amount of free space at the end of the window. */
1463185029Spjd    uInt wsize = s->w_size;
1464168404Spjd
1465168404Spjd    do {
1466168404Spjd        more = (unsigned)(s->window_size -(ulg)s->lookahead -(ulg)s->strstart);
1467168404Spjd
1468168404Spjd        /* Deal with !@#$% 64K limit: */
1469168404Spjd        if (more == 0 && s->strstart == 0 && s->lookahead == 0) {
1470168404Spjd            more = wsize;
1471168404Spjd
1472168404Spjd        } else if (more == (unsigned)(-1)) {
1473168404Spjd            /* Very unlikely, but possible on 16 bit machine if strstart == 0
1474168404Spjd             * and lookahead == 1 (input done one byte at time)
1475168404Spjd             */
1476168404Spjd            more--;
1477168404Spjd
1478168404Spjd        /* If the window is almost full and there is insufficient lookahead,
1479168404Spjd         * move the upper half to the lower one to make room in the upper half.
1480168404Spjd         */
1481168404Spjd        } else if (s->strstart >= wsize+MAX_DIST(s)) {
1482168404Spjd
1483168404Spjd            zmemcpy((charf *)s->window, (charf *)s->window+wsize,
1484168404Spjd                   (unsigned)wsize);
1485168404Spjd            s->match_start -= wsize;
1486168404Spjd            s->strstart    -= wsize; /* we now have strstart >= MAX_DIST */
1487168404Spjd            s->block_start -= (long) wsize;
1488219089Spjd
1489168404Spjd            /* Slide the hash table (could be avoided with 32 bit values
1490168404Spjd               at the expense of memory usage). We slide even when level == 0
1491168404Spjd               to keep the hash table consistent if we switch back to level > 0
1492168404Spjd               later. (Using level 0 permanently is not an optimal usage of
1493219089Spjd               zlib, so we don't care about this pathological case.)
1494168404Spjd             */
1495168404Spjd            n = s->hash_size;
1496168404Spjd            p = &s->head[n];
1497168404Spjd            do {
1498168404Spjd                m = *--p;
1499219089Spjd                *p = (Pos)(m >= wsize ? m-wsize : NIL);
1500168404Spjd            } while (--n);
1501168404Spjd
1502219089Spjd            n = wsize;
1503168404Spjd            p = &s->prev[n];
1504219089Spjd            do {
1505219089Spjd                m = *--p;
1506219089Spjd                *p = (Pos)(m >= wsize ? m-wsize : NIL);
1507219089Spjd                /* If n is not on any hash chain, prev[n] is garbage but
1508219089Spjd                 * its value will never be used.
1509219089Spjd                 */
1510219089Spjd            } while (--n);
1511219089Spjd            more += wsize;
1512219089Spjd        }
1513219089Spjd        if (s->strm->avail_in == 0) return;
1514219089Spjd
1515219089Spjd        /* If there was no sliding:
1516219089Spjd         *    strstart <= WSIZE+MAX_DIST-1 && lookahead <= MIN_LOOKAHEAD - 1 &&
1517219089Spjd         *    more == window_size - lookahead - strstart
1518219089Spjd         * => more >= window_size - (MIN_LOOKAHEAD-1 + WSIZE + MAX_DIST-1)
1519185029Spjd         * => more >= window_size - 2*WSIZE + 2
1520168404Spjd         * In the BIG_MEM or MMAP case (not yet supported),
1521168404Spjd         *   window_size == input_size + MIN_LOOKAHEAD  &&
1522185029Spjd         *   strstart + s->lookahead <= input_size => more >= MIN_LOOKAHEAD.
1523168404Spjd         * Otherwise, window_size == 2*WSIZE so more >= 2.
1524168404Spjd         * If there was sliding, more >= WSIZE. So in all cases, more >= 2.
1525168404Spjd         */
1526168404Spjd        Assert(more >= 2, "more < 2");
1527185029Spjd
1528185029Spjd        n = read_buf(s->strm, (charf *)s->window + s->strstart + s->lookahead,
1529168404Spjd                     more);
1530185029Spjd        s->lookahead += n;
1531168404Spjd
1532168404Spjd        /* Initialize the hash value now that we have some input: */
1533185029Spjd        if (s->lookahead >= MIN_MATCH) {
1534168404Spjd            s->ins_h = s->window[s->strstart];
1535168404Spjd            UPDATE_HASH(s, s->ins_h, s->window[s->strstart+1]);
1536168404Spjd#if MIN_MATCH != 3
1537185029Spjd            Call UPDATE_HASH() MIN_MATCH-3 more times
1538168404Spjd#endif
1539185029Spjd        }
1540168404Spjd        /* If the whole input has less than MIN_MATCH bytes, ins_h is garbage,
1541168404Spjd         * but this is not important since only literal bytes will be emitted.
1542168404Spjd         */
1543168404Spjd
1544219089Spjd    } while (s->lookahead < MIN_LOOKAHEAD && s->strm->avail_in != 0);
1545168404Spjd}
1546168404Spjd
1547168404Spjd/* ===========================================================================
1548185029Spjd * Flush the current block, with given end-of-file flag.
1549168404Spjd * IN assertion: strstart is set to the end of the current match.
1550219089Spjd */
1551219089Spjd#define FLUSH_BLOCK_ONLY(s, eof) { \
1552185029Spjd   _tr_flush_block(s, (s->block_start >= 0L ? \
1553168404Spjd                   (charf *)&s->window[(unsigned)s->block_start] : \
1554168404Spjd                   (charf *)Z_NULL), \
1555168404Spjd		(ulg)((long)s->strstart - s->block_start), \
1556168404Spjd		(eof)); \
1557168404Spjd   s->block_start = s->strstart; \
1558168404Spjd   flush_pending(s->strm); \
1559185029Spjd   Tracev((stderr,"[FLUSH]")); \
1560168404Spjd}
1561185029Spjd
1562168404Spjd/* Same but force premature exit if necessary. */
1563185029Spjd#define FLUSH_BLOCK(s, eof) { \
1564185029Spjd   FLUSH_BLOCK_ONLY(s, eof); \
1565185029Spjd   if (s->strm->avail_out == 0) return (eof) ? finish_started : need_more; \
1566168404Spjd}
1567168404Spjd
1568209962Smm/* ===========================================================================
1569168404Spjd * Copy without compression as much as possible from the input stream, return
1570209962Smm * the current block state.
1571168404Spjd * This function does not insert new strings in the dictionary since
1572185029Spjd * uncompressible data is probably not useful. This function is used
1573185029Spjd * only for the level=0 compression option.
1574185029Spjd * NOTE: this function should be optimized to avoid extra copying from
1575168404Spjd * window to pending_buf.
1576168404Spjd */
1577168404Spjdlocal block_state deflate_stored(s, flush)
1578168404Spjd    deflate_state *s;
1579168404Spjd    int flush;
1580185029Spjd{
1581185029Spjd    /* Stored blocks are limited to 0xffff bytes, pending_buf is limited
1582168404Spjd     * to pending_buf_size, and each stored block has a 5 byte header:
1583219089Spjd     */
1584168404Spjd    ulg max_block_size = 0xffff;
1585219089Spjd    ulg max_start;
1586219089Spjd
1587219089Spjd    if (max_block_size > s->pending_buf_size - 5) {
1588219089Spjd        max_block_size = s->pending_buf_size - 5;
1589219089Spjd    }
1590185029Spjd
1591168404Spjd    /* Copy as much as possible from input to output: */
1592185029Spjd    for (;;) {
1593185029Spjd        /* Fill the window as much as possible: */
1594185029Spjd        if (s->lookahead <= 1) {
1595185029Spjd
1596185029Spjd            Assert(s->strstart < s->w_size+MAX_DIST(s) ||
1597185029Spjd		   s->block_start >= (long)s->w_size, "slide too late");
1598185029Spjd
1599185029Spjd            fill_window(s);
1600185029Spjd            if (s->lookahead == 0 && flush == Z_NO_FLUSH) return need_more;
1601185029Spjd
1602185029Spjd            if (s->lookahead == 0) break; /* flush the current block */
1603185029Spjd        }
1604185029Spjd	Assert(s->block_start >= 0L, "block gone");
1605185029Spjd
1606185029Spjd	s->strstart += s->lookahead;
1607185029Spjd	s->lookahead = 0;
1608185029Spjd
1609185029Spjd	/* Emit a stored block if pending_buf will be full: */
1610185029Spjd 	max_start = s->block_start + max_block_size;
1611185029Spjd        if (s->strstart == 0 || (ulg)s->strstart >= max_start) {
1612185029Spjd	    /* strstart == 0 is possible when wraparound on 16-bit machine */
1613185029Spjd	    s->lookahead = (uInt)(s->strstart - max_start);
1614185029Spjd	    s->strstart = (uInt)max_start;
1615185029Spjd            FLUSH_BLOCK(s, 0);
1616185029Spjd	}
1617185029Spjd	/* Flush if we may have to slide, otherwise block_start may become
1618185029Spjd         * negative and the data will be gone:
1619185029Spjd         */
1620185029Spjd        if (s->strstart - (uInt)s->block_start >= MAX_DIST(s)) {
1621185029Spjd            FLUSH_BLOCK(s, 0);
1622219089Spjd	}
1623185029Spjd    }
1624185029Spjd    FLUSH_BLOCK(s, flush == Z_FINISH);
1625168404Spjd    return flush == Z_FINISH ? finish_done : block_done;
1626168404Spjd}
1627219089Spjd
1628219089Spjd/* ===========================================================================
1629185029Spjd * Compress as much as possible from the input stream, return the current
1630185029Spjd * block state.
1631185029Spjd * This function does not perform lazy evaluation of matches and inserts
1632185029Spjd * new strings in the dictionary only for unmatched strings or for short
1633185029Spjd * matches. It is used only for the fast compression options.
1634185029Spjd */
1635219089Spjdlocal block_state deflate_fast(s, flush)
1636219089Spjd    deflate_state *s;
1637185029Spjd    int flush;
1638185029Spjd{
1639168404Spjd    IPos hash_head = NIL; /* head of the hash chain */
1640168404Spjd    int bflush;           /* set if current block must be flushed */
1641168404Spjd
1642168404Spjd    for (;;) {
1643185029Spjd        /* Make sure that we always have enough lookahead, except
1644185029Spjd         * at the end of the input file. We need MAX_MATCH bytes
1645185029Spjd         * for the next match, plus MIN_MATCH bytes to insert the
1646185029Spjd         * string following the next match.
1647185029Spjd         */
1648185029Spjd        if (s->lookahead < MIN_LOOKAHEAD) {
1649185029Spjd            fill_window(s);
1650185029Spjd            if (s->lookahead < MIN_LOOKAHEAD && flush == Z_NO_FLUSH) {
1651185029Spjd	        return need_more;
1652185029Spjd	    }
1653185029Spjd            if (s->lookahead == 0) break; /* flush the current block */
1654185029Spjd        }
1655185029Spjd
1656185029Spjd        /* Insert the string window[strstart .. strstart+2] in the
1657185029Spjd         * dictionary, and set hash_head to the head of the hash chain:
1658185029Spjd         */
1659185029Spjd        if (s->lookahead >= MIN_MATCH) {
1660185029Spjd            INSERT_STRING(s, s->strstart, hash_head);
1661185029Spjd        }
1662185029Spjd
1663219089Spjd        /* Find the longest match, discarding those <= prev_length.
1664219089Spjd         * At this point we have always match_length < MIN_MATCH
1665185029Spjd         */
1666185029Spjd        if (hash_head != NIL && s->strstart - hash_head <= MAX_DIST(s)) {
1667185029Spjd            /* To simplify the code, we prevent matches with the string
1668185029Spjd             * of window index 0 (in particular we have to avoid a match
1669185029Spjd             * of the string with itself at the start of the input file).
1670185029Spjd             */
1671185029Spjd            if (s->strategy != Z_HUFFMAN_ONLY) {
1672185029Spjd                s->match_length = longest_match (s, hash_head);
1673185029Spjd            }
1674219089Spjd            /* longest_match() sets match_start */
1675185029Spjd        }
1676185029Spjd        if (s->match_length >= MIN_MATCH) {
1677185029Spjd            check_match(s, s->strstart, s->match_start, s->match_length);
1678185029Spjd
1679185029Spjd            bflush = _tr_tally(s, s->strstart - s->match_start,
1680185029Spjd                               s->match_length - MIN_MATCH);
1681185029Spjd
1682185029Spjd            s->lookahead -= s->match_length;
1683185029Spjd
1684185029Spjd            /* Insert new strings in the hash table only if the match length
1685185029Spjd             * is not too large. This saves time but degrades compression.
1686219089Spjd             */
1687219089Spjd            if (s->match_length <= s->max_insert_length &&
1688209962Smm                s->lookahead >= MIN_MATCH) {
1689185029Spjd                s->match_length--; /* string at strstart already in hash table */
1690209962Smm                do {
1691185029Spjd                    s->strstart++;
1692185029Spjd                    INSERT_STRING(s, s->strstart, hash_head);
1693185029Spjd                    /* strstart never exceeds WSIZE-MAX_MATCH, so there are
1694185029Spjd                     * always MIN_MATCH bytes ahead.
1695185029Spjd                     */
1696185029Spjd                } while (--s->match_length != 0);
1697185029Spjd                s->strstart++;
1698185029Spjd            } else {
1699185029Spjd                s->strstart += s->match_length;
1700219089Spjd                s->match_length = 0;
1701219089Spjd                s->ins_h = s->window[s->strstart];
1702219089Spjd                UPDATE_HASH(s, s->ins_h, s->window[s->strstart+1]);
1703185029Spjd#if MIN_MATCH != 3
1704219089Spjd                Call UPDATE_HASH() MIN_MATCH-3 more times
1705219089Spjd#endif
1706219089Spjd                /* If lookahead < MIN_MATCH, ins_h is garbage, but it does not
1707219089Spjd                 * matter since it will be recomputed at next deflate call.
1708219089Spjd                 */
1709219089Spjd            }
1710219089Spjd        } else {
1711168404Spjd            /* No match, output a literal byte */
1712168404Spjd            Tracevv((stderr,"%c", s->window[s->strstart]));
1713168404Spjd            bflush = _tr_tally (s, 0, s->window[s->strstart]);
1714168404Spjd            s->lookahead--;
1715168404Spjd            s->strstart++;
1716168404Spjd        }
1717168404Spjd        if (bflush) FLUSH_BLOCK(s, 0);
1718168404Spjd    }
1719185029Spjd    FLUSH_BLOCK(s, flush == Z_FINISH);
1720168404Spjd    return flush == Z_FINISH ? finish_done : block_done;
1721209962Smm}
1722209962Smm
1723168404Spjd/* ===========================================================================
1724168404Spjd * Same as above, but achieves better compression. We use a lazy
1725168404Spjd * evaluation for matches: a match is finally adopted only if there is
1726185029Spjd * no better match at the next window position.
1727185029Spjd */
1728185029Spjdlocal block_state deflate_slow(s, flush)
1729185029Spjd    deflate_state *s;
1730185029Spjd    int flush;
1731185029Spjd{
1732185029Spjd    IPos hash_head = NIL;    /* head of hash chain */
1733185029Spjd    int bflush;              /* set if current block must be flushed */
1734185029Spjd
1735185029Spjd    /* Process the input block. */
1736185029Spjd    for (;;) {
1737185029Spjd        /* Make sure that we always have enough lookahead, except
1738185029Spjd         * at the end of the input file. We need MAX_MATCH bytes
1739185029Spjd         * for the next match, plus MIN_MATCH bytes to insert the
1740185029Spjd         * string following the next match.
1741185029Spjd         */
1742185029Spjd        if (s->lookahead < MIN_LOOKAHEAD) {
1743185029Spjd            fill_window(s);
1744185029Spjd            if (s->lookahead < MIN_LOOKAHEAD && flush == Z_NO_FLUSH) {
1745219089Spjd	        return need_more;
1746219089Spjd	    }
1747219089Spjd            if (s->lookahead == 0) break; /* flush the current block */
1748219089Spjd        }
1749185029Spjd
1750185029Spjd        /* Insert the string window[strstart .. strstart+2] in the
1751219089Spjd         * dictionary, and set hash_head to the head of the hash chain:
1752219089Spjd         */
1753219089Spjd        if (s->lookahead >= MIN_MATCH) {
1754219089Spjd            INSERT_STRING(s, s->strstart, hash_head);
1755219089Spjd        }
1756185029Spjd
1757185029Spjd        /* Find the longest match, discarding those <= prev_length.
1758185029Spjd         */
1759185029Spjd        s->prev_length = s->match_length, s->prev_match = s->match_start;
1760185029Spjd        s->match_length = MIN_MATCH-1;
1761185029Spjd
1762185029Spjd        if (hash_head != NIL && s->prev_length < s->max_lazy_match &&
1763219089Spjd            s->strstart - hash_head <= MAX_DIST(s)) {
1764219089Spjd            /* To simplify the code, we prevent matches with the string
1765219089Spjd             * of window index 0 (in particular we have to avoid a match
1766219089Spjd             * of the string with itself at the start of the input file).
1767219089Spjd             */
1768219089Spjd            if (s->strategy != Z_HUFFMAN_ONLY) {
1769219089Spjd                s->match_length = longest_match (s, hash_head);
1770219089Spjd            }
1771219089Spjd            /* longest_match() sets match_start */
1772219089Spjd
1773185029Spjd            if (s->match_length <= 5 && (s->strategy == Z_FILTERED ||
1774185029Spjd                 (s->match_length == MIN_MATCH &&
1775185029Spjd                  s->strstart - s->match_start > TOO_FAR))) {
1776185029Spjd
1777219089Spjd                /* If prev_match is also MIN_MATCH, match_start is garbage
1778185029Spjd                 * but we will ignore the current match anyway.
1779185029Spjd                 */
1780185029Spjd                s->match_length = MIN_MATCH-1;
1781185029Spjd            }
1782185029Spjd        }
1783185029Spjd        /* If there was a match at the previous step and the current
1784219089Spjd         * match is not better, output the previous match:
1785219089Spjd         */
1786209962Smm        if (s->prev_length >= MIN_MATCH && s->match_length <= s->prev_length) {
1787185029Spjd            uInt max_insert = s->strstart + s->lookahead - MIN_MATCH;
1788209962Smm            /* Do not insert strings in hash table beyond this. */
1789185029Spjd
1790185029Spjd            check_match(s, s->strstart-1, s->prev_match, s->prev_length);
1791185029Spjd
1792185029Spjd            bflush = _tr_tally(s, s->strstart -1 - s->prev_match,
1793185029Spjd                               s->prev_length - MIN_MATCH);
1794185029Spjd
1795185029Spjd            /* Insert in hash table all strings up to the end of the match.
1796185029Spjd             * strstart-1 and strstart are already inserted. If there is not
1797219089Spjd             * enough lookahead, the last two strings are not inserted in
1798219089Spjd             * the hash table.
1799219089Spjd             */
1800219089Spjd            s->lookahead -= s->prev_length-1;
1801219089Spjd            s->prev_length -= 2;
1802219089Spjd            do {
1803219089Spjd                if (++s->strstart <= max_insert) {
1804219089Spjd                    INSERT_STRING(s, s->strstart, hash_head);
1805185029Spjd                }
1806185029Spjd            } while (--s->prev_length != 0);
1807185029Spjd            s->match_available = 0;
1808185029Spjd            s->match_length = MIN_MATCH-1;
1809185029Spjd            s->strstart++;
1810185029Spjd
1811168404Spjd            if (bflush) FLUSH_BLOCK(s, 0);
1812185029Spjd
1813168404Spjd        } else if (s->match_available) {
1814168404Spjd            /* If there was no match at the previous position, output a
1815219089Spjd             * single literal. If there was a match but the current match
1816185029Spjd             * is longer, truncate the previous match to a single literal.
1817185029Spjd             */
1818185029Spjd            Tracevv((stderr,"%c", s->window[s->strstart-1]));
1819168404Spjd            if (_tr_tally (s, 0, s->window[s->strstart-1])) {
1820207334Spjd                FLUSH_BLOCK_ONLY(s, 0);
1821168404Spjd            }
1822199156Spjd            s->strstart++;
1823168404Spjd            s->lookahead--;
1824185029Spjd            if (s->strm->avail_out == 0) return need_more;
1825209962Smm        } else {
1826168404Spjd            /* There is no previous match to compare with, wait for
1827168404Spjd             * the next step to decide.
1828168404Spjd             */
1829168404Spjd            s->match_available = 1;
1830168404Spjd            s->strstart++;
1831168404Spjd            s->lookahead--;
1832168404Spjd        }
1833168404Spjd    }
1834168404Spjd    Assert (flush != Z_NO_FLUSH, "no flush?");
1835168404Spjd    if (s->match_available) {
1836168404Spjd        Tracevv((stderr,"%c", s->window[s->strstart-1]));
1837168404Spjd        _tr_tally (s, 0, s->window[s->strstart-1]);
1838168404Spjd        s->match_available = 0;
1839168404Spjd    }
1840168404Spjd    FLUSH_BLOCK(s, flush == Z_FINISH);
1841168404Spjd    return flush == Z_FINISH ? finish_done : block_done;
1842219089Spjd}
1843185029Spjd/* --- deflate.c */
1844185029Spjd
1845185029Spjd/* +++ trees.c */
1846185029Spjd/* trees.c -- output deflated data using Huffman coding
1847185029Spjd * Copyright (C) 1995-1996 Jean-loup Gailly
1848168404Spjd * For conditions of distribution and use, see copyright notice in zlib.h
1849185029Spjd */
1850185029Spjd
1851185029Spjd/*
1852185029Spjd *  ALGORITHM
1853209962Smm *
1854209962Smm *      The "deflation" process uses several Huffman trees. The more
1855185029Spjd *      common source values are represented by shorter bit sequences.
1856185029Spjd *
1857185029Spjd *      Each code tree is stored in a compressed form which is itself
1858185029Spjd * a Huffman encoding of the lengths of all the code strings (in
1859185029Spjd * ascending order by source values).  The actual code strings are
1860185029Spjd * reconstructed from the lengths in the inflate process, as described
1861185029Spjd * in the deflate specification.
1862185029Spjd *
1863185029Spjd *  REFERENCES
1864185029Spjd *
1865209962Smm *      Deutsch, L.P.,"'Deflate' Compressed Data Format Specification".
1866168404Spjd *      Available in ftp.uu.net:/pub/archiving/zip/doc/deflate-1.1.doc
1867168404Spjd *
1868219089Spjd *      Storer, James A.
1869219089Spjd *          Data Compression:  Methods and Theory, pp. 49-50.
1870219089Spjd *          Computer Science Press, 1988.  ISBN 0-7167-8156-5.
1871219089Spjd *
1872219089Spjd *      Sedgewick, R.
1873219089Spjd *          Algorithms, p290.
1874219089Spjd *          Addison-Wesley, 1983. ISBN 0-201-06672-6.
1875219089Spjd */
1876219089Spjd
1877219089Spjd/* From: trees.c,v 1.11 1996/07/24 13:41:06 me Exp $ */
1878219089Spjd
1879219089Spjd/* #include "deflate.h" */
1880168404Spjd
1881168404Spjd#ifdef DEBUG_ZLIB
1882209962Smm#  include <ctype.h>
1883168404Spjd#endif
1884209962Smm
1885168404Spjd/* ===========================================================================
1886168404Spjd * Constants
1887168404Spjd */
1888168404Spjd
1889168404Spjd#define MAX_BL_BITS 7
1890168404Spjd/* Bit length codes must not exceed MAX_BL_BITS bits */
1891185029Spjd
1892168404Spjd#define END_BLOCK 256
1893168404Spjd/* end of block literal code */
1894168404Spjd
1895185029Spjd#define REP_3_6      16
1896185029Spjd/* repeat previous bit length 3-6 times (2 bits of repeat count) */
1897168404Spjd
1898219089Spjd#define REPZ_3_10    17
1899219089Spjd/* repeat a zero length 3-10 times  (3 bits of repeat count) */
1900168404Spjd
1901199156Spjd#define REPZ_11_138  18
1902219089Spjd/* repeat a zero length 11-138 times  (7 bits of repeat count) */
1903219089Spjd
1904168404Spjdlocal int extra_lbits[LENGTH_CODES] /* extra bits for each length code */
1905168404Spjd   = {0,0,0,0,0,0,0,0,1,1,1,1,2,2,2,2,3,3,3,3,4,4,4,4,5,5,5,5,0};
1906219089Spjd
1907168404Spjdlocal int extra_dbits[D_CODES] /* extra bits for each distance code */
1908199156Spjd   = {0,0,0,0,1,1,2,2,3,3,4,4,5,5,6,6,7,7,8,8,9,9,10,10,11,11,12,12,13,13};
1909199156Spjd
1910199156Spjdlocal int extra_blbits[BL_CODES]/* extra bits for each bit length code */
1911185029Spjd   = {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,2,3,7};
1912168404Spjd
1913168404Spjdlocal uch bl_order[BL_CODES]
1914185029Spjd   = {16,17,18,0,8,7,9,6,10,5,11,4,12,3,13,2,14,1,15};
1915185029Spjd/* The lengths of the bit length codes are sent in order of decreasing
1916219089Spjd * probability, to avoid transmitting the lengths for unused bit length codes.
1917185029Spjd */
1918219089Spjd
1919219089Spjd#define Buf_size (8 * 2*sizeof(char))
1920219089Spjd/* Number of bits used within bi_buf. (bi_buf might be implemented on
1921219089Spjd * more than 16 bits on some systems.)
1922219089Spjd */
1923219089Spjd
1924185029Spjd/* ===========================================================================
1925185029Spjd * Local data. These are initialized only once.
1926185029Spjd */
1927185029Spjd
1928185029Spjdlocal ct_data static_ltree[L_CODES+2];
1929185029Spjd/* The static literal tree. Since the bit lengths are imposed, there is no
1930168404Spjd * need for the L_CODES extra codes used during heap construction. However
1931168404Spjd * The codes 286 and 287 are needed to build a canonical tree (see _tr_init
1932168404Spjd * below).
1933168404Spjd */
1934168404Spjd
1935207334Spjdlocal ct_data static_dtree[D_CODES];
1936207334Spjd/* The static distance tree. (Actually a trivial tree since all codes use
1937207334Spjd * 5 bits.)
1938185029Spjd */
1939209962Smm
1940209962Smmlocal uch dist_code[512];
1941219089Spjd/* distance codes. The first 256 values correspond to the distances
1942185029Spjd * 3 .. 258, the last 256 values correspond to the top 8 bits of
1943185029Spjd * the 15 bit distances.
1944168404Spjd */
1945209962Smm
1946185029Spjdlocal uch length_code[MAX_MATCH-MIN_MATCH+1];
1947168404Spjd/* length code for each normalized match length (0 == MIN_MATCH) */
1948219089Spjd
1949209962Smmlocal int base_length[LENGTH_CODES];
1950209962Smm/* First normalized length for each code (0 = MIN_MATCH) */
1951209962Smm
1952209962Smmlocal int base_dist[D_CODES];
1953209962Smm/* First normalized distance for each code (0 = distance of 1) */
1954209962Smm
1955209962Smmstruct static_tree_desc_s {
1956209962Smm    ct_data *static_tree;        /* static tree or NULL */
1957209962Smm    intf    *extra_bits;         /* extra bits for each code or NULL */
1958209962Smm    int     extra_base;          /* base index for extra_bits */
1959209962Smm    int     elems;               /* max number of elements in the tree */
1960207334Spjd    int     max_length;          /* max bit length for the codes */
1961207334Spjd};
1962168404Spjd
1963185029Spjdlocal static_tree_desc  static_l_desc =
1964168404Spjd{static_ltree, extra_lbits, LITERALS+1, L_CODES, MAX_BITS};
1965219089Spjd
1966168404Spjdlocal static_tree_desc  static_d_desc =
1967219089Spjd{static_dtree, extra_dbits, 0,          D_CODES, MAX_BITS};
1968168404Spjd
1969219089Spjdlocal static_tree_desc  static_bl_desc =
1970219089Spjd{(ct_data *)0, extra_blbits, 0,      BL_CODES, MAX_BL_BITS};
1971219089Spjd
1972219089Spjd/* ===========================================================================
1973219089Spjd * Local (static) routines in this file.
1974219089Spjd */
1975219089Spjd
1976219089Spjdlocal void tr_static_init OF((void));
1977219089Spjdlocal void init_block     OF((deflate_state *s));
1978219089Spjdlocal void pqdownheap     OF((deflate_state *s, ct_data *tree, int k));
1979219089Spjdlocal void gen_bitlen     OF((deflate_state *s, tree_desc *desc));
1980219089Spjdlocal void gen_codes      OF((ct_data *tree, int max_code, ushf *bl_count));
1981219089Spjdlocal void build_tree     OF((deflate_state *s, tree_desc *desc));
1982219089Spjdlocal void scan_tree      OF((deflate_state *s, ct_data *tree, int max_code));
1983219089Spjdlocal void send_tree      OF((deflate_state *s, ct_data *tree, int max_code));
1984168404Spjdlocal int  build_bl_tree  OF((deflate_state *s));
1985168404Spjdlocal void send_all_trees OF((deflate_state *s, int lcodes, int dcodes,
1986168404Spjd                              int blcodes));
1987219089Spjdlocal void compress_block OF((deflate_state *s, ct_data *ltree,
1988168404Spjd                              ct_data *dtree));
1989168404Spjdlocal void set_data_type  OF((deflate_state *s));
1990219089Spjdlocal unsigned bi_reverse OF((unsigned value, int length));
1991219089Spjdlocal void bi_windup      OF((deflate_state *s));
1992219089Spjdlocal void bi_flush       OF((deflate_state *s));
1993219089Spjdlocal void copy_block     OF((deflate_state *s, charf *buf, unsigned len,
1994168404Spjd                              int header));
1995219089Spjd
1996219089Spjd#ifndef DEBUG_ZLIB
1997168404Spjd#  define send_code(s, c, tree) send_bits(s, tree[(c)].Code, tree[(c)].Len)
1998168404Spjd   /* Send a code of the given tree. c and tree must not have side effects */
1999219089Spjd
2000219089Spjd#else /* DEBUG_ZLIB */
2001219089Spjd#  define send_code(s, c, tree) \
2002219089Spjd     { if (verbose>2) fprintf(stderr,"\ncd %3d ",(c)); \
2003219089Spjd       send_bits(s, tree[c].Code, tree[c].Len); }
2004168404Spjd#endif
2005168404Spjd
2006168404Spjd#define d_code(dist) \
2007168404Spjd   ((dist) < 256 ? dist_code[dist] : dist_code[256+((dist)>>7)])
2008219089Spjd/* Mapping from a distance to a distance code. dist is the distance - 1 and
2009219089Spjd * must not have side effects. dist_code[256] and dist_code[257] are never
2010168404Spjd * used.
2011219089Spjd */
2012219089Spjd
2013219089Spjd/* ===========================================================================
2014219089Spjd * Output a short LSB first on the stream.
2015219089Spjd * IN assertion: there is enough room in pendingBuf.
2016219089Spjd */
2017219089Spjd#define put_short(s, w) { \
2018219089Spjd    put_byte(s, (uch)((w) & 0xff)); \
2019219089Spjd    put_byte(s, (uch)((ush)(w) >> 8)); \
2020219089Spjd}
2021219089Spjd
2022219089Spjd/* ===========================================================================
2023219089Spjd * Send a value on a given number of bits.
2024219089Spjd * IN assertion: length <= 16 and value fits in length bits.
2025219089Spjd */
2026219089Spjd#ifdef DEBUG_ZLIB
2027219089Spjdlocal void send_bits      OF((deflate_state *s, int value, int length));
2028219089Spjd
2029219089Spjdlocal void send_bits(s, value, length)
2030219089Spjd    deflate_state *s;
2031219089Spjd    int value;  /* value to send */
2032219089Spjd    int length; /* number of bits */
2033219089Spjd{
2034219089Spjd    Tracevv((stderr," l %2d v %4x ", length, value));
2035219089Spjd    Assert(length > 0 && length <= 15, "invalid length");
2036219089Spjd    s->bits_sent += (ulg)length;
2037219089Spjd
2038219089Spjd    /* If not enough room in bi_buf, use (valid) bits from bi_buf and
2039219089Spjd     * (16 - bi_valid) bits from value, leaving (width - (16-bi_valid))
2040219089Spjd     * unused bits in value.
2041219089Spjd     */
2042219089Spjd    if (s->bi_valid > (int)Buf_size - length) {
2043219089Spjd        s->bi_buf |= (value << s->bi_valid);
2044219089Spjd        put_short(s, s->bi_buf);
2045219089Spjd        s->bi_buf = (ush)value >> (Buf_size - s->bi_valid);
2046219089Spjd        s->bi_valid += length - Buf_size;
2047219089Spjd    } else {
2048219089Spjd        s->bi_buf |= value << s->bi_valid;
2049219089Spjd        s->bi_valid += length;
2050219089Spjd    }
2051219089Spjd}
2052219089Spjd#else /* !DEBUG_ZLIB */
2053219089Spjd
2054219089Spjd#define send_bits(s, value, length) \
2055219089Spjd{ int len = (length);\
2056219089Spjd  if ((s)->bi_valid > (int)Buf_size - len) {\
2057219089Spjd    int val = (value);\
2058219089Spjd    (s)->bi_buf |= (val << (s)->bi_valid);\
2059219089Spjd    put_short((s), (s)->bi_buf);\
2060219089Spjd    (s)->bi_buf = (ush)val >> (Buf_size - (s)->bi_valid);\
2061219089Spjd    (s)->bi_valid += len - Buf_size;\
2062219089Spjd  } else {\
2063219089Spjd    (s)->bi_buf |= (value) << (s)->bi_valid;\
2064219089Spjd    (s)->bi_valid += len;\
2065219089Spjd  }\
2066219089Spjd}
2067219089Spjd#endif /* DEBUG_ZLIB */
2068219089Spjd
2069219089Spjd/* the arguments must not have side effects */
2070219089Spjd
2071219089Spjd/* ===========================================================================
2072219089Spjd * Initialize the various 'constant' tables. In a multi-threaded environment,
2073219089Spjd * this function may be called by two threads concurrently, but this is
2074219089Spjd * harmless since both invocations do exactly the same thing.
2075219089Spjd */
2076168404Spjdlocal void tr_static_init()
2077168404Spjd{
2078168404Spjd    static int static_init_done = 0;
2079168404Spjd    int n;        /* iterates over tree elements */
2080219089Spjd    int bits;     /* bit counter */
2081168404Spjd    int length;   /* length value */
2082168404Spjd    int code;     /* code value */
2083168404Spjd    int dist;     /* distance index */
2084168404Spjd    ush bl_count[MAX_BITS+1];
2085168404Spjd    /* number of codes at each bit length for an optimal tree */
2086168404Spjd
2087168404Spjd    if (static_init_done) return;
2088219089Spjd
2089219089Spjd    /* Initialize the mapping length (0..255) -> length code (0..28) */
2090219089Spjd    length = 0;
2091219089Spjd    for (code = 0; code < LENGTH_CODES-1; code++) {
2092168404Spjd        base_length[code] = length;
2093168404Spjd        for (n = 0; n < (1<<extra_lbits[code]); n++) {
2094168404Spjd            length_code[length++] = (uch)code;
2095168404Spjd        }
2096168404Spjd    }
2097168404Spjd    Assert (length == 256, "tr_static_init: length != 256");
2098168404Spjd    /* Note that the length 255 (match length 258) can be represented
2099168404Spjd     * in two different ways: code 284 + 5 bits or code 285, so we
2100168404Spjd     * overwrite length_code[255] to use the best encoding:
2101168404Spjd     */
2102168404Spjd    length_code[length-1] = (uch)code;
2103168404Spjd
2104168404Spjd    /* Initialize the mapping dist (0..32K) -> dist code (0..29) */
2105185029Spjd    dist = 0;
2106185029Spjd    for (code = 0 ; code < 16; code++) {
2107168404Spjd        base_dist[code] = dist;
2108168404Spjd        for (n = 0; n < (1<<extra_dbits[code]); n++) {
2109168404Spjd            dist_code[dist++] = (uch)code;
2110168404Spjd        }
2111168404Spjd    }
2112168404Spjd    Assert (dist == 256, "tr_static_init: dist != 256");
2113168404Spjd    dist >>= 7; /* from now on, all distances are divided by 128 */
2114168404Spjd    for ( ; code < D_CODES; code++) {
2115168404Spjd        base_dist[code] = dist << 7;
2116219089Spjd        for (n = 0; n < (1<<(extra_dbits[code]-7)); n++) {
2117219089Spjd            dist_code[256 + dist++] = (uch)code;
2118219089Spjd        }
2119219089Spjd    }
2120219089Spjd    Assert (dist == 256, "tr_static_init: 256+dist != 512");
2121219089Spjd
2122219089Spjd    /* Construct the codes of the static literal tree */
2123219089Spjd    for (bits = 0; bits <= MAX_BITS; bits++) bl_count[bits] = 0;
2124219089Spjd    n = 0;
2125219089Spjd    while (n <= 143) static_ltree[n++].Len = 8, bl_count[8]++;
2126219089Spjd    while (n <= 255) static_ltree[n++].Len = 9, bl_count[9]++;
2127168404Spjd    while (n <= 279) static_ltree[n++].Len = 7, bl_count[7]++;
2128168404Spjd    while (n <= 287) static_ltree[n++].Len = 8, bl_count[8]++;
2129219089Spjd    /* Codes 286 and 287 do not exist, but we must include them in the
2130219089Spjd     * tree construction to get a canonical Huffman tree (longest code
2131219089Spjd     * all ones)
2132219089Spjd     */
2133219089Spjd    gen_codes((ct_data *)static_ltree, L_CODES+1, bl_count);
2134168404Spjd
2135168404Spjd    /* The static distance tree is trivial: */
2136219089Spjd    for (n = 0; n < D_CODES; n++) {
2137168404Spjd        static_dtree[n].Len = 5;
2138168404Spjd        static_dtree[n].Code = bi_reverse((unsigned)n, 5);
2139219089Spjd    }
2140219089Spjd    static_init_done = 1;
2141219089Spjd}
2142219089Spjd
2143219089Spjd/* ===========================================================================
2144219089Spjd * Initialize the tree data structures for a new zlib stream.
2145219089Spjd */
2146219089Spjdvoid _tr_init(s)
2147219089Spjd    deflate_state *s;
2148219089Spjd{
2149219089Spjd    tr_static_init();
2150219089Spjd
2151219089Spjd    s->compressed_len = 0L;
2152219089Spjd
2153219089Spjd    s->l_desc.dyn_tree = s->dyn_ltree;
2154219089Spjd    s->l_desc.stat_desc = &static_l_desc;
2155219089Spjd
2156219089Spjd    s->d_desc.dyn_tree = s->dyn_dtree;
2157219089Spjd    s->d_desc.stat_desc = &static_d_desc;
2158219089Spjd
2159219089Spjd    s->bl_desc.dyn_tree = s->bl_tree;
2160219089Spjd    s->bl_desc.stat_desc = &static_bl_desc;
2161219089Spjd
2162219089Spjd    s->bi_buf = 0;
2163219089Spjd    s->bi_valid = 0;
2164219089Spjd    s->last_eob_len = 8; /* enough lookahead for inflate */
2165219089Spjd#ifdef DEBUG_ZLIB
2166219089Spjd    s->bits_sent = 0L;
2167219089Spjd#endif
2168219089Spjd
2169219089Spjd    /* Initialize the first block of the first file: */
2170219089Spjd    init_block(s);
2171219089Spjd}
2172219089Spjd
2173219089Spjd/* ===========================================================================
2174219089Spjd * Initialize a new block.
2175219089Spjd */
2176219089Spjdlocal void init_block(s)
2177219089Spjd    deflate_state *s;
2178219089Spjd{
2179219089Spjd    int n; /* iterates over tree elements */
2180219089Spjd
2181219089Spjd    /* Initialize the trees. */
2182219089Spjd    for (n = 0; n < L_CODES;  n++) s->dyn_ltree[n].Freq = 0;
2183219089Spjd    for (n = 0; n < D_CODES;  n++) s->dyn_dtree[n].Freq = 0;
2184219089Spjd    for (n = 0; n < BL_CODES; n++) s->bl_tree[n].Freq = 0;
2185219089Spjd
2186219089Spjd    s->dyn_ltree[END_BLOCK].Freq = 1;
2187219089Spjd    s->opt_len = s->static_len = 0L;
2188219089Spjd    s->last_lit = s->matches = 0;
2189219089Spjd}
2190219089Spjd
2191219089Spjd#define SMALLEST 1
2192219089Spjd/* Index within the heap array of least frequent node in the Huffman tree */
2193
2194
2195/* ===========================================================================
2196 * Remove the smallest element from the heap and recreate the heap with
2197 * one less element. Updates heap and heap_len.
2198 */
2199#define pqremove(s, tree, top) \
2200{\
2201    top = s->heap[SMALLEST]; \
2202    s->heap[SMALLEST] = s->heap[s->heap_len--]; \
2203    pqdownheap(s, tree, SMALLEST); \
2204}
2205
2206/* ===========================================================================
2207 * Compares to subtrees, using the tree depth as tie breaker when
2208 * the subtrees have equal frequency. This minimizes the worst case length.
2209 */
2210#define smaller(tree, n, m, depth) \
2211   (tree[n].Freq < tree[m].Freq || \
2212   (tree[n].Freq == tree[m].Freq && depth[n] <= depth[m]))
2213
2214/* ===========================================================================
2215 * Restore the heap property by moving down the tree starting at node k,
2216 * exchanging a node with the smallest of its two sons if necessary, stopping
2217 * when the heap property is re-established (each father smaller than its
2218 * two sons).
2219 */
2220local void pqdownheap(s, tree, k)
2221    deflate_state *s;
2222    ct_data *tree;  /* the tree to restore */
2223    int k;               /* node to move down */
2224{
2225    int v = s->heap[k];
2226    int j = k << 1;  /* left son of k */
2227    while (j <= s->heap_len) {
2228        /* Set j to the smallest of the two sons: */
2229        if (j < s->heap_len &&
2230            smaller(tree, s->heap[j+1], s->heap[j], s->depth)) {
2231            j++;
2232        }
2233        /* Exit if v is smaller than both sons */
2234        if (smaller(tree, v, s->heap[j], s->depth)) break;
2235
2236        /* Exchange v with the smallest son */
2237        s->heap[k] = s->heap[j];  k = j;
2238
2239        /* And continue down the tree, setting j to the left son of k */
2240        j <<= 1;
2241    }
2242    s->heap[k] = v;
2243}
2244
2245/* ===========================================================================
2246 * Compute the optimal bit lengths for a tree and update the total bit length
2247 * for the current block.
2248 * IN assertion: the fields freq and dad are set, heap[heap_max] and
2249 *    above are the tree nodes sorted by increasing frequency.
2250 * OUT assertions: the field len is set to the optimal bit length, the
2251 *     array bl_count contains the frequencies for each bit length.
2252 *     The length opt_len is updated; static_len is also updated if stree is
2253 *     not null.
2254 */
2255local void gen_bitlen(s, desc)
2256    deflate_state *s;
2257    tree_desc *desc;    /* the tree descriptor */
2258{
2259    ct_data *tree  = desc->dyn_tree;
2260    int max_code   = desc->max_code;
2261    ct_data *stree = desc->stat_desc->static_tree;
2262    intf *extra    = desc->stat_desc->extra_bits;
2263    int base       = desc->stat_desc->extra_base;
2264    int max_length = desc->stat_desc->max_length;
2265    int h;              /* heap index */
2266    int n, m;           /* iterate over the tree elements */
2267    int bits;           /* bit length */
2268    int xbits;          /* extra bits */
2269    ush f;              /* frequency */
2270    int overflow = 0;   /* number of elements with bit length too large */
2271
2272    for (bits = 0; bits <= MAX_BITS; bits++) s->bl_count[bits] = 0;
2273
2274    /* In a first pass, compute the optimal bit lengths (which may
2275     * overflow in the case of the bit length tree).
2276     */
2277    tree[s->heap[s->heap_max]].Len = 0; /* root of the heap */
2278
2279    for (h = s->heap_max+1; h < HEAP_SIZE; h++) {
2280        n = s->heap[h];
2281        bits = tree[tree[n].Dad].Len + 1;
2282        if (bits > max_length) bits = max_length, overflow++;
2283        tree[n].Len = (ush)bits;
2284        /* We overwrite tree[n].Dad which is no longer needed */
2285
2286        if (n > max_code) continue; /* not a leaf node */
2287
2288        s->bl_count[bits]++;
2289        xbits = 0;
2290        if (n >= base) xbits = extra[n-base];
2291        f = tree[n].Freq;
2292        s->opt_len += (ulg)f * (bits + xbits);
2293        if (stree) s->static_len += (ulg)f * (stree[n].Len + xbits);
2294    }
2295    if (overflow == 0) return;
2296
2297    Trace((stderr,"\nbit length overflow\n"));
2298    /* This happens for example on obj2 and pic of the Calgary corpus */
2299
2300    /* Find the first bit length which could increase: */
2301    do {
2302        bits = max_length-1;
2303        while (s->bl_count[bits] == 0) bits--;
2304        s->bl_count[bits]--;      /* move one leaf down the tree */
2305        s->bl_count[bits+1] += 2; /* move one overflow item as its brother */
2306        s->bl_count[max_length]--;
2307        /* The brother of the overflow item also moves one step up,
2308         * but this does not affect bl_count[max_length]
2309         */
2310        overflow -= 2;
2311    } while (overflow > 0);
2312
2313    /* Now recompute all bit lengths, scanning in increasing frequency.
2314     * h is still equal to HEAP_SIZE. (It is simpler to reconstruct all
2315     * lengths instead of fixing only the wrong ones. This idea is taken
2316     * from 'ar' written by Haruhiko Okumura.)
2317     */
2318    for (bits = max_length; bits != 0; bits--) {
2319        n = s->bl_count[bits];
2320        while (n != 0) {
2321            m = s->heap[--h];
2322            if (m > max_code) continue;
2323            if (tree[m].Len != (unsigned) bits) {
2324                Trace((stderr,"code %d bits %d->%d\n", m, tree[m].Len, bits));
2325                s->opt_len += ((long)bits - (long)tree[m].Len)
2326                              *(long)tree[m].Freq;
2327                tree[m].Len = (ush)bits;
2328            }
2329            n--;
2330        }
2331    }
2332}
2333
2334/* ===========================================================================
2335 * Generate the codes for a given tree and bit counts (which need not be
2336 * optimal).
2337 * IN assertion: the array bl_count contains the bit length statistics for
2338 * the given tree and the field len is set for all tree elements.
2339 * OUT assertion: the field code is set for all tree elements of non
2340 *     zero code length.
2341 */
2342local void gen_codes (tree, max_code, bl_count)
2343    ct_data *tree;             /* the tree to decorate */
2344    int max_code;              /* largest code with non zero frequency */
2345    ushf *bl_count;            /* number of codes at each bit length */
2346{
2347    ush next_code[MAX_BITS+1]; /* next code value for each bit length */
2348    ush code = 0;              /* running code value */
2349    int bits;                  /* bit index */
2350    int n;                     /* code index */
2351
2352    /* The distribution counts are first used to generate the code values
2353     * without bit reversal.
2354     */
2355    for (bits = 1; bits <= MAX_BITS; bits++) {
2356        next_code[bits] = code = (code + bl_count[bits-1]) << 1;
2357    }
2358    /* Check that the bit counts in bl_count are consistent. The last code
2359     * must be all ones.
2360     */
2361    Assert (code + bl_count[MAX_BITS]-1 == (1<<MAX_BITS)-1,
2362            "inconsistent bit counts");
2363    Tracev((stderr,"\ngen_codes: max_code %d ", max_code));
2364
2365    for (n = 0;  n <= max_code; n++) {
2366        int len = tree[n].Len;
2367        if (len == 0) continue;
2368        /* Now reverse the bits */
2369        tree[n].Code = bi_reverse(next_code[len]++, len);
2370
2371        Tracecv(tree != static_ltree, (stderr,"\nn %3d %c l %2d c %4x (%x) ",
2372             n, (isgraph(n) ? n : ' '), len, tree[n].Code, next_code[len]-1));
2373    }
2374}
2375
2376/* ===========================================================================
2377 * Construct one Huffman tree and assigns the code bit strings and lengths.
2378 * Update the total bit length for the current block.
2379 * IN assertion: the field freq is set for all tree elements.
2380 * OUT assertions: the fields len and code are set to the optimal bit length
2381 *     and corresponding code. The length opt_len is updated; static_len is
2382 *     also updated if stree is not null. The field max_code is set.
2383 */
2384local void build_tree(s, desc)
2385    deflate_state *s;
2386    tree_desc *desc; /* the tree descriptor */
2387{
2388    ct_data *tree   = desc->dyn_tree;
2389    ct_data *stree  = desc->stat_desc->static_tree;
2390    int elems       = desc->stat_desc->elems;
2391    int n, m;          /* iterate over heap elements */
2392    int max_code = -1; /* largest code with non zero frequency */
2393    int node;          /* new node being created */
2394
2395    /* Construct the initial heap, with least frequent element in
2396     * heap[SMALLEST]. The sons of heap[n] are heap[2*n] and heap[2*n+1].
2397     * heap[0] is not used.
2398     */
2399    s->heap_len = 0, s->heap_max = HEAP_SIZE;
2400
2401    for (n = 0; n < elems; n++) {
2402        if (tree[n].Freq != 0) {
2403            s->heap[++(s->heap_len)] = max_code = n;
2404            s->depth[n] = 0;
2405        } else {
2406            tree[n].Len = 0;
2407        }
2408    }
2409
2410    /* The pkzip format requires that at least one distance code exists,
2411     * and that at least one bit should be sent even if there is only one
2412     * possible code. So to avoid special checks later on we force at least
2413     * two codes of non zero frequency.
2414     */
2415    while (s->heap_len < 2) {
2416        node = s->heap[++(s->heap_len)] = (max_code < 2 ? ++max_code : 0);
2417        tree[node].Freq = 1;
2418        s->depth[node] = 0;
2419        s->opt_len--; if (stree) s->static_len -= stree[node].Len;
2420        /* node is 0 or 1 so it does not have extra bits */
2421    }
2422    desc->max_code = max_code;
2423
2424    /* The elements heap[heap_len/2+1 .. heap_len] are leaves of the tree,
2425     * establish sub-heaps of increasing lengths:
2426     */
2427    for (n = s->heap_len/2; n >= 1; n--) pqdownheap(s, tree, n);
2428
2429    /* Construct the Huffman tree by repeatedly combining the least two
2430     * frequent nodes.
2431     */
2432    node = elems;              /* next internal node of the tree */
2433    do {
2434        pqremove(s, tree, n);  /* n = node of least frequency */
2435        m = s->heap[SMALLEST]; /* m = node of next least frequency */
2436
2437        s->heap[--(s->heap_max)] = n; /* keep the nodes sorted by frequency */
2438        s->heap[--(s->heap_max)] = m;
2439
2440        /* Create a new node father of n and m */
2441        tree[node].Freq = tree[n].Freq + tree[m].Freq;
2442        s->depth[node] = (uch) (MAX(s->depth[n], s->depth[m]) + 1);
2443        tree[n].Dad = tree[m].Dad = (ush)node;
2444#ifdef DUMP_BL_TREE
2445        if (tree == s->bl_tree) {
2446            fprintf(stderr,"\nnode %d(%d), sons %d(%d) %d(%d)",
2447                    node, tree[node].Freq, n, tree[n].Freq, m, tree[m].Freq);
2448        }
2449#endif
2450        /* and insert the new node in the heap */
2451        s->heap[SMALLEST] = node++;
2452        pqdownheap(s, tree, SMALLEST);
2453
2454    } while (s->heap_len >= 2);
2455
2456    s->heap[--(s->heap_max)] = s->heap[SMALLEST];
2457
2458    /* At this point, the fields freq and dad are set. We can now
2459     * generate the bit lengths.
2460     */
2461    gen_bitlen(s, (tree_desc *)desc);
2462
2463    /* The field len is now set, we can generate the bit codes */
2464    gen_codes ((ct_data *)tree, max_code, s->bl_count);
2465}
2466
2467/* ===========================================================================
2468 * Scan a literal or distance tree to determine the frequencies of the codes
2469 * in the bit length tree.
2470 */
2471local void scan_tree (s, tree, max_code)
2472    deflate_state *s;
2473    ct_data *tree;   /* the tree to be scanned */
2474    int max_code;    /* and its largest code of non zero frequency */
2475{
2476    int n;                     /* iterates over all tree elements */
2477    int prevlen = -1;          /* last emitted length */
2478    int curlen;                /* length of current code */
2479    int nextlen = tree[0].Len; /* length of next code */
2480    int count = 0;             /* repeat count of the current code */
2481    int max_count = 7;         /* max repeat count */
2482    int min_count = 4;         /* min repeat count */
2483
2484    if (nextlen == 0) max_count = 138, min_count = 3;
2485    tree[max_code+1].Len = (ush)0xffff; /* guard */
2486
2487    for (n = 0; n <= max_code; n++) {
2488        curlen = nextlen; nextlen = tree[n+1].Len;
2489        if (++count < max_count && curlen == nextlen) {
2490            continue;
2491        } else if (count < min_count) {
2492            s->bl_tree[curlen].Freq += count;
2493        } else if (curlen != 0) {
2494            if (curlen != prevlen) s->bl_tree[curlen].Freq++;
2495            s->bl_tree[REP_3_6].Freq++;
2496        } else if (count <= 10) {
2497            s->bl_tree[REPZ_3_10].Freq++;
2498        } else {
2499            s->bl_tree[REPZ_11_138].Freq++;
2500        }
2501        count = 0; prevlen = curlen;
2502        if (nextlen == 0) {
2503            max_count = 138, min_count = 3;
2504        } else if (curlen == nextlen) {
2505            max_count = 6, min_count = 3;
2506        } else {
2507            max_count = 7, min_count = 4;
2508        }
2509    }
2510}
2511
2512/* ===========================================================================
2513 * Send a literal or distance tree in compressed form, using the codes in
2514 * bl_tree.
2515 */
2516local void send_tree (s, tree, max_code)
2517    deflate_state *s;
2518    ct_data *tree; /* the tree to be scanned */
2519    int max_code;       /* and its largest code of non zero frequency */
2520{
2521    int n;                     /* iterates over all tree elements */
2522    int prevlen = -1;          /* last emitted length */
2523    int curlen;                /* length of current code */
2524    int nextlen = tree[0].Len; /* length of next code */
2525    int count = 0;             /* repeat count of the current code */
2526    int max_count = 7;         /* max repeat count */
2527    int min_count = 4;         /* min repeat count */
2528
2529    /* tree[max_code+1].Len = -1; */  /* guard already set */
2530    if (nextlen == 0) max_count = 138, min_count = 3;
2531
2532    for (n = 0; n <= max_code; n++) {
2533        curlen = nextlen; nextlen = tree[n+1].Len;
2534        if (++count < max_count && curlen == nextlen) {
2535            continue;
2536        } else if (count < min_count) {
2537            do { send_code(s, curlen, s->bl_tree); } while (--count != 0);
2538
2539        } else if (curlen != 0) {
2540            if (curlen != prevlen) {
2541                send_code(s, curlen, s->bl_tree); count--;
2542            }
2543            Assert(count >= 3 && count <= 6, " 3_6?");
2544            send_code(s, REP_3_6, s->bl_tree); send_bits(s, count-3, 2);
2545
2546        } else if (count <= 10) {
2547            send_code(s, REPZ_3_10, s->bl_tree); send_bits(s, count-3, 3);
2548
2549        } else {
2550            send_code(s, REPZ_11_138, s->bl_tree); send_bits(s, count-11, 7);
2551        }
2552        count = 0; prevlen = curlen;
2553        if (nextlen == 0) {
2554            max_count = 138, min_count = 3;
2555        } else if (curlen == nextlen) {
2556            max_count = 6, min_count = 3;
2557        } else {
2558            max_count = 7, min_count = 4;
2559        }
2560    }
2561}
2562
2563/* ===========================================================================
2564 * Construct the Huffman tree for the bit lengths and return the index in
2565 * bl_order of the last bit length code to send.
2566 */
2567local int build_bl_tree(s)
2568    deflate_state *s;
2569{
2570    int max_blindex;  /* index of last bit length code of non zero freq */
2571
2572    /* Determine the bit length frequencies for literal and distance trees */
2573    scan_tree(s, (ct_data *)s->dyn_ltree, s->l_desc.max_code);
2574    scan_tree(s, (ct_data *)s->dyn_dtree, s->d_desc.max_code);
2575
2576    /* Build the bit length tree: */
2577    build_tree(s, (tree_desc *)(&(s->bl_desc)));
2578    /* opt_len now includes the length of the tree representations, except
2579     * the lengths of the bit lengths codes and the 5+5+4 bits for the counts.
2580     */
2581
2582    /* Determine the number of bit length codes to send. The pkzip format
2583     * requires that at least 4 bit length codes be sent. (appnote.txt says
2584     * 3 but the actual value used is 4.)
2585     */
2586    for (max_blindex = BL_CODES-1; max_blindex >= 3; max_blindex--) {
2587        if (s->bl_tree[bl_order[max_blindex]].Len != 0) break;
2588    }
2589    /* Update opt_len to include the bit length tree and counts */
2590    s->opt_len += 3*(max_blindex+1) + 5+5+4;
2591    Tracev((stderr, "\ndyn trees: dyn %ld, stat %ld",
2592            s->opt_len, s->static_len));
2593
2594    return max_blindex;
2595}
2596
2597/* ===========================================================================
2598 * Send the header for a block using dynamic Huffman trees: the counts, the
2599 * lengths of the bit length codes, the literal tree and the distance tree.
2600 * IN assertion: lcodes >= 257, dcodes >= 1, blcodes >= 4.
2601 */
2602local void send_all_trees(s, lcodes, dcodes, blcodes)
2603    deflate_state *s;
2604    int lcodes, dcodes, blcodes; /* number of codes for each tree */
2605{
2606    int rank;                    /* index in bl_order */
2607
2608    Assert (lcodes >= 257 && dcodes >= 1 && blcodes >= 4, "not enough codes");
2609    Assert (lcodes <= L_CODES && dcodes <= D_CODES && blcodes <= BL_CODES,
2610            "too many codes");
2611    Tracev((stderr, "\nbl counts: "));
2612    send_bits(s, lcodes-257, 5); /* not +255 as stated in appnote.txt */
2613    send_bits(s, dcodes-1,   5);
2614    send_bits(s, blcodes-4,  4); /* not -3 as stated in appnote.txt */
2615    for (rank = 0; rank < blcodes; rank++) {
2616        Tracev((stderr, "\nbl code %2d ", bl_order[rank]));
2617        send_bits(s, s->bl_tree[bl_order[rank]].Len, 3);
2618    }
2619    Tracev((stderr, "\nbl tree: sent %ld", s->bits_sent));
2620
2621    send_tree(s, (ct_data *)s->dyn_ltree, lcodes-1); /* literal tree */
2622    Tracev((stderr, "\nlit tree: sent %ld", s->bits_sent));
2623
2624    send_tree(s, (ct_data *)s->dyn_dtree, dcodes-1); /* distance tree */
2625    Tracev((stderr, "\ndist tree: sent %ld", s->bits_sent));
2626}
2627
2628/* ===========================================================================
2629 * Send a stored block
2630 */
2631void _tr_stored_block(s, buf, stored_len, eof)
2632    deflate_state *s;
2633    charf *buf;       /* input block */
2634    ulg stored_len;   /* length of input block */
2635    int eof;          /* true if this is the last block for a file */
2636{
2637    send_bits(s, (STORED_BLOCK<<1)+eof, 3);  /* send block type */
2638    s->compressed_len = (s->compressed_len + 3 + 7) & (ulg)~7L;
2639    s->compressed_len += (stored_len + 4) << 3;
2640
2641    copy_block(s, buf, (unsigned)stored_len, 1); /* with header */
2642}
2643
2644/* Send just the `stored block' type code without any length bytes or data.
2645 */
2646void _tr_stored_type_only(s)
2647    deflate_state *s;
2648{
2649    send_bits(s, (STORED_BLOCK << 1), 3);
2650    bi_windup(s);
2651    s->compressed_len = (s->compressed_len + 3) & ~7L;
2652}
2653
2654
2655/* ===========================================================================
2656 * Send one empty static block to give enough lookahead for inflate.
2657 * This takes 10 bits, of which 7 may remain in the bit buffer.
2658 * The current inflate code requires 9 bits of lookahead. If the
2659 * last two codes for the previous block (real code plus EOB) were coded
2660 * on 5 bits or less, inflate may have only 5+3 bits of lookahead to decode
2661 * the last real code. In this case we send two empty static blocks instead
2662 * of one. (There are no problems if the previous block is stored or fixed.)
2663 * To simplify the code, we assume the worst case of last real code encoded
2664 * on one bit only.
2665 */
2666void _tr_align(s)
2667    deflate_state *s;
2668{
2669    send_bits(s, STATIC_TREES<<1, 3);
2670    send_code(s, END_BLOCK, static_ltree);
2671    s->compressed_len += 10L; /* 3 for block type, 7 for EOB */
2672    bi_flush(s);
2673    /* Of the 10 bits for the empty block, we have already sent
2674     * (10 - bi_valid) bits. The lookahead for the last real code (before
2675     * the EOB of the previous block) was thus at least one plus the length
2676     * of the EOB plus what we have just sent of the empty static block.
2677     */
2678    if (1 + s->last_eob_len + 10 - s->bi_valid < 9) {
2679        send_bits(s, STATIC_TREES<<1, 3);
2680        send_code(s, END_BLOCK, static_ltree);
2681        s->compressed_len += 10L;
2682        bi_flush(s);
2683    }
2684    s->last_eob_len = 7;
2685}
2686
2687/* ===========================================================================
2688 * Determine the best encoding for the current block: dynamic trees, static
2689 * trees or store, and output the encoded block to the zip file. This function
2690 * returns the total compressed length for the file so far.
2691 */
2692ulg _tr_flush_block(s, buf, stored_len, eof)
2693    deflate_state *s;
2694    charf *buf;       /* input block, or NULL if too old */
2695    ulg stored_len;   /* length of input block */
2696    int eof;          /* true if this is the last block for a file */
2697{
2698    ulg opt_lenb, static_lenb; /* opt_len and static_len in bytes */
2699    int max_blindex = 0;  /* index of last bit length code of non zero freq */
2700
2701    /* Build the Huffman trees unless a stored block is forced */
2702    if (s->level > 0) {
2703
2704	 /* Check if the file is ascii or binary */
2705	if (s->data_type == Z_UNKNOWN) set_data_type(s);
2706
2707	/* Construct the literal and distance trees */
2708	build_tree(s, (tree_desc *)(&(s->l_desc)));
2709	Tracev((stderr, "\nlit data: dyn %ld, stat %ld", s->opt_len,
2710		s->static_len));
2711
2712	build_tree(s, (tree_desc *)(&(s->d_desc)));
2713	Tracev((stderr, "\ndist data: dyn %ld, stat %ld", s->opt_len,
2714		s->static_len));
2715	/* At this point, opt_len and static_len are the total bit lengths of
2716	 * the compressed block data, excluding the tree representations.
2717	 */
2718
2719	/* Build the bit length tree for the above two trees, and get the index
2720	 * in bl_order of the last bit length code to send.
2721	 */
2722	max_blindex = build_bl_tree(s);
2723
2724	/* Determine the best encoding. Compute first the block length in bytes*/
2725	opt_lenb = (s->opt_len+3+7)>>3;
2726	static_lenb = (s->static_len+3+7)>>3;
2727
2728	Tracev((stderr, "\nopt %lu(%lu) stat %lu(%lu) stored %lu lit %u ",
2729		opt_lenb, s->opt_len, static_lenb, s->static_len, stored_len,
2730		s->last_lit));
2731
2732	if (static_lenb <= opt_lenb) opt_lenb = static_lenb;
2733
2734    } else {
2735        Assert(buf != (char*)0, "lost buf");
2736	opt_lenb = static_lenb = stored_len + 5; /* force a stored block */
2737    }
2738
2739    /* If compression failed and this is the first and last block,
2740     * and if the .zip file can be seeked (to rewrite the local header),
2741     * the whole file is transformed into a stored file:
2742     */
2743#ifdef STORED_FILE_OK
2744#  ifdef FORCE_STORED_FILE
2745    if (eof && s->compressed_len == 0L) { /* force stored file */
2746#  else
2747    if (stored_len <= opt_lenb && eof && s->compressed_len==0L && seekable()) {
2748#  endif
2749        /* Since LIT_BUFSIZE <= 2*WSIZE, the input data must be there: */
2750        if (buf == (charf*)0) error ("block vanished");
2751
2752        copy_block(s, buf, (unsigned)stored_len, 0); /* without header */
2753        s->compressed_len = stored_len << 3;
2754        s->method = STORED;
2755    } else
2756#endif /* STORED_FILE_OK */
2757
2758#ifdef FORCE_STORED
2759    if (buf != (char*)0) { /* force stored block */
2760#else
2761    if (stored_len+4 <= opt_lenb && buf != (char*)0) {
2762                       /* 4: two words for the lengths */
2763#endif
2764        /* The test buf != NULL is only necessary if LIT_BUFSIZE > WSIZE.
2765         * Otherwise we can't have processed more than WSIZE input bytes since
2766         * the last block flush, because compression would have been
2767         * successful. If LIT_BUFSIZE <= WSIZE, it is never too late to
2768         * transform a block into a stored block.
2769         */
2770        _tr_stored_block(s, buf, stored_len, eof);
2771
2772#ifdef FORCE_STATIC
2773    } else if (static_lenb >= 0) { /* force static trees */
2774#else
2775    } else if (static_lenb == opt_lenb) {
2776#endif
2777        send_bits(s, (STATIC_TREES<<1)+eof, 3);
2778        compress_block(s, (ct_data *)static_ltree, (ct_data *)static_dtree);
2779        s->compressed_len += 3 + s->static_len;
2780    } else {
2781        send_bits(s, (DYN_TREES<<1)+eof, 3);
2782        send_all_trees(s, s->l_desc.max_code+1, s->d_desc.max_code+1,
2783                       max_blindex+1);
2784        compress_block(s, (ct_data *)s->dyn_ltree, (ct_data *)s->dyn_dtree);
2785        s->compressed_len += 3 + s->opt_len;
2786    }
2787    Assert (s->compressed_len == s->bits_sent, "bad compressed size");
2788    init_block(s);
2789
2790    if (eof) {
2791        bi_windup(s);
2792        s->compressed_len += 7;  /* align on byte boundary */
2793    }
2794    Tracev((stderr,"\ncomprlen %lu(%lu) ", s->compressed_len>>3,
2795           s->compressed_len-7*eof));
2796
2797    return s->compressed_len >> 3;
2798}
2799
2800/* ===========================================================================
2801 * Save the match info and tally the frequency counts. Return true if
2802 * the current block must be flushed.
2803 */
2804int _tr_tally (s, dist, lc)
2805    deflate_state *s;
2806    unsigned dist;  /* distance of matched string */
2807    unsigned lc;    /* match length-MIN_MATCH or unmatched char (if dist==0) */
2808{
2809    s->d_buf[s->last_lit] = (ush)dist;
2810    s->l_buf[s->last_lit++] = (uch)lc;
2811    if (dist == 0) {
2812        /* lc is the unmatched char */
2813        s->dyn_ltree[lc].Freq++;
2814    } else {
2815        s->matches++;
2816        /* Here, lc is the match length - MIN_MATCH */
2817        dist--;             /* dist = match distance - 1 */
2818        Assert((ush)dist < (ush)MAX_DIST(s) &&
2819               (ush)lc <= (ush)(MAX_MATCH-MIN_MATCH) &&
2820               (ush)d_code(dist) < (ush)D_CODES,  "_tr_tally: bad match");
2821
2822        s->dyn_ltree[length_code[lc]+LITERALS+1].Freq++;
2823        s->dyn_dtree[d_code(dist)].Freq++;
2824    }
2825
2826    /* Try to guess if it is profitable to stop the current block here */
2827    if (s->level > 2 && (s->last_lit & 0xfff) == 0) {
2828        /* Compute an upper bound for the compressed length */
2829        ulg out_length = (ulg)s->last_lit*8L;
2830        ulg in_length = (ulg)((long)s->strstart - s->block_start);
2831        int dcode;
2832        for (dcode = 0; dcode < D_CODES; dcode++) {
2833            out_length += (ulg)s->dyn_dtree[dcode].Freq *
2834                (5L+extra_dbits[dcode]);
2835        }
2836        out_length >>= 3;
2837        Tracev((stderr,"\nlast_lit %u, in %ld, out ~%ld(%ld%%) ",
2838               s->last_lit, in_length, out_length,
2839               100L - out_length*100L/in_length));
2840        if (s->matches < s->last_lit/2 && out_length < in_length/2) return 1;
2841    }
2842    return (s->last_lit == s->lit_bufsize-1);
2843    /* We avoid equality with lit_bufsize because of wraparound at 64K
2844     * on 16 bit machines and because stored blocks are restricted to
2845     * 64K-1 bytes.
2846     */
2847}
2848
2849/* ===========================================================================
2850 * Send the block data compressed using the given Huffman trees
2851 */
2852local void compress_block(s, ltree, dtree)
2853    deflate_state *s;
2854    ct_data *ltree; /* literal tree */
2855    ct_data *dtree; /* distance tree */
2856{
2857    unsigned dist;      /* distance of matched string */
2858    int lc;             /* match length or unmatched char (if dist == 0) */
2859    unsigned lx = 0;    /* running index in l_buf */
2860    unsigned code;      /* the code to send */
2861    int extra;          /* number of extra bits to send */
2862
2863    if (s->last_lit != 0) do {
2864        dist = s->d_buf[lx];
2865        lc = s->l_buf[lx++];
2866        if (dist == 0) {
2867            send_code(s, lc, ltree); /* send a literal byte */
2868            Tracecv(isgraph(lc), (stderr," '%c' ", lc));
2869        } else {
2870            /* Here, lc is the match length - MIN_MATCH */
2871            code = length_code[lc];
2872            send_code(s, code+LITERALS+1, ltree); /* send the length code */
2873            extra = extra_lbits[code];
2874            if (extra != 0) {
2875                lc -= base_length[code];
2876                send_bits(s, lc, extra);       /* send the extra length bits */
2877            }
2878            dist--; /* dist is now the match distance - 1 */
2879            code = d_code(dist);
2880            Assert (code < D_CODES, "bad d_code");
2881
2882            send_code(s, code, dtree);       /* send the distance code */
2883            extra = extra_dbits[code];
2884            if (extra != 0) {
2885                dist -= base_dist[code];
2886                send_bits(s, dist, extra);   /* send the extra distance bits */
2887            }
2888        } /* literal or match pair ? */
2889
2890        /* Check that the overlay between pending_buf and d_buf+l_buf is ok: */
2891        Assert(s->pending < s->lit_bufsize + 2*lx, "pendingBuf overflow");
2892
2893    } while (lx < s->last_lit);
2894
2895    send_code(s, END_BLOCK, ltree);
2896    s->last_eob_len = ltree[END_BLOCK].Len;
2897}
2898
2899/* ===========================================================================
2900 * Set the data type to ASCII or BINARY, using a crude approximation:
2901 * binary if more than 20% of the bytes are <= 6 or >= 128, ascii otherwise.
2902 * IN assertion: the fields freq of dyn_ltree are set and the total of all
2903 * frequencies does not exceed 64K (to fit in an int on 16 bit machines).
2904 */
2905local void set_data_type(s)
2906    deflate_state *s;
2907{
2908    int n = 0;
2909    unsigned ascii_freq = 0;
2910    unsigned bin_freq = 0;
2911    while (n < 7)        bin_freq += s->dyn_ltree[n++].Freq;
2912    while (n < 128)    ascii_freq += s->dyn_ltree[n++].Freq;
2913    while (n < LITERALS) bin_freq += s->dyn_ltree[n++].Freq;
2914    s->data_type = (Byte)(bin_freq > (ascii_freq >> 2) ? Z_BINARY : Z_ASCII);
2915}
2916
2917/* ===========================================================================
2918 * Reverse the first len bits of a code, using straightforward code (a faster
2919 * method would use a table)
2920 * IN assertion: 1 <= len <= 15
2921 */
2922local unsigned bi_reverse(code, len)
2923    unsigned code; /* the value to invert */
2924    int len;       /* its bit length */
2925{
2926    register unsigned res = 0;
2927    do {
2928        res |= code & 1;
2929        code >>= 1, res <<= 1;
2930    } while (--len > 0);
2931    return res >> 1;
2932}
2933
2934/* ===========================================================================
2935 * Flush the bit buffer, keeping at most 7 bits in it.
2936 */
2937local void bi_flush(s)
2938    deflate_state *s;
2939{
2940    if (s->bi_valid == 16) {
2941        put_short(s, s->bi_buf);
2942        s->bi_buf = 0;
2943        s->bi_valid = 0;
2944    } else if (s->bi_valid >= 8) {
2945        put_byte(s, (Byte)s->bi_buf);
2946        s->bi_buf >>= 8;
2947        s->bi_valid -= 8;
2948    }
2949}
2950
2951/* ===========================================================================
2952 * Flush the bit buffer and align the output on a byte boundary
2953 */
2954local void bi_windup(s)
2955    deflate_state *s;
2956{
2957    if (s->bi_valid > 8) {
2958        put_short(s, s->bi_buf);
2959    } else if (s->bi_valid > 0) {
2960        put_byte(s, (Byte)s->bi_buf);
2961    }
2962    s->bi_buf = 0;
2963    s->bi_valid = 0;
2964#ifdef DEBUG_ZLIB
2965    s->bits_sent = (s->bits_sent+7) & ~7;
2966#endif
2967}
2968
2969/* ===========================================================================
2970 * Copy a stored block, storing first the length and its
2971 * one's complement if requested.
2972 */
2973local void copy_block(s, buf, len, header)
2974    deflate_state *s;
2975    charf    *buf;    /* the input data */
2976    unsigned len;     /* its length */
2977    int      header;  /* true if block header must be written */
2978{
2979    bi_windup(s);        /* align on byte boundary */
2980    s->last_eob_len = 8; /* enough lookahead for inflate */
2981
2982    if (header) {
2983        put_short(s, (ush)len);
2984        put_short(s, (ush)~len);
2985#ifdef DEBUG_ZLIB
2986        s->bits_sent += 2*16;
2987#endif
2988    }
2989#ifdef DEBUG_ZLIB
2990    s->bits_sent += (ulg)len<<3;
2991#endif
2992    /* bundle up the put_byte(s, *buf++) calls */
2993    zmemcpy(&s->pending_buf[s->pending], buf, len);
2994    s->pending += len;
2995}
2996/* --- trees.c */
2997
2998/* +++ inflate.c */
2999/* inflate.c -- zlib interface to inflate modules
3000 * Copyright (C) 1995-1996 Mark Adler
3001 * For conditions of distribution and use, see copyright notice in zlib.h
3002 */
3003
3004/* #include "zutil.h" */
3005
3006/* +++ infblock.h */
3007/* infblock.h -- header to use infblock.c
3008 * Copyright (C) 1995-1996 Mark Adler
3009 * For conditions of distribution and use, see copyright notice in zlib.h
3010 */
3011
3012/* WARNING: this file should *not* be used by applications. It is
3013   part of the implementation of the compression library and is
3014   subject to change. Applications should only use zlib.h.
3015 */
3016
3017struct inflate_blocks_state;
3018typedef struct inflate_blocks_state FAR inflate_blocks_statef;
3019
3020extern inflate_blocks_statef * inflate_blocks_new OF((
3021    z_streamp z,
3022    check_func c,               /* check function */
3023    uInt w));                   /* window size */
3024
3025extern int inflate_blocks OF((
3026    inflate_blocks_statef *,
3027    z_streamp ,
3028    int));                      /* initial return code */
3029
3030extern void inflate_blocks_reset OF((
3031    inflate_blocks_statef *,
3032    z_streamp ,
3033    uLongf *));                  /* check value on output */
3034
3035extern int inflate_blocks_free OF((
3036    inflate_blocks_statef *,
3037    z_streamp ,
3038    uLongf *));                  /* check value on output */
3039
3040extern void inflate_set_dictionary OF((
3041    inflate_blocks_statef *s,
3042    const Bytef *d,  /* dictionary */
3043    uInt  n));       /* dictionary length */
3044
3045extern int inflate_addhistory OF((
3046    inflate_blocks_statef *,
3047    z_streamp));
3048
3049extern int inflate_packet_flush OF((
3050    inflate_blocks_statef *));
3051/* --- infblock.h */
3052
3053#ifndef NO_DUMMY_DECL
3054struct inflate_blocks_state {int dummy;}; /* for buggy compilers */
3055#endif
3056
3057/* inflate private state */
3058struct internal_state {
3059
3060  /* mode */
3061  enum {
3062      METHOD,   /* waiting for method byte */
3063      FLAG,     /* waiting for flag byte */
3064      DICT4,    /* four dictionary check bytes to go */
3065      DICT3,    /* three dictionary check bytes to go */
3066      DICT2,    /* two dictionary check bytes to go */
3067      DICT1,    /* one dictionary check byte to go */
3068      DICT0,    /* waiting for inflateSetDictionary */
3069      BLOCKS,   /* decompressing blocks */
3070      CHECK4,   /* four check bytes to go */
3071      CHECK3,   /* three check bytes to go */
3072      CHECK2,   /* two check bytes to go */
3073      CHECK1,   /* one check byte to go */
3074      DONE,     /* finished check, done */
3075      BAD}      /* got an error--stay here */
3076    mode;               /* current inflate mode */
3077
3078  /* mode dependent information */
3079  union {
3080    uInt method;        /* if FLAGS, method byte */
3081    struct {
3082      uLong was;                /* computed check value */
3083      uLong need;               /* stream check value */
3084    } check;            /* if CHECK, check values to compare */
3085    uInt marker;        /* if BAD, inflateSync's marker bytes count */
3086  } sub;        /* submode */
3087
3088  /* mode independent information */
3089  int  nowrap;          /* flag for no wrapper */
3090  uInt wbits;           /* log2(window size)  (8..15, defaults to 15) */
3091  inflate_blocks_statef
3092    *blocks;            /* current inflate_blocks state */
3093
3094};
3095
3096
3097int inflateReset(z)
3098z_streamp z;
3099{
3100  uLong c;
3101
3102  if (z == Z_NULL || z->state == Z_NULL)
3103    return Z_STREAM_ERROR;
3104  z->total_in = z->total_out = 0;
3105  z->msg = Z_NULL;
3106  z->state->mode = z->state->nowrap ? BLOCKS : METHOD;
3107  inflate_blocks_reset(z->state->blocks, z, &c);
3108  Trace((stderr, "inflate: reset\n"));
3109  return Z_OK;
3110}
3111
3112
3113int inflateEnd(z)
3114z_streamp z;
3115{
3116  uLong c;
3117
3118  if (z == Z_NULL || z->state == Z_NULL || z->zfree == Z_NULL)
3119    return Z_STREAM_ERROR;
3120  if (z->state->blocks != Z_NULL)
3121    inflate_blocks_free(z->state->blocks, z, &c);
3122  ZFREE(z, z->state);
3123  z->state = Z_NULL;
3124  Trace((stderr, "inflate: end\n"));
3125  return Z_OK;
3126}
3127
3128
3129int inflateInit2_(z, w, version, stream_size)
3130z_streamp z;
3131int w;
3132const char *version;
3133int stream_size;
3134{
3135  if (version == Z_NULL || version[0] != ZLIB_VERSION[0] ||
3136      stream_size != sizeof(z_stream))
3137      return Z_VERSION_ERROR;
3138
3139  /* initialize state */
3140  if (z == Z_NULL)
3141    return Z_STREAM_ERROR;
3142  z->msg = Z_NULL;
3143#ifndef NO_ZCFUNCS
3144  if (z->zalloc == Z_NULL)
3145  {
3146    z->zalloc = zcalloc;
3147    z->opaque = (voidpf)0;
3148  }
3149  if (z->zfree == Z_NULL) z->zfree = zcfree;
3150#endif
3151  if ((z->state = (struct internal_state FAR *)
3152       ZALLOC(z,1,sizeof(struct internal_state))) == Z_NULL)
3153    return Z_MEM_ERROR;
3154  z->state->blocks = Z_NULL;
3155
3156  /* handle undocumented nowrap option (no zlib header or check) */
3157  z->state->nowrap = 0;
3158  if (w < 0)
3159  {
3160    w = - w;
3161    z->state->nowrap = 1;
3162  }
3163
3164  /* set window size */
3165  if (w < 8 || w > 15)
3166  {
3167    inflateEnd(z);
3168    return Z_STREAM_ERROR;
3169  }
3170  z->state->wbits = (uInt)w;
3171
3172  /* create inflate_blocks state */
3173  if ((z->state->blocks =
3174      inflate_blocks_new(z, z->state->nowrap ? Z_NULL : adler32, (uInt)1 << w))
3175      == Z_NULL)
3176  {
3177    inflateEnd(z);
3178    return Z_MEM_ERROR;
3179  }
3180  Trace((stderr, "inflate: allocated\n"));
3181
3182  /* reset state */
3183  inflateReset(z);
3184  return Z_OK;
3185}
3186
3187
3188int inflateInit_(z, version, stream_size)
3189z_streamp z;
3190const char *version;
3191int stream_size;
3192{
3193  return inflateInit2_(z, DEF_WBITS, version, stream_size);
3194}
3195
3196
3197#define NEEDBYTE {if(z->avail_in==0)goto empty;r=Z_OK;}
3198#define NEXTBYTE (z->avail_in--,z->total_in++,*z->next_in++)
3199
3200int inflate(z, f)
3201z_streamp z;
3202int f;
3203{
3204  int r;
3205  uInt b;
3206
3207  if (z == Z_NULL || z->state == Z_NULL || z->next_in == Z_NULL || f < 0)
3208    return Z_STREAM_ERROR;
3209  r = Z_BUF_ERROR;
3210  while (1) switch (z->state->mode)
3211  {
3212    case METHOD:
3213      NEEDBYTE
3214      if (((z->state->sub.method = NEXTBYTE) & 0xf) != Z_DEFLATED)
3215      {
3216        z->state->mode = BAD;
3217        z->msg = (char*)"unknown compression method";
3218        z->state->sub.marker = 5;       /* can't try inflateSync */
3219        break;
3220      }
3221      if ((z->state->sub.method >> 4) + 8 > z->state->wbits)
3222      {
3223        z->state->mode = BAD;
3224        z->msg = (char*)"invalid window size";
3225        z->state->sub.marker = 5;       /* can't try inflateSync */
3226        break;
3227      }
3228      z->state->mode = FLAG;
3229    case FLAG:
3230      NEEDBYTE
3231      b = NEXTBYTE;
3232      if (((z->state->sub.method << 8) + b) % 31)
3233      {
3234        z->state->mode = BAD;
3235        z->msg = (char*)"incorrect header check";
3236        z->state->sub.marker = 5;       /* can't try inflateSync */
3237        break;
3238      }
3239      Trace((stderr, "inflate: zlib header ok\n"));
3240      if (!(b & PRESET_DICT))
3241      {
3242        z->state->mode = BLOCKS;
3243	break;
3244      }
3245      z->state->mode = DICT4;
3246    case DICT4:
3247      NEEDBYTE
3248      z->state->sub.check.need = (uLong)NEXTBYTE << 24;
3249      z->state->mode = DICT3;
3250    case DICT3:
3251      NEEDBYTE
3252      z->state->sub.check.need += (uLong)NEXTBYTE << 16;
3253      z->state->mode = DICT2;
3254    case DICT2:
3255      NEEDBYTE
3256      z->state->sub.check.need += (uLong)NEXTBYTE << 8;
3257      z->state->mode = DICT1;
3258    case DICT1:
3259      NEEDBYTE
3260      z->state->sub.check.need += (uLong)NEXTBYTE;
3261      z->adler = z->state->sub.check.need;
3262      z->state->mode = DICT0;
3263      return Z_NEED_DICT;
3264    case DICT0:
3265      z->state->mode = BAD;
3266      z->msg = (char*)"need dictionary";
3267      z->state->sub.marker = 0;       /* can try inflateSync */
3268      return Z_STREAM_ERROR;
3269    case BLOCKS:
3270      r = inflate_blocks(z->state->blocks, z, r);
3271      if (f == Z_PACKET_FLUSH && z->avail_in == 0 && z->avail_out != 0)
3272	  r = inflate_packet_flush(z->state->blocks);
3273      if (r == Z_DATA_ERROR)
3274      {
3275        z->state->mode = BAD;
3276        z->state->sub.marker = 0;       /* can try inflateSync */
3277        break;
3278      }
3279      if (r != Z_STREAM_END)
3280        return r;
3281      r = Z_OK;
3282      inflate_blocks_reset(z->state->blocks, z, &z->state->sub.check.was);
3283      if (z->state->nowrap)
3284      {
3285        z->state->mode = DONE;
3286        break;
3287      }
3288      z->state->mode = CHECK4;
3289    case CHECK4:
3290      NEEDBYTE
3291      z->state->sub.check.need = (uLong)NEXTBYTE << 24;
3292      z->state->mode = CHECK3;
3293    case CHECK3:
3294      NEEDBYTE
3295      z->state->sub.check.need += (uLong)NEXTBYTE << 16;
3296      z->state->mode = CHECK2;
3297    case CHECK2:
3298      NEEDBYTE
3299      z->state->sub.check.need += (uLong)NEXTBYTE << 8;
3300      z->state->mode = CHECK1;
3301    case CHECK1:
3302      NEEDBYTE
3303      z->state->sub.check.need += (uLong)NEXTBYTE;
3304
3305      if (z->state->sub.check.was != z->state->sub.check.need)
3306      {
3307        z->state->mode = BAD;
3308        z->msg = (char*)"incorrect data check";
3309        z->state->sub.marker = 5;       /* can't try inflateSync */
3310        break;
3311      }
3312      Trace((stderr, "inflate: zlib check ok\n"));
3313      z->state->mode = DONE;
3314    case DONE:
3315      return Z_STREAM_END;
3316    case BAD:
3317      return Z_DATA_ERROR;
3318    default:
3319      return Z_STREAM_ERROR;
3320  }
3321
3322 empty:
3323  if (f != Z_PACKET_FLUSH)
3324    return r;
3325  z->state->mode = BAD;
3326  z->msg = (char *)"need more for packet flush";
3327  z->state->sub.marker = 0;       /* can try inflateSync */
3328  return Z_DATA_ERROR;
3329}
3330
3331
3332int inflateSetDictionary(z, dictionary, dictLength)
3333z_streamp z;
3334const Bytef *dictionary;
3335uInt  dictLength;
3336{
3337  uInt length = dictLength;
3338
3339  if (z == Z_NULL || z->state == Z_NULL || z->state->mode != DICT0)
3340    return Z_STREAM_ERROR;
3341
3342  if (adler32(1L, dictionary, dictLength) != z->adler) return Z_DATA_ERROR;
3343  z->adler = 1L;
3344
3345  if (length >= ((uInt)1<<z->state->wbits))
3346  {
3347    length = (1<<z->state->wbits)-1;
3348    dictionary += dictLength - length;
3349  }
3350  inflate_set_dictionary(z->state->blocks, dictionary, length);
3351  z->state->mode = BLOCKS;
3352  return Z_OK;
3353}
3354
3355/*
3356 * This subroutine adds the data at next_in/avail_in to the output history
3357 * without performing any output.  The output buffer must be "caught up";
3358 * i.e. no pending output (hence s->read equals s->write), and the state must
3359 * be BLOCKS (i.e. we should be willing to see the start of a series of
3360 * BLOCKS).  On exit, the output will also be caught up, and the checksum
3361 * will have been updated if need be.
3362 */
3363
3364int inflateIncomp(z)
3365z_stream *z;
3366{
3367    if (z->state->mode != BLOCKS)
3368	return Z_DATA_ERROR;
3369    return inflate_addhistory(z->state->blocks, z);
3370}
3371
3372
3373int inflateSync(z)
3374z_streamp z;
3375{
3376  uInt n;       /* number of bytes to look at */
3377  Bytef *p;     /* pointer to bytes */
3378  uInt m;       /* number of marker bytes found in a row */
3379  uLong r, w;   /* temporaries to save total_in and total_out */
3380
3381  /* set up */
3382  if (z == Z_NULL || z->state == Z_NULL)
3383    return Z_STREAM_ERROR;
3384  if (z->state->mode != BAD)
3385  {
3386    z->state->mode = BAD;
3387    z->state->sub.marker = 0;
3388  }
3389  if ((n = z->avail_in) == 0)
3390    return Z_BUF_ERROR;
3391  p = z->next_in;
3392  m = z->state->sub.marker;
3393
3394  /* search */
3395  while (n && m < 4)
3396  {
3397    if (*p == (Byte)(m < 2 ? 0 : 0xff))
3398      m++;
3399    else if (*p)
3400      m = 0;
3401    else
3402      m = 4 - m;
3403    p++, n--;
3404  }
3405
3406  /* restore */
3407  z->total_in += p - z->next_in;
3408  z->next_in = p;
3409  z->avail_in = n;
3410  z->state->sub.marker = m;
3411
3412  /* return no joy or set up to restart on a new block */
3413  if (m != 4)
3414    return Z_DATA_ERROR;
3415  r = z->total_in;  w = z->total_out;
3416  inflateReset(z);
3417  z->total_in = r;  z->total_out = w;
3418  z->state->mode = BLOCKS;
3419  return Z_OK;
3420}
3421
3422#undef NEEDBYTE
3423#undef NEXTBYTE
3424/* --- inflate.c */
3425
3426/* +++ infblock.c */
3427/* infblock.c -- interpret and process block types to last block
3428 * Copyright (C) 1995-1996 Mark Adler
3429 * For conditions of distribution and use, see copyright notice in zlib.h
3430 */
3431
3432/* #include "zutil.h" */
3433/* #include "infblock.h" */
3434
3435/* +++ inftrees.h */
3436/* inftrees.h -- header to use inftrees.c
3437 * Copyright (C) 1995-1996 Mark Adler
3438 * For conditions of distribution and use, see copyright notice in zlib.h
3439 */
3440
3441/* WARNING: this file should *not* be used by applications. It is
3442   part of the implementation of the compression library and is
3443   subject to change. Applications should only use zlib.h.
3444 */
3445
3446/* Huffman code lookup table entry--this entry is four bytes for machines
3447   that have 16-bit pointers (e.g. PC's in the small or medium model). */
3448
3449typedef struct inflate_huft_s FAR inflate_huft;
3450
3451struct inflate_huft_s {
3452  union {
3453    struct {
3454      Byte Exop;        /* number of extra bits or operation */
3455      Byte Bits;        /* number of bits in this code or subcode */
3456    } what;
3457    Bytef *pad;         /* pad structure to a power of 2 (4 bytes for */
3458  } word;               /*  16-bit, 8 bytes for 32-bit machines) */
3459  union {
3460    uInt Base;          /* literal, length base, or distance base */
3461    inflate_huft *Next; /* pointer to next level of table */
3462  } more;
3463};
3464
3465#ifdef DEBUG_ZLIB
3466  extern uInt inflate_hufts;
3467#endif
3468
3469extern int inflate_trees_bits OF((
3470    uIntf *,                    /* 19 code lengths */
3471    uIntf *,                    /* bits tree desired/actual depth */
3472    inflate_huft * FAR *,       /* bits tree result */
3473    z_streamp ));               /* for zalloc, zfree functions */
3474
3475extern int inflate_trees_dynamic OF((
3476    uInt,                       /* number of literal/length codes */
3477    uInt,                       /* number of distance codes */
3478    uIntf *,                    /* that many (total) code lengths */
3479    uIntf *,                    /* literal desired/actual bit depth */
3480    uIntf *,                    /* distance desired/actual bit depth */
3481    inflate_huft * FAR *,       /* literal/length tree result */
3482    inflate_huft * FAR *,       /* distance tree result */
3483    z_streamp ));               /* for zalloc, zfree functions */
3484
3485extern int inflate_trees_fixed OF((
3486    uIntf *,                    /* literal desired/actual bit depth */
3487    uIntf *,                    /* distance desired/actual bit depth */
3488    inflate_huft * FAR *,       /* literal/length tree result */
3489    inflate_huft * FAR *));     /* distance tree result */
3490
3491extern int inflate_trees_free OF((
3492    inflate_huft *,             /* tables to free */
3493    z_streamp ));               /* for zfree function */
3494
3495/* --- inftrees.h */
3496
3497/* +++ infcodes.h */
3498/* infcodes.h -- header to use infcodes.c
3499 * Copyright (C) 1995-1996 Mark Adler
3500 * For conditions of distribution and use, see copyright notice in zlib.h
3501 */
3502
3503/* WARNING: this file should *not* be used by applications. It is
3504   part of the implementation of the compression library and is
3505   subject to change. Applications should only use zlib.h.
3506 */
3507
3508struct inflate_codes_state;
3509typedef struct inflate_codes_state FAR inflate_codes_statef;
3510
3511extern inflate_codes_statef *inflate_codes_new OF((
3512    uInt, uInt,
3513    inflate_huft *, inflate_huft *,
3514    z_streamp ));
3515
3516extern int inflate_codes OF((
3517    inflate_blocks_statef *,
3518    z_streamp ,
3519    int));
3520
3521extern void inflate_codes_free OF((
3522    inflate_codes_statef *,
3523    z_streamp ));
3524
3525/* --- infcodes.h */
3526
3527/* +++ infutil.h */
3528/* infutil.h -- types and macros common to blocks and codes
3529 * Copyright (C) 1995-1996 Mark Adler
3530 * For conditions of distribution and use, see copyright notice in zlib.h
3531 */
3532
3533/* WARNING: this file should *not* be used by applications. It is
3534   part of the implementation of the compression library and is
3535   subject to change. Applications should only use zlib.h.
3536 */
3537
3538#ifndef _INFUTIL_H
3539#define _INFUTIL_H
3540
3541typedef enum {
3542      TYPE,     /* get type bits (3, including end bit) */
3543      LENS,     /* get lengths for stored */
3544      STORED,   /* processing stored block */
3545      TABLE,    /* get table lengths */
3546      BTREE,    /* get bit lengths tree for a dynamic block */
3547      DTREE,    /* get length, distance trees for a dynamic block */
3548      CODES,    /* processing fixed or dynamic block */
3549      DRY,      /* output remaining window bytes */
3550      DONEB,    /* finished last block, done */
3551      BADB}     /* got a data error--stuck here */
3552inflate_block_mode;
3553
3554/* inflate blocks semi-private state */
3555struct inflate_blocks_state {
3556
3557  /* mode */
3558  inflate_block_mode  mode;     /* current inflate_block mode */
3559
3560  /* mode dependent information */
3561  union {
3562    uInt left;          /* if STORED, bytes left to copy */
3563    struct {
3564      uInt table;               /* table lengths (14 bits) */
3565      uInt index;               /* index into blens (or border) */
3566      uIntf *blens;             /* bit lengths of codes */
3567      uInt bb;                  /* bit length tree depth */
3568      inflate_huft *tb;         /* bit length decoding tree */
3569    } trees;            /* if DTREE, decoding info for trees */
3570    struct {
3571      inflate_huft *tl;
3572      inflate_huft *td;         /* trees to free */
3573      inflate_codes_statef
3574         *codes;
3575    } decode;           /* if CODES, current state */
3576  } sub;                /* submode */
3577  uInt last;            /* true if this block is the last block */
3578
3579  /* mode independent information */
3580  uInt bitk;            /* bits in bit buffer */
3581  uLong bitb;           /* bit buffer */
3582  Bytef *window;        /* sliding window */
3583  Bytef *end;           /* one byte after sliding window */
3584  Bytef *read;          /* window read pointer */
3585  Bytef *write;         /* window write pointer */
3586  check_func checkfn;   /* check function */
3587  uLong check;          /* check on output */
3588
3589};
3590
3591
3592/* defines for inflate input/output */
3593/*   update pointers and return */
3594#define UPDBITS {s->bitb=b;s->bitk=k;}
3595#define UPDIN {z->avail_in=n;z->total_in+=p-z->next_in;z->next_in=p;}
3596#define UPDOUT {s->write=q;}
3597#define UPDATE {UPDBITS UPDIN UPDOUT}
3598#define LEAVE {UPDATE return inflate_flush(s,z,r);}
3599/*   get bytes and bits */
3600#define LOADIN {p=z->next_in;n=z->avail_in;b=s->bitb;k=s->bitk;}
3601#define NEEDBYTE {if(n)r=Z_OK;else LEAVE}
3602#define NEXTBYTE (n--,*p++)
3603#define NEEDBITS(j) {while(k<(j)){NEEDBYTE;b|=((uLong)NEXTBYTE)<<k;k+=8;}}
3604#define DUMPBITS(j) {b>>=(j);k-=(j);}
3605/*   output bytes */
3606#define WAVAIL (uInt)(q<s->read?s->read-q-1:s->end-q)
3607#define LOADOUT {q=s->write;m=(uInt)WAVAIL;}
3608#define WWRAP {if(q==s->end&&s->read!=s->window){q=s->window;m=(uInt)WAVAIL;}}
3609#define FLUSH {UPDOUT r=inflate_flush(s,z,r); LOADOUT}
3610#define NEEDOUT {if(m==0){WWRAP if(m==0){FLUSH WWRAP if(m==0) LEAVE}}r=Z_OK;}
3611#define OUTBYTE(a) {*q++=(Byte)(a);m--;}
3612/*   load local pointers */
3613#define LOAD {LOADIN LOADOUT}
3614
3615/* masks for lower bits (size given to avoid silly warnings with Visual C++) */
3616extern uInt inflate_mask[17];
3617
3618/* copy as much as possible from the sliding window to the output area */
3619extern int inflate_flush OF((
3620    inflate_blocks_statef *,
3621    z_streamp ,
3622    int));
3623
3624#ifndef NO_DUMMY_DECL
3625struct internal_state      {int dummy;}; /* for buggy compilers */
3626#endif
3627
3628#endif
3629/* --- infutil.h */
3630
3631#ifndef NO_DUMMY_DECL
3632struct inflate_codes_state {int dummy;}; /* for buggy compilers */
3633#endif
3634
3635/* Table for deflate from PKZIP's appnote.txt. */
3636local const uInt border[] = { /* Order of the bit length code lengths */
3637        16, 17, 18, 0, 8, 7, 9, 6, 10, 5, 11, 4, 12, 3, 13, 2, 14, 1, 15};
3638
3639/*
3640   Notes beyond the 1.93a appnote.txt:
3641
3642   1. Distance pointers never point before the beginning of the output
3643      stream.
3644   2. Distance pointers can point back across blocks, up to 32k away.
3645   3. There is an implied maximum of 7 bits for the bit length table and
3646      15 bits for the actual data.
3647   4. If only one code exists, then it is encoded using one bit.  (Zero
3648      would be more efficient, but perhaps a little confusing.)  If two
3649      codes exist, they are coded using one bit each (0 and 1).
3650   5. There is no way of sending zero distance codes--a dummy must be
3651      sent if there are none.  (History: a pre 2.0 version of PKZIP would
3652      store blocks with no distance codes, but this was discovered to be
3653      too harsh a criterion.)  Valid only for 1.93a.  2.04c does allow
3654      zero distance codes, which is sent as one code of zero bits in
3655      length.
3656   6. There are up to 286 literal/length codes.  Code 256 represents the
3657      end-of-block.  Note however that the static length tree defines
3658      288 codes just to fill out the Huffman codes.  Codes 286 and 287
3659      cannot be used though, since there is no length base or extra bits
3660      defined for them.  Similarily, there are up to 30 distance codes.
3661      However, static trees define 32 codes (all 5 bits) to fill out the
3662      Huffman codes, but the last two had better not show up in the data.
3663   7. Unzip can check dynamic Huffman blocks for complete code sets.
3664      The exception is that a single code would not be complete (see #4).
3665   8. The five bits following the block type is really the number of
3666      literal codes sent minus 257.
3667   9. Length codes 8,16,16 are interpreted as 13 length codes of 8 bits
3668      (1+6+6).  Therefore, to output three times the length, you output
3669      three codes (1+1+1), whereas to output four times the same length,
3670      you only need two codes (1+3).  Hmm.
3671  10. In the tree reconstruction algorithm, Code = Code + Increment
3672      only if BitLength(i) is not zero.  (Pretty obvious.)
3673  11. Correction: 4 Bits: # of Bit Length codes - 4     (4 - 19)
3674  12. Note: length code 284 can represent 227-258, but length code 285
3675      really is 258.  The last length deserves its own, short code
3676      since it gets used a lot in very redundant files.  The length
3677      258 is special since 258 - 3 (the min match length) is 255.
3678  13. The literal/length and distance code bit lengths are read as a
3679      single stream of lengths.  It is possible (and advantageous) for
3680      a repeat code (16, 17, or 18) to go across the boundary between
3681      the two sets of lengths.
3682 */
3683
3684
3685void inflate_blocks_reset(s, z, c)
3686inflate_blocks_statef *s;
3687z_streamp z;
3688uLongf *c;
3689{
3690  if (s->checkfn != Z_NULL)
3691    *c = s->check;
3692  if (s->mode == BTREE || s->mode == DTREE)
3693    ZFREE(z, s->sub.trees.blens);
3694  if (s->mode == CODES)
3695  {
3696    inflate_codes_free(s->sub.decode.codes, z);
3697    inflate_trees_free(s->sub.decode.td, z);
3698    inflate_trees_free(s->sub.decode.tl, z);
3699  }
3700  s->mode = TYPE;
3701  s->bitk = 0;
3702  s->bitb = 0;
3703  s->read = s->write = s->window;
3704  if (s->checkfn != Z_NULL)
3705    z->adler = s->check = (*s->checkfn)(0L, Z_NULL, 0);
3706  Trace((stderr, "inflate:   blocks reset\n"));
3707}
3708
3709
3710inflate_blocks_statef *inflate_blocks_new(z, c, w)
3711z_streamp z;
3712check_func c;
3713uInt w;
3714{
3715  inflate_blocks_statef *s;
3716
3717  if ((s = (inflate_blocks_statef *)ZALLOC
3718       (z,1,sizeof(struct inflate_blocks_state))) == Z_NULL)
3719    return s;
3720  if ((s->window = (Bytef *)ZALLOC(z, 1, w)) == Z_NULL)
3721  {
3722    ZFREE(z, s);
3723    return Z_NULL;
3724  }
3725  s->end = s->window + w;
3726  s->checkfn = c;
3727  s->mode = TYPE;
3728  Trace((stderr, "inflate:   blocks allocated\n"));
3729  inflate_blocks_reset(s, z, &s->check);
3730  return s;
3731}
3732
3733
3734#ifdef DEBUG_ZLIB
3735  extern uInt inflate_hufts;
3736#endif
3737int inflate_blocks(s, z, r)
3738inflate_blocks_statef *s;
3739z_streamp z;
3740int r;
3741{
3742  uInt t;               /* temporary storage */
3743  uLong b;              /* bit buffer */
3744  uInt k;               /* bits in bit buffer */
3745  Bytef *p;             /* input data pointer */
3746  uInt n;               /* bytes available there */
3747  Bytef *q;             /* output window write pointer */
3748  uInt m;               /* bytes to end of window or read pointer */
3749
3750  /* copy input/output information to locals (UPDATE macro restores) */
3751  LOAD
3752
3753  /* process input based on current state */
3754  while (1) switch (s->mode)
3755  {
3756    case TYPE:
3757      NEEDBITS(3)
3758      t = (uInt)b & 7;
3759      s->last = t & 1;
3760      switch (t >> 1)
3761      {
3762        case 0:                         /* stored */
3763          Trace((stderr, "inflate:     stored block%s\n",
3764                 s->last ? " (last)" : ""));
3765          DUMPBITS(3)
3766          t = k & 7;                    /* go to byte boundary */
3767          DUMPBITS(t)
3768          s->mode = LENS;               /* get length of stored block */
3769          break;
3770        case 1:                         /* fixed */
3771          Trace((stderr, "inflate:     fixed codes block%s\n",
3772                 s->last ? " (last)" : ""));
3773          {
3774            uInt bl, bd;
3775            inflate_huft *tl, *td;
3776
3777            inflate_trees_fixed(&bl, &bd, &tl, &td);
3778            s->sub.decode.codes = inflate_codes_new(bl, bd, tl, td, z);
3779            if (s->sub.decode.codes == Z_NULL)
3780            {
3781              r = Z_MEM_ERROR;
3782              LEAVE
3783            }
3784            s->sub.decode.tl = Z_NULL;  /* don't try to free these */
3785            s->sub.decode.td = Z_NULL;
3786          }
3787          DUMPBITS(3)
3788          s->mode = CODES;
3789          break;
3790        case 2:                         /* dynamic */
3791          Trace((stderr, "inflate:     dynamic codes block%s\n",
3792                 s->last ? " (last)" : ""));
3793          DUMPBITS(3)
3794          s->mode = TABLE;
3795          break;
3796        case 3:                         /* illegal */
3797          DUMPBITS(3)
3798          s->mode = BADB;
3799          z->msg = (char*)"invalid block type";
3800          r = Z_DATA_ERROR;
3801          LEAVE
3802      }
3803      break;
3804    case LENS:
3805      NEEDBITS(32)
3806      if ((((~b) >> 16) & 0xffff) != (b & 0xffff))
3807      {
3808        s->mode = BADB;
3809        z->msg = (char*)"invalid stored block lengths";
3810        r = Z_DATA_ERROR;
3811        LEAVE
3812      }
3813      s->sub.left = (uInt)b & 0xffff;
3814      b = k = 0;                      /* dump bits */
3815      Tracev((stderr, "inflate:       stored length %u\n", s->sub.left));
3816      s->mode = s->sub.left ? STORED : (s->last ? DRY : TYPE);
3817      break;
3818    case STORED:
3819      if (n == 0)
3820        LEAVE
3821      NEEDOUT
3822      t = s->sub.left;
3823      if (t > n) t = n;
3824      if (t > m) t = m;
3825      zmemcpy(q, p, t);
3826      p += t;  n -= t;
3827      q += t;  m -= t;
3828      if ((s->sub.left -= t) != 0)
3829        break;
3830      Tracev((stderr, "inflate:       stored end, %lu total out\n",
3831              z->total_out + (q >= s->read ? q - s->read :
3832              (s->end - s->read) + (q - s->window))));
3833      s->mode = s->last ? DRY : TYPE;
3834      break;
3835    case TABLE:
3836      NEEDBITS(14)
3837      s->sub.trees.table = t = (uInt)b & 0x3fff;
3838#ifndef PKZIP_BUG_WORKAROUND
3839      if ((t & 0x1f) > 29 || ((t >> 5) & 0x1f) > 29)
3840      {
3841        s->mode = BADB;
3842        z->msg = (char*)"too many length or distance symbols";
3843        r = Z_DATA_ERROR;
3844        LEAVE
3845      }
3846#endif
3847      t = 258 + (t & 0x1f) + ((t >> 5) & 0x1f);
3848      if (t < 19)
3849        t = 19;
3850      if ((s->sub.trees.blens = (uIntf*)ZALLOC(z, t, sizeof(uInt))) == Z_NULL)
3851      {
3852        r = Z_MEM_ERROR;
3853        LEAVE
3854      }
3855      DUMPBITS(14)
3856      s->sub.trees.index = 0;
3857      Tracev((stderr, "inflate:       table sizes ok\n"));
3858      s->mode = BTREE;
3859    case BTREE:
3860      while (s->sub.trees.index < 4 + (s->sub.trees.table >> 10))
3861      {
3862        NEEDBITS(3)
3863        s->sub.trees.blens[border[s->sub.trees.index++]] = (uInt)b & 7;
3864        DUMPBITS(3)
3865      }
3866      while (s->sub.trees.index < 19)
3867        s->sub.trees.blens[border[s->sub.trees.index++]] = 0;
3868      s->sub.trees.bb = 7;
3869      t = inflate_trees_bits(s->sub.trees.blens, &s->sub.trees.bb,
3870                             &s->sub.trees.tb, z);
3871      if (t != Z_OK)
3872      {
3873        r = t;
3874        if (r == Z_DATA_ERROR) {
3875          ZFREE(z, s->sub.trees.blens);
3876          s->mode = BADB;
3877        }
3878        LEAVE
3879      }
3880      s->sub.trees.index = 0;
3881      Tracev((stderr, "inflate:       bits tree ok\n"));
3882      s->mode = DTREE;
3883    case DTREE:
3884      while (t = s->sub.trees.table,
3885             s->sub.trees.index < 258 + (t & 0x1f) + ((t >> 5) & 0x1f))
3886      {
3887        inflate_huft *h;
3888        uInt i, j, c;
3889
3890        t = s->sub.trees.bb;
3891        NEEDBITS(t)
3892        h = s->sub.trees.tb + ((uInt)b & inflate_mask[t]);
3893        t = h->word.what.Bits;
3894        c = h->more.Base;
3895        if (c < 16)
3896        {
3897          DUMPBITS(t)
3898          s->sub.trees.blens[s->sub.trees.index++] = c;
3899        }
3900        else /* c == 16..18 */
3901        {
3902          i = c == 18 ? 7 : c - 14;
3903          j = c == 18 ? 11 : 3;
3904          NEEDBITS(t + i)
3905          DUMPBITS(t)
3906          j += (uInt)b & inflate_mask[i];
3907          DUMPBITS(i)
3908          i = s->sub.trees.index;
3909          t = s->sub.trees.table;
3910          if (i + j > 258 + (t & 0x1f) + ((t >> 5) & 0x1f) ||
3911              (c == 16 && i < 1))
3912          {
3913            inflate_trees_free(s->sub.trees.tb, z);
3914            ZFREE(z, s->sub.trees.blens);
3915            s->mode = BADB;
3916            z->msg = (char*)"invalid bit length repeat";
3917            r = Z_DATA_ERROR;
3918            LEAVE
3919          }
3920          c = c == 16 ? s->sub.trees.blens[i - 1] : 0;
3921          do {
3922            s->sub.trees.blens[i++] = c;
3923          } while (--j);
3924          s->sub.trees.index = i;
3925        }
3926      }
3927      inflate_trees_free(s->sub.trees.tb, z);
3928      s->sub.trees.tb = Z_NULL;
3929      {
3930        uInt bl, bd;
3931        inflate_huft *tl, *td;
3932        inflate_codes_statef *c;
3933
3934        bl = 9;         /* must be <= 9 for lookahead assumptions */
3935        bd = 6;         /* must be <= 9 for lookahead assumptions */
3936        t = s->sub.trees.table;
3937#ifdef DEBUG_ZLIB
3938      inflate_hufts = 0;
3939#endif
3940        t = inflate_trees_dynamic(257 + (t & 0x1f), 1 + ((t >> 5) & 0x1f),
3941                                  s->sub.trees.blens, &bl, &bd, &tl, &td, z);
3942        if (t != Z_OK)
3943        {
3944          if (t == (uInt)Z_DATA_ERROR) {
3945            ZFREE(z, s->sub.trees.blens);
3946            s->mode = BADB;
3947          }
3948          r = t;
3949          LEAVE
3950        }
3951        Tracev((stderr, "inflate:       trees ok, %d * %d bytes used\n",
3952              inflate_hufts, sizeof(inflate_huft)));
3953        if ((c = inflate_codes_new(bl, bd, tl, td, z)) == Z_NULL)
3954        {
3955          inflate_trees_free(td, z);
3956          inflate_trees_free(tl, z);
3957          r = Z_MEM_ERROR;
3958          LEAVE
3959        }
3960	/*
3961	 * this ZFREE must occur *BEFORE* we mess with sub.decode, because
3962	 * sub.trees is union'd with sub.decode.
3963	 */
3964        ZFREE(z, s->sub.trees.blens);
3965        s->sub.decode.codes = c;
3966        s->sub.decode.tl = tl;
3967        s->sub.decode.td = td;
3968      }
3969      s->mode = CODES;
3970    case CODES:
3971      UPDATE
3972      if ((r = inflate_codes(s, z, r)) != Z_STREAM_END)
3973        return inflate_flush(s, z, r);
3974      r = Z_OK;
3975      inflate_codes_free(s->sub.decode.codes, z);
3976      inflate_trees_free(s->sub.decode.td, z);
3977      inflate_trees_free(s->sub.decode.tl, z);
3978      LOAD
3979      Tracev((stderr, "inflate:       codes end, %lu total out\n",
3980              z->total_out + (q >= s->read ? q - s->read :
3981              (s->end - s->read) + (q - s->window))));
3982      if (!s->last)
3983      {
3984        s->mode = TYPE;
3985        break;
3986      }
3987      if (k > 7)              /* return unused byte, if any */
3988      {
3989        Assert(k < 16, "inflate_codes grabbed too many bytes")
3990        k -= 8;
3991        n++;
3992        p--;                    /* can always return one */
3993      }
3994      s->mode = DRY;
3995    case DRY:
3996      FLUSH
3997      if (s->read != s->write)
3998        LEAVE
3999      s->mode = DONEB;
4000    case DONEB:
4001      r = Z_STREAM_END;
4002      LEAVE
4003    case BADB:
4004      r = Z_DATA_ERROR;
4005      LEAVE
4006    default:
4007      r = Z_STREAM_ERROR;
4008      LEAVE
4009  }
4010}
4011
4012
4013int inflate_blocks_free(s, z, c)
4014inflate_blocks_statef *s;
4015z_streamp z;
4016uLongf *c;
4017{
4018  inflate_blocks_reset(s, z, c);
4019  ZFREE(z, s->window);
4020  ZFREE(z, s);
4021  Trace((stderr, "inflate:   blocks freed\n"));
4022  return Z_OK;
4023}
4024
4025
4026void inflate_set_dictionary(s, d, n)
4027inflate_blocks_statef *s;
4028const Bytef *d;
4029uInt  n;
4030{
4031  zmemcpy((charf *)s->window, d, n);
4032  s->read = s->write = s->window + n;
4033}
4034
4035/*
4036 * This subroutine adds the data at next_in/avail_in to the output history
4037 * without performing any output.  The output buffer must be "caught up";
4038 * i.e. no pending output (hence s->read equals s->write), and the state must
4039 * be BLOCKS (i.e. we should be willing to see the start of a series of
4040 * BLOCKS).  On exit, the output will also be caught up, and the checksum
4041 * will have been updated if need be.
4042 */
4043int inflate_addhistory(s, z)
4044inflate_blocks_statef *s;
4045z_stream *z;
4046{
4047    uLong b;              /* bit buffer */  /* NOT USED HERE */
4048    uInt k;               /* bits in bit buffer */ /* NOT USED HERE */
4049    uInt t;               /* temporary storage */
4050    Bytef *p;             /* input data pointer */
4051    uInt n;               /* bytes available there */
4052    Bytef *q;             /* output window write pointer */
4053    uInt m;               /* bytes to end of window or read pointer */
4054
4055    if (s->read != s->write)
4056	return Z_STREAM_ERROR;
4057    if (s->mode != TYPE)
4058	return Z_DATA_ERROR;
4059
4060    /* we're ready to rock */
4061    LOAD
4062    /* while there is input ready, copy to output buffer, moving
4063     * pointers as needed.
4064     */
4065    while (n) {
4066	t = n;  /* how many to do */
4067	/* is there room until end of buffer? */
4068	if (t > m) t = m;
4069	/* update check information */
4070	if (s->checkfn != Z_NULL)
4071	    s->check = (*s->checkfn)(s->check, q, t);
4072	zmemcpy(q, p, t);
4073	q += t;
4074	p += t;
4075	n -= t;
4076	z->total_out += t;
4077	s->read = q;    /* drag read pointer forward */
4078/*      WWRAP  */ 	/* expand WWRAP macro by hand to handle s->read */
4079	if (q == s->end) {
4080	    s->read = q = s->window;
4081	    m = WAVAIL;
4082	}
4083    }
4084    UPDATE
4085    return Z_OK;
4086}
4087
4088
4089/*
4090 * At the end of a Deflate-compressed PPP packet, we expect to have seen
4091 * a `stored' block type value but not the (zero) length bytes.
4092 */
4093int inflate_packet_flush(s)
4094    inflate_blocks_statef *s;
4095{
4096    if (s->mode != LENS)
4097	return Z_DATA_ERROR;
4098    s->mode = TYPE;
4099    return Z_OK;
4100}
4101/* --- infblock.c */
4102
4103/* +++ inftrees.c */
4104/* inftrees.c -- generate Huffman trees for efficient decoding
4105 * Copyright (C) 1995-1996 Mark Adler
4106 * For conditions of distribution and use, see copyright notice in zlib.h
4107 */
4108
4109/* #include "zutil.h" */
4110/* #include "inftrees.h" */
4111
4112char inflate_copyright[] = " inflate 1.0.4 Copyright 1995-1996 Mark Adler ";
4113/*
4114  If you use the zlib library in a product, an acknowledgment is welcome
4115  in the documentation of your product. If for some reason you cannot
4116  include such an acknowledgment, I would appreciate that you keep this
4117  copyright string in the executable of your product.
4118 */
4119
4120#ifndef NO_DUMMY_DECL
4121struct internal_state  {int dummy;}; /* for buggy compilers */
4122#endif
4123
4124/* simplify the use of the inflate_huft type with some defines */
4125#define base more.Base
4126#define next more.Next
4127#define exop word.what.Exop
4128#define bits word.what.Bits
4129
4130
4131local int huft_build OF((
4132    uIntf *,            /* code lengths in bits */
4133    uInt,               /* number of codes */
4134    uInt,               /* number of "simple" codes */
4135    const uIntf *,      /* list of base values for non-simple codes */
4136    const uIntf *,      /* list of extra bits for non-simple codes */
4137    inflate_huft * FAR*,/* result: starting table */
4138    uIntf *,            /* maximum lookup bits (returns actual) */
4139    z_streamp ));       /* for zalloc function */
4140
4141local voidpf falloc OF((
4142    voidpf,             /* opaque pointer (not used) */
4143    uInt,               /* number of items */
4144    uInt));             /* size of item */
4145
4146/* Tables for deflate from PKZIP's appnote.txt. */
4147local const uInt cplens[31] = { /* Copy lengths for literal codes 257..285 */
4148        3, 4, 5, 6, 7, 8, 9, 10, 11, 13, 15, 17, 19, 23, 27, 31,
4149        35, 43, 51, 59, 67, 83, 99, 115, 131, 163, 195, 227, 258, 0, 0};
4150        /* see note #13 above about 258 */
4151local const uInt cplext[31] = { /* Extra bits for literal codes 257..285 */
4152        0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2,
4153        3, 3, 3, 3, 4, 4, 4, 4, 5, 5, 5, 5, 0, 112, 112}; /* 112==invalid */
4154local const uInt cpdist[30] = { /* Copy offsets for distance codes 0..29 */
4155        1, 2, 3, 4, 5, 7, 9, 13, 17, 25, 33, 49, 65, 97, 129, 193,
4156        257, 385, 513, 769, 1025, 1537, 2049, 3073, 4097, 6145,
4157        8193, 12289, 16385, 24577};
4158local const uInt cpdext[30] = { /* Extra bits for distance codes */
4159        0, 0, 0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6,
4160        7, 7, 8, 8, 9, 9, 10, 10, 11, 11,
4161        12, 12, 13, 13};
4162
4163/*
4164   Huffman code decoding is performed using a multi-level table lookup.
4165   The fastest way to decode is to simply build a lookup table whose
4166   size is determined by the longest code.  However, the time it takes
4167   to build this table can also be a factor if the data being decoded
4168   is not very long.  The most common codes are necessarily the
4169   shortest codes, so those codes dominate the decoding time, and hence
4170   the speed.  The idea is you can have a shorter table that decodes the
4171   shorter, more probable codes, and then point to subsidiary tables for
4172   the longer codes.  The time it costs to decode the longer codes is
4173   then traded against the time it takes to make longer tables.
4174
4175   This results of this trade are in the variables lbits and dbits
4176   below.  lbits is the number of bits the first level table for literal/
4177   length codes can decode in one step, and dbits is the same thing for
4178   the distance codes.  Subsequent tables are also less than or equal to
4179   those sizes.  These values may be adjusted either when all of the
4180   codes are shorter than that, in which case the longest code length in
4181   bits is used, or when the shortest code is *longer* than the requested
4182   table size, in which case the length of the shortest code in bits is
4183   used.
4184
4185   There are two different values for the two tables, since they code a
4186   different number of possibilities each.  The literal/length table
4187   codes 286 possible values, or in a flat code, a little over eight
4188   bits.  The distance table codes 30 possible values, or a little less
4189   than five bits, flat.  The optimum values for speed end up being
4190   about one bit more than those, so lbits is 8+1 and dbits is 5+1.
4191   The optimum values may differ though from machine to machine, and
4192   possibly even between compilers.  Your mileage may vary.
4193 */
4194
4195
4196/* If BMAX needs to be larger than 16, then h and x[] should be uLong. */
4197#define BMAX 15         /* maximum bit length of any code */
4198#define N_MAX 288       /* maximum number of codes in any set */
4199
4200#ifdef DEBUG_ZLIB
4201  uInt inflate_hufts;
4202#endif
4203
4204local int huft_build(b, n, s, d, e, t, m, zs)
4205uIntf *b;               /* code lengths in bits (all assumed <= BMAX) */
4206uInt n;                 /* number of codes (assumed <= N_MAX) */
4207uInt s;                 /* number of simple-valued codes (0..s-1) */
4208const uIntf *d;         /* list of base values for non-simple codes */
4209const uIntf *e;         /* list of extra bits for non-simple codes */
4210inflate_huft * FAR *t;  /* result: starting table */
4211uIntf *m;               /* maximum lookup bits, returns actual */
4212z_streamp zs;           /* for zalloc function */
4213/* Given a list of code lengths and a maximum table size, make a set of
4214   tables to decode that set of codes.  Return Z_OK on success, Z_BUF_ERROR
4215   if the given code set is incomplete (the tables are still built in this
4216   case), Z_DATA_ERROR if the input is invalid (an over-subscribed set of
4217   lengths), or Z_MEM_ERROR if not enough memory. */
4218{
4219
4220  uInt a;                       /* counter for codes of length k */
4221  uInt c[BMAX+1];               /* bit length count table */
4222  uInt f;                       /* i repeats in table every f entries */
4223  int g;                        /* maximum code length */
4224  int h;                        /* table level */
4225  register uInt i;              /* counter, current code */
4226  register uInt j;              /* counter */
4227  register int k;               /* number of bits in current code */
4228  int l;                        /* bits per table (returned in m) */
4229  register uIntf *p;            /* pointer into c[], b[], or v[] */
4230  inflate_huft *q;              /* points to current table */
4231  struct inflate_huft_s r;      /* table entry for structure assignment */
4232  inflate_huft *u[BMAX];        /* table stack */
4233  uInt v[N_MAX];                /* values in order of bit length */
4234  register int w;               /* bits before this table == (l * h) */
4235  uInt x[BMAX+1];               /* bit offsets, then code stack */
4236  uIntf *xp;                    /* pointer into x */
4237  int y;                        /* number of dummy codes added */
4238  uInt z;                       /* number of entries in current table */
4239
4240
4241  /* Generate counts for each bit length */
4242  p = c;
4243#define C0 *p++ = 0;
4244#define C2 C0 C0 C0 C0
4245#define C4 C2 C2 C2 C2
4246  C4                            /* clear c[]--assume BMAX+1 is 16 */
4247  p = b;  i = n;
4248  do {
4249    c[*p++]++;                  /* assume all entries <= BMAX */
4250  } while (--i);
4251  if (c[0] == n)                /* null input--all zero length codes */
4252  {
4253    *t = (inflate_huft *)Z_NULL;
4254    *m = 0;
4255    return Z_OK;
4256  }
4257
4258
4259  /* Find minimum and maximum length, bound *m by those */
4260  l = *m;
4261  for (j = 1; j <= BMAX; j++)
4262    if (c[j])
4263      break;
4264  k = j;                        /* minimum code length */
4265  if ((uInt)l < j)
4266    l = j;
4267  for (i = BMAX; i; i--)
4268    if (c[i])
4269      break;
4270  g = i;                        /* maximum code length */
4271  if ((uInt)l > i)
4272    l = i;
4273  *m = l;
4274
4275
4276  /* Adjust last length count to fill out codes, if needed */
4277  for (y = 1 << j; j < i; j++, y <<= 1)
4278    if ((y -= c[j]) < 0)
4279      return Z_DATA_ERROR;
4280  if ((y -= c[i]) < 0)
4281    return Z_DATA_ERROR;
4282  c[i] += y;
4283
4284
4285  /* Generate starting offsets into the value table for each length */
4286  x[1] = j = 0;
4287  p = c + 1;  xp = x + 2;
4288  while (--i) {                 /* note that i == g from above */
4289    *xp++ = (j += *p++);
4290  }
4291
4292
4293  /* Make a table of values in order of bit lengths */
4294  p = b;  i = 0;
4295  do {
4296    if ((j = *p++) != 0)
4297      v[x[j]++] = i;
4298  } while (++i < n);
4299  n = x[g];                   /* set n to length of v */
4300
4301
4302  /* Generate the Huffman codes and for each, make the table entries */
4303  x[0] = i = 0;                 /* first Huffman code is zero */
4304  p = v;                        /* grab values in bit order */
4305  h = -1;                       /* no tables yet--level -1 */
4306  w = -l;                       /* bits decoded == (l * h) */
4307  u[0] = (inflate_huft *)Z_NULL;        /* just to keep compilers happy */
4308  q = (inflate_huft *)Z_NULL;   /* ditto */
4309  z = 0;                        /* ditto */
4310
4311  /* go through the bit lengths (k already is bits in shortest code) */
4312  for (; k <= g; k++)
4313  {
4314    a = c[k];
4315    while (a--)
4316    {
4317      /* here i is the Huffman code of length k bits for value *p */
4318      /* make tables up to required level */
4319      while (k > w + l)
4320      {
4321        h++;
4322        w += l;                 /* previous table always l bits */
4323
4324        /* compute minimum size table less than or equal to l bits */
4325        z = g - w;
4326        z = z > (uInt)l ? l : z;        /* table size upper limit */
4327        if ((f = 1 << (j = k - w)) > a + 1)     /* try a k-w bit table */
4328        {                       /* too few codes for k-w bit table */
4329          f -= a + 1;           /* deduct codes from patterns left */
4330          xp = c + k;
4331          if (j < z)
4332            while (++j < z)     /* try smaller tables up to z bits */
4333            {
4334              if ((f <<= 1) <= *++xp)
4335                break;          /* enough codes to use up j bits */
4336              f -= *xp;         /* else deduct codes from patterns */
4337            }
4338        }
4339        z = 1 << j;             /* table entries for j-bit table */
4340
4341        /* allocate and link in new table */
4342        if ((q = (inflate_huft *)ZALLOC
4343             (zs,z + 1,sizeof(inflate_huft))) == Z_NULL)
4344        {
4345          if (h)
4346            inflate_trees_free(u[0], zs);
4347          return Z_MEM_ERROR;   /* not enough memory */
4348        }
4349#ifdef DEBUG_ZLIB
4350        inflate_hufts += z + 1;
4351#endif
4352        *t = q + 1;             /* link to list for huft_free() */
4353        *(t = &(q->next)) = Z_NULL;
4354        u[h] = ++q;             /* table starts after link */
4355
4356        /* connect to last table, if there is one */
4357        if (h)
4358        {
4359          x[h] = i;             /* save pattern for backing up */
4360          r.bits = (Byte)l;     /* bits to dump before this table */
4361          r.exop = (Byte)j;     /* bits in this table */
4362          r.next = q;           /* pointer to this table */
4363          j = i >> (w - l);     /* (get around Turbo C bug) */
4364          u[h-1][j] = r;        /* connect to last table */
4365        }
4366      }
4367
4368      /* set up table entry in r */
4369      r.bits = (Byte)(k - w);
4370      if (p >= v + n)
4371        r.exop = 128 + 64;      /* out of values--invalid code */
4372      else if (*p < s)
4373      {
4374        r.exop = (Byte)(*p < 256 ? 0 : 32 + 64);     /* 256 is end-of-block */
4375        r.base = *p++;          /* simple code is just the value */
4376      }
4377      else
4378      {
4379        r.exop = (Byte)(e[*p - s] + 16 + 64);/* non-simple--look up in lists */
4380        r.base = d[*p++ - s];
4381      }
4382
4383      /* fill code-like entries with r */
4384      f = 1 << (k - w);
4385      for (j = i >> w; j < z; j += f)
4386        q[j] = r;
4387
4388      /* backwards increment the k-bit code i */
4389      for (j = 1 << (k - 1); i & j; j >>= 1)
4390        i ^= j;
4391      i ^= j;
4392
4393      /* backup over finished tables */
4394      while ((i & ((1 << w) - 1)) != x[h])
4395      {
4396        h--;                    /* don't need to update q */
4397        w -= l;
4398      }
4399    }
4400  }
4401
4402
4403  /* Return Z_BUF_ERROR if we were given an incomplete table */
4404  return y != 0 && g != 1 ? Z_BUF_ERROR : Z_OK;
4405}
4406
4407
4408int inflate_trees_bits(c, bb, tb, z)
4409uIntf *c;               /* 19 code lengths */
4410uIntf *bb;              /* bits tree desired/actual depth */
4411inflate_huft * FAR *tb; /* bits tree result */
4412z_streamp z;            /* for zfree function */
4413{
4414  int r;
4415
4416  r = huft_build(c, 19, 19, (uIntf*)Z_NULL, (uIntf*)Z_NULL, tb, bb, z);
4417  if (r == Z_DATA_ERROR)
4418    z->msg = (char*)"oversubscribed dynamic bit lengths tree";
4419  else if (r == Z_BUF_ERROR || *bb == 0)
4420  {
4421    inflate_trees_free(*tb, z);
4422    z->msg = (char*)"incomplete dynamic bit lengths tree";
4423    r = Z_DATA_ERROR;
4424  }
4425  return r;
4426}
4427
4428
4429int inflate_trees_dynamic(nl, nd, c, bl, bd, tl, td, z)
4430uInt nl;                /* number of literal/length codes */
4431uInt nd;                /* number of distance codes */
4432uIntf *c;               /* that many (total) code lengths */
4433uIntf *bl;              /* literal desired/actual bit depth */
4434uIntf *bd;              /* distance desired/actual bit depth */
4435inflate_huft * FAR *tl; /* literal/length tree result */
4436inflate_huft * FAR *td; /* distance tree result */
4437z_streamp z;            /* for zfree function */
4438{
4439  int r;
4440
4441  /* build literal/length tree */
4442  r = huft_build(c, nl, 257, cplens, cplext, tl, bl, z);
4443  if (r != Z_OK || *bl == 0)
4444  {
4445    if (r == Z_DATA_ERROR)
4446      z->msg = (char*)"oversubscribed literal/length tree";
4447    else if (r != Z_MEM_ERROR)
4448    {
4449      inflate_trees_free(*tl, z);
4450      z->msg = (char*)"incomplete literal/length tree";
4451      r = Z_DATA_ERROR;
4452    }
4453    return r;
4454  }
4455
4456  /* build distance tree */
4457  r = huft_build(c + nl, nd, 0, cpdist, cpdext, td, bd, z);
4458  if (r != Z_OK || (*bd == 0 && nl > 257))
4459  {
4460    if (r == Z_DATA_ERROR)
4461      z->msg = (char*)"oversubscribed distance tree";
4462    else if (r == Z_BUF_ERROR) {
4463#ifdef PKZIP_BUG_WORKAROUND
4464      r = Z_OK;
4465    }
4466#else
4467      inflate_trees_free(*td, z);
4468      z->msg = (char*)"incomplete distance tree";
4469      r = Z_DATA_ERROR;
4470    }
4471    else if (r != Z_MEM_ERROR)
4472    {
4473      z->msg = (char*)"empty distance tree with lengths";
4474      r = Z_DATA_ERROR;
4475    }
4476    inflate_trees_free(*tl, z);
4477    return r;
4478#endif
4479  }
4480
4481  /* done */
4482  return Z_OK;
4483}
4484
4485
4486/* build fixed tables only once--keep them here */
4487local int fixed_built = 0;
4488#define FIXEDH 530      /* number of hufts used by fixed tables */
4489local inflate_huft fixed_mem[FIXEDH];
4490local uInt fixed_bl;
4491local uInt fixed_bd;
4492local inflate_huft *fixed_tl;
4493local inflate_huft *fixed_td;
4494
4495
4496local voidpf falloc(q, n, s)
4497voidpf q;       /* opaque pointer */
4498uInt n;         /* number of items */
4499uInt s;         /* size of item */
4500{
4501  Assert(s == sizeof(inflate_huft) && n <= *(intf *)q,
4502         "inflate_trees falloc overflow");
4503  *(intf *)q -= n+s-s; /* s-s to avoid warning */
4504  return (voidpf)(fixed_mem + *(intf *)q);
4505}
4506
4507
4508int inflate_trees_fixed(bl, bd, tl, td)
4509uIntf *bl;               /* literal desired/actual bit depth */
4510uIntf *bd;               /* distance desired/actual bit depth */
4511inflate_huft * FAR *tl;  /* literal/length tree result */
4512inflate_huft * FAR *td;  /* distance tree result */
4513{
4514  /* build fixed tables if not already (multiple overlapped executions ok) */
4515  if (!fixed_built)
4516  {
4517    int k;              /* temporary variable */
4518    unsigned c[288];    /* length list for huft_build */
4519    z_stream z;         /* for falloc function */
4520    int f = FIXEDH;     /* number of hufts left in fixed_mem */
4521
4522    /* set up fake z_stream for memory routines */
4523    z.zalloc = falloc;
4524    z.zfree = Z_NULL;
4525    z.opaque = (voidpf)&f;
4526
4527    /* literal table */
4528    for (k = 0; k < 144; k++)
4529      c[k] = 8;
4530    for (; k < 256; k++)
4531      c[k] = 9;
4532    for (; k < 280; k++)
4533      c[k] = 7;
4534    for (; k < 288; k++)
4535      c[k] = 8;
4536    fixed_bl = 7;
4537    huft_build(c, 288, 257, cplens, cplext, &fixed_tl, &fixed_bl, &z);
4538
4539    /* distance table */
4540    for (k = 0; k < 30; k++)
4541      c[k] = 5;
4542    fixed_bd = 5;
4543    huft_build(c, 30, 0, cpdist, cpdext, &fixed_td, &fixed_bd, &z);
4544
4545    /* done */
4546    Assert(f == 0, "invalid build of fixed tables");
4547    fixed_built = 1;
4548  }
4549  *bl = fixed_bl;
4550  *bd = fixed_bd;
4551  *tl = fixed_tl;
4552  *td = fixed_td;
4553  return Z_OK;
4554}
4555
4556
4557int inflate_trees_free(t, z)
4558inflate_huft *t;        /* table to free */
4559z_streamp z;            /* for zfree function */
4560/* Free the malloc'ed tables built by huft_build(), which makes a linked
4561   list of the tables it made, with the links in a dummy first entry of
4562   each table. */
4563{
4564  register inflate_huft *p, *q, *r;
4565
4566  /* Reverse linked list */
4567  p = Z_NULL;
4568  q = t;
4569  while (q != Z_NULL)
4570  {
4571    r = (q - 1)->next;
4572    (q - 1)->next = p;
4573    p = q;
4574    q = r;
4575  }
4576  /* Go through linked list, freeing from the malloced (t[-1]) address. */
4577  while (p != Z_NULL)
4578  {
4579    q = (--p)->next;
4580    ZFREE(z,p);
4581    p = q;
4582  }
4583  return Z_OK;
4584}
4585/* --- inftrees.c */
4586
4587/* +++ infcodes.c */
4588/* infcodes.c -- process literals and length/distance pairs
4589 * Copyright (C) 1995-1996 Mark Adler
4590 * For conditions of distribution and use, see copyright notice in zlib.h
4591 */
4592
4593/* #include "zutil.h" */
4594/* #include "inftrees.h" */
4595/* #include "infblock.h" */
4596/* #include "infcodes.h" */
4597/* #include "infutil.h" */
4598
4599/* +++ inffast.h */
4600/* inffast.h -- header to use inffast.c
4601 * Copyright (C) 1995-1996 Mark Adler
4602 * For conditions of distribution and use, see copyright notice in zlib.h
4603 */
4604
4605/* WARNING: this file should *not* be used by applications. It is
4606   part of the implementation of the compression library and is
4607   subject to change. Applications should only use zlib.h.
4608 */
4609
4610extern int inflate_fast OF((
4611    uInt,
4612    uInt,
4613    inflate_huft *,
4614    inflate_huft *,
4615    inflate_blocks_statef *,
4616    z_streamp ));
4617/* --- inffast.h */
4618
4619/* simplify the use of the inflate_huft type with some defines */
4620#define base more.Base
4621#define next more.Next
4622#define exop word.what.Exop
4623#define bits word.what.Bits
4624
4625/* inflate codes private state */
4626struct inflate_codes_state {
4627
4628  /* mode */
4629  enum {        /* waiting for "i:"=input, "o:"=output, "x:"=nothing */
4630      START,    /* x: set up for LEN */
4631      LEN,      /* i: get length/literal/eob next */
4632      LENEXT,   /* i: getting length extra (have base) */
4633      DIST,     /* i: get distance next */
4634      DISTEXT,  /* i: getting distance extra */
4635      COPY,     /* o: copying bytes in window, waiting for space */
4636      LIT,      /* o: got literal, waiting for output space */
4637      WASH,     /* o: got eob, possibly still output waiting */
4638      END,      /* x: got eob and all data flushed */
4639      BADCODE}  /* x: got error */
4640    mode;               /* current inflate_codes mode */
4641
4642  /* mode dependent information */
4643  uInt len;
4644  union {
4645    struct {
4646      inflate_huft *tree;       /* pointer into tree */
4647      uInt need;                /* bits needed */
4648    } code;             /* if LEN or DIST, where in tree */
4649    uInt lit;           /* if LIT, literal */
4650    struct {
4651      uInt get;                 /* bits to get for extra */
4652      uInt dist;                /* distance back to copy from */
4653    } copy;             /* if EXT or COPY, where and how much */
4654  } sub;                /* submode */
4655
4656  /* mode independent information */
4657  Byte lbits;           /* ltree bits decoded per branch */
4658  Byte dbits;           /* dtree bits decoder per branch */
4659  inflate_huft *ltree;          /* literal/length/eob tree */
4660  inflate_huft *dtree;          /* distance tree */
4661
4662};
4663
4664
4665inflate_codes_statef *inflate_codes_new(bl, bd, tl, td, z)
4666uInt bl, bd;
4667inflate_huft *tl;
4668inflate_huft *td; /* need separate declaration for Borland C++ */
4669z_streamp z;
4670{
4671  inflate_codes_statef *c;
4672
4673  if ((c = (inflate_codes_statef *)
4674       ZALLOC(z,1,sizeof(struct inflate_codes_state))) != Z_NULL)
4675  {
4676    c->mode = START;
4677    c->lbits = (Byte)bl;
4678    c->dbits = (Byte)bd;
4679    c->ltree = tl;
4680    c->dtree = td;
4681    Tracev((stderr, "inflate:       codes new\n"));
4682  }
4683  return c;
4684}
4685
4686
4687int inflate_codes(s, z, r)
4688inflate_blocks_statef *s;
4689z_streamp z;
4690int r;
4691{
4692  uInt j;               /* temporary storage */
4693  inflate_huft *t;      /* temporary pointer */
4694  uInt e;               /* extra bits or operation */
4695  uLong b;              /* bit buffer */
4696  uInt k;               /* bits in bit buffer */
4697  Bytef *p;             /* input data pointer */
4698  uInt n;               /* bytes available there */
4699  Bytef *q;             /* output window write pointer */
4700  uInt m;               /* bytes to end of window or read pointer */
4701  Bytef *f;             /* pointer to copy strings from */
4702  inflate_codes_statef *c = s->sub.decode.codes;  /* codes state */
4703
4704  /* copy input/output information to locals (UPDATE macro restores) */
4705  LOAD
4706
4707  /* process input and output based on current state */
4708  while (1) switch (c->mode)
4709  {             /* waiting for "i:"=input, "o:"=output, "x:"=nothing */
4710    case START:         /* x: set up for LEN */
4711#ifndef SLOW
4712      if (m >= 258 && n >= 10)
4713      {
4714        UPDATE
4715        r = inflate_fast(c->lbits, c->dbits, c->ltree, c->dtree, s, z);
4716        LOAD
4717        if (r != Z_OK)
4718        {
4719          c->mode = r == Z_STREAM_END ? WASH : BADCODE;
4720          break;
4721        }
4722      }
4723#endif /* !SLOW */
4724      c->sub.code.need = c->lbits;
4725      c->sub.code.tree = c->ltree;
4726      c->mode = LEN;
4727    case LEN:           /* i: get length/literal/eob next */
4728      j = c->sub.code.need;
4729      NEEDBITS(j)
4730      t = c->sub.code.tree + ((uInt)b & inflate_mask[j]);
4731      DUMPBITS(t->bits)
4732      e = (uInt)(t->exop);
4733      if (e == 0)               /* literal */
4734      {
4735        c->sub.lit = t->base;
4736        Tracevv((stderr, t->base >= 0x20 && t->base < 0x7f ?
4737                 "inflate:         literal '%c'\n" :
4738                 "inflate:         literal 0x%02x\n", t->base));
4739        c->mode = LIT;
4740        break;
4741      }
4742      if (e & 16)               /* length */
4743      {
4744        c->sub.copy.get = e & 15;
4745        c->len = t->base;
4746        c->mode = LENEXT;
4747        break;
4748      }
4749      if ((e & 64) == 0)        /* next table */
4750      {
4751        c->sub.code.need = e;
4752        c->sub.code.tree = t->next;
4753        break;
4754      }
4755      if (e & 32)               /* end of block */
4756      {
4757        Tracevv((stderr, "inflate:         end of block\n"));
4758        c->mode = WASH;
4759        break;
4760      }
4761      c->mode = BADCODE;        /* invalid code */
4762      z->msg = (char*)"invalid literal/length code";
4763      r = Z_DATA_ERROR;
4764      LEAVE
4765    case LENEXT:        /* i: getting length extra (have base) */
4766      j = c->sub.copy.get;
4767      NEEDBITS(j)
4768      c->len += (uInt)b & inflate_mask[j];
4769      DUMPBITS(j)
4770      c->sub.code.need = c->dbits;
4771      c->sub.code.tree = c->dtree;
4772      Tracevv((stderr, "inflate:         length %u\n", c->len));
4773      c->mode = DIST;
4774    case DIST:          /* i: get distance next */
4775      j = c->sub.code.need;
4776      NEEDBITS(j)
4777      t = c->sub.code.tree + ((uInt)b & inflate_mask[j]);
4778      DUMPBITS(t->bits)
4779      e = (uInt)(t->exop);
4780      if (e & 16)               /* distance */
4781      {
4782        c->sub.copy.get = e & 15;
4783        c->sub.copy.dist = t->base;
4784        c->mode = DISTEXT;
4785        break;
4786      }
4787      if ((e & 64) == 0)        /* next table */
4788      {
4789        c->sub.code.need = e;
4790        c->sub.code.tree = t->next;
4791        break;
4792      }
4793      c->mode = BADCODE;        /* invalid code */
4794      z->msg = (char*)"invalid distance code";
4795      r = Z_DATA_ERROR;
4796      LEAVE
4797    case DISTEXT:       /* i: getting distance extra */
4798      j = c->sub.copy.get;
4799      NEEDBITS(j)
4800      c->sub.copy.dist += (uInt)b & inflate_mask[j];
4801      DUMPBITS(j)
4802      Tracevv((stderr, "inflate:         distance %u\n", c->sub.copy.dist));
4803      c->mode = COPY;
4804    case COPY:          /* o: copying bytes in window, waiting for space */
4805#ifndef __TURBOC__ /* Turbo C bug for following expression */
4806      f = (uInt)(q - s->window) < c->sub.copy.dist ?
4807          s->end - (c->sub.copy.dist - (q - s->window)) :
4808          q - c->sub.copy.dist;
4809#else
4810      f = q - c->sub.copy.dist;
4811      if ((uInt)(q - s->window) < c->sub.copy.dist)
4812        f = s->end - (c->sub.copy.dist - (uInt)(q - s->window));
4813#endif
4814      while (c->len)
4815      {
4816        NEEDOUT
4817        OUTBYTE(*f++)
4818        if (f == s->end)
4819          f = s->window;
4820        c->len--;
4821      }
4822      c->mode = START;
4823      break;
4824    case LIT:           /* o: got literal, waiting for output space */
4825      NEEDOUT
4826      OUTBYTE(c->sub.lit)
4827      c->mode = START;
4828      break;
4829    case WASH:          /* o: got eob, possibly more output */
4830      FLUSH
4831      if (s->read != s->write)
4832        LEAVE
4833      c->mode = END;
4834    case END:
4835      r = Z_STREAM_END;
4836      LEAVE
4837    case BADCODE:       /* x: got error */
4838      r = Z_DATA_ERROR;
4839      LEAVE
4840    default:
4841      r = Z_STREAM_ERROR;
4842      LEAVE
4843  }
4844}
4845
4846
4847void inflate_codes_free(c, z)
4848inflate_codes_statef *c;
4849z_streamp z;
4850{
4851  ZFREE(z, c);
4852  Tracev((stderr, "inflate:       codes free\n"));
4853}
4854/* --- infcodes.c */
4855
4856/* +++ infutil.c */
4857/* inflate_util.c -- data and routines common to blocks and codes
4858 * Copyright (C) 1995-1996 Mark Adler
4859 * For conditions of distribution and use, see copyright notice in zlib.h
4860 */
4861
4862/* #include "zutil.h" */
4863/* #include "infblock.h" */
4864/* #include "inftrees.h" */
4865/* #include "infcodes.h" */
4866/* #include "infutil.h" */
4867
4868#ifndef NO_DUMMY_DECL
4869struct inflate_codes_state {int dummy;}; /* for buggy compilers */
4870#endif
4871
4872/* And'ing with mask[n] masks the lower n bits */
4873uInt inflate_mask[17] = {
4874    0x0000,
4875    0x0001, 0x0003, 0x0007, 0x000f, 0x001f, 0x003f, 0x007f, 0x00ff,
4876    0x01ff, 0x03ff, 0x07ff, 0x0fff, 0x1fff, 0x3fff, 0x7fff, 0xffff
4877};
4878
4879
4880/* copy as much as possible from the sliding window to the output area */
4881int inflate_flush(s, z, r)
4882inflate_blocks_statef *s;
4883z_streamp z;
4884int r;
4885{
4886  uInt n;
4887  Bytef *p;
4888  Bytef *q;
4889
4890  /* local copies of source and destination pointers */
4891  p = z->next_out;
4892  q = s->read;
4893
4894  /* compute number of bytes to copy as far as end of window */
4895  n = (uInt)((q <= s->write ? s->write : s->end) - q);
4896  if (n > z->avail_out) n = z->avail_out;
4897  if (n && r == Z_BUF_ERROR) r = Z_OK;
4898
4899  /* update counters */
4900  z->avail_out -= n;
4901  z->total_out += n;
4902
4903  /* update check information */
4904  if (s->checkfn != Z_NULL)
4905    z->adler = s->check = (*s->checkfn)(s->check, q, n);
4906
4907  /* copy as far as end of window */
4908  if (p != Z_NULL) {
4909    zmemcpy(p, q, n);
4910    p += n;
4911  }
4912  q += n;
4913
4914  /* see if more to copy at beginning of window */
4915  if (q == s->end)
4916  {
4917    /* wrap pointers */
4918    q = s->window;
4919    if (s->write == s->end)
4920      s->write = s->window;
4921
4922    /* compute bytes to copy */
4923    n = (uInt)(s->write - q);
4924    if (n > z->avail_out) n = z->avail_out;
4925    if (n && r == Z_BUF_ERROR) r = Z_OK;
4926
4927    /* update counters */
4928    z->avail_out -= n;
4929    z->total_out += n;
4930
4931    /* update check information */
4932    if (s->checkfn != Z_NULL)
4933      z->adler = s->check = (*s->checkfn)(s->check, q, n);
4934
4935    /* copy */
4936    if (p != Z_NULL) {
4937      zmemcpy(p, q, n);
4938      p += n;
4939    }
4940    q += n;
4941  }
4942
4943  /* update pointers */
4944  z->next_out = p;
4945  s->read = q;
4946
4947  /* done */
4948  return r;
4949}
4950/* --- infutil.c */
4951
4952/* +++ inffast.c */
4953/* inffast.c -- process literals and length/distance pairs fast
4954 * Copyright (C) 1995-1996 Mark Adler
4955 * For conditions of distribution and use, see copyright notice in zlib.h
4956 */
4957
4958/* #include "zutil.h" */
4959/* #include "inftrees.h" */
4960/* #include "infblock.h" */
4961/* #include "infcodes.h" */
4962/* #include "infutil.h" */
4963/* #include "inffast.h" */
4964
4965#ifndef NO_DUMMY_DECL
4966struct inflate_codes_state {int dummy;}; /* for buggy compilers */
4967#endif
4968
4969/* simplify the use of the inflate_huft type with some defines */
4970#define base more.Base
4971#define next more.Next
4972#define exop word.what.Exop
4973#define bits word.what.Bits
4974
4975/* macros for bit input with no checking and for returning unused bytes */
4976#define GRABBITS(j) {while(k<(j)){b|=((uLong)NEXTBYTE)<<k;k+=8;}}
4977#define UNGRAB {n+=(c=k>>3);p-=c;k&=7;}
4978
4979/* Called with number of bytes left to write in window at least 258
4980   (the maximum string length) and number of input bytes available
4981   at least ten.  The ten bytes are six bytes for the longest length/
4982   distance pair plus four bytes for overloading the bit buffer. */
4983
4984int inflate_fast(bl, bd, tl, td, s, z)
4985uInt bl, bd;
4986inflate_huft *tl;
4987inflate_huft *td; /* need separate declaration for Borland C++ */
4988inflate_blocks_statef *s;
4989z_streamp z;
4990{
4991  inflate_huft *t;      /* temporary pointer */
4992  uInt e;               /* extra bits or operation */
4993  uLong b;              /* bit buffer */
4994  uInt k;               /* bits in bit buffer */
4995  Bytef *p;             /* input data pointer */
4996  uInt n;               /* bytes available there */
4997  Bytef *q;             /* output window write pointer */
4998  uInt m;               /* bytes to end of window or read pointer */
4999  uInt ml;              /* mask for literal/length tree */
5000  uInt md;              /* mask for distance tree */
5001  uInt c;               /* bytes to copy */
5002  uInt d;               /* distance back to copy from */
5003  Bytef *r;             /* copy source pointer */
5004
5005  /* load input, output, bit values */
5006  LOAD
5007
5008  /* initialize masks */
5009  ml = inflate_mask[bl];
5010  md = inflate_mask[bd];
5011
5012  /* do until not enough input or output space for fast loop */
5013  do {                          /* assume called with m >= 258 && n >= 10 */
5014    /* get literal/length code */
5015    GRABBITS(20)                /* max bits for literal/length code */
5016    if ((e = (t = tl + ((uInt)b & ml))->exop) == 0)
5017    {
5018      DUMPBITS(t->bits)
5019      Tracevv((stderr, t->base >= 0x20 && t->base < 0x7f ?
5020                "inflate:         * literal '%c'\n" :
5021                "inflate:         * literal 0x%02x\n", t->base));
5022      *q++ = (Byte)t->base;
5023      m--;
5024      continue;
5025    }
5026    do {
5027      DUMPBITS(t->bits)
5028      if (e & 16)
5029      {
5030        /* get extra bits for length */
5031        e &= 15;
5032        c = t->base + ((uInt)b & inflate_mask[e]);
5033        DUMPBITS(e)
5034        Tracevv((stderr, "inflate:         * length %u\n", c));
5035
5036        /* decode distance base of block to copy */
5037        GRABBITS(15);           /* max bits for distance code */
5038        e = (t = td + ((uInt)b & md))->exop;
5039        do {
5040          DUMPBITS(t->bits)
5041          if (e & 16)
5042          {
5043            /* get extra bits to add to distance base */
5044            e &= 15;
5045            GRABBITS(e)         /* get extra bits (up to 13) */
5046            d = t->base + ((uInt)b & inflate_mask[e]);
5047            DUMPBITS(e)
5048            Tracevv((stderr, "inflate:         * distance %u\n", d));
5049
5050            /* do the copy */
5051            m -= c;
5052            if ((uInt)(q - s->window) >= d)     /* offset before dest */
5053            {                                   /*  just copy */
5054              r = q - d;
5055              *q++ = *r++;  c--;        /* minimum count is three, */
5056              *q++ = *r++;  c--;        /*  so unroll loop a little */
5057            }
5058            else                        /* else offset after destination */
5059            {
5060              e = d - (uInt)(q - s->window); /* bytes from offset to end */
5061              r = s->end - e;           /* pointer to offset */
5062              if (c > e)                /* if source crosses, */
5063              {
5064                c -= e;                 /* copy to end of window */
5065                do {
5066                  *q++ = *r++;
5067                } while (--e);
5068                r = s->window;          /* copy rest from start of window */
5069              }
5070            }
5071            do {                        /* copy all or what's left */
5072              *q++ = *r++;
5073            } while (--c);
5074            break;
5075          }
5076          else if ((e & 64) == 0)
5077            e = (t = t->next + ((uInt)b & inflate_mask[e]))->exop;
5078          else
5079          {
5080            z->msg = (char*)"invalid distance code";
5081            UNGRAB
5082            UPDATE
5083            return Z_DATA_ERROR;
5084          }
5085        } while (1);
5086        break;
5087      }
5088      if ((e & 64) == 0)
5089      {
5090        if ((e = (t = t->next + ((uInt)b & inflate_mask[e]))->exop) == 0)
5091        {
5092          DUMPBITS(t->bits)
5093          Tracevv((stderr, t->base >= 0x20 && t->base < 0x7f ?
5094                    "inflate:         * literal '%c'\n" :
5095                    "inflate:         * literal 0x%02x\n", t->base));
5096          *q++ = (Byte)t->base;
5097          m--;
5098          break;
5099        }
5100      }
5101      else if (e & 32)
5102      {
5103        Tracevv((stderr, "inflate:         * end of block\n"));
5104        UNGRAB
5105        UPDATE
5106        return Z_STREAM_END;
5107      }
5108      else
5109      {
5110        z->msg = (char*)"invalid literal/length code";
5111        UNGRAB
5112        UPDATE
5113        return Z_DATA_ERROR;
5114      }
5115    } while (1);
5116  } while (m >= 258 && n >= 10);
5117
5118  /* not enough input or output--restore pointers and return */
5119  UNGRAB
5120  UPDATE
5121  return Z_OK;
5122}
5123/* --- inffast.c */
5124
5125/* +++ zutil.c */
5126/* zutil.c -- target dependent utility functions for the compression library
5127 * Copyright (C) 1995-1996 Jean-loup Gailly.
5128 * For conditions of distribution and use, see copyright notice in zlib.h
5129 */
5130
5131/* From: zutil.c,v 1.17 1996/07/24 13:41:12 me Exp $ */
5132
5133#ifdef DEBUG_ZLIB
5134#include <stdio.h>
5135#endif
5136
5137/* #include "zutil.h" */
5138
5139#ifndef NO_DUMMY_DECL
5140struct internal_state      {int dummy;}; /* for buggy compilers */
5141#endif
5142
5143#ifndef STDC
5144extern void exit OF((int));
5145#endif
5146
5147static const char *z_errmsg[10] = {
5148"need dictionary",     /* Z_NEED_DICT       2  */
5149"stream end",          /* Z_STREAM_END      1  */
5150"",                    /* Z_OK              0  */
5151"file error",          /* Z_ERRNO         (-1) */
5152"stream error",        /* Z_STREAM_ERROR  (-2) */
5153"data error",          /* Z_DATA_ERROR    (-3) */
5154"insufficient memory", /* Z_MEM_ERROR     (-4) */
5155"buffer error",        /* Z_BUF_ERROR     (-5) */
5156"incompatible version",/* Z_VERSION_ERROR (-6) */
5157""};
5158
5159
5160const char *zlibVersion()
5161{
5162    return ZLIB_VERSION;
5163}
5164
5165#ifdef DEBUG_ZLIB
5166void z_error (m)
5167    char *m;
5168{
5169    fprintf(stderr, "%s\n", m);
5170    exit(1);
5171}
5172#endif
5173
5174#ifndef HAVE_MEMCPY
5175
5176void zmemcpy(dest, source, len)
5177    Bytef* dest;
5178    Bytef* source;
5179    uInt  len;
5180{
5181    if (len == 0) return;
5182    do {
5183        *dest++ = *source++; /* ??? to be unrolled */
5184    } while (--len != 0);
5185}
5186
5187int zmemcmp(s1, s2, len)
5188    Bytef* s1;
5189    Bytef* s2;
5190    uInt  len;
5191{
5192    uInt j;
5193
5194    for (j = 0; j < len; j++) {
5195        if (s1[j] != s2[j]) return 2*(s1[j] > s2[j])-1;
5196    }
5197    return 0;
5198}
5199
5200void zmemzero(dest, len)
5201    Bytef* dest;
5202    uInt  len;
5203{
5204    if (len == 0) return;
5205    do {
5206        *dest++ = 0;  /* ??? to be unrolled */
5207    } while (--len != 0);
5208}
5209#endif
5210
5211#ifdef __TURBOC__
5212#if (defined( __BORLANDC__) || !defined(SMALL_MEDIUM)) && !defined(__32BIT__)
5213/* Small and medium model in Turbo C are for now limited to near allocation
5214 * with reduced MAX_WBITS and MAX_MEM_LEVEL
5215 */
5216#  define MY_ZCALLOC
5217
5218/* Turbo C malloc() does not allow dynamic allocation of 64K bytes
5219 * and farmalloc(64K) returns a pointer with an offset of 8, so we
5220 * must fix the pointer. Warning: the pointer must be put back to its
5221 * original form in order to free it, use zcfree().
5222 */
5223
5224#define MAX_PTR 10
5225/* 10*64K = 640K */
5226
5227local int next_ptr = 0;
5228
5229typedef struct ptr_table_s {
5230    voidpf org_ptr;
5231    voidpf new_ptr;
5232} ptr_table;
5233
5234local ptr_table table[MAX_PTR];
5235/* This table is used to remember the original form of pointers
5236 * to large buffers (64K). Such pointers are normalized with a zero offset.
5237 * Since MSDOS is not a preemptive multitasking OS, this table is not
5238 * protected from concurrent access. This hack doesn't work anyway on
5239 * a protected system like OS/2. Use Microsoft C instead.
5240 */
5241
5242voidpf zcalloc (voidpf opaque, unsigned items, unsigned size)
5243{
5244    voidpf buf = opaque; /* just to make some compilers happy */
5245    ulg bsize = (ulg)items*size;
5246
5247    /* If we allocate less than 65520 bytes, we assume that farmalloc
5248     * will return a usable pointer which doesn't have to be normalized.
5249     */
5250    if (bsize < 65520L) {
5251        buf = farmalloc(bsize);
5252        if (*(ush*)&buf != 0) return buf;
5253    } else {
5254        buf = farmalloc(bsize + 16L);
5255    }
5256    if (buf == NULL || next_ptr >= MAX_PTR) return NULL;
5257    table[next_ptr].org_ptr = buf;
5258
5259    /* Normalize the pointer to seg:0 */
5260    *((ush*)&buf+1) += ((ush)((uch*)buf-0) + 15) >> 4;
5261    *(ush*)&buf = 0;
5262    table[next_ptr++].new_ptr = buf;
5263    return buf;
5264}
5265
5266void  zcfree (voidpf opaque, voidpf ptr)
5267{
5268    int n;
5269    if (*(ush*)&ptr != 0) { /* object < 64K */
5270        farfree(ptr);
5271        return;
5272    }
5273    /* Find the original pointer */
5274    for (n = 0; n < next_ptr; n++) {
5275        if (ptr != table[n].new_ptr) continue;
5276
5277        farfree(table[n].org_ptr);
5278        while (++n < next_ptr) {
5279            table[n-1] = table[n];
5280        }
5281        next_ptr--;
5282        return;
5283    }
5284    ptr = opaque; /* just to make some compilers happy */
5285    Assert(0, "zcfree: ptr not found");
5286}
5287#endif
5288#endif /* __TURBOC__ */
5289
5290
5291#if defined(M_I86) && !defined(__32BIT__)
5292/* Microsoft C in 16-bit mode */
5293
5294#  define MY_ZCALLOC
5295
5296#if (!defined(_MSC_VER) || (_MSC_VER < 600))
5297#  define _halloc  halloc
5298#  define _hfree   hfree
5299#endif
5300
5301voidpf zcalloc (voidpf opaque, unsigned items, unsigned size)
5302{
5303    if (opaque) opaque = 0; /* to make compiler happy */
5304    return _halloc((long)items, size);
5305}
5306
5307void  zcfree (voidpf opaque, voidpf ptr)
5308{
5309    if (opaque) opaque = 0; /* to make compiler happy */
5310    _hfree(ptr);
5311}
5312
5313#endif /* MSC */
5314
5315
5316#ifndef MY_ZCALLOC /* Any system without a special alloc function */
5317
5318#ifndef STDC
5319extern voidp  calloc OF((uInt items, uInt size));
5320extern void   free   OF((voidpf ptr));
5321#endif
5322
5323voidpf zcalloc (opaque, items, size)
5324    voidpf opaque;
5325    unsigned items;
5326    unsigned size;
5327{
5328    if (opaque) items += size - size; /* make compiler happy */
5329    return (voidpf)calloc(items, size);
5330}
5331
5332void  zcfree (opaque, ptr)
5333    voidpf opaque;
5334    voidpf ptr;
5335{
5336    free(ptr);
5337    if (opaque) return; /* make compiler happy */
5338}
5339
5340#endif /* MY_ZCALLOC */
5341/* --- zutil.c */
5342
5343/* +++ adler32.c */
5344/* adler32.c -- compute the Adler-32 checksum of a data stream
5345 * Copyright (C) 1995-1996 Mark Adler
5346 * For conditions of distribution and use, see copyright notice in zlib.h
5347 */
5348
5349/* From: adler32.c,v 1.10 1996/05/22 11:52:18 me Exp $ */
5350
5351/* #include "zlib.h" */
5352
5353#define BASE 65521L /* largest prime smaller than 65536 */
5354#define NMAX 5552
5355/* NMAX is the largest n such that 255n(n+1)/2 + (n+1)(BASE-1) <= 2^32-1 */
5356
5357#define DO1(buf,i)  {s1 += buf[(i)]; s2 += s1;}
5358#define DO2(buf,i)  DO1(buf,i); DO1(buf,(i)+1);
5359#define DO4(buf,i)  DO2(buf,i); DO2(buf,(i)+2);
5360#define DO8(buf,i)  DO4(buf,i); DO4(buf,(i)+4);
5361#define DO16(buf)   DO8(buf,0); DO8(buf,8);
5362
5363/* ========================================================================= */
5364uLong adler32(adler, buf, len)
5365    uLong adler;
5366    const Bytef *buf;
5367    uInt len;
5368{
5369    unsigned long s1 = adler & 0xffff;
5370    unsigned long s2 = (adler >> 16) & 0xffff;
5371    int k;
5372
5373    if (buf == Z_NULL) return 1L;
5374
5375    while (len > 0) {
5376        k = len < NMAX ? len : NMAX;
5377        len -= k;
5378        while (k >= 16) {
5379            DO16(buf);
5380	    buf += 16;
5381            k -= 16;
5382        }
5383        if (k != 0) do {
5384            s1 += *buf++;
5385	    s2 += s1;
5386        } while (--k);
5387        s1 %= BASE;
5388        s2 %= BASE;
5389    }
5390    return (s2 << 16) | s1;
5391}
5392/* --- adler32.c */
5393
5394#ifdef _KERNEL
5395static int
5396zlib_modevent(module_t mod, int type, void *unused)
5397{
5398	switch (type) {
5399	case MOD_LOAD:
5400		return 0;
5401	case MOD_UNLOAD:
5402		return 0;
5403	}
5404	return EINVAL;
5405}
5406
5407static moduledata_t zlib_mod = {
5408	"zlib",
5409	zlib_modevent,
5410	0
5411};
5412DECLARE_MODULE(zlib, zlib_mod, SI_SUB_DRIVERS, SI_ORDER_FIRST);
5413MODULE_VERSION(zlib, 1);
5414#endif /* _KERNEL */
5415