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)
  }
 }