|
| 1 | +// SPDX-License-Identifier: Apache-2.0 |
| 2 | +// Copyright 2024 Aztec Labs. |
| 3 | +pragma solidity >=0.8.27; |
| 4 | + |
| 5 | +import {Errors} from "./Errors.sol"; |
| 6 | + |
| 7 | +/** |
| 8 | + * @title SlashPayloadLib |
| 9 | + * @author Aztec Labs |
| 10 | + * @notice Library for encoding immutable arguments for SlashPayloadCloneable contracts |
| 11 | + * @dev Provides utilities for encoding validator addresses and amounts into a format |
| 12 | + * suitable for use with EIP-1167 minimal proxy clones with immutable arguments |
| 13 | + */ |
| 14 | +library SlashPayloadLib { |
| 15 | + /** |
| 16 | + * @notice Encode immutable arguments for SlashPayloadCloneable clones |
| 17 | + * @dev Encodes data in the format expected by SlashPayloadCloneable._getImmutableArgs() |
| 18 | + * Layout: [validatorSelection(20 bytes)][arrayLength(32 bytes)][validators+amounts array data] |
| 19 | + * Each validator entry: [address(20 bytes)][amount(12 bytes for uint96)] |
| 20 | + * @param _validatorSelection Address of the validator selection contract |
| 21 | + * @param _validators Array of validator addresses to slash |
| 22 | + * @param _amounts Array of amounts to slash for each validator (uint96 values) |
| 23 | + * @return Encoded arguments for use with cloneDeterministicWithImmutableArgs |
| 24 | + */ |
| 25 | + function encodeImmutableArgs(address _validatorSelection, address[] memory _validators, uint96[] memory _amounts) |
| 26 | + internal |
| 27 | + pure |
| 28 | + returns (bytes memory) |
| 29 | + { |
| 30 | + require( |
| 31 | + _validators.length == _amounts.length, Errors.SlashPayload_ArraySizeMismatch(_validators.length, _amounts.length) |
| 32 | + ); |
| 33 | + |
| 34 | + // Calculate total size: 20 bytes (address) + 32 bytes (length) + (20 + 12) * length |
| 35 | + uint256 dataSize = 52 + 32 * _validators.length; |
| 36 | + bytes memory data = new bytes(dataSize); |
| 37 | + |
| 38 | + assembly { |
| 39 | + let ptr := add(data, 0x20) |
| 40 | + |
| 41 | + // Store validator selection address (20 bytes) |
| 42 | + // Shift left by 96 bits (12 bytes) to align to the left of the 32-byte slot |
| 43 | + mstore(ptr, shl(96, _validatorSelection)) |
| 44 | + ptr := add(ptr, 0x14) // Move 20 bytes forward |
| 45 | + |
| 46 | + // Store array length (32 bytes) |
| 47 | + mstore(ptr, mload(_validators)) |
| 48 | + ptr := add(ptr, 0x20) // Move 32 bytes forward |
| 49 | + |
| 50 | + // Store validators and amounts |
| 51 | + let len := mload(_validators) |
| 52 | + let validatorsPtr := add(_validators, 0x20) |
| 53 | + let amountsPtr := add(_amounts, 0x20) |
| 54 | + |
| 55 | + for { let i := 0 } lt(i, len) { i := add(i, 1) } { |
| 56 | + // Store validator address (20 bytes) |
| 57 | + // Shift left by 96 bits to align to the left of the 32-byte slot |
| 58 | + mstore(ptr, shl(96, mload(add(validatorsPtr, mul(i, 0x20))))) |
| 59 | + ptr := add(ptr, 0x14) // Move 20 bytes forward |
| 60 | + |
| 61 | + // Store amount (12 bytes for uint96) |
| 62 | + // Shift left by 160 bits (20 bytes) to align to the left of the remaining space |
| 63 | + mstore(ptr, shl(160, mload(add(amountsPtr, mul(i, 0x20))))) |
| 64 | + ptr := add(ptr, 0x0c) // Move 12 bytes forward |
| 65 | + } |
| 66 | + } |
| 67 | + |
| 68 | + return data; |
| 69 | + } |
| 70 | +} |
0 commit comments