bitcoinjs-lib/ts_src/script_number.ts

72 lines
1.6 KiB
TypeScript
Raw Normal View History

2019-03-03 15:07:49 +01:00
export function decode(
buffer: Buffer,
maxLength?: number,
minimal?: boolean,
): number {
maxLength = maxLength || 4;
minimal = minimal === undefined ? true : minimal;
2019-03-03 15:07:49 +01:00
const length = buffer.length;
if (length === 0) return 0;
if (length > maxLength) throw new TypeError('Script number overflow');
2016-01-04 15:59:14 +01:00
if (minimal) {
if ((buffer[length - 1] & 0x7f) === 0) {
2019-03-03 15:07:49 +01:00
if (length <= 1 || (buffer[length - 2] & 0x80) === 0)
throw new Error('Non-minimally encoded script number');
2016-01-04 15:59:14 +01:00
}
}
2016-01-04 16:18:50 +01:00
// 40-bit
if (length === 5) {
2019-03-03 15:07:49 +01:00
const a = buffer.readUInt32LE(0);
const b = buffer.readUInt8(4);
2016-01-04 15:59:14 +01:00
2019-03-03 15:07:49 +01:00
if (b & 0x80) return -((b & ~0x80) * 0x100000000 + a);
return b * 0x100000000 + a;
}
2016-01-04 15:59:14 +01:00
// 32-bit / 24-bit / 16-bit / 8-bit
2019-03-03 15:07:49 +01:00
let result = 0;
for (var i = 0; i < length; ++i) {
2019-03-03 15:07:49 +01:00
result |= buffer[i] << (8 * i);
2016-01-04 15:59:14 +01:00
}
2019-03-03 15:07:49 +01:00
if (buffer[length - 1] & 0x80)
return -(result & ~(0x80 << (8 * (length - 1))));
return result;
2016-01-04 15:59:14 +01:00
}
2019-03-03 15:07:49 +01:00
function scriptNumSize(i: number): number {
return i > 0x7fffffff
? 5
: i > 0x7fffff
? 4
: i > 0x7fff
? 3
: i > 0x7f
? 2
: i > 0x00
? 1
: 0;
2016-01-04 15:59:14 +01:00
}
2019-03-03 15:07:49 +01:00
export function encode(number: number): Buffer {
let value = Math.abs(number);
const size = scriptNumSize(value);
const buffer = Buffer.allocUnsafe(size);
const negative = number < 0;
2016-01-04 15:59:14 +01:00
for (var i = 0; i < size; ++i) {
2019-03-03 15:07:49 +01:00
buffer.writeUInt8(value & 0xff, i);
value >>= 8;
2016-01-04 15:59:14 +01:00
}
if (buffer[size - 1] & 0x80) {
2019-03-03 15:07:49 +01:00
buffer.writeUInt8(negative ? 0x80 : 0x00, size - 1);
2016-01-04 15:59:14 +01:00
} else if (negative) {
2019-03-03 15:07:49 +01:00
buffer[size - 1] |= 0x80;
2016-01-04 15:59:14 +01:00
}
2019-03-03 15:07:49 +01:00
return buffer;
2016-01-04 15:59:14 +01:00
}