39 lines
1.4 KiB
JavaScript
39 lines
1.4 KiB
JavaScript
|
"use strict";
|
||
|
|
import { BigNumber } from "@ethersproject/bignumber";
|
||
|
|
import { MaxUint256, NegativeOne, One, Zero } from "@ethersproject/constants";
|
||
|
|
import { Coder } from "./abstract-coder";
|
||
|
|
export class NumberCoder extends Coder {
|
||
|
|
constructor(size, signed, localName) {
|
||
|
|
const name = ((signed ? "int" : "uint") + (size * 8));
|
||
|
|
super(name, name, localName, false);
|
||
|
|
this.size = size;
|
||
|
|
this.signed = signed;
|
||
|
|
}
|
||
|
|
encode(writer, value) {
|
||
|
|
let v = BigNumber.from(value);
|
||
|
|
// Check bounds are safe for encoding
|
||
|
|
let maxUintValue = MaxUint256.maskn(writer.wordSize * 8);
|
||
|
|
if (this.signed) {
|
||
|
|
let bounds = maxUintValue.maskn(this.size * 8 - 1);
|
||
|
|
if (v.gt(bounds) || v.lt(bounds.add(One).mul(NegativeOne))) {
|
||
|
|
this._throwError("value out-of-bounds", value);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else if (v.lt(Zero) || v.gt(maxUintValue.maskn(this.size * 8))) {
|
||
|
|
this._throwError("value out-of-bounds", value);
|
||
|
|
}
|
||
|
|
v = v.toTwos(this.size * 8).maskn(this.size * 8);
|
||
|
|
if (this.signed) {
|
||
|
|
v = v.fromTwos(this.size * 8).toTwos(8 * writer.wordSize);
|
||
|
|
}
|
||
|
|
return writer.writeValue(v);
|
||
|
|
}
|
||
|
|
decode(reader) {
|
||
|
|
let value = reader.readValue().maskn(this.size * 8);
|
||
|
|
if (this.signed) {
|
||
|
|
value = value.fromTwos(this.size * 8);
|
||
|
|
}
|
||
|
|
return reader.coerce(this.name, value);
|
||
|
|
}
|
||
|
|
}
|