From 773f5a792127f705c4d1eace26a3c588650102c5 Mon Sep 17 00:00:00 2001 From: VoR0220 Date: Tue, 5 Jun 2018 14:16:38 -0500 Subject: [PATCH] convert console logs to embark logger and this baby is ready to rock Signed-off-by: VoR0220 --- lib/modules/fuzzer/index.js | 21 +++------------------ 1 file changed, 3 insertions(+), 18 deletions(-) diff --git a/lib/modules/fuzzer/index.js b/lib/modules/fuzzer/index.js index f2f6f2ab1..0a1d96863 100644 --- a/lib/modules/fuzzer/index.js +++ b/lib/modules/fuzzer/index.js @@ -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; }