371 lines
13 KiB
Rust
371 lines
13 KiB
Rust
#![allow(clippy::integer_arithmetic)]
|
|
use crate::clock::DEFAULT_MS_PER_SLOT;
|
|
use crate::message::Message;
|
|
use crate::secp256k1_program;
|
|
use log::*;
|
|
|
|
#[derive(Serialize, Deserialize, PartialEq, Eq, Clone, Debug, AbiExample)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct FeeCalculator {
|
|
// The current cost of a signature This amount may increase/decrease over time based on
|
|
// cluster processing load.
|
|
pub lamports_per_signature: u64,
|
|
}
|
|
|
|
impl Default for FeeCalculator {
|
|
fn default() -> Self {
|
|
Self {
|
|
lamports_per_signature: 0,
|
|
}
|
|
}
|
|
}
|
|
|
|
pub struct FeeConfig {
|
|
pub secp256k1_program_enabled: bool,
|
|
}
|
|
|
|
impl Default for FeeConfig {
|
|
fn default() -> Self {
|
|
Self {
|
|
secp256k1_program_enabled: true,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FeeCalculator {
|
|
pub fn new(lamports_per_signature: u64) -> Self {
|
|
Self {
|
|
lamports_per_signature,
|
|
}
|
|
}
|
|
|
|
pub fn calculate_fee(&self, message: &Message) -> u64 {
|
|
self.calculate_fee_with_config(message, &FeeConfig::default())
|
|
}
|
|
|
|
pub fn calculate_fee_with_config(&self, message: &Message, fee_config: &FeeConfig) -> u64 {
|
|
let mut num_secp256k1_signatures: u64 = 0;
|
|
if fee_config.secp256k1_program_enabled {
|
|
for instruction in &message.instructions {
|
|
let program_index = instruction.program_id_index as usize;
|
|
// Transaction may not be sanitized here
|
|
if program_index < message.account_keys.len() {
|
|
let id = message.account_keys[program_index];
|
|
if secp256k1_program::check_id(&id) && !instruction.data.is_empty() {
|
|
num_secp256k1_signatures += instruction.data[0] as u64;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
self.lamports_per_signature
|
|
* (u64::from(message.header.num_required_signatures) + num_secp256k1_signatures)
|
|
}
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, PartialEq, Eq, Clone, Debug, AbiExample)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct FeeRateGovernor {
|
|
// The current cost of a signature This amount may increase/decrease over time based on
|
|
// cluster processing load.
|
|
#[serde(skip)]
|
|
pub lamports_per_signature: u64,
|
|
|
|
// The target cost of a signature when the cluster is operating around target_signatures_per_slot
|
|
// signatures
|
|
pub target_lamports_per_signature: u64,
|
|
|
|
// Used to estimate the desired processing capacity of the cluster. As the signatures for
|
|
// recent slots are fewer/greater than this value, lamports_per_signature will decrease/increase
|
|
// for the next slot. A value of 0 disables lamports_per_signature fee adjustments
|
|
pub target_signatures_per_slot: u64,
|
|
|
|
pub min_lamports_per_signature: u64,
|
|
pub max_lamports_per_signature: u64,
|
|
|
|
// What portion of collected fees are to be destroyed, as a fraction of std::u8::MAX
|
|
pub burn_percent: u8,
|
|
}
|
|
|
|
pub const DEFAULT_TARGET_LAMPORTS_PER_SIGNATURE: u64 = 10_000;
|
|
pub const DEFAULT_TARGET_SIGNATURES_PER_SLOT: u64 = 50 * DEFAULT_MS_PER_SLOT;
|
|
|
|
// Percentage of tx fees to burn
|
|
pub const DEFAULT_BURN_PERCENT: u8 = 50;
|
|
|
|
impl Default for FeeRateGovernor {
|
|
fn default() -> Self {
|
|
Self {
|
|
lamports_per_signature: 0,
|
|
target_lamports_per_signature: DEFAULT_TARGET_LAMPORTS_PER_SIGNATURE,
|
|
target_signatures_per_slot: DEFAULT_TARGET_SIGNATURES_PER_SLOT,
|
|
min_lamports_per_signature: 0,
|
|
max_lamports_per_signature: 0,
|
|
burn_percent: DEFAULT_BURN_PERCENT,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl FeeRateGovernor {
|
|
pub fn new(target_lamports_per_signature: u64, target_signatures_per_slot: u64) -> Self {
|
|
let base_fee_rate_governor = Self {
|
|
target_lamports_per_signature,
|
|
lamports_per_signature: target_lamports_per_signature,
|
|
target_signatures_per_slot,
|
|
..FeeRateGovernor::default()
|
|
};
|
|
|
|
Self::new_derived(&base_fee_rate_governor, 0)
|
|
}
|
|
|
|
pub fn new_derived(
|
|
base_fee_rate_governor: &FeeRateGovernor,
|
|
latest_signatures_per_slot: u64,
|
|
) -> Self {
|
|
let mut me = base_fee_rate_governor.clone();
|
|
|
|
if me.target_signatures_per_slot > 0 {
|
|
// lamports_per_signature can range from 50% to 1000% of
|
|
// target_lamports_per_signature
|
|
me.min_lamports_per_signature = std::cmp::max(1, me.target_lamports_per_signature / 2);
|
|
me.max_lamports_per_signature = me.target_lamports_per_signature * 10;
|
|
|
|
// What the cluster should charge at `latest_signatures_per_slot`
|
|
let desired_lamports_per_signature =
|
|
me.max_lamports_per_signature
|
|
.min(me.min_lamports_per_signature.max(
|
|
me.target_lamports_per_signature
|
|
* std::cmp::min(latest_signatures_per_slot, std::u32::MAX as u64)
|
|
as u64
|
|
/ me.target_signatures_per_slot as u64,
|
|
));
|
|
|
|
trace!(
|
|
"desired_lamports_per_signature: {}",
|
|
desired_lamports_per_signature
|
|
);
|
|
|
|
let gap = desired_lamports_per_signature as i64
|
|
- base_fee_rate_governor.lamports_per_signature as i64;
|
|
|
|
if gap == 0 {
|
|
me.lamports_per_signature = desired_lamports_per_signature;
|
|
} else {
|
|
// Adjust fee by 5% of target_lamports_per_signature to produce a smooth
|
|
// increase/decrease in fees over time.
|
|
let gap_adjust =
|
|
std::cmp::max(1, me.target_lamports_per_signature / 20) as i64 * gap.signum();
|
|
|
|
trace!(
|
|
"lamports_per_signature gap is {}, adjusting by {}",
|
|
gap,
|
|
gap_adjust
|
|
);
|
|
|
|
me.lamports_per_signature =
|
|
me.max_lamports_per_signature
|
|
.min(me.min_lamports_per_signature.max(
|
|
(base_fee_rate_governor.lamports_per_signature as i64 + gap_adjust)
|
|
as u64,
|
|
));
|
|
}
|
|
} else {
|
|
me.lamports_per_signature = base_fee_rate_governor.target_lamports_per_signature;
|
|
me.min_lamports_per_signature = me.target_lamports_per_signature;
|
|
me.max_lamports_per_signature = me.target_lamports_per_signature;
|
|
}
|
|
debug!(
|
|
"new_derived(): lamports_per_signature: {}",
|
|
me.lamports_per_signature
|
|
);
|
|
me
|
|
}
|
|
|
|
/// calculate unburned fee from a fee total, returns (unburned, burned)
|
|
pub fn burn(&self, fees: u64) -> (u64, u64) {
|
|
let burned = fees * u64::from(self.burn_percent) / 100;
|
|
(fees - burned, burned)
|
|
}
|
|
|
|
/// create a FeeCalculator based on current cluster signature throughput
|
|
pub fn create_fee_calculator(&self) -> FeeCalculator {
|
|
FeeCalculator::new(self.lamports_per_signature)
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use crate::{pubkey::Pubkey, system_instruction};
|
|
|
|
#[test]
|
|
fn test_fee_rate_governor_burn() {
|
|
let mut fee_rate_governor = FeeRateGovernor::default();
|
|
assert_eq!(fee_rate_governor.burn(2), (1, 1));
|
|
|
|
fee_rate_governor.burn_percent = 0;
|
|
assert_eq!(fee_rate_governor.burn(2), (2, 0));
|
|
|
|
fee_rate_governor.burn_percent = 100;
|
|
assert_eq!(fee_rate_governor.burn(2), (0, 2));
|
|
}
|
|
|
|
#[test]
|
|
fn test_fee_calculator_calculate_fee() {
|
|
// Default: no fee.
|
|
let message = Message::default();
|
|
assert_eq!(FeeCalculator::default().calculate_fee(&message), 0);
|
|
|
|
// No signature, no fee.
|
|
assert_eq!(FeeCalculator::new(1).calculate_fee(&message), 0);
|
|
|
|
// One signature, a fee.
|
|
let pubkey0 = Pubkey::new(&[0; 32]);
|
|
let pubkey1 = Pubkey::new(&[1; 32]);
|
|
let ix0 = system_instruction::transfer(&pubkey0, &pubkey1, 1);
|
|
let message = Message::new(&[ix0], Some(&pubkey0));
|
|
assert_eq!(FeeCalculator::new(2).calculate_fee(&message), 2);
|
|
|
|
// Two signatures, double the fee.
|
|
let ix0 = system_instruction::transfer(&pubkey0, &pubkey1, 1);
|
|
let ix1 = system_instruction::transfer(&pubkey1, &pubkey0, 1);
|
|
let message = Message::new(&[ix0, ix1], Some(&pubkey0));
|
|
assert_eq!(FeeCalculator::new(2).calculate_fee(&message), 4);
|
|
}
|
|
|
|
#[test]
|
|
fn test_fee_calculator_calculate_fee_secp256k1() {
|
|
use crate::instruction::Instruction;
|
|
let pubkey0 = Pubkey::new(&[0; 32]);
|
|
let pubkey1 = Pubkey::new(&[1; 32]);
|
|
let ix0 = system_instruction::transfer(&pubkey0, &pubkey1, 1);
|
|
let mut secp_instruction = Instruction {
|
|
program_id: crate::secp256k1_program::id(),
|
|
accounts: vec![],
|
|
data: vec![],
|
|
};
|
|
let mut secp_instruction2 = Instruction {
|
|
program_id: crate::secp256k1_program::id(),
|
|
accounts: vec![],
|
|
data: vec![1],
|
|
};
|
|
|
|
let message = Message::new(
|
|
&[
|
|
ix0.clone(),
|
|
secp_instruction.clone(),
|
|
secp_instruction2.clone(),
|
|
],
|
|
Some(&pubkey0),
|
|
);
|
|
assert_eq!(FeeCalculator::new(1).calculate_fee(&message), 2);
|
|
assert_eq!(
|
|
FeeCalculator::new(1).calculate_fee_with_config(
|
|
&message,
|
|
&FeeConfig {
|
|
secp256k1_program_enabled: false
|
|
}
|
|
),
|
|
1
|
|
);
|
|
|
|
secp_instruction.data = vec![0];
|
|
secp_instruction2.data = vec![10];
|
|
let message = Message::new(&[ix0, secp_instruction, secp_instruction2], Some(&pubkey0));
|
|
assert_eq!(FeeCalculator::new(1).calculate_fee(&message), 11);
|
|
}
|
|
|
|
#[test]
|
|
fn test_fee_rate_governor_derived_default() {
|
|
solana_logger::setup();
|
|
|
|
let f0 = FeeRateGovernor::default();
|
|
assert_eq!(
|
|
f0.target_signatures_per_slot,
|
|
DEFAULT_TARGET_SIGNATURES_PER_SLOT
|
|
);
|
|
assert_eq!(
|
|
f0.target_lamports_per_signature,
|
|
DEFAULT_TARGET_LAMPORTS_PER_SIGNATURE
|
|
);
|
|
assert_eq!(f0.lamports_per_signature, 0);
|
|
|
|
let f1 = FeeRateGovernor::new_derived(&f0, DEFAULT_TARGET_SIGNATURES_PER_SLOT);
|
|
assert_eq!(
|
|
f1.target_signatures_per_slot,
|
|
DEFAULT_TARGET_SIGNATURES_PER_SLOT
|
|
);
|
|
assert_eq!(
|
|
f1.target_lamports_per_signature,
|
|
DEFAULT_TARGET_LAMPORTS_PER_SIGNATURE
|
|
);
|
|
assert_eq!(
|
|
f1.lamports_per_signature,
|
|
DEFAULT_TARGET_LAMPORTS_PER_SIGNATURE / 2
|
|
); // min
|
|
}
|
|
|
|
#[test]
|
|
fn test_fee_rate_governor_derived_adjust() {
|
|
solana_logger::setup();
|
|
|
|
let mut f = FeeRateGovernor {
|
|
target_lamports_per_signature: 100,
|
|
target_signatures_per_slot: 100,
|
|
..FeeRateGovernor::default()
|
|
};
|
|
f = FeeRateGovernor::new_derived(&f, 0);
|
|
|
|
// Ramp fees up
|
|
let mut count = 0;
|
|
loop {
|
|
let last_lamports_per_signature = f.lamports_per_signature;
|
|
|
|
f = FeeRateGovernor::new_derived(&f, std::u64::MAX);
|
|
info!("[up] f.lamports_per_signature={}", f.lamports_per_signature);
|
|
|
|
// some maximum target reached
|
|
if f.lamports_per_signature == last_lamports_per_signature {
|
|
break;
|
|
}
|
|
// shouldn't take more than 1000 steps to get to minimum
|
|
assert!(count < 1000);
|
|
count += 1;
|
|
}
|
|
|
|
// Ramp fees down
|
|
let mut count = 0;
|
|
loop {
|
|
let last_lamports_per_signature = f.lamports_per_signature;
|
|
f = FeeRateGovernor::new_derived(&f, 0);
|
|
|
|
info!(
|
|
"[down] f.lamports_per_signature={}",
|
|
f.lamports_per_signature
|
|
);
|
|
|
|
// some minimum target reached
|
|
if f.lamports_per_signature == last_lamports_per_signature {
|
|
break;
|
|
}
|
|
|
|
// shouldn't take more than 1000 steps to get to minimum
|
|
assert!(count < 1000);
|
|
count += 1;
|
|
}
|
|
|
|
// Arrive at target rate
|
|
let mut count = 0;
|
|
while f.lamports_per_signature != f.target_lamports_per_signature {
|
|
f = FeeRateGovernor::new_derived(&f, f.target_signatures_per_slot);
|
|
info!(
|
|
"[target] f.lamports_per_signature={}",
|
|
f.lamports_per_signature
|
|
);
|
|
// shouldn't take more than 100 steps to get to target
|
|
assert!(count < 100);
|
|
count += 1;
|
|
}
|
|
}
|
|
}
|