123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426 |
- var assert = require('minimalistic-assert');
- var Buffer = require('buffer').Buffer;
- function WBuf() {
- this.buffers = [];
- this.toReserve = 0;
- this.size = 0;
- this.maxSize = 0;
- this.avail = 0;
- this.last = null;
- this.offset = 0;
- // Used in slicing
- this.sliceQueue = null;
- this.forceReserve = false;
- // Mostly a constant
- this.reserveRate = 64;
- }
- module.exports = WBuf;
- WBuf.prototype.reserve = function reserve(n) {
- this.toReserve += n;
- // Force reservation of extra bytes
- if (this.forceReserve)
- this.toReserve = Math.max(this.toReserve, this.reserveRate);
- };
- WBuf.prototype._ensure = function _ensure(n) {
- if (this.avail >= n)
- return;
- if (this.toReserve === 0)
- this.toReserve = this.reserveRate;
- this.toReserve = Math.max(n - this.avail, this.toReserve);
- if (this.avail === 0)
- this._next();
- };
- WBuf.prototype._next = function _next() {
- var buf;
- if (this.sliceQueue === null) {
- // Most common case
- buf = new Buffer(this.toReserve);
- } else {
- // Only for `.slice()` results
- buf = this.sliceQueue.shift();
- if (this.sliceQueue.length === 0)
- this.sliceQueue = null;
- }
- this.toReserve = 0;
- this.buffers.push(buf);
- this.avail = buf.length;
- this.offset = 0;
- this.last = buf;
- };
- WBuf.prototype._rangeCheck = function _rangeCheck() {
- if (this.maxSize !== 0 && this.size > this.maxSize)
- throw new RangeError('WBuf overflow');
- };
- WBuf.prototype._move = function _move(n) {
- this.size += n;
- if (this.avail === 0)
- this.last = null;
- this._rangeCheck();
- };
- WBuf.prototype.slice = function slice(start, end) {
- assert(0 <= start && start <= this.size);
- assert(0 <= end && end <= this.size);
- if (this.last === null)
- this._next();
- var res = new WBuf();
- // Only last chunk is requested
- if (start >= this.size - this.offset) {
- res.buffers.push(this.last);
- res.last = this.last;
- res.offset = start - this.size + this.offset;
- res.maxSize = end - start;
- res.avail = res.maxSize;
- return res;
- }
- var startIndex = -1;
- var startOffset = 0;
- var endIndex = -1;
- // Find buffer indices
- var offset = 0;
- for (var i = 0; i < this.buffers.length; i++) {
- var buf = this.buffers[i];
- var next = offset + buf.length;
- // Found the start
- if (start >= offset && start <= next) {
- startIndex = i;
- startOffset = start - offset;
- if (endIndex !== -1)
- break;
- }
- if (end >= offset && end <= next) {
- endIndex = i;
- if (startIndex !== -1)
- break;
- }
- offset = next;
- }
- res.last = this.buffers[startIndex];
- res.offset = startOffset;
- res.maxSize = end - start;
- // Multi-buffer slice
- if (startIndex < endIndex) {
- res.sliceQueue = this.buffers.slice(startIndex + 1, endIndex + 1);
- res.last = res.last.slice(res.offset);
- res.offset = 0;
- }
- res.avail = res.last.length - res.offset;
- res.buffers.push(res.last);
- return res;
- };
- WBuf.prototype.skip = function skip(n) {
- if (n === 0)
- return this.slice(this.size, this.size);
- this._ensure(n);
- var left = n;
- while (left > 0) {
- var toSkip = Math.min(left, this.avail);
- left -= toSkip;
- this.size += toSkip;
- if (toSkip === this.avail) {
- if (left !== 0) {
- this._next();
- } else {
- this.avail -= toSkip;
- this.offset += toSkip;
- }
- } else {
- this.offset += toSkip;
- this.avail -= toSkip;
- }
- }
- this._rangeCheck();
- return this.slice(this.size - n, this.size);
- };
- WBuf.prototype.write = function write(str) {
- var len = 0;
- for (var i = 0; i < str.length; i++) {
- var c = str.charCodeAt(i);
- if (c > 255)
- len += 2;
- else
- len += 1;
- }
- this.reserve(len);
- for (var i = 0; i < str.length; i++) {
- var c = str.charCodeAt(i);
- var hi = c >>> 8;
- var lo = c & 0xff;
- if (hi)
- this.writeUInt8(hi);
- this.writeUInt8(lo);
- }
- };
- WBuf.prototype.copyFrom = function copyFrom(buf, start, end) {
- var off = start === undefined ? 0 : start;
- var len = end === undefined ? buf.length : end;
- if (off === len)
- return;
- this._ensure(len - off);
- while (off < len) {
- var toCopy = Math.min(len - off, this.avail);
- buf.copy(this.last, this.offset, off, off + toCopy);
- off += toCopy;
- this.size += toCopy;
- if (toCopy === this.avail) {
- if (off !== len) {
- this._next();
- } else {
- this.avail = 0;
- this.offset += toCopy;
- }
- } else {
- this.offset += toCopy;
- this.avail -= toCopy;
- }
- }
- this._rangeCheck();
- };
- WBuf.prototype.writeUInt8 = function writeUInt8(v) {
- this._ensure(1);
- this.last[this.offset++] = v;
- this.avail--;
- this._move(1);
- };
- WBuf.prototype.writeUInt16BE = function writeUInt16BE(v) {
- this._ensure(2);
- // Fast case - everything fits into the last buffer
- if (this.avail >= 2) {
- this.last.writeUInt16BE(v, this.offset);
- this.offset += 2;
- this.avail -= 2;
- // One byte here, one byte there
- } else {
- this.last[this.offset] = (v >>> 8);
- this._next();
- this.last[this.offset++] = v & 0xff;
- this.avail--;
- }
- this._move(2);
- };
- WBuf.prototype.writeUInt24BE = function writeUInt24BE(v) {
- this._ensure(3);
- // Fast case - everything fits into the last buffer
- if (this.avail >= 3) {
- this.last.writeUInt16BE(v >>> 8, this.offset);
- this.last[this.offset + 2] = v & 0xff;
- this.offset += 3;
- this.avail -= 3;
- this._move(3);
- // Two bytes here
- } else if (this.avail >= 2) {
- this.last.writeUInt16BE(v >>> 8, this.offset);
- this._next();
- this.last[this.offset++] = v & 0xff;
- this.avail--;
- this._move(3);
- // Just one byte here
- } else {
- this.last[this.offset] = v >>> 16;
- this._move(1);
- this._next();
- this.writeUInt16BE(v & 0xffff);
- }
- };
- WBuf.prototype.writeUInt32BE = function writeUInt32BE(v) {
- this._ensure(4);
- // Fast case - everything fits into the last buffer
- if (this.avail >= 4) {
- this.last.writeUInt32BE(v, this.offset);
- this.offset += 4;
- this.avail -= 4;
- this._move(4);
- // Three bytes here
- } else if (this.avail >= 3) {
- this.writeUInt24BE(v >>> 8);
- this._next();
- this.last[this.offset++] = v & 0xff;
- this.avail--;
- this._move(1);
- // Slow case, who cares
- } else {
- this.writeUInt16BE(v >>> 16);
- this.writeUInt16BE(v & 0xffff);
- }
- };
- WBuf.prototype.writeUInt16LE = function writeUInt16LE(num) {
- var r = ((num & 0xff) << 8) | (num >>> 8);
- this.writeUInt16BE(r);
- };
- WBuf.prototype.writeUInt24LE = function writeUInt24LE(num) {
- var r = ((num & 0xff) << 16) | (((num >>> 8) & 0xff) << 8) | (num >>> 16);
- this.writeUInt24BE(r);
- };
- WBuf.prototype.writeUInt32LE = function writeUInt32LE(num) {
- this._ensure(4);
- // Fast case - everything fits into the last buffer
- if (this.avail >= 4) {
- this.last.writeUInt32LE(num, this.offset);
- this.offset += 4;
- this.avail -= 4;
- this._move(4);
- // Three bytes here
- } else if (this.avail >= 3) {
- this.writeUInt24LE(num & 0xffffff);
- this._next();
- this.last[this.offset++] = num >>> 24;
- this.avail--;
- this._move(1);
- // Slow case, who cares
- } else {
- this.writeUInt16LE(num & 0xffff);
- this.writeUInt16LE(num >>> 16);
- }
- };
- WBuf.prototype.render = function render() {
- var left = this.size;
- var out = [];
- for (var i = 0; i < this.buffers.length && left >= 0; i++) {
- var buf = this.buffers[i];
- left -= buf.length;
- if (left >= 0) {
- out.push(buf);
- } else {
- out.push(buf.slice(0, buf.length + left));
- }
- }
- return out;
- };
- // Signed APIs
- WBuf.prototype.writeInt8 = function writeInt8(num) {
- if (num < 0)
- return this.writeUInt8(0x100 + num);
- else
- return this.writeUInt8(num);
- };
- function toUnsigned16(num) {
- if (num < 0)
- return 0x10000 + num;
- else
- return num;
- }
- WBuf.prototype.writeInt16LE = function writeInt16LE(num) {
- this.writeUInt16LE(toUnsigned16(num));
- };
- WBuf.prototype.writeInt16BE = function writeInt16BE(num) {
- this.writeUInt16BE(toUnsigned16(num));
- };
- function toUnsigned24(num) {
- if (num < 0)
- return 0x1000000 + num;
- else
- return num;
- }
- WBuf.prototype.writeInt24LE = function writeInt24LE(num) {
- this.writeUInt24LE(toUnsigned24(num));
- };
- WBuf.prototype.writeInt24BE = function writeInt24BE(num) {
- this.writeUInt24BE(toUnsigned24(num));
- };
- function toUnsigned32(num) {
- if (num < 0)
- return (0xffffffff + num) + 1;
- else
- return num;
- }
- WBuf.prototype.writeInt32LE = function writeInt32LE(num) {
- this.writeUInt32LE(toUnsigned32(num));
- };
- WBuf.prototype.writeInt32BE = function writeInt32BE(num) {
- this.writeUInt32BE(toUnsigned32(num));
- };
- WBuf.prototype.writeComb = function writeComb(size, endian, value) {
- if (size === 1)
- return this.writeUInt8(value);
- if (endian === 'le') {
- if (size === 2)
- this.writeUInt16LE(value);
- else if (size === 3)
- this.writeUInt24LE(value);
- else if (size === 4)
- this.writeUInt32LE(value);
- } else {
- if (size === 2)
- this.writeUInt16BE(value);
- else if (size === 3)
- this.writeUInt24BE(value);
- else if (size === 4)
- this.writeUInt32BE(value);
- }
- };
|