modularisation

src/basicPanel.js

@@ -38,7 +38,7 @@ module.exports = React.createClass({
         ret.push(
           <tr key={key}>
             <td>
-              {this.props.data[key]}
+              <pre>{this.props.data[key]}</pre>
             </td>
           </tr>)
       }

src/memoryPanel.js

@@ -50,13 +50,13 @@ module.exports = React.createClass({
         ret.push(
           <tr key={key}>
             <td>
-              {memSlot.address}
+              <pre>{memSlot.address}</pre>
             </td>
             <td>
-              {memSlot.content.raw}
+              <pre>{memSlot.content.raw}</pre>
             </td>
             <td>
-              {memSlot.content.ascii}
+              <pre>{memSlot.content.ascii}</pre>
             </td>
           </tr>)
       }

src/storagePanel.js

@@ -49,10 +49,10 @@ module.exports = React.createClass({
         ret.push(
           <tr key={key}>
             <td>
-              {key}
+              <pre>{key}</pre>
             </td>
             <td>
-              {data[key]}
+              <pre>{data[key]}</pre>
             </td>
           </tr>)
       }

src/traceAnalyser.js

+'use strict'
+function TraceAnalyser (_cache) {
+  this.traceCache = _cache
+  this.trace = null
+}
+
+TraceAnalyser.prototype.analyse = function (trace, callback) {
+  this.trace = trace
+  var currentDepth = 0
+  var context = {
+    currentStorageAddress: trace[0].address,
+    previousStorageAddress: trace[0].address
+  }
+  var callStack = []
+  for (var k in this.trace) {
+    var step = this.trace[k]
+    this.buildCalldata(k, step)
+    this.buildMemory(k, step)
+    var depth = this.buildDepth(k, step, currentDepth, callStack)
+    if (depth) {
+      currentDepth = depth
+    }
+    context = this.buildStorage(k, step, context)
+  }
+  callback(null, true)
+}
+
+TraceAnalyser.prototype.buildCalldata = function (index, step) {
+  if (step.calldata) {
+    this.traceCache.pushCallDataChanges(index)
+  }
+}
+
+TraceAnalyser.prototype.buildMemory = function (index, step) {
+  if (step.memory) {
+    this.traceCache.pushMemoryChanges(index)
+  }
+}
+
+TraceAnalyser.prototype.buildStorage = function (index, step, context) {
+  if (step.address) {
+    // new context
+    context.currentStorageAddress = step.address
+    this.traceCache.pushStoreChanges(index, context.currentStorageAddress)
+  } else if (step.inst === 'SSTORE') {
+    this.traceCache.pushStoreChanges(index, context.currentStorageAddress, step.stack[step.stack.length - 1], step.stack[step.stack.length - 2])
+  } else if (!step.address && step.depth) {
+    // returned from context
+    context.currentStorageAddress = context.previousStorageAddress
+    this.traceCache.pushStoreChanges(index, context.currentStorageAddress)
+  }
+  return context
+}
+
+TraceAnalyser.prototype.buildDepth = function (index, step, currentDepth, callStack) {
+  if (step.depth === undefined) return
+  if (step.depth > currentDepth) {
+    if (index === 0) {
+      callStack.push('0x' + step.address) // new context
+    } else {
+      // getting the address from the stack
+      var callTrace = this.trace[index - 1]
+      var address = callTrace.stack[callTrace.stack.length - 2]
+      callStack.push(address) // new context
+    }
+  } else if (step.depth < currentDepth) {
+    callStack.pop() // returning from context
+  }
+  this.traceCache.pushCallStack(index, {
+    stack: callStack.slice(0),
+    depth: step.depth
+  })
+  this.traceCache.pushDepthChanges(index)
+  return step.depth
+}
+
+module.exports = TraceAnalyser

src/traceCache.js

+'use strict'
+function TraceCache () {
+  this.init()
+}
+
+TraceCache.prototype.init = function () {
+  // ...Changes contains index in the vmtrace of the corresponding changes
+  this.depthChanges = []
+  this.memoryChanges = []
+  this.callDataChanges = []
+  this.storageChanges = []
+  this.sstore = {} // all sstore occurence in the trace
+  this.callStack = {} // contains all callStack by vmtrace index (we need to rebuild it, callstack is not included in the vmtrace)
+}
+
+TraceCache.prototype.pushCallDataChanges = function (value) {
+  this.callDataChanges.push(value)
+}
+
+TraceCache.prototype.pushMemoryChanges = function (value) {
+  this.memoryChanges.push(value)
+}
+
+TraceCache.prototype.pushDepthChanges = function (value) {
+  this.depthChanges.push(value)
+}
+
+TraceCache.prototype.pushCallStack = function (index, callStack) {
+  this.callStack[index] = callStack
+}
+
+TraceCache.prototype.pushStoreChanges = function (index, address, key, value) {
+  this.sstore[index] = {
+    'address': address,
+    'key': key,
+    'value': value
+  }
+  this.storageChanges.push(index)
+}
+
+TraceCache.prototype.rebuildStorage = function (address, storage, index) {
+  for (var k in this.storageChanges) {
+    var changesIndex = this.storageChanges[k]
+    if (changesIndex > index) {
+      return storage
+    }
+    var sstore = this.sstore[changesIndex]
+    if (sstore.address === address && sstore.key) {
+      storage[sstore.key] = sstore.value
+    }
+  }
+  return storage
+}
+
+module.exports = TraceCache

src/traceManagerUtil.js

+module.exports = {
+  // util section
+  findLowerBound: function (target, changes) {
+    if (changes.length === 0) {
+      return undefined
+    }
+
+    if (changes.length === 1) {
+      if (changes[0] > target) {
+        // we only a closest maximum, returning O
+        return 0
+      } else {
+        return changes[0]
+      }
+    }
+
+    var middle = Math.floor(changes.length / 2)
+    if (changes[middle] > target) {
+      return this.findLowerBound(target, changes.slice(0, middle))
+    } else if (changes[middle] < target) {
+      return this.findLowerBound(target, changes.slice(middle, changes.length))
+    } else {
+      return changes[middle]
+    }
+  },
+
+  resolveCalledAddress: function (vmTraceIndex, trace) {
+    var address = trace[vmTraceIndex].address
+    if (vmTraceIndex > 0) {
+      var stack = trace[vmTraceIndex - 1].stack // callcode, delegatecall, ...
+      address = stack[stack.length - 2]
+    }
+    return address
+  }
+
+}

src/traceRetriever.js

+'use strict'
+function TraceRetriever (_web3) {
+  this.web3 = _web3
+  this.storages = {} // contains all intial storage (by addresses)
+}
+
+TraceRetriever.prototype.getTrace = function (blockNumber, txNumber, callback) {
+  this.web3.debug.trace(blockNumber, parseInt(txNumber), function (error, result) {
+    callback(error, result)
+  })
+}
+
+TraceRetriever.prototype.getStorage = function (blockNumber, txIndex, address, callback) {
+  if (this.storages[address]) {
+    callback(null, this.storages[address])
+  } else {
+    var self = this
+    this.web3.debug.storageAt(blockNumber, txIndex, address, function (error, result) {
+      self.storages[address] = result
+      callback(error, result)
+    })
+  }
+}
+
+module.exports = TraceRetriever

src/traceStepManager.js

+'use strict'
+function TraceStepManager (_traceAnalyser) {
+  this.traceAnalyser = _traceAnalyser
+}
+
+TraceStepManager.prototype.isCallInstruction = function (index) {
+  var state = this.traceAnalyser.trace[index]
+  return state.instname === 'CALL' || state.instname === 'CALLCODE' || state.instname === 'CREATE' || state.instname === 'DELEGATECALL'
+}
+
+TraceStepManager.prototype.isReturnInstruction = function (index) {
+  var state = this.traceAnalyser.trace[index]
+  return state.instname === 'RETURN'
+}
+
+TraceStepManager.prototype.findStepOverBack = function (currentStep) {
+  if (this.isReturnInstruction(currentStep - 1)) {
+    return this.findStepOutBack(currentStep)
+  } else {
+    return currentStep - 1
+  }
+}
+
+TraceStepManager.prototype.findStepOverForward = function (currentStep) {
+  if (this.isCallInstruction(currentStep)) {
+    return this.findStepOutForward(currentStep)
+  } else {
+    return currentStep + 1
+  }
+}
+
+TraceStepManager.prototype.findStepOutBack = function (currentStep) {
+  var i = currentStep - 1
+  var depth = 0
+  while (--i >= 0) {
+    if (this.isCallInstruction(i)) {
+      if (depth === 0) {
+        break
+      } else {
+        depth--
+      }
+    } else if (this.isReturnInstruction(i)) {
+      depth++
+    }
+  }
+  return i
+}
+
+TraceStepManager.prototype.findStepOutForward = function (currentStep) {
+  var i = currentStep
+  var depth = 0
+  while (++i < this.traceAnalyser.length) {
+    if (this.isReturnInstruction(i)) {
+      if (depth === 0) {
+        break
+      } else {
+        depth--
+      }
+    } else if (this.isCallInstruction(i)) {
+      depth++
+    }
+  }
+  return i + 1
+}
+
+module.exports = TraceStepManager