embarklabs
/
embark
mirror of https://github.com/embarklabs/embark.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

convert console logs to embark logger and this baby is ready to rock 

Signed-off-by: VoR0220 <catalanor0220@gmail.com>
This commit is contained in:
VoR0220 2018-06-05 14:16:38 -05:00
parent a7629d2ac0
commit 773f5a7921
1 changed files with 3 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
lib/modules/fuzzer/index.js
View File

@ -16,25 +16,20 @@ class ContractFuzzer {
// main function to call, takes in iteration number and a contract and returns a map object
// composed of method names -> fuzzed inputs.
generateFuzz(iterations, contract) {
const self = this;
let fuzzMap = {};
for (let i = 0; i < iterations; i++) contract.abiDefinition.filter((x) => x.inputs && x.inputs.length != 0).forEach((abiMethod) => {
let name = abiMethod.type === "constructor" ? "constructor" : abiMethod.name;
console.log("name");
let inputTypes = abiMethod.inputs.map(input => input.type);
console.log("INPUT TYPES:", inputTypes);
let fuzzedInputs = _.map(inputTypes, this.getTypeFuzz.bind(this));
console.log("FUZZED INPUTS:", fuzzedInputs);
fuzzMap[name] = fuzzedInputs;
console.log("FUZZ KEYS SO FAR:", Object.keys(fuzzMap));
console.log("FUZZ VALS SO FAR:", Object.values(fuzzMap));
});
console.log("FUZZ:", Object.keys(fuzzMap), Object.values(fuzzMap));
for (let key in fuzzMap) self.logger.info(key + ":" + fuzzMap[key]);
return fuzzMap;
}
getTypeFuzz(typeString) {
const self = this;
console.log("TYPESTRING:", typeString);
// Group 0: uint256[3]
// Group 1: uint256
// Group 2: uint
@ -42,22 +37,15 @@ class ContractFuzzer {
// Group 4: [3]
// Group 5: 3
let regexObj = typeString.match(/((bool|int|uint|bytes|string|address)([0-9]*)?)(\[([0-9]*)\])*$/);
console.log("REGEX OBJ:", regexObj);
let type = regexObj[1];
console.log("type:", type);
let kind = regexObj[2];
console.log("kind:", kind);
let size = regexObj[3];
console.log("size:", size);
let array = regexObj[4];
console.log("array:", array);
let arraySize = regexObj[5];
console.log("array size:", arraySize);
switch(true) {
case array !== undefined: {
// if it's a dynamic array pick a number between 1 and 256 for length of array
let length = arraySize === undefined || arraySize === null || arraySize === '' ? Math.floor((Math.random() * 256) + 1) : arraySize;
console.log("LENGTH: ", length);
return self.generateArrayOfType(length, type);
}
case kind == "bool":
@ -81,10 +69,7 @@ class ContractFuzzer {
generateArrayOfType(length, type) {
var arr = [];
for (var i = 0; i < length; i++) {
arr.push(this.getTypeFuzz(type));
}
console.log("Final Array:", arr);
for (var i = 0; i < length; i++) arr.push(this.getTypeFuzz(type));
return arr;
}