feat: rewards v2.2 (#1659)

**Motivation:**

We need to support Rewards v2.2 :
- Rewards that are linear to allocated unique stake (Needs to be both
retroactive and future-looking).
- Rewards that are linear to total stake (Needs to be both retroactive
and future-looking).

TDD:
https://www.notion.so/eigen-labs/PRD-Rewards-v2-2-Operator-Set-Rewards-with-Unique-Total-Stake-28513c11c3e080cbb856e41ddc6362f5

**Modifications:**

Updates in `RewardsCoordinator`:
- New `createUniqueStakeRewardsSubmission` function with
`UniqueStakeRewardsSubmissionCreated` event emission
- New `PAUSED_UNIQUE_STAKE_REWARDS_SUBMISSION` constant
- New `isUniqueStakeRewardsSubmissionHash` mapping
- New `createTotalStakeRewardsSubmission` function with
`TotalStakeRewardsSubmissionCreated` event emission
- New `PAUSED_TOTAL_STAKE_REWARDS_SUBMISSION` constant
- New `isTotalStakeRewardsSubmissionHash` mapping
- Updated storage gap to 33 slots (from 35 slots)

Updated Bindings

**Result:**

Support for Rewards v2.2

Author: 0xrajath

pkg/bindings/IRewardsCoordinator/binding.go

@@ -203,6 +203,52 @@ contract RewardsCoordinator is
         }
     }
 
+    /// @inheritdoc IRewardsCoordinator
+    function createUniqueStakeRewardsSubmission(
+        OperatorSet calldata operatorSet,
+        RewardsSubmission[] calldata rewardsSubmissions
+    ) external onlyWhenNotPaused(PAUSED_UNIQUE_STAKE_REWARDS_SUBMISSION) checkCanCall(operatorSet.avs) nonReentrant {
+        require(allocationManager.isOperatorSet(operatorSet), InvalidOperatorSet());
+        for (uint256 i = 0; i < rewardsSubmissions.length; ++i) {
+            RewardsSubmission calldata rewardsSubmission = rewardsSubmissions[i];
+            uint256 nonce = submissionNonce[operatorSet.avs];
+            bytes32 rewardsSubmissionHash = keccak256(abi.encode(operatorSet.avs, nonce, rewardsSubmission));
+
+            _validateRewardsSubmission(rewardsSubmission);
+
+            isUniqueStakeRewardsSubmissionHash[operatorSet.avs][rewardsSubmissionHash] = true;
+            submissionNonce[operatorSet.avs] = nonce + 1;
+
+            emit UniqueStakeRewardsSubmissionCreated(
+                msg.sender, rewardsSubmissionHash, operatorSet, nonce, rewardsSubmission
+            );
+            rewardsSubmission.token.safeTransferFrom(msg.sender, address(this), rewardsSubmission.amount);
+        }
+    }
+
+    /// @inheritdoc IRewardsCoordinator
+    function createTotalStakeRewardsSubmission(
+        OperatorSet calldata operatorSet,
+        RewardsSubmission[] calldata rewardsSubmissions
+    ) external onlyWhenNotPaused(PAUSED_TOTAL_STAKE_REWARDS_SUBMISSION) checkCanCall(operatorSet.avs) nonReentrant {
+        require(allocationManager.isOperatorSet(operatorSet), InvalidOperatorSet());
+        for (uint256 i = 0; i < rewardsSubmissions.length; ++i) {
+            RewardsSubmission calldata rewardsSubmission = rewardsSubmissions[i];
+            uint256 nonce = submissionNonce[operatorSet.avs];
+            bytes32 rewardsSubmissionHash = keccak256(abi.encode(operatorSet.avs, nonce, rewardsSubmission));
+
+            _validateRewardsSubmission(rewardsSubmission);
+
+            isTotalStakeRewardsSubmissionHash[operatorSet.avs][rewardsSubmissionHash] = true;
+            submissionNonce[operatorSet.avs] = nonce + 1;
+
+            emit TotalStakeRewardsSubmissionCreated(
+                msg.sender, rewardsSubmissionHash, operatorSet, nonce, rewardsSubmission
+            );
+            rewardsSubmission.token.safeTransferFrom(msg.sender, address(this), rewardsSubmission.amount);
+        }
+    }
+
     /// @inheritdoc IRewardsCoordinator
     function processClaim(
         RewardsMerkleClaim calldata claim,

pkg/bindings/RewardsCoordinator/binding.go

@@ -32,6 +32,10 @@ abstract contract RewardsCoordinatorStorage is IRewardsCoordinator {
     uint8 internal constant PAUSED_OPERATOR_SET_SPLIT = 8;
     /// @dev Index for flag that pauses calling setOperatorSetPerformanceRewardsSubmission
     uint8 internal constant PAUSED_OPERATOR_DIRECTED_OPERATOR_SET_REWARDS_SUBMISSION = 9;
+    /// @dev Index for flag that pauses calling createOperatorSetRewardsSubmission
+    uint8 internal constant PAUSED_UNIQUE_STAKE_REWARDS_SUBMISSION = 10;
+    /// @dev Index for flag that pauses calling createTotalStakeRewardsSubmission
+    uint8 internal constant PAUSED_TOTAL_STAKE_REWARDS_SUBMISSION = 11;
 
     /// @dev Salt for the earner leaf, meant to distinguish from tokenLeaf since they have the same sized data
     uint8 internal constant EARNER_LEAF_SALT = 0;
@@ -130,8 +134,13 @@ abstract contract RewardsCoordinatorStorage is IRewardsCoordinator {
     mapping(address avs => mapping(bytes32 hash => bool valid)) public
         isOperatorDirectedOperatorSetRewardsSubmissionHash;
 
-    // Construction
+    /// @notice Returns whether a `hash` is a `valid` unique stake rewards submission hash for a given `avs`.
+    mapping(address avs => mapping(bytes32 hash => bool valid)) public isUniqueStakeRewardsSubmissionHash;
+
+    /// @notice Returns whether a `hash` is a `valid` total stake rewards submission hash for a given `avs`.
+    mapping(address avs => mapping(bytes32 hash => bool valid)) public isTotalStakeRewardsSubmissionHash;
 
+    // Construction
     constructor(
         IDelegationManager _delegationManager,
         IStrategyManager _strategyManager,
@@ -161,5 +170,5 @@ abstract contract RewardsCoordinatorStorage is IRewardsCoordinator {
      * variables without shifting down storage in the inheritance chain.
      * See https://docs.openzeppelin.com/contracts/4.x/upgradeable#storage_gaps
      */
-    uint256[35] private __gap;
+    uint256[33] private __gap;
 }

pkg/bindings/RewardsCoordinatorStorage/binding.go

@@ -333,6 +333,38 @@ interface IRewardsCoordinatorEvents is IRewardsCoordinatorTypes {
         OperatorDirectedRewardsSubmission operatorDirectedRewardsSubmission
     );
 
+    /**
+     * @notice Emitted when an AVS creates a valid `UniqueStakeRewardsSubmission` for an operator set.
+     * @param caller The address calling `createUniqueStakeRewardsSubmission`.
+     * @param rewardsSubmissionHash Keccak256 hash of (`avs`, `submissionNonce` and `rewardsSubmission`).
+     * @param operatorSet The operatorSet on behalf of which the rewards are being submitted.
+     * @param submissionNonce Current nonce of the avs. Used to generate a unique submission hash.
+     * @param rewardsSubmission The Rewards Submission. Contains the token, start timestamp, duration, strategies and multipliers.
+     */
+    event UniqueStakeRewardsSubmissionCreated(
+        address indexed caller,
+        bytes32 indexed rewardsSubmissionHash,
+        OperatorSet operatorSet,
+        uint256 submissionNonce,
+        RewardsSubmission rewardsSubmission
+    );
+
+    /**
+     * @notice Emitted when an AVS creates a valid `TotalStakeRewardsSubmission` for an operator set.
+     * @param caller The address calling `createTotalStakeRewardsSubmission`.
+     * @param rewardsSubmissionHash Keccak256 hash of (`avs`, `submissionNonce` and `rewardsSubmission`).
+     * @param operatorSet The operatorSet on behalf of which the rewards are being submitted.
+     * @param submissionNonce Current nonce of the avs. Used to generate a unique submission hash.
+     * @param rewardsSubmission The Rewards Submission. Contains the token, start timestamp, duration, strategies and multipliers.
+     */
+    event TotalStakeRewardsSubmissionCreated(
+        address indexed caller,
+        bytes32 indexed rewardsSubmissionHash,
+        OperatorSet operatorSet,
+        uint256 submissionNonce,
+        RewardsSubmission rewardsSubmission
+    );
+
     /// @notice rewardsUpdater is responsible for submitting DistributionRoots, only owner can set rewardsUpdater
     event RewardsUpdaterSet(address indexed oldRewardsUpdater, address indexed newRewardsUpdater);
 
@@ -515,6 +547,40 @@ interface IRewardsCoordinator is IRewardsCoordinatorErrors, IRewardsCoordinatorE
         OperatorDirectedRewardsSubmission[] calldata operatorDirectedRewardsSubmissions
     ) external;
 
+    /**
+     * @notice Creates a new unique stake rewards submission for an operator set, to be split amongst the operators and
+     * set of stakers delegated to operators. The operators have to both be registered and allocate unique slashable stake to the operator set to be rewarded.
+     * @param operatorSet The operator set for which the rewards are being submitted
+     * @param rewardsSubmissions The rewards submissions being created
+     * @dev Expected to be called by the AVS that created the operator set
+     * @dev The duration of the `rewardsSubmission` cannot exceed `MAX_REWARDS_DURATION`
+     * @dev The duration of the `rewardsSubmission` cannot be 0 and must be a multiple of `CALCULATION_INTERVAL_SECONDS`
+     * @dev The tokens are sent to the `RewardsCoordinator` contract
+     * @dev Strategies must be in ascending order of addresses to check for duplicates
+     * @dev This function will revert if the `rewardsSubmissions` is malformed.
+     */
+    function createUniqueStakeRewardsSubmission(
+        OperatorSet calldata operatorSet,
+        RewardsSubmission[] calldata rewardsSubmissions
+    ) external;
+
+    /**
+     * @notice Creates a new total stake rewards submission for an operator set, to be split amongst the operators and
+     * set of stakers delegated to operators. The operators have to just be registered to the operator set to be rewarded.
+     * @param operatorSet The operator set for which the rewards are being submitted
+     * @param rewardsSubmissions The rewards submissions being created
+     * @dev Expected to be called by the AVS that created the operator set
+     * @dev The duration of the `rewardsSubmission` cannot exceed `MAX_REWARDS_DURATION`
+     * @dev The duration of the `rewardsSubmission` cannot be 0 and must be a multiple of `CALCULATION_INTERVAL_SECONDS`
+     * @dev The tokens are sent to the `RewardsCoordinator` contract
+     * @dev Strategies must be in ascending order of addresses to check for duplicates
+     * @dev This function will revert if the `rewardsSubmissions` is malformed.
+     */
+    function createTotalStakeRewardsSubmission(
+        OperatorSet calldata operatorSet,
+        RewardsSubmission[] calldata rewardsSubmissions
+    ) external;
+
     /**
      * @notice Claim rewards against a given root (read from _distributionRoots[claim.rootIndex]).
      * Earnings are cumulative so earners don't have to claim against all distribution roots they have earnings for,