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;
|