update secure pda example

coral-xyz · Jul 12, 2022 · 4a8e9d3 · 4a8e9d3
1 parent 2b70674
commit 4a8e9d3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/programs/8-pda-sharing/secure/src/lib.rs b/programs/8-pda-sharing/secure/src/lib.rs
@@ -13,6 +13,13 @@ pub mod pda_sharing_secure {
             ctx.accounts.pool.withdraw_destination.as_ref(),
             &[ctx.accounts.pool.bump],
         ];
+        let (address, expected_bump) = Pubkey::find_program_address(seeds, ctx.program_id);
+        if address != ctx.accounts.pool.key() {
+            return Err(ProgramError::InvalidArgument);
+        }
+        if expected_bump != ctx.accounts.pool.bump {
+            return Err(ProgramError::InvalidArgument);
+        }
         token::transfer(ctx.accounts.transfer_ctx().with_signer(&[seeds]), amount)
     }
 }