2021-07-16 00:31:22 -07:00
|
|
|
#![cfg(feature = "full")]
|
|
|
|
|
2021-08-17 15:17:56 -07:00
|
|
|
use {
|
2022-01-17 01:43:17 -08:00
|
|
|
crate::{
|
|
|
|
borsh::try_from_slice_unchecked,
|
|
|
|
entrypoint::HEAP_LENGTH as MIN_HEAP_FRAME_BYTES,
|
|
|
|
feature_set::{requestable_heap_size, FeatureSet},
|
|
|
|
instruction::{Instruction, InstructionError},
|
|
|
|
transaction::{SanitizedTransaction, TransactionError},
|
|
|
|
},
|
2021-08-17 15:17:56 -07:00
|
|
|
borsh::{BorshDeserialize, BorshSchema, BorshSerialize},
|
2022-01-17 01:43:17 -08:00
|
|
|
std::sync::Arc,
|
2021-07-16 00:31:22 -07:00
|
|
|
};
|
|
|
|
|
|
|
|
crate::declare_id!("ComputeBudget111111111111111111111111111111");
|
|
|
|
|
2022-01-17 01:43:17 -08:00
|
|
|
const MAX_UNITS: u32 = 1_000_000;
|
|
|
|
const MAX_HEAP_FRAME_BYTES: u32 = 256 * 1024;
|
|
|
|
|
2021-07-16 00:31:22 -07:00
|
|
|
/// Compute Budget Instructions
|
|
|
|
#[derive(
|
|
|
|
Serialize,
|
|
|
|
Deserialize,
|
|
|
|
BorshSerialize,
|
|
|
|
BorshDeserialize,
|
|
|
|
BorshSchema,
|
|
|
|
Debug,
|
|
|
|
Clone,
|
|
|
|
PartialEq,
|
|
|
|
AbiExample,
|
|
|
|
AbiEnumVisitor,
|
|
|
|
)]
|
|
|
|
pub enum ComputeBudgetInstruction {
|
|
|
|
/// Request a specific maximum number of compute units the transaction is
|
|
|
|
/// allowed to consume.
|
2021-10-12 20:56:24 -07:00
|
|
|
RequestUnits(u32),
|
2021-10-19 21:01:39 -07:00
|
|
|
/// Request a specific transaction-wide program heap frame size in bytes.
|
|
|
|
/// The value requested must be a multiple of 1024. This new heap frame size
|
|
|
|
/// applies to each program executed, including all calls to CPIs.
|
|
|
|
RequestHeapFrame(u32),
|
2021-07-16 00:31:22 -07:00
|
|
|
}
|
2021-07-22 10:18:51 -07:00
|
|
|
impl ComputeBudgetInstruction {
|
|
|
|
/// Create a `ComputeBudgetInstruction::RequestUnits` `Instruction`
|
2021-10-12 20:56:24 -07:00
|
|
|
pub fn request_units(units: u32) -> Instruction {
|
2021-07-22 10:18:51 -07:00
|
|
|
Instruction::new_with_borsh(id(), &ComputeBudgetInstruction::RequestUnits(units), vec![])
|
|
|
|
}
|
2021-10-19 21:01:39 -07:00
|
|
|
/// Create a `ComputeBudgetInstruction::RequestHeapFrame` `Instruction`
|
|
|
|
pub fn request_heap_frame(bytes: u32) -> Instruction {
|
|
|
|
Instruction::new_with_borsh(
|
|
|
|
id(),
|
|
|
|
&ComputeBudgetInstruction::RequestHeapFrame(bytes),
|
|
|
|
vec![],
|
|
|
|
)
|
|
|
|
}
|
2021-07-16 00:31:22 -07:00
|
|
|
}
|
2022-01-17 01:43:17 -08:00
|
|
|
|
|
|
|
#[derive(Clone, Copy, Debug, AbiExample, PartialEq)]
|
|
|
|
pub struct ComputeBudget {
|
|
|
|
/// Number of compute units that an instruction is allowed. Compute units
|
|
|
|
/// are consumed by program execution, resources they use, etc...
|
|
|
|
pub max_units: u64,
|
|
|
|
/// Number of compute units consumed by a log_u64 call
|
|
|
|
pub log_64_units: u64,
|
|
|
|
/// Number of compute units consumed by a create_program_address call
|
|
|
|
pub create_program_address_units: u64,
|
|
|
|
/// Number of compute units consumed by an invoke call (not including the cost incurred by
|
|
|
|
/// the called program)
|
|
|
|
pub invoke_units: u64,
|
|
|
|
/// Maximum cross-program invocation depth allowed
|
|
|
|
pub max_invoke_depth: usize,
|
|
|
|
/// Base number of compute units consumed to call SHA256
|
|
|
|
pub sha256_base_cost: u64,
|
|
|
|
/// Incremental number of units consumed by SHA256 (based on bytes)
|
|
|
|
pub sha256_byte_cost: u64,
|
|
|
|
/// Maximum BPF to BPF call depth
|
|
|
|
pub max_call_depth: usize,
|
|
|
|
/// Size of a stack frame in bytes, must match the size specified in the LLVM BPF backend
|
|
|
|
pub stack_frame_size: usize,
|
|
|
|
/// Number of compute units consumed by logging a `Pubkey`
|
|
|
|
pub log_pubkey_units: u64,
|
|
|
|
/// Maximum cross-program invocation instruction size
|
|
|
|
pub max_cpi_instruction_size: usize,
|
|
|
|
/// Number of account data bytes per conpute unit charged during a cross-program invocation
|
|
|
|
pub cpi_bytes_per_unit: u64,
|
|
|
|
/// Base number of compute units consumed to get a sysvar
|
|
|
|
pub sysvar_base_cost: u64,
|
|
|
|
/// Number of compute units consumed to call secp256k1_recover
|
|
|
|
pub secp256k1_recover_cost: u64,
|
|
|
|
/// Number of compute units consumed to do a syscall without any work
|
|
|
|
pub syscall_base_cost: u64,
|
|
|
|
/// Number of compute units consumed to call zktoken_crypto_op
|
|
|
|
pub zk_token_elgamal_op_cost: u64,
|
|
|
|
/// Optional program heap region size, if `None` then loader default
|
|
|
|
pub heap_size: Option<usize>,
|
|
|
|
/// Number of compute units per additional 32k heap above the default (~.5
|
|
|
|
/// us per 32k at 15 units/us rounded up)
|
|
|
|
pub heap_cost: u64,
|
|
|
|
/// Memory operation syscall base cost
|
|
|
|
pub mem_op_base_cost: u64,
|
|
|
|
}
|
|
|
|
impl Default for ComputeBudget {
|
|
|
|
fn default() -> Self {
|
|
|
|
Self::new()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
impl ComputeBudget {
|
|
|
|
pub fn new() -> Self {
|
|
|
|
ComputeBudget {
|
|
|
|
max_units: 200_000,
|
|
|
|
log_64_units: 100,
|
|
|
|
create_program_address_units: 1500,
|
|
|
|
invoke_units: 1000,
|
|
|
|
max_invoke_depth: 4,
|
|
|
|
sha256_base_cost: 85,
|
|
|
|
sha256_byte_cost: 1,
|
|
|
|
max_call_depth: 64,
|
|
|
|
stack_frame_size: 4_096,
|
|
|
|
log_pubkey_units: 100,
|
|
|
|
max_cpi_instruction_size: 1280, // IPv6 Min MTU size
|
|
|
|
cpi_bytes_per_unit: 250, // ~50MB at 200,000 units
|
|
|
|
sysvar_base_cost: 100,
|
|
|
|
secp256k1_recover_cost: 25_000,
|
|
|
|
syscall_base_cost: 100,
|
|
|
|
zk_token_elgamal_op_cost: 25_000,
|
|
|
|
heap_size: None,
|
|
|
|
heap_cost: 8,
|
|
|
|
mem_op_base_cost: 15,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
pub fn process_transaction(
|
|
|
|
&mut self,
|
|
|
|
tx: &SanitizedTransaction,
|
|
|
|
feature_set: Arc<FeatureSet>,
|
|
|
|
) -> Result<(), TransactionError> {
|
|
|
|
let error = TransactionError::InstructionError(0, InstructionError::InvalidInstructionData);
|
|
|
|
// Compute budget instruction must be in the 1st 3 instructions (avoid
|
|
|
|
// nonce marker), otherwise ignored
|
|
|
|
for (program_id, instruction) in tx.message().program_instructions_iter().take(3) {
|
|
|
|
if check_id(program_id) {
|
|
|
|
match try_from_slice_unchecked(&instruction.data) {
|
|
|
|
Ok(ComputeBudgetInstruction::RequestUnits(units)) => {
|
|
|
|
if units > MAX_UNITS {
|
|
|
|
return Err(error);
|
|
|
|
}
|
|
|
|
self.max_units = units as u64;
|
|
|
|
}
|
|
|
|
Ok(ComputeBudgetInstruction::RequestHeapFrame(bytes)) => {
|
|
|
|
if !feature_set.is_active(&requestable_heap_size::id())
|
|
|
|
|| bytes > MAX_HEAP_FRAME_BYTES
|
|
|
|
|| bytes < MIN_HEAP_FRAME_BYTES as u32
|
|
|
|
|| bytes % 1024 != 0
|
|
|
|
{
|
|
|
|
return Err(error);
|
|
|
|
}
|
|
|
|
self.heap_size = Some(bytes as usize);
|
|
|
|
}
|
|
|
|
_ => return Err(error),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
|
|
|
use {
|
|
|
|
super::*,
|
|
|
|
crate::{
|
|
|
|
hash::Hash, message::Message, pubkey::Pubkey, signature::Keypair, signer::Signer,
|
|
|
|
transaction::Transaction,
|
|
|
|
},
|
|
|
|
};
|
|
|
|
|
|
|
|
macro_rules! test {
|
|
|
|
( $instructions: expr, $expected_error: expr, $expected_budget: expr ) => {
|
|
|
|
let payer_keypair = Keypair::new();
|
|
|
|
let tx = SanitizedTransaction::from_transaction_for_tests(Transaction::new(
|
|
|
|
&[&payer_keypair],
|
|
|
|
Message::new($instructions, Some(&payer_keypair.pubkey())),
|
|
|
|
Hash::default(),
|
|
|
|
));
|
|
|
|
let feature_set = Arc::new(FeatureSet::all_enabled());
|
|
|
|
let mut compute_budget = ComputeBudget::default();
|
|
|
|
let result = compute_budget.process_transaction(&tx, feature_set);
|
|
|
|
assert_eq!($expected_error as Result<(), TransactionError>, result);
|
|
|
|
assert_eq!(compute_budget, $expected_budget);
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_process_transaction() {
|
|
|
|
// Units
|
|
|
|
test!(&[], Ok(()), ComputeBudget::default());
|
|
|
|
test!(
|
|
|
|
&[
|
|
|
|
ComputeBudgetInstruction::request_units(1),
|
|
|
|
Instruction::new_with_bincode(Pubkey::new_unique(), &0, vec![]),
|
|
|
|
],
|
|
|
|
Ok(()),
|
|
|
|
ComputeBudget {
|
|
|
|
max_units: 1,
|
|
|
|
..ComputeBudget::default()
|
|
|
|
}
|
|
|
|
);
|
|
|
|
test!(
|
|
|
|
&[
|
|
|
|
ComputeBudgetInstruction::request_units(MAX_UNITS + 1),
|
|
|
|
Instruction::new_with_bincode(Pubkey::new_unique(), &0, vec![]),
|
|
|
|
],
|
|
|
|
Err(TransactionError::InstructionError(
|
|
|
|
0,
|
|
|
|
InstructionError::InvalidInstructionData,
|
|
|
|
)),
|
|
|
|
ComputeBudget::default()
|
|
|
|
);
|
|
|
|
test!(
|
|
|
|
&[
|
|
|
|
Instruction::new_with_bincode(Pubkey::new_unique(), &0, vec![]),
|
|
|
|
ComputeBudgetInstruction::request_units(MAX_UNITS),
|
|
|
|
],
|
|
|
|
Ok(()),
|
|
|
|
ComputeBudget {
|
|
|
|
max_units: MAX_UNITS as u64,
|
|
|
|
..ComputeBudget::default()
|
|
|
|
}
|
|
|
|
);
|
|
|
|
test!(
|
|
|
|
&[
|
|
|
|
Instruction::new_with_bincode(Pubkey::new_unique(), &0, vec![]),
|
|
|
|
Instruction::new_with_bincode(Pubkey::new_unique(), &0, vec![]),
|
|
|
|
Instruction::new_with_bincode(Pubkey::new_unique(), &0, vec![]),
|
|
|
|
ComputeBudgetInstruction::request_units(1),
|
|
|
|
],
|
|
|
|
Ok(()),
|
|
|
|
ComputeBudget::default()
|
|
|
|
);
|
|
|
|
|
|
|
|
// HeapFrame
|
|
|
|
test!(&[], Ok(()), ComputeBudget::default());
|
|
|
|
test!(
|
|
|
|
&[
|
|
|
|
ComputeBudgetInstruction::request_heap_frame(40 * 1024),
|
|
|
|
Instruction::new_with_bincode(Pubkey::new_unique(), &0, vec![]),
|
|
|
|
],
|
|
|
|
Ok(()),
|
|
|
|
ComputeBudget {
|
|
|
|
heap_size: Some(40 * 1024),
|
|
|
|
..ComputeBudget::default()
|
|
|
|
}
|
|
|
|
);
|
|
|
|
test!(
|
|
|
|
&[
|
|
|
|
ComputeBudgetInstruction::request_heap_frame(40 * 1024 + 1),
|
|
|
|
Instruction::new_with_bincode(Pubkey::new_unique(), &0, vec![]),
|
|
|
|
],
|
|
|
|
Err(TransactionError::InstructionError(
|
|
|
|
0,
|
|
|
|
InstructionError::InvalidInstructionData,
|
|
|
|
)),
|
|
|
|
ComputeBudget::default()
|
|
|
|
);
|
|
|
|
test!(
|
|
|
|
&[
|
|
|
|
ComputeBudgetInstruction::request_heap_frame(31 * 1024),
|
|
|
|
Instruction::new_with_bincode(Pubkey::new_unique(), &0, vec![]),
|
|
|
|
],
|
|
|
|
Err(TransactionError::InstructionError(
|
|
|
|
0,
|
|
|
|
InstructionError::InvalidInstructionData,
|
|
|
|
)),
|
|
|
|
ComputeBudget::default()
|
|
|
|
);
|
|
|
|
test!(
|
|
|
|
&[
|
|
|
|
ComputeBudgetInstruction::request_heap_frame(MAX_HEAP_FRAME_BYTES + 1),
|
|
|
|
Instruction::new_with_bincode(Pubkey::new_unique(), &0, vec![]),
|
|
|
|
],
|
|
|
|
Err(TransactionError::InstructionError(
|
|
|
|
0,
|
|
|
|
InstructionError::InvalidInstructionData,
|
|
|
|
)),
|
|
|
|
ComputeBudget::default()
|
|
|
|
);
|
|
|
|
test!(
|
|
|
|
&[
|
|
|
|
Instruction::new_with_bincode(Pubkey::new_unique(), &0, vec![]),
|
|
|
|
ComputeBudgetInstruction::request_heap_frame(MAX_HEAP_FRAME_BYTES),
|
|
|
|
],
|
|
|
|
Ok(()),
|
|
|
|
ComputeBudget {
|
|
|
|
heap_size: Some(MAX_HEAP_FRAME_BYTES as usize),
|
|
|
|
..ComputeBudget::default()
|
|
|
|
}
|
|
|
|
);
|
|
|
|
test!(
|
|
|
|
&[
|
|
|
|
Instruction::new_with_bincode(Pubkey::new_unique(), &0, vec![]),
|
|
|
|
Instruction::new_with_bincode(Pubkey::new_unique(), &0, vec![]),
|
|
|
|
Instruction::new_with_bincode(Pubkey::new_unique(), &0, vec![]),
|
|
|
|
ComputeBudgetInstruction::request_heap_frame(1), // ignored
|
|
|
|
],
|
|
|
|
Ok(()),
|
|
|
|
ComputeBudget::default()
|
|
|
|
);
|
|
|
|
|
|
|
|
// Combined
|
|
|
|
test!(
|
|
|
|
&[
|
|
|
|
Instruction::new_with_bincode(Pubkey::new_unique(), &0, vec![]),
|
|
|
|
ComputeBudgetInstruction::request_heap_frame(MAX_HEAP_FRAME_BYTES),
|
|
|
|
ComputeBudgetInstruction::request_units(MAX_UNITS),
|
|
|
|
],
|
|
|
|
Ok(()),
|
|
|
|
ComputeBudget {
|
|
|
|
max_units: MAX_UNITS as u64,
|
|
|
|
heap_size: Some(MAX_HEAP_FRAME_BYTES as usize),
|
|
|
|
..ComputeBudget::default()
|
|
|
|
}
|
|
|
|
);
|
|
|
|
}
|
|
|
|
}
|