fix loader instruction checker (#18047)

This commit is contained in:
Jack May 2021-06-18 11:37:39 -07:00 committed by GitHub
parent 491e4450cc
commit d18e02ef44
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions

View File

@ -189,11 +189,11 @@ pub fn upgrade(
}
pub fn is_upgrade_instruction(instruction_data: &[u8]) -> bool {
3 == instruction_data[0]
!instruction_data.is_empty() && 3 == instruction_data[0]
}
pub fn is_set_authority_instruction(instruction_data: &[u8]) -> bool {
4 == instruction_data[0]
!instruction_data.is_empty() && 4 == instruction_data[0]
}
/// Returns the instructions required to set a buffers's authority.
@ -331,6 +331,7 @@ mod tests {
#[test]
fn test_is_set_authority_instruction() {
assert!(!is_set_authority_instruction(&[]));
assert_is_instruction(
is_set_authority_instruction,
UpgradeableLoaderInstruction::SetAuthority {},
@ -339,6 +340,7 @@ mod tests {
#[test]
fn test_is_upgrade_instruction() {
assert!(!is_upgrade_instruction(&[]));
assert_is_instruction(
is_upgrade_instruction,
UpgradeableLoaderInstruction::Upgrade {},