2021-12-03 09:00:31 -08:00
|
|
|
use {
|
|
|
|
byteorder::{ByteOrder, LittleEndian, WriteBytesExt},
|
|
|
|
solana_rbpf::{aligned_memory::AlignedMemory, ebpf::HOST_ALIGN},
|
|
|
|
solana_sdk::{
|
|
|
|
bpf_loader_deprecated,
|
|
|
|
entrypoint::{BPF_ALIGN_OF_U128, MAX_PERMITTED_DATA_INCREASE},
|
|
|
|
instruction::InstructionError,
|
|
|
|
pubkey::Pubkey,
|
|
|
|
system_instruction::MAX_PERMITTED_DATA_LENGTH,
|
2022-01-03 14:30:56 -08:00
|
|
|
transaction_context::{InstructionContext, TransactionContext},
|
2021-12-03 09:00:31 -08:00
|
|
|
},
|
|
|
|
std::{io::prelude::*, mem::size_of},
|
2020-08-11 16:11:52 -07:00
|
|
|
};
|
|
|
|
|
|
|
|
/// Look for a duplicate account and return its position if found
|
2022-01-03 14:30:56 -08:00
|
|
|
pub fn is_duplicate(
|
|
|
|
instruction_context: &InstructionContext,
|
|
|
|
index_in_instruction: usize,
|
|
|
|
) -> Option<usize> {
|
|
|
|
let index_in_transaction = instruction_context.get_index_in_transaction(index_in_instruction);
|
|
|
|
(instruction_context.get_number_of_program_accounts()..index_in_instruction).position(
|
|
|
|
|index_in_instruction| {
|
|
|
|
instruction_context.get_index_in_transaction(index_in_instruction)
|
|
|
|
== index_in_transaction
|
|
|
|
},
|
|
|
|
)
|
2020-08-11 16:11:52 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
pub fn serialize_parameters(
|
2022-01-03 14:30:56 -08:00
|
|
|
transaction_context: &TransactionContext,
|
|
|
|
instruction_context: &InstructionContext,
|
2021-09-17 23:09:47 -07:00
|
|
|
) -> Result<(AlignedMemory, Vec<usize>), InstructionError> {
|
2022-01-03 14:30:56 -08:00
|
|
|
let is_loader_deprecated = *instruction_context
|
|
|
|
.try_borrow_program_account(transaction_context)?
|
|
|
|
.get_owner()
|
|
|
|
== bpf_loader_deprecated::id();
|
|
|
|
if is_loader_deprecated {
|
|
|
|
serialize_parameters_unaligned(transaction_context, instruction_context)
|
2020-08-11 16:11:52 -07:00
|
|
|
} else {
|
2022-01-03 14:30:56 -08:00
|
|
|
serialize_parameters_aligned(transaction_context, instruction_context)
|
2020-08-11 16:11:52 -07:00
|
|
|
}
|
2021-09-17 23:09:47 -07:00
|
|
|
.and_then(|buffer| {
|
2022-01-03 14:30:56 -08:00
|
|
|
let account_lengths = (instruction_context.get_number_of_program_accounts()
|
|
|
|
..instruction_context.get_number_of_accounts())
|
|
|
|
.map(|index_in_instruction| {
|
|
|
|
Ok(instruction_context
|
|
|
|
.try_borrow_account(transaction_context, index_in_instruction)?
|
|
|
|
.get_data()
|
|
|
|
.len())
|
|
|
|
})
|
2021-09-17 23:09:47 -07:00
|
|
|
.collect::<Result<Vec<usize>, InstructionError>>()?;
|
|
|
|
Ok((buffer, account_lengths))
|
|
|
|
})
|
2020-08-11 16:11:52 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
pub fn deserialize_parameters(
|
2022-01-03 14:30:56 -08:00
|
|
|
transaction_context: &TransactionContext,
|
|
|
|
instruction_context: &InstructionContext,
|
2020-08-11 16:11:52 -07:00
|
|
|
buffer: &[u8],
|
2021-09-17 23:09:47 -07:00
|
|
|
account_lengths: &[usize],
|
2021-09-28 01:13:03 -07:00
|
|
|
do_support_realloc: bool,
|
2020-08-11 16:11:52 -07:00
|
|
|
) -> Result<(), InstructionError> {
|
2022-01-03 14:30:56 -08:00
|
|
|
let is_loader_deprecated = *instruction_context
|
|
|
|
.try_borrow_program_account(transaction_context)?
|
|
|
|
.get_owner()
|
|
|
|
== bpf_loader_deprecated::id();
|
|
|
|
if is_loader_deprecated {
|
2022-01-05 00:39:37 -08:00
|
|
|
deserialize_parameters_unaligned(
|
|
|
|
transaction_context,
|
|
|
|
instruction_context,
|
|
|
|
buffer,
|
|
|
|
account_lengths,
|
|
|
|
)
|
2020-08-11 16:11:52 -07:00
|
|
|
} else {
|
2022-01-03 14:30:56 -08:00
|
|
|
deserialize_parameters_aligned(
|
|
|
|
transaction_context,
|
|
|
|
instruction_context,
|
|
|
|
buffer,
|
|
|
|
account_lengths,
|
|
|
|
do_support_realloc,
|
|
|
|
)
|
2020-08-11 16:11:52 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn serialize_parameters_unaligned(
|
2022-01-03 14:30:56 -08:00
|
|
|
transaction_context: &TransactionContext,
|
|
|
|
instruction_context: &InstructionContext,
|
2021-04-20 11:07:30 -07:00
|
|
|
) -> Result<AlignedMemory, InstructionError> {
|
2020-09-10 11:13:35 -07:00
|
|
|
// Calculate size in order to alloc once
|
|
|
|
let mut size = size_of::<u64>();
|
2022-01-03 14:30:56 -08:00
|
|
|
for index_in_instruction in instruction_context.get_number_of_program_accounts()
|
|
|
|
..instruction_context.get_number_of_accounts()
|
|
|
|
{
|
|
|
|
let duplicate = is_duplicate(instruction_context, index_in_instruction);
|
2021-03-11 19:28:21 -08:00
|
|
|
size += 1; // dup
|
2022-01-03 14:30:56 -08:00
|
|
|
if duplicate.is_none() {
|
|
|
|
let data_len = instruction_context
|
|
|
|
.try_borrow_account(transaction_context, index_in_instruction)?
|
|
|
|
.get_data()
|
|
|
|
.len();
|
|
|
|
size += size_of::<u8>() // is_signer
|
|
|
|
+ size_of::<u8>() // is_writable
|
|
|
|
+ size_of::<Pubkey>() // key
|
|
|
|
+ size_of::<u64>() // lamports
|
|
|
|
+ size_of::<u64>() // data len
|
|
|
|
+ data_len // data
|
|
|
|
+ size_of::<Pubkey>() // owner
|
|
|
|
+ size_of::<u8>() // executable
|
|
|
|
+ size_of::<u64>(); // rent_epoch
|
2020-09-10 11:13:35 -07:00
|
|
|
}
|
|
|
|
}
|
2021-03-11 19:28:21 -08:00
|
|
|
size += size_of::<u64>() // instruction data len
|
2022-01-03 14:30:56 -08:00
|
|
|
+ instruction_context.get_instruction_data().len() // instruction data
|
2021-03-11 19:28:21 -08:00
|
|
|
+ size_of::<Pubkey>(); // program id
|
2021-04-20 11:07:30 -07:00
|
|
|
let mut v = AlignedMemory::new(size, HOST_ALIGN);
|
2020-09-10 11:13:35 -07:00
|
|
|
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_u64::<LittleEndian>(instruction_context.get_number_of_instruction_accounts() as u64)
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
for index_in_instruction in instruction_context.get_number_of_program_accounts()
|
|
|
|
..instruction_context.get_number_of_accounts()
|
|
|
|
{
|
|
|
|
let duplicate = is_duplicate(instruction_context, index_in_instruction);
|
|
|
|
if let Some(position) = duplicate {
|
2021-04-19 13:30:31 -07:00
|
|
|
v.write_u8(position as u8)
|
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2020-08-11 16:11:52 -07:00
|
|
|
} else {
|
2022-01-03 14:30:56 -08:00
|
|
|
let borrowed_account = instruction_context
|
|
|
|
.try_borrow_account(transaction_context, index_in_instruction)?;
|
2021-04-19 13:30:31 -07:00
|
|
|
v.write_u8(std::u8::MAX)
|
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_u8(borrowed_account.is_signer() as u8)
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_u8(borrowed_account.is_writable() as u8)
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_all(borrowed_account.get_key().as_ref())
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_u64::<LittleEndian>(borrowed_account.get_lamports())
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_u64::<LittleEndian>(borrowed_account.get_data().len() as u64)
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_all(borrowed_account.get_data())
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_all(borrowed_account.get_owner().as_ref())
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_u8(borrowed_account.is_executable() as u8)
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_u64::<LittleEndian>(borrowed_account.get_rent_epoch() as u64)
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2020-08-11 16:11:52 -07:00
|
|
|
}
|
|
|
|
}
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_u64::<LittleEndian>(instruction_context.get_instruction_data().len() as u64)
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_all(instruction_context.get_instruction_data())
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_all(
|
|
|
|
instruction_context
|
|
|
|
.try_borrow_program_account(transaction_context)?
|
|
|
|
.get_key()
|
|
|
|
.as_ref(),
|
|
|
|
)
|
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2020-08-11 16:11:52 -07:00
|
|
|
Ok(v)
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn deserialize_parameters_unaligned(
|
2022-01-03 14:30:56 -08:00
|
|
|
transaction_context: &TransactionContext,
|
|
|
|
instruction_context: &InstructionContext,
|
2020-08-11 16:11:52 -07:00
|
|
|
buffer: &[u8],
|
2022-01-05 00:39:37 -08:00
|
|
|
account_lengths: &[usize],
|
2020-08-11 16:11:52 -07:00
|
|
|
) -> Result<(), InstructionError> {
|
2020-08-17 13:38:42 -07:00
|
|
|
let mut start = size_of::<u64>(); // number of accounts
|
2022-01-05 00:39:37 -08:00
|
|
|
for (index_in_instruction, pre_len) in (instruction_context.get_number_of_program_accounts()
|
|
|
|
..instruction_context.get_number_of_accounts())
|
|
|
|
.zip(account_lengths.iter())
|
2021-09-17 23:09:47 -07:00
|
|
|
{
|
2022-01-03 14:30:56 -08:00
|
|
|
let duplicate = is_duplicate(instruction_context, index_in_instruction);
|
2020-08-11 16:11:52 -07:00
|
|
|
start += 1; // is_dup
|
2022-01-03 14:30:56 -08:00
|
|
|
if duplicate.is_none() {
|
|
|
|
let mut borrowed_account = instruction_context
|
|
|
|
.try_borrow_account(transaction_context, index_in_instruction)?;
|
2021-05-28 09:50:25 -07:00
|
|
|
start += size_of::<u8>(); // is_signer
|
|
|
|
start += size_of::<u8>(); // is_writable
|
|
|
|
start += size_of::<Pubkey>(); // key
|
2022-01-03 14:30:56 -08:00
|
|
|
let _ = borrowed_account.set_lamports(LittleEndian::read_u64(&buffer[start..]));
|
2021-05-28 09:50:25 -07:00
|
|
|
start += size_of::<u64>() // lamports
|
2020-08-17 13:38:42 -07:00
|
|
|
+ size_of::<u64>(); // data length
|
2022-01-05 00:39:37 -08:00
|
|
|
let _ = borrowed_account.set_data(&buffer[start..start + pre_len]);
|
|
|
|
start += pre_len // data
|
2020-08-17 13:38:42 -07:00
|
|
|
+ size_of::<Pubkey>() // owner
|
|
|
|
+ size_of::<u8>() // executable
|
|
|
|
+ size_of::<u64>(); // rent_epoch
|
2020-08-11 16:11:52 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn serialize_parameters_aligned(
|
2022-01-03 14:30:56 -08:00
|
|
|
transaction_context: &TransactionContext,
|
|
|
|
instruction_context: &InstructionContext,
|
2021-04-20 11:07:30 -07:00
|
|
|
) -> Result<AlignedMemory, InstructionError> {
|
2020-09-10 11:13:35 -07:00
|
|
|
// Calculate size in order to alloc once
|
|
|
|
let mut size = size_of::<u64>();
|
2022-01-03 14:30:56 -08:00
|
|
|
for index_in_instruction in instruction_context.get_number_of_program_accounts()
|
|
|
|
..instruction_context.get_number_of_accounts()
|
|
|
|
{
|
|
|
|
let duplicate = is_duplicate(instruction_context, index_in_instruction);
|
2021-03-11 19:28:21 -08:00
|
|
|
size += 1; // dup
|
2022-01-03 14:30:56 -08:00
|
|
|
if duplicate.is_some() {
|
2021-03-11 19:28:21 -08:00
|
|
|
size += 7; // padding to 64-bit aligned
|
|
|
|
} else {
|
2022-01-03 14:30:56 -08:00
|
|
|
let data_len = instruction_context
|
|
|
|
.try_borrow_account(transaction_context, index_in_instruction)?
|
|
|
|
.get_data()
|
|
|
|
.len();
|
|
|
|
size += size_of::<u8>() // is_signer
|
|
|
|
+ size_of::<u8>() // is_writable
|
|
|
|
+ size_of::<u8>() // executable
|
|
|
|
+ 4 // padding to 128-bit aligned
|
|
|
|
+ size_of::<Pubkey>() // key
|
|
|
|
+ size_of::<Pubkey>() // owner
|
|
|
|
+ size_of::<u64>() // lamports
|
|
|
|
+ size_of::<u64>() // data len
|
|
|
|
+ data_len
|
|
|
|
+ MAX_PERMITTED_DATA_INCREASE
|
|
|
|
+ (data_len as *const u8).align_offset(BPF_ALIGN_OF_U128)
|
|
|
|
+ size_of::<u64>(); // rent epoch
|
2020-09-10 11:13:35 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
size += size_of::<u64>() // data len
|
2022-01-03 14:30:56 -08:00
|
|
|
+ instruction_context.get_instruction_data().len()
|
2020-09-10 11:13:35 -07:00
|
|
|
+ size_of::<Pubkey>(); // program id;
|
2021-04-20 11:07:30 -07:00
|
|
|
let mut v = AlignedMemory::new(size, HOST_ALIGN);
|
2020-09-10 11:13:35 -07:00
|
|
|
|
|
|
|
// Serialize into the buffer
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_u64::<LittleEndian>(instruction_context.get_number_of_instruction_accounts() as u64)
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
for index_in_instruction in instruction_context.get_number_of_program_accounts()
|
|
|
|
..instruction_context.get_number_of_accounts()
|
|
|
|
{
|
|
|
|
let duplicate = is_duplicate(instruction_context, index_in_instruction);
|
|
|
|
if let Some(position) = duplicate {
|
2021-04-19 13:30:31 -07:00
|
|
|
v.write_u8(position as u8)
|
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
|
|
|
v.write_all(&[0u8, 0, 0, 0, 0, 0, 0])
|
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?; // 7 bytes of padding to make 64-bit aligned
|
2020-08-11 16:11:52 -07:00
|
|
|
} else {
|
2022-01-03 14:30:56 -08:00
|
|
|
let borrowed_account = instruction_context
|
|
|
|
.try_borrow_account(transaction_context, index_in_instruction)?;
|
2021-04-19 13:30:31 -07:00
|
|
|
v.write_u8(std::u8::MAX)
|
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_u8(borrowed_account.is_signer() as u8)
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_u8(borrowed_account.is_writable() as u8)
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_u8(borrowed_account.is_executable() as u8)
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
|
|
|
v.write_all(&[0u8, 0, 0, 0])
|
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?; // 4 bytes of padding to make 128-bit aligned
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_all(borrowed_account.get_key().as_ref())
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_all(borrowed_account.get_owner().as_ref())
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_u64::<LittleEndian>(borrowed_account.get_lamports())
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_u64::<LittleEndian>(borrowed_account.get_data().len() as u64)
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_all(borrowed_account.get_data())
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2021-05-19 13:43:59 -07:00
|
|
|
v.resize(
|
2021-04-20 11:07:30 -07:00
|
|
|
MAX_PERMITTED_DATA_INCREASE
|
2021-11-30 10:41:15 -08:00
|
|
|
+ (v.write_index() as *const u8).align_offset(BPF_ALIGN_OF_U128),
|
2020-08-17 10:24:34 -07:00
|
|
|
0,
|
2021-04-20 11:07:30 -07:00
|
|
|
)
|
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_u64::<LittleEndian>(borrowed_account.get_rent_epoch() as u64)
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2020-08-11 16:11:52 -07:00
|
|
|
}
|
|
|
|
}
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_u64::<LittleEndian>(instruction_context.get_instruction_data().len() as u64)
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_all(instruction_context.get_instruction_data())
|
2021-04-19 13:30:31 -07:00
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2022-01-03 14:30:56 -08:00
|
|
|
v.write_all(
|
|
|
|
instruction_context
|
|
|
|
.try_borrow_program_account(transaction_context)?
|
|
|
|
.get_key()
|
|
|
|
.as_ref(),
|
|
|
|
)
|
|
|
|
.map_err(|_| InstructionError::InvalidArgument)?;
|
2020-08-11 16:11:52 -07:00
|
|
|
Ok(v)
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn deserialize_parameters_aligned(
|
2022-01-03 14:30:56 -08:00
|
|
|
transaction_context: &TransactionContext,
|
|
|
|
instruction_context: &InstructionContext,
|
2020-08-11 16:11:52 -07:00
|
|
|
buffer: &[u8],
|
2021-09-17 23:09:47 -07:00
|
|
|
account_lengths: &[usize],
|
2021-09-28 01:13:03 -07:00
|
|
|
do_support_realloc: bool,
|
2020-08-11 16:11:52 -07:00
|
|
|
) -> Result<(), InstructionError> {
|
2020-08-17 13:38:42 -07:00
|
|
|
let mut start = size_of::<u64>(); // number of accounts
|
2022-01-03 14:30:56 -08:00
|
|
|
for (index_in_instruction, pre_len) in (instruction_context.get_number_of_program_accounts()
|
|
|
|
..instruction_context.get_number_of_accounts())
|
2021-09-17 23:09:47 -07:00
|
|
|
.zip(account_lengths.iter())
|
|
|
|
{
|
2022-01-03 14:30:56 -08:00
|
|
|
let duplicate = is_duplicate(instruction_context, index_in_instruction);
|
2020-08-17 13:38:42 -07:00
|
|
|
start += size_of::<u8>(); // position
|
2022-01-03 14:30:56 -08:00
|
|
|
if duplicate.is_some() {
|
2020-08-17 13:38:42 -07:00
|
|
|
start += 7; // padding to 64-bit aligned
|
2021-05-28 09:50:25 -07:00
|
|
|
} else {
|
2022-01-03 14:30:56 -08:00
|
|
|
let mut borrowed_account = instruction_context
|
|
|
|
.try_borrow_account(transaction_context, index_in_instruction)?;
|
2020-08-17 13:38:42 -07:00
|
|
|
start += size_of::<u8>() // is_signer
|
2020-09-10 11:13:35 -07:00
|
|
|
+ size_of::<u8>() // is_writable
|
|
|
|
+ size_of::<u8>() // executable
|
|
|
|
+ 4 // padding to 128-bit aligned
|
|
|
|
+ size_of::<Pubkey>(); // key
|
2022-01-03 14:30:56 -08:00
|
|
|
let _ = borrowed_account.set_owner(&buffer[start..start + size_of::<Pubkey>()]);
|
2020-08-17 13:38:42 -07:00
|
|
|
start += size_of::<Pubkey>(); // owner
|
2022-01-03 14:30:56 -08:00
|
|
|
let _ = borrowed_account.set_lamports(LittleEndian::read_u64(&buffer[start..]));
|
2020-08-17 13:38:42 -07:00
|
|
|
start += size_of::<u64>(); // lamports
|
|
|
|
let post_len = LittleEndian::read_u64(&buffer[start..]) as usize;
|
|
|
|
start += size_of::<u64>(); // data length
|
2021-09-28 01:13:03 -07:00
|
|
|
let data_end = if do_support_realloc {
|
|
|
|
if post_len.saturating_sub(*pre_len) > MAX_PERMITTED_DATA_INCREASE
|
|
|
|
|| post_len > MAX_PERMITTED_DATA_LENGTH as usize
|
|
|
|
{
|
|
|
|
return Err(InstructionError::InvalidRealloc);
|
|
|
|
}
|
|
|
|
start + post_len
|
|
|
|
} else {
|
|
|
|
let mut data_end = start + *pre_len;
|
|
|
|
if post_len != *pre_len
|
|
|
|
&& (post_len.saturating_sub(*pre_len)) <= MAX_PERMITTED_DATA_INCREASE
|
|
|
|
{
|
|
|
|
data_end = start + post_len;
|
|
|
|
}
|
|
|
|
data_end
|
|
|
|
};
|
2022-01-03 14:30:56 -08:00
|
|
|
let _ = borrowed_account.set_data(&buffer[start..data_end]);
|
2021-09-17 23:09:47 -07:00
|
|
|
start += *pre_len + MAX_PERMITTED_DATA_INCREASE; // data
|
2021-11-30 10:41:15 -08:00
|
|
|
start += (start as *const u8).align_offset(BPF_ALIGN_OF_U128);
|
2020-08-17 13:38:42 -07:00
|
|
|
start += size_of::<u64>(); // rent_epoch
|
2020-08-11 16:11:52 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
2021-12-03 09:00:31 -08:00
|
|
|
use {
|
|
|
|
super::*,
|
|
|
|
solana_program_runtime::invoke_context::{prepare_mock_invoke_context, InvokeContext},
|
|
|
|
solana_sdk::{
|
2022-01-05 00:39:37 -08:00
|
|
|
account::{Account, AccountSharedData, ReadableAccount, WritableAccount},
|
2021-12-03 09:00:31 -08:00
|
|
|
account_info::AccountInfo,
|
|
|
|
bpf_loader,
|
|
|
|
entrypoint::deserialize,
|
2021-12-17 05:01:12 -08:00
|
|
|
instruction::AccountMeta,
|
2021-12-03 09:00:31 -08:00
|
|
|
},
|
|
|
|
std::{
|
|
|
|
cell::RefCell,
|
|
|
|
rc::Rc,
|
|
|
|
slice::{from_raw_parts, from_raw_parts_mut},
|
|
|
|
},
|
2020-08-11 16:11:52 -07:00
|
|
|
};
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_serialize_parameters() {
|
2020-10-19 12:12:08 -07:00
|
|
|
let program_id = solana_sdk::pubkey::new_rand();
|
2021-12-17 05:01:12 -08:00
|
|
|
let transaction_accounts = vec![
|
2021-11-08 09:51:36 -08:00
|
|
|
(
|
|
|
|
program_id,
|
2021-12-17 05:01:12 -08:00
|
|
|
AccountSharedData::from(Account {
|
2021-11-08 09:51:36 -08:00
|
|
|
lamports: 0,
|
|
|
|
data: vec![],
|
|
|
|
owner: bpf_loader::id(),
|
|
|
|
executable: true,
|
|
|
|
rent_epoch: 0,
|
2021-12-17 05:01:12 -08:00
|
|
|
}),
|
2021-11-08 09:51:36 -08:00
|
|
|
),
|
|
|
|
(
|
2021-12-17 05:01:12 -08:00
|
|
|
solana_sdk::pubkey::new_rand(),
|
|
|
|
AccountSharedData::from(Account {
|
2021-11-08 09:51:36 -08:00
|
|
|
lamports: 1,
|
|
|
|
data: vec![1u8, 2, 3, 4, 5],
|
|
|
|
owner: bpf_loader::id(),
|
|
|
|
executable: false,
|
|
|
|
rent_epoch: 100,
|
2021-12-17 05:01:12 -08:00
|
|
|
}),
|
2021-11-08 09:51:36 -08:00
|
|
|
),
|
|
|
|
(
|
|
|
|
solana_sdk::pubkey::new_rand(),
|
2021-12-17 05:01:12 -08:00
|
|
|
AccountSharedData::from(Account {
|
2021-11-08 09:51:36 -08:00
|
|
|
lamports: 2,
|
|
|
|
data: vec![11u8, 12, 13, 14, 15, 16, 17, 18, 19],
|
|
|
|
owner: bpf_loader::id(),
|
|
|
|
executable: true,
|
|
|
|
rent_epoch: 200,
|
2021-12-17 05:01:12 -08:00
|
|
|
}),
|
2021-11-08 09:51:36 -08:00
|
|
|
),
|
|
|
|
(
|
|
|
|
solana_sdk::pubkey::new_rand(),
|
2021-12-17 05:01:12 -08:00
|
|
|
AccountSharedData::from(Account {
|
2021-11-08 09:51:36 -08:00
|
|
|
lamports: 3,
|
|
|
|
data: vec![],
|
|
|
|
owner: bpf_loader::id(),
|
|
|
|
executable: false,
|
|
|
|
rent_epoch: 3100,
|
2021-12-17 05:01:12 -08:00
|
|
|
}),
|
2021-11-08 09:51:36 -08:00
|
|
|
),
|
|
|
|
(
|
2021-12-17 05:01:12 -08:00
|
|
|
solana_sdk::pubkey::new_rand(),
|
|
|
|
AccountSharedData::from(Account {
|
2021-11-08 09:51:36 -08:00
|
|
|
lamports: 4,
|
|
|
|
data: vec![1u8, 2, 3, 4, 5],
|
|
|
|
owner: bpf_loader::id(),
|
|
|
|
executable: false,
|
|
|
|
rent_epoch: 100,
|
2021-12-17 05:01:12 -08:00
|
|
|
}),
|
2021-11-08 09:51:36 -08:00
|
|
|
),
|
|
|
|
(
|
|
|
|
solana_sdk::pubkey::new_rand(),
|
2021-12-17 05:01:12 -08:00
|
|
|
AccountSharedData::from(Account {
|
2021-11-08 09:51:36 -08:00
|
|
|
lamports: 5,
|
|
|
|
data: vec![11u8, 12, 13, 14, 15, 16, 17, 18, 19],
|
|
|
|
owner: bpf_loader::id(),
|
|
|
|
executable: true,
|
|
|
|
rent_epoch: 200,
|
2021-12-17 05:01:12 -08:00
|
|
|
}),
|
2021-11-08 09:51:36 -08:00
|
|
|
),
|
|
|
|
(
|
|
|
|
solana_sdk::pubkey::new_rand(),
|
2021-12-17 05:01:12 -08:00
|
|
|
AccountSharedData::from(Account {
|
2021-11-08 09:51:36 -08:00
|
|
|
lamports: 6,
|
|
|
|
data: vec![],
|
|
|
|
owner: bpf_loader::id(),
|
|
|
|
executable: false,
|
|
|
|
rent_epoch: 3100,
|
2021-12-17 05:01:12 -08:00
|
|
|
}),
|
2021-11-08 09:51:36 -08:00
|
|
|
),
|
2020-10-19 12:23:14 -07:00
|
|
|
];
|
2022-01-03 14:30:56 -08:00
|
|
|
let instruction_accounts = [1, 1, 2, 3, 4, 4, 5, 6]
|
|
|
|
.into_iter()
|
|
|
|
.enumerate()
|
|
|
|
.map(|(index_in_instruction, index_in_transaction)| AccountMeta {
|
|
|
|
pubkey: transaction_accounts[index_in_transaction].0,
|
2021-12-17 05:01:12 -08:00
|
|
|
is_signer: false,
|
2022-01-03 14:30:56 -08:00
|
|
|
is_writable: index_in_instruction >= 4,
|
|
|
|
})
|
|
|
|
.collect();
|
2020-08-11 16:11:52 -07:00
|
|
|
let instruction_data = vec![1u8, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11];
|
2021-11-08 09:51:36 -08:00
|
|
|
let program_indices = [0];
|
2022-01-05 00:39:37 -08:00
|
|
|
let mut original_accounts = transaction_accounts.clone();
|
2021-12-30 06:46:36 -08:00
|
|
|
let preparation = prepare_mock_invoke_context(
|
2022-01-03 14:30:56 -08:00
|
|
|
transaction_accounts,
|
2021-12-30 06:46:36 -08:00
|
|
|
instruction_accounts,
|
|
|
|
&program_indices,
|
|
|
|
);
|
2022-01-19 13:40:09 -08:00
|
|
|
let mut transaction_context =
|
|
|
|
TransactionContext::new(preparation.transaction_accounts, 1, 1);
|
2022-01-03 14:30:56 -08:00
|
|
|
let mut invoke_context = InvokeContext::new_mock(&mut transaction_context, &[]);
|
2021-11-08 09:51:36 -08:00
|
|
|
invoke_context
|
2022-01-03 14:30:56 -08:00
|
|
|
.push(
|
|
|
|
&preparation.instruction_accounts,
|
|
|
|
&program_indices,
|
|
|
|
&instruction_data,
|
|
|
|
)
|
|
|
|
.unwrap();
|
|
|
|
let instruction_context = invoke_context
|
|
|
|
.transaction_context
|
|
|
|
.get_current_instruction_context()
|
2021-11-08 09:51:36 -08:00
|
|
|
.unwrap();
|
2020-08-11 16:11:52 -07:00
|
|
|
|
|
|
|
// check serialize_parameters_aligned
|
|
|
|
|
2022-01-03 14:30:56 -08:00
|
|
|
let (mut serialized, account_lengths) =
|
|
|
|
serialize_parameters(invoke_context.transaction_context, instruction_context).unwrap();
|
2021-03-11 19:28:21 -08:00
|
|
|
|
2020-08-11 16:11:52 -07:00
|
|
|
let (de_program_id, de_accounts, de_instruction_data) =
|
2021-04-20 11:07:30 -07:00
|
|
|
unsafe { deserialize(&mut serialized.as_slice_mut()[0] as *mut u8) };
|
|
|
|
|
2020-08-11 16:11:52 -07:00
|
|
|
assert_eq!(&program_id, de_program_id);
|
|
|
|
assert_eq!(instruction_data, de_instruction_data);
|
|
|
|
assert_eq!(
|
2021-11-30 10:41:15 -08:00
|
|
|
(&de_instruction_data[0] as *const u8).align_offset(BPF_ALIGN_OF_U128),
|
2020-08-11 16:11:52 -07:00
|
|
|
0
|
|
|
|
);
|
2021-12-17 05:01:12 -08:00
|
|
|
for account_info in de_accounts {
|
2022-01-03 14:30:56 -08:00
|
|
|
let index_in_transaction = invoke_context
|
|
|
|
.transaction_context
|
|
|
|
.find_index_of_account(account_info.key)
|
|
|
|
.unwrap();
|
|
|
|
let account = invoke_context
|
|
|
|
.transaction_context
|
|
|
|
.get_account_at_index(index_in_transaction)
|
|
|
|
.borrow();
|
2021-04-30 11:16:58 -07:00
|
|
|
assert_eq!(account.lamports(), account_info.lamports());
|
2021-04-20 14:41:16 -07:00
|
|
|
assert_eq!(account.data(), &account_info.data.borrow()[..]);
|
2021-04-26 08:40:11 -07:00
|
|
|
assert_eq!(account.owner(), account_info.owner);
|
2021-04-28 06:53:54 -07:00
|
|
|
assert_eq!(account.executable(), account_info.executable);
|
2021-04-27 11:51:13 -07:00
|
|
|
assert_eq!(account.rent_epoch(), account_info.rent_epoch);
|
2020-08-11 16:11:52 -07:00
|
|
|
|
|
|
|
assert_eq!(
|
2021-11-30 10:41:15 -08:00
|
|
|
(*account_info.lamports.borrow() as *const u64).align_offset(BPF_ALIGN_OF_U128),
|
2020-08-11 16:11:52 -07:00
|
|
|
0
|
|
|
|
);
|
|
|
|
assert_eq!(
|
|
|
|
account_info
|
|
|
|
.data
|
|
|
|
.borrow()
|
|
|
|
.as_ptr()
|
2021-11-30 10:41:15 -08:00
|
|
|
.align_offset(BPF_ALIGN_OF_U128),
|
2020-08-11 16:11:52 -07:00
|
|
|
0
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
2022-01-05 00:39:37 -08:00
|
|
|
for index_in_transaction in 1..original_accounts.len() {
|
|
|
|
let mut account = invoke_context
|
|
|
|
.transaction_context
|
|
|
|
.get_account_at_index(index_in_transaction)
|
|
|
|
.borrow_mut();
|
|
|
|
account.set_lamports(0);
|
|
|
|
account.set_data(vec![0; 0]);
|
|
|
|
account.set_owner(Pubkey::default());
|
|
|
|
}
|
2021-09-17 23:09:47 -07:00
|
|
|
deserialize_parameters(
|
2022-01-03 14:30:56 -08:00
|
|
|
invoke_context.transaction_context,
|
|
|
|
instruction_context,
|
2021-09-17 23:09:47 -07:00
|
|
|
serialized.as_slice(),
|
|
|
|
&account_lengths,
|
2021-09-28 01:13:03 -07:00
|
|
|
true,
|
2021-09-17 23:09:47 -07:00
|
|
|
)
|
|
|
|
.unwrap();
|
2022-01-05 00:39:37 -08:00
|
|
|
for (index_in_transaction, (_key, original_account)) in original_accounts.iter().enumerate()
|
|
|
|
{
|
2022-01-03 14:30:56 -08:00
|
|
|
let account = invoke_context
|
|
|
|
.transaction_context
|
|
|
|
.get_account_at_index(index_in_transaction)
|
|
|
|
.borrow();
|
2022-01-05 00:39:37 -08:00
|
|
|
assert_eq!(&*account, original_account);
|
2021-03-11 19:28:21 -08:00
|
|
|
}
|
|
|
|
|
2020-08-11 16:11:52 -07:00
|
|
|
// check serialize_parameters_unaligned
|
2022-01-05 00:39:37 -08:00
|
|
|
original_accounts[0]
|
|
|
|
.1
|
|
|
|
.set_owner(bpf_loader_deprecated::id());
|
2022-01-03 14:30:56 -08:00
|
|
|
let _ = invoke_context
|
|
|
|
.transaction_context
|
|
|
|
.get_current_instruction_context()
|
|
|
|
.unwrap()
|
|
|
|
.try_borrow_account(invoke_context.transaction_context, 0)
|
|
|
|
.unwrap()
|
|
|
|
.set_owner(bpf_loader_deprecated::id().as_ref());
|
2020-08-11 16:11:52 -07:00
|
|
|
|
2022-01-03 14:30:56 -08:00
|
|
|
let (mut serialized, account_lengths) =
|
|
|
|
serialize_parameters(invoke_context.transaction_context, instruction_context).unwrap();
|
2021-03-11 19:28:21 -08:00
|
|
|
|
2020-08-11 16:11:52 -07:00
|
|
|
let (de_program_id, de_accounts, de_instruction_data) =
|
2021-04-20 11:07:30 -07:00
|
|
|
unsafe { deserialize_unaligned(&mut serialized.as_slice_mut()[0] as *mut u8) };
|
2020-08-11 16:11:52 -07:00
|
|
|
assert_eq!(&program_id, de_program_id);
|
|
|
|
assert_eq!(instruction_data, de_instruction_data);
|
2021-12-17 05:01:12 -08:00
|
|
|
for account_info in de_accounts {
|
2022-01-03 14:30:56 -08:00
|
|
|
let index_in_transaction = invoke_context
|
|
|
|
.transaction_context
|
|
|
|
.find_index_of_account(account_info.key)
|
|
|
|
.unwrap();
|
|
|
|
let account = invoke_context
|
|
|
|
.transaction_context
|
|
|
|
.get_account_at_index(index_in_transaction)
|
|
|
|
.borrow();
|
2021-04-30 11:16:58 -07:00
|
|
|
assert_eq!(account.lamports(), account_info.lamports());
|
2021-04-20 14:41:16 -07:00
|
|
|
assert_eq!(account.data(), &account_info.data.borrow()[..]);
|
2021-04-26 08:40:11 -07:00
|
|
|
assert_eq!(account.owner(), account_info.owner);
|
2021-04-28 06:53:54 -07:00
|
|
|
assert_eq!(account.executable(), account_info.executable);
|
2021-04-27 11:51:13 -07:00
|
|
|
assert_eq!(account.rent_epoch(), account_info.rent_epoch);
|
2020-08-11 16:11:52 -07:00
|
|
|
}
|
2021-03-11 19:28:21 -08:00
|
|
|
|
2022-01-05 00:39:37 -08:00
|
|
|
for index_in_transaction in 1..original_accounts.len() {
|
|
|
|
let mut account = invoke_context
|
|
|
|
.transaction_context
|
|
|
|
.get_account_at_index(index_in_transaction)
|
|
|
|
.borrow_mut();
|
|
|
|
account.set_lamports(0);
|
|
|
|
account.set_data(vec![0; 0]);
|
|
|
|
}
|
2021-03-11 19:28:21 -08:00
|
|
|
deserialize_parameters(
|
2022-01-03 14:30:56 -08:00
|
|
|
invoke_context.transaction_context,
|
|
|
|
instruction_context,
|
2021-04-20 11:07:30 -07:00
|
|
|
serialized.as_slice(),
|
2021-09-17 23:09:47 -07:00
|
|
|
&account_lengths,
|
2021-09-28 01:13:03 -07:00
|
|
|
true,
|
2021-03-11 19:28:21 -08:00
|
|
|
)
|
|
|
|
.unwrap();
|
2022-01-05 00:39:37 -08:00
|
|
|
for (index_in_transaction, (_key, original_account)) in original_accounts.iter().enumerate()
|
|
|
|
{
|
2022-01-03 14:30:56 -08:00
|
|
|
let account = invoke_context
|
|
|
|
.transaction_context
|
|
|
|
.get_account_at_index(index_in_transaction)
|
|
|
|
.borrow();
|
2022-01-05 00:39:37 -08:00
|
|
|
assert_eq!(&*account, original_account);
|
2021-03-11 19:28:21 -08:00
|
|
|
}
|
2020-08-11 16:11:52 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// the old bpf_loader in-program deserializer bpf_loader::id()
|
|
|
|
#[allow(clippy::type_complexity)]
|
|
|
|
pub unsafe fn deserialize_unaligned<'a>(
|
|
|
|
input: *mut u8,
|
|
|
|
) -> (&'a Pubkey, Vec<AccountInfo<'a>>, &'a [u8]) {
|
|
|
|
let mut offset: usize = 0;
|
|
|
|
|
|
|
|
// number of accounts present
|
|
|
|
|
|
|
|
#[allow(clippy::cast_ptr_alignment)]
|
|
|
|
let num_accounts = *(input.add(offset) as *const u64) as usize;
|
|
|
|
offset += size_of::<u64>();
|
|
|
|
|
|
|
|
// account Infos
|
|
|
|
|
|
|
|
let mut accounts = Vec::with_capacity(num_accounts);
|
|
|
|
for _ in 0..num_accounts {
|
|
|
|
let dup_info = *(input.add(offset) as *const u8);
|
|
|
|
offset += size_of::<u8>();
|
|
|
|
if dup_info == std::u8::MAX {
|
|
|
|
#[allow(clippy::cast_ptr_alignment)]
|
|
|
|
let is_signer = *(input.add(offset) as *const u8) != 0;
|
|
|
|
offset += size_of::<u8>();
|
|
|
|
|
|
|
|
#[allow(clippy::cast_ptr_alignment)]
|
|
|
|
let is_writable = *(input.add(offset) as *const u8) != 0;
|
|
|
|
offset += size_of::<u8>();
|
|
|
|
|
|
|
|
let key: &Pubkey = &*(input.add(offset) as *const Pubkey);
|
|
|
|
offset += size_of::<Pubkey>();
|
|
|
|
|
|
|
|
#[allow(clippy::cast_ptr_alignment)]
|
|
|
|
let lamports = Rc::new(RefCell::new(&mut *(input.add(offset) as *mut u64)));
|
|
|
|
offset += size_of::<u64>();
|
|
|
|
|
|
|
|
#[allow(clippy::cast_ptr_alignment)]
|
|
|
|
let data_len = *(input.add(offset) as *const u64) as usize;
|
|
|
|
offset += size_of::<u64>();
|
|
|
|
|
|
|
|
let data = Rc::new(RefCell::new({
|
|
|
|
from_raw_parts_mut(input.add(offset), data_len)
|
|
|
|
}));
|
|
|
|
offset += data_len;
|
|
|
|
|
|
|
|
let owner: &Pubkey = &*(input.add(offset) as *const Pubkey);
|
|
|
|
offset += size_of::<Pubkey>();
|
|
|
|
|
|
|
|
#[allow(clippy::cast_ptr_alignment)]
|
|
|
|
let executable = *(input.add(offset) as *const u8) != 0;
|
|
|
|
offset += size_of::<u8>();
|
|
|
|
|
|
|
|
#[allow(clippy::cast_ptr_alignment)]
|
|
|
|
let rent_epoch = *(input.add(offset) as *const u64);
|
|
|
|
offset += size_of::<u64>();
|
|
|
|
|
|
|
|
accounts.push(AccountInfo {
|
2021-04-18 10:27:36 -07:00
|
|
|
key,
|
2020-08-11 16:11:52 -07:00
|
|
|
is_signer,
|
|
|
|
is_writable,
|
|
|
|
lamports,
|
|
|
|
data,
|
|
|
|
owner,
|
|
|
|
executable,
|
|
|
|
rent_epoch,
|
|
|
|
});
|
|
|
|
} else {
|
|
|
|
// duplicate account, clone the original
|
|
|
|
accounts.push(accounts[dup_info as usize].clone());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// instruction data
|
|
|
|
|
|
|
|
#[allow(clippy::cast_ptr_alignment)]
|
|
|
|
let instruction_data_len = *(input.add(offset) as *const u64) as usize;
|
|
|
|
offset += size_of::<u64>();
|
|
|
|
|
|
|
|
let instruction_data = { from_raw_parts(input.add(offset), instruction_data_len) };
|
|
|
|
offset += instruction_data_len;
|
|
|
|
|
|
|
|
// program Id
|
|
|
|
|
|
|
|
let program_id: &Pubkey = &*(input.add(offset) as *const Pubkey);
|
|
|
|
|
|
|
|
(program_id, accounts, instruction_data)
|
|
|
|
}
|
|
|
|
}
|