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.
316 lines
7.5 KiB
316 lines
7.5 KiB
// Copyright 2012 The Obvious Corporation. |
|
|
|
/* |
|
* leb: LEB128 utilities. |
|
*/ |
|
|
|
/* |
|
* Modules used |
|
*/ |
|
"use strict"; |
|
|
|
import Long from "@xtuc/long"; |
|
import * as bits from "./bits"; |
|
import * as bufs from "./bufs"; |
|
/* |
|
* Module variables |
|
*/ |
|
|
|
/** The minimum possible 32-bit signed int. */ |
|
|
|
var MIN_INT32 = -0x80000000; |
|
/** The maximum possible 32-bit signed int. */ |
|
|
|
var MAX_INT32 = 0x7fffffff; |
|
/** The maximum possible 32-bit unsigned int. */ |
|
|
|
var MAX_UINT32 = 0xffffffff; |
|
/** The minimum possible 64-bit signed int. */ |
|
// const MIN_INT64 = -0x8000000000000000; |
|
|
|
/** |
|
* The maximum possible 64-bit signed int that is representable as a |
|
* JavaScript number. |
|
*/ |
|
// const MAX_INT64 = 0x7ffffffffffffc00; |
|
|
|
/** |
|
* The maximum possible 64-bit unsigned int that is representable as a |
|
* JavaScript number. |
|
*/ |
|
// const MAX_UINT64 = 0xfffffffffffff800; |
|
|
|
/* |
|
* Helper functions |
|
*/ |
|
|
|
/** |
|
* Determines the number of bits required to encode the number |
|
* represented in the given buffer as a signed value. The buffer is |
|
* taken to represent a signed number in little-endian form. |
|
* |
|
* The number of bits to encode is the (zero-based) bit number of the |
|
* highest-order non-sign-matching bit, plus two. For example: |
|
* |
|
* 11111011 01110101 |
|
* high low |
|
* |
|
* The sign bit here is 1 (that is, it's a negative number). The highest |
|
* bit number that doesn't match the sign is bit #10 (where the lowest-order |
|
* bit is bit #0). So, we have to encode at least 12 bits total. |
|
* |
|
* As a special degenerate case, the numbers 0 and -1 each require just one bit. |
|
*/ |
|
|
|
function signedBitCount(buffer) { |
|
return bits.highOrder(bits.getSign(buffer) ^ 1, buffer) + 2; |
|
} |
|
/** |
|
* Determines the number of bits required to encode the number |
|
* represented in the given buffer as an unsigned value. The buffer is |
|
* taken to represent an unsigned number in little-endian form. |
|
* |
|
* The number of bits to encode is the (zero-based) bit number of the |
|
* highest-order 1 bit, plus one. For example: |
|
* |
|
* 00011000 01010011 |
|
* high low |
|
* |
|
* The highest-order 1 bit here is bit #12 (where the lowest-order bit |
|
* is bit #0). So, we have to encode at least 13 bits total. |
|
* |
|
* As a special degenerate case, the number 0 requires 1 bit. |
|
*/ |
|
|
|
|
|
function unsignedBitCount(buffer) { |
|
var result = bits.highOrder(1, buffer) + 1; |
|
return result ? result : 1; |
|
} |
|
/** |
|
* Common encoder for both signed and unsigned ints. This takes a |
|
* bigint-ish buffer, returning an LEB128-encoded buffer. |
|
*/ |
|
|
|
|
|
function encodeBufferCommon(buffer, signed) { |
|
var signBit; |
|
var bitCount; |
|
|
|
if (signed) { |
|
signBit = bits.getSign(buffer); |
|
bitCount = signedBitCount(buffer); |
|
} else { |
|
signBit = 0; |
|
bitCount = unsignedBitCount(buffer); |
|
} |
|
|
|
var byteCount = Math.ceil(bitCount / 7); |
|
var result = bufs.alloc(byteCount); |
|
|
|
for (var i = 0; i < byteCount; i++) { |
|
var payload = bits.extract(buffer, i * 7, 7, signBit); |
|
result[i] = payload | 0x80; |
|
} // Mask off the top bit of the last byte, to indicate the end of the |
|
// encoding. |
|
|
|
|
|
result[byteCount - 1] &= 0x7f; |
|
return result; |
|
} |
|
/** |
|
* Gets the byte-length of the value encoded in the given buffer at |
|
* the given index. |
|
*/ |
|
|
|
|
|
function encodedLength(encodedBuffer, index) { |
|
var result = 0; |
|
|
|
while (encodedBuffer[index + result] >= 0x80) { |
|
result++; |
|
} |
|
|
|
result++; // to account for the last byte |
|
|
|
if (index + result > encodedBuffer.length) {// FIXME(sven): seems to cause false positives |
|
// throw new Error("integer representation too long"); |
|
} |
|
|
|
return result; |
|
} |
|
/** |
|
* Common decoder for both signed and unsigned ints. This takes an |
|
* LEB128-encoded buffer, returning a bigint-ish buffer. |
|
*/ |
|
|
|
|
|
function decodeBufferCommon(encodedBuffer, index, signed) { |
|
index = index === undefined ? 0 : index; |
|
var length = encodedLength(encodedBuffer, index); |
|
var bitLength = length * 7; |
|
var byteLength = Math.ceil(bitLength / 8); |
|
var result = bufs.alloc(byteLength); |
|
var outIndex = 0; |
|
|
|
while (length > 0) { |
|
bits.inject(result, outIndex, 7, encodedBuffer[index]); |
|
outIndex += 7; |
|
index++; |
|
length--; |
|
} |
|
|
|
var signBit; |
|
var signByte; |
|
|
|
if (signed) { |
|
// Sign-extend the last byte. |
|
var lastByte = result[byteLength - 1]; |
|
var endBit = outIndex % 8; |
|
|
|
if (endBit !== 0) { |
|
var shift = 32 - endBit; // 32 because JS bit ops work on 32-bit ints. |
|
|
|
lastByte = result[byteLength - 1] = lastByte << shift >> shift & 0xff; |
|
} |
|
|
|
signBit = lastByte >> 7; |
|
signByte = signBit * 0xff; |
|
} else { |
|
signBit = 0; |
|
signByte = 0; |
|
} // Slice off any superfluous bytes, that is, ones that add no meaningful |
|
// bits (because the value would be the same if they were removed). |
|
|
|
|
|
while (byteLength > 1 && result[byteLength - 1] === signByte && (!signed || result[byteLength - 2] >> 7 === signBit)) { |
|
byteLength--; |
|
} |
|
|
|
result = bufs.resize(result, byteLength); |
|
return { |
|
value: result, |
|
nextIndex: index |
|
}; |
|
} |
|
/* |
|
* Exported bindings |
|
*/ |
|
|
|
|
|
function encodeIntBuffer(buffer) { |
|
return encodeBufferCommon(buffer, true); |
|
} |
|
|
|
function decodeIntBuffer(encodedBuffer, index) { |
|
return decodeBufferCommon(encodedBuffer, index, true); |
|
} |
|
|
|
function encodeInt32(num) { |
|
var buf = bufs.alloc(4); |
|
buf.writeInt32LE(num, 0); |
|
var result = encodeIntBuffer(buf); |
|
bufs.free(buf); |
|
return result; |
|
} |
|
|
|
function decodeInt32(encodedBuffer, index) { |
|
var result = decodeIntBuffer(encodedBuffer, index); |
|
var parsed = bufs.readInt(result.value); |
|
var value = parsed.value; |
|
bufs.free(result.value); |
|
|
|
if (value < MIN_INT32 || value > MAX_INT32) { |
|
throw new Error("integer too large"); |
|
} |
|
|
|
return { |
|
value: value, |
|
nextIndex: result.nextIndex |
|
}; |
|
} |
|
|
|
function encodeInt64(num) { |
|
var buf = bufs.alloc(8); |
|
bufs.writeInt64(num, buf); |
|
var result = encodeIntBuffer(buf); |
|
bufs.free(buf); |
|
return result; |
|
} |
|
|
|
function decodeInt64(encodedBuffer, index) { |
|
var result = decodeIntBuffer(encodedBuffer, index); |
|
var value = Long.fromBytesLE(result.value, false); |
|
bufs.free(result.value); |
|
return { |
|
value: value, |
|
nextIndex: result.nextIndex, |
|
lossy: false |
|
}; |
|
} |
|
|
|
function encodeUIntBuffer(buffer) { |
|
return encodeBufferCommon(buffer, false); |
|
} |
|
|
|
function decodeUIntBuffer(encodedBuffer, index) { |
|
return decodeBufferCommon(encodedBuffer, index, false); |
|
} |
|
|
|
function encodeUInt32(num) { |
|
var buf = bufs.alloc(4); |
|
buf.writeUInt32LE(num, 0); |
|
var result = encodeUIntBuffer(buf); |
|
bufs.free(buf); |
|
return result; |
|
} |
|
|
|
function decodeUInt32(encodedBuffer, index) { |
|
var result = decodeUIntBuffer(encodedBuffer, index); |
|
var parsed = bufs.readUInt(result.value); |
|
var value = parsed.value; |
|
bufs.free(result.value); |
|
|
|
if (value > MAX_UINT32) { |
|
throw new Error("integer too large"); |
|
} |
|
|
|
return { |
|
value: value, |
|
nextIndex: result.nextIndex |
|
}; |
|
} |
|
|
|
function encodeUInt64(num) { |
|
var buf = bufs.alloc(8); |
|
bufs.writeUInt64(num, buf); |
|
var result = encodeUIntBuffer(buf); |
|
bufs.free(buf); |
|
return result; |
|
} |
|
|
|
function decodeUInt64(encodedBuffer, index) { |
|
var result = decodeUIntBuffer(encodedBuffer, index); |
|
var value = Long.fromBytesLE(result.value, true); |
|
bufs.free(result.value); |
|
return { |
|
value: value, |
|
nextIndex: result.nextIndex, |
|
lossy: false |
|
}; |
|
} |
|
|
|
export default { |
|
decodeInt32: decodeInt32, |
|
decodeInt64: decodeInt64, |
|
decodeIntBuffer: decodeIntBuffer, |
|
decodeUInt32: decodeUInt32, |
|
decodeUInt64: decodeUInt64, |
|
decodeUIntBuffer: decodeUIntBuffer, |
|
encodeInt32: encodeInt32, |
|
encodeInt64: encodeInt64, |
|
encodeIntBuffer: encodeIntBuffer, |
|
encodeUInt32: encodeUInt32, |
|
encodeUInt64: encodeUInt64, |
|
encodeUIntBuffer: encodeUIntBuffer |
|
}; |