fix region checks (#11651)

This commit is contained in:
Jack May
2020-08-16 23:11:52 -07:00
committed by GitHub
parent 23fa84b322
commit 768b386f0a
2 changed files with 4 additions and 7 deletions

View File

@ -365,10 +365,10 @@ pub fn syscall_create_program_address(
)
})
.collect::<Result<Vec<_>, EbpfError<BPFError>>>()?;
let program_id = translate_type!(Pubkey, program_id_addr, rw_regions)?;
let program_id = translate_type!(Pubkey, program_id_addr, ro_regions)?;
let new_address =
Pubkey::create_program_address(&seeds, program_id).map_err(SyscallError::BadSeeds)?;
let address = translate_slice_mut!(u8, address_addr, 32, ro_regions)?;
let address = translate_slice_mut!(u8, address_addr, 32, rw_regions)?;
address.copy_from_slice(new_address.as_ref());
Ok(0)
}

View File

@ -139,16 +139,13 @@ impl Pubkey {
address_bytes_addr: *const u8,
) -> u64;
};
let bytes = [
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0,
];
let mut bytes = [0; 32];
let result = unsafe {
sol_create_program_address(
seeds as *const _ as *const u8,
seeds.len() as u64,
program_id as *const _ as *const u8,
&bytes as *const _ as *const u8,
&mut bytes as *mut _ as *mut u8,
)
};
match result {