Add an option to get non inclusion proof with hashed key

commitground · Nov 24, 2018 · 47d4f77 · 47d4f77
1 parent 42d6f5a
commit 47d4f77
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/contracts/tree.sol b/contracts/tree.sol
@@ -114,12 +114,21 @@ library PatriciaTree {
         bytes32 potentialSiblingValue,
         uint branchMask,
         bytes32[] _siblings
+    ) {
+        return getNonInclusionProofWithHashedKey(tree, keccak256(key));
+    }
+
+    function getNonInclusionProofWithHashedKey(Tree storage tree, bytes32 hashedKey) internal view returns (
+        bytes32 potentialSiblingLabel,
+        bytes32 potentialSiblingValue,
+        uint branchMask,
+        bytes32[] _siblings
     ){
         uint length;
         uint numSiblings;
 
         // Start from root edge
-        D.Label memory label = D.Label(keccak256(key), 256);
+        D.Label memory label = D.Label(hashedKey, 256);
         D.Edge memory e = tree.rootEdge;
         bytes32[256] memory siblings;