deflate.c

   1 /* deflate.c -- compress data using the deflation algorithm
   2  * Copyright (C) 1995-2011 Jean-loup Gailly and Mark Adler
   3  * For conditions of distribution and use, see copyright notice in zlib.h
   4  */
   5
   6 /*
   7  *  ALGORITHM
   8  *
   9  *      The "deflation" process depends on being able to identify portions
  10  *      of the input text which are identical to earlier input (within a
  11  *      sliding window trailing behind the input currently being processed).
  12  *
  13  *      The most straightforward technique turns out to be the fastest for
  14  *      most input files: try all possible matches and select the longest.
  15  *      The key feature of this algorithm is that insertions into the string
  16  *      dictionary are very simple and thus fast, and deletions are avoided
  17  *      completely. Insertions are performed at each input character, whereas
  18  *      string matches are performed only when the previous match ends. So it
  19  *      is preferable to spend more time in matches to allow very fast string
  20  *      insertions and avoid deletions. The matching algorithm for small
  21  *      strings is inspired from that of Rabin & Karp. A brute force approach
  22  *      is used to find longer strings when a small match has been found.
  23  *      A similar algorithm is used in comic (by Jan-Mark Wams) and freeze
  24  *      (by Leonid Broukhis).
  25  *         A previous version of this file used a more sophisticated algorithm
  26  *      (by Fiala and Greene) which is guaranteed to run in linear amortized
  27  *      time, but has a larger average cost, uses more memory and is patented.
  28  *      However the F&G algorithm may be faster for some highly redundant
  29  *      files if the parameter max_chain_length (described below) is too large.
  30  *
  31  *  ACKNOWLEDGEMENTS
  32  *
  33  *      The idea of lazy evaluation of matches is due to Jan-Mark Wams, and
  34  *      I found it in 'freeze' written by Leonid Broukhis.
  35  *      Thanks to many people for bug reports and testing.
  36  *
  37  *  REFERENCES
  38  *
  39  *      Deutsch, L.P.,"DEFLATE Compressed Data Format Specification".
  40  *      Available in http://tools.ietf.org/html/rfc1951
  41  *
  42  *      A description of the Rabin and Karp algorithm is given in the book
  43  *         "Algorithms" by R. Sedgewick, Addison-Wesley, p252.
  44  *
  45  *      Fiala,E.R., and Greene,D.H.
  46  *         Data Compression with Finite Windows, Comm.ACM, 32,4 (1989) 490-595
  47  *
  48  */
  49
  50 /* @(#) $Id$ */
  51
  52 #include "deflate.h"
  53
  54 const char deflate_copyright[] =
  55    " deflate 1.2.5.3 Copyright 1995-2011 Jean-loup Gailly and Mark Adler ";
  56 /*
  57   If you use the zlib library in a product, an acknowledgment is welcome
  58   in the documentation of your product. If for some reason you cannot
  59   include such an acknowledgment, I would appreciate that you keep this
  60   copyright string in the executable of your product.
  61  */
  62
  63 /* ===========================================================================
  64  *  Function prototypes.
  65  */
  66 typedef enum {
  67     need_more,      /* block not completed, need more input or more output */
  68     block_done,     /* block flush performed */
  69     finish_started, /* finish started, need only more output at next deflate */
  70     finish_done     /* finish done, accept no more input or output */
  71 } block_state;
  72
  73 typedef block_state (*compress_func) OF((deflate_state *s, int flush));
  74 /* Compression function. Returns the block state after the call. */
  75
  76 local void fill_window    OF((deflate_state *s));
  77 local block_state deflate_stored OF((deflate_state *s, int flush));
  78 local block_state deflate_fast   OF((deflate_state *s, int flush));
  79 #ifndef FASTEST
  80 local block_state deflate_slow   OF((deflate_state *s, int flush));
  81 #endif
  82 local block_state deflate_rle    OF((deflate_state *s, int flush));
  83 local block_state deflate_huff   OF((deflate_state *s, int flush));
  84 local void lm_init        OF((deflate_state *s));
  85 local void putShortMSB    OF((deflate_state *s, uInt b));
  86 local void flush_pending  OF((z_streamp strm));
  87 local int read_buf        OF((z_streamp strm, Bytef *buf, unsigned size));
  88 #ifdef ASMV
  89       void match_init OF((void)); /* asm code initialization */
  90       uInt longest_match  OF((deflate_state *s, IPos cur_match));
  91 #else
  92 local uInt longest_match  OF((deflate_state *s, IPos cur_match));
  93 #endif
  94
  95 #ifdef DEBUG
  96 local  void check_match OF((deflate_state *s, IPos start, IPos match,
  97                             int length));
  98 #endif
  99
 100 /* ===========================================================================
 101  * Local data
 102  */
 103
 104 #define NIL 0
 105 /* Tail of hash chains */
 106
 107 #ifndef TOO_FAR
 108 #  define TOO_FAR 4096
 109 #endif
 110 /* Matches of length 3 are discarded if their distance exceeds TOO_FAR */
 111
 112 /* Values for max_lazy_match, good_match and max_chain_length, depending on
 113  * the desired pack level (0..9). The values given below have been tuned to
 114  * exclude worst case performance for pathological files. Better values may be
 115  * found for specific files.
 116  */
 117 typedef struct config_s {
 118    ush good_length; /* reduce lazy search above this match length */
 119    ush max_lazy;    /* do not perform lazy search above this match length */
 120    ush nice_length; /* quit search above this match length */
 121    ush max_chain;
 122    compress_func func;
 123 } config;
 124
 125 #ifdef FASTEST
 126 local const config configuration_table[2] = {
 127 /*      good lazy nice chain */
 128 /* 0 */ {0,    0,  0,    0, deflate_stored},  /* store only */
 129 /* 1 */ {4,    4,  8,    4, deflate_fast}}; /* max speed, no lazy matches */
 130 #else
 131 local const config configuration_table[10] = {
 132 /*      good lazy nice chain */
 133 /* 0 */ {0,    0,  0,    0, deflate_stored},  /* store only */
 134 /* 1 */ {4,    4,  8,    4, deflate_fast}, /* max speed, no lazy matches */
 135 /* 2 */ {4,    5, 16,    8, deflate_fast},
 136 /* 3 */ {4,    6, 32,   32, deflate_fast},
 137
 138 /* 4 */ {4,    4, 16,   16, deflate_slow},  /* lazy matches */
 139 /* 5 */ {8,   16, 32,   32, deflate_slow},
 140 /* 6 */ {8,   16, 128, 128, deflate_slow},
 141 /* 7 */ {8,   32, 128, 256, deflate_slow},
 142 /* 8 */ {32, 128, 258, 1024, deflate_slow},
 143 /* 9 */ {32, 258, 258, 4096, deflate_slow}}; /* max compression */
 144 #endif
 145
 146 /* Note: the deflate() code requires max_lazy >= MIN_MATCH and max_chain >= 4
 147  * For deflate_fast() (levels <= 3) good is ignored and lazy has a different
 148  * meaning.
 149  */
 150
 151 #define EQUAL 0
 152 /* result of memcmp for equal strings */
 153
 154 #ifndef NO_DUMMY_DECL
 155 struct static_tree_desc_s {int dummy;}; /* for buggy compilers */
 156 #endif
 157
 158 /* rank Z_BLOCK between Z_NO_FLUSH and Z_PARTIAL_FLUSH */
 159 #define RANK(f) (((f) << 1) - ((f) > 4 ? 9 : 0))
 160
 161 /* ===========================================================================
 162  * Update a hash value with the given input byte
 163  * IN  assertion: all calls to to UPDATE_HASH are made with consecutive
 164  *    input characters, so that a running hash key can be computed from the
 165  *    previous key instead of complete recalculation each time.
 166  */
 167 #define UPDATE_HASH(s,h,c) (h = (((h)<<s->hash_shift) ^ (c)) & s->hash_mask)
 168
 169
 170 /* ===========================================================================
 171  * Insert string str in the dictionary and set match_head to the previous head
 172  * of the hash chain (the most recent string with same hash key). Return
 173  * the previous length of the hash chain.
 174  * If this file is compiled with -DFASTEST, the compression level is forced
 175  * to 1, and no hash chains are maintained.
 176  * IN  assertion: all calls to to INSERT_STRING are made with consecutive
 177  *    input characters and the first MIN_MATCH bytes of str are valid
 178  *    (except for the last MIN_MATCH-1 bytes of the input file).
 179  */
 180 #ifdef FASTEST
 181 #define INSERT_STRING(s, str, match_head) \
 182    (UPDATE_HASH(s, s->ins_h, s->window[(str) + (MIN_MATCH-1)]), \
 183     match_head = s->head[s->ins_h], \
 184     s->head[s->ins_h] = (Pos)(str))
 185 #else
 186 #define INSERT_STRING(s, str, match_head) \
 187    (UPDATE_HASH(s, s->ins_h, s->window[(str) + (MIN_MATCH-1)]), \
 188     match_head = s->prev[(str) & s->w_mask] = s->head[s->ins_h], \
 189     s->head[s->ins_h] = (Pos)(str))
 190 #endif
 191
 192 /* ===========================================================================
 193  * Initialize the hash table (avoiding 64K overflow for 16 bit systems).
 194  * prev[] will be initialized on the fly.
 195  */
 196 #define CLEAR_HASH(s) \
 197     s->head[s->hash_size-1] = NIL; \
 198     zmemzero((Bytef *)s->head, (unsigned)(s->hash_size-1)*sizeof(*s->head));
 199
 200 /* ========================================================================= */
 201 int ZEXPORT deflateInit_(strm, level, version, stream_size)
 202     z_streamp strm;
 203     int level;
 204     const char *version;
 205     int stream_size;
 206 {
 207     return deflateInit2_(strm, level, Z_DEFLATED, MAX_WBITS, DEF_MEM_LEVEL,
 208                          Z_DEFAULT_STRATEGY, version, stream_size);
 209     /* To do: ignore strm->next_in if we use it as window */
 210 }
 211
 212 /* ========================================================================= */
 213 int ZEXPORT deflateInit2_(strm, level, method, windowBits, memLevel, strategy,
 214                   version, stream_size)
 215     z_streamp strm;
 216     int  level;
 217     int  method;
 218     int  windowBits;
 219     int  memLevel;
 220     int  strategy;
 221     const char *version;
 222     int stream_size;
 223 {
 224     deflate_state *s;
 225     int wrap = 1;
 226     static const char my_version[] = ZLIB_VERSION;
 227
 228     ushf *overlay;
 229     /* We overlay pending_buf and d_buf+l_buf. This works since the average
 230      * output size for (length,distance) codes is <= 24 bits.
 231      */
 232
 233     if (version == Z_NULL || version[0] != my_version[0] ||
 234         stream_size != sizeof(z_stream)) {
 235         return Z_VERSION_ERROR;
 236     }
 237     if (strm == Z_NULL) return Z_STREAM_ERROR;
 238
 239     strm->msg = Z_NULL;
 240     if (strm->zalloc == (alloc_func)0) {
 241 #ifdef Z_SOLO
 242         return Z_STREAM_ERROR;
 243 #else
 244         strm->zalloc = zcalloc;
 245         strm->opaque = (voidpf)0;
 246 #endif
 247     }
 248     if (strm->zfree == (free_func)0)
 249 #ifdef Z_SOLO
 250         return Z_STREAM_ERROR;
 251 #else
 252         strm->zfree = zcfree;
 253 #endif
 254
 255 #ifdef FASTEST
 256     if (level != 0) level = 1;
 257 #else
 258     if (level == Z_DEFAULT_COMPRESSION) level = 6;
 259 #endif
 260
 261     if (windowBits < 0) { /* suppress zlib wrapper */
 262         wrap = 0;
 263         windowBits = -windowBits;
 264     }
 265 #ifdef GZIP
 266     else if (windowBits > 15) {
 267         wrap = 2;       /* write gzip wrapper instead */
 268         windowBits -= 16;
 269     }
 270 #endif
 271     if (memLevel < 1 || memLevel > MAX_MEM_LEVEL || method != Z_DEFLATED ||
 272         windowBits < 8 || windowBits > 15 || level < 0 || level > 9 ||
 273         strategy < 0 || strategy > Z_FIXED) {
 274         return Z_STREAM_ERROR;
 275     }
 276     if (windowBits == 8) windowBits = 9;  /* until 256-byte window bug fixed */
 277     s = (deflate_state *) ZALLOC(strm, 1, sizeof(deflate_state));
 278     if (s == Z_NULL) return Z_MEM_ERROR;
 279     strm->state = (struct internal_state FAR *)s;
 280     s->strm = strm;
 281
 282     s->wrap = wrap;
 283     s->gzhead = Z_NULL;
 284     s->w_bits = windowBits;
 285     s->w_size = 1 << s->w_bits;
 286     s->w_mask = s->w_size - 1;
 287
 288     s->hash_bits = memLevel + 7;
 289     s->hash_size = 1 << s->hash_bits;
 290     s->hash_mask = s->hash_size - 1;
 291     s->hash_shift =  ((s->hash_bits+MIN_MATCH-1)/MIN_MATCH);
 292
 293     s->window = (Bytef *) ZALLOC(strm, s->w_size, 2*sizeof(Byte));
 294     s->prev   = (Posf *)  ZALLOC(strm, s->w_size, sizeof(Pos));
 295     s->head   = (Posf *)  ZALLOC(strm, s->hash_size, sizeof(Pos));
 296
 297     s->high_water = 0;      /* nothing written to s->window yet */
 298
 299     s->lit_bufsize = 1 << (memLevel + 6); /* 16K elements by default */
 300
 301     overlay = (ushf *) ZALLOC(strm, s->lit_bufsize, sizeof(ush)+2);
 302     s->pending_buf = (uchf *) overlay;
 303     s->pending_buf_size = (ulg)s->lit_bufsize * (sizeof(ush)+2L);
 304
 305     if (s->window == Z_NULL || s->prev == Z_NULL || s->head == Z_NULL ||
 306         s->pending_buf == Z_NULL) {
 307         s->status = FINISH_STATE;
 308         strm->msg = (char*)ERR_MSG(Z_MEM_ERROR);
 309         deflateEnd (strm);
 310         return Z_MEM_ERROR;
 311     }
 312     s->d_buf = overlay + s->lit_bufsize/sizeof(ush);
 313     s->l_buf = s->pending_buf + (1+sizeof(ush))*s->lit_bufsize;
 314
 315     s->level = level;
 316     s->strategy = strategy;
 317     s->method = (Byte)method;
 318
 319     return deflateReset(strm);
 320 }
 321
 322 /* ========================================================================= */
 323 int ZEXPORT deflateSetDictionary (strm, dictionary, dictLength)
 324     z_streamp strm;
 325     const Bytef *dictionary;
 326     uInt  dictLength;
 327 {
 328     deflate_state *s;
 329     uInt str, n;
 330     int wrap;
 331     unsigned avail;
 332     unsigned char *next;
 333
 334     if (strm == Z_NULL || strm->state == Z_NULL || dictionary == Z_NULL)
 335         return Z_STREAM_ERROR;
 336     s = strm->state;
 337     wrap = s->wrap;
 338     if (wrap == 2 || (wrap == 1 && s->status != INIT_STATE) || s->lookahead)
 339         return Z_STREAM_ERROR;
 340
 341     /* when using zlib wrappers, compute Adler-32 for provided dictionary */
 342     if (wrap == 1)
 343         strm->adler = adler32(strm->adler, dictionary, dictLength);
 344     s->wrap = 0;                    /* avoid computing Adler-32 in read_buf */
 345
 346     /* if dictionary would fill window, just replace the history */
 347     if (dictLength >= s->w_size) {
 348         if (wrap == 0) {            /* already empty otherwise */
 349             CLEAR_HASH(s);
 350             s->strstart = 0;
 351             s->block_start = 0L;
 352         }
 353         dictionary += dictLength - s->w_size;  /* use the tail */
 354         dictLength = s->w_size;
 355     }
 356
 357     /* insert dictionary into window and hash */
 358     avail = strm->avail_in;
 359     next = strm->next_in;
 360     strm->avail_in = dictLength;
 361     strm->next_in = (Bytef *)dictionary;
 362     fill_window(s);
 363     while (s->lookahead >= MIN_MATCH) {
 364         str = s->strstart;
 365         n = s->lookahead - (MIN_MATCH-1);
 366         do {
 367             UPDATE_HASH(s, s->ins_h, s->window[str + MIN_MATCH-1]);
 368 #ifndef FASTEST
 369             s->prev[str & s->w_mask] = s->head[s->ins_h];
 370 #endif
 371             s->head[s->ins_h] = (Pos)str;
 372             str++;
 373         } while (--n);
 374         s->strstart = str;
 375         s->lookahead = MIN_MATCH-1;
 376         fill_window(s);
 377     }
 378     s->strstart += s->lookahead;
 379     s->block_start = (long)s->strstart;
 380     s->lookahead = 0;
 381     s->match_length = s->prev_length = MIN_MATCH-1;
 382     s->match_available = 0;
 383     strm->next_in = next;
 384     strm->avail_in = avail;
 385     s->wrap = wrap;
 386     return Z_OK;
 387 }
 388
 389 /* ========================================================================= */
 390 int ZEXPORT deflateResetKeep (strm)
 391     z_streamp strm;
 392 {
 393     deflate_state *s;
 394
 395     if (strm == Z_NULL || strm->state == Z_NULL ||
 396         strm->zalloc == (alloc_func)0 || strm->zfree == (free_func)0) {
 397         return Z_STREAM_ERROR;
 398     }
 399
 400     strm->total_in = strm->total_out = 0;
 401     strm->msg = Z_NULL; /* use zfree if we ever allocate msg dynamically */
 402     strm->data_type = Z_UNKNOWN;
 403
 404     s = (deflate_state *)strm->state;
 405     s->pending = 0;
 406     s->pending_out = s->pending_buf;
 407
 408     if (s->wrap < 0) {
 409         s->wrap = -s->wrap; /* was made negative by deflate(..., Z_FINISH); */
 410     }
 411     s->status = s->wrap ? INIT_STATE : BUSY_STATE;
 412     strm->adler =
 413 #ifdef GZIP
 414         s->wrap == 2 ? crc32(0L, Z_NULL, 0) :
 415 #endif
 416         adler32(0L, Z_NULL, 0);
 417     s->last_flush = Z_NO_FLUSH;
 418
 419     _tr_init(s);
 420
 421     return Z_OK;
 422 }
 423
 424 /* ========================================================================= */
 425 int ZEXPORT deflateReset (strm)
 426     z_streamp strm;
 427 {
 428     int ret;
 429
 430     ret = deflateResetKeep(strm);
 431     if (ret == Z_OK)
 432         lm_init(strm->state);
 433     return ret;
 434 }
 435
 436 /* ========================================================================= */
 437 int ZEXPORT deflateSetHeader (strm, head)
 438     z_streamp strm;
 439     gz_headerp head;
 440 {
 441     if (strm == Z_NULL || strm->state == Z_NULL) return Z_STREAM_ERROR;
 442     if (strm->state->wrap != 2) return Z_STREAM_ERROR;
 443     strm->state->gzhead = head;
 444     return Z_OK;
 445 }
 446
 447 /* ========================================================================= */
 448 int ZEXPORT deflatePending (strm, pending, bits)
 449     unsigned *pending;
 450     int *bits;
 451     z_streamp strm;
 452 {
 453     if (strm == Z_NULL || strm->state == Z_NULL) return Z_STREAM_ERROR;
 454     if (pending != Z_NULL)
 455         *pending = strm->state->pending;
 456     if (bits != Z_NULL)
 457         *bits = strm->state->bi_valid;
 458     return Z_OK;
 459 }
 460
 461 /* ========================================================================= */
 462 int ZEXPORT deflatePrime (strm, bits, value)
 463     z_streamp strm;
 464     int bits;
 465     int value;
 466 {
 467     if (strm == Z_NULL || strm->state == Z_NULL) return Z_STREAM_ERROR;
 468     strm->state->bi_valid = bits;
 469     strm->state->bi_buf = (ush)(value & ((1 << bits) - 1));
 470     return Z_OK;
 471 }
 472
 473 /* ========================================================================= */
 474 int ZEXPORT deflateParams(strm, level, strategy)
 475     z_streamp strm;
 476     int level;
 477     int strategy;
 478 {
 479     deflate_state *s;
 480     compress_func func;
 481     int err = Z_OK;
 482
 483     if (strm == Z_NULL || strm->state == Z_NULL) return Z_STREAM_ERROR;
 484     s = strm->state;
 485
 486 #ifdef FASTEST
 487     if (level != 0) level = 1;
 488 #else
 489     if (level == Z_DEFAULT_COMPRESSION) level = 6;
 490 #endif
 491     if (level < 0 || level > 9 || strategy < 0 || strategy > Z_FIXED) {
 492         return Z_STREAM_ERROR;
 493     }
 494     func = configuration_table[s->level].func;
 495
 496     if ((strategy != s->strategy || func != configuration_table[level].func) &&
 497         strm->total_in != 0) {
 498         /* Flush the last buffer: */
 499         err = deflate(strm, Z_BLOCK);
 500     }
 501     if (s->level != level) {
 502         s->level = level;
 503         s->max_lazy_match   = configuration_table[level].max_lazy;
 504         s->good_match       = configuration_table[level].good_length;
 505         s->nice_match       = configuration_table[level].nice_length;
 506         s->max_chain_length = configuration_table[level].max_chain;
 507     }
 508     s->strategy = strategy;
 509     return err;
 510 }
 511
 512 /* ========================================================================= */
 513 int ZEXPORT deflateTune(strm, good_length, max_lazy, nice_length, max_chain)
 514     z_streamp strm;
 515     int good_length;
 516     int max_lazy;
 517     int nice_length;
 518     int max_chain;
 519 {
 520     deflate_state *s;
 521
 522     if (strm == Z_NULL || strm->state == Z_NULL) return Z_STREAM_ERROR;
 523     s = strm->state;
 524     s->good_match = good_length;
 525     s->max_lazy_match = max_lazy;
 526     s->nice_match = nice_length;
 527     s->max_chain_length = max_chain;
 528     return Z_OK;
 529 }
 530
 531 /* =========================================================================
 532  * For the default windowBits of 15 and memLevel of 8, this function returns
 533  * a close to exact, as well as small, upper bound on the compressed size.
 534  * They are coded as constants here for a reason--if the #define's are
 535  * changed, then this function needs to be changed as well.  The return
 536  * value for 15 and 8 only works for those exact settings.
 537  *
 538  * For any setting other than those defaults for windowBits and memLevel,
 539  * the value returned is a conservative worst case for the maximum expansion
 540  * resulting from using fixed blocks instead of stored blocks, which deflate
 541  * can emit on compressed data for some combinations of the parameters.
 542  *
 543  * This function could be more sophisticated to provide closer upper bounds for
 544  * every combination of windowBits and memLevel.  But even the conservative
 545  * upper bound of about 14% expansion does not seem onerous for output buffer
 546  * allocation.
 547  */
 548 uLong ZEXPORT deflateBound(strm, sourceLen)
 549     z_streamp strm;
 550     uLong sourceLen;
 551 {
 552     deflate_state *s;
 553     uLong complen, wraplen;
 554     Bytef *str;
 555
 556     /* conservative upper bound for compressed data */
 557     complen = sourceLen +
 558               ((sourceLen + 7) >> 3) + ((sourceLen + 63) >> 6) + 5;
 559
 560     /* if can't get parameters, return conservative bound plus zlib wrapper */
 561     if (strm == Z_NULL || strm->state == Z_NULL)
 562         return complen + 6;
 563
 564     /* compute wrapper length */
 565     s = strm->state;
 566     switch (s->wrap) {
 567     case 0:                                 /* raw deflate */
 568         wraplen = 0;
 569         break;
 570     case 1:                                 /* zlib wrapper */
 571         wraplen = 6 + (s->strstart ? 4 : 0);
 572         break;
 573     case 2:                                 /* gzip wrapper */
 574         wraplen = 18;
 575         if (s->gzhead != Z_NULL) {          /* user-supplied gzip header */
 576             if (s->gzhead->extra != Z_NULL)
 577                 wraplen += 2 + s->gzhead->extra_len;
 578             str = s->gzhead->name;
 579             if (str != Z_NULL)
 580                 do {
 581                     wraplen++;
 582                 } while (*str++);
 583             str = s->gzhead->comment;
 584             if (str != Z_NULL)
 585                 do {
 586                     wraplen++;
 587                 } while (*str++);
 588             if (s->gzhead->hcrc)
 589                 wraplen += 2;
 590         }
 591         break;
 592     default:                                /* for compiler happiness */
 593         wraplen = 6;
 594     }
 595
 596     /* if not default parameters, return conservative bound */
 597     if (s->w_bits != 15 || s->hash_bits != 8 + 7)
 598         return complen + wraplen;
 599
 600     /* default settings: return tight bound for that case */
 601     return sourceLen + (sourceLen >> 12) + (sourceLen >> 14) +
 602            (sourceLen >> 25) + 13 - 6 + wraplen;
 603 }
 604
 605 /* =========================================================================
 606  * Put a short in the pending buffer. The 16-bit value is put in MSB order.
 607  * IN assertion: the stream state is correct and there is enough room in
 608  * pending_buf.
 609  */
 610 local void putShortMSB (s, b)
 611     deflate_state *s;
 612     uInt b;
 613 {
 614     put_byte(s, (Byte)(b >> 8));
 615     put_byte(s, (Byte)(b & 0xff));
 616 }
 617
 618 /* =========================================================================
 619  * Flush as much pending output as possible. All deflate() output goes
 620  * through this function so some applications may wish to modify it
 621  * to avoid allocating a large strm->next_out buffer and copying into it.
 622  * (See also read_buf()).
 623  */
 624 local void flush_pending(strm)
 625     z_streamp strm;
 626 {
 627     unsigned len = strm->state->pending;
 628
 629     if (len > strm->avail_out) len = strm->avail_out;
 630     if (len == 0) return;
 631
 632     zmemcpy(strm->next_out, strm->state->pending_out, len);
 633     strm->next_out  += len;
 634     strm->state->pending_out  += len;
 635     strm->total_out += len;
 636     strm->avail_out  -= len;
 637     strm->state->pending -= len;
 638     if (strm->state->pending == 0) {
 639         strm->state->pending_out = strm->state->pending_buf;
 640     }
 641 }
 642
 643 /* ========================================================================= */
 644 int ZEXPORT deflate (strm, flush)
 645     z_streamp strm;
 646     int flush;
 647 {
 648     int old_flush; /* value of flush param for previous deflate call */
 649     deflate_state *s;
 650
 651     if (strm == Z_NULL || strm->state == Z_NULL ||
 652         flush > Z_BLOCK || flush < 0) {
 653         return Z_STREAM_ERROR;
 654     }
 655     s = strm->state;
 656
 657     if (strm->next_out == Z_NULL ||
 658         (strm->next_in == Z_NULL && strm->avail_in != 0) ||
 659         (s->status == FINISH_STATE && flush != Z_FINISH)) {
 660         ERR_RETURN(strm, Z_STREAM_ERROR);
 661     }
 662     if (strm->avail_out == 0) ERR_RETURN(strm, Z_BUF_ERROR);
 663
 664     s->strm = strm; /* just in case */
 665     old_flush = s->last_flush;
 666     s->last_flush = flush;
 667
 668     /* Write the header */
 669     if (s->status == INIT_STATE) {
 670 #ifdef GZIP
 671         if (s->wrap == 2) {
 672             strm->adler = crc32(0L, Z_NULL, 0);
 673             put_byte(s, 31);
 674             put_byte(s, 139);
 675             put_byte(s, 8);
 676             if (s->gzhead == Z_NULL) {
 677                 put_byte(s, 0);
 678                 put_byte(s, 0);
 679                 put_byte(s, 0);
 680                 put_byte(s, 0);
 681                 put_byte(s, 0);
 682                 put_byte(s, s->level == 9 ? 2 :
 683                             (s->strategy >= Z_HUFFMAN_ONLY || s->level < 2 ?
 684                              4 : 0));
 685                 put_byte(s, OS_CODE);
 686                 s->status = BUSY_STATE;
 687             }
 688             else {
 689                 put_byte(s, (s->gzhead->text ? 1 : 0) +
 690                             (s->gzhead->hcrc ? 2 : 0) +
 691                             (s->gzhead->extra == Z_NULL ? 0 : 4) +
 692                             (s->gzhead->name == Z_NULL ? 0 : 8) +
 693                             (s->gzhead->comment == Z_NULL ? 0 : 16)
 694                         );
 695                 put_byte(s, (Byte)(s->gzhead->time & 0xff));
 696                 put_byte(s, (Byte)((s->gzhead->time >> 8) & 0xff));
 697                 put_byte(s, (Byte)((s->gzhead->time >> 16) & 0xff));
 698                 put_byte(s, (Byte)((s->gzhead->time >> 24) & 0xff));
 699                 put_byte(s, s->level == 9 ? 2 :
 700                             (s->strategy >= Z_HUFFMAN_ONLY || s->level < 2 ?
 701                              4 : 0));
 702                 put_byte(s, s->gzhead->os & 0xff);
 703                 if (s->gzhead->extra != Z_NULL) {
 704                     put_byte(s, s->gzhead->extra_len & 0xff);
 705                     put_byte(s, (s->gzhead->extra_len >> 8) & 0xff);
 706                 }
 707                 if (s->gzhead->hcrc)
 708                     strm->adler = crc32(strm->adler, s->pending_buf,
 709                                         s->pending);
 710                 s->gzindex = 0;
 711                 s->status = EXTRA_STATE;
 712             }
 713         }
 714         else
 715 #endif
 716         {
 717             uInt header = (Z_DEFLATED + ((s->w_bits-8)<<4)) << 8;
 718             uInt level_flags;
 719
 720             if (s->strategy >= Z_HUFFMAN_ONLY || s->level < 2)
 721                 level_flags = 0;
 722             else if (s->level < 6)
 723                 level_flags = 1;
 724             else if (s->level == 6)
 725                 level_flags = 2;
 726             else
 727                 level_flags = 3;
 728             header |= (level_flags << 6);
 729             if (s->strstart != 0) header |= PRESET_DICT;
 730             header += 31 - (header % 31);
 731
 732             s->status = BUSY_STATE;
 733             putShortMSB(s, header);
 734
 735             /* Save the adler32 of the preset dictionary: */
 736             if (s->strstart != 0) {
 737                 putShortMSB(s, (uInt)(strm->adler >> 16));
 738                 putShortMSB(s, (uInt)(strm->adler & 0xffff));
 739             }
 740             strm->adler = adler32(0L, Z_NULL, 0);
 741         }
 742     }
 743 #ifdef GZIP
 744     if (s->status == EXTRA_STATE) {
 745         if (s->gzhead->extra != Z_NULL) {
 746             uInt beg = s->pending;  /* start of bytes to update crc */
 747
 748             while (s->gzindex < (s->gzhead->extra_len & 0xffff)) {
 749                 if (s->pending == s->pending_buf_size) {
 750                     if (s->gzhead->hcrc && s->pending > beg)
 751                         strm->adler = crc32(strm->adler, s->pending_buf + beg,
 752                                             s->pending - beg);
 753                     flush_pending(strm);
 754                     beg = s->pending;
 755                     if (s->pending == s->pending_buf_size)
 756                         break;
 757                 }
 758                 put_byte(s, s->gzhead->extra[s->gzindex]);
 759                 s->gzindex++;
 760             }
 761             if (s->gzhead->hcrc && s->pending > beg)
 762                 strm->adler = crc32(strm->adler, s->pending_buf + beg,
 763                                     s->pending - beg);
 764             if (s->gzindex == s->gzhead->extra_len) {
 765                 s->gzindex = 0;
 766                 s->status = NAME_STATE;
 767             }
 768         }
 769         else
 770             s->status = NAME_STATE;
 771     }
 772     if (s->status == NAME_STATE) {
 773         if (s->gzhead->name != Z_NULL) {
 774             uInt beg = s->pending;  /* start of bytes to update crc */
 775             int val;
 776
 777             do {
 778                 if (s->pending == s->pending_buf_size) {
 779                     if (s->gzhead->hcrc && s->pending > beg)
 780                         strm->adler = crc32(strm->adler, s->pending_buf + beg,
 781                                             s->pending - beg);
 782                     flush_pending(strm);
 783                     beg = s->pending;
 784                     if (s->pending == s->pending_buf_size) {
 785                         val = 1;
 786                         break;
 787                     }
 788                 }
 789                 val = s->gzhead->name[s->gzindex++];
 790                 put_byte(s, val);
 791             } while (val != 0);
 792             if (s->gzhead->hcrc && s->pending > beg)
 793                 strm->adler = crc32(strm->adler, s->pending_buf + beg,
 794                                     s->pending - beg);
 795             if (val == 0) {
 796                 s->gzindex = 0;
 797                 s->status = COMMENT_STATE;
 798             }
 799         }
 800         else
 801             s->status = COMMENT_STATE;
 802     }
 803     if (s->status == COMMENT_STATE) {
 804         if (s->gzhead->comment != Z_NULL) {
 805             uInt beg = s->pending;  /* start of bytes to update crc */
 806             int val;
 807
 808             do {
 809                 if (s->pending == s->pending_buf_size) {
 810                     if (s->gzhead->hcrc && s->pending > beg)
 811                         strm->adler = crc32(strm->adler, s->pending_buf + beg,
 812                                             s->pending - beg);
 813                     flush_pending(strm);
 814                     beg = s->pending;
 815                     if (s->pending == s->pending_buf_size) {
 816                         val = 1;
 817                         break;
 818                     }
 819                 }
 820                 val = s->gzhead->comment[s->gzindex++];
 821                 put_byte(s, val);
 822             } while (val != 0);
 823             if (s->gzhead->hcrc && s->pending > beg)
 824                 strm->adler = crc32(strm->adler, s->pending_buf + beg,
 825                                     s->pending - beg);
 826             if (val == 0)
 827                 s->status = HCRC_STATE;
 828         }
 829         else
 830             s->status = HCRC_STATE;
 831     }
 832     if (s->status == HCRC_STATE) {
 833         if (s->gzhead->hcrc) {
 834             if (s->pending + 2 > s->pending_buf_size)
 835                 flush_pending(strm);
 836             if (s->pending + 2 <= s->pending_buf_size) {
 837                 put_byte(s, (Byte)(strm->adler & 0xff));
 838                 put_byte(s, (Byte)((strm->adler >> 8) & 0xff));
 839                 strm->adler = crc32(0L, Z_NULL, 0);
 840                 s->status = BUSY_STATE;
 841             }
 842         }
 843         else
 844             s->status = BUSY_STATE;
 845     }
 846 #endif
 847
 848     /* Flush as much pending output as possible */
 849     if (s->pending != 0) {
 850         flush_pending(strm);
 851         if (strm->avail_out == 0) {
 852             /* Since avail_out is 0, deflate will be called again with
 853              * more output space, but possibly with both pending and
 854              * avail_in equal to zero. There won't be anything to do,
 855              * but this is not an error situation so make sure we
 856              * return OK instead of BUF_ERROR at next call of deflate:
 857              */
 858             s->last_flush = -1;
 859             return Z_OK;
 860         }
 861
 862     /* Make sure there is something to do and avoid duplicate consecutive
 863      * flushes. For repeated and useless calls with Z_FINISH, we keep
 864      * returning Z_STREAM_END instead of Z_BUF_ERROR.
 865      */
 866     } else if (strm->avail_in == 0 && RANK(flush) <= RANK(old_flush) &&
 867                flush != Z_FINISH) {
 868         ERR_RETURN(strm, Z_BUF_ERROR);
 869     }
 870
 871     /* User must not provide more input after the first FINISH: */
 872     if (s->status == FINISH_STATE && strm->avail_in != 0) {
 873         ERR_RETURN(strm, Z_BUF_ERROR);
 874     }
 875
 876     /* Start a new block or continue the current one.
 877      */
 878     if (strm->avail_in != 0 || s->lookahead != 0 ||
 879         (flush != Z_NO_FLUSH && s->status != FINISH_STATE)) {
 880         block_state bstate;
 881
 882         bstate = s->strategy == Z_HUFFMAN_ONLY ? deflate_huff(s, flush) :
 883                     (s->strategy == Z_RLE ? deflate_rle(s, flush) :
 884                         (*(configuration_table[s->level].func))(s, flush));
 885
 886         if (bstate == finish_started || bstate == finish_done) {
 887             s->status = FINISH_STATE;
 888         }
 889         if (bstate == need_more || bstate == finish_started) {
 890             if (strm->avail_out == 0) {
 891                 s->last_flush = -1; /* avoid BUF_ERROR next call, see above */
 892             }
 893             return Z_OK;
 894             /* If flush != Z_NO_FLUSH && avail_out == 0, the next call
 895              * of deflate should use the same flush parameter to make sure
 896              * that the flush is complete. So we don't have to output an
 897              * empty block here, this will be done at next call. This also
 898              * ensures that for a very small output buffer, we emit at most
 899              * one empty block.
 900              */
 901         }
 902         if (bstate == block_done) {
 903             if (flush == Z_PARTIAL_FLUSH) {
 904                 _tr_align(s);
 905             } else if (flush != Z_BLOCK) { /* FULL_FLUSH or SYNC_FLUSH */
 906                 _tr_stored_block(s, (char*)0, 0L, 0);
 907                 /* For a full flush, this empty block will be recognized
 908                  * as a special marker by inflate_sync().
 909                  */
 910                 if (flush == Z_FULL_FLUSH) {
 911                     CLEAR_HASH(s);             /* forget history */
 912                     if (s->lookahead == 0) {
 913                         s->strstart = 0;
 914                         s->block_start = 0L;
 915                     }
 916                 }
 917             }
 918             flush_pending(strm);
 919             if (strm->avail_out == 0) {
 920               s->last_flush = -1; /* avoid BUF_ERROR at next call, see above */
 921               return Z_OK;
 922             }
 923         }
 924     }
 925     Assert(strm->avail_out > 0, "bug2");
 926
 927     if (flush != Z_FINISH) return Z_OK;
 928     if (s->wrap <= 0) return Z_STREAM_END;
 929
 930     /* Write the trailer */
 931 #ifdef GZIP
 932     if (s->wrap == 2) {
 933         put_byte(s, (Byte)(strm->adler & 0xff));
 934         put_byte(s, (Byte)((strm->adler >> 8) & 0xff));
 935         put_byte(s, (Byte)((strm->adler >> 16) & 0xff));
 936         put_byte(s, (Byte)((strm->adler >> 24) & 0xff));
 937         put_byte(s, (Byte)(strm->total_in & 0xff));
 938         put_byte(s, (Byte)((strm->total_in >> 8) & 0xff));
 939         put_byte(s, (Byte)((strm->total_in >> 16) & 0xff));
 940         put_byte(s, (Byte)((strm->total_in >> 24) & 0xff));
 941     }
 942     else
 943 #endif
 944     {
 945         putShortMSB(s, (uInt)(strm->adler >> 16));
 946         putShortMSB(s, (uInt)(strm->adler & 0xffff));
 947     }
 948     flush_pending(strm);
 949     /* If avail_out is zero, the application will call deflate again
 950      * to flush the rest.
 951      */
 952     if (s->wrap > 0) s->wrap = -s->wrap; /* write the trailer only once! */
 953     return s->pending != 0 ? Z_OK : Z_STREAM_END;
 954 }
 955
 956 /* ========================================================================= */
 957 int ZEXPORT deflateEnd (strm)
 958     z_streamp strm;
 959 {
 960     int status;
 961
 962     if (strm == Z_NULL || strm->state == Z_NULL) return Z_STREAM_ERROR;
 963
 964     status = strm->state->status;
 965     if (status != INIT_STATE &&
 966         status != EXTRA_STATE &&
 967         status != NAME_STATE &&
 968         status != COMMENT_STATE &&
 969         status != HCRC_STATE &&
 970         status != BUSY_STATE &&
 971         status != FINISH_STATE) {
 972       return Z_STREAM_ERROR;
 973     }
 974
 975     /* Deallocate in reverse order of allocations: */
 976     TRY_FREE(strm, strm->state->pending_buf);
 977     TRY_FREE(strm, strm->state->head);
 978     TRY_FREE(strm, strm->state->prev);
 979     TRY_FREE(strm, strm->state->window);
 980
 981     ZFREE(strm, strm->state);
 982     strm->state = Z_NULL;
 983
 984     return status == BUSY_STATE ? Z_DATA_ERROR : Z_OK;
 985 }
 986
 987 /* =========================================================================
 988  * Copy the source state to the destination state.
 989  * To simplify the source, this is not supported for 16-bit MSDOS (which
 990  * doesn't have enough memory anyway to duplicate compression states).
 991  */
 992 int ZEXPORT deflateCopy (dest, source)
 993     z_streamp dest;
 994     z_streamp source;
 995 {
 996 #ifdef MAXSEG_64K
 997     return Z_STREAM_ERROR;
 998 #else
 999     deflate_state *ds;
1000     deflate_state *ss;
1001     ushf *overlay;
1002
1003
1004     if (source == Z_NULL || dest == Z_NULL || source->state == Z_NULL) {
1005         return Z_STREAM_ERROR;
1006     }
1007
1008     ss = source->state;
1009
1010     zmemcpy((voidpf)dest, (voidpf)source, sizeof(z_stream));
1011
1012     ds = (deflate_state *) ZALLOC(dest, 1, sizeof(deflate_state));
1013     if (ds == Z_NULL) return Z_MEM_ERROR;
1014     dest->state = (struct internal_state FAR *) ds;
1015     zmemcpy((voidpf)ds, (voidpf)ss, sizeof(deflate_state));
1016     ds->strm = dest;
1017
1018     ds->window = (Bytef *) ZALLOC(dest, ds->w_size, 2*sizeof(Byte));
1019     ds->prev   = (Posf *)  ZALLOC(dest, ds->w_size, sizeof(Pos));
1020     ds->head   = (Posf *)  ZALLOC(dest, ds->hash_size, sizeof(Pos));
1021     overlay = (ushf *) ZALLOC(dest, ds->lit_bufsize, sizeof(ush)+2);
1022     ds->pending_buf = (uchf *) overlay;
1023
1024     if (ds->window == Z_NULL || ds->prev == Z_NULL || ds->head == Z_NULL ||
1025         ds->pending_buf == Z_NULL) {
1026         deflateEnd (dest);
1027         return Z_MEM_ERROR;
1028     }
1029     /* following zmemcpy do not work for 16-bit MSDOS */
1030     zmemcpy(ds->window, ss->window, ds->w_size * 2 * sizeof(Byte));
1031     zmemcpy((voidpf)ds->prev, (voidpf)ss->prev, ds->w_size * sizeof(Pos));
1032     zmemcpy((voidpf)ds->head, (voidpf)ss->head, ds->hash_size * sizeof(Pos));
1033     zmemcpy(ds->pending_buf, ss->pending_buf, (uInt)ds->pending_buf_size);
1034
1035     ds->pending_out = ds->pending_buf + (ss->pending_out - ss->pending_buf);
1036     ds->d_buf = overlay + ds->lit_bufsize/sizeof(ush);
1037     ds->l_buf = ds->pending_buf + (1+sizeof(ush))*ds->lit_bufsize;
1038
1039     ds->l_desc.dyn_tree = ds->dyn_ltree;
1040     ds->d_desc.dyn_tree = ds->dyn_dtree;
1041     ds->bl_desc.dyn_tree = ds->bl_tree;
1042
1043     return Z_OK;
1044 #endif /* MAXSEG_64K */
1045 }
1046
1047 /* ===========================================================================
1048  * Read a new buffer from the current input stream, update the adler32
1049  * and total number of bytes read.  All deflate() input goes through
1050  * this function so some applications may wish to modify it to avoid
1051  * allocating a large strm->next_in buffer and copying from it.
1052  * (See also flush_pending()).
1053  */
1054 local int read_buf(strm, buf, size)
1055     z_streamp strm;
1056     Bytef *buf;
1057     unsigned size;
1058 {
1059     unsigned len = strm->avail_in;
1060
1061     if (len > size) len = size;
1062     if (len == 0) return 0;
1063
1064     strm->avail_in  -= len;
1065
1066     zmemcpy(buf, strm->next_in, len);
1067     if (strm->state->wrap == 1) {
1068         strm->adler = adler32(strm->adler, buf, len);
1069     }
1070 #ifdef GZIP
1071     else if (strm->state->wrap == 2) {
1072         strm->adler = crc32(strm->adler, buf, len);
1073     }
1074 #endif
1075     strm->next_in  += len;
1076     strm->total_in += len;
1077
1078     return (int)len;
1079 }
1080
1081 /* ===========================================================================
1082  * Initialize the "longest match" routines for a new zlib stream
1083  */
1084 local void lm_init (s)
1085     deflate_state *s;
1086 {
1087     s->window_size = (ulg)2L*s->w_size;
1088
1089     CLEAR_HASH(s);
1090
1091     /* Set the default configuration parameters:
1092      */
1093     s->max_lazy_match   = configuration_table[s->level].max_lazy;
1094     s->good_match       = configuration_table[s->level].good_length;
1095     s->nice_match       = configuration_table[s->level].nice_length;
1096     s->max_chain_length = configuration_table[s->level].max_chain;
1097
1098     s->strstart = 0;
1099     s->block_start = 0L;
1100     s->lookahead = 0;
1101     s->match_length = s->prev_length = MIN_MATCH-1;
1102     s->match_available = 0;
1103     s->ins_h = 0;
1104 #ifndef FASTEST
1105 #ifdef ASMV
1106     match_init(); /* initialize the asm code */
1107 #endif
1108 #endif
1109 }
1110
1111 #ifndef FASTEST
1112 /* ===========================================================================
1113  * Set match_start to the longest match starting at the given string and
1114  * return its length. Matches shorter or equal to prev_length are discarded,
1115  * in which case the result is equal to prev_length and match_start is
1116  * garbage.
1117  * IN assertions: cur_match is the head of the hash chain for the current
1118  *   string (strstart) and its distance is <= MAX_DIST, and prev_length >= 1
1119  * OUT assertion: the match length is not greater than s->lookahead.
1120  */
1121 #ifndef ASMV
1122 /* For 80x86 and 680x0, an optimized version will be provided in match.asm or
1123  * match.S. The code will be functionally equivalent.
1124  */
1125 local uInt longest_match(s, cur_match)
1126     deflate_state *s;
1127     IPos cur_match;                             /* current match */
1128 {
1129     unsigned chain_length = s->max_chain_length;/* max hash chain length */
1130     register Bytef *scan = s->window + s->strstart; /* current string */
1131     register Bytef *match;                       /* matched string */
1132     register int len;                           /* length of current match */
1133     int best_len = s->prev_length;              /* best match length so far */
1134     int nice_match = s->nice_match;             /* stop if match long enough */
1135     IPos limit = s->strstart > (IPos)MAX_DIST(s) ?
1136         s->strstart - (IPos)MAX_DIST(s) : NIL;
1137     /* Stop when cur_match becomes <= limit. To simplify the code,
1138      * we prevent matches with the string of window index 0.
1139      */
1140     Posf *prev = s->prev;
1141     uInt wmask = s->w_mask;
1142
1143 #ifdef UNALIGNED_OK
1144     /* Compare two bytes at a time. Note: this is not always beneficial.
1145      * Try with and without -DUNALIGNED_OK to check.
1146      */
1147     register Bytef *strend = s->window + s->strstart + MAX_MATCH - 1;
1148     register ush scan_start = *(ushf*)scan;
1149     register ush scan_end   = *(ushf*)(scan+best_len-1);
1150 #else
1151     register Bytef *strend = s->window + s->strstart + MAX_MATCH;
1152     register Byte scan_end1  = scan[best_len-1];
1153     register Byte scan_end   = scan[best_len];
1154 #endif
1155
1156     /* The code is optimized for HASH_BITS >= 8 and MAX_MATCH-2 multiple of 16.
1157      * It is easy to get rid of this optimization if necessary.
1158      */
1159     Assert(s->hash_bits >= 8 && MAX_MATCH == 258, "Code too clever");
1160
1161     /* Do not waste too much time if we already have a good match: */
1162     if (s->prev_length >= s->good_match) {
1163         chain_length >>= 2;
1164     }
1165     /* Do not look for matches beyond the end of the input. This is necessary
1166      * to make deflate deterministic.
1167      */
1168     if ((uInt)nice_match > s->lookahead) nice_match = s->lookahead;
1169
1170     Assert((ulg)s->strstart <= s->window_size-MIN_LOOKAHEAD, "need lookahead");
1171
1172     do {
1173         Assert(cur_match < s->strstart, "no future");
1174         match = s->window + cur_match;
1175
1176         /* Skip to next match if the match length cannot increase
1177          * or if the match length is less than 2.  Note that the checks below
1178          * for insufficient lookahead only occur occasionally for performance
1179          * reasons.  Therefore uninitialized memory will be accessed, and
1180          * conditional jumps will be made that depend on those values.
1181          * However the length of the match is limited to the lookahead, so
1182          * the output of deflate is not affected by the uninitialized values.
1183          */
1184 #if (defined(UNALIGNED_OK) && MAX_MATCH == 258)
1185         /* This code assumes sizeof(unsigned short) == 2. Do not use
1186          * UNALIGNED_OK if your compiler uses a different size.
1187          */
1188         if (*(ushf*)(match+best_len-1) != scan_end ||
1189             *(ushf*)match != scan_start) continue;
1190
1191         /* It is not necessary to compare scan[2] and match[2] since they are
1192          * always equal when the other bytes match, given that the hash keys
1193          * are equal and that HASH_BITS >= 8. Compare 2 bytes at a time at
1194          * strstart+3, +5, ... up to strstart+257. We check for insufficient
1195          * lookahead only every 4th comparison; the 128th check will be made
1196          * at strstart+257. If MAX_MATCH-2 is not a multiple of 8, it is
1197          * necessary to put more guard bytes at the end of the window, or
1198          * to check more often for insufficient lookahead.
1199          */
1200         Assert(scan[2] == match[2], "scan[2]?");
1201         scan++, match++;
1202         do {
1203         } while (*(ushf*)(scan+=2) == *(ushf*)(match+=2) &&
1204                  *(ushf*)(scan+=2) == *(ushf*)(match+=2) &&
1205                  *(ushf*)(scan+=2) == *(ushf*)(match+=2) &&
1206                  *(ushf*)(scan+=2) == *(ushf*)(match+=2) &&
1207                  scan < strend);
1208         /* The funny "do {}" generates better code on most compilers */
1209
1210         /* Here, scan <= window+strstart+257 */
1211         Assert(scan <= s->window+(unsigned)(s->window_size-1), "wild scan");
1212         if (*scan == *match) scan++;
1213
1214         len = (MAX_MATCH - 1) - (int)(strend-scan);
1215         scan = strend - (MAX_MATCH-1);
1216
1217 #else /* UNALIGNED_OK */
1218
1219         if (match[best_len]   != scan_end  ||
1220             match[best_len-1] != scan_end1 ||
1221             *match            != *scan     ||
1222             *++match          != scan[1])      continue;
1223
1224         /* The check at best_len-1 can be removed because it will be made
1225          * again later. (This heuristic is not always a win.)
1226          * It is not necessary to compare scan[2] and match[2] since they
1227          * are always equal when the other bytes match, given that
1228          * the hash keys are equal and that HASH_BITS >= 8.
1229          */
1230         scan += 2, match++;
1231         Assert(*scan == *match, "match[2]?");
1232
1233         /* We check for insufficient lookahead only every 8th comparison;
1234          * the 256th check will be made at strstart+258.
1235          */
1236         do {
1237         } while (*++scan == *++match && *++scan == *++match &&
1238                  *++scan == *++match && *++scan == *++match &&
1239                  *++scan == *++match && *++scan == *++match &&
1240                  *++scan == *++match && *++scan == *++match &&
1241                  scan < strend);
1242
1243         Assert(scan <= s->window+(unsigned)(s->window_size-1), "wild scan");
1244
1245         len = MAX_MATCH - (int)(strend - scan);
1246         scan = strend - MAX_MATCH;
1247
1248 #endif /* UNALIGNED_OK */
1249
1250         if (len > best_len) {
1251             s->match_start = cur_match;
1252             best_len = len;
1253             if (len >= nice_match) break;
1254 #ifdef UNALIGNED_OK
1255             scan_end = *(ushf*)(scan+best_len-1);
1256 #else
1257             scan_end1  = scan[best_len-1];
1258             scan_end   = scan[best_len];
1259 #endif
1260         }
1261     } while ((cur_match = prev[cur_match & wmask]) > limit
1262              && --chain_length != 0);
1263
1264     if ((uInt)best_len <= s->lookahead) return (uInt)best_len;
1265     return s->lookahead;
1266 }
1267 #endif /* ASMV */
1268
1269 #else /* FASTEST */
1270
1271 /* ---------------------------------------------------------------------------
1272  * Optimized version for FASTEST only
1273  */
1274 local uInt longest_match(s, cur_match)
1275     deflate_state *s;
1276     IPos cur_match;                             /* current match */
1277 {
1278     register Bytef *scan = s->window + s->strstart; /* current string */
1279     register Bytef *match;                       /* matched string */
1280     register int len;                           /* length of current match */
1281     register Bytef *strend = s->window + s->strstart + MAX_MATCH;
1282
1283     /* The code is optimized for HASH_BITS >= 8 and MAX_MATCH-2 multiple of 16.
1284      * It is easy to get rid of this optimization if necessary.
1285      */
1286     Assert(s->hash_bits >= 8 && MAX_MATCH == 258, "Code too clever");
1287
1288     Assert((ulg)s->strstart <= s->window_size-MIN_LOOKAHEAD, "need lookahead");
1289
1290     Assert(cur_match < s->strstart, "no future");
1291
1292     match = s->window + cur_match;
1293
1294     /* Return failure if the match length is less than 2:
1295      */
1296     if (match[0] != scan[0] || match[1] != scan[1]) return MIN_MATCH-1;
1297
1298     /* The check at best_len-1 can be removed because it will be made
1299      * again later. (This heuristic is not always a win.)
1300      * It is not necessary to compare scan[2] and match[2] since they
1301      * are always equal when the other bytes match, given that
1302      * the hash keys are equal and that HASH_BITS >= 8.
1303      */
1304     scan += 2, match += 2;
1305     Assert(*scan == *match, "match[2]?");
1306
1307     /* We check for insufficient lookahead only every 8th comparison;
1308      * the 256th check will be made at strstart+258.
1309      */
1310     do {
1311     } while (*++scan == *++match && *++scan == *++match &&
1312              *++scan == *++match && *++scan == *++match &&
1313              *++scan == *++match && *++scan == *++match &&
1314              *++scan == *++match && *++scan == *++match &&
1315              scan < strend);
1316
1317     Assert(scan <= s->window+(unsigned)(s->window_size-1), "wild scan");
1318
1319     len = MAX_MATCH - (int)(strend - scan);
1320
1321     if (len < MIN_MATCH) return MIN_MATCH - 1;
1322
1323     s->match_start = cur_match;
1324     return (uInt)len <= s->lookahead ? (uInt)len : s->lookahead;
1325 }
1326
1327 #endif /* FASTEST */
1328
1329 #ifdef DEBUG
1330 /* ===========================================================================
1331  * Check that the match at match_start is indeed a match.
1332  */
1333 local void check_match(s, start, match, length)
1334     deflate_state *s;
1335     IPos start, match;
1336     int length;
1337 {
1338     /* check that the match is indeed a match */
1339     if (zmemcmp(s->window + match,
1340                 s->window + start, length) != EQUAL) {
1341         fprintf(stderr, " start %u, match %u, length %d\n",
1342                 start, match, length);
1343         do {
1344             fprintf(stderr, "%c%c", s->window[match++], s->window[start++]);
1345         } while (--length != 0);
1346         z_error("invalid match");
1347     }
1348     if (z_verbose > 1) {
1349         fprintf(stderr,"\\[%d,%d]", start-match, length);
1350         do { putc(s->window[start++], stderr); } while (--length != 0);
1351     }
1352 }
1353 #else
1354 #  define check_match(s, start, match, length)
1355 #endif /* DEBUG */
1356
1357 /* ===========================================================================
1358  * Fill the window when the lookahead becomes insufficient.
1359  * Updates strstart and lookahead.
1360  *
1361  * IN assertion: lookahead < MIN_LOOKAHEAD
1362  * OUT assertions: strstart <= window_size-MIN_LOOKAHEAD
1363  *    At least one byte has been read, or avail_in == 0; reads are
1364  *    performed for at least two bytes (required for the zip translate_eol
1365  *    option -- not supported here).
1366  */
1367 local void fill_window(s)
1368     deflate_state *s;
1369 {
1370     register unsigned n, m;
1371     register Posf *p;
1372     unsigned more;    /* Amount of free space at the end of the window. */
1373     uInt wsize = s->w_size;
1374
1375     Assert(s->lookahead < MIN_LOOKAHEAD, "already enough lookahead");
1376
1377     do {
1378         more = (unsigned)(s->window_size -(ulg)s->lookahead -(ulg)s->strstart);
1379
1380         /* Deal with !@#$% 64K limit: */
1381         if (sizeof(int) <= 2) {
1382             if (more == 0 && s->strstart == 0 && s->lookahead == 0) {
1383                 more = wsize;
1384
1385             } else if (more == (unsigned)(-1)) {
1386                 /* Very unlikely, but possible on 16 bit machine if
1387                  * strstart == 0 && lookahead == 1 (input done a byte at time)
1388                  */
1389                 more--;
1390             }
1391         }
1392
1393         /* If the window is almost full and there is insufficient lookahead,
1394          * move the upper half to the lower one to make room in the upper half.
1395          */
1396         if (s->strstart >= wsize+MAX_DIST(s)) {
1397
1398             zmemcpy(s->window, s->window+wsize, (unsigned)wsize);
1399             s->match_start -= wsize;
1400             s->strstart    -= wsize; /* we now have strstart >= MAX_DIST */
1401             s->block_start -= (long) wsize;
1402
1403             /* Slide the hash table (could be avoided with 32 bit values
1404                at the expense of memory usage). We slide even when level == 0
1405                to keep the hash table consistent if we switch back to level > 0
1406                later. (Using level 0 permanently is not an optimal usage of
1407                zlib, so we don't care about this pathological case.)
1408              */
1409             n = s->hash_size;
1410             p = &s->head[n];
1411             do {
1412                 m = *--p;
1413                 *p = (Pos)(m >= wsize ? m-wsize : NIL);
1414             } while (--n);
1415
1416             n = wsize;
1417 #ifndef FASTEST
1418             p = &s->prev[n];
1419             do {
1420                 m = *--p;
1421                 *p = (Pos)(m >= wsize ? m-wsize : NIL);
1422                 /* If n is not on any hash chain, prev[n] is garbage but
1423                  * its value will never be used.
1424                  */
1425             } while (--n);
1426 #endif
1427             more += wsize;
1428         }
1429         if (s->strm->avail_in == 0) break;
1430
1431         /* If there was no sliding:
1432          *    strstart <= WSIZE+MAX_DIST-1 && lookahead <= MIN_LOOKAHEAD - 1 &&
1433          *    more == window_size - lookahead - strstart
1434          * => more >= window_size - (MIN_LOOKAHEAD-1 + WSIZE + MAX_DIST-1)
1435          * => more >= window_size - 2*WSIZE + 2
1436          * In the BIG_MEM or MMAP case (not yet supported),
1437          *   window_size == input_size + MIN_LOOKAHEAD  &&
1438          *   strstart + s->lookahead <= input_size => more >= MIN_LOOKAHEAD.
1439          * Otherwise, window_size == 2*WSIZE so more >= 2.
1440          * If there was sliding, more >= WSIZE. So in all cases, more >= 2.
1441          */
1442         Assert(more >= 2, "more < 2");
1443
1444         n = read_buf(s->strm, s->window + s->strstart + s->lookahead, more);
1445         s->lookahead += n;
1446
1447         /* Initialize the hash value now that we have some input: */
1448         if (s->lookahead >= MIN_MATCH) {
1449             s->ins_h = s->window[s->strstart];
1450             UPDATE_HASH(s, s->ins_h, s->window[s->strstart+1]);
1451 #if MIN_MATCH != 3
1452             Call UPDATE_HASH() MIN_MATCH-3 more times
1453 #endif
1454         }
1455         /* If the whole input has less than MIN_MATCH bytes, ins_h is garbage,
1456          * but this is not important since only literal bytes will be emitted.
1457          */
1458
1459     } while (s->lookahead < MIN_LOOKAHEAD && s->strm->avail_in != 0);
1460
1461     /* If the WIN_INIT bytes after the end of the current data have never been
1462      * written, then zero those bytes in order to avoid memory check reports of
1463      * the use of uninitialized (or uninitialised as Julian writes) bytes by
1464      * the longest match routines.  Update the high water mark for the next
1465      * time through here.  WIN_INIT is set to MAX_MATCH since the longest match
1466      * routines allow scanning to strstart + MAX_MATCH, ignoring lookahead.
1467      */
1468     if (s->high_water < s->window_size) {
1469         ulg curr = s->strstart + (ulg)(s->lookahead);
1470         ulg init;
1471
1472         if (s->high_water < curr) {
1473             /* Previous high water mark below current data -- zero WIN_INIT
1474              * bytes or up to end of window, whichever is less.
1475              */
1476             init = s->window_size - curr;
1477             if (init > WIN_INIT)
1478                 init = WIN_INIT;
1479             zmemzero(s->window + curr, (unsigned)init);
1480             s->high_water = curr + init;
1481         }
1482         else if (s->high_water < (ulg)curr + WIN_INIT) {
1483             /* High water mark at or above current data, but below current data
1484              * plus WIN_INIT -- zero out to current data plus WIN_INIT, or up
1485              * to end of window, whichever is less.
1486              */
1487             init = (ulg)curr + WIN_INIT - s->high_water;
1488             if (init > s->window_size - s->high_water)
1489                 init = s->window_size - s->high_water;
1490             zmemzero(s->window + s->high_water, (unsigned)init);
1491             s->high_water += init;
1492         }
1493     }
1494
1495     Assert((ulg)s->strstart <= s->window_size - MIN_LOOKAHEAD,
1496            "not enough room for search");
1497 }
1498
1499 /* ===========================================================================
1500  * Flush the current block, with given end-of-file flag.
1501  * IN assertion: strstart is set to the end of the current match.
1502  */
1503 #define FLUSH_BLOCK_ONLY(s, last) { \
1504    _tr_flush_block(s, (s->block_start >= 0L ? \
1505                    (charf *)&s->window[(unsigned)s->block_start] : \
1506                    (charf *)Z_NULL), \
1507                 (ulg)((long)s->strstart - s->block_start), \
1508                 (last)); \
1509    s->block_start = s->strstart; \
1510    flush_pending(s->strm); \
1511    Tracev((stderr,"[FLUSH]")); \
1512 }
1513
1514 /* Same but force premature exit if necessary. */
1515 #define FLUSH_BLOCK(s, last) { \
1516    FLUSH_BLOCK_ONLY(s, last); \
1517    if (s->strm->avail_out == 0) return (last) ? finish_started : need_more; \
1518 }
1519
1520 /* ===========================================================================
1521  * Copy without compression as much as possible from the input stream, return
1522  * the current block state.
1523  * This function does not insert new strings in the dictionary since
1524  * uncompressible data is probably not useful. This function is used
1525  * only for the level=0 compression option.
1526  * NOTE: this function should be optimized to avoid extra copying from
1527  * window to pending_buf.
1528  */
1529 local block_state deflate_stored(s, flush)
1530     deflate_state *s;
1531     int flush;
1532 {
1533     /* Stored blocks are limited to 0xffff bytes, pending_buf is limited
1534      * to pending_buf_size, and each stored block has a 5 byte header:
1535      */
1536     ulg max_block_size = 0xffff;
1537     ulg max_start;
1538
1539     if (max_block_size > s->pending_buf_size - 5) {
1540         max_block_size = s->pending_buf_size - 5;
1541     }
1542
1543     /* Copy as much as possible from input to output: */
1544     for (;;) {
1545         /* Fill the window as much as possible: */
1546         if (s->lookahead <= 1) {
1547
1548             Assert(s->strstart < s->w_size+MAX_DIST(s) ||
1549                    s->block_start >= (long)s->w_size, "slide too late");
1550
1551             fill_window(s);
1552             if (s->lookahead == 0 && flush == Z_NO_FLUSH) return need_more;
1553
1554             if (s->lookahead == 0) break; /* flush the current block */
1555         }
1556         Assert(s->block_start >= 0L, "block gone");
1557
1558         s->strstart += s->lookahead;
1559         s->lookahead = 0;
1560
1561         /* Emit a stored block if pending_buf will be full: */
1562         max_start = s->block_start + max_block_size;
1563         if (s->strstart == 0 || (ulg)s->strstart >= max_start) {
1564             /* strstart == 0 is possible when wraparound on 16-bit machine */
1565             s->lookahead = (uInt)(s->strstart - max_start);
1566             s->strstart = (uInt)max_start;
1567             FLUSH_BLOCK(s, 0);
1568         }
1569         /* Flush if we may have to slide, otherwise block_start may become
1570          * negative and the data will be gone:
1571          */
1572         if (s->strstart - (uInt)s->block_start >= MAX_DIST(s)) {
1573             FLUSH_BLOCK(s, 0);
1574         }
1575     }
1576     if (flush == Z_FINISH) {
1577         FLUSH_BLOCK(s, 1);
1578         return finish_done;
1579     }
1580     if ((long)s->strstart > s->block_start)
1581         FLUSH_BLOCK(s, 0);
1582     return block_done;
1583 }
1584
1585 /* ===========================================================================
1586  * Compress as much as possible from the input stream, return the current
1587  * block state.
1588  * This function does not perform lazy evaluation of matches and inserts
1589  * new strings in the dictionary only for unmatched strings or for short
1590  * matches. It is used only for the fast compression options.
1591  */
1592 local block_state deflate_fast(s, flush)
1593     deflate_state *s;
1594     int flush;
1595 {
1596     IPos hash_head;       /* head of the hash chain */
1597     int bflush;           /* set if current block must be flushed */
1598
1599     for (;;) {
1600         /* Make sure that we always have enough lookahead, except
1601          * at the end of the input file. We need MAX_MATCH bytes
1602          * for the next match, plus MIN_MATCH bytes to insert the
1603          * string following the next match.
1604          */
1605         if (s->lookahead < MIN_LOOKAHEAD) {
1606             fill_window(s);
1607             if (s->lookahead < MIN_LOOKAHEAD && flush == Z_NO_FLUSH) {
1608                 return need_more;
1609             }
1610             if (s->lookahead == 0) break; /* flush the current block */
1611         }
1612
1613         /* Insert the string window[strstart .. strstart+2] in the
1614          * dictionary, and set hash_head to the head of the hash chain:
1615          */
1616         hash_head = NIL;
1617         if (s->lookahead >= MIN_MATCH) {
1618             INSERT_STRING(s, s->strstart, hash_head);
1619         }
1620
1621         /* Find the longest match, discarding those <= prev_length.
1622          * At this point we have always match_length < MIN_MATCH
1623          */
1624         if (hash_head != NIL && s->strstart - hash_head <= MAX_DIST(s)) {
1625             /* To simplify the code, we prevent matches with the string
1626              * of window index 0 (in particular we have to avoid a match
1627              * of the string with itself at the start of the input file).
1628              */
1629             s->match_length = longest_match (s, hash_head);
1630             /* longest_match() sets match_start */
1631         }
1632         if (s->match_length >= MIN_MATCH) {
1633             check_match(s, s->strstart, s->match_start, s->match_length);
1634
1635             _tr_tally_dist(s, s->strstart - s->match_start,
1636                            s->match_length - MIN_MATCH, bflush);
1637
1638             s->lookahead -= s->match_length;
1639
1640             /* Insert new strings in the hash table only if the match length
1641              * is not too large. This saves time but degrades compression.
1642              */
1643 #ifndef FASTEST
1644             if (s->match_length <= s->max_insert_length &&
1645                 s->lookahead >= MIN_MATCH) {
1646                 s->match_length--; /* string at strstart already in table */
1647                 do {
1648                     s->strstart++;
1649                     INSERT_STRING(s, s->strstart, hash_head);
1650                     /* strstart never exceeds WSIZE-MAX_MATCH, so there are
1651                      * always MIN_MATCH bytes ahead.
1652                      */
1653                 } while (--s->match_length != 0);
1654                 s->strstart++;
1655             } else
1656 #endif
1657             {
1658                 s->strstart += s->match_length;
1659                 s->match_length = 0;
1660                 s->ins_h = s->window[s->strstart];
1661                 UPDATE_HASH(s, s->ins_h, s->window[s->strstart+1]);
1662 #if MIN_MATCH != 3
1663                 Call UPDATE_HASH() MIN_MATCH-3 more times
1664 #endif
1665                 /* If lookahead < MIN_MATCH, ins_h is garbage, but it does not
1666                  * matter since it will be recomputed at next deflate call.
1667                  */
1668             }
1669         } else {
1670             /* No match, output a literal byte */
1671             Tracevv((stderr,"%c", s->window[s->strstart]));
1672             _tr_tally_lit (s, s->window[s->strstart], bflush);
1673             s->lookahead--;
1674             s->strstart++;
1675         }
1676         if (bflush) FLUSH_BLOCK(s, 0);
1677     }
1678     if (flush == Z_FINISH) {
1679         FLUSH_BLOCK(s, 1);
1680         return finish_done;
1681     }
1682     if (s->last_lit)
1683         FLUSH_BLOCK(s, 0);
1684     return block_done;
1685 }
1686
1687 #ifndef FASTEST
1688 /* ===========================================================================
1689  * Same as above, but achieves better compression. We use a lazy
1690  * evaluation for matches: a match is finally adopted only if there is
1691  * no better match at the next window position.
1692  */
1693 local block_state deflate_slow(s, flush)
1694     deflate_state *s;
1695     int flush;
1696 {
1697     IPos hash_head;          /* head of hash chain */
1698     int bflush;              /* set if current block must be flushed */
1699
1700     /* Process the input block. */
1701     for (;;) {
1702         /* Make sure that we always have enough lookahead, except
1703          * at the end of the input file. We need MAX_MATCH bytes
1704          * for the next match, plus MIN_MATCH bytes to insert the
1705          * string following the next match.
1706          */
1707         if (s->lookahead < MIN_LOOKAHEAD) {
1708             fill_window(s);
1709             if (s->lookahead < MIN_LOOKAHEAD && flush == Z_NO_FLUSH) {
1710                 return need_more;
1711             }
1712             if (s->lookahead == 0) break; /* flush the current block */
1713         }
1714
1715         /* Insert the string window[strstart .. strstart+2] in the
1716          * dictionary, and set hash_head to the head of the hash chain:
1717          */
1718         hash_head = NIL;
1719         if (s->lookahead >= MIN_MATCH) {
1720             INSERT_STRING(s, s->strstart, hash_head);
1721         }
1722
1723         /* Find the longest match, discarding those <= prev_length.
1724          */
1725         s->prev_length = s->match_length, s->prev_match = s->match_start;
1726         s->match_length = MIN_MATCH-1;
1727
1728         if (hash_head != NIL && s->prev_length < s->max_lazy_match &&
1729             s->strstart - hash_head <= MAX_DIST(s)) {
1730             /* To simplify the code, we prevent matches with the string
1731              * of window index 0 (in particular we have to avoid a match
1732              * of the string with itself at the start of the input file).
1733              */
1734             s->match_length = longest_match (s, hash_head);
1735             /* longest_match() sets match_start */
1736
1737             if (s->match_length <= 5 && (s->strategy == Z_FILTERED
1738 #if TOO_FAR <= 32767
1739                 || (s->match_length == MIN_MATCH &&
1740                     s->strstart - s->match_start > TOO_FAR)
1741 #endif
1742                 )) {
1743
1744                 /* If prev_match is also MIN_MATCH, match_start is garbage
1745                  * but we will ignore the current match anyway.
1746                  */
1747                 s->match_length = MIN_MATCH-1;
1748             }
1749         }
1750         /* If there was a match at the previous step and the current
1751          * match is not better, output the previous match:
1752          */
1753         if (s->prev_length >= MIN_MATCH && s->match_length <= s->prev_length) {
1754             uInt max_insert = s->strstart + s->lookahead - MIN_MATCH;
1755             /* Do not insert strings in hash table beyond this. */
1756
1757             check_match(s, s->strstart-1, s->prev_match, s->prev_length);
1758
1759             _tr_tally_dist(s, s->strstart -1 - s->prev_match,
1760                            s->prev_length - MIN_MATCH, bflush);
1761
1762             /* Insert in hash table all strings up to the end of the match.
1763              * strstart-1 and strstart are already inserted. If there is not
1764              * enough lookahead, the last two strings are not inserted in
1765              * the hash table.
1766              */
1767             s->lookahead -= s->prev_length-1;
1768             s->prev_length -= 2;
1769             do {
1770                 if (++s->strstart <= max_insert) {
1771                     INSERT_STRING(s, s->strstart, hash_head);
1772                 }
1773             } while (--s->prev_length != 0);
1774             s->match_available = 0;
1775             s->match_length = MIN_MATCH-1;
1776             s->strstart++;
1777
1778             if (bflush) FLUSH_BLOCK(s, 0);
1779
1780         } else if (s->match_available) {
1781             /* If there was no match at the previous position, output a
1782              * single literal. If there was a match but the current match
1783              * is longer, truncate the previous match to a single literal.
1784              */
1785             Tracevv((stderr,"%c", s->window[s->strstart-1]));
1786             _tr_tally_lit(s, s->window[s->strstart-1], bflush);
1787             if (bflush) {
1788                 FLUSH_BLOCK_ONLY(s, 0);
1789             }
1790             s->strstart++;
1791             s->lookahead--;
1792             if (s->strm->avail_out == 0) return need_more;
1793         } else {
1794             /* There is no previous match to compare with, wait for
1795              * the next step to decide.
1796              */
1797             s->match_available = 1;
1798             s->strstart++;
1799             s->lookahead--;
1800         }
1801     }
1802     Assert (flush != Z_NO_FLUSH, "no flush?");
1803     if (s->match_available) {
1804         Tracevv((stderr,"%c", s->window[s->strstart-1]));
1805         _tr_tally_lit(s, s->window[s->strstart-1], bflush);
1806         s->match_available = 0;
1807     }
1808     if (flush == Z_FINISH) {
1809         FLUSH_BLOCK(s, 1);
1810         return finish_done;
1811     }
1812     if (s->last_lit)
1813         FLUSH_BLOCK(s, 0);
1814     return block_done;
1815 }
1816 #endif /* FASTEST */
1817
1818 /* ===========================================================================
1819  * For Z_RLE, simply look for runs of bytes, generate matches only of distance
1820  * one.  Do not maintain a hash table.  (It will be regenerated if this run of
1821  * deflate switches away from Z_RLE.)
1822  */
1823 local block_state deflate_rle(s, flush)
1824     deflate_state *s;
1825     int flush;
1826 {
1827     int bflush;             /* set if current block must be flushed */
1828     uInt prev;              /* byte at distance one to match */
1829     Bytef *scan, *strend;   /* scan goes up to strend for length of run */
1830
1831     for (;;) {
1832         /* Make sure that we always have enough lookahead, except
1833          * at the end of the input file. We need MAX_MATCH bytes
1834          * for the longest run, plus one for the unrolled loop.
1835          */
1836         if (s->lookahead <= MAX_MATCH) {
1837             fill_window(s);
1838             if (s->lookahead <= MAX_MATCH && flush == Z_NO_FLUSH) {
1839                 return need_more;
1840             }
1841             if (s->lookahead == 0) break; /* flush the current block */
1842         }
1843
1844         /* See how many times the previous byte repeats */
1845         s->match_length = 0;
1846         if (s->lookahead >= MIN_MATCH && s->strstart > 0) {
1847             scan = s->window + s->strstart - 1;
1848             prev = *scan;
1849             if (prev == *++scan && prev == *++scan && prev == *++scan) {
1850                 strend = s->window + s->strstart + MAX_MATCH;
1851                 do {
1852                 } while (prev == *++scan && prev == *++scan &&
1853                          prev == *++scan && prev == *++scan &&
1854                          prev == *++scan && prev == *++scan &&
1855                          prev == *++scan && prev == *++scan &&
1856                          scan < strend);
1857                 s->match_length = MAX_MATCH - (int)(strend - scan);
1858                 if (s->match_length > s->lookahead)
1859                     s->match_length = s->lookahead;
1860             }
1861             Assert(scan <= s->window+(uInt)(s->window_size-1), "wild scan");
1862         }
1863
1864         /* Emit match if have run of MIN_MATCH or longer, else emit literal */
1865         if (s->match_length >= MIN_MATCH) {
1866             check_match(s, s->strstart, s->strstart - 1, s->match_length);
1867
1868             _tr_tally_dist(s, 1, s->match_length - MIN_MATCH, bflush);
1869
1870             s->lookahead -= s->match_length;
1871             s->strstart += s->match_length;
1872             s->match_length = 0;
1873         } else {
1874             /* No match, output a literal byte */
1875             Tracevv((stderr,"%c", s->window[s->strstart]));
1876             _tr_tally_lit (s, s->window[s->strstart], bflush);
1877             s->lookahead--;
1878             s->strstart++;
1879         }
1880         if (bflush) FLUSH_BLOCK(s, 0);
1881     }
1882     if (flush == Z_FINISH) {
1883         FLUSH_BLOCK(s, 1);
1884         return finish_done;
1885     }
1886     if (s->last_lit)
1887         FLUSH_BLOCK(s, 0);
1888     return block_done;
1889 }
1890
1891 /* ===========================================================================
1892  * For Z_HUFFMAN_ONLY, do not look for matches.  Do not maintain a hash table.
1893  * (It will be regenerated if this run of deflate switches away from Huffman.)
1894  */
1895 local block_state deflate_huff(s, flush)
1896     deflate_state *s;
1897     int flush;
1898 {
1899     int bflush;             /* set if current block must be flushed */
1900
1901     for (;;) {
1902         /* Make sure that we have a literal to write. */
1903         if (s->lookahead == 0) {
1904             fill_window(s);
1905             if (s->lookahead == 0) {
1906                 if (flush == Z_NO_FLUSH)
1907                     return need_more;
1908                 break;      /* flush the current block */
1909             }
1910         }
1911
1912         /* Output a literal byte */
1913         s->match_length = 0;
1914         Tracevv((stderr,"%c", s->window[s->strstart]));
1915         _tr_tally_lit (s, s->window[s->strstart], bflush);
1916         s->lookahead--;
1917         s->strstart++;
1918         if (bflush) FLUSH_BLOCK(s, 0);
1919     }
1920     if (flush == Z_FINISH) {
1921         FLUSH_BLOCK(s, 1);
1922         return finish_done;
1923     }
1924     if (s->last_lit)
1925         FLUSH_BLOCK(s, 0);
1926     return block_done;
1927 }