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

lamports += to checked_add (#16841)

This commit is contained in:
Jeff Washington (jwash) 2021-04-27 09:56:18 -05:00 committed by GitHub
parent 47ca7063f2
commit 3887169db0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 18 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
programs/vote/src/vote_state/mod.rs
View File

@ -686,7 +686,9 @@ pub fn withdraw<S: std::hash::BuildHasher>(
_ => (),
}
vote_account.try_account_ref_mut()?.lamports -= lamports;
to_account.try_account_ref_mut()?.lamports += lamports;
to_account
.try_account_ref_mut()?
.checked_add_lamports(lamports)?;
Ok(())
}

9
runtime/src/bank.rs
View File

@ -5449,7 +5449,10 @@ pub(crate) mod tests {
if let Ok(instruction) = bincode::deserialize(data) {
match instruction {
MockInstruction::Deduction => {
keyed_accounts[1].account.borrow_mut().lamports += 1;
keyed_accounts[1]
.account
.borrow_mut()
.checked_add_lamports(1)?;
keyed_accounts[2].account.borrow_mut().lamports -= 1;
Ok(())
}
@ -9868,7 +9871,7 @@ pub(crate) mod tests {
let mut to_account = keyed_accounts[1].try_account_ref_mut()?;
let mut dup_account = keyed_accounts[2].try_account_ref_mut()?;
dup_account.lamports -= lamports;
to_account.lamports += lamports;
to_account.checked_add_lamports(lamports).unwrap();
}
keyed_accounts[0]
.try_account_ref_mut()?
@ -10359,7 +10362,7 @@ pub(crate) mod tests {
let keyed_accounts = invoke_context.get_keyed_accounts()?;
assert_eq!(42, keyed_accounts[0].lamports().unwrap());
let mut account = keyed_accounts[0].try_account_ref_mut()?;
account.lamports += 1;
account.checked_add_lamports(1)?;
Ok(())
}

11
runtime/src/message_processor.rs
View File

@ -1828,7 +1828,10 @@ mod tests {
MockSystemInstruction::Correct => Ok(()),
MockSystemInstruction::AttemptCredit { lamports } => {
keyed_accounts[0].account.borrow_mut().lamports -= lamports;
keyed_accounts[1].account.borrow_mut().lamports += lamports;
keyed_accounts[1]
.account
.borrow_mut()
.checked_add_lamports(lamports)?;
Ok(())
}
// Change data in a read-only account
@ -2005,11 +2008,13 @@ mod tests {
let mut to_account = keyed_accounts[1].try_account_ref_mut()?;
let mut dup_account = keyed_accounts[2].try_account_ref_mut()?;
dup_account.lamports -= lamports;
to_account.lamports += lamports;
to_account.checked_add_lamports(lamports)?;
dup_account.set_data(vec![data]);
}
keyed_accounts[0].try_account_ref_mut()?.lamports -= lamports;
keyed_accounts[1].try_account_ref_mut()?.lamports += lamports;
keyed_accounts[1]
.try_account_ref_mut()?
.checked_add_lamports(lamports)?;
Ok(())
}
}

2
runtime/src/system_instruction_processor.rs
View File

@ -190,7 +190,7 @@ fn transfer_verified(
}
from.try_account_ref_mut()?.lamports -= lamports;
to.try_account_ref_mut()?.lamports += lamports;
to.try_account_ref_mut()?.checked_add_lamports(lamports)?;
Ok(())
}