From b5c45b260540ba1e6b9010ca8e223f8a8cad7673 Mon Sep 17 00:00:00 2001
From: Giannis Chatziveroglou <gchatz@mit.edu>
Date: Tue, 3 Jan 2023 18:44:01 +0200
Subject: [PATCH] move to anchor checks

---
 .../reward_distributor/init_reward_distributor.rs            | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/programs/cardinal-rewards-center/src/reward_distribution/reward_distributor/init_reward_distributor.rs b/programs/cardinal-rewards-center/src/reward_distribution/reward_distributor/init_reward_distributor.rs
index aa9a906d..079383ab 100644
--- a/programs/cardinal-rewards-center/src/reward_distribution/reward_distributor/init_reward_distributor.rs
+++ b/programs/cardinal-rewards-center/src/reward_distribution/reward_distributor/init_reward_distributor.rs
@@ -32,6 +32,7 @@ pub struct InitRewardDistributorCtx<'info> {
         bump,
     )]
     reward_distributor: Box<Account<'info, RewardDistributor>>,
+    #[account(constraint = authority.key() != stake_pool.authority.key() @ ErrorCode::InvalidAuthority)]
     stake_pool: Box<Account<'info, StakePool>>,
     #[account(mut)]
     reward_mint: Box<Account<'info, Mint>>,
@@ -58,10 +59,6 @@ pub fn handler(ctx: Context<InitRewardDistributorCtx>, ix: InitRewardDistributor
     reward_distributor.max_reward_seconds_received = ix.max_reward_seconds_received;
     reward_distributor.claim_rewards_payment_info = ix.claim_rewards_payment_info;
 
-    if ctx.accounts.authority.key() != ctx.accounts.stake_pool.authority.key() {
-        return Err(error!(ErrorCode::InvalidAuthority));
-    }
-
     assert_payment_info(ctx.accounts.stake_pool.key(), Action::ClaimRewards, ix.claim_rewards_payment_info)?;
     Ok(())
 }