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

Drop 'unchecked' from get_subset_mut()

This commit is contained in:
Michael Vines 2019-03-19 15:51:44 -07:00 committed by Grimes
parent 682b1b89b3
commit 70b21b3795
1 changed files with 9 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
runtime/src/runtime.rs
View File

@ -18,7 +18,7 @@ pub fn has_duplicates<T: PartialEq>(xs: &[T]) -> bool {
} }
/// Get mut references to a subset of elements. /// Get mut references to a subset of elements.
fn get_subset_unchecked_mut<'a, T>( fn get_subset_mut<'a, T>(
xs: &'a mut [T], xs: &'a mut [T],
indexes: &[u8], indexes: &[u8],
) -> Result<Vec<&'a mut T>, InstructionError> { ) -> Result<Vec<&'a mut T>, InstructionError> {
@ -215,7 +215,7 @@ impl Runtime {
) -> Result<(), TransactionError> { ) -> Result<(), TransactionError> {
for (instruction_index, instruction) in tx.instructions.iter().enumerate() { for (instruction_index, instruction) in tx.instructions.iter().enumerate() {
let executable_accounts = &mut loaders[instruction.program_ids_index as usize]; let executable_accounts = &mut loaders[instruction.program_ids_index as usize];
let mut program_accounts = get_subset_unchecked_mut(tx_accounts, &instruction.accounts) let mut program_accounts = get_subset_mut(tx_accounts, &instruction.accounts)
.map_err(|err| TransactionError::InstructionError(instruction_index as u8, err))?; .map_err(|err| TransactionError::InstructionError(instruction_index as u8, err))?;
self.execute_instruction( self.execute_instruction(
tx, tx,
@ -242,31 +242,27 @@ mod tests {
} }
#[test] #[test]
fn test_get_subset_unchecked_mut() { fn test_get_subset_mut() {
assert_eq!(get_subset_mut(&mut [7, 8], &[0]).unwrap(), vec![&mut 7]);
assert_eq!( assert_eq!(
get_subset_unchecked_mut(&mut [7, 8], &[0]).unwrap(), get_subset_mut(&mut [7, 8], &[0, 1]).unwrap(),
vec![&mut 7]
);
assert_eq!(
get_subset_unchecked_mut(&mut [7, 8], &[0, 1]).unwrap(),
vec![&mut 7, &mut 8] vec![&mut 7, &mut 8]
); );
} }
#[test] #[test]
fn test_get_subset_unchecked_mut_duplicate_index() { fn test_get_subset_mut_duplicate_index() {
// This panics, because it assumes duplicate detection is done elsewhere.
assert_eq!( assert_eq!(
get_subset_unchecked_mut(&mut [7, 8], &[0, 0]).unwrap_err(), get_subset_mut(&mut [7, 8], &[0, 0]).unwrap_err(),
InstructionError::DuplicateAccountIndex InstructionError::DuplicateAccountIndex
); );
} }
#[test] #[test]
#[should_panic] #[should_panic]
fn test_get_subset_unchecked_mut_out_of_bounds() { fn test_get_subset_mut_out_of_bounds() {
// This panics, because it assumes bounds validation is done elsewhere. // This panics, because it assumes bounds validation is done elsewhere.
get_subset_unchecked_mut(&mut [7, 8], &[2]).unwrap(); get_subset_mut(&mut [7, 8], &[2]).unwrap();
} }
#[test] #[test]