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