MessageHashUtils: Add toDataWithIntendedValidatorHash(address, bytes32) (#5081)

Co-authored-by: Hadrien Croubois <hadrien.croubois@gmail.com>
Co-authored-by: Arr00 <13561405+arr00@users.noreply.github.com>
This commit is contained in:
StackOverflowExcept1on
2025-03-01 03:00:43 +07:00
committed by GitHub
parent a9b1f58b00
commit a4b0d89900
4 changed files with 87 additions and 5 deletions

View File

@ -0,0 +1,33 @@
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.20;
import {Test} from "forge-std/Test.sol";
import {MessageHashUtils} from "@openzeppelin/contracts/utils/cryptography/MessageHashUtils.sol";
contract MessageHashUtilsTest is Test {
function testToDataWithIntendedValidatorHash(address validator, bytes memory data) external pure {
assertEq(
MessageHashUtils.toDataWithIntendedValidatorHash(validator, data),
MessageHashUtils.toDataWithIntendedValidatorHash(_dirty(validator), data)
);
}
function testToDataWithIntendedValidatorHash(address validator, bytes32 messageHash) external pure {
assertEq(
MessageHashUtils.toDataWithIntendedValidatorHash(validator, messageHash),
MessageHashUtils.toDataWithIntendedValidatorHash(_dirty(validator), messageHash)
);
assertEq(
MessageHashUtils.toDataWithIntendedValidatorHash(validator, messageHash),
MessageHashUtils.toDataWithIntendedValidatorHash(validator, abi.encodePacked(messageHash))
);
}
function _dirty(address input) private pure returns (address output) {
assembly ("memory-safe") {
output := or(input, shl(160, not(0)))
}
}
}