blockworks-foundation
/
solana
mirror of https://github.com/blockworks-foundation/solana.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

checked_sub_lamports (#16918)

This commit is contained in:
Jeff Washington (jwash) 2021-04-29 16:34:44 -05:00 committed by GitHub
parent 63c56b57a9
commit 94edd6140c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
programs/budget/src/budget_processor.rs
View File

@ -35,7 +35,9 @@ fn apply_signature(
if let Some(key) = witness_keyed_account.signer_key() {
if &payment.to == key {
budget_state.pending_budget = None;
contract_keyed_account.try_account_ref_mut()?.lamports -= payment.lamports;
contract_keyed_account
.try_account_ref_mut()?
.checked_sub_lamports(payment.lamports)?;
witness_keyed_account
.try_account_ref_mut()?
.checked_add_lamports(payment.lamports)?;
@ -48,7 +50,9 @@ fn apply_signature(
return Err(BudgetError::DestinationMissing.into());
}
budget_state.pending_budget = None;
contract_keyed_account.try_account_ref_mut()?.lamports -= payment.lamports;
contract_keyed_account
.try_account_ref_mut()?
.checked_sub_lamports(payment.lamports)?;
to_keyed_account
.try_account_ref_mut()?
.checked_add_lamports(payment.lamports)?;
@ -81,7 +85,9 @@ fn apply_timestamp(
return Err(BudgetError::DestinationMissing.into());
}
budget_state.pending_budget = None;
contract_keyed_account.try_account_ref_mut()?.lamports -= payment.lamports;
contract_keyed_account
.try_account_ref_mut()?
.checked_sub_lamports(payment.lamports)?;
to_keyed_account
.try_account_ref_mut()?
.checked_add_lamports(payment.lamports)?;
@ -114,7 +120,9 @@ fn apply_account_data(
return Err(BudgetError::DestinationMissing.into());
}
budget_state.pending_budget = None;
contract_keyed_account.try_account_ref_mut()?.lamports -= payment.lamports;
contract_keyed_account
.try_account_ref_mut()?
.checked_sub_lamports(payment.lamports)?;
to_keyed_account
.try_account_ref_mut()?
.checked_add_lamports(payment.lamports)?;