switch back to InputStream

stateless/test_block_witness.nim

@@ -27,27 +27,29 @@ proc isValidBranch(branch: openArray[seq[byte]], rootHash: KeccakHash, key, valu
 proc testGetBranch(tester: Tester, rootHash: KeccakHash, testStatusIMPL: var TestStatus) =
   var trie = initHexaryTrie(tester.memdb, rootHash)
 
-  try:
-    for address in tester.address:
-      var wb = initWitnessBuilder(tester.memdb, rootHash)
-      var witness = wb.getBranchRecurse(address)
-
-      var db = newMemoryDB()
-      var tb = initTreeBuilder(witness, db)
-      var root = tb.treeNode()
-      check root.data == rootHash.data
-      #echo "ROOT: ", root.data.toHex
-      #echo "rootHash: ", rootHash.data.toHex
-
-      #var stackBranch = wb.getBranchStack(address)
-      #check recurseBranch == stackBranch
-      #
-      #var branch = wb.getBranch(address)
-      #let account = trie.get(address)
-      #check isValidBranch(branch, trie.rootHash, address, account)
-  except:
-    debugEcho "MSG: ", getCurrentExceptionMsg()
-    quit(1)
+  #try:
+  for address in tester.address:
+    var wb = initWitnessBuilder(tester.memdb, rootHash)
+    var witness = wb.getBranchRecurse(address)
+  
+    var db = newMemoryDB()
+    var input = memoryInput(witness)
+    var tb = initTreeBuilder(input, db)
+    var root = tb.treeNode()
+    check root.data == rootHash.data
+    
+    #echo "ROOT: ", root.data.toHex
+    #echo "rootHash: ", rootHash.data.toHex
+  
+    #var stackBranch = wb.getBranchStack(address)
+    #check recurseBranch == stackBranch
+    #
+    #var branch = wb.getBranch(address)
+    #let account = trie.get(address)
+    #check isValidBranch(branch, trie.rootHash, address, account)
+  #except:
+    #debugEcho "MSG: ", getCurrentExceptionMsg()
+    #quit(1)
 
 func parseHash256(n: JsonNode, name: string): Hash256 =
   hexToByteArray(n[name].getStr(), result.data)
@@ -95,8 +97,8 @@ proc testFixtureGST(node: JsonNode, testStatusIMPL: var TestStatus) =
 
   fixture["pre"].testBlockWitness(emptyRlpHash, testStatusIMPL)
 
-suite "Block Witness":
-  jsonTest("newBlockChainTests", "witnessBuilderBC", testFixtureBC)
+#suite "Block Witness":
+  #jsonTest("newBlockChainTests", "witnessBuilderBC", testFixtureBC)
 
 suite "Block Witness":
   jsonTest("GeneralStateTests", "witnessBuilderGST", testFixtureGST)

stateless/test_witness_keys.nim

@@ -1,6 +1,7 @@
 import
   randutils, stew/byteutils, random,
   eth/[common, rlp], eth/trie/[hexary, db, trie_defs],
+  faststreams/input_stream,
   ../stateless/[witness_from_tree, tree_from_witness]
 
 proc runTest(keyBytes: int, valBytes: int, numPairs: int) =
@@ -20,9 +21,10 @@ proc runTest(keyBytes: int, valBytes: int, numPairs: int) =
 
   var wb = initWitnessBuilder(memDB, rootHash)
   var witness = wb.getBranchRecurse(keys[0])
+  var input = memoryInput(witness)
 
-  var db = newMemoryDB()
-  var tb = initTreeBuilder(witness, db)
+  var db = newMemoryDB()  
+  var tb = initTreeBuilder(input, db)
   var root = tb.treeNode()
   debugEcho "root: ", root.data.toHex
   debugEcho "rootHash: ", rootHash.data.toHex

stateless/tree_from_witness.nim

@@ -11,23 +11,23 @@ type
     data*: array[32, byte]
 
   TreeBuilder = object
-    data: seq[byte]
-    pos: int
-    #input: InputStream
+    #data: seq[byte]
+    #pos: int
+    input: InputStream
     db: DB
     root: KeccakHash
 
 # InputStream is unstable, so we hack our own inputstream
-#proc initTreeBuilder*(input: InputStream, db: DB): TreeBuilder =
-  #result.input = input
-  #result.db = db
-  #result.root = emptyRlpHash
-
-proc initTreeBuilder*(input: openArray[byte], db: DB): TreeBuilder =
-  result.data = @input
+proc initTreeBuilder*(input: InputStream, db: DB): TreeBuilder =
+  result.input = input
   result.db = db
   result.root = emptyRlpHash
 
+#proc initTreeBuilder*(input: openArray[byte], db: DB): TreeBuilder =
+#  result.data = @input
+#  result.db = db
+#  result.root = emptyRlpHash
+
 func rootHash*(t: TreeBuilder): KeccakHash {.inline.} =
   t.root
 
@@ -35,7 +35,19 @@ proc writeNode(t: var TreeBuilder, n: openArray[byte]): KeccakHash =
   result = keccak(n)
   t.db.put(result.data, n)
 
-proc readByte(t: var TreeBuilder): byte =
+template readByte(t: var TreeBuilder): byte =
+  t.input.read
+
+template len(t: TreeBuilder): int =
+  t.input.len
+
+template peek(t: TreeBuilder): byte =
+  t.input.peek
+
+template read(t: var TreeBuilder, len: int): auto =
+  t.input.read(len)
+
+#[proc readByte(t: var TreeBuilder): byte =
   if t.pos < t.data.len:
     result = t.data[t.pos]
     inc t.pos
@@ -51,6 +63,7 @@ template read(t: var TreeBuilder, len: int): auto =
   let pos = t.pos
   inc(t.pos, len)
   toOpenArray(t.data, pos, pos + len - 1)
+]#
 
 proc readU32(t: var TreeBuilder): int =
   # TODO: what if the value overflow int32.high?