use arrayref::{array_mut_ref, array_ref, array_refs, mut_array_refs};
use solana_program::{
program_error::ProgramError,
program_pack::{Pack, Sealed},
};
#[repr(C)]
#[derive(Clone, Copy, Debug, Default, PartialEq)]
#[cfg_attr(feature = "fuzz", derive(arbitrary::Arbitrary))]
pub struct Fees {
pub admin_trade_fee_numerator: u64,
pub admin_trade_fee_denominator: u64,
pub admin_withdraw_fee_numerator: u64,
pub admin_withdraw_fee_denominator: u64,
pub trade_fee_numerator: u64,
pub trade_fee_denominator: u64,
pub withdraw_fee_numerator: u64,
pub withdraw_fee_denominator: u64,
}
impl Sealed for Fees {}
impl Pack for Fees {
const LEN: usize = 64;
fn unpack_from_slice(input: &[u8]) -> Result<Self, ProgramError> {
let input = array_ref![input, 0, 64];
#[allow(clippy::ptr_offset_with_cast)]
let (
admin_trade_fee_numerator,
admin_trade_fee_denominator,
admin_withdraw_fee_numerator,
admin_withdraw_fee_denominator,
trade_fee_numerator,
trade_fee_denominator,
withdraw_fee_numerator,
withdraw_fee_denominator,
) = array_refs![input, 8, 8, 8, 8, 8, 8, 8, 8];
Ok(Self {
admin_trade_fee_numerator: u64::from_le_bytes(*admin_trade_fee_numerator),
admin_trade_fee_denominator: u64::from_le_bytes(*admin_trade_fee_denominator),
admin_withdraw_fee_numerator: u64::from_le_bytes(*admin_withdraw_fee_numerator),
admin_withdraw_fee_denominator: u64::from_le_bytes(*admin_withdraw_fee_denominator),
trade_fee_numerator: u64::from_le_bytes(*trade_fee_numerator),
trade_fee_denominator: u64::from_le_bytes(*trade_fee_denominator),
withdraw_fee_numerator: u64::from_le_bytes(*withdraw_fee_numerator),
withdraw_fee_denominator: u64::from_le_bytes(*withdraw_fee_denominator),
})
}
fn pack_into_slice(&self, output: &mut [u8]) {
let output = array_mut_ref![output, 0, 64];
let (
admin_trade_fee_numerator,
admin_trade_fee_denominator,
admin_withdraw_fee_numerator,
admin_withdraw_fee_denominator,
trade_fee_numerator,
trade_fee_denominator,
withdraw_fee_numerator,
withdraw_fee_denominator,
) = mut_array_refs![output, 8, 8, 8, 8, 8, 8, 8, 8];
*admin_trade_fee_numerator = self.admin_trade_fee_numerator.to_le_bytes();
*admin_trade_fee_denominator = self.admin_trade_fee_denominator.to_le_bytes();
*admin_withdraw_fee_numerator = self.admin_withdraw_fee_numerator.to_le_bytes();
*admin_withdraw_fee_denominator = self.admin_withdraw_fee_denominator.to_le_bytes();
*trade_fee_numerator = self.trade_fee_numerator.to_le_bytes();
*trade_fee_denominator = self.trade_fee_denominator.to_le_bytes();
*withdraw_fee_numerator = self.withdraw_fee_numerator.to_le_bytes();
*withdraw_fee_denominator = self.withdraw_fee_denominator.to_le_bytes();
}
}
#[cfg(test)]
#[allow(clippy::unwrap_used)]
mod tests {
use super::*;
#[test]
fn pack_fees() {
let admin_trade_fee_numerator = 1;
let admin_trade_fee_denominator = 2;
let admin_withdraw_fee_numerator = 3;
let admin_withdraw_fee_denominator = 4;
let trade_fee_numerator = 5;
let trade_fee_denominator = 6;
let withdraw_fee_numerator = 7;
let withdraw_fee_denominator = 8;
let fees = Fees {
admin_trade_fee_numerator,
admin_trade_fee_denominator,
admin_withdraw_fee_numerator,
admin_withdraw_fee_denominator,
trade_fee_numerator,
trade_fee_denominator,
withdraw_fee_numerator,
withdraw_fee_denominator,
};
let mut packed = [0u8; Fees::LEN];
Pack::pack_into_slice(&fees, &mut packed[..]);
let unpacked = Fees::unpack_from_slice(&packed).unwrap();
assert_eq!(fees, unpacked);
let mut packed = vec![];
packed.extend_from_slice(&admin_trade_fee_numerator.to_le_bytes());
packed.extend_from_slice(&admin_trade_fee_denominator.to_le_bytes());
packed.extend_from_slice(&admin_withdraw_fee_numerator.to_le_bytes());
packed.extend_from_slice(&admin_withdraw_fee_denominator.to_le_bytes());
packed.extend_from_slice(&trade_fee_numerator.to_le_bytes());
packed.extend_from_slice(&trade_fee_denominator.to_le_bytes());
packed.extend_from_slice(&withdraw_fee_numerator.to_le_bytes());
packed.extend_from_slice(&withdraw_fee_denominator.to_le_bytes());
let unpacked = Fees::unpack_from_slice(&packed).unwrap();
assert_eq!(fees, unpacked);
}
}