This repository has been archived by the owner on Apr 19, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathMerkleProof.t.sol
147 lines (125 loc) · 6.04 KB
/
MerkleProof.t.sol
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
// SPDX-License-Identifier: UNLICENSED
pragma solidity ^0.8.13;
import "forge-std/console.sol";
import "forge-std/Test.sol";
import {MerkleProof as SnowbridgeMerkleProof} from "snowbridge/utils/MerkleProof.sol";
import {MerkleProof as OpenZeppelinMerkleProof} from "openzeppelin/utils/cryptography/MerkleProof.sol";
// import {OptimisedMerkleProof as OpenZeppelinMerkleProof} from "src/OptimisedMerkleProof.sol";
contract MerkleProofTest is Test {
uint256 leafIndex = 3;
bytes[] leaves = [
bytes(hex"0000000000000000000000000000000000000000000000000000000000000000"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000001"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000002"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000003"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000004"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000005"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000006"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000007"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000008"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000009"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000010"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000011"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000012"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000013"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000014"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000015"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000016"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000017"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000018"),
bytes(hex"0000000000000000000000000000000000000000000000000000000000000019")
];
function setUp() external {}
function testUnoptimisedProofLengthIndex() external {
bytes32 expectedRoot = hex"18bbfdde0b580e2bbe3d45449000b98b7e98f66fc58f187ade7de5343eb3a15c";
bytes32[] memory proof = new bytes32[](5);
proof[0] = bytes32(
hex"58e9a6ca105dd9ed36c41988e9cf55b722660d6b76f83ebc322fecc94477b2ab"
);
proof[1] = bytes32(
hex"b9ea4098a5754a60725baf50899ccd2e84f4e9cd26c6ee5f8df8e164e6dffa09"
);
proof[2] = bytes32(
hex"5c2ad896a3f86e0d3391c5c2ab35e31d6dc16f8843bcdb8bb77d93b733612a18"
);
proof[3] = bytes32(
hex"df340c91de0b21ac6130307f468b025f0a665789dd7e8f586096566c387e8131"
);
proof[4] = bytes32(
hex"f50a7b2cd8561cab32e7f43f50c64e9ad3838c2c25e45e61e6cf00a434270580"
);
bytes32 hashedLeaf = hashLeaf(leaves[leafIndex]);
bytes32 computedRoot = SnowbridgeMerkleProof
.computeRootFromProofAtPosition(
hashedLeaf,
leafIndex,
leaves.length,
proof
);
assertEq(computedRoot, expectedRoot);
}
function testUnoptimisedProofSidesArray() external {
bytes32 expectedRoot = hex"18bbfdde0b580e2bbe3d45449000b98b7e98f66fc58f187ade7de5343eb3a15c";
bytes32[] memory proof = new bytes32[](5);
proof[0] = bytes32(
hex"58e9a6ca105dd9ed36c41988e9cf55b722660d6b76f83ebc322fecc94477b2ab"
);
proof[1] = bytes32(
hex"b9ea4098a5754a60725baf50899ccd2e84f4e9cd26c6ee5f8df8e164e6dffa09"
);
proof[2] = bytes32(
hex"5c2ad896a3f86e0d3391c5c2ab35e31d6dc16f8843bcdb8bb77d93b733612a18"
);
proof[3] = bytes32(
hex"df340c91de0b21ac6130307f468b025f0a665789dd7e8f586096566c387e8131"
);
proof[4] = bytes32(
hex"f50a7b2cd8561cab32e7f43f50c64e9ad3838c2c25e45e61e6cf00a434270580"
);
bool[] memory hashSides = new bool[](7);
hashSides[0] = true;
hashSides[1] = true;
hashSides[2] = false;
hashSides[3] = false;
hashSides[4] = false;
bytes32 hashedLeaf = hashLeaf(leaves[leafIndex]);
bytes32 computedRoot = SnowbridgeMerkleProof
.computeRootFromProofAndSide(hashedLeaf, proof, hashSides);
assertEq(computedRoot, expectedRoot);
}
function testOptimisedProof() external {
bytes32 expectedRoot = hex"b08ed46dee2199775e04c69dc62b43614b404d05cf09b83bd064e6594d0715e9";
bytes32[] memory proof = new bytes32[](4);
proof[0] = bytes32(
hex"75b9ce82e2f316dd49c7fc23ba632b1b06c2eb9b40e86abb6fa4ba3b940dbf4b"
);
proof[1] = bytes32(
hex"3e8fda3dd4a61ef41110c4b250cdf7ac07717402207d418a36a2bc5c511daebf"
);
proof[2] = bytes32(
hex"52f43dcfa7482ed14dea86ede03ab138ebc7125eb97525b09f4715b615da18cf"
);
proof[3] = bytes32(
hex"5642a45a5810d90f9aadaceeb5bd365b791fd637d0f986b6d19869f2d616a859"
);
bytes32 hashedLeaf = hashLeaf(leaves[leafIndex]);
bytes32 computedRoot = OpenZeppelinMerkleProof.processProof(
proof,
hashedLeaf
);
assertEq(computedRoot, expectedRoot);
}
function hashLeaf(bytes storage leaf) private pure returns (bytes32) {
bytes memory encodedLeaf = abi.encode(leaf);
// Leaves are double-hashed to prevent second preimage attacks
bytes32 hashedLeaf = keccak256(abi.encode(keccak256(encodedLeaf)));
// // Uncomment for intermediate leaf values
// console.log("leaf:");
// console.logBytes(leaf);
// console.log("encoded leaf:");
// console.logBytes(encodedLeaf);
// console.log("leaf hash:");
// console.logBytes32(hashedLeaf);
return hashedLeaf;
}
}