parent
77628b02cb
commit
0b8c303fac
@ -1,37 +1,61 @@ |
|||||||
const web3Utils = require('web3-utils') |
const web3Utils = require('web3-utils') |
||||||
|
|
||||||
|
/** |
||||||
|
* Writes data from the VM step to the in-memory |
||||||
|
* coverage map constructed by the Instrumenter. |
||||||
|
*/ |
||||||
class DataCollector { |
class DataCollector { |
||||||
constructor(instrumentationData={}){ |
constructor(instrumentationData={}){ |
||||||
this.instrumentationData = instrumentationData; |
this.instrumentationData = instrumentationData; |
||||||
|
|
||||||
|
this.validOpcodes = { |
||||||
|
"PUSH1": true, |
||||||
|
} |
||||||
} |
} |
||||||
|
|
||||||
|
/** |
||||||
|
* VM step event handler. Detects instrumentation hashes when they are pushed to the |
||||||
|
* top of the stack. This runs millions of times - trying to keep it fast. |
||||||
|
* @param {Object} info vm step info |
||||||
|
*/ |
||||||
step(info){ |
step(info){ |
||||||
const self = this; |
try { |
||||||
if (typeof info !== 'object' || !info.opcode ) return; |
if (this.validOpcodes[info.opcode.name] && info.stack.length > 0){ |
||||||
|
const idx = info.stack.length - 1; |
||||||
if (info.opcode.name.includes("PUSH1") && info.stack.length > 0){ |
let hash = web3Utils.toHex(info.stack[idx]).toString(); |
||||||
const idx = info.stack.length - 1; |
hash = this._normalizeHash(hash); |
||||||
let hash = web3Utils.toHex(info.stack[idx]).toString(); |
|
||||||
hash = self._normalizeHash(hash); |
|
||||||
|
|
||||||
if(self.instrumentationData[hash]){ |
if(this.instrumentationData[hash]){ |
||||||
self.instrumentationData[hash].hits++; |
this.instrumentationData[hash].hits++; |
||||||
|
} |
||||||
} |
} |
||||||
} |
} catch (err) { /*Ignore*/ }; |
||||||
} |
|
||||||
|
|
||||||
_setInstrumentationData(data){ |
|
||||||
this.instrumentationData = data; |
|
||||||
} |
} |
||||||
|
|
||||||
|
/** |
||||||
|
* Left-pads zero prefixed bytes 32 hashes to length 66. The '59' in the |
||||||
|
* comparison below is arbitrary. It provides a margin for recurring zeros |
||||||
|
* but prevents left-padding shorter irrelevant hashes (like fn sigs) |
||||||
|
* |
||||||
|
* @param {String} hash data hash from evm stack. |
||||||
|
* @return {String} 0x prefixed hash of length 66. |
||||||
|
*/ |
||||||
_normalizeHash(hash){ |
_normalizeHash(hash){ |
||||||
if (hash.length < 66 && hash.length > 52){ |
if (hash.length < 66 && hash.length > 59){ |
||||||
hash = hash.slice(2); |
hash = hash.slice(2); |
||||||
while(hash.length < 64) hash = '0' + hash; |
while(hash.length < 64) hash = '0' + hash; |
||||||
hash = '0x' + hash |
hash = '0x' + hash |
||||||
} |
} |
||||||
return hash; |
return hash; |
||||||
} |
} |
||||||
|
|
||||||
|
/** |
||||||
|
* Unit test helper |
||||||
|
* @param {Object} data Instrumenter.instrumentationData |
||||||
|
*/ |
||||||
|
_setInstrumentationData(data){ |
||||||
|
this.instrumentationData = data; |
||||||
|
} |
||||||
} |
} |
||||||
|
|
||||||
module.exports = DataCollector; |
module.exports = DataCollector; |
Loading…
Reference in new issue