mirror of https://github.com/embarklabs/embark.git
75 lines
2.9 KiB
JavaScript
75 lines
2.9 KiB
JavaScript
/*global web3*/
|
|
const async = require('async');
|
|
const _ = require('underscore');
|
|
const ContractFuzzer = require('../fuzzer');
|
|
|
|
class GasEstimator {
|
|
constructor(embark) {
|
|
this.embark = embark;
|
|
this.logger = embark.logger;
|
|
this.events = embark.events;
|
|
this.fuzzer = new ContractFuzzer(embark);
|
|
}
|
|
|
|
estimateGas(contractName, cb) {
|
|
const self = this;
|
|
let gasMap = {};
|
|
self.events.request('contracts:contract', contractName, (contract) => {
|
|
self.logger.info("Generating three rounds of fuzz for contract -- ", contractName);
|
|
let fuzzMap = self.fuzzer.generateFuzz(3, contract);
|
|
self.logger.info("-- Beginning gastimation for contract -- " + contractName);
|
|
let contractObj = new web3.eth.Contract(contract.abiDefinition, contract.deployedAddress);
|
|
async.each(contract.abiDefinition.filter((x) => x.type !== "event"),
|
|
(abiMethod, gasCb) => {
|
|
let name = abiMethod.name;
|
|
if (abiMethod.type === "constructor") {
|
|
// already provided for us
|
|
gasMap['constructor'] = contract.gasEstimates.creation.totalCost.toString();
|
|
return gasCb(null, name, abiMethod.type);
|
|
} else if (abiMethod.type == "fallback") {
|
|
gasMap['fallback'] = contract.gasEstimates.external[""].toString();
|
|
return gasCb(null, name, abiMethod.type);
|
|
} else if (
|
|
(abiMethod.inputs === null || abiMethod.inputs === undefined || abiMethod.inputs.length === 0)
|
|
) {
|
|
// just run it and register it
|
|
contractObj.methods[name]
|
|
.apply(contractObj.methods[name], [])
|
|
.estimateGas((err, gasAmount) => {
|
|
if (err) return gasCb(err, name, abiMethod.type);
|
|
gasMap[name] = gasAmount;
|
|
return gasCb(null, name, abiMethod.type);
|
|
});
|
|
} else {
|
|
// async concatenate all the fuzz values and their gas cost outputs and check for equality
|
|
async.concat(fuzzMap[name], (values, getVarianceCb) => {
|
|
contractObj.methods[name].apply(contractObj.methods[name], values)
|
|
.estimateGas((err, gasAmount) => {
|
|
getVarianceCb(err, gasAmount);
|
|
});
|
|
}, (err, variance) => {
|
|
if (err) {
|
|
return gasCb(err, name, abiMethod.type);
|
|
} else if (_.isEqual(variance[0], variance[1]) && _.isEqual(variance[1], variance[2])) {
|
|
gasMap[name] = variance[0];
|
|
} else {
|
|
gasMap[name] = 'infinite';
|
|
}
|
|
return gasCb(null, name, abiMethod.type);
|
|
});
|
|
}
|
|
},
|
|
(err, name, type) => {
|
|
if (err) {
|
|
if (type === "constructor" || type === "fallback") name = type;
|
|
return cb(err, null, name);
|
|
}
|
|
cb(null, gasMap, null);
|
|
}
|
|
);
|
|
});
|
|
}
|
|
}
|
|
|
|
module.exports = GasEstimator;
|