sbpf-runtime 0.2.3

pub mod abort;
pub mod crypto;
pub mod log;
pub mod memory;
pub mod pda;
pub mod return_data;
pub mod sysvar;

use {
    crate::{
        config::{ExecutionCost, SysvarContext},
        cpi::request::{self, CpiRequest},
        runtime::LogCollector,
    },
    sbpf_vm::{
        compute::ComputeMeter, errors::SbpfVmResult, memory::Memory, syscalls::SyscallHandler,
    },
    solana_address::Address,
};

const ACCOUNT_META_SIZE: u64 = 34;
const ACCOUNT_INFO_BYTE_SIZE: u64 = 80;

pub struct RuntimeSyscallHandler {
    pub costs: ExecutionCost,
    pub program_id: Address,
    pub sysvars: SysvarContext,
    pub pending_cpi: Option<CpiRequest>,
    pub return_data: crate::cpi::ReturnData,
    pub log_collector: LogCollector,
}

impl RuntimeSyscallHandler {
    pub fn new(
        costs: ExecutionCost,
        program_id: Address,
        sysvars: SysvarContext,
        log_collector: LogCollector,
    ) -> Self {
        Self {
            costs,
            program_id,
            sysvars,
            pending_cpi: None,
            return_data: None,
            log_collector,
        }
    }
}

// Consume CUs for CPI.
fn consume_cpi_compute_units(
    request: &CpiRequest,
    compute: &ComputeMeter,
    costs: &ExecutionCost,
) -> SbpfVmResult<()> {
    // Base invoke cost.
    compute.consume(costs.invoke_units)?;

    // Instruction data and account meta cost.
    let data_cost = request.data.len() as u64 / costs.cpi_bytes_per_unit;
    let meta_cost = (request.accounts.len() as u64 * ACCOUNT_META_SIZE) / costs.cpi_bytes_per_unit;
    compute.consume(data_cost + meta_cost)?;

    // Account info translation cost.
    let account_info_cost =
        (request.caller_accounts.len() as u64 * ACCOUNT_INFO_BYTE_SIZE) / costs.cpi_bytes_per_unit;
    compute.consume(account_info_cost)?;

    // Per-account data cost (skip duplicates).
    let mut seen = Vec::with_capacity(request.accounts.len());
    for meta in request.accounts.iter() {
        if seen.contains(&meta.pubkey) {
            continue;
        }
        seen.push(meta.pubkey);
        if let Some(caller) = request
            .caller_accounts
            .iter()
            .find(|c| c.pubkey == meta.pubkey)
        {
            let acct_cost = caller.data_len / costs.cpi_bytes_per_unit;
            compute.consume(acct_cost)?;
        }
    }
    Ok(())
}

impl SyscallHandler for RuntimeSyscallHandler {
    fn handle(
        &mut self,
        name: &str,
        registers: [u64; 5],
        memory: &mut Memory,
        compute: ComputeMeter,
    ) -> SbpfVmResult<u64> {
        match name {
            "sol_log_" => log::sol_log(
                registers,
                memory,
                &compute,
                &self.costs,
                &self.log_collector,
            ),
            "sol_log_64_" => log::sol_log_64(registers, &compute, &self.costs, &self.log_collector),
            "sol_log_pubkey" => log::sol_log_pubkey(
                registers,
                memory,
                &compute,
                &self.costs,
                &self.log_collector,
            ),
            "sol_log_compute_units_" => {
                log::sol_log_compute_units(&compute, &self.costs, &self.log_collector)
            }
            "sol_remaining_compute_units" => {
                log::sol_remaining_compute_units(&compute, &self.costs)
            }

            "sol_memcpy_" => memory::sol_memcpy(registers, memory, &compute, &self.costs),
            "sol_memmove_" => memory::sol_memmove(registers, memory, &compute, &self.costs),
            "sol_memset_" => memory::sol_memset(registers, memory, &compute, &self.costs),
            "sol_memcmp_" => memory::sol_memcmp(registers, memory, &compute, &self.costs),

            "abort" => abort::abort(),
            "sol_panic_" => abort::sol_panic(registers, memory),

            "sol_sha256" => crypto::sol_sha256(registers, memory, &compute, &self.costs),
            "sol_keccak256" => crypto::sol_keccak256(registers, memory, &compute, &self.costs),
            "sol_blake3" => crypto::sol_blake3(registers, memory, &compute, &self.costs),

            "sol_create_program_address" => {
                pda::sol_create_program_address(registers, memory, &compute, &self.costs)
            }
            "sol_try_find_program_address" => {
                pda::sol_try_find_program_address(registers, memory, &compute, &self.costs)
            }

            "sol_get_clock_sysvar" => sysvar::sol_get_clock_sysvar(
                registers,
                memory,
                &compute,
                &self.costs,
                &self.sysvars,
            ),
            "sol_get_rent_sysvar" => {
                sysvar::sol_get_rent_sysvar(registers, memory, &compute, &self.costs, &self.sysvars)
            }
            "sol_get_epoch_schedule_sysvar" => sysvar::sol_get_epoch_schedule_sysvar(
                registers,
                memory,
                &compute,
                &self.costs,
                &self.sysvars,
            ),
            "sol_get_last_restart_slot_sysvar" => sysvar::sol_get_last_restart_slot_sysvar(
                registers,
                memory,
                &compute,
                &self.costs,
                &self.sysvars,
            ),

            "sol_set_return_data" => {
                let (result, data) = return_data::sol_set_return_data(
                    registers,
                    memory,
                    &compute,
                    &self.costs,
                    &self.program_id,
                )?;
                self.return_data = data;
                Ok(result)
            }
            "sol_get_return_data" => return_data::sol_get_return_data(
                registers,
                memory,
                &compute,
                &self.costs,
                &self.return_data,
            ),

            "sol_invoke_signed_c" => {
                let request = request::parse_cpi_c(registers, memory, &self.program_id)?;
                consume_cpi_compute_units(&request, &compute, &self.costs)?;
                self.pending_cpi = Some(request);
                Ok(0)
            }
            "sol_invoke_signed_rust" => {
                let request = request::parse_cpi_rust(registers, memory, &self.program_id)?;
                consume_cpi_compute_units(&request, &compute, &self.costs)?;
                self.pending_cpi = Some(request);
                Ok(0)
            }

            _ => {
                compute.consume(self.costs.syscall_base_cost)?;
                eprintln!("Unknown syscall: {}", name);
                Ok(0)
            }
        }
    }
}

#[cfg(test)]
mod tests {
    use {
        self::test_helpers::{make_memory, meter},
        super::*,
        crate::cpi::request::{CallerAccountInfo, CpiAccountMeta},
        sbpf_vm::errors::SbpfVmError,
        std::{cell::RefCell, rc::Rc},
    };

    pub(crate) mod test_helpers {
        use {
            crate::config::ExecutionCost,
            sbpf_vm::{compute::ComputeMeter, memory::Memory},
        };

        pub fn make_memory() -> Memory {
            Memory::new(vec![], vec![], 4096, 64 * 1024)
        }

        pub fn costs() -> ExecutionCost {
            ExecutionCost::default()
        }

        pub fn meter(limit: u64) -> ComputeMeter {
            ComputeMeter::new(limit)
        }
    }

    const LIMIT: u64 = 1_000_000;

    fn addr(seed: u8) -> Address {
        Address::new_from_array([seed; 32])
    }

    fn meta(pubkey: Address) -> CpiAccountMeta {
        CpiAccountMeta {
            pubkey,
            is_signer: false,
            is_writable: true,
        }
    }

    fn caller(pubkey: Address, data_len: u64) -> CallerAccountInfo {
        CallerAccountInfo {
            pubkey,
            lamports_addr: 0,
            data_addr: 0,
            data_len,
            owner_addr: 0,
            vm_data_len_addr: 0,
            is_writable: true,
        }
    }

    fn handler() -> RuntimeSyscallHandler {
        RuntimeSyscallHandler::new(
            ExecutionCost::default(),
            addr(7),
            SysvarContext::default(),
            Rc::new(RefCell::new(Vec::new())),
        )
    }

    #[test]
    fn consume_cpi_compute_units_dedups_and_charges_data() {
        let costs = ExecutionCost::default();
        let a = addr(1);
        let b = addr(2);
        let c = addr(3);
        // `a` is duplicated (should be charged once); `c` has no caller entry.
        let request = CpiRequest {
            program_id: addr(9),
            accounts: vec![meta(a), meta(b), meta(c), meta(a)],
            data: vec![0u8; 800],
            caller_accounts: vec![caller(a, 1000), caller(b, 500)],
            signers: Vec::new(),
        };
        let compute = meter(LIMIT);
        consume_cpi_compute_units(&request, &compute, &costs).unwrap();

        let expected = costs.invoke_units + 3 + 4 + 2;
        assert_eq!(compute.get_consumed(), expected);
    }

    #[test]
    fn consume_cpi_compute_units_no_caller_accounts() {
        let costs = ExecutionCost::default();
        let request = CpiRequest {
            program_id: addr(9),
            accounts: vec![meta(addr(1)), meta(addr(2))],
            data: Vec::new(),
            caller_accounts: Vec::new(),
            signers: Vec::new(),
        };
        let compute = meter(LIMIT);
        consume_cpi_compute_units(&request, &compute, &costs).unwrap();
        assert_eq!(compute.get_consumed(), costs.invoke_units);
    }

    #[test]
    fn consume_cpi_compute_units_out_of_budget() {
        let costs = ExecutionCost::default();
        let request = CpiRequest {
            program_id: addr(9),
            accounts: vec![meta(addr(1))],
            data: Vec::new(),
            caller_accounts: Vec::new(),
            signers: Vec::new(),
        };
        let compute = ComputeMeter::new(10); // less than invoke_units
        assert!(consume_cpi_compute_units(&request, &compute, &costs).is_err());
    }

    #[test]
    fn handle_unknown_syscall_charges_base_and_returns_zero() {
        let mut h = handler();
        let mut memory = make_memory();
        let compute = meter(LIMIT);
        let out = h
            .handle("sol_does_not_exist", [0; 5], &mut memory, compute.clone())
            .unwrap();
        assert_eq!(out, 0);
        assert_eq!(compute.get_consumed(), h.costs.syscall_base_cost);
    }

    #[test]
    fn handle_abort_returns_abort_error() {
        let mut h = handler();
        let mut memory = make_memory();
        let compute = meter(LIMIT);
        let err = h.handle("abort", [0; 5], &mut memory, compute).unwrap_err();
        assert!(matches!(err, SbpfVmError::Abort));
    }

    #[test]
    fn handle_log_64_writes_log() {
        let mut h = handler();
        let mut memory = make_memory();
        let compute = meter(LIMIT);
        h.handle("sol_log_64_", [1, 2, 3, 4, 5], &mut memory, compute)
            .unwrap();
        assert!(!h.log_collector.borrow().is_empty());
    }

    #[test]
    fn handle_remaining_compute_units_reports_remaining() {
        let mut h = handler();
        let mut memory = make_memory();
        let compute = meter(LIMIT);
        let out = h
            .handle(
                "sol_remaining_compute_units",
                [0; 5],
                &mut memory,
                compute.clone(),
            )
            .unwrap();
        // Reports the remaining budget after charging its own cost.
        assert_eq!(out, compute.get_remaining());
    }
}