108 lines
3.1 KiB
Nim
108 lines
3.1 KiB
Nim
|
import
|
||
|
|
json, downloader, stint, strutils,
|
||
|
|
../nimbus/tracer, chronicles, prestate
|
||
|
|
|
||
|
|
proc fakeAlloc(n: JsonNode) =
|
||
|
|
const
|
||
|
|
chunk = repeat('0', 64)
|
||
|
|
|
||
|
|
for i in 1 ..< n.len:
|
||
|
|
if not n[i].hasKey("memory"): return
|
||
|
|
let
|
||
|
|
prevMem = n[i-1]["memory"]
|
||
|
|
currMem = n[i]["memory"]
|
||
|
|
|
||
|
|
if currMem.len > prevMem.len:
|
||
|
|
let diff = currMem.len - prevMem.len
|
||
|
|
for _ in 0 ..< diff:
|
||
|
|
prevMem.add %chunk
|
||
|
|
|
||
|
|
proc requestPostState(n: JsonNode, jsTracer: string): JsonNode =
|
||
|
|
let txs = n["transactions"]
|
||
|
|
if txs.len > 0:
|
||
|
|
result = newJArray()
|
||
|
|
let tracer = %{
|
||
|
|
"tracer": %jsTracer
|
||
|
|
}
|
||
|
|
for tx in txs:
|
||
|
|
let txHash = tx["hash"]
|
||
|
|
let txTrace = request("debug_traceTransaction", %[txHash, tracer])
|
||
|
|
if txTrace.kind == JNull:
|
||
|
|
error "requested postState not available", txHash=txHash
|
||
|
|
raise newException(ValueError, "Error when retrieving transaction postState")
|
||
|
|
result.add txTrace
|
||
|
|
|
||
|
|
proc requestPostState(thisBlock: Block): JsonNode =
|
||
|
|
let
|
||
|
|
tmp = readFile("poststate_tracer.js.template")
|
||
|
|
tracer = tmp % [ $thisBlock.jsonData["miner"] ]
|
||
|
|
requestPostState(thisBlock.jsonData, tracer)
|
||
|
|
|
||
|
|
proc copyAccount(acc: JsonNode): JsonNode =
|
||
|
|
result = newJObject()
|
||
|
|
result["balance"] = newJString(acc["balance"].getStr)
|
||
|
|
result["nonce"] = newJString(acc["nonce"].getStr)
|
||
|
|
result["code"] = newJString(acc["code"].getStr)
|
||
|
|
var storage = newJObject()
|
||
|
|
for k, v in acc["storage"]:
|
||
|
|
storage[k] = newJString(v.getStr)
|
||
|
|
result["storage"] = storage
|
||
|
|
|
||
|
|
proc updateAccount(a, b: JsonNode) =
|
||
|
|
a["balance"] = newJString(b["balance"].getStr)
|
||
|
|
a["nonce"] = newJString(b["nonce"].getStr)
|
||
|
|
a["code"] = newJString(b["code"].getStr)
|
||
|
|
var storage = a["storage"]
|
||
|
|
for k, v in b["storage"]:
|
||
|
|
storage[k] = newJString(v.getStr)
|
||
|
|
|
||
|
|
proc processPostState(postState: JsonNode): JsonNode =
|
||
|
|
var accounts = newJObject()
|
||
|
|
|
||
|
|
for state in postState:
|
||
|
|
for address, account in state:
|
||
|
|
if accounts.hasKey(address):
|
||
|
|
updateAccount(accounts[address], account)
|
||
|
|
else:
|
||
|
|
accounts[address] = copyAccount(account)
|
||
|
|
|
||
|
|
result = accounts
|
||
|
|
|
||
|
|
proc generatePremixData(nimbus: JsonNode, blockNumber: Uint256, thisBlock: Block, postState, accounts: JsonNode) =
|
||
|
|
let
|
||
|
|
receipts = toJson(thisBlock.receipts)
|
||
|
|
txTraces = nimbus["txTraces"]
|
||
|
|
|
||
|
|
for trace in txTraces:
|
||
|
|
trace["structLogs"].fakeAlloc()
|
||
|
|
|
||
|
|
var geth = %{
|
||
|
|
"blockNumber": %blockNumber.toHex,
|
||
|
|
"txTraces": thisBlock.traces,
|
||
|
|
"receipts": receipts,
|
||
|
|
"block": thisBlock.jsonData,
|
||
|
|
"postState": postState,
|
||
|
|
"accounts": accounts
|
||
|
|
}
|
||
|
|
|
||
|
|
var metaData = %{
|
||
|
|
"nimbus": nimbus,
|
||
|
|
"geth": geth
|
||
|
|
}
|
||
|
|
|
||
|
|
var data = "var debugMetaData = " & metaData.pretty & "\n"
|
||
|
|
writeFile("premixData.js", data)
|
||
|
|
|
||
|
|
proc main() =
|
||
|
|
let
|
||
|
|
nimbus = parseFile("debug_meta_data.json")
|
||
|
|
blockNumber = UInt256.fromHex(nimbus["blockNumber"].getStr())
|
||
|
|
thisBlock = downloader.requestBlock(blockNumber, {DownloadReceipts, DownloadTxTrace})
|
||
|
|
postState = requestPostState(thisBlock)
|
||
|
|
accounts = processPostState(postState)
|
||
|
|
|
||
|
|
generatePremixData(nimbus, blockNumber, thisBlock, postState, accounts)
|
||
|
|
generatePrestate(nimbus, blockNumber, thisBlock)
|
||
|
|
|
||
|
|
main()
|