blob: 14e419d2ea18a4cc066671fb31721e5bbe5f1285 [file] [log] [blame]
'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