You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					393 lines
				
				9.5 KiB
			
		
		
			
		
	
	
					393 lines
				
				9.5 KiB
			| 
											3 years ago
										 | 'use strict' | ||
|  | var DuplexStream = require('readable-stream').Duplex | ||
|  |   , util         = require('util') | ||
|  |   , Buffer       = require('safe-buffer').Buffer | ||
|  | 
 | ||
|  | function BufferList (callback) { | ||
|  |   if (!(this instanceof BufferList)) | ||
|  |     return new BufferList(callback) | ||
|  | 
 | ||
|  |   this._bufs  = [] | ||
|  |   this.length = 0 | ||
|  | 
 | ||
|  |   if (typeof callback == 'function') { | ||
|  |     this._callback = callback | ||
|  | 
 | ||
|  |     var piper = function piper (err) { | ||
|  |       if (this._callback) { | ||
|  |         this._callback(err) | ||
|  |         this._callback = null | ||
|  |       } | ||
|  |     }.bind(this) | ||
|  | 
 | ||
|  |     this.on('pipe', function onPipe (src) { | ||
|  |       src.on('error', piper) | ||
|  |     }) | ||
|  |     this.on('unpipe', function onUnpipe (src) { | ||
|  |       src.removeListener('error', piper) | ||
|  |     }) | ||
|  |   } else { | ||
|  |     this.append(callback) | ||
|  |   } | ||
|  | 
 | ||
|  |   DuplexStream.call(this) | ||
|  | } | ||
|  | 
 | ||
|  | 
 | ||
|  | util.inherits(BufferList, DuplexStream) | ||
|  | 
 | ||
|  | 
 | ||
|  | BufferList.prototype._offset = function _offset (offset) { | ||
|  |   var tot = 0, i = 0, _t | ||
|  |   if (offset === 0) return [ 0, 0 ] | ||
|  |   for (; i < this._bufs.length; i++) { | ||
|  |     _t = tot + this._bufs[i].length | ||
|  |     if (offset < _t || i == this._bufs.length - 1) { | ||
|  |       return [ i, offset - tot ] | ||
|  |     } | ||
|  |     tot = _t | ||
|  |   } | ||
|  | } | ||
|  | 
 | ||
|  | BufferList.prototype._reverseOffset = function (blOffset) { | ||
|  |   var bufferId = blOffset[0] | ||
|  |   var offset = blOffset[1] | ||
|  |   for (var i = 0; i < bufferId; i++) { | ||
|  |     offset += this._bufs[i].length | ||
|  |   } | ||
|  |   return offset | ||
|  | } | ||
|  | 
 | ||
|  | BufferList.prototype.append = function append (buf) { | ||
|  |   var i = 0 | ||
|  | 
 | ||
|  |   if (Buffer.isBuffer(buf)) { | ||
|  |     this._appendBuffer(buf) | ||
|  |   } else if (Array.isArray(buf)) { | ||
|  |     for (; i < buf.length; i++) | ||
|  |       this.append(buf[i]) | ||
|  |   } else if (buf instanceof BufferList) { | ||
|  |     // unwrap argument into individual BufferLists
 | ||
|  |     for (; i < buf._bufs.length; i++) | ||
|  |       this.append(buf._bufs[i]) | ||
|  |   } else if (buf != null) { | ||
|  |     // coerce number arguments to strings, since Buffer(number) does
 | ||
|  |     // uninitialized memory allocation
 | ||
|  |     if (typeof buf == 'number') | ||
|  |       buf = buf.toString() | ||
|  | 
 | ||
|  |     this._appendBuffer(Buffer.from(buf)) | ||
|  |   } | ||
|  | 
 | ||
|  |   return this | ||
|  | } | ||
|  | 
 | ||
|  | 
 | ||
|  | BufferList.prototype._appendBuffer = function appendBuffer (buf) { | ||
|  |   this._bufs.push(buf) | ||
|  |   this.length += buf.length | ||
|  | } | ||
|  | 
 | ||
|  | 
 | ||
|  | BufferList.prototype._write = function _write (buf, encoding, callback) { | ||
|  |   this._appendBuffer(buf) | ||
|  | 
 | ||
|  |   if (typeof callback == 'function') | ||
|  |     callback() | ||
|  | } | ||
|  | 
 | ||
|  | 
 | ||
|  | BufferList.prototype._read = function _read (size) { | ||
|  |   if (!this.length) | ||
|  |     return this.push(null) | ||
|  | 
 | ||
|  |   size = Math.min(size, this.length) | ||
|  |   this.push(this.slice(0, size)) | ||
|  |   this.consume(size) | ||
|  | } | ||
|  | 
 | ||
|  | 
 | ||
|  | BufferList.prototype.end = function end (chunk) { | ||
|  |   DuplexStream.prototype.end.call(this, chunk) | ||
|  | 
 | ||
|  |   if (this._callback) { | ||
|  |     this._callback(null, this.slice()) | ||
|  |     this._callback = null | ||
|  |   } | ||
|  | } | ||
|  | 
 | ||
|  | 
 | ||
|  | BufferList.prototype.get = function get (index) { | ||
|  |   if (index > this.length || index < 0) { | ||
|  |     return undefined | ||
|  |   } | ||
|  |   var offset = this._offset(index) | ||
|  |   return this._bufs[offset[0]][offset[1]] | ||
|  | } | ||
|  | 
 | ||
|  | 
 | ||
|  | BufferList.prototype.slice = function slice (start, end) { | ||
|  |   if (typeof start == 'number' && start < 0) | ||
|  |     start += this.length | ||
|  |   if (typeof end == 'number' && end < 0) | ||
|  |     end += this.length | ||
|  |   return this.copy(null, 0, start, end) | ||
|  | } | ||
|  | 
 | ||
|  | 
 | ||
|  | BufferList.prototype.copy = function copy (dst, dstStart, srcStart, srcEnd) { | ||
|  |   if (typeof srcStart != 'number' || srcStart < 0) | ||
|  |     srcStart = 0 | ||
|  |   if (typeof srcEnd != 'number' || srcEnd > this.length) | ||
|  |     srcEnd = this.length | ||
|  |   if (srcStart >= this.length) | ||
|  |     return dst || Buffer.alloc(0) | ||
|  |   if (srcEnd <= 0) | ||
|  |     return dst || Buffer.alloc(0) | ||
|  | 
 | ||
|  |   var copy   = !!dst | ||
|  |     , off    = this._offset(srcStart) | ||
|  |     , len    = srcEnd - srcStart | ||
|  |     , bytes  = len | ||
|  |     , bufoff = (copy && dstStart) || 0 | ||
|  |     , start  = off[1] | ||
|  |     , l | ||
|  |     , i | ||
|  | 
 | ||
|  |   // copy/slice everything
 | ||
|  |   if (srcStart === 0 && srcEnd == this.length) { | ||
|  |     if (!copy) { // slice, but full concat if multiple buffers
 | ||
|  |       return this._bufs.length === 1 | ||
|  |         ? this._bufs[0] | ||
|  |         : Buffer.concat(this._bufs, this.length) | ||
|  |     } | ||
|  | 
 | ||
|  |     // copy, need to copy individual buffers
 | ||
|  |     for (i = 0; i < this._bufs.length; i++) { | ||
|  |       this._bufs[i].copy(dst, bufoff) | ||
|  |       bufoff += this._bufs[i].length | ||
|  |     } | ||
|  | 
 | ||
|  |     return dst | ||
|  |   } | ||
|  | 
 | ||
|  |   // easy, cheap case where it's a subset of one of the buffers
 | ||
|  |   if (bytes <= this._bufs[off[0]].length - start) { | ||
|  |     return copy | ||
|  |       ? this._bufs[off[0]].copy(dst, dstStart, start, start + bytes) | ||
|  |       : this._bufs[off[0]].slice(start, start + bytes) | ||
|  |   } | ||
|  | 
 | ||
|  |   if (!copy) // a slice, we need something to copy in to
 | ||
|  |     dst = Buffer.allocUnsafe(len) | ||
|  | 
 | ||
|  |   for (i = off[0]; i < this._bufs.length; i++) { | ||
|  |     l = this._bufs[i].length - start | ||
|  | 
 | ||
|  |     if (bytes > l) { | ||
|  |       this._bufs[i].copy(dst, bufoff, start) | ||
|  |       bufoff += l | ||
|  |     } else { | ||
|  |       this._bufs[i].copy(dst, bufoff, start, start + bytes) | ||
|  |       bufoff += l | ||
|  |       break | ||
|  |     } | ||
|  | 
 | ||
|  |     bytes -= l | ||
|  | 
 | ||
|  |     if (start) | ||
|  |       start = 0 | ||
|  |   } | ||
|  | 
 | ||
|  |   // safeguard so that we don't return uninitialized memory
 | ||
|  |   if (dst.length > bufoff) return dst.slice(0, bufoff) | ||
|  | 
 | ||
|  |   return dst | ||
|  | } | ||
|  | 
 | ||
|  | BufferList.prototype.shallowSlice = function shallowSlice (start, end) { | ||
|  |   start = start || 0 | ||
|  |   end = typeof end !== 'number' ? this.length : end | ||
|  | 
 | ||
|  |   if (start < 0) | ||
|  |     start += this.length | ||
|  |   if (end < 0) | ||
|  |     end += this.length | ||
|  | 
 | ||
|  |   if (start === end) { | ||
|  |     return new BufferList() | ||
|  |   } | ||
|  |   var startOffset = this._offset(start) | ||
|  |     , endOffset = this._offset(end) | ||
|  |     , buffers = this._bufs.slice(startOffset[0], endOffset[0] + 1) | ||
|  | 
 | ||
|  |   if (endOffset[1] == 0) | ||
|  |     buffers.pop() | ||
|  |   else | ||
|  |     buffers[buffers.length-1] = buffers[buffers.length-1].slice(0, endOffset[1]) | ||
|  | 
 | ||
|  |   if (startOffset[1] != 0) | ||
|  |     buffers[0] = buffers[0].slice(startOffset[1]) | ||
|  | 
 | ||
|  |   return new BufferList(buffers) | ||
|  | } | ||
|  | 
 | ||
|  | BufferList.prototype.toString = function toString (encoding, start, end) { | ||
|  |   return this.slice(start, end).toString(encoding) | ||
|  | } | ||
|  | 
 | ||
|  | BufferList.prototype.consume = function consume (bytes) { | ||
|  |   // first, normalize the argument, in accordance with how Buffer does it
 | ||
|  |   bytes = Math.trunc(bytes) | ||
|  |   // do nothing if not a positive number
 | ||
|  |   if (Number.isNaN(bytes) || bytes <= 0) return this | ||
|  | 
 | ||
|  |   while (this._bufs.length) { | ||
|  |     if (bytes >= this._bufs[0].length) { | ||
|  |       bytes -= this._bufs[0].length | ||
|  |       this.length -= this._bufs[0].length | ||
|  |       this._bufs.shift() | ||
|  |     } else { | ||
|  |       this._bufs[0] = this._bufs[0].slice(bytes) | ||
|  |       this.length -= bytes | ||
|  |       break | ||
|  |     } | ||
|  |   } | ||
|  |   return this | ||
|  | } | ||
|  | 
 | ||
|  | 
 | ||
|  | BufferList.prototype.duplicate = function duplicate () { | ||
|  |   var i = 0 | ||
|  |     , copy = new BufferList() | ||
|  | 
 | ||
|  |   for (; i < this._bufs.length; i++) | ||
|  |     copy.append(this._bufs[i]) | ||
|  | 
 | ||
|  |   return copy | ||
|  | } | ||
|  | 
 | ||
|  | 
 | ||
|  | BufferList.prototype.destroy = function destroy () { | ||
|  |   this._bufs.length = 0 | ||
|  |   this.length = 0 | ||
|  |   this.push(null) | ||
|  | } | ||
|  | 
 | ||
|  | 
 | ||
|  | BufferList.prototype.indexOf = function (search, offset, encoding) { | ||
|  |   if (encoding === undefined && typeof offset === 'string') { | ||
|  |     encoding = offset | ||
|  |     offset = undefined | ||
|  |   } | ||
|  |   if (typeof search === 'function' || Array.isArray(search)) { | ||
|  |     throw new TypeError('The "value" argument must be one of type string, Buffer, BufferList, or Uint8Array.') | ||
|  |   } else if (typeof search === 'number') { | ||
|  |       search = Buffer.from([search]) | ||
|  |   } else if (typeof search === 'string') { | ||
|  |     search = Buffer.from(search, encoding) | ||
|  |   } else if (search instanceof BufferList) { | ||
|  |     search = search.slice() | ||
|  |   } else if (!Buffer.isBuffer(search)) { | ||
|  |     search = Buffer.from(search) | ||
|  |   } | ||
|  | 
 | ||
|  |   offset = Number(offset || 0) | ||
|  |   if (isNaN(offset)) { | ||
|  |     offset = 0 | ||
|  |   } | ||
|  | 
 | ||
|  |   if (offset < 0) { | ||
|  |     offset = this.length + offset | ||
|  |   } | ||
|  | 
 | ||
|  |   if (offset < 0) { | ||
|  |     offset = 0 | ||
|  |   } | ||
|  | 
 | ||
|  |   if (search.length === 0) { | ||
|  |     return offset > this.length ? this.length : offset | ||
|  |   } | ||
|  | 
 | ||
|  |   var blOffset = this._offset(offset) | ||
|  |   var blIndex = blOffset[0] // index of which internal buffer we're working on
 | ||
|  |   var buffOffset = blOffset[1] // offset of the internal buffer we're working on
 | ||
|  | 
 | ||
|  |   // scan over each buffer
 | ||
|  |   for (blIndex; blIndex < this._bufs.length; blIndex++) { | ||
|  |     var buff = this._bufs[blIndex] | ||
|  |     while(buffOffset < buff.length) { | ||
|  |       var availableWindow = buff.length - buffOffset | ||
|  |       if (availableWindow >= search.length) { | ||
|  |         var nativeSearchResult = buff.indexOf(search, buffOffset) | ||
|  |         if (nativeSearchResult !== -1) { | ||
|  |           return this._reverseOffset([blIndex, nativeSearchResult]) | ||
|  |         } | ||
|  |         buffOffset = buff.length - search.length + 1 // end of native search window
 | ||
|  |       } else { | ||
|  |         var revOffset = this._reverseOffset([blIndex, buffOffset]) | ||
|  |         if (this._match(revOffset, search)) { | ||
|  |           return revOffset | ||
|  |         } | ||
|  |         buffOffset++ | ||
|  |       } | ||
|  |     } | ||
|  |     buffOffset = 0 | ||
|  |   } | ||
|  |   return -1 | ||
|  | } | ||
|  | 
 | ||
|  | BufferList.prototype._match = function(offset, search) { | ||
|  |   if (this.length - offset < search.length) { | ||
|  |     return false | ||
|  |   } | ||
|  |   for (var searchOffset = 0; searchOffset < search.length ; searchOffset++) { | ||
|  |     if(this.get(offset + searchOffset) !== search[searchOffset]){ | ||
|  |       return false | ||
|  |     } | ||
|  |   } | ||
|  |   return true | ||
|  | } | ||
|  | 
 | ||
|  | 
 | ||
|  | ;(function () { | ||
|  |   var methods = { | ||
|  |       'readDoubleBE' : 8 | ||
|  |     , 'readDoubleLE' : 8 | ||
|  |     , 'readFloatBE'  : 4 | ||
|  |     , 'readFloatLE'  : 4 | ||
|  |     , 'readInt32BE'  : 4 | ||
|  |     , 'readInt32LE'  : 4 | ||
|  |     , 'readUInt32BE' : 4 | ||
|  |     , 'readUInt32LE' : 4 | ||
|  |     , 'readInt16BE'  : 2 | ||
|  |     , 'readInt16LE'  : 2 | ||
|  |     , 'readUInt16BE' : 2 | ||
|  |     , 'readUInt16LE' : 2 | ||
|  |     , 'readInt8'     : 1 | ||
|  |     , 'readUInt8'    : 1 | ||
|  |     , 'readIntBE'    : null | ||
|  |     , 'readIntLE'    : null | ||
|  |     , 'readUIntBE'   : null | ||
|  |     , 'readUIntLE'   : null | ||
|  |   } | ||
|  | 
 | ||
|  |   for (var m in methods) { | ||
|  |     (function (m) { | ||
|  |       if (methods[m] === null) { | ||
|  |         BufferList.prototype[m] = function (offset, byteLength) { | ||
|  |           return this.slice(offset, offset + byteLength)[m](0, byteLength) | ||
|  |         } | ||
|  |       } | ||
|  |       else { | ||
|  |         BufferList.prototype[m] = function (offset) { | ||
|  |           return this.slice(offset, offset + methods[m])[m](0) | ||
|  |         } | ||
|  |       } | ||
|  |     }(m)) | ||
|  |   } | ||
|  | }()) | ||
|  | 
 | ||
|  | 
 | ||
|  | module.exports = BufferList |