| OLD | NEW |
| 1 /* inftrees.c -- generate Huffman trees for efficient decoding | 1 /* inftrees.c -- generate Huffman trees for efficient decoding |
| 2 * Copyright (C) 1995-2005 Mark Adler | 2 * Copyright (C) 1995-2010 Mark Adler |
| 3 * For conditions of distribution and use, see copyright notice in zlib.h | 3 * For conditions of distribution and use, see copyright notice in zlib.h |
| 4 */ | 4 */ |
| 5 | 5 |
| 6 #include "zutil.h" | 6 #include "zutil.h" |
| 7 #include "inftrees.h" | 7 #include "inftrees.h" |
| 8 | 8 |
| 9 #define MAXBITS 15 | 9 #define MAXBITS 15 |
| 10 | 10 |
| 11 const char inflate_copyright[] = | 11 const char inflate_copyright[] = |
| 12 " inflate 1.2.3 Copyright 1995-2005 Mark Adler "; | 12 " inflate 1.2.5 Copyright 1995-2010 Mark Adler "; |
| 13 /* | 13 /* |
| 14 If you use the zlib library in a product, an acknowledgment is welcome | 14 If you use the zlib library in a product, an acknowledgment is welcome |
| 15 in the documentation of your product. If for some reason you cannot | 15 in the documentation of your product. If for some reason you cannot |
| 16 include such an acknowledgment, I would appreciate that you keep this | 16 include such an acknowledgment, I would appreciate that you keep this |
| 17 copyright string in the executable of your product. | 17 copyright string in the executable of your product. |
| 18 */ | 18 */ |
| 19 | 19 |
| 20 /* | 20 /* |
| 21 Build a set of tables to decode the provided canonical Huffman code. | 21 Build a set of tables to decode the provided canonical Huffman code. |
| 22 The code lengths are lens[0..codes-1]. The result starts at *table, | 22 The code lengths are lens[0..codes-1]. The result starts at *table, |
| 23 whose indices are 0..2^bits-1. work is a writable array of at least | 23 whose indices are 0..2^bits-1. work is a writable array of at least |
| 24 lens shorts, which is used as a work area. type is the type of code | 24 lens shorts, which is used as a work area. type is the type of code |
| 25 to be generated, CODES, LENS, or DISTS. On return, zero is success, | 25 to be generated, CODES, LENS, or DISTS. On return, zero is success, |
| 26 -1 is an invalid code, and +1 means that ENOUGH isn't enough. table | 26 -1 is an invalid code, and +1 means that ENOUGH isn't enough. table |
| 27 on return points to the next available entry's address. bits is the | 27 on return points to the next available entry's address. bits is the |
| 28 requested root table index bits, and on return it is the actual root | 28 requested root table index bits, and on return it is the actual root |
| 29 table index bits. It will differ if the request is greater than the | 29 table index bits. It will differ if the request is greater than the |
| 30 longest code or if it is less than the shortest code. | 30 longest code or if it is less than the shortest code. |
| 31 */ | 31 */ |
| 32 int inflate_table(type, lens, codes, table, bits, work) | 32 int ZLIB_INTERNAL inflate_table(type, lens, codes, table, bits, work) |
| 33 codetype type; | 33 codetype type; |
| 34 unsigned short FAR *lens; | 34 unsigned short FAR *lens; |
| 35 unsigned codes; | 35 unsigned codes; |
| 36 code FAR * FAR *table; | 36 code FAR * FAR *table; |
| 37 unsigned FAR *bits; | 37 unsigned FAR *bits; |
| 38 unsigned short FAR *work; | 38 unsigned short FAR *work; |
| 39 { | 39 { |
| 40 unsigned len; /* a code's length in bits */ | 40 unsigned len; /* a code's length in bits */ |
| 41 unsigned sym; /* index of code symbols */ | 41 unsigned sym; /* index of code symbols */ |
| 42 unsigned min, max; /* minimum and maximum code lengths */ | 42 unsigned min, max; /* minimum and maximum code lengths */ |
| 43 unsigned root; /* number of index bits for root table */ | 43 unsigned root; /* number of index bits for root table */ |
| 44 unsigned curr; /* number of index bits for current table */ | 44 unsigned curr; /* number of index bits for current table */ |
| 45 unsigned drop; /* code bits to drop for sub-table */ | 45 unsigned drop; /* code bits to drop for sub-table */ |
| 46 int left; /* number of prefix codes available */ | 46 int left; /* number of prefix codes available */ |
| 47 unsigned used; /* code entries in table used */ | 47 unsigned used; /* code entries in table used */ |
| 48 unsigned huff; /* Huffman code */ | 48 unsigned huff; /* Huffman code */ |
| 49 unsigned incr; /* for incrementing code, index */ | 49 unsigned incr; /* for incrementing code, index */ |
| 50 unsigned fill; /* index for replicating entries */ | 50 unsigned fill; /* index for replicating entries */ |
| 51 unsigned low; /* low bits for current root entry */ | 51 unsigned low; /* low bits for current root entry */ |
| 52 unsigned mask; /* mask for low root bits */ | 52 unsigned mask; /* mask for low root bits */ |
| 53 code this; /* table entry for duplication */ | 53 code here; /* table entry for duplication */ |
| 54 code FAR *next; /* next available space in table */ | 54 code FAR *next; /* next available space in table */ |
| 55 const unsigned short FAR *base; /* base value table to use */ | 55 const unsigned short FAR *base; /* base value table to use */ |
| 56 const unsigned short FAR *extra; /* extra bits table to use */ | 56 const unsigned short FAR *extra; /* extra bits table to use */ |
| 57 int end; /* use base and extra for symbol > end */ | 57 int end; /* use base and extra for symbol > end */ |
| 58 unsigned short count[MAXBITS+1]; /* number of codes of each length */ | 58 unsigned short count[MAXBITS+1]; /* number of codes of each length */ |
| 59 unsigned short offs[MAXBITS+1]; /* offsets in table for each length */ | 59 unsigned short offs[MAXBITS+1]; /* offsets in table for each length */ |
| 60 static const unsigned short lbase[31] = { /* Length codes 257..285 base */ | 60 static const unsigned short lbase[31] = { /* Length codes 257..285 base */ |
| 61 3, 4, 5, 6, 7, 8, 9, 10, 11, 13, 15, 17, 19, 23, 27, 31, | 61 3, 4, 5, 6, 7, 8, 9, 10, 11, 13, 15, 17, 19, 23, 27, 31, |
| 62 35, 43, 51, 59, 67, 83, 99, 115, 131, 163, 195, 227, 258, 0, 0}; | 62 35, 43, 51, 59, 67, 83, 99, 115, 131, 163, 195, 227, 258, 0, 0}; |
| 63 static const unsigned short lext[31] = { /* Length codes 257..285 extra */ | 63 static const unsigned short lext[31] = { /* Length codes 257..285 extra */ |
| 64 16, 16, 16, 16, 16, 16, 16, 16, 17, 17, 17, 17, 18, 18, 18, 18, | 64 16, 16, 16, 16, 16, 16, 16, 16, 17, 17, 17, 17, 18, 18, 18, 18, |
| 65 19, 19, 19, 19, 20, 20, 20, 20, 21, 21, 21, 21, 16, 201, 196}; | 65 19, 19, 19, 19, 20, 20, 20, 20, 21, 21, 21, 21, 16, 73, 195}; |
| 66 static const unsigned short dbase[32] = { /* Distance codes 0..29 base */ | 66 static const unsigned short dbase[32] = { /* Distance codes 0..29 base */ |
| 67 1, 2, 3, 4, 5, 7, 9, 13, 17, 25, 33, 49, 65, 97, 129, 193, | 67 1, 2, 3, 4, 5, 7, 9, 13, 17, 25, 33, 49, 65, 97, 129, 193, |
| 68 257, 385, 513, 769, 1025, 1537, 2049, 3073, 4097, 6145, | 68 257, 385, 513, 769, 1025, 1537, 2049, 3073, 4097, 6145, |
| 69 8193, 12289, 16385, 24577, 0, 0}; | 69 8193, 12289, 16385, 24577, 0, 0}; |
| 70 static const unsigned short dext[32] = { /* Distance codes 0..29 extra */ | 70 static const unsigned short dext[32] = { /* Distance codes 0..29 extra */ |
| 71 16, 16, 16, 16, 17, 17, 18, 18, 19, 19, 20, 20, 21, 21, 22, 22, | 71 16, 16, 16, 16, 17, 17, 18, 18, 19, 19, 20, 20, 21, 21, 22, 22, |
| 72 23, 23, 24, 24, 25, 25, 26, 26, 27, 27, | 72 23, 23, 24, 24, 25, 25, 26, 26, 27, 27, |
| 73 28, 28, 29, 29, 64, 64}; | 73 28, 28, 29, 29, 64, 64}; |
| 74 | 74 |
| 75 /* | 75 /* |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 108 count[len] = 0; | 108 count[len] = 0; |
| 109 for (sym = 0; sym < codes; sym++) | 109 for (sym = 0; sym < codes; sym++) |
| 110 count[lens[sym]]++; | 110 count[lens[sym]]++; |
| 111 | 111 |
| 112 /* bound code lengths, force root to be within code lengths */ | 112 /* bound code lengths, force root to be within code lengths */ |
| 113 root = *bits; | 113 root = *bits; |
| 114 for (max = MAXBITS; max >= 1; max--) | 114 for (max = MAXBITS; max >= 1; max--) |
| 115 if (count[max] != 0) break; | 115 if (count[max] != 0) break; |
| 116 if (root > max) root = max; | 116 if (root > max) root = max; |
| 117 if (max == 0) { /* no symbols to code at all */ | 117 if (max == 0) { /* no symbols to code at all */ |
| 118 this.op = (unsigned char)64; /* invalid code marker */ | 118 here.op = (unsigned char)64; /* invalid code marker */ |
| 119 this.bits = (unsigned char)1; | 119 here.bits = (unsigned char)1; |
| 120 this.val = (unsigned short)0; | 120 here.val = (unsigned short)0; |
| 121 *(*table)++ = this; /* make a table to force an error */ | 121 *(*table)++ = here; /* make a table to force an error */ |
| 122 *(*table)++ = this; | 122 *(*table)++ = here; |
| 123 *bits = 1; | 123 *bits = 1; |
| 124 return 0; /* no symbols, but wait for decoding to report error */ | 124 return 0; /* no symbols, but wait for decoding to report error */ |
| 125 } | 125 } |
| 126 for (min = 1; min <= MAXBITS; min++) | 126 for (min = 1; min < max; min++) |
| 127 if (count[min] != 0) break; | 127 if (count[min] != 0) break; |
| 128 if (root < min) root = min; | 128 if (root < min) root = min; |
| 129 | 129 |
| 130 /* check for an over-subscribed or incomplete set of lengths */ | 130 /* check for an over-subscribed or incomplete set of lengths */ |
| 131 left = 1; | 131 left = 1; |
| 132 for (len = 1; len <= MAXBITS; len++) { | 132 for (len = 1; len <= MAXBITS; len++) { |
| 133 left <<= 1; | 133 left <<= 1; |
| 134 left -= count[len]; | 134 left -= count[len]; |
| 135 if (left < 0) return -1; /* over-subscribed */ | 135 if (left < 0) return -1; /* over-subscribed */ |
| 136 } | 136 } |
| (...skipping 22 matching lines...) Expand all Loading... |
| 159 of the low root bits of huff. This is saved in low to check for when a | 159 of the low root bits of huff. This is saved in low to check for when a |
| 160 new sub-table should be started. drop is zero when the root table is | 160 new sub-table should be started. drop is zero when the root table is |
| 161 being filled, and drop is root when sub-tables are being filled. | 161 being filled, and drop is root when sub-tables are being filled. |
| 162 | 162 |
| 163 When a new sub-table is needed, it is necessary to look ahead in the | 163 When a new sub-table is needed, it is necessary to look ahead in the |
| 164 code lengths to determine what size sub-table is needed. The length | 164 code lengths to determine what size sub-table is needed. The length |
| 165 counts are used for this, and so count[] is decremented as codes are | 165 counts are used for this, and so count[] is decremented as codes are |
| 166 entered in the tables. | 166 entered in the tables. |
| 167 | 167 |
| 168 used keeps track of how many table entries have been allocated from the | 168 used keeps track of how many table entries have been allocated from the |
| 169 provided *table space. It is checked when a LENS table is being made | 169 provided *table space. It is checked for LENS and DIST tables against |
| 170 against the space in *table, ENOUGH, minus the maximum space needed by | 170 the constants ENOUGH_LENS and ENOUGH_DISTS to guard against changes in |
| 171 the worst case distance code, MAXD. This should never happen, but the | 171 the initial root table size constants. See the comments in inftrees.h |
| 172 sufficiency of ENOUGH has not been proven exhaustively, hence the check. | 172 for more information. |
| 173 This assumes that when type == LENS, bits == 9. | |
| 174 | 173 |
| 175 sym increments through all symbols, and the loop terminates when | 174 sym increments through all symbols, and the loop terminates when |
| 176 all codes of length max, i.e. all codes, have been processed. This | 175 all codes of length max, i.e. all codes, have been processed. This |
| 177 routine permits incomplete codes, so another loop after this one fills | 176 routine permits incomplete codes, so another loop after this one fills |
| 178 in the rest of the decoding tables with invalid code markers. | 177 in the rest of the decoding tables with invalid code markers. |
| 179 */ | 178 */ |
| 180 | 179 |
| 181 /* set up for code type */ | 180 /* set up for code type */ |
| 182 switch (type) { | 181 switch (type) { |
| 183 case CODES: | 182 case CODES: |
| (...skipping 18 matching lines...) Expand all Loading... |
| 202 sym = 0; /* starting code symbol */ | 201 sym = 0; /* starting code symbol */ |
| 203 len = min; /* starting code length */ | 202 len = min; /* starting code length */ |
| 204 next = *table; /* current table to fill in */ | 203 next = *table; /* current table to fill in */ |
| 205 curr = root; /* current table index bits */ | 204 curr = root; /* current table index bits */ |
| 206 drop = 0; /* current bits to drop from code for index */ | 205 drop = 0; /* current bits to drop from code for index */ |
| 207 low = (unsigned)(-1); /* trigger new sub-table when len > root */ | 206 low = (unsigned)(-1); /* trigger new sub-table when len > root */ |
| 208 used = 1U << root; /* use root table entries */ | 207 used = 1U << root; /* use root table entries */ |
| 209 mask = used - 1; /* mask for comparing low */ | 208 mask = used - 1; /* mask for comparing low */ |
| 210 | 209 |
| 211 /* check available table space */ | 210 /* check available table space */ |
| 212 if (type == LENS && used >= ENOUGH - MAXD) | 211 if ((type == LENS && used >= ENOUGH_LENS) || |
| 212 (type == DISTS && used >= ENOUGH_DISTS)) |
| 213 return 1; | 213 return 1; |
| 214 | 214 |
| 215 /* process all codes and make table entries */ | 215 /* process all codes and make table entries */ |
| 216 for (;;) { | 216 for (;;) { |
| 217 /* create table entry */ | 217 /* create table entry */ |
| 218 this.bits = (unsigned char)(len - drop); | 218 here.bits = (unsigned char)(len - drop); |
| 219 if ((int)(work[sym]) < end) { | 219 if ((int)(work[sym]) < end) { |
| 220 this.op = (unsigned char)0; | 220 here.op = (unsigned char)0; |
| 221 this.val = work[sym]; | 221 here.val = work[sym]; |
| 222 } | 222 } |
| 223 else if ((int)(work[sym]) > end) { | 223 else if ((int)(work[sym]) > end) { |
| 224 this.op = (unsigned char)(extra[work[sym]]); | 224 here.op = (unsigned char)(extra[work[sym]]); |
| 225 this.val = base[work[sym]]; | 225 here.val = base[work[sym]]; |
| 226 } | 226 } |
| 227 else { | 227 else { |
| 228 this.op = (unsigned char)(32 + 64); /* end of block */ | 228 here.op = (unsigned char)(32 + 64); /* end of block */ |
| 229 this.val = 0; | 229 here.val = 0; |
| 230 } | 230 } |
| 231 | 231 |
| 232 /* replicate for those indices with low len bits equal to huff */ | 232 /* replicate for those indices with low len bits equal to huff */ |
| 233 incr = 1U << (len - drop); | 233 incr = 1U << (len - drop); |
| 234 fill = 1U << curr; | 234 fill = 1U << curr; |
| 235 min = fill; /* save offset to next table */ | 235 min = fill; /* save offset to next table */ |
| 236 do { | 236 do { |
| 237 fill -= incr; | 237 fill -= incr; |
| 238 next[(huff >> drop) + fill] = this; | 238 next[(huff >> drop) + fill] = here; |
| 239 } while (fill != 0); | 239 } while (fill != 0); |
| 240 | 240 |
| 241 /* backwards increment the len-bit code huff */ | 241 /* backwards increment the len-bit code huff */ |
| 242 incr = 1U << (len - 1); | 242 incr = 1U << (len - 1); |
| 243 while (huff & incr) | 243 while (huff & incr) |
| 244 incr >>= 1; | 244 incr >>= 1; |
| 245 if (incr != 0) { | 245 if (incr != 0) { |
| 246 huff &= incr - 1; | 246 huff &= incr - 1; |
| 247 huff += incr; | 247 huff += incr; |
| 248 } | 248 } |
| (...skipping 21 matching lines...) Expand all Loading... |
| 270 left = (int)(1 << curr); | 270 left = (int)(1 << curr); |
| 271 while (curr + drop < max) { | 271 while (curr + drop < max) { |
| 272 left -= count[curr + drop]; | 272 left -= count[curr + drop]; |
| 273 if (left <= 0) break; | 273 if (left <= 0) break; |
| 274 curr++; | 274 curr++; |
| 275 left <<= 1; | 275 left <<= 1; |
| 276 } | 276 } |
| 277 | 277 |
| 278 /* check for enough space */ | 278 /* check for enough space */ |
| 279 used += 1U << curr; | 279 used += 1U << curr; |
| 280 if (type == LENS && used >= ENOUGH - MAXD) | 280 if ((type == LENS && used >= ENOUGH_LENS) || |
| 281 (type == DISTS && used >= ENOUGH_DISTS)) |
| 281 return 1; | 282 return 1; |
| 282 | 283 |
| 283 /* point entry in root table to sub-table */ | 284 /* point entry in root table to sub-table */ |
| 284 low = huff & mask; | 285 low = huff & mask; |
| 285 (*table)[low].op = (unsigned char)curr; | 286 (*table)[low].op = (unsigned char)curr; |
| 286 (*table)[low].bits = (unsigned char)root; | 287 (*table)[low].bits = (unsigned char)root; |
| 287 (*table)[low].val = (unsigned short)(next - *table); | 288 (*table)[low].val = (unsigned short)(next - *table); |
| 288 } | 289 } |
| 289 } | 290 } |
| 290 | 291 |
| 291 /* | 292 /* |
| 292 Fill in rest of table for incomplete codes. This loop is similar to the | 293 Fill in rest of table for incomplete codes. This loop is similar to the |
| 293 loop above in incrementing huff for table indices. It is assumed that | 294 loop above in incrementing huff for table indices. It is assumed that |
| 294 len is equal to curr + drop, so there is no loop needed to increment | 295 len is equal to curr + drop, so there is no loop needed to increment |
| 295 through high index bits. When the current sub-table is filled, the loop | 296 through high index bits. When the current sub-table is filled, the loop |
| 296 drops back to the root table to fill in any remaining entries there. | 297 drops back to the root table to fill in any remaining entries there. |
| 297 */ | 298 */ |
| 298 this.op = (unsigned char)64; /* invalid code marker */ | 299 here.op = (unsigned char)64; /* invalid code marker */ |
| 299 this.bits = (unsigned char)(len - drop); | 300 here.bits = (unsigned char)(len - drop); |
| 300 this.val = (unsigned short)0; | 301 here.val = (unsigned short)0; |
| 301 while (huff != 0) { | 302 while (huff != 0) { |
| 302 /* when done with sub-table, drop back to root table */ | 303 /* when done with sub-table, drop back to root table */ |
| 303 if (drop != 0 && (huff & mask) != low) { | 304 if (drop != 0 && (huff & mask) != low) { |
| 304 drop = 0; | 305 drop = 0; |
| 305 len = root; | 306 len = root; |
| 306 next = *table; | 307 next = *table; |
| 307 this.bits = (unsigned char)len; | 308 here.bits = (unsigned char)len; |
| 308 } | 309 } |
| 309 | 310 |
| 310 /* put invalid code marker in table */ | 311 /* put invalid code marker in table */ |
| 311 next[huff >> drop] = this; | 312 next[huff >> drop] = here; |
| 312 | 313 |
| 313 /* backwards increment the len-bit code huff */ | 314 /* backwards increment the len-bit code huff */ |
| 314 incr = 1U << (len - 1); | 315 incr = 1U << (len - 1); |
| 315 while (huff & incr) | 316 while (huff & incr) |
| 316 incr >>= 1; | 317 incr >>= 1; |
| 317 if (incr != 0) { | 318 if (incr != 0) { |
| 318 huff &= incr - 1; | 319 huff &= incr - 1; |
| 319 huff += incr; | 320 huff += incr; |
| 320 } | 321 } |
| 321 else | 322 else |
| 322 huff = 0; | 323 huff = 0; |
| 323 } | 324 } |
| 324 | 325 |
| 325 /* set return parameters */ | 326 /* set return parameters */ |
| 326 *table += used; | 327 *table += used; |
| 327 *bits = root; | 328 *bits = root; |
| 328 return 0; | 329 return 0; |
| 329 } | 330 } |
| OLD | NEW |