186 lines
6.3 KiB
Rust
186 lines
6.3 KiB
Rust
//! Various constraints as required for production environments
|
|
|
|
use crate::{
|
|
curve::base::{CurveType, SwapCurve},
|
|
curve::{constant_product::ConstantProductCurve, flat::FlatCurve},
|
|
error::SwapError,
|
|
};
|
|
|
|
use solana_program::program_error::ProgramError;
|
|
|
|
#[cfg(feature = "production")]
|
|
use std::env;
|
|
|
|
/// Encodes fee constraints, used in multihost environments where the program
|
|
/// may be used by multiple frontends, to ensure that proper fees are being
|
|
/// assessed.
|
|
/// Since this struct needs to be created at compile-time, we only have access
|
|
/// to const functions and constructors. Since SwapCurve contains a Box, it
|
|
/// cannot be used, so we have to split the curves based on their types.
|
|
pub struct FeeConstraints<'a> {
|
|
/// Owner of the program
|
|
pub owner_key: &'a str,
|
|
/// Valid flat curves for the program
|
|
pub valid_flat_curves: &'a [FlatCurve],
|
|
/// Valid constant product curves for the program
|
|
pub valid_constant_product_curves: &'a [ConstantProductCurve],
|
|
}
|
|
|
|
impl<'a> FeeConstraints<'a> {
|
|
/// Checks that the provided curve is valid for the given constraints
|
|
pub fn validate_curve(&self, swap_curve: &SwapCurve) -> Result<(), ProgramError> {
|
|
let valid_swap_curves: Vec<SwapCurve> = match swap_curve.curve_type {
|
|
CurveType::Flat => self
|
|
.valid_flat_curves
|
|
.iter()
|
|
.map(|x| SwapCurve {
|
|
curve_type: swap_curve.curve_type,
|
|
calculator: Box::new(x.clone()),
|
|
})
|
|
.collect(),
|
|
CurveType::ConstantProduct => self
|
|
.valid_constant_product_curves
|
|
.iter()
|
|
.map(|x| SwapCurve {
|
|
curve_type: swap_curve.curve_type,
|
|
calculator: Box::new(x.clone()),
|
|
})
|
|
.collect(),
|
|
};
|
|
if valid_swap_curves.iter().any(|x| *x == *swap_curve) {
|
|
Ok(())
|
|
} else {
|
|
Err(SwapError::InvalidFee.into())
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(feature = "production")]
|
|
const OWNER_KEY: &'static str = env!("SWAP_PROGRAM_OWNER_FEE_ADDRESS");
|
|
#[cfg(feature = "production")]
|
|
const VALID_CONSTANT_PRODUCT_CURVES: &[ConstantProductCurve] = &[ConstantProductCurve {
|
|
trade_fee_numerator: 25,
|
|
trade_fee_denominator: 10000,
|
|
owner_trade_fee_numerator: 5,
|
|
owner_trade_fee_denominator: 10000,
|
|
owner_withdraw_fee_numerator: 0,
|
|
owner_withdraw_fee_denominator: 0,
|
|
host_fee_numerator: 20,
|
|
host_fee_denominator: 100,
|
|
}];
|
|
#[cfg(feature = "production")]
|
|
const VALID_FLAT_CURVES: &[FlatCurve] = &[FlatCurve {
|
|
trade_fee_numerator: 25,
|
|
trade_fee_denominator: 10000,
|
|
owner_trade_fee_numerator: 5,
|
|
owner_trade_fee_denominator: 10000,
|
|
owner_withdraw_fee_numerator: 0,
|
|
owner_withdraw_fee_denominator: 0,
|
|
host_fee_numerator: 20,
|
|
host_fee_denominator: 100,
|
|
}];
|
|
|
|
/// Fee structure defined by program creator in order to enforce certain
|
|
/// fees when others use the program. Adds checks on pool creation and
|
|
/// swapping to ensure the correct fees and account owners are passed.
|
|
pub const FEE_CONSTRAINTS: Option<FeeConstraints> = {
|
|
#[cfg(feature = "production")]
|
|
{
|
|
Some(FeeConstraints {
|
|
owner_key: OWNER_KEY,
|
|
valid_constant_product_curves: VALID_CONSTANT_PRODUCT_CURVES,
|
|
valid_flat_curves: VALID_FLAT_CURVES,
|
|
})
|
|
}
|
|
#[cfg(not(feature = "production"))]
|
|
{
|
|
None
|
|
}
|
|
};
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
use crate::curve::{base::CurveType, constant_product::ConstantProductCurve};
|
|
|
|
#[test]
|
|
fn validate_fees() {
|
|
let trade_fee_numerator = 1;
|
|
let trade_fee_denominator = 4;
|
|
let owner_trade_fee_numerator = 2;
|
|
let owner_trade_fee_denominator = 5;
|
|
let owner_withdraw_fee_numerator = 4;
|
|
let owner_withdraw_fee_denominator = 10;
|
|
let host_fee_numerator = 10;
|
|
let host_fee_denominator = 100;
|
|
let owner_key = "";
|
|
let curve_type = CurveType::ConstantProduct;
|
|
let mut calculator = ConstantProductCurve {
|
|
trade_fee_numerator,
|
|
trade_fee_denominator,
|
|
owner_trade_fee_numerator,
|
|
owner_trade_fee_denominator,
|
|
owner_withdraw_fee_numerator,
|
|
owner_withdraw_fee_denominator,
|
|
host_fee_numerator,
|
|
host_fee_denominator,
|
|
};
|
|
let swap_curve = SwapCurve {
|
|
curve_type,
|
|
calculator: Box::new(calculator.clone()),
|
|
};
|
|
let fee_constraints = FeeConstraints {
|
|
owner_key,
|
|
valid_constant_product_curves: &[calculator.clone()],
|
|
valid_flat_curves: &[],
|
|
};
|
|
|
|
fee_constraints.validate_curve(&swap_curve).unwrap();
|
|
|
|
calculator.trade_fee_numerator = trade_fee_numerator - 1;
|
|
let swap_curve = SwapCurve {
|
|
curve_type,
|
|
calculator: Box::new(calculator.clone()),
|
|
};
|
|
assert_eq!(
|
|
Err(SwapError::InvalidFee.into()),
|
|
fee_constraints.validate_curve(&swap_curve),
|
|
);
|
|
calculator.trade_fee_numerator = trade_fee_numerator;
|
|
|
|
calculator.trade_fee_denominator = trade_fee_denominator - 1;
|
|
let swap_curve = SwapCurve {
|
|
curve_type,
|
|
calculator: Box::new(calculator.clone()),
|
|
};
|
|
assert_eq!(
|
|
Err(SwapError::InvalidFee.into()),
|
|
fee_constraints.validate_curve(&swap_curve),
|
|
);
|
|
calculator.trade_fee_denominator = trade_fee_denominator;
|
|
|
|
calculator.owner_trade_fee_numerator = owner_trade_fee_numerator - 1;
|
|
let swap_curve = SwapCurve {
|
|
curve_type,
|
|
calculator: Box::new(calculator.clone()),
|
|
};
|
|
assert_eq!(
|
|
Err(SwapError::InvalidFee.into()),
|
|
fee_constraints.validate_curve(&swap_curve),
|
|
);
|
|
calculator.owner_trade_fee_numerator = owner_trade_fee_numerator;
|
|
|
|
calculator.owner_trade_fee_denominator = owner_trade_fee_denominator - 1;
|
|
let swap_curve = SwapCurve {
|
|
curve_type,
|
|
calculator: Box::new(calculator.clone()),
|
|
};
|
|
assert_eq!(
|
|
Err(SwapError::InvalidFee.into()),
|
|
fee_constraints.validate_curve(&swap_curve),
|
|
);
|
|
calculator.owner_trade_fee_denominator = owner_trade_fee_denominator;
|
|
}
|
|
}
|