Merge pull request #743 from schemar/generate_proof_data

Generate proof data

.eslintrc.json

@@ -56,5 +56,12 @@
     "artifacts": false,
     "contract": false,
     "assert": false
+  },
+  "settings": {
+    "import/resolver": {
+      "node": {
+        "extensions": [".js", ".jsx", ".ts", ".tsx"]
+      }
+    }
   }
 }

tools/fuzzy_proof_generator_tool/generate.ts

@@ -0,0 +1,53 @@
+import * as fs from 'fs';
+import * as path from 'path';
+
+import FuzzyProofGenerator from './src/FuzzyProofGenerator';
+import ProofData from './src/ProofData';
+import patterns from './patterns';
+
+interface HexProof {
+  value: string;
+  encodedPath: string;
+  rlpParentNodes: string;
+  root: string;
+}
+
+/** Converts a buffer to a hex string with a leading 0x. */
+function bufferToHex(buffer: Buffer): string {
+  return `0x${buffer.toString('hex')}`;
+}
+
+/** Converts a proof to a hex proof. */
+function proofToHexProof(proof: ProofData): HexProof {
+  return {
+    value: bufferToHex(proof.value),
+    encodedPath: bufferToHex(proof.encodedPath),
+    rlpParentNodes: bufferToHex(proof.rlpParentNodes),
+    root: bufferToHex(proof.root),
+  };
+}
+
+const proofs: ProofData[] = [];
+
+// Run 4 times to generate more data.
+for (let i = 0; i < 4; i += 1) {
+  patterns.forEach(
+    (pattern: string): number => proofs.push(FuzzyProofGenerator.generateByPattern(pattern)),
+  );
+}
+
+// Convert to hex to be in line with the rest of the code base.
+const hexProofs: HexProof[] = proofs.map(
+  (proof: ProofData): HexProof => proofToHexProof(proof),
+);
+
+const filePath: string = path.join(
+  __dirname,
+  '..',
+  '..',
+  'test',
+  'data',
+  'generatedProofData.json',
+);
+
+fs.writeFileSync(filePath, JSON.stringify(hexProofs));

tools/fuzzy_proof_generator_tool/patterns.ts

@@ -0,0 +1,19 @@
+const patterns: string[] = [
+  'l',
+  'bl',
+  'bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbl',
+  'ebl',
+  'bebl',
+  'bbebbl',
+  'bebbbbl',
+  'bbbebbbbbl',
+  'ebebl',
+  'ebebebebebl',
+  'bbebbebebebebbbebebl',
+  'bebb',
+  'bebebb',
+  'ebb',
+  'ebebebebbb',
+];
+
+export { patterns as default };

tools/fuzzy_proof_generator_tool/src/FuzzyProofGenerator.ts

@@ -20,18 +20,13 @@ import { BranchNode } from './BranchNode';
 import { BranchKeys } from './BranchNode';
 import ExtensionNode from './ExtensionNode';
 import LeafNode from './LeafNode';
+import ProofData from './ProofData';
 
 import assert = require('assert');
 import ethUtil = require('ethereumjs-util');
 import rlp = require('rlp');
 import crypto = require('crypto');
 
-interface ProofData {
-  value: Buffer;
-  encodedPath: Buffer;
-  rlpParentNodes: Buffer;
-  root: Buffer;
-}
 
 const FuzzyProofGenerator = {
 

tools/fuzzy_proof_generator_tool/src/ProofData.ts

@@ -0,0 +1,6 @@
+export default interface ProofData {
+  value: Buffer;
+  encodedPath: Buffer;
+  rlpParentNodes: Buffer;
+  root: Buffer;
+}