ethers.js/src.ts/utils/abi-coder.ts

1089 lines
33 KiB
TypeScript
Raw Normal View History

2018-06-13 22:39:39 +03:00
'use strict';
// See: https://github.com/ethereum/wiki/wiki/Ethereum-Contract-ABI
import { NegativeOne, Zero, One, MaxUint256 } from '../constants';
import * as errors from '../errors';
2018-06-13 22:39:39 +03:00
import { getAddress } from './address';
import { BigNumber, bigNumberify } from './bignumber';
import { arrayify, concat, hexlify, padZeros } from './bytes';
2018-06-13 22:39:39 +03:00
import { toUtf8Bytes, toUtf8String } from './utf8';
import { deepCopy, defineReadOnly, shallowCopy } from './properties';
2018-06-13 22:39:39 +03:00
///////////////////////////////
// Imported Types
import { Arrayish } from './bytes';
import { BigNumberish } from './bignumber';
///////////////////////////////
// Exported Types
export type CoerceFunc = (type: string, value: any) => any;
export type ParamType = {
name?: string,
type: string,
indexed?: boolean,
components?: Array<any>
};
// @TODO: should this just be a combined Fragment?
export type EventFragment = {
type: string
name: string,
anonymous: boolean,
inputs: Array<ParamType>,
};
export type FunctionFragment = {
type: string
name: string,
constant: boolean,
inputs: Array<ParamType>,
outputs: Array<ParamType>,
payable: boolean,
stateMutability: string,
2018-08-02 00:24:11 +03:00
gas: BigNumber
};
///////////////////////////////
2018-06-13 22:39:39 +03:00
const paramTypeBytes = new RegExp(/^bytes([0-9]*)$/);
const paramTypeNumber = new RegExp(/^(u?int)([0-9]*)$/);
const paramTypeArray = new RegExp(/^(.*)\[([0-9]*)\]$/);
2018-06-15 11:18:17 +03:00
export const defaultCoerceFunc: CoerceFunc = function(type: string, value: any): any {
2018-06-13 22:39:39 +03:00
var match = type.match(paramTypeNumber)
if (match && parseInt(match[2]) <= 48) { return value.toNumber(); }
return value;
}
///////////////////////////////////
// Parsing for Solidity Signatures
const regexParen = new RegExp("^([^)(]*)\\((.*)\\)([^)(]*)$");
const regexIdentifier = new RegExp("^[A-Za-z_][A-Za-z0-9_]*$");
function verifyType(type: string): string {
// These need to be transformed to their full description
if (type.match(/^uint($|[^1-9])/)) {
type = 'uint256' + type.substring(4);
} else if (type.match(/^int($|[^1-9])/)) {
type = 'int256' + type.substring(3);
}
return type;
}
type ParseState = {
allowArray?: boolean,
allowName?: boolean,
allowParams?: boolean,
allowType?: boolean,
readArray?: boolean,
};
type ParseNode = {
parent?: any,
type?: string,
name?: string,
state?: ParseState,
indexed?: boolean,
components?: Array<any>
};
function parseParam(param: string, allowIndexed?: boolean): ParamType {
let originalParam = param;
function throwError(i: number) {
throw new Error('unexpected character "' + originalParam[i] + '" at position ' + i + ' in "' + originalParam + '"');
2018-06-13 22:39:39 +03:00
}
param = param.replace(/\s/g, ' ');
2018-06-13 22:39:39 +03:00
var parent: ParseNode = { type: '', name: '', state: { allowType: true } };
var node = parent;
for (var i = 0; i < param.length; i++) {
var c = param[i];
switch (c) {
case '(':
if (!node.state.allowParams) { throwError(i); }
node.state.allowType = false;
node.type = verifyType(node.type);
node.components = [ { type: '', name: '', parent: node, state: { allowType: true } } ];
node = node.components[0];
break;
case ')':
delete node.state;
if (allowIndexed && node.name === 'indexed') {
node.indexed = true;
node.name = '';
}
2018-06-13 22:39:39 +03:00
node.type = verifyType(node.type);
var child = node;
node = node.parent;
if (!node) { throwError(i); }
delete child.parent;
node.state.allowParams = false;
node.state.allowName = true;
node.state.allowArray = true;
break;
case ',':
delete node.state;
if (allowIndexed && node.name === 'indexed') {
node.indexed = true;
node.name = '';
}
2018-06-13 22:39:39 +03:00
node.type = verifyType(node.type);
var sibling: ParseNode = { type: '', name: '', parent: node.parent, state: { allowType: true } };
node.parent.components.push(sibling);
delete node.parent;
node = sibling;
break;
// Hit a space...
case ' ':
// If reading type, the type is done and may read a param or name
if (node.state.allowType) {
if (node.type !== '') {
node.type = verifyType(node.type);
delete node.state.allowType;
node.state.allowName = true;
node.state.allowParams = true;
}
}
// If reading name, the name is done
if (node.state.allowName) {
if (node.name !== '') {
if (allowIndexed && node.name === 'indexed') {
node.indexed = true;
node.name = '';
} else {
node.state.allowName = false;
}
}
}
break;
case '[':
if (!node.state.allowArray) { throwError(i); }
node.type += c;
node.state.allowArray = false;
node.state.allowName = false;
node.state.readArray = true;
break;
case ']':
if (!node.state.readArray) { throwError(i); }
node.type += c;
node.state.readArray = false;
node.state.allowArray = true;
node.state.allowName = true;
break;
default:
if (node.state.allowType) {
node.type += c;
node.state.allowParams = true;
node.state.allowArray = true;
} else if (node.state.allowName) {
node.name += c;
delete node.state.allowArray;
} else if (node.state.readArray) {
node.type += c;
} else {
throwError(i);
}
}
}
if (node.parent) { throw new Error("unexpected eof"); }
delete parent.state;
if (allowIndexed && node.name === 'indexed') {
node.indexed = true;
node.name = '';
}
2018-06-13 22:39:39 +03:00
parent.type = verifyType(parent.type);
return (<ParamType>parent);
}
2018-06-15 11:18:17 +03:00
2018-06-13 22:39:39 +03:00
// @TODO: Better return type
2018-06-15 11:18:17 +03:00
function parseSignatureEvent(fragment: string): EventFragment {
2018-06-13 22:39:39 +03:00
var abi: EventFragment = {
2018-06-13 22:39:39 +03:00
anonymous: false,
inputs: [],
name: '',
type: 'event'
}
var match = fragment.match(regexParen);
if (!match) { throw new Error('invalid event: ' + fragment); }
abi.name = match[1].trim();
splitNesting(match[2]).forEach(function(param) {
param = parseParam(param, true);
param.indexed = !!param.indexed;
abi.inputs.push(param);
});
match[3].split(' ').forEach(function(modifier) {
switch(modifier) {
case 'anonymous':
abi.anonymous = true;
break;
case '':
break;
default:
errors.info('unknown modifier: ' + modifier);
2018-06-13 22:39:39 +03:00
}
});
if (abi.name && !abi.name.match(regexIdentifier)) {
throw new Error('invalid identifier: "' + abi.name + '"');
}
return abi;
}
2018-06-15 11:18:17 +03:00
function parseSignatureFunction(fragment: string): FunctionFragment {
var abi: FunctionFragment = {
2018-06-13 22:39:39 +03:00
constant: false,
2018-08-02 00:24:11 +03:00
gas: null,
2018-06-13 22:39:39 +03:00
inputs: [],
name: '',
outputs: [],
payable: false,
stateMutability: null, // @TODO: Should this be initialized to 'nonpayable'?
2018-06-13 22:39:39 +03:00
type: 'function'
};
2018-08-02 00:24:11 +03:00
let comps = fragment.split('@');
if (comps.length !== 1) {
if (comps.length > 2) {
throw new Error('invalid signature');
}
if (!comps[1].match(/^[0-9]+$/)) {
throw new Error('invalid signature gas');
}
abi.gas = bigNumberify(comps[1]);
fragment = comps[0];
}
comps = fragment.split(' returns ');
2018-06-13 22:39:39 +03:00
var left = comps[0].match(regexParen);
if (!left) { throw new Error('invalid signature'); }
abi.name = left[1].trim();
if (!abi.name.match(regexIdentifier)) {
throw new Error('invalid identifier: "' + left[1] + '"');
}
splitNesting(left[2]).forEach(function(param) {
abi.inputs.push(parseParam(param));
});
left[3].split(' ').forEach(function(modifier) {
switch (modifier) {
case 'constant':
abi.constant = true;
break;
case 'payable':
abi.payable = true;
abi.stateMutability = 'payable';
2018-06-13 22:39:39 +03:00
break;
case 'pure':
abi.constant = true;
abi.stateMutability = 'pure';
break;
case 'view':
abi.constant = true;
abi.stateMutability = 'view';
break;
case 'external':
case 'public':
2018-06-13 22:39:39 +03:00
case '':
break;
default:
errors.info('unknown modifier: ' + modifier);
2018-06-13 22:39:39 +03:00
}
});
// We have outputs
if (comps.length > 1) {
var right = comps[1].match(regexParen);
if (right[1].trim() != '' || right[3].trim() != '') {
throw new Error('unexpected tokens');
}
splitNesting(right[2]).forEach(function(param) {
abi.outputs.push(parseParam(param));
});
}
if (abi.name === 'constructor') {
abi.type = "constructor";
if (abi.outputs.length) {
throw new Error('constructor may not have outputs');
}
delete abi.name;
delete abi.outputs;
}
2018-06-13 22:39:39 +03:00
return abi;
}
export function parseParamType(type: string): ParamType {
return parseParam(type, true);
}
2018-06-13 22:39:39 +03:00
// @TODO: Allow a second boolean to expose names
export function formatParamType(paramType: ParamType): string {
return getParamCoder(defaultCoerceFunc, paramType).type;
}
// @TODO: Allow a second boolean to expose names and modifiers
export function formatSignature(fragment: EventFragment | FunctionFragment): string {
return fragment.name + '(' + fragment.inputs.map((i) => formatParamType(i)).join(',') + ')';
}
2018-06-15 11:18:17 +03:00
export function parseSignature(fragment: string): EventFragment | FunctionFragment {
2018-06-13 22:39:39 +03:00
if(typeof(fragment) === 'string') {
// Make sure the "returns" is surrounded by a space and all whitespace is exactly one space
fragment = fragment.replace(/\s/g, ' ');
2018-06-13 22:39:39 +03:00
fragment = fragment.replace(/\(/g, ' (').replace(/\)/g, ') ').replace(/\s+/g, ' ');
fragment = fragment.trim();
if (fragment.substring(0, 6) === 'event ') {
return parseSignatureEvent(fragment.substring(6).trim());
} else {
if (fragment.substring(0, 9) === 'function ') {
fragment = fragment.substring(9);
}
return parseSignatureFunction(fragment.trim());
}
}
2018-06-15 11:18:17 +03:00
throw new Error('unknown signature');
2018-06-13 22:39:39 +03:00
}
///////////////////////////////////
// Coders
type DecodedResult = { consumed: number, value: any };
abstract class Coder {
readonly coerceFunc: CoerceFunc;
readonly name: string;
readonly type: string;
readonly localName: string;
readonly dynamic: boolean;
constructor(coerceFunc: CoerceFunc, name: string, type: string, localName: string, dynamic: boolean) {
this.coerceFunc = coerceFunc;
this.name = name;
this.type = type;
this.localName = localName;
this.dynamic = dynamic;
}
abstract encode(value: any): Uint8Array;
abstract decode(data: Uint8Array, offset: number): DecodedResult;
}
// Clones the functionality of an existing Coder, but without a localName
class CoderAnonymous extends Coder {
private coder: Coder;
constructor(coder: Coder) {
super(coder.coerceFunc, coder.name, coder.type, undefined, coder.dynamic);
defineReadOnly(this, 'coder', coder);
}
encode(value: any): Uint8Array { return this.coder.encode(value); }
decode(data: Uint8Array, offset: number): DecodedResult { return this.coder.decode(data, offset); }
}
2018-06-13 22:39:39 +03:00
class CoderNull extends Coder {
constructor(coerceFunc: CoerceFunc, localName: string) {
2018-06-13 22:39:39 +03:00
super(coerceFunc, 'null', '', localName, false);
}
encode(value: any): Uint8Array {
return arrayify([]);
}
decode(data: Uint8Array, offset: number): DecodedResult {
if (offset > data.length) { throw new Error('invalid null'); }
return {
consumed: 0,
value: this.coerceFunc('null', undefined)
}
}
}
class CoderNumber extends Coder {
readonly size: number;
readonly signed: boolean;
constructor(coerceFunc: CoerceFunc, size: number, signed: boolean, localName: string) {
2018-06-13 22:39:39 +03:00
const name = ((signed ? 'int': 'uint') + (size * 8));
super(coerceFunc, name, name, localName, false);
this.size = size;
this.signed = signed;
}
encode(value: BigNumberish): Uint8Array {
try {
let v = bigNumberify(value);
if (this.signed) {
let bounds = MaxUint256.maskn(this.size * 8 - 1);
if (v.gt(bounds)) { throw new Error('out-of-bounds'); }
bounds = bounds.add(One).mul(NegativeOne);
if (v.lt(bounds)) { throw new Error('out-of-bounds'); }
} else if (v.lt(Zero) || v.gt(MaxUint256.maskn(this.size * 8))) {
throw new Error('out-of-bounds');
}
2018-06-13 22:39:39 +03:00
v = v.toTwos(this.size * 8).maskn(this.size * 8);
if (this.signed) {
v = v.fromTwos(this.size * 8).toTwos(256);
}
2018-06-13 22:39:39 +03:00
return padZeros(arrayify(v), 32);
} catch (error) {
errors.throwError('invalid number value', errors.INVALID_ARGUMENT, {
arg: this.localName,
coderType: this.name,
2018-06-13 22:39:39 +03:00
value: value
});
}
return null;
}
decode(data: Uint8Array, offset: number): DecodedResult {
if (data.length < offset + 32) {
errors.throwError('insufficient data for ' + this.name + ' type', errors.INVALID_ARGUMENT, {
arg: this.localName,
coderType: this.name,
value: hexlify(data.slice(offset, offset + 32))
});
}
var junkLength = 32 - this.size;
var value = bigNumberify(data.slice(offset + junkLength, offset + 32));
if (this.signed) {
value = value.fromTwos(this.size * 8);
} else {
value = value.maskn(this.size * 8);
}
return {
consumed: 32,
value: this.coerceFunc(this.name, value),
}
}
}
var uint256Coder = new CoderNumber(function(type: string, value: any) { return value; }, 32, false, 'none');
2018-06-13 22:39:39 +03:00
class CoderBoolean extends Coder {
constructor(coerceFunc: CoerceFunc, localName: string) {
2018-06-13 22:39:39 +03:00
super(coerceFunc, 'bool', 'bool', localName, false);
}
encode(value: boolean): Uint8Array {
return uint256Coder.encode(!!value ? 1: 0);
}
decode(data: Uint8Array, offset: number): DecodedResult {
try {
var result = uint256Coder.decode(data, offset);
} catch (error) {
if (error.reason === 'insufficient data for uint256 type') {
errors.throwError('insufficient data for boolean type', errors.INVALID_ARGUMENT, {
arg: this.localName,
coderType: 'boolean',
value: error.value
});
}
throw error;
}
return {
consumed: result.consumed,
value: this.coerceFunc('bool', !result.value.isZero())
}
}
}
class CoderFixedBytes extends Coder {
readonly length: number;
constructor(coerceFunc: CoerceFunc, length: number, localName: string) {
2018-06-13 22:39:39 +03:00
const name = ('bytes' + length);
super(coerceFunc, name, name, localName, false);
this.length = length;
}
encode(value: Arrayish): Uint8Array {
var result = new Uint8Array(32);
try {
let data = arrayify(value);
if (data.length !== this.length) { throw new Error('incorrect data length'); }
2018-06-13 22:39:39 +03:00
result.set(data);
} catch (error) {
errors.throwError('invalid ' + this.name + ' value', errors.INVALID_ARGUMENT, {
arg: this.localName,
coderType: this.name,
2018-06-13 22:39:39 +03:00
value: (error.value || value)
});
}
return result;
}
decode(data: Uint8Array, offset: number): DecodedResult {
if (data.length < offset + 32) {
errors.throwError('insufficient data for ' + name + ' type', errors.INVALID_ARGUMENT, {
arg: this.localName,
coderType: this.name,
value: hexlify(data.slice(offset, offset + 32))
});
}
return {
consumed: 32,
value: this.coerceFunc(this.name, hexlify(data.slice(offset, offset + this.length)))
}
}
}
class CoderAddress extends Coder {
constructor(coerceFunc: CoerceFunc, localName: string) {
2018-06-13 22:39:39 +03:00
super(coerceFunc, 'address', 'address', localName, false);
}
encode(value: string): Uint8Array {
let result = new Uint8Array(32);
try {
result.set(arrayify(getAddress(value)), 12);
} catch (error) {
errors.throwError('invalid address', errors.INVALID_ARGUMENT, {
arg: this.localName,
coderType: 'address',
2018-06-13 22:39:39 +03:00
value: value
});
}
return result;
}
decode(data: Uint8Array, offset: number): DecodedResult {
if (data.length < offset + 32) {
errors.throwError('insufficuent data for address type', errors.INVALID_ARGUMENT, {
arg: this.localName,
coderType: 'address',
value: hexlify(data.slice(offset, offset + 32))
});
}
return {
consumed: 32,
value: this.coerceFunc('address', getAddress(hexlify(data.slice(offset + 12, offset + 32))))
}
}
}
function _encodeDynamicBytes(value: Uint8Array): Uint8Array {
var dataLength = 32 * Math.ceil(value.length / 32);
2018-06-13 22:39:39 +03:00
var padding = new Uint8Array(dataLength - value.length);
return concat([
uint256Coder.encode(value.length),
value,
padding
]);
}
function _decodeDynamicBytes(data: Uint8Array, offset: number, localName: string): DecodedResult {
if (data.length < offset + 32) {
errors.throwError('insufficient data for dynamicBytes length', errors.INVALID_ARGUMENT, {
arg: localName,
coderType: 'dynamicBytes',
value: hexlify(data.slice(offset, offset + 32))
});
}
var length = uint256Coder.decode(data, offset).value;
try {
length = length.toNumber();
} catch (error) {
errors.throwError('dynamic bytes count too large', errors.INVALID_ARGUMENT, {
arg: localName,
coderType: 'dynamicBytes',
value: length.toString()
});
}
if (data.length < offset + 32 + length) {
errors.throwError('insufficient data for dynamicBytes type', errors.INVALID_ARGUMENT, {
arg: localName,
coderType: 'dynamicBytes',
value: hexlify(data.slice(offset, offset + 32 + length))
});
}
return {
consumed: 32 + 32 * Math.ceil(length / 32),
2018-06-13 22:39:39 +03:00
value: data.slice(offset + 32, offset + 32 + length),
}
}
class CoderDynamicBytes extends Coder {
constructor(coerceFunc: CoerceFunc, localName: string) {
2018-06-13 22:39:39 +03:00
super(coerceFunc, 'bytes', 'bytes', localName, true);
}
encode(value: Arrayish): Uint8Array {
try {
return _encodeDynamicBytes(arrayify(value));
} catch (error) {
errors.throwError('invalid bytes value', errors.INVALID_ARGUMENT, {
arg: this.localName,
coderType: 'bytes',
2018-06-13 22:39:39 +03:00
value: error.value
});
}
return null;
}
decode(data: Uint8Array, offset: number): DecodedResult {
var result = _decodeDynamicBytes(data, offset, this.localName);
result.value = this.coerceFunc('bytes', hexlify(result.value));
return result;
}
}
class CoderString extends Coder {
constructor(coerceFunc: CoerceFunc, localName: string) {
2018-06-13 22:39:39 +03:00
super(coerceFunc, 'string', 'string', localName, true);
}
encode(value: string): Uint8Array {
if (typeof(value) !== 'string') {
errors.throwError('invalid string value', errors.INVALID_ARGUMENT, {
arg: this.localName,
coderType: 'string',
2018-06-13 22:39:39 +03:00
value: value
});
}
return _encodeDynamicBytes(toUtf8Bytes(value));
}
decode(data: Uint8Array, offset: number): DecodedResult {
var result = _decodeDynamicBytes(data, offset, this.localName);
result.value = this.coerceFunc('string', toUtf8String(result.value));
return result;
}
}
function alignSize(size: number): number {
return 32 * Math.ceil(size / 32);
2018-06-13 22:39:39 +03:00
}
function pack(coders: Array<Coder>, values: Array<any>): Uint8Array {
if (Array.isArray(values)) {
// do nothing
} else if (values && typeof(values) === 'object') {
var arrayValues: Array<any> = [];
2018-06-13 22:39:39 +03:00
coders.forEach(function(coder) {
arrayValues.push((<any>values)[coder.localName]);
2018-06-13 22:39:39 +03:00
});
values = arrayValues;
} else {
errors.throwError('invalid tuple value', errors.INVALID_ARGUMENT, {
coderType: 'tuple',
value: values
});
}
if (coders.length !== values.length) {
errors.throwError('types/value length mismatch', errors.INVALID_ARGUMENT, {
coderType: 'tuple',
value: values
});
}
var parts: Array<{ dynamic: boolean, value: any }> = [];
2018-06-13 22:39:39 +03:00
coders.forEach(function(coder, index) {
parts.push({ dynamic: coder.dynamic, value: coder.encode(values[index]) });
});
var staticSize = 0, dynamicSize = 0;
parts.forEach(function(part) {
if (part.dynamic) {
staticSize += 32;
dynamicSize += alignSize(part.value.length);
} else {
staticSize += alignSize(part.value.length);
}
});
var offset = 0, dynamicOffset = staticSize;
var data = new Uint8Array(staticSize + dynamicSize);
parts.forEach(function(part) {
if (part.dynamic) {
//uint256Coder.encode(dynamicOffset).copy(data, offset);
data.set(uint256Coder.encode(dynamicOffset), offset);
offset += 32;
//part.value.copy(data, dynamicOffset); @TODO
data.set(part.value, dynamicOffset);
dynamicOffset += alignSize(part.value.length);
} else {
//part.value.copy(data, offset); @TODO
data.set(part.value, offset);
offset += alignSize(part.value.length);
}
});
return data;
}
function unpack(coders: Array<Coder>, data: Uint8Array, offset: number): DecodedResult {
var baseOffset = offset;
var consumed = 0;
var value: any = [];
2018-06-13 22:39:39 +03:00
coders.forEach(function(coder) {
if (coder.dynamic) {
var dynamicOffset = uint256Coder.decode(data, offset);
var result = coder.decode(data, baseOffset + dynamicOffset.value.toNumber());
// The dynamic part is leap-frogged somewhere else; doesn't count towards size
result.consumed = dynamicOffset.consumed;
} else {
var result = coder.decode(data, offset);
}
if (result.value != undefined) {
value.push(result.value);
}
offset += result.consumed;
consumed += result.consumed;
});
coders.forEach(function(coder: Coder, index: number) {
let name: string = coder.localName;
if (!name) { return; }
if (name === 'length') { name = '_length'; }
if (value[name] != null) { return; }
value[name] = value[index];
});
return {
value: value,
consumed: consumed
}
}
class CoderArray extends Coder {
readonly coder: Coder;
readonly length: number;
constructor(coerceFunc: CoerceFunc, coder: Coder, length: number, localName: string) {
2018-06-13 22:39:39 +03:00
const type = (coder.type + '[' + (length >= 0 ? length: '') + ']');
const dynamic = (length === -1 || coder.dynamic);
super(coerceFunc, 'array', type, localName, dynamic);
this.coder = coder;
this.length = length;
}
encode(value: Array<any>): Uint8Array {
if (!Array.isArray(value)) {
errors.throwError('expected array value', errors.INVALID_ARGUMENT, {
arg: this.localName,
coderType: 'array',
value: value
});
}
var count = this.length;
var result = new Uint8Array(0);
if (count === -1) {
count = value.length;
result = uint256Coder.encode(count);
}
errors.checkArgumentCount(count, value.length, ' in coder array' + (this.localName? (" "+ this.localName): ""));
2018-06-13 22:39:39 +03:00
var coders = [];
for (var i = 0; i < value.length; i++) { coders.push(this.coder); }
return concat([result, pack(coders, value)]);
}
decode(data: Uint8Array, offset: number) {
2018-06-13 22:39:39 +03:00
// @TODO:
//if (data.length < offset + length * 32) { throw new Error('invalid array'); }
var consumed = 0;
var count = this.length;
if (count === -1) {
try {
var decodedLength = uint256Coder.decode(data, offset);
} catch (error) {
errors.throwError('insufficient data for dynamic array length', errors.INVALID_ARGUMENT, {
arg: this.localName,
coderType: 'array',
value: error.value
});
}
try {
count = decodedLength.value.toNumber();
} catch (error) {
errors.throwError('array count too large', errors.INVALID_ARGUMENT, {
arg: this.localName,
coderType: 'array',
value: decodedLength.value.toString()
});
}
consumed += decodedLength.consumed;
offset += decodedLength.consumed;
}
var coders = [];
for (var i = 0; i < count; i++) { coders.push(new CoderAnonymous(this.coder)); }
2018-06-13 22:39:39 +03:00
var result = unpack(coders, data, offset);
result.consumed += consumed;
result.value = this.coerceFunc(this.type, result.value);
return result;
}
}
class CoderTuple extends Coder {
readonly coders: Array<Coder>;
constructor(coerceFunc: CoerceFunc, coders: Array<Coder>, localName: string) {
2018-06-13 22:39:39 +03:00
var dynamic = false;
var types: Array<string> = [];
2018-06-13 22:39:39 +03:00
coders.forEach(function(coder) {
if (coder.dynamic) { dynamic = true; }
types.push(coder.type);
});
var type = ('tuple(' + types.join(',') + ')');
super(coerceFunc, 'tuple', type, localName, dynamic);
this.coders = coders;
}
2018-06-13 22:39:39 +03:00
encode(value: Array<any>): Uint8Array {
return pack(this.coders, value);
}
2018-06-13 22:39:39 +03:00
decode(data: Uint8Array, offset: number): DecodedResult {
var result = unpack(this.coders, data, offset);
result.value = this.coerceFunc(this.type, result.value);
return result;
}
}
/*
function getTypes(coders) {
var type = coderTuple(coders).type;
return type.substring(6, type.length - 1);
}
*/
function splitNesting(value: string): Array<any> {
value = value.trim();
2018-06-13 22:39:39 +03:00
var result = [];
var accum = '';
var depth = 0;
for (var offset = 0; offset < value.length; offset++) {
var c = value[offset];
if (c === ',' && depth === 0) {
result.push(accum);
accum = '';
} else {
accum += c;
if (c === '(') {
depth++;
} else if (c === ')') {
depth--;
if (depth === -1) {
throw new Error('unbalanced parenthsis');
}
}
}
}
if (accum) { result.push(accum); }
2018-06-13 22:39:39 +03:00
return result;
}
// @TODO: Is there a way to return "class"?
const paramTypeSimple: { [key: string]: any } = {
2018-06-13 22:39:39 +03:00
address: CoderAddress,
bool: CoderBoolean,
string: CoderString,
bytes: CoderDynamicBytes,
};
function getTupleParamCoder(coerceFunc: CoerceFunc, components: Array<any>, localName: string): CoderTuple {
if (!components) { components = []; }
var coders: Array<Coder> = [];
2018-06-13 22:39:39 +03:00
components.forEach(function(component) {
coders.push(getParamCoder(coerceFunc, component));
});
return new CoderTuple(coerceFunc, coders, localName);
}
function getParamCoder(coerceFunc: CoerceFunc, param: ParamType): Coder {
var coder = paramTypeSimple[param.type];
if (coder) { return new coder(coerceFunc, param.name); }
var match = param.type.match(paramTypeNumber);
if (match) {
let size = parseInt(match[2] || "256");
if (size === 0 || size > 256 || (size % 8) !== 0) {
errors.throwError('invalid ' + match[1] + ' bit length', errors.INVALID_ARGUMENT, {
arg: 'param',
value: param
});
}
return new CoderNumber(coerceFunc, size / 8, (match[1] === 'int'), param.name);
}
var match = param.type.match(paramTypeBytes);
if (match) {
let size = parseInt(match[1]);
if (size === 0 || size > 32) {
errors.throwError('invalid bytes length', errors.INVALID_ARGUMENT, {
arg: 'param',
value: param
});
}
return new CoderFixedBytes(coerceFunc, size, param.name);
}
var match = param.type.match(paramTypeArray);
if (match) {
let size = parseInt(match[2] || "-1");
param = shallowCopy(param);
2018-06-13 22:39:39 +03:00
param.type = match[1];
param = deepCopy(param);
2018-06-13 22:39:39 +03:00
return new CoderArray(coerceFunc, getParamCoder(coerceFunc, param), size, param.name);
}
if (param.type.substring(0, 5) === 'tuple') {
return getTupleParamCoder(coerceFunc, param.components, param.name);
}
if (param.type === '') {
return new CoderNull(coerceFunc, param.name);
}
errors.throwError('invalid type', errors.INVALID_ARGUMENT, {
arg: 'type',
value: param.type
});
return null;
}
export class AbiCoder {
readonly coerceFunc: CoerceFunc;
constructor(coerceFunc?: CoerceFunc) {
2018-06-15 11:18:17 +03:00
errors.checkNew(this, AbiCoder);
2018-06-13 22:39:39 +03:00
if (!coerceFunc) { coerceFunc = defaultCoerceFunc; }
2018-06-15 11:18:17 +03:00
defineReadOnly(this, 'coerceFunc', coerceFunc);
2018-06-13 22:39:39 +03:00
}
encode(types: Array<string | ParamType>, values: Array<any>): string {
if (types.length !== values.length) {
errors.throwError('types/values length mismatch', errors.INVALID_ARGUMENT, {
count: { types: types.length, values: values.length },
value: { types: types, values: values }
});
}
var coders: Array<Coder> = [];
types.forEach(function(type) {
// Convert types to type objects
// - "uint foo" => { type: "uint", name: "foo" }
// - "tuple(uint, uint)" => { type: "tuple", components: [ { type: "uint" }, { type: "uint" }, ] }
2018-06-15 11:18:17 +03:00
let typeObject: ParamType = null;
2018-06-13 22:39:39 +03:00
if (typeof(type) === 'string') {
2018-06-15 11:18:17 +03:00
typeObject = parseParam(type);
} else {
typeObject = type;
2018-06-13 22:39:39 +03:00
}
2018-06-15 11:18:17 +03:00
coders.push(getParamCoder(this.coerceFunc, typeObject));
2018-06-13 22:39:39 +03:00
}, this);
return hexlify(new CoderTuple(this.coerceFunc, coders, '_').encode(values));
}
2018-07-27 00:31:08 +03:00
decode(types: Array<string | ParamType>, data: Arrayish): any {
2018-06-13 22:39:39 +03:00
var coders: Array<Coder> = [];
2018-06-13 22:39:39 +03:00
types.forEach(function(type) {
// See encode for details
2018-06-15 11:18:17 +03:00
let typeObject: ParamType = null;
2018-06-13 22:39:39 +03:00
if (typeof(type) === 'string') {
2018-06-15 11:18:17 +03:00
typeObject = parseParam(type);
} else {
typeObject = deepCopy(type);
2018-06-13 22:39:39 +03:00
}
2018-06-15 11:18:17 +03:00
coders.push(getParamCoder(this.coerceFunc, typeObject));
2018-06-13 22:39:39 +03:00
}, this);
return new CoderTuple(this.coerceFunc, coders, '_').decode(arrayify(data), 0).value;
}
}
export const defaultAbiCoder: AbiCoder = new AbiCoder();