| 'use strict' |
| /* eslint camelcase: "off" */ |
| |
| var assert = require('assert') |
| |
| var Zstream = require('pako/lib/zlib/zstream') |
| var zlib_deflate = require('pako/lib/zlib/deflate.js') |
| var zlib_inflate = require('pako/lib/zlib/inflate.js') |
| var constants = require('pako/lib/zlib/constants') |
| |
| for (var key in constants) { |
| exports[key] = constants[key] |
| } |
| |
| // zlib modes |
| exports.NONE = 0 |
| exports.DEFLATE = 1 |
| exports.INFLATE = 2 |
| exports.GZIP = 3 |
| exports.GUNZIP = 4 |
| exports.DEFLATERAW = 5 |
| exports.INFLATERAW = 6 |
| exports.UNZIP = 7 |
| |
| var GZIP_HEADER_ID1 = 0x1f |
| var GZIP_HEADER_ID2 = 0x8b |
| |
| /** |
| * Emulate Node's zlib C++ layer for use by the JS layer in index.js |
| */ |
| function Zlib (mode) { |
| if (typeof mode !== 'number' || mode < exports.DEFLATE || mode > exports.UNZIP) { |
| throw new TypeError('Bad argument') |
| } |
| |
| this.dictionary = null |
| this.err = 0 |
| this.flush = 0 |
| this.init_done = false |
| this.level = 0 |
| this.memLevel = 0 |
| this.mode = mode |
| this.strategy = 0 |
| this.windowBits = 0 |
| this.write_in_progress = false |
| this.pending_close = false |
| this.gzip_id_bytes_read = 0 |
| } |
| |
| Zlib.prototype.close = function () { |
| if (this.write_in_progress) { |
| this.pending_close = true |
| return |
| } |
| |
| this.pending_close = false |
| |
| assert(this.init_done, 'close before init') |
| assert(this.mode <= exports.UNZIP) |
| |
| if (this.mode === exports.DEFLATE || this.mode === exports.GZIP || this.mode === exports.DEFLATERAW) { |
| zlib_deflate.deflateEnd(this.strm) |
| } else if (this.mode === exports.INFLATE || this.mode === exports.GUNZIP || |
| this.mode === exports.INFLATERAW || this.mode === exports.UNZIP) { |
| zlib_inflate.inflateEnd(this.strm) |
| } |
| |
| this.mode = exports.NONE |
| |
| this.dictionary = null |
| } |
| |
| Zlib.prototype.write = function (flush, input, in_off, in_len, out, out_off, out_len) { |
| return this._write(true, flush, input, in_off, in_len, out, out_off, out_len) |
| } |
| |
| Zlib.prototype.writeSync = function (flush, input, in_off, in_len, out, out_off, out_len) { |
| return this._write(false, flush, input, in_off, in_len, out, out_off, out_len) |
| } |
| |
| Zlib.prototype._write = function (async, flush, input, in_off, in_len, out, out_off, out_len) { |
| assert.equal(arguments.length, 8) |
| |
| assert(this.init_done, 'write before init') |
| assert(this.mode !== exports.NONE, 'already finalized') |
| assert.equal(false, this.write_in_progress, 'write already in progress') |
| assert.equal(false, this.pending_close, 'close is pending') |
| |
| this.write_in_progress = true |
| |
| assert.equal(false, flush === undefined, 'must provide flush value') |
| |
| this.write_in_progress = true |
| |
| if (flush !== exports.Z_NO_FLUSH && |
| flush !== exports.Z_PARTIAL_FLUSH && |
| flush !== exports.Z_SYNC_FLUSH && |
| flush !== exports.Z_FULL_FLUSH && |
| flush !== exports.Z_FINISH && |
| flush !== exports.Z_BLOCK) { |
| throw new Error('Invalid flush value') |
| } |
| |
| if (input == null) { |
| input = Buffer.alloc(0) |
| in_len = 0 |
| in_off = 0 |
| } |
| |
| this.strm.avail_in = in_len |
| this.strm.input = input |
| this.strm.next_in = in_off |
| this.strm.avail_out = out_len |
| this.strm.output = out |
| this.strm.next_out = out_off |
| this.flush = flush |
| |
| if (!async) { |
| // sync version |
| this._process() |
| |
| if (this._checkError()) { |
| return this._afterSync() |
| } |
| return |
| } |
| |
| // async version |
| var self = this |
| process.nextTick(function () { |
| self._process() |
| self._after() |
| }) |
| |
| return this |
| } |
| |
| Zlib.prototype._afterSync = function () { |
| var avail_out = this.strm.avail_out |
| var avail_in = this.strm.avail_in |
| |
| this.write_in_progress = false |
| |
| return [avail_in, avail_out] |
| } |
| |
| Zlib.prototype._process = function () { |
| var next_expected_header_byte = null |
| |
| // If the avail_out is left at 0, then it means that it ran out |
| // of room. If there was avail_out left over, then it means |
| // that all of the input was consumed. |
| switch (this.mode) { |
| case exports.DEFLATE: |
| case exports.GZIP: |
| case exports.DEFLATERAW: |
| this.err = zlib_deflate.deflate(this.strm, this.flush) |
| break |
| case exports.UNZIP: |
| if (this.strm.avail_in > 0) { |
| next_expected_header_byte = this.strm.next_in |
| } |
| |
| switch (this.gzip_id_bytes_read) { |
| case 0: |
| if (next_expected_header_byte === null) { |
| break |
| } |
| |
| if (this.strm.input[next_expected_header_byte] === GZIP_HEADER_ID1) { |
| this.gzip_id_bytes_read = 1 |
| next_expected_header_byte++ |
| |
| if (this.strm.avail_in === 1) { |
| // The only available byte was already read. |
| break |
| } |
| } else { |
| this.mode = exports.INFLATE |
| break |
| } |
| |
| // fallthrough |
| case 1: |
| if (next_expected_header_byte === null) { |
| break |
| } |
| |
| if (this.strm.input[next_expected_header_byte] === GZIP_HEADER_ID2) { |
| this.gzip_id_bytes_read = 2 |
| this.mode = exports.GUNZIP |
| } else { |
| // There is no actual difference between INFLATE and INFLATERAW |
| // (after initialization). |
| this.mode = exports.INFLATE |
| } |
| |
| break |
| default: |
| throw new Error('invalid number of gzip magic number bytes read') |
| } |
| |
| // fallthrough |
| case exports.INFLATE: |
| case exports.GUNZIP: |
| case exports.INFLATERAW: |
| this.err = zlib_inflate.inflate(this.strm, this.flush) |
| |
| // If data was encoded with dictionary |
| if (this.err === exports.Z_NEED_DICT && this.dictionary) { |
| // Load it |
| this.err = zlib_inflate.inflateSetDictionary(this.strm, this.dictionary) |
| if (this.err === exports.Z_OK) { |
| // And try to decode again |
| this.err = zlib_inflate.inflate(this.strm, this.flush) |
| } else if (this.err === exports.Z_DATA_ERROR) { |
| // Both inflateSetDictionary() and inflate() return Z_DATA_ERROR. |
| // Make it possible for After() to tell a bad dictionary from bad |
| // input. |
| this.err = exports.Z_NEED_DICT |
| } |
| } |
| while (this.strm.avail_in > 0 && |
| this.mode === exports.GUNZIP && |
| this.err === exports.Z_STREAM_END && |
| this.strm.next_in[0] !== 0x00) { |
| // Bytes remain in input buffer. Perhaps this is another compressed |
| // member in the same archive, or just trailing garbage. |
| // Trailing zero bytes are okay, though, since they are frequently |
| // used for padding. |
| |
| this.reset() |
| this.err = zlib_inflate.inflate(this.strm, this.flush) |
| } |
| break |
| default: |
| throw new Error('Unknown mode ' + this.mode) |
| } |
| } |
| |
| Zlib.prototype._checkError = function () { |
| // Acceptable error states depend on the type of zlib stream. |
| switch (this.err) { |
| case exports.Z_OK: |
| case exports.Z_BUF_ERROR: |
| if (this.strm.avail_out !== 0 && this.flush === exports.Z_FINISH) { |
| this._error('unexpected end of file') |
| return false |
| } |
| break |
| case exports.Z_STREAM_END: |
| // normal statuses, not fatal |
| break |
| case exports.Z_NEED_DICT: |
| if (this.dictionary == null) { |
| this._error('Missing dictionary') |
| } else { |
| this._error('Bad dictionary') |
| } |
| return false |
| default: |
| // something else. |
| this._error('Zlib error') |
| return false |
| } |
| |
| return true |
| } |
| |
| Zlib.prototype._after = function () { |
| if (!this._checkError()) { |
| return |
| } |
| |
| var avail_out = this.strm.avail_out |
| var avail_in = this.strm.avail_in |
| |
| this.write_in_progress = false |
| |
| // call the write() cb |
| this.callback(avail_in, avail_out) |
| |
| if (this.pending_close) { |
| this.close() |
| } |
| } |
| |
| Zlib.prototype._error = function (message) { |
| if (this.strm.msg) { |
| message = this.strm.msg |
| } |
| this.onerror(message, this.err) |
| |
| // no hope of rescue. |
| this.write_in_progress = false |
| if (this.pending_close) { |
| this.close() |
| } |
| } |
| |
| Zlib.prototype.init = function (windowBits, level, memLevel, strategy, dictionary) { |
| assert(arguments.length === 4 || arguments.length === 5, 'init(windowBits, level, memLevel, strategy, [dictionary])') |
| |
| assert(windowBits >= 8 && windowBits <= 15, 'invalid windowBits') |
| assert(level >= -1 && level <= 9, 'invalid compression level') |
| |
| assert(memLevel >= 1 && memLevel <= 9, 'invalid memlevel') |
| |
| assert(strategy === exports.Z_FILTERED || |
| strategy === exports.Z_HUFFMAN_ONLY || |
| strategy === exports.Z_RLE || |
| strategy === exports.Z_FIXED || |
| strategy === exports.Z_DEFAULT_STRATEGY, 'invalid strategy') |
| |
| this._init(level, windowBits, memLevel, strategy, dictionary) |
| this._setDictionary() |
| } |
| |
| Zlib.prototype.params = function () { |
| throw new Error('deflateParams Not supported') |
| } |
| |
| Zlib.prototype.reset = function () { |
| this._reset() |
| this._setDictionary() |
| } |
| |
| Zlib.prototype._init = function (level, windowBits, memLevel, strategy, dictionary) { |
| this.level = level |
| this.windowBits = windowBits |
| this.memLevel = memLevel |
| this.strategy = strategy |
| |
| this.flush = exports.Z_NO_FLUSH |
| |
| this.err = exports.Z_OK |
| |
| if (this.mode === exports.GZIP || this.mode === exports.GUNZIP) { |
| this.windowBits += 16 |
| } |
| |
| if (this.mode === exports.UNZIP) { |
| this.windowBits += 32 |
| } |
| |
| if (this.mode === exports.DEFLATERAW || this.mode === exports.INFLATERAW) { |
| this.windowBits = -1 * this.windowBits |
| } |
| |
| this.strm = new Zstream() |
| |
| switch (this.mode) { |
| case exports.DEFLATE: |
| case exports.GZIP: |
| case exports.DEFLATERAW: |
| this.err = zlib_deflate.deflateInit2( |
| this.strm, |
| this.level, |
| exports.Z_DEFLATED, |
| this.windowBits, |
| this.memLevel, |
| this.strategy |
| ) |
| break |
| case exports.INFLATE: |
| case exports.GUNZIP: |
| case exports.INFLATERAW: |
| case exports.UNZIP: |
| this.err = zlib_inflate.inflateInit2( |
| this.strm, |
| this.windowBits |
| ) |
| break |
| default: |
| throw new Error('Unknown mode ' + this.mode) |
| } |
| |
| if (this.err !== exports.Z_OK) { |
| this._error('Init error') |
| } |
| |
| this.dictionary = dictionary |
| |
| this.write_in_progress = false |
| this.init_done = true |
| } |
| |
| Zlib.prototype._setDictionary = function () { |
| if (this.dictionary == null) { |
| return |
| } |
| |
| this.err = exports.Z_OK |
| |
| switch (this.mode) { |
| case exports.DEFLATE: |
| case exports.DEFLATERAW: |
| this.err = zlib_deflate.deflateSetDictionary(this.strm, this.dictionary) |
| break |
| default: |
| break |
| } |
| |
| if (this.err !== exports.Z_OK) { |
| this._error('Failed to set dictionary') |
| } |
| } |
| |
| Zlib.prototype._reset = function () { |
| this.err = exports.Z_OK |
| |
| switch (this.mode) { |
| case exports.DEFLATE: |
| case exports.DEFLATERAW: |
| case exports.GZIP: |
| this.err = zlib_deflate.deflateReset(this.strm) |
| break |
| case exports.INFLATE: |
| case exports.INFLATERAW: |
| case exports.GUNZIP: |
| this.err = zlib_inflate.inflateReset(this.strm) |
| break |
| default: |
| break |
| } |
| |
| if (this.err !== exports.Z_OK) { |
| this._error('Failed to reset stream') |
| } |
| } |
| |
| exports.Zlib = Zlib |