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