1 | 'use strict';
|
---|
2 |
|
---|
3 |
|
---|
4 | var zlib_inflate = require('./zlib/inflate');
|
---|
5 | var utils = require('./utils/common');
|
---|
6 | var strings = require('./utils/strings');
|
---|
7 | var c = require('./zlib/constants');
|
---|
8 | var msg = require('./zlib/messages');
|
---|
9 | var ZStream = require('./zlib/zstream');
|
---|
10 | var GZheader = require('./zlib/gzheader');
|
---|
11 |
|
---|
12 | var toString = Object.prototype.toString;
|
---|
13 |
|
---|
14 | /**
|
---|
15 | * class Inflate
|
---|
16 | *
|
---|
17 | * Generic JS-style wrapper for zlib calls. If you don't need
|
---|
18 | * streaming behaviour - use more simple functions: [[inflate]]
|
---|
19 | * and [[inflateRaw]].
|
---|
20 | **/
|
---|
21 |
|
---|
22 | /* internal
|
---|
23 | * inflate.chunks -> Array
|
---|
24 | *
|
---|
25 | * Chunks of output data, if [[Inflate#onData]] not overridden.
|
---|
26 | **/
|
---|
27 |
|
---|
28 | /**
|
---|
29 | * Inflate.result -> Uint8Array|Array|String
|
---|
30 | *
|
---|
31 | * Uncompressed result, generated by default [[Inflate#onData]]
|
---|
32 | * and [[Inflate#onEnd]] handlers. Filled after you push last chunk
|
---|
33 | * (call [[Inflate#push]] with `Z_FINISH` / `true` param) or if you
|
---|
34 | * push a chunk with explicit flush (call [[Inflate#push]] with
|
---|
35 | * `Z_SYNC_FLUSH` param).
|
---|
36 | **/
|
---|
37 |
|
---|
38 | /**
|
---|
39 | * Inflate.err -> Number
|
---|
40 | *
|
---|
41 | * Error code after inflate finished. 0 (Z_OK) on success.
|
---|
42 | * Should be checked if broken data possible.
|
---|
43 | **/
|
---|
44 |
|
---|
45 | /**
|
---|
46 | * Inflate.msg -> String
|
---|
47 | *
|
---|
48 | * Error message, if [[Inflate.err]] != 0
|
---|
49 | **/
|
---|
50 |
|
---|
51 |
|
---|
52 | /**
|
---|
53 | * new Inflate(options)
|
---|
54 | * - options (Object): zlib inflate options.
|
---|
55 | *
|
---|
56 | * Creates new inflator instance with specified params. Throws exception
|
---|
57 | * on bad params. Supported options:
|
---|
58 | *
|
---|
59 | * - `windowBits`
|
---|
60 | * - `dictionary`
|
---|
61 | *
|
---|
62 | * [http://zlib.net/manual.html#Advanced](http://zlib.net/manual.html#Advanced)
|
---|
63 | * for more information on these.
|
---|
64 | *
|
---|
65 | * Additional options, for internal needs:
|
---|
66 | *
|
---|
67 | * - `chunkSize` - size of generated data chunks (16K by default)
|
---|
68 | * - `raw` (Boolean) - do raw inflate
|
---|
69 | * - `to` (String) - if equal to 'string', then result will be converted
|
---|
70 | * from utf8 to utf16 (javascript) string. When string output requested,
|
---|
71 | * chunk length can differ from `chunkSize`, depending on content.
|
---|
72 | *
|
---|
73 | * By default, when no options set, autodetect deflate/gzip data format via
|
---|
74 | * wrapper header.
|
---|
75 | *
|
---|
76 | * ##### Example:
|
---|
77 | *
|
---|
78 | * ```javascript
|
---|
79 | * var pako = require('pako')
|
---|
80 | * , chunk1 = Uint8Array([1,2,3,4,5,6,7,8,9])
|
---|
81 | * , chunk2 = Uint8Array([10,11,12,13,14,15,16,17,18,19]);
|
---|
82 | *
|
---|
83 | * var inflate = new pako.Inflate({ level: 3});
|
---|
84 | *
|
---|
85 | * inflate.push(chunk1, false);
|
---|
86 | * inflate.push(chunk2, true); // true -> last chunk
|
---|
87 | *
|
---|
88 | * if (inflate.err) { throw new Error(inflate.err); }
|
---|
89 | *
|
---|
90 | * console.log(inflate.result);
|
---|
91 | * ```
|
---|
92 | **/
|
---|
93 | function Inflate(options) {
|
---|
94 | if (!(this instanceof Inflate)) return new Inflate(options);
|
---|
95 |
|
---|
96 | this.options = utils.assign({
|
---|
97 | chunkSize: 16384,
|
---|
98 | windowBits: 0,
|
---|
99 | to: ''
|
---|
100 | }, options || {});
|
---|
101 |
|
---|
102 | var opt = this.options;
|
---|
103 |
|
---|
104 | // Force window size for `raw` data, if not set directly,
|
---|
105 | // because we have no header for autodetect.
|
---|
106 | if (opt.raw && (opt.windowBits >= 0) && (opt.windowBits < 16)) {
|
---|
107 | opt.windowBits = -opt.windowBits;
|
---|
108 | if (opt.windowBits === 0) { opt.windowBits = -15; }
|
---|
109 | }
|
---|
110 |
|
---|
111 | // If `windowBits` not defined (and mode not raw) - set autodetect flag for gzip/deflate
|
---|
112 | if ((opt.windowBits >= 0) && (opt.windowBits < 16) &&
|
---|
113 | !(options && options.windowBits)) {
|
---|
114 | opt.windowBits += 32;
|
---|
115 | }
|
---|
116 |
|
---|
117 | // Gzip header has no info about windows size, we can do autodetect only
|
---|
118 | // for deflate. So, if window size not set, force it to max when gzip possible
|
---|
119 | if ((opt.windowBits > 15) && (opt.windowBits < 48)) {
|
---|
120 | // bit 3 (16) -> gzipped data
|
---|
121 | // bit 4 (32) -> autodetect gzip/deflate
|
---|
122 | if ((opt.windowBits & 15) === 0) {
|
---|
123 | opt.windowBits |= 15;
|
---|
124 | }
|
---|
125 | }
|
---|
126 |
|
---|
127 | this.err = 0; // error code, if happens (0 = Z_OK)
|
---|
128 | this.msg = ''; // error message
|
---|
129 | this.ended = false; // used to avoid multiple onEnd() calls
|
---|
130 | this.chunks = []; // chunks of compressed data
|
---|
131 |
|
---|
132 | this.strm = new ZStream();
|
---|
133 | this.strm.avail_out = 0;
|
---|
134 |
|
---|
135 | var status = zlib_inflate.inflateInit2(
|
---|
136 | this.strm,
|
---|
137 | opt.windowBits
|
---|
138 | );
|
---|
139 |
|
---|
140 | if (status !== c.Z_OK) {
|
---|
141 | throw new Error(msg[status]);
|
---|
142 | }
|
---|
143 |
|
---|
144 | this.header = new GZheader();
|
---|
145 |
|
---|
146 | zlib_inflate.inflateGetHeader(this.strm, this.header);
|
---|
147 |
|
---|
148 | // Setup dictionary
|
---|
149 | if (opt.dictionary) {
|
---|
150 | // Convert data if needed
|
---|
151 | if (typeof opt.dictionary === 'string') {
|
---|
152 | opt.dictionary = strings.string2buf(opt.dictionary);
|
---|
153 | } else if (toString.call(opt.dictionary) === '[object ArrayBuffer]') {
|
---|
154 | opt.dictionary = new Uint8Array(opt.dictionary);
|
---|
155 | }
|
---|
156 | if (opt.raw) { //In raw mode we need to set the dictionary early
|
---|
157 | status = zlib_inflate.inflateSetDictionary(this.strm, opt.dictionary);
|
---|
158 | if (status !== c.Z_OK) {
|
---|
159 | throw new Error(msg[status]);
|
---|
160 | }
|
---|
161 | }
|
---|
162 | }
|
---|
163 | }
|
---|
164 |
|
---|
165 | /**
|
---|
166 | * Inflate#push(data[, mode]) -> Boolean
|
---|
167 | * - data (Uint8Array|Array|ArrayBuffer|String): input data
|
---|
168 | * - mode (Number|Boolean): 0..6 for corresponding Z_NO_FLUSH..Z_TREE modes.
|
---|
169 | * See constants. Skipped or `false` means Z_NO_FLUSH, `true` means Z_FINISH.
|
---|
170 | *
|
---|
171 | * Sends input data to inflate pipe, generating [[Inflate#onData]] calls with
|
---|
172 | * new output chunks. Returns `true` on success. The last data block must have
|
---|
173 | * mode Z_FINISH (or `true`). That will flush internal pending buffers and call
|
---|
174 | * [[Inflate#onEnd]]. For interim explicit flushes (without ending the stream) you
|
---|
175 | * can use mode Z_SYNC_FLUSH, keeping the decompression context.
|
---|
176 | *
|
---|
177 | * On fail call [[Inflate#onEnd]] with error code and return false.
|
---|
178 | *
|
---|
179 | * We strongly recommend to use `Uint8Array` on input for best speed (output
|
---|
180 | * format is detected automatically). Also, don't skip last param and always
|
---|
181 | * use the same type in your code (boolean or number). That will improve JS speed.
|
---|
182 | *
|
---|
183 | * For regular `Array`-s make sure all elements are [0..255].
|
---|
184 | *
|
---|
185 | * ##### Example
|
---|
186 | *
|
---|
187 | * ```javascript
|
---|
188 | * push(chunk, false); // push one of data chunks
|
---|
189 | * ...
|
---|
190 | * push(chunk, true); // push last chunk
|
---|
191 | * ```
|
---|
192 | **/
|
---|
193 | Inflate.prototype.push = function (data, mode) {
|
---|
194 | var strm = this.strm;
|
---|
195 | var chunkSize = this.options.chunkSize;
|
---|
196 | var dictionary = this.options.dictionary;
|
---|
197 | var status, _mode;
|
---|
198 | var next_out_utf8, tail, utf8str;
|
---|
199 |
|
---|
200 | // Flag to properly process Z_BUF_ERROR on testing inflate call
|
---|
201 | // when we check that all output data was flushed.
|
---|
202 | var allowBufError = false;
|
---|
203 |
|
---|
204 | if (this.ended) { return false; }
|
---|
205 | _mode = (mode === ~~mode) ? mode : ((mode === true) ? c.Z_FINISH : c.Z_NO_FLUSH);
|
---|
206 |
|
---|
207 | // Convert data if needed
|
---|
208 | if (typeof data === 'string') {
|
---|
209 | // Only binary strings can be decompressed on practice
|
---|
210 | strm.input = strings.binstring2buf(data);
|
---|
211 | } else if (toString.call(data) === '[object ArrayBuffer]') {
|
---|
212 | strm.input = new Uint8Array(data);
|
---|
213 | } else {
|
---|
214 | strm.input = data;
|
---|
215 | }
|
---|
216 |
|
---|
217 | strm.next_in = 0;
|
---|
218 | strm.avail_in = strm.input.length;
|
---|
219 |
|
---|
220 | do {
|
---|
221 | if (strm.avail_out === 0) {
|
---|
222 | strm.output = new utils.Buf8(chunkSize);
|
---|
223 | strm.next_out = 0;
|
---|
224 | strm.avail_out = chunkSize;
|
---|
225 | }
|
---|
226 |
|
---|
227 | status = zlib_inflate.inflate(strm, c.Z_NO_FLUSH); /* no bad return value */
|
---|
228 |
|
---|
229 | if (status === c.Z_NEED_DICT && dictionary) {
|
---|
230 | status = zlib_inflate.inflateSetDictionary(this.strm, dictionary);
|
---|
231 | }
|
---|
232 |
|
---|
233 | if (status === c.Z_BUF_ERROR && allowBufError === true) {
|
---|
234 | status = c.Z_OK;
|
---|
235 | allowBufError = false;
|
---|
236 | }
|
---|
237 |
|
---|
238 | if (status !== c.Z_STREAM_END && status !== c.Z_OK) {
|
---|
239 | this.onEnd(status);
|
---|
240 | this.ended = true;
|
---|
241 | return false;
|
---|
242 | }
|
---|
243 |
|
---|
244 | if (strm.next_out) {
|
---|
245 | if (strm.avail_out === 0 || status === c.Z_STREAM_END || (strm.avail_in === 0 && (_mode === c.Z_FINISH || _mode === c.Z_SYNC_FLUSH))) {
|
---|
246 |
|
---|
247 | if (this.options.to === 'string') {
|
---|
248 |
|
---|
249 | next_out_utf8 = strings.utf8border(strm.output, strm.next_out);
|
---|
250 |
|
---|
251 | tail = strm.next_out - next_out_utf8;
|
---|
252 | utf8str = strings.buf2string(strm.output, next_out_utf8);
|
---|
253 |
|
---|
254 | // move tail
|
---|
255 | strm.next_out = tail;
|
---|
256 | strm.avail_out = chunkSize - tail;
|
---|
257 | if (tail) { utils.arraySet(strm.output, strm.output, next_out_utf8, tail, 0); }
|
---|
258 |
|
---|
259 | this.onData(utf8str);
|
---|
260 |
|
---|
261 | } else {
|
---|
262 | this.onData(utils.shrinkBuf(strm.output, strm.next_out));
|
---|
263 | }
|
---|
264 | }
|
---|
265 | }
|
---|
266 |
|
---|
267 | // When no more input data, we should check that internal inflate buffers
|
---|
268 | // are flushed. The only way to do it when avail_out = 0 - run one more
|
---|
269 | // inflate pass. But if output data not exists, inflate return Z_BUF_ERROR.
|
---|
270 | // Here we set flag to process this error properly.
|
---|
271 | //
|
---|
272 | // NOTE. Deflate does not return error in this case and does not needs such
|
---|
273 | // logic.
|
---|
274 | if (strm.avail_in === 0 && strm.avail_out === 0) {
|
---|
275 | allowBufError = true;
|
---|
276 | }
|
---|
277 |
|
---|
278 | } while ((strm.avail_in > 0 || strm.avail_out === 0) && status !== c.Z_STREAM_END);
|
---|
279 |
|
---|
280 | if (status === c.Z_STREAM_END) {
|
---|
281 | _mode = c.Z_FINISH;
|
---|
282 | }
|
---|
283 |
|
---|
284 | // Finalize on the last chunk.
|
---|
285 | if (_mode === c.Z_FINISH) {
|
---|
286 | status = zlib_inflate.inflateEnd(this.strm);
|
---|
287 | this.onEnd(status);
|
---|
288 | this.ended = true;
|
---|
289 | return status === c.Z_OK;
|
---|
290 | }
|
---|
291 |
|
---|
292 | // callback interim results if Z_SYNC_FLUSH.
|
---|
293 | if (_mode === c.Z_SYNC_FLUSH) {
|
---|
294 | this.onEnd(c.Z_OK);
|
---|
295 | strm.avail_out = 0;
|
---|
296 | return true;
|
---|
297 | }
|
---|
298 |
|
---|
299 | return true;
|
---|
300 | };
|
---|
301 |
|
---|
302 |
|
---|
303 | /**
|
---|
304 | * Inflate#onData(chunk) -> Void
|
---|
305 | * - chunk (Uint8Array|Array|String): output data. Type of array depends
|
---|
306 | * on js engine support. When string output requested, each chunk
|
---|
307 | * will be string.
|
---|
308 | *
|
---|
309 | * By default, stores data blocks in `chunks[]` property and glue
|
---|
310 | * those in `onEnd`. Override this handler, if you need another behaviour.
|
---|
311 | **/
|
---|
312 | Inflate.prototype.onData = function (chunk) {
|
---|
313 | this.chunks.push(chunk);
|
---|
314 | };
|
---|
315 |
|
---|
316 |
|
---|
317 | /**
|
---|
318 | * Inflate#onEnd(status) -> Void
|
---|
319 | * - status (Number): inflate status. 0 (Z_OK) on success,
|
---|
320 | * other if not.
|
---|
321 | *
|
---|
322 | * Called either after you tell inflate that the input stream is
|
---|
323 | * complete (Z_FINISH) or should be flushed (Z_SYNC_FLUSH)
|
---|
324 | * or if an error happened. By default - join collected chunks,
|
---|
325 | * free memory and fill `results` / `err` properties.
|
---|
326 | **/
|
---|
327 | Inflate.prototype.onEnd = function (status) {
|
---|
328 | // On success - join
|
---|
329 | if (status === c.Z_OK) {
|
---|
330 | if (this.options.to === 'string') {
|
---|
331 | // Glue & convert here, until we teach pako to send
|
---|
332 | // utf8 aligned strings to onData
|
---|
333 | this.result = this.chunks.join('');
|
---|
334 | } else {
|
---|
335 | this.result = utils.flattenChunks(this.chunks);
|
---|
336 | }
|
---|
337 | }
|
---|
338 | this.chunks = [];
|
---|
339 | this.err = status;
|
---|
340 | this.msg = this.strm.msg;
|
---|
341 | };
|
---|
342 |
|
---|
343 |
|
---|
344 | /**
|
---|
345 | * inflate(data[, options]) -> Uint8Array|Array|String
|
---|
346 | * - data (Uint8Array|Array|String): input data to decompress.
|
---|
347 | * - options (Object): zlib inflate options.
|
---|
348 | *
|
---|
349 | * Decompress `data` with inflate/ungzip and `options`. Autodetect
|
---|
350 | * format via wrapper header by default. That's why we don't provide
|
---|
351 | * separate `ungzip` method.
|
---|
352 | *
|
---|
353 | * Supported options are:
|
---|
354 | *
|
---|
355 | * - windowBits
|
---|
356 | *
|
---|
357 | * [http://zlib.net/manual.html#Advanced](http://zlib.net/manual.html#Advanced)
|
---|
358 | * for more information.
|
---|
359 | *
|
---|
360 | * Sugar (options):
|
---|
361 | *
|
---|
362 | * - `raw` (Boolean) - say that we work with raw stream, if you don't wish to specify
|
---|
363 | * negative windowBits implicitly.
|
---|
364 | * - `to` (String) - if equal to 'string', then result will be converted
|
---|
365 | * from utf8 to utf16 (javascript) string. When string output requested,
|
---|
366 | * chunk length can differ from `chunkSize`, depending on content.
|
---|
367 | *
|
---|
368 | *
|
---|
369 | * ##### Example:
|
---|
370 | *
|
---|
371 | * ```javascript
|
---|
372 | * var pako = require('pako')
|
---|
373 | * , input = pako.deflate([1,2,3,4,5,6,7,8,9])
|
---|
374 | * , output;
|
---|
375 | *
|
---|
376 | * try {
|
---|
377 | * output = pako.inflate(input);
|
---|
378 | * } catch (err)
|
---|
379 | * console.log(err);
|
---|
380 | * }
|
---|
381 | * ```
|
---|
382 | **/
|
---|
383 | function inflate(input, options) {
|
---|
384 | var inflator = new Inflate(options);
|
---|
385 |
|
---|
386 | inflator.push(input, true);
|
---|
387 |
|
---|
388 | // That will never happens, if you don't cheat with options :)
|
---|
389 | if (inflator.err) { throw inflator.msg || msg[inflator.err]; }
|
---|
390 |
|
---|
391 | return inflator.result;
|
---|
392 | }
|
---|
393 |
|
---|
394 |
|
---|
395 | /**
|
---|
396 | * inflateRaw(data[, options]) -> Uint8Array|Array|String
|
---|
397 | * - data (Uint8Array|Array|String): input data to decompress.
|
---|
398 | * - options (Object): zlib inflate options.
|
---|
399 | *
|
---|
400 | * The same as [[inflate]], but creates raw data, without wrapper
|
---|
401 | * (header and adler32 crc).
|
---|
402 | **/
|
---|
403 | function inflateRaw(input, options) {
|
---|
404 | options = options || {};
|
---|
405 | options.raw = true;
|
---|
406 | return inflate(input, options);
|
---|
407 | }
|
---|
408 |
|
---|
409 |
|
---|
410 | /**
|
---|
411 | * ungzip(data[, options]) -> Uint8Array|Array|String
|
---|
412 | * - data (Uint8Array|Array|String): input data to decompress.
|
---|
413 | * - options (Object): zlib inflate options.
|
---|
414 | *
|
---|
415 | * Just shortcut to [[inflate]], because it autodetects format
|
---|
416 | * by header.content. Done for convenience.
|
---|
417 | **/
|
---|
418 |
|
---|
419 |
|
---|
420 | exports.Inflate = Inflate;
|
---|
421 | exports.inflate = inflate;
|
---|
422 | exports.inflateRaw = inflateRaw;
|
---|
423 | exports.ungzip = inflate;
|
---|