Skip to content

Commit d1deccd

Browse files
committed
refactor: use check_data_is_zeroed set_discriminator
1 parent ee8304f commit d1deccd

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

program-libs/account-checks/src/checks.rs

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -77,9 +77,8 @@ pub fn check_account_info<T: Discriminator>(
7777
/// 1. discriminator is uninitialized
7878
/// 2. sets discriminator
7979
pub fn set_discriminator<T: Discriminator>(bytes: &mut [u8]) -> Result<(), AccountError> {
80-
if bytes[0..DISCRIMINATOR_LEN] != [0; DISCRIMINATOR_LEN] {
81-
return Err(AccountError::AlreadyInitialized);
82-
}
80+
check_data_is_zeroed::<DISCRIMINATOR_LEN>(bytes)
81+
.map_err(|_| AccountError::AlreadyInitialized)?;
8382
bytes[0..DISCRIMINATOR_LEN].copy_from_slice(&T::DISCRIMINATOR);
8483
Ok(())
8584
}

0 commit comments

Comments
 (0)