gaspersic/solana
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

checked_sub_lamports (#16918)

Browse Source
This commit is contained in:
Jeff Washington (jwash)
2021-04-29 16:34:44 -05:00
committed by GitHub
parent 63c56b57a9
commit 94edd6140c
1 changed files with 12 additions and 4 deletions
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 let Some(key) = witness_keyed_account.signer_key() {
if &payment.to == key { if &payment.to == key {
budget_state.pending_budget = None; 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 witness_keyed_account
.try_account_ref_mut()? .try_account_ref_mut()?
.checked_add_lamports(payment.lamports)?; .checked_add_lamports(payment.lamports)?;
@ -48,7 +50,9 @@ fn apply_signature(
return Err(BudgetError::DestinationMissing.into()); return Err(BudgetError::DestinationMissing.into());
} }
budget_state.pending_budget = None; 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 to_keyed_account
.try_account_ref_mut()? .try_account_ref_mut()?
.checked_add_lamports(payment.lamports)?; .checked_add_lamports(payment.lamports)?;
@ -81,7 +85,9 @@ fn apply_timestamp(
return Err(BudgetError::DestinationMissing.into()); return Err(BudgetError::DestinationMissing.into());
} }
budget_state.pending_budget = None; 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 to_keyed_account
.try_account_ref_mut()? .try_account_ref_mut()?
.checked_add_lamports(payment.lamports)?; .checked_add_lamports(payment.lamports)?;
@ -114,7 +120,9 @@ fn apply_account_data(
return Err(BudgetError::DestinationMissing.into()); return Err(BudgetError::DestinationMissing.into());
} }
budget_state.pending_budget = None; 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 to_keyed_account
.try_account_ref_mut()? .try_account_ref_mut()?
.checked_add_lamports(payment.lamports)?; .checked_add_lamports(payment.lamports)?;