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