63 lines
2.3 KiB
Solidity
63 lines
2.3 KiB
Solidity
// SPDX-License-Identifier: MIT
|
|
|
|
pragma solidity ^0.8.20;
|
|
|
|
import {MerkleProof} from "../utils/cryptography/MerkleProof.sol";
|
|
|
|
// This could be a library, but then we would have to add it to the Stateless.sol mock for upgradeable tests
|
|
abstract contract MerkleProofCustomHashMock {
|
|
function customHash(bytes32 a, bytes32 b) internal pure returns (bytes32) {
|
|
return a < b ? sha256(abi.encode(a, b)) : sha256(abi.encode(b, a));
|
|
}
|
|
|
|
function verify(bytes32[] calldata proof, bytes32 root, bytes32 leaf) internal view returns (bool) {
|
|
return MerkleProof.verify(proof, root, leaf, customHash);
|
|
}
|
|
|
|
function processProof(bytes32[] calldata proof, bytes32 leaf) internal view returns (bytes32) {
|
|
return MerkleProof.processProof(proof, leaf, customHash);
|
|
}
|
|
|
|
function verifyCalldata(bytes32[] calldata proof, bytes32 root, bytes32 leaf) internal view returns (bool) {
|
|
return MerkleProof.verifyCalldata(proof, root, leaf, customHash);
|
|
}
|
|
|
|
function processProofCalldata(bytes32[] calldata proof, bytes32 leaf) internal view returns (bytes32) {
|
|
return MerkleProof.processProofCalldata(proof, leaf, customHash);
|
|
}
|
|
|
|
function multiProofVerify(
|
|
bytes32[] calldata proof,
|
|
bool[] calldata proofFlags,
|
|
bytes32 root,
|
|
bytes32[] calldata leaves
|
|
) internal view returns (bool) {
|
|
return MerkleProof.multiProofVerify(proof, proofFlags, root, leaves, customHash);
|
|
}
|
|
|
|
function processMultiProof(
|
|
bytes32[] calldata proof,
|
|
bool[] calldata proofFlags,
|
|
bytes32[] calldata leaves
|
|
) internal view returns (bytes32) {
|
|
return MerkleProof.processMultiProof(proof, proofFlags, leaves, customHash);
|
|
}
|
|
|
|
function multiProofVerifyCalldata(
|
|
bytes32[] calldata proof,
|
|
bool[] calldata proofFlags,
|
|
bytes32 root,
|
|
bytes32[] calldata leaves
|
|
) internal view returns (bool) {
|
|
return MerkleProof.multiProofVerifyCalldata(proof, proofFlags, root, leaves, customHash);
|
|
}
|
|
|
|
function processMultiProofCalldata(
|
|
bytes32[] calldata proof,
|
|
bool[] calldata proofFlags,
|
|
bytes32[] calldata leaves
|
|
) internal view returns (bytes32) {
|
|
return MerkleProof.processMultiProofCalldata(proof, proofFlags, leaves, customHash);
|
|
}
|
|
}
|