puff.c 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838
  1. /*
  2. * puff.c
  3. * Copyright (C) 2002-2010 Mark Adler
  4. * For conditions of distribution and use, see copyright notice in puff.h
  5. * version 2.2, 25 Apr 2010
  6. *
  7. * puff.c is a simple inflate written to be an unambiguous way to specify the
  8. * deflate format. It is not written for speed but rather simplicity. As a
  9. * side benefit, this code might actually be useful when small code is more
  10. * important than speed, such as bootstrap applications. For typical deflate
  11. * data, zlib's inflate() is about four times as fast as puff(). zlib's
  12. * inflate compiles to around 20K on my machine, whereas puff.c compiles to
  13. * around 4K on my machine (a PowerPC using GNU cc). If the faster decode()
  14. * function here is used, then puff() is only twice as slow as zlib's
  15. * inflate().
  16. *
  17. * All dynamically allocated memory comes from the stack. The stack required
  18. * is less than 2K bytes. This code is compatible with 16-bit int's and
  19. * assumes that long's are at least 32 bits. puff.c uses the short data type,
  20. * assumed to be 16 bits, for arrays in order to to conserve memory. The code
  21. * works whether integers are stored big endian or little endian.
  22. *
  23. * In the comments below are "Format notes" that describe the inflate process
  24. * and document some of the less obvious aspects of the format. This source
  25. * code is meant to supplement RFC 1951, which formally describes the deflate
  26. * format:
  27. *
  28. * http://www.zlib.org/rfc-deflate.html
  29. */
  30. /*
  31. * Change history:
  32. *
  33. * 1.0 10 Feb 2002 - First version
  34. * 1.1 17 Feb 2002 - Clarifications of some comments and notes
  35. * - Update puff() dest and source pointers on negative
  36. * errors to facilitate debugging deflators
  37. * - Remove longest from struct huffman -- not needed
  38. * - Simplify offs[] index in construct()
  39. * - Add input size and checking, using longjmp() to
  40. * maintain easy readability
  41. * - Use short data type for large arrays
  42. * - Use pointers instead of long to specify source and
  43. * destination sizes to avoid arbitrary 4 GB limits
  44. * 1.2 17 Mar 2002 - Add faster version of decode(), doubles speed (!),
  45. * but leave simple version for readabilty
  46. * - Make sure invalid distances detected if pointers
  47. * are 16 bits
  48. * - Fix fixed codes table error
  49. * - Provide a scanning mode for determining size of
  50. * uncompressed data
  51. * 1.3 20 Mar 2002 - Go back to lengths for puff() parameters [Gailly]
  52. * - Add a puff.h file for the interface
  53. * - Add braces in puff() for else do [Gailly]
  54. * - Use indexes instead of pointers for readability
  55. * 1.4 31 Mar 2002 - Simplify construct() code set check
  56. * - Fix some comments
  57. * - Add FIXLCODES #define
  58. * 1.5 6 Apr 2002 - Minor comment fixes
  59. * 1.6 7 Aug 2002 - Minor format changes
  60. * 1.7 3 Mar 2003 - Added test code for distribution
  61. * - Added zlib-like license
  62. * 1.8 9 Jan 2004 - Added some comments on no distance codes case
  63. * 1.9 21 Feb 2008 - Fix bug on 16-bit integer architectures [Pohland]
  64. * - Catch missing end-of-block symbol error
  65. * 2.0 25 Jul 2008 - Add #define to permit distance too far back
  66. * - Add option in TEST code for puff to write the data
  67. * - Add option in TEST code to skip input bytes
  68. * - Allow TEST code to read from piped stdin
  69. * 2.1 4 Apr 2010 - Avoid variable initialization for happier compilers
  70. * - Avoid unsigned comparisons for even happier compilers
  71. * 2.2 25 Apr 2010 - Fix bug in variable initializations [Oberhumer]
  72. * - Add const where appropriate [Oberhumer]
  73. * - Split if's and ?'s for coverage testing
  74. * - Break out test code to separate file
  75. * - Move NIL to puff.h
  76. * - Allow incomplete code only if single code length is 1
  77. * - Add full code coverage test to Makefile
  78. */
  79. #include <setjmp.h> /* for setjmp(), longjmp(), and jmp_buf */
  80. #include "puff.h" /* prototype for puff() */
  81. #define local static /* for local function definitions */
  82. /*
  83. * Maximums for allocations and loops. It is not useful to change these --
  84. * they are fixed by the deflate format.
  85. */
  86. #define MAXBITS 15 /* maximum bits in a code */
  87. #define MAXLCODES 286 /* maximum number of literal/length codes */
  88. #define MAXDCODES 30 /* maximum number of distance codes */
  89. #define MAXCODES (MAXLCODES+MAXDCODES) /* maximum codes lengths to read */
  90. #define FIXLCODES 288 /* number of fixed literal/length codes */
  91. /* input and output state */
  92. struct state {
  93. /* output state */
  94. unsigned char *out; /* output buffer */
  95. unsigned long outlen; /* available space at out */
  96. unsigned long outcnt; /* bytes written to out so far */
  97. /* input state */
  98. const unsigned char *in; /* input buffer */
  99. unsigned long inlen; /* available input at in */
  100. unsigned long incnt; /* bytes read so far */
  101. int bitbuf; /* bit buffer */
  102. int bitcnt; /* number of bits in bit buffer */
  103. /* input limit error return state for bits() and decode() */
  104. jmp_buf env;
  105. };
  106. /*
  107. * Return need bits from the input stream. This always leaves less than
  108. * eight bits in the buffer. bits() works properly for need == 0.
  109. *
  110. * Format notes:
  111. *
  112. * - Bits are stored in bytes from the least significant bit to the most
  113. * significant bit. Therefore bits are dropped from the bottom of the bit
  114. * buffer, using shift right, and new bytes are appended to the top of the
  115. * bit buffer, using shift left.
  116. */
  117. local int bits(struct state *s, int need)
  118. {
  119. long val; /* bit accumulator (can use up to 20 bits) */
  120. /* load at least need bits into val */
  121. val = s->bitbuf;
  122. while (s->bitcnt < need) {
  123. if (s->incnt == s->inlen)
  124. longjmp(s->env, 1); /* out of input */
  125. val |= (long)(s->in[s->incnt++]) << s->bitcnt; /* load eight bits */
  126. s->bitcnt += 8;
  127. }
  128. /* drop need bits and update buffer, always zero to seven bits left */
  129. s->bitbuf = (int)(val >> need);
  130. s->bitcnt -= need;
  131. /* return need bits, zeroing the bits above that */
  132. return (int)(val & ((1L << need) - 1));
  133. }
  134. /*
  135. * Process a stored block.
  136. *
  137. * Format notes:
  138. *
  139. * - After the two-bit stored block type (00), the stored block length and
  140. * stored bytes are byte-aligned for fast copying. Therefore any leftover
  141. * bits in the byte that has the last bit of the type, as many as seven, are
  142. * discarded. The value of the discarded bits are not defined and should not
  143. * be checked against any expectation.
  144. *
  145. * - The second inverted copy of the stored block length does not have to be
  146. * checked, but it's probably a good idea to do so anyway.
  147. *
  148. * - A stored block can have zero length. This is sometimes used to byte-align
  149. * subsets of the compressed data for random access or partial recovery.
  150. */
  151. local int stored(struct state *s)
  152. {
  153. unsigned len; /* length of stored block */
  154. /* discard leftover bits from current byte (assumes s->bitcnt < 8) */
  155. s->bitbuf = 0;
  156. s->bitcnt = 0;
  157. /* get length and check against its one's complement */
  158. if (s->incnt + 4 > s->inlen)
  159. return 2; /* not enough input */
  160. len = s->in[s->incnt++];
  161. len |= s->in[s->incnt++] << 8;
  162. if (s->in[s->incnt++] != (~len & 0xff) ||
  163. s->in[s->incnt++] != ((~len >> 8) & 0xff))
  164. return -2; /* didn't match complement! */
  165. /* copy len bytes from in to out */
  166. if (s->incnt + len > s->inlen)
  167. return 2; /* not enough input */
  168. if (s->out != NIL) {
  169. if (s->outcnt + len > s->outlen)
  170. return 1; /* not enough output space */
  171. while (len--)
  172. s->out[s->outcnt++] = s->in[s->incnt++];
  173. }
  174. else { /* just scanning */
  175. s->outcnt += len;
  176. s->incnt += len;
  177. }
  178. /* done with a valid stored block */
  179. return 0;
  180. }
  181. /*
  182. * Huffman code decoding tables. count[1..MAXBITS] is the number of symbols of
  183. * each length, which for a canonical code are stepped through in order.
  184. * symbol[] are the symbol values in canonical order, where the number of
  185. * entries is the sum of the counts in count[]. The decoding process can be
  186. * seen in the function decode() below.
  187. */
  188. struct huffman {
  189. short *count; /* number of symbols of each length */
  190. short *symbol; /* canonically ordered symbols */
  191. };
  192. /*
  193. * Decode a code from the stream s using huffman table h. Return the symbol or
  194. * a negative value if there is an error. If all of the lengths are zero, i.e.
  195. * an empty code, or if the code is incomplete and an invalid code is received,
  196. * then -10 is returned after reading MAXBITS bits.
  197. *
  198. * Format notes:
  199. *
  200. * - The codes as stored in the compressed data are bit-reversed relative to
  201. * a simple integer ordering of codes of the same lengths. Hence below the
  202. * bits are pulled from the compressed data one at a time and used to
  203. * build the code value reversed from what is in the stream in order to
  204. * permit simple integer comparisons for decoding. A table-based decoding
  205. * scheme (as used in zlib) does not need to do this reversal.
  206. *
  207. * - The first code for the shortest length is all zeros. Subsequent codes of
  208. * the same length are simply integer increments of the previous code. When
  209. * moving up a length, a zero bit is appended to the code. For a complete
  210. * code, the last code of the longest length will be all ones.
  211. *
  212. * - Incomplete codes are handled by this decoder, since they are permitted
  213. * in the deflate format. See the format notes for fixed() and dynamic().
  214. */
  215. #ifdef SLOW
  216. local int decode(struct state *s, const struct huffman *h)
  217. {
  218. int len; /* current number of bits in code */
  219. int code; /* len bits being decoded */
  220. int first; /* first code of length len */
  221. int count; /* number of codes of length len */
  222. int index; /* index of first code of length len in symbol table */
  223. code = first = index = 0;
  224. for (len = 1; len <= MAXBITS; len++) {
  225. code |= bits(s, 1); /* get next bit */
  226. count = h->count[len];
  227. if (code - count < first) /* if length len, return symbol */
  228. return h->symbol[index + (code - first)];
  229. index += count; /* else update for next length */
  230. first += count;
  231. first <<= 1;
  232. code <<= 1;
  233. }
  234. return -10; /* ran out of codes */
  235. }
  236. /*
  237. * A faster version of decode() for real applications of this code. It's not
  238. * as readable, but it makes puff() twice as fast. And it only makes the code
  239. * a few percent larger.
  240. */
  241. #else /* !SLOW */
  242. local int decode(struct state *s, const struct huffman *h)
  243. {
  244. int len; /* current number of bits in code */
  245. int code; /* len bits being decoded */
  246. int first; /* first code of length len */
  247. int count; /* number of codes of length len */
  248. int index; /* index of first code of length len in symbol table */
  249. int bitbuf; /* bits from stream */
  250. int left; /* bits left in next or left to process */
  251. short *next; /* next number of codes */
  252. bitbuf = s->bitbuf;
  253. left = s->bitcnt;
  254. code = first = index = 0;
  255. len = 1;
  256. next = h->count + 1;
  257. while (1) {
  258. while (left--) {
  259. code |= bitbuf & 1;
  260. bitbuf >>= 1;
  261. count = *next++;
  262. if (code - count < first) { /* if length len, return symbol */
  263. s->bitbuf = bitbuf;
  264. s->bitcnt = (s->bitcnt - len) & 7;
  265. return h->symbol[index + (code - first)];
  266. }
  267. index += count; /* else update for next length */
  268. first += count;
  269. first <<= 1;
  270. code <<= 1;
  271. len++;
  272. }
  273. left = (MAXBITS+1) - len;
  274. if (left == 0)
  275. break;
  276. if (s->incnt == s->inlen)
  277. longjmp(s->env, 1); /* out of input */
  278. bitbuf = s->in[s->incnt++];
  279. if (left > 8)
  280. left = 8;
  281. }
  282. return -10; /* ran out of codes */
  283. }
  284. #endif /* SLOW */
  285. /*
  286. * Given the list of code lengths length[0..n-1] representing a canonical
  287. * Huffman code for n symbols, construct the tables required to decode those
  288. * codes. Those tables are the number of codes of each length, and the symbols
  289. * sorted by length, retaining their original order within each length. The
  290. * return value is zero for a complete code set, negative for an over-
  291. * subscribed code set, and positive for an incomplete code set. The tables
  292. * can be used if the return value is zero or positive, but they cannot be used
  293. * if the return value is negative. If the return value is zero, it is not
  294. * possible for decode() using that table to return an error--any stream of
  295. * enough bits will resolve to a symbol. If the return value is positive, then
  296. * it is possible for decode() using that table to return an error for received
  297. * codes past the end of the incomplete lengths.
  298. *
  299. * Not used by decode(), but used for error checking, h->count[0] is the number
  300. * of the n symbols not in the code. So n - h->count[0] is the number of
  301. * codes. This is useful for checking for incomplete codes that have more than
  302. * one symbol, which is an error in a dynamic block.
  303. *
  304. * Assumption: for all i in 0..n-1, 0 <= length[i] <= MAXBITS
  305. * This is assured by the construction of the length arrays in dynamic() and
  306. * fixed() and is not verified by construct().
  307. *
  308. * Format notes:
  309. *
  310. * - Permitted and expected examples of incomplete codes are one of the fixed
  311. * codes and any code with a single symbol which in deflate is coded as one
  312. * bit instead of zero bits. See the format notes for fixed() and dynamic().
  313. *
  314. * - Within a given code length, the symbols are kept in ascending order for
  315. * the code bits definition.
  316. */
  317. local int construct(struct huffman *h, const short *length, int n)
  318. {
  319. int symbol; /* current symbol when stepping through length[] */
  320. int len; /* current length when stepping through h->count[] */
  321. int left; /* number of possible codes left of current length */
  322. short offs[MAXBITS+1]; /* offsets in symbol table for each length */
  323. /* count number of codes of each length */
  324. for (len = 0; len <= MAXBITS; len++)
  325. h->count[len] = 0;
  326. for (symbol = 0; symbol < n; symbol++)
  327. (h->count[length[symbol]])++; /* assumes lengths are within bounds */
  328. if (h->count[0] == n) /* no codes! */
  329. return 0; /* complete, but decode() will fail */
  330. /* check for an over-subscribed or incomplete set of lengths */
  331. left = 1; /* one possible code of zero length */
  332. for (len = 1; len <= MAXBITS; len++) {
  333. left <<= 1; /* one more bit, double codes left */
  334. left -= h->count[len]; /* deduct count from possible codes */
  335. if (left < 0)
  336. return left; /* over-subscribed--return negative */
  337. } /* left > 0 means incomplete */
  338. /* generate offsets into symbol table for each length for sorting */
  339. offs[1] = 0;
  340. for (len = 1; len < MAXBITS; len++)
  341. offs[len + 1] = offs[len] + h->count[len];
  342. /*
  343. * put symbols in table sorted by length, by symbol order within each
  344. * length
  345. */
  346. for (symbol = 0; symbol < n; symbol++)
  347. if (length[symbol] != 0)
  348. h->symbol[offs[length[symbol]]++] = symbol;
  349. /* return zero for complete set, positive for incomplete set */
  350. return left;
  351. }
  352. /*
  353. * Decode literal/length and distance codes until an end-of-block code.
  354. *
  355. * Format notes:
  356. *
  357. * - Compressed data that is after the block type if fixed or after the code
  358. * description if dynamic is a combination of literals and length/distance
  359. * pairs terminated by and end-of-block code. Literals are simply Huffman
  360. * coded bytes. A length/distance pair is a coded length followed by a
  361. * coded distance to represent a string that occurs earlier in the
  362. * uncompressed data that occurs again at the current location.
  363. *
  364. * - Literals, lengths, and the end-of-block code are combined into a single
  365. * code of up to 286 symbols. They are 256 literals (0..255), 29 length
  366. * symbols (257..285), and the end-of-block symbol (256).
  367. *
  368. * - There are 256 possible lengths (3..258), and so 29 symbols are not enough
  369. * to represent all of those. Lengths 3..10 and 258 are in fact represented
  370. * by just a length symbol. Lengths 11..257 are represented as a symbol and
  371. * some number of extra bits that are added as an integer to the base length
  372. * of the length symbol. The number of extra bits is determined by the base
  373. * length symbol. These are in the static arrays below, lens[] for the base
  374. * lengths and lext[] for the corresponding number of extra bits.
  375. *
  376. * - The reason that 258 gets its own symbol is that the longest length is used
  377. * often in highly redundant files. Note that 258 can also be coded as the
  378. * base value 227 plus the maximum extra value of 31. While a good deflate
  379. * should never do this, it is not an error, and should be decoded properly.
  380. *
  381. * - If a length is decoded, including its extra bits if any, then it is
  382. * followed a distance code. There are up to 30 distance symbols. Again
  383. * there are many more possible distances (1..32768), so extra bits are added
  384. * to a base value represented by the symbol. The distances 1..4 get their
  385. * own symbol, but the rest require extra bits. The base distances and
  386. * corresponding number of extra bits are below in the static arrays dist[]
  387. * and dext[].
  388. *
  389. * - Literal bytes are simply written to the output. A length/distance pair is
  390. * an instruction to copy previously uncompressed bytes to the output. The
  391. * copy is from distance bytes back in the output stream, copying for length
  392. * bytes.
  393. *
  394. * - Distances pointing before the beginning of the output data are not
  395. * permitted.
  396. *
  397. * - Overlapped copies, where the length is greater than the distance, are
  398. * allowed and common. For example, a distance of one and a length of 258
  399. * simply copies the last byte 258 times. A distance of four and a length of
  400. * twelve copies the last four bytes three times. A simple forward copy
  401. * ignoring whether the length is greater than the distance or not implements
  402. * this correctly. You should not use memcpy() since its behavior is not
  403. * defined for overlapped arrays. You should not use memmove() or bcopy()
  404. * since though their behavior -is- defined for overlapping arrays, it is
  405. * defined to do the wrong thing in this case.
  406. */
  407. local int codes(struct state *s,
  408. const struct huffman *lencode,
  409. const struct huffman *distcode)
  410. {
  411. int symbol; /* decoded symbol */
  412. int len; /* length for copy */
  413. unsigned dist; /* distance for copy */
  414. static const short lens[29] = { /* Size base for length codes 257..285 */
  415. 3, 4, 5, 6, 7, 8, 9, 10, 11, 13, 15, 17, 19, 23, 27, 31,
  416. 35, 43, 51, 59, 67, 83, 99, 115, 131, 163, 195, 227, 258};
  417. static const short lext[29] = { /* Extra bits for length codes 257..285 */
  418. 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2,
  419. 3, 3, 3, 3, 4, 4, 4, 4, 5, 5, 5, 5, 0};
  420. static const short dists[30] = { /* Offset base for distance codes 0..29 */
  421. 1, 2, 3, 4, 5, 7, 9, 13, 17, 25, 33, 49, 65, 97, 129, 193,
  422. 257, 385, 513, 769, 1025, 1537, 2049, 3073, 4097, 6145,
  423. 8193, 12289, 16385, 24577};
  424. static const short dext[30] = { /* Extra bits for distance codes 0..29 */
  425. 0, 0, 0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6,
  426. 7, 7, 8, 8, 9, 9, 10, 10, 11, 11,
  427. 12, 12, 13, 13};
  428. /* decode literals and length/distance pairs */
  429. do {
  430. symbol = decode(s, lencode);
  431. if (symbol < 0)
  432. return symbol; /* invalid symbol */
  433. if (symbol < 256) { /* literal: symbol is the byte */
  434. /* write out the literal */
  435. if (s->out != NIL) {
  436. if (s->outcnt == s->outlen)
  437. return 1;
  438. s->out[s->outcnt] = symbol;
  439. }
  440. s->outcnt++;
  441. }
  442. else if (symbol > 256) { /* length */
  443. /* get and compute length */
  444. symbol -= 257;
  445. if (symbol >= 29)
  446. return -10; /* invalid fixed code */
  447. len = lens[symbol] + bits(s, lext[symbol]);
  448. /* get and check distance */
  449. symbol = decode(s, distcode);
  450. if (symbol < 0)
  451. return symbol; /* invalid symbol */
  452. dist = dists[symbol] + bits(s, dext[symbol]);
  453. #ifndef INFLATE_ALLOW_INVALID_DISTANCE_TOOFAR_ARRR
  454. if (dist > s->outcnt)
  455. return -11; /* distance too far back */
  456. #endif
  457. /* copy length bytes from distance bytes back */
  458. if (s->out != NIL) {
  459. if (s->outcnt + len > s->outlen)
  460. return 1;
  461. while (len--) {
  462. s->out[s->outcnt] =
  463. #ifdef INFLATE_ALLOW_INVALID_DISTANCE_TOOFAR_ARRR
  464. dist > s->outcnt ?
  465. 0 :
  466. #endif
  467. s->out[s->outcnt - dist];
  468. s->outcnt++;
  469. }
  470. }
  471. else
  472. s->outcnt += len;
  473. }
  474. } while (symbol != 256); /* end of block symbol */
  475. /* done with a valid fixed or dynamic block */
  476. return 0;
  477. }
  478. /*
  479. * Process a fixed codes block.
  480. *
  481. * Format notes:
  482. *
  483. * - This block type can be useful for compressing small amounts of data for
  484. * which the size of the code descriptions in a dynamic block exceeds the
  485. * benefit of custom codes for that block. For fixed codes, no bits are
  486. * spent on code descriptions. Instead the code lengths for literal/length
  487. * codes and distance codes are fixed. The specific lengths for each symbol
  488. * can be seen in the "for" loops below.
  489. *
  490. * - The literal/length code is complete, but has two symbols that are invalid
  491. * and should result in an error if received. This cannot be implemented
  492. * simply as an incomplete code since those two symbols are in the "middle"
  493. * of the code. They are eight bits long and the longest literal/length\
  494. * code is nine bits. Therefore the code must be constructed with those
  495. * symbols, and the invalid symbols must be detected after decoding.
  496. *
  497. * - The fixed distance codes also have two invalid symbols that should result
  498. * in an error if received. Since all of the distance codes are the same
  499. * length, this can be implemented as an incomplete code. Then the invalid
  500. * codes are detected while decoding.
  501. */
  502. local int fixed(struct state *s)
  503. {
  504. static int virgin = 1;
  505. static short lencnt[MAXBITS+1], lensym[FIXLCODES];
  506. static short distcnt[MAXBITS+1], distsym[MAXDCODES];
  507. static struct huffman lencode, distcode;
  508. /* build fixed huffman tables if first call (may not be thread safe) */
  509. if (virgin) {
  510. int symbol;
  511. short lengths[FIXLCODES];
  512. /* construct lencode and distcode */
  513. lencode.count = lencnt;
  514. lencode.symbol = lensym;
  515. distcode.count = distcnt;
  516. distcode.symbol = distsym;
  517. /* literal/length table */
  518. for (symbol = 0; symbol < 144; symbol++)
  519. lengths[symbol] = 8;
  520. for (; symbol < 256; symbol++)
  521. lengths[symbol] = 9;
  522. for (; symbol < 280; symbol++)
  523. lengths[symbol] = 7;
  524. for (; symbol < FIXLCODES; symbol++)
  525. lengths[symbol] = 8;
  526. construct(&lencode, lengths, FIXLCODES);
  527. /* distance table */
  528. for (symbol = 0; symbol < MAXDCODES; symbol++)
  529. lengths[symbol] = 5;
  530. construct(&distcode, lengths, MAXDCODES);
  531. /* do this just once */
  532. virgin = 0;
  533. }
  534. /* decode data until end-of-block code */
  535. return codes(s, &lencode, &distcode);
  536. }
  537. /*
  538. * Process a dynamic codes block.
  539. *
  540. * Format notes:
  541. *
  542. * - A dynamic block starts with a description of the literal/length and
  543. * distance codes for that block. New dynamic blocks allow the compressor to
  544. * rapidly adapt to changing data with new codes optimized for that data.
  545. *
  546. * - The codes used by the deflate format are "canonical", which means that
  547. * the actual bits of the codes are generated in an unambiguous way simply
  548. * from the number of bits in each code. Therefore the code descriptions
  549. * are simply a list of code lengths for each symbol.
  550. *
  551. * - The code lengths are stored in order for the symbols, so lengths are
  552. * provided for each of the literal/length symbols, and for each of the
  553. * distance symbols.
  554. *
  555. * - If a symbol is not used in the block, this is represented by a zero as
  556. * as the code length. This does not mean a zero-length code, but rather
  557. * that no code should be created for this symbol. There is no way in the
  558. * deflate format to represent a zero-length code.
  559. *
  560. * - The maximum number of bits in a code is 15, so the possible lengths for
  561. * any code are 1..15.
  562. *
  563. * - The fact that a length of zero is not permitted for a code has an
  564. * interesting consequence. Normally if only one symbol is used for a given
  565. * code, then in fact that code could be represented with zero bits. However
  566. * in deflate, that code has to be at least one bit. So for example, if
  567. * only a single distance base symbol appears in a block, then it will be
  568. * represented by a single code of length one, in particular one 0 bit. This
  569. * is an incomplete code, since if a 1 bit is received, it has no meaning,
  570. * and should result in an error. So incomplete distance codes of one symbol
  571. * should be permitted, and the receipt of invalid codes should be handled.
  572. *
  573. * - It is also possible to have a single literal/length code, but that code
  574. * must be the end-of-block code, since every dynamic block has one. This
  575. * is not the most efficient way to create an empty block (an empty fixed
  576. * block is fewer bits), but it is allowed by the format. So incomplete
  577. * literal/length codes of one symbol should also be permitted.
  578. *
  579. * - If there are only literal codes and no lengths, then there are no distance
  580. * codes. This is represented by one distance code with zero bits.
  581. *
  582. * - The list of up to 286 length/literal lengths and up to 30 distance lengths
  583. * are themselves compressed using Huffman codes and run-length encoding. In
  584. * the list of code lengths, a 0 symbol means no code, a 1..15 symbol means
  585. * that length, and the symbols 16, 17, and 18 are run-length instructions.
  586. * Each of 16, 17, and 18 are follwed by extra bits to define the length of
  587. * the run. 16 copies the last length 3 to 6 times. 17 represents 3 to 10
  588. * zero lengths, and 18 represents 11 to 138 zero lengths. Unused symbols
  589. * are common, hence the special coding for zero lengths.
  590. *
  591. * - The symbols for 0..18 are Huffman coded, and so that code must be
  592. * described first. This is simply a sequence of up to 19 three-bit values
  593. * representing no code (0) or the code length for that symbol (1..7).
  594. *
  595. * - A dynamic block starts with three fixed-size counts from which is computed
  596. * the number of literal/length code lengths, the number of distance code
  597. * lengths, and the number of code length code lengths (ok, you come up with
  598. * a better name!) in the code descriptions. For the literal/length and
  599. * distance codes, lengths after those provided are considered zero, i.e. no
  600. * code. The code length code lengths are received in a permuted order (see
  601. * the order[] array below) to make a short code length code length list more
  602. * likely. As it turns out, very short and very long codes are less likely
  603. * to be seen in a dynamic code description, hence what may appear initially
  604. * to be a peculiar ordering.
  605. *
  606. * - Given the number of literal/length code lengths (nlen) and distance code
  607. * lengths (ndist), then they are treated as one long list of nlen + ndist
  608. * code lengths. Therefore run-length coding can and often does cross the
  609. * boundary between the two sets of lengths.
  610. *
  611. * - So to summarize, the code description at the start of a dynamic block is
  612. * three counts for the number of code lengths for the literal/length codes,
  613. * the distance codes, and the code length codes. This is followed by the
  614. * code length code lengths, three bits each. This is used to construct the
  615. * code length code which is used to read the remainder of the lengths. Then
  616. * the literal/length code lengths and distance lengths are read as a single
  617. * set of lengths using the code length codes. Codes are constructed from
  618. * the resulting two sets of lengths, and then finally you can start
  619. * decoding actual compressed data in the block.
  620. *
  621. * - For reference, a "typical" size for the code description in a dynamic
  622. * block is around 80 bytes.
  623. */
  624. local int dynamic(struct state *s)
  625. {
  626. int nlen, ndist, ncode; /* number of lengths in descriptor */
  627. int index; /* index of lengths[] */
  628. int err; /* construct() return value */
  629. short lengths[MAXCODES]; /* descriptor code lengths */
  630. short lencnt[MAXBITS+1], lensym[MAXLCODES]; /* lencode memory */
  631. short distcnt[MAXBITS+1], distsym[MAXDCODES]; /* distcode memory */
  632. struct huffman lencode, distcode; /* length and distance codes */
  633. static const short order[19] = /* permutation of code length codes */
  634. {16, 17, 18, 0, 8, 7, 9, 6, 10, 5, 11, 4, 12, 3, 13, 2, 14, 1, 15};
  635. /* construct lencode and distcode */
  636. lencode.count = lencnt;
  637. lencode.symbol = lensym;
  638. distcode.count = distcnt;
  639. distcode.symbol = distsym;
  640. /* get number of lengths in each table, check lengths */
  641. nlen = bits(s, 5) + 257;
  642. ndist = bits(s, 5) + 1;
  643. ncode = bits(s, 4) + 4;
  644. if (nlen > MAXLCODES || ndist > MAXDCODES)
  645. return -3; /* bad counts */
  646. /* read code length code lengths (really), missing lengths are zero */
  647. for (index = 0; index < ncode; index++)
  648. lengths[order[index]] = bits(s, 3);
  649. for (; index < 19; index++)
  650. lengths[order[index]] = 0;
  651. /* build huffman table for code lengths codes (use lencode temporarily) */
  652. err = construct(&lencode, lengths, 19);
  653. if (err != 0) /* require complete code set here */
  654. return -4;
  655. /* read length/literal and distance code length tables */
  656. index = 0;
  657. while (index < nlen + ndist) {
  658. int symbol; /* decoded value */
  659. int len; /* last length to repeat */
  660. symbol = decode(s, &lencode);
  661. if (symbol < 16) /* length in 0..15 */
  662. lengths[index++] = symbol;
  663. else { /* repeat instruction */
  664. len = 0; /* assume repeating zeros */
  665. if (symbol == 16) { /* repeat last length 3..6 times */
  666. if (index == 0)
  667. return -5; /* no last length! */
  668. len = lengths[index - 1]; /* last length */
  669. symbol = 3 + bits(s, 2);
  670. }
  671. else if (symbol == 17) /* repeat zero 3..10 times */
  672. symbol = 3 + bits(s, 3);
  673. else /* == 18, repeat zero 11..138 times */
  674. symbol = 11 + bits(s, 7);
  675. if (index + symbol > nlen + ndist)
  676. return -6; /* too many lengths! */
  677. while (symbol--) /* repeat last or zero symbol times */
  678. lengths[index++] = len;
  679. }
  680. }
  681. /* check for end-of-block code -- there better be one! */
  682. if (lengths[256] == 0)
  683. return -9;
  684. /* build huffman table for literal/length codes */
  685. err = construct(&lencode, lengths, nlen);
  686. if (err && (err < 0 || nlen != lencode.count[0] + lencode.count[1]))
  687. return -7; /* incomplete code ok only for single length 1 code */
  688. /* build huffman table for distance codes */
  689. err = construct(&distcode, lengths + nlen, ndist);
  690. if (err && (err < 0 || ndist != distcode.count[0] + distcode.count[1]))
  691. return -8; /* incomplete code ok only for single length 1 code */
  692. /* decode data until end-of-block code */
  693. return codes(s, &lencode, &distcode);
  694. }
  695. /*
  696. * Inflate source to dest. On return, destlen and sourcelen are updated to the
  697. * size of the uncompressed data and the size of the deflate data respectively.
  698. * On success, the return value of puff() is zero. If there is an error in the
  699. * source data, i.e. it is not in the deflate format, then a negative value is
  700. * returned. If there is not enough input available or there is not enough
  701. * output space, then a positive error is returned. In that case, destlen and
  702. * sourcelen are not updated to facilitate retrying from the beginning with the
  703. * provision of more input data or more output space. In the case of invalid
  704. * inflate data (a negative error), the dest and source pointers are updated to
  705. * facilitate the debugging of deflators.
  706. *
  707. * puff() also has a mode to determine the size of the uncompressed output with
  708. * no output written. For this dest must be (unsigned char *)0. In this case,
  709. * the input value of *destlen is ignored, and on return *destlen is set to the
  710. * size of the uncompressed output.
  711. *
  712. * The return codes are:
  713. *
  714. * 2: available inflate data did not terminate
  715. * 1: output space exhausted before completing inflate
  716. * 0: successful inflate
  717. * -1: invalid block type (type == 3)
  718. * -2: stored block length did not match one's complement
  719. * -3: dynamic block code description: too many length or distance codes
  720. * -4: dynamic block code description: code lengths codes incomplete
  721. * -5: dynamic block code description: repeat lengths with no first length
  722. * -6: dynamic block code description: repeat more than specified lengths
  723. * -7: dynamic block code description: invalid literal/length code lengths
  724. * -8: dynamic block code description: invalid distance code lengths
  725. * -9: dynamic block code description: missing end-of-block code
  726. * -10: invalid literal/length or distance code in fixed or dynamic block
  727. * -11: distance is too far back in fixed or dynamic block
  728. *
  729. * Format notes:
  730. *
  731. * - Three bits are read for each block to determine the kind of block and
  732. * whether or not it is the last block. Then the block is decoded and the
  733. * process repeated if it was not the last block.
  734. *
  735. * - The leftover bits in the last byte of the deflate data after the last
  736. * block (if it was a fixed or dynamic block) are undefined and have no
  737. * expected values to check.
  738. */
  739. int puff(unsigned char *dest, /* pointer to destination pointer */
  740. unsigned long *destlen, /* amount of output space */
  741. const unsigned char *source, /* pointer to source data pointer */
  742. unsigned long *sourcelen) /* amount of input available */
  743. {
  744. struct state s; /* input/output state */
  745. int last, type; /* block information */
  746. int err; /* return value */
  747. /* initialize output state */
  748. s.out = dest;
  749. s.outlen = *destlen; /* ignored if dest is NIL */
  750. s.outcnt = 0;
  751. /* initialize input state */
  752. s.in = source;
  753. s.inlen = *sourcelen;
  754. s.incnt = 0;
  755. s.bitbuf = 0;
  756. s.bitcnt = 0;
  757. /* return if bits() or decode() tries to read past available input */
  758. if (setjmp(s.env) != 0) /* if came back here via longjmp() */
  759. err = 2; /* then skip do-loop, return error */
  760. else {
  761. /* process blocks until last block or error */
  762. do {
  763. last = bits(&s, 1); /* one if last block */
  764. type = bits(&s, 2); /* block type 0..3 */
  765. err = type == 0 ?
  766. stored(&s) :
  767. (type == 1 ?
  768. fixed(&s) :
  769. (type == 2 ?
  770. dynamic(&s) :
  771. -1)); /* type == 3, invalid */
  772. if (err != 0)
  773. break; /* return with error */
  774. } while (!last);
  775. }
  776. /* update the lengths and return */
  777. if (err <= 0) {
  778. *destlen = s.outcnt;
  779. *sourcelen = s.incnt;
  780. }
  781. return err;
  782. }