use crate::{alloc, BPFError};
use alloc::Alloc;
use curve25519_dalek::{ristretto::RistrettoPoint, scalar::Scalar};
use solana_rbpf::{
    ebpf::{hash_symbol_name, MM_HEAP_START},
    error::EbpfError,
    memory_region::{AccessType, MemoryMapping},
    vm::{EbpfVm, Syscall, SyscallObject},
};
use solana_runtime::{
    feature_set::{
        pubkey_log_syscall_enabled, ristretto_mul_syscall_enabled, sha256_syscall_enabled,
    },
    message_processor::MessageProcessor,
    process_instruction::{ComputeMeter, InvokeContext, Logger},
};
use solana_sdk::{
    account::Account,
    account_info::AccountInfo,
    bpf_loader, bpf_loader_deprecated,
    entrypoint::{MAX_PERMITTED_DATA_INCREASE, SUCCESS},
    hash::{Hasher, HASH_BYTES},
    instruction::{AccountMeta, Instruction, InstructionError},
    keyed_account::KeyedAccount,
    message::Message,
    program_error::ProgramError,
    pubkey::{Pubkey, PubkeyError},
};
use std::{
    alloc::Layout,
    cell::{RefCell, RefMut},
    convert::TryFrom,
    mem::{align_of, size_of},
    rc::Rc,
    slice::from_raw_parts_mut,
    str::{from_utf8, Utf8Error},
};
use thiserror::Error as ThisError;

/// Error definitions
#[derive(Debug, ThisError, PartialEq)]
pub enum SyscallError {
    #[error("{0}: {1:?}")]
    InvalidString(Utf8Error, Vec<u8>),
    #[error("BPF program called abort()!")]
    Abort,
    #[error("BPF program Panicked in {0} at {1}:{2}")]
    Panic(String, u64, u64),
    #[error("cannot borrow invoke context")]
    InvokeContextBorrowFailed,
    #[error("malformed signer seed: {0}: {1:?}")]
    MalformedSignerSeed(Utf8Error, Vec<u8>),
    #[error("Could not create program address with signer seeds: {0}")]
    BadSeeds(PubkeyError),
    #[error("Program id is not supported by cross-program invocations")]
    ProgramNotSupported,
    #[error("{0}")]
    InstructionError(InstructionError),
    #[error("Cross-program invocation with unauthorized signer or writable account")]
    PrivilegeEscalation,
    #[error("Unaligned pointer")]
    UnalignedPointer,
}
impl From<SyscallError> for EbpfError<BPFError> {
    fn from(error: SyscallError) -> Self {
        EbpfError::UserError(error.into())
    }
}

trait SyscallConsume {
    fn consume(&mut self, amount: u64) -> Result<(), EbpfError<BPFError>>;
}
impl SyscallConsume for Rc<RefCell<dyn ComputeMeter>> {
    fn consume(&mut self, amount: u64) -> Result<(), EbpfError<BPFError>> {
        self.try_borrow_mut()
            .map_err(|_| SyscallError::InvokeContextBorrowFailed)?
            .consume(amount)
            .map_err(SyscallError::InstructionError)?;
        Ok(())
    }
}

/// Program heap allocators are intended to allocate/free from a given
/// chunk of memory.  The specific allocator implementation is
/// selectable at build-time.
/// Only one allocator is currently supported

/// Simple bump allocator, never frees
use crate::allocator_bump::BPFAllocator;

pub fn register_syscalls<'a>(
    loader_id: &'a Pubkey,
    vm: &mut EbpfVm<'a, BPFError, crate::ThisInstructionMeter>,
    callers_keyed_accounts: &'a [KeyedAccount<'a>],
    invoke_context: &'a mut dyn InvokeContext,
    heap: Vec<u8>,
) -> Result<(), EbpfError<BPFError>> {
    let compute_budget = invoke_context.get_compute_budget();

    // Syscall functions common across languages

    vm.register_syscall(hash_symbol_name(b"abort"), Syscall::Function(syscall_abort))?;
    vm.register_syscall(
        hash_symbol_name(b"sol_panic_"),
        Syscall::Object(Box::new(SyscallPanic { loader_id })),
    )?;
    vm.register_syscall(
        hash_symbol_name(b"sol_log_"),
        Syscall::Object(Box::new(SyscallLog {
            cost: compute_budget.log_units,
            compute_meter: invoke_context.get_compute_meter(),
            logger: invoke_context.get_logger(),
            loader_id,
        })),
    )?;
    vm.register_syscall(
        hash_symbol_name(b"sol_log_64_"),
        Syscall::Object(Box::new(SyscallLogU64 {
            cost: compute_budget.log_64_units,
            compute_meter: invoke_context.get_compute_meter(),
            logger: invoke_context.get_logger(),
        })),
    )?;

    if invoke_context.is_feature_active(&pubkey_log_syscall_enabled::id()) {
        vm.register_syscall(
            hash_symbol_name(b"sol_log_pubkey"),
            Syscall::Object(Box::new(SyscallLogPubkey {
                cost: compute_budget.log_pubkey_units,
                compute_meter: invoke_context.get_compute_meter(),
                logger: invoke_context.get_logger(),
                loader_id,
            })),
        )?;
    }

    if invoke_context.is_feature_active(&sha256_syscall_enabled::id()) {
        vm.register_syscall(
            hash_symbol_name(b"sol_sha256"),
            Syscall::Object(Box::new(SyscallSha256 {
                sha256_base_cost: compute_budget.sha256_base_cost,
                sha256_byte_cost: compute_budget.sha256_byte_cost,
                compute_meter: invoke_context.get_compute_meter(),
                loader_id,
            })),
        )?;
    }

    if invoke_context.is_feature_active(&ristretto_mul_syscall_enabled::id()) {
        vm.register_syscall(
            hash_symbol_name(b"sol_ristretto_mul"),
            Syscall::Object(Box::new(SyscallRistrettoMul {
                cost: 0,
                compute_meter: invoke_context.get_compute_meter(),
                loader_id,
            })),
        )?;
    }

    vm.register_syscall(
        hash_symbol_name(b"sol_create_program_address"),
        Syscall::Object(Box::new(SyscallCreateProgramAddress {
            cost: compute_budget.create_program_address_units,
            compute_meter: invoke_context.get_compute_meter(),
            loader_id,
        })),
    )?;

    // Cross-program invocation syscalls

    let invoke_context = Rc::new(RefCell::new(invoke_context));
    vm.register_syscall(
        hash_symbol_name(b"sol_invoke_signed_c"),
        Syscall::Object(Box::new(SyscallInvokeSignedC {
            callers_keyed_accounts,
            invoke_context: invoke_context.clone(),
            loader_id,
        })),
    )?;
    vm.register_syscall(
        hash_symbol_name(b"sol_invoke_signed_rust"),
        Syscall::Object(Box::new(SyscallInvokeSignedRust {
            callers_keyed_accounts,
            invoke_context: invoke_context.clone(),
            loader_id,
        })),
    )?;

    // Memory allocator

    vm.register_syscall(
        hash_symbol_name(b"sol_alloc_free_"),
        Syscall::Object(Box::new(SyscallAllocFree {
            aligned: *loader_id != bpf_loader_deprecated::id(),
            allocator: BPFAllocator::new(heap, MM_HEAP_START),
        })),
    )?;

    Ok(())
}

#[macro_export]
macro_rules! translate {
    ($memory_mapping:expr, $access_type:expr, $vm_addr:expr, $len:expr, $loader_id:expr) => {
        $memory_mapping.map::<BPFError>($access_type, $vm_addr as u64, $len as u64)
    };
}

#[macro_export]
macro_rules! translate_type_mut {
    ($memory_mapping:expr, $access_type:expr, $vm_addr:expr, $t:ty, $loader_id:expr) => {{
        if $loader_id != &bpf_loader_deprecated::id()
            && ($vm_addr as u64 as *mut $t).align_offset(align_of::<$t>()) != 0
        {
            Err(SyscallError::UnalignedPointer.into())
        } else {
            unsafe {
                match translate!(
                    $memory_mapping,
                    $access_type,
                    $vm_addr,
                    size_of::<$t>(),
                    $loader_id
                ) {
                    Ok(value) => Ok(&mut *(value as *mut $t)),
                    Err(e) => Err(e),
                }
            }
        }
    }};
}
#[macro_export]
macro_rules! translate_type {
    ($memory_mapping:expr, $access_type:expr, $vm_addr:expr, $t:ty, $loader_id:expr) => {
        match translate_type_mut!($memory_mapping, $access_type, $vm_addr, $t, $loader_id) {
            Ok(value) => Ok(&*value),
            Err(e) => Err(e),
        }
    };
}

#[macro_export]
macro_rules! translate_slice_mut {
    ($memory_mapping:expr, $access_type:expr, $vm_addr:expr, $t:ty, $len:expr, $loader_id:expr) => {{
        if $loader_id != &bpf_loader_deprecated::id()
            && ($vm_addr as u64 as *mut $t).align_offset(align_of::<$t>()) != 0
        {
            Err(SyscallError::UnalignedPointer.into())
        } else if $len == 0 {
            Ok(unsafe { from_raw_parts_mut(0x1 as *mut $t, $len as usize) })
        } else {
            match translate!(
                $memory_mapping,
                $access_type,
                $vm_addr,
                $len as usize * size_of::<$t>(),
                $loader_id
            ) {
                Ok(value) => Ok(unsafe { from_raw_parts_mut(value as *mut $t, $len as usize) }),
                Err(e) => Err(e),
            }
        }
    }};
}
#[macro_export]
macro_rules! translate_slice {
    ($memory_mapping:expr, $access_type:expr, $vm_addr:expr, $t:ty, $len:expr, $loader_id:expr) => {
        match translate_slice_mut!(
            $memory_mapping,
            $access_type,
            $vm_addr,
            $t,
            $len,
            $loader_id
        ) {
            Ok(value) => Ok(&*value),
            Err(e) => Err(e),
        }
    };
}

/// Take a virtual pointer to a string (points to BPF VM memory space), translate it
/// pass it to a user-defined work function
fn translate_string_and_do(
    memory_mapping: &MemoryMapping,
    access_type: AccessType,
    addr: u64,
    len: u64,
    loader_id: &Pubkey,
    work: &mut dyn FnMut(&str) -> Result<u64, EbpfError<BPFError>>,
) -> Result<u64, EbpfError<BPFError>> {
    let buf = translate_slice!(memory_mapping, access_type, addr, u8, len, loader_id)?;
    let i = match buf.iter().position(|byte| *byte == 0) {
        Some(i) => i,
        None => len as usize,
    };
    match from_utf8(&buf[..i]) {
        Ok(message) => work(message),
        Err(err) => Err(SyscallError::InvalidString(err, buf[..i].to_vec()).into()),
    }
}

/// Abort syscall functions, called when the BPF program calls `abort()`
/// LLVM will insert calls to `abort()` if it detects an untenable situation,
/// `abort()` is not intended to be called explicitly by the program.
/// Causes the BPF program to be halted immediately
pub fn syscall_abort(
    _arg1: u64,
    _arg2: u64,
    _arg3: u64,
    _arg4: u64,
    _arg5: u64,
    _memory_mapping: &MemoryMapping,
) -> Result<u64, EbpfError<BPFError>> {
    Err(SyscallError::Abort.into())
}

/// Panic syscall function, called when the BPF program calls 'sol_panic_()`
/// Causes the BPF program to be halted immediately
/// Log a user's info message
pub struct SyscallPanic<'a> {
    loader_id: &'a Pubkey,
}
impl<'a> SyscallObject<BPFError> for SyscallPanic<'a> {
    fn call(
        &mut self,
        file: u64,
        len: u64,
        line: u64,
        column: u64,
        _arg5: u64,
        memory_mapping: &MemoryMapping,
    ) -> Result<u64, EbpfError<BPFError>> {
        translate_string_and_do(
            memory_mapping,
            AccessType::Load,
            file,
            len,
            &self.loader_id,
            &mut |string: &str| Err(SyscallError::Panic(string.to_string(), line, column).into()),
        )
    }
}

/// Log a user's info message
pub struct SyscallLog<'a> {
    cost: u64,
    compute_meter: Rc<RefCell<dyn ComputeMeter>>,
    logger: Rc<RefCell<dyn Logger>>,
    loader_id: &'a Pubkey,
}
impl<'a> SyscallObject<BPFError> for SyscallLog<'a> {
    fn call(
        &mut self,
        addr: u64,
        len: u64,
        _arg3: u64,
        _arg4: u64,
        _arg5: u64,
        memory_mapping: &MemoryMapping,
    ) -> Result<u64, EbpfError<BPFError>> {
        self.compute_meter.consume(self.cost)?;
        let mut logger = self
            .logger
            .try_borrow_mut()
            .map_err(|_| SyscallError::InvokeContextBorrowFailed)?;
        if logger.log_enabled() {
            translate_string_and_do(
                memory_mapping,
                AccessType::Load,
                addr,
                len,
                &self.loader_id,
                &mut |string: &str| {
                    logger.log(&format!("Program log: {}", string));
                    Ok(0)
                },
            )?;
        }
        Ok(0)
    }
}

/// Log 5 64-bit values
pub struct SyscallLogU64 {
    cost: u64,
    compute_meter: Rc<RefCell<dyn ComputeMeter>>,
    logger: Rc<RefCell<dyn Logger>>,
}
impl SyscallObject<BPFError> for SyscallLogU64 {
    fn call(
        &mut self,
        arg1: u64,
        arg2: u64,
        arg3: u64,
        arg4: u64,
        arg5: u64,
        _memory_mapping: &MemoryMapping,
    ) -> Result<u64, EbpfError<BPFError>> {
        self.compute_meter.consume(self.cost)?;
        let mut logger = self
            .logger
            .try_borrow_mut()
            .map_err(|_| SyscallError::InvokeContextBorrowFailed)?;
        if logger.log_enabled() {
            logger.log(&format!(
                "Program log: {:#x}, {:#x}, {:#x}, {:#x}, {:#x}",
                arg1, arg2, arg3, arg4, arg5
            ));
        }
        Ok(0)
    }
}

/// Log 5 64-bit values
pub struct SyscallLogPubkey<'a> {
    cost: u64,
    compute_meter: Rc<RefCell<dyn ComputeMeter>>,
    logger: Rc<RefCell<dyn Logger>>,
    loader_id: &'a Pubkey,
}
impl<'a> SyscallObject<BPFError> for SyscallLogPubkey<'a> {
    fn call(
        &mut self,
        pubkey_addr: u64,
        _arg2: u64,
        _arg3: u64,
        _arg4: u64,
        _arg5: u64,
        memory_mapping: &MemoryMapping,
    ) -> Result<u64, EbpfError<BPFError>> {
        self.compute_meter.consume(self.cost)?;
        let mut logger = self
            .logger
            .try_borrow_mut()
            .map_err(|_| SyscallError::InvokeContextBorrowFailed)?;
        if logger.log_enabled() {
            let pubkey = translate_type!(
                memory_mapping,
                AccessType::Load,
                pubkey_addr,
                Pubkey,
                self.loader_id
            )?;
            logger.log(&format!("Program log: {}", pubkey));
        }
        Ok(0)
    }
}

/// Dynamic memory allocation syscall called when the BPF program calls
/// `sol_alloc_free_()`.  The allocator is expected to allocate/free
/// from/to a given chunk of memory and enforce size restrictions.  The
/// memory chunk is given to the allocator during allocator creation and
/// information about that memory (start address and size) is passed
/// to the VM to use for enforcement.
pub struct SyscallAllocFree {
    aligned: bool,
    allocator: BPFAllocator,
}
impl SyscallObject<BPFError> for SyscallAllocFree {
    fn call(
        &mut self,
        size: u64,
        free_addr: u64,
        _arg3: u64,
        _arg4: u64,
        _arg5: u64,
        _memory_mapping: &MemoryMapping,
    ) -> Result<u64, EbpfError<BPFError>> {
        let align = if self.aligned {
            align_of::<u128>()
        } else {
            align_of::<u8>()
        };
        let layout = match Layout::from_size_align(size as usize, align) {
            Ok(layout) => layout,
            Err(_) => return Ok(0),
        };
        if free_addr == 0 {
            match self.allocator.alloc(layout) {
                Ok(addr) => Ok(addr as u64),
                Err(_) => Ok(0),
            }
        } else {
            self.allocator.dealloc(free_addr, layout);
            Ok(0)
        }
    }
}

/// Create a program address
struct SyscallCreateProgramAddress<'a> {
    cost: u64,
    compute_meter: Rc<RefCell<dyn ComputeMeter>>,
    loader_id: &'a Pubkey,
}
impl<'a> SyscallObject<BPFError> for SyscallCreateProgramAddress<'a> {
    fn call(
        &mut self,
        seeds_addr: u64,
        seeds_len: u64,
        program_id_addr: u64,
        address_addr: u64,
        _arg5: u64,
        memory_mapping: &MemoryMapping,
    ) -> Result<u64, EbpfError<BPFError>> {
        self.compute_meter.consume(self.cost)?;
        // TODO need ref?
        let untranslated_seeds = translate_slice!(
            memory_mapping,
            AccessType::Load,
            seeds_addr,
            &[&u8],
            seeds_len,
            self.loader_id
        )?;
        let seeds = untranslated_seeds
            .iter()
            .map(|untranslated_seed| {
                translate_slice!(
                    memory_mapping,
                    AccessType::Load,
                    untranslated_seed.as_ptr(),
                    u8,
                    untranslated_seed.len(),
                    self.loader_id
                )
            })
            .collect::<Result<Vec<_>, EbpfError<BPFError>>>()?;
        let program_id = translate_type!(
            memory_mapping,
            AccessType::Load,
            program_id_addr,
            Pubkey,
            self.loader_id
        )?;

        let new_address = match Pubkey::create_program_address(&seeds, program_id)
            .map_err(SyscallError::BadSeeds)
        {
            Ok(address) => address,
            Err(_) => return Ok(1),
        };
        let address = translate_slice_mut!(
            memory_mapping,
            AccessType::Store,
            address_addr,
            u8,
            32,
            self.loader_id
        )?;
        address.copy_from_slice(new_address.as_ref());
        Ok(0)
    }
}

/// SHA256
pub struct SyscallSha256<'a> {
    sha256_base_cost: u64,
    sha256_byte_cost: u64,
    compute_meter: Rc<RefCell<dyn ComputeMeter>>,
    loader_id: &'a Pubkey,
}
impl<'a> SyscallObject<BPFError> for SyscallSha256<'a> {
    fn call(
        &mut self,
        vals_addr: u64,
        vals_len: u64,
        result_addr: u64,
        _arg4: u64,
        _arg5: u64,
        memory_mapping: &MemoryMapping,
    ) -> Result<u64, EbpfError<BPFError>> {
        self.compute_meter.consume(self.sha256_base_cost)?;
        let hash_result = translate_slice_mut!(
            memory_mapping,
            AccessType::Store,
            result_addr,
            u8,
            HASH_BYTES,
            self.loader_id
        )?;
        let mut hasher = Hasher::default();
        if vals_len > 0 {
            let vals = translate_slice!(
                memory_mapping,
                AccessType::Load,
                vals_addr,
                &[u8],
                vals_len,
                self.loader_id
            )?;
            for val in vals.iter() {
                let bytes = translate_slice!(
                    memory_mapping,
                    AccessType::Load,
                    val.as_ptr(),
                    u8,
                    val.len(),
                    self.loader_id
                )?;
                self.compute_meter
                    .consume(self.sha256_byte_cost * (val.len() as u64 / 2))?;
                hasher.hash(bytes);
            }
        }
        hash_result.copy_from_slice(&hasher.result().to_bytes());
        Ok(0)
    }
}

/// Ristretto point multiply
pub struct SyscallRistrettoMul<'a> {
    cost: u64,
    compute_meter: Rc<RefCell<dyn ComputeMeter>>,
    loader_id: &'a Pubkey,
}
impl<'a> SyscallObject<BPFError> for SyscallRistrettoMul<'a> {
    fn call(
        &mut self,
        point_addr: u64,
        scalar_addr: u64,
        result_addr: u64,
        _arg4: u64,
        _arg5: u64,
        memory_mapping: &MemoryMapping,
    ) -> Result<u64, EbpfError<BPFError>> {
        self.compute_meter.consume(self.cost)?;

        let point = translate_type!(
            memory_mapping,
            AccessType::Load,
            point_addr,
            RistrettoPoint,
            self.loader_id
        )?;
        let scalar = translate_type!(
            memory_mapping,
            AccessType::Load,
            scalar_addr,
            Scalar,
            self.loader_id
        )?;
        let result = translate_type_mut!(
            memory_mapping,
            AccessType::Store,
            result_addr,
            RistrettoPoint,
            self.loader_id
        )?;

        *result = point * scalar;

        Ok(0)
    }
}

// Cross-program invocation syscalls

struct AccountReferences<'a> {
    lamports: &'a mut u64,
    owner: &'a mut Pubkey,
    data: &'a mut [u8],
    ref_to_len_in_vm: &'a mut u64,
    serialized_len_ptr: &'a mut u64,
}
type TranslatedAccounts<'a> = (Vec<Rc<RefCell<Account>>>, Vec<AccountReferences<'a>>);

/// Implemented by language specific data structure translators
trait SyscallInvokeSigned<'a> {
    fn get_context_mut(&self) -> Result<RefMut<&'a mut dyn InvokeContext>, EbpfError<BPFError>>;
    fn get_callers_keyed_accounts(&self) -> &'a [KeyedAccount<'a>];
    fn translate_instruction(
        &self,
        addr: u64,
        memory_mapping: &MemoryMapping,
    ) -> Result<Instruction, EbpfError<BPFError>>;
    fn translate_accounts(
        &self,
        message: &Message,
        account_infos_addr: u64,
        account_infos_len: usize,
        memory_mapping: &MemoryMapping,
    ) -> Result<TranslatedAccounts<'a>, EbpfError<BPFError>>;
    fn translate_signers(
        &self,
        program_id: &Pubkey,
        signers_seeds_addr: u64,
        signers_seeds_len: usize,
        memory_mapping: &MemoryMapping,
    ) -> Result<Vec<Pubkey>, EbpfError<BPFError>>;
}

/// Cross-program invocation called from Rust
pub struct SyscallInvokeSignedRust<'a> {
    callers_keyed_accounts: &'a [KeyedAccount<'a>],
    invoke_context: Rc<RefCell<&'a mut dyn InvokeContext>>,
    loader_id: &'a Pubkey,
}
impl<'a> SyscallInvokeSigned<'a> for SyscallInvokeSignedRust<'a> {
    fn get_context_mut(&self) -> Result<RefMut<&'a mut dyn InvokeContext>, EbpfError<BPFError>> {
        self.invoke_context
            .try_borrow_mut()
            .map_err(|_| SyscallError::InvokeContextBorrowFailed.into())
    }
    fn get_callers_keyed_accounts(&self) -> &'a [KeyedAccount<'a>] {
        self.callers_keyed_accounts
    }
    fn translate_instruction(
        &self,
        addr: u64,
        memory_mapping: &MemoryMapping,
    ) -> Result<Instruction, EbpfError<BPFError>> {
        let ix = translate_type!(
            memory_mapping,
            AccessType::Load,
            addr,
            Instruction,
            self.loader_id
        )?;
        let accounts = translate_slice!(
            memory_mapping,
            AccessType::Load,
            ix.accounts.as_ptr(),
            AccountMeta,
            ix.accounts.len(),
            self.loader_id
        )?
        .to_vec();
        let data = translate_slice!(
            memory_mapping,
            AccessType::Load,
            ix.data.as_ptr(),
            u8,
            ix.data.len(),
            self.loader_id
        )?
        .to_vec();
        Ok(Instruction {
            program_id: ix.program_id,
            accounts,
            data,
        })
    }

    fn translate_accounts(
        &self,
        message: &Message,
        account_infos_addr: u64,
        account_infos_len: usize,
        memory_mapping: &MemoryMapping,
    ) -> Result<TranslatedAccounts<'a>, EbpfError<BPFError>> {
        let account_infos = if account_infos_len > 0 {
            translate_slice!(
                memory_mapping,
                AccessType::Load,
                account_infos_addr,
                AccountInfo,
                account_infos_len,
                self.loader_id
            )?
        } else {
            &[]
        };

        let mut accounts = Vec::with_capacity(message.account_keys.len());
        let mut refs = Vec::with_capacity(message.account_keys.len());
        'root: for account_key in message.account_keys.iter() {
            for account_info in account_infos.iter() {
                let key = translate_type!(
                    memory_mapping,
                    AccessType::Load,
                    account_info.key as *const _,
                    Pubkey,
                    self.loader_id
                )?;
                if account_key == key {
                    let lamports = {
                        // Double translate lamports out of RefCell
                        let ptr = translate_type!(
                            memory_mapping,
                            AccessType::Load,
                            account_info.lamports.as_ptr(),
                            u64,
                            self.loader_id
                        )?;
                        translate_type_mut!(
                            memory_mapping,
                            AccessType::Store,
                            *ptr,
                            u64,
                            self.loader_id
                        )?
                    };
                    let owner = translate_type_mut!(
                        memory_mapping,
                        AccessType::Load,
                        account_info.owner as *const _,
                        Pubkey,
                        self.loader_id
                    )?;
                    let (data, ref_to_len_in_vm, serialized_len_ptr) = {
                        // Double translate data out of RefCell
                        let data = *translate_type!(
                            memory_mapping,
                            AccessType::Load,
                            account_info.data.as_ptr(),
                            &[u8],
                            self.loader_id
                        )?;
                        let translated = translate!(
                            memory_mapping,
                            AccessType::Load,
                            account_info.data.as_ptr(),
                            8,
                            self.loader_id
                        )? as *mut u64;
                        let ref_to_len_in_vm = unsafe { &mut *translated.offset(1) };
                        let ref_of_len_in_input_buffer = unsafe { data.as_ptr().offset(-8) };
                        let serialized_len_ptr = translate_type_mut!(
                            memory_mapping,
                            AccessType::Store,
                            ref_of_len_in_input_buffer,
                            u64,
                            self.loader_id
                        )?;
                        (
                            translate_slice_mut!(
                                memory_mapping,
                                AccessType::Store,
                                data.as_ptr(),
                                u8,
                                data.len(),
                                self.loader_id
                            )?,
                            ref_to_len_in_vm,
                            serialized_len_ptr,
                        )
                    };

                    accounts.push(Rc::new(RefCell::new(Account {
                        lamports: *lamports,
                        data: data.to_vec(),
                        executable: account_info.executable,
                        owner: *owner,
                        rent_epoch: account_info.rent_epoch,
                    })));
                    refs.push(AccountReferences {
                        lamports,
                        owner,
                        data,
                        ref_to_len_in_vm,
                        serialized_len_ptr,
                    });
                    continue 'root;
                }
            }
            return Err(SyscallError::InstructionError(InstructionError::MissingAccount).into());
        }

        Ok((accounts, refs))
    }

    fn translate_signers(
        &self,
        program_id: &Pubkey,
        signers_seeds_addr: u64,
        signers_seeds_len: usize,
        memory_mapping: &MemoryMapping,
    ) -> Result<Vec<Pubkey>, EbpfError<BPFError>> {
        let mut signers = Vec::new();
        if signers_seeds_len > 0 {
            let signers_seeds = translate_slice!(
                memory_mapping,
                AccessType::Load,
                signers_seeds_addr,
                &[&[u8]],
                signers_seeds_len,
                self.loader_id
            )?;
            for signer_seeds in signers_seeds.iter() {
                let untranslated_seeds = translate_slice!(
                    memory_mapping,
                    AccessType::Load,
                    signer_seeds.as_ptr(),
                    &[u8],
                    signer_seeds.len(),
                    self.loader_id
                )?;
                let seeds = untranslated_seeds
                    .iter()
                    .map(|untranslated_seed| {
                        translate_slice!(
                            memory_mapping,
                            AccessType::Load,
                            untranslated_seed.as_ptr(),
                            u8,
                            untranslated_seed.len(),
                            self.loader_id
                        )
                    })
                    .collect::<Result<Vec<_>, EbpfError<BPFError>>>()?;
                let signer = Pubkey::create_program_address(&seeds, program_id)
                    .map_err(SyscallError::BadSeeds)?;
                signers.push(signer);
            }
            Ok(signers)
        } else {
            Ok(vec![])
        }
    }
}
impl<'a> SyscallObject<BPFError> for SyscallInvokeSignedRust<'a> {
    fn call(
        &mut self,
        instruction_addr: u64,
        account_infos_addr: u64,
        account_infos_len: u64,
        signers_seeds_addr: u64,
        signers_seeds_len: u64,
        memory_mapping: &MemoryMapping,
    ) -> Result<u64, EbpfError<BPFError>> {
        call(
            self,
            instruction_addr,
            account_infos_addr,
            account_infos_len,
            signers_seeds_addr,
            signers_seeds_len,
            memory_mapping,
        )
    }
}

/// Rust representation of C's SolInstruction
#[derive(Debug)]
struct SolInstruction {
    program_id_addr: u64,
    accounts_addr: u64,
    accounts_len: usize,
    data_addr: u64,
    data_len: usize,
}

/// Rust representation of C's SolAccountMeta
#[derive(Debug)]
struct SolAccountMeta {
    pubkey_addr: u64,
    is_writable: bool,
    is_signer: bool,
}

/// Rust representation of C's SolAccountInfo
#[derive(Debug)]
struct SolAccountInfo {
    key_addr: u64,
    lamports_addr: u64,
    data_len: u64,
    data_addr: u64,
    owner_addr: u64,
    rent_epoch: u64,
    is_signer: bool,
    is_writable: bool,
    executable: bool,
}

/// Rust representation of C's SolSignerSeed
#[derive(Debug)]
struct SolSignerSeedC {
    addr: u64,
    len: u64,
}

/// Rust representation of C's SolSignerSeeds
#[derive(Debug)]
struct SolSignerSeedsC {
    addr: u64,
    len: u64,
}

/// Cross-program invocation called from C
pub struct SyscallInvokeSignedC<'a> {
    callers_keyed_accounts: &'a [KeyedAccount<'a>],
    invoke_context: Rc<RefCell<&'a mut dyn InvokeContext>>,
    loader_id: &'a Pubkey,
}
impl<'a> SyscallInvokeSigned<'a> for SyscallInvokeSignedC<'a> {
    fn get_context_mut(&self) -> Result<RefMut<&'a mut dyn InvokeContext>, EbpfError<BPFError>> {
        self.invoke_context
            .try_borrow_mut()
            .map_err(|_| SyscallError::InvokeContextBorrowFailed.into())
    }
    fn get_callers_keyed_accounts(&self) -> &'a [KeyedAccount<'a>] {
        self.callers_keyed_accounts
    }
    fn translate_instruction(
        &self,
        addr: u64,
        memory_mapping: &MemoryMapping,
    ) -> Result<Instruction, EbpfError<BPFError>> {
        let ix_c = translate_type!(
            memory_mapping,
            AccessType::Load,
            addr,
            SolInstruction,
            self.loader_id
        )?;
        let program_id = translate_type!(
            memory_mapping,
            AccessType::Load,
            ix_c.program_id_addr,
            Pubkey,
            self.loader_id
        )?;
        let meta_cs = translate_slice!(
            memory_mapping,
            AccessType::Load,
            ix_c.accounts_addr,
            SolAccountMeta,
            ix_c.accounts_len,
            self.loader_id
        )?;
        let data = translate_slice!(
            memory_mapping,
            AccessType::Load,
            ix_c.data_addr,
            u8,
            ix_c.data_len,
            self.loader_id
        )?
        .to_vec();
        let accounts = meta_cs
            .iter()
            .map(|meta_c| {
                let pubkey = translate_type!(
                    memory_mapping,
                    AccessType::Load,
                    meta_c.pubkey_addr,
                    Pubkey,
                    self.loader_id
                )?;
                Ok(AccountMeta {
                    pubkey: *pubkey,
                    is_signer: meta_c.is_signer,
                    is_writable: meta_c.is_writable,
                })
            })
            .collect::<Result<Vec<AccountMeta>, EbpfError<BPFError>>>()?;

        Ok(Instruction {
            program_id: *program_id,
            accounts,
            data,
        })
    }

    fn translate_accounts(
        &self,
        message: &Message,
        account_infos_addr: u64,
        account_infos_len: usize,
        memory_mapping: &MemoryMapping,
    ) -> Result<TranslatedAccounts<'a>, EbpfError<BPFError>> {
        let account_infos = translate_slice!(
            memory_mapping,
            AccessType::Load,
            account_infos_addr,
            SolAccountInfo,
            account_infos_len,
            self.loader_id
        )?;
        let mut accounts = Vec::with_capacity(message.account_keys.len());
        let mut refs = Vec::with_capacity(message.account_keys.len());
        'root: for account_key in message.account_keys.iter() {
            for account_info in account_infos.iter() {
                let key = translate_type!(
                    memory_mapping,
                    AccessType::Load,
                    account_info.key_addr,
                    Pubkey,
                    self.loader_id
                )?;
                if account_key == key {
                    let lamports = translate_type_mut!(
                        memory_mapping,
                        AccessType::Store,
                        account_info.lamports_addr,
                        u64,
                        self.loader_id
                    )?;
                    let owner = translate_type_mut!(
                        memory_mapping,
                        AccessType::Load,
                        account_info.owner_addr,
                        Pubkey,
                        self.loader_id
                    )?;
                    let data = translate_slice_mut!(
                        memory_mapping,
                        AccessType::Store,
                        account_info.data_addr,
                        u8,
                        account_info.data_len,
                        self.loader_id
                    )?;
                    let ref_to_len_in_vm =
                        unsafe { &mut *(&account_info.data_len as *const u64 as u64 as *mut u64) };
                    let ref_of_len_in_input_buffer =
                        unsafe { (account_info.data_addr as *mut u8).offset(-8) };
                    let serialized_len_ptr = translate_type_mut!(
                        memory_mapping,
                        AccessType::Store,
                        ref_of_len_in_input_buffer,
                        u64,
                        self.loader_id
                    )?;

                    accounts.push(Rc::new(RefCell::new(Account {
                        lamports: *lamports,
                        data: data.to_vec(),
                        executable: account_info.executable,
                        owner: *owner,
                        rent_epoch: account_info.rent_epoch,
                    })));
                    refs.push(AccountReferences {
                        lamports,
                        owner,
                        data,
                        ref_to_len_in_vm,
                        serialized_len_ptr,
                    });
                    continue 'root;
                }
            }
            return Err(SyscallError::InstructionError(InstructionError::MissingAccount).into());
        }

        Ok((accounts, refs))
    }

    fn translate_signers(
        &self,
        program_id: &Pubkey,
        signers_seeds_addr: u64,
        signers_seeds_len: usize,
        memory_mapping: &MemoryMapping,
    ) -> Result<Vec<Pubkey>, EbpfError<BPFError>> {
        if signers_seeds_len > 0 {
            let signers_seeds = translate_slice!(
                memory_mapping,
                AccessType::Load,
                signers_seeds_addr,
                SolSignerSeedC,
                signers_seeds_len,
                self.loader_id
            )?;
            Ok(signers_seeds
                .iter()
                .map(|signer_seeds| {
                    let seeds = translate_slice!(
                        memory_mapping,
                        AccessType::Store,
                        signer_seeds.addr,
                        SolSignerSeedC,
                        signer_seeds.len,
                        self.loader_id
                    )?;
                    let seeds_bytes = seeds
                        .iter()
                        .map(|seed| {
                            translate_slice!(
                                memory_mapping,
                                AccessType::Load,
                                seed.addr,
                                u8,
                                seed.len,
                                self.loader_id
                            )
                        })
                        .collect::<Result<Vec<_>, EbpfError<BPFError>>>()?;
                    Pubkey::create_program_address(&seeds_bytes, program_id)
                        .map_err(|err| SyscallError::BadSeeds(err).into())
                })
                .collect::<Result<Vec<_>, EbpfError<BPFError>>>()?)
        } else {
            Ok(vec![])
        }
    }
}
impl<'a> SyscallObject<BPFError> for SyscallInvokeSignedC<'a> {
    fn call(
        &mut self,
        instruction_addr: u64,
        account_infos_addr: u64,
        account_infos_len: u64,
        signers_seeds_addr: u64,
        signers_seeds_len: u64,
        memory_mapping: &MemoryMapping,
    ) -> Result<u64, EbpfError<BPFError>> {
        call(
            self,
            instruction_addr,
            account_infos_addr,
            account_infos_len,
            signers_seeds_addr,
            signers_seeds_len,
            memory_mapping,
        )
    }
}

fn verify_instruction<'a>(
    syscall: &dyn SyscallInvokeSigned<'a>,
    instruction: &Instruction,
    signers: &[Pubkey],
) -> Result<(), EbpfError<BPFError>> {
    let callers_keyed_accounts = syscall.get_callers_keyed_accounts();

    // Check for privilege escalation
    for account in instruction.accounts.iter() {
        let keyed_account = callers_keyed_accounts
            .iter()
            .find_map(|keyed_account| {
                if &account.pubkey == keyed_account.unsigned_key() {
                    Some(keyed_account)
                } else {
                    None
                }
            })
            .ok_or(SyscallError::InstructionError(
                InstructionError::MissingAccount,
            ))?;
        // Readonly account cannot become writable
        if account.is_writable && !keyed_account.is_writable() {
            return Err(SyscallError::PrivilegeEscalation.into());
        }

        if account.is_signer && // If message indicates account is signed
        !( // one of the following needs to be true:
            keyed_account.signer_key().is_some() // Signed in the parent instruction
            || signers.contains(&account.pubkey) // Signed by the program
        ) {
            return Err(SyscallError::PrivilegeEscalation.into());
        }
    }

    Ok(())
}

/// Call process instruction, common to both Rust and C
fn call<'a>(
    syscall: &mut dyn SyscallInvokeSigned<'a>,
    instruction_addr: u64,
    account_infos_addr: u64,
    account_infos_len: u64,
    signers_seeds_addr: u64,
    signers_seeds_len: u64,
    memory_mapping: &MemoryMapping,
) -> Result<u64, EbpfError<BPFError>> {
    let mut invoke_context = syscall.get_context_mut()?;
    invoke_context
        .get_compute_meter()
        .consume(invoke_context.get_compute_budget().invoke_units)?;

    // Translate data passed from the VM

    let instruction = syscall.translate_instruction(instruction_addr, &memory_mapping)?;
    let caller_program_id = invoke_context
        .get_caller()
        .map_err(SyscallError::InstructionError)?;
    let signers = syscall.translate_signers(
        caller_program_id,
        signers_seeds_addr,
        signers_seeds_len as usize,
        memory_mapping,
    )?;
    verify_instruction(syscall, &instruction, &signers)?;
    invoke_context.record_instruction(&instruction);
    let message = Message::new(&[instruction], None);
    let callee_program_id_index = message.instructions[0].program_id_index as usize;
    let callee_program_id = message.account_keys[callee_program_id_index];
    let (accounts, account_refs) = syscall.translate_accounts(
        &message,
        account_infos_addr,
        account_infos_len as usize,
        memory_mapping,
    )?;

    // Process instruction

    let program_account = (*accounts[callee_program_id_index]).clone();
    if !program_account.borrow().executable {
        return Err(SyscallError::InstructionError(InstructionError::AccountNotExecutable).into());
    }
    let executable_accounts = vec![(callee_program_id, program_account)];
    let mut message_processor = MessageProcessor::default();
    for (program_id, process_instruction) in invoke_context.get_programs().iter() {
        message_processor.add_program(*program_id, *process_instruction);
    }
    message_processor.add_loader(bpf_loader::id(), crate::process_instruction);
    message_processor.add_loader(bpf_loader_deprecated::id(), crate::process_instruction);

    #[allow(clippy::deref_addrof)]
    match message_processor.process_cross_program_instruction(
        &message,
        &executable_accounts,
        &accounts,
        *(&mut *invoke_context),
    ) {
        Ok(()) => (),
        Err(err) => match ProgramError::try_from(err) {
            Ok(err) => return Ok(err.into()),
            Err(err) => return Err(SyscallError::InstructionError(err).into()),
        },
    }

    // Copy results back into caller's AccountInfos

    for (i, (account, account_ref)) in accounts.iter().zip(account_refs).enumerate() {
        let account = account.borrow();
        if message.is_writable(i) && !account.executable {
            *account_ref.lamports = account.lamports;
            *account_ref.owner = account.owner;
            if account_ref.data.len() != account.data.len() {
                *account_ref.ref_to_len_in_vm = account.data.len() as u64;
                *account_ref.serialized_len_ptr = account.data.len() as u64;
                if !account_ref.data.is_empty() {
                    // Only support for `CreateAccount` at this time.
                    // Need a way to limit total realloc size accross multiple CPI calls
                    return Err(
                        SyscallError::InstructionError(InstructionError::InvalidRealloc).into(),
                    );
                }
                if account.data.len() > account_ref.data.len() + MAX_PERMITTED_DATA_INCREASE {
                    return Err(
                        SyscallError::InstructionError(InstructionError::InvalidRealloc).into(),
                    );
                }
            }
            account_ref
                .data
                .clone_from_slice(&account.data[0..account_ref.data.len()]);
        }
    }

    Ok(SUCCESS)
}

#[cfg(test)]
mod tests {
    use super::*;
    use solana_rbpf::memory_region::MemoryRegion;
    use solana_runtime::bpf_test_utils::{MockComputeMeter, MockLogger};
    use solana_sdk::hash::hashv;
    use std::str::FromStr;

    macro_rules! assert_access_violation {
        ($result:expr, $va:expr, $len:expr) => {
            match $result {
                Err(EbpfError::AccessViolation(_, _, va, len, _)) if $va == va && len == len => (),
                _ => panic!(),
            }
        };
    }

    #[test]
    fn test_translate() {
        const START: u64 = 100;
        const LENGTH: u64 = 1000;
        let data = vec![0u8; LENGTH as usize];
        let addr = data.as_ptr() as u64;
        let memory_mapping = MemoryMapping::new_from_regions(vec![MemoryRegion::new_from_slice(
            &data, START, false,
        )]);

        let cases = vec![
            (true, START, 0, addr),
            (true, START, 1, addr),
            (true, START, LENGTH, addr),
            (true, START + 1, LENGTH - 1, addr + 1),
            (false, START + 1, LENGTH, 0),
            (true, START + LENGTH - 1, 1, addr + LENGTH - 1),
            (true, START + LENGTH, 0, addr + LENGTH),
            (false, START + LENGTH, 1, 0),
            (false, START, LENGTH + 1, 0),
            (false, 0, 0, 0),
            (false, 0, 1, 0),
            (false, START - 1, 0, 0),
            (false, START - 1, 1, 0),
            (true, START + LENGTH / 2, LENGTH / 2, addr + LENGTH / 2),
        ];
        for (ok, start, length, value) in cases {
            if ok {
                assert_eq!(
                    translate!(
                        memory_mapping,
                        AccessType::Load,
                        start,
                        length,
                        &bpf_loader::id()
                    )
                    .unwrap(),
                    value
                )
            } else {
                assert!(translate!(
                    memory_mapping,
                    AccessType::Load,
                    start,
                    length,
                    &bpf_loader::id()
                )
                .is_err())
            }
        }
    }

    #[test]
    fn test_translate_type() {
        // Pubkey
        let pubkey = solana_sdk::pubkey::new_rand();
        let addr = &pubkey as *const _ as u64;
        let memory_mapping = MemoryMapping::new_from_regions(vec![MemoryRegion {
            host_addr: addr,
            vm_addr: 100,
            len: std::mem::size_of::<Pubkey>() as u64,
            is_writable: false,
        }]);
        let translated_pubkey = translate_type!(
            memory_mapping,
            AccessType::Load,
            100,
            Pubkey,
            &bpf_loader::id()
        )
        .unwrap();
        assert_eq!(pubkey, *translated_pubkey);

        // Instruction
        let instruction = Instruction::new(
            solana_sdk::pubkey::new_rand(),
            &"foobar",
            vec![AccountMeta::new(solana_sdk::pubkey::new_rand(), false)],
        );
        let addr = &instruction as *const _ as u64;
        let mut memory_mapping = MemoryMapping::new_from_regions(vec![MemoryRegion {
            host_addr: addr,
            vm_addr: 96,
            len: std::mem::size_of::<Instruction>() as u64,
            is_writable: false,
        }]);
        let translated_instruction = translate_type!(
            memory_mapping,
            AccessType::Load,
            96,
            Instruction,
            &bpf_loader::id()
        )
        .unwrap();
        assert_eq!(instruction, *translated_instruction);
        memory_mapping.resize_region::<BPFError>(0, 1).unwrap();
        assert!(translate_type!(
            memory_mapping,
            AccessType::Load,
            100,
            Instruction,
            &bpf_loader::id()
        )
        .is_err());
    }

    #[test]
    fn test_translate_slice() {
        // zero len
        let good_data = vec![1u8, 2, 3, 4, 5];
        let data: Vec<u8> = vec![];
        assert_eq!(0x1 as *const u8, data.as_ptr());
        let addr = good_data.as_ptr() as *const _ as u64;
        let memory_mapping = MemoryMapping::new_from_regions(vec![MemoryRegion {
            host_addr: addr,
            vm_addr: 100,
            len: good_data.len() as u64,
            is_writable: false,
        }]);
        let translated_data = translate_slice!(
            memory_mapping,
            AccessType::Load,
            data.as_ptr(),
            u8,
            data.len(),
            &bpf_loader::id()
        )
        .unwrap();
        assert_eq!(data, translated_data);
        assert_eq!(0, translated_data.len());

        // u8
        let mut data = vec![1u8, 2, 3, 4, 5];
        let addr = data.as_ptr() as *const _ as u64;
        let memory_mapping = MemoryMapping::new_from_regions(vec![MemoryRegion {
            host_addr: addr,
            vm_addr: 100,
            len: data.len() as u64,
            is_writable: false,
        }]);
        let translated_data = translate_slice!(
            memory_mapping,
            AccessType::Load,
            100,
            u8,
            data.len(),
            &bpf_loader::id()
        )
        .unwrap();
        assert_eq!(data, translated_data);
        data[0] = 10;
        assert_eq!(data, translated_data);

        // Pubkeys
        let mut data = vec![solana_sdk::pubkey::new_rand(); 5];
        let addr = data.as_ptr() as *const _ as u64;
        let memory_mapping = MemoryMapping::new_from_regions(vec![MemoryRegion {
            host_addr: addr,
            vm_addr: 100,
            len: (data.len() * std::mem::size_of::<Pubkey>()) as u64,
            is_writable: false,
        }]);
        let translated_data = translate_slice!(
            memory_mapping,
            AccessType::Load,
            100,
            Pubkey,
            data.len(),
            &bpf_loader::id()
        )
        .unwrap();
        assert_eq!(data, translated_data);
        data[0] = solana_sdk::pubkey::new_rand(); // Both should point to same place
        assert_eq!(data, translated_data);
    }

    #[test]
    fn test_translate_string_and_do() {
        let string = "Gaggablaghblagh!";
        let addr = string.as_ptr() as *const _ as u64;
        let memory_mapping = MemoryMapping::new_from_regions(vec![MemoryRegion {
            host_addr: addr,
            vm_addr: 100,
            len: string.len() as u64,
            is_writable: false,
        }]);
        assert_eq!(
            42,
            translate_string_and_do(
                &memory_mapping,
                AccessType::Load,
                100,
                string.len() as u64,
                &bpf_loader::id(),
                &mut |string: &str| {
                    assert_eq!(string, "Gaggablaghblagh!");
                    Ok(42)
                }
            )
            .unwrap()
        );
    }

    #[test]
    #[should_panic(expected = "UserError(SyscallError(Abort))")]
    fn test_syscall_abort() {
        let memory_mapping = MemoryMapping::new_from_regions(vec![MemoryRegion::default()]);
        syscall_abort(0, 0, 0, 0, 0, &memory_mapping).unwrap();
    }

    #[test]
    #[should_panic(expected = "UserError(SyscallError(Panic(\"Gaggablaghblagh!\", 42, 84)))")]
    fn test_syscall_sol_panic() {
        let string = "Gaggablaghblagh!";
        let addr = string.as_ptr() as *const _ as u64;
        let memory_mapping = MemoryMapping::new_from_regions(vec![MemoryRegion {
            host_addr: addr,
            vm_addr: 100,
            len: string.len() as u64,
            is_writable: false,
        }]);
        let mut syscall_panic = SyscallPanic {
            loader_id: &bpf_loader::id(),
        };
        syscall_panic
            .call(100, string.len() as u64, 42, 84, 0, &memory_mapping)
            .unwrap();
    }

    #[test]
    fn test_syscall_sol_log() {
        let string = "Gaggablaghblagh!";
        let addr = string.as_ptr() as *const _ as u64;

        let compute_meter: Rc<RefCell<dyn ComputeMeter>> =
            Rc::new(RefCell::new(MockComputeMeter { remaining: 3 }));
        let log = Rc::new(RefCell::new(vec![]));
        let logger: Rc<RefCell<dyn Logger>> =
            Rc::new(RefCell::new(MockLogger { log: log.clone() }));
        let mut syscall_sol_log = SyscallLog {
            cost: 1,
            compute_meter,
            logger,
            loader_id: &bpf_loader::id(),
        };
        let memory_mapping = MemoryMapping::new_from_regions(vec![MemoryRegion {
            host_addr: addr,
            vm_addr: 100,
            len: string.len() as u64,
            is_writable: false,
        }]);

        syscall_sol_log
            .call(100, string.len() as u64, 0, 0, 0, &memory_mapping)
            .unwrap();
        assert_eq!(log.borrow().len(), 1);
        assert_eq!(log.borrow()[0], "Program log: Gaggablaghblagh!");

        assert_access_violation!(
            syscall_sol_log.call(
                101, // AccessViolation
                string.len() as u64,
                0,
                0,
                0,
                &memory_mapping
            ),
            101,
            string.len() as u64
        );
        assert_access_violation!(
            syscall_sol_log.call(
                100,
                string.len() as u64 * 2, // AccessViolation
                0,
                0,
                0,
                &memory_mapping
            ),
            100,
            string.len() as u64 * 2
        );

        assert_eq!(
            Err(EbpfError::UserError(BPFError::SyscallError(
                SyscallError::InstructionError(InstructionError::ComputationalBudgetExceeded)
            ))),
            syscall_sol_log.call(100, string.len() as u64, 0, 0, 0, &memory_mapping)
        );
    }

    #[test]
    fn test_syscall_sol_log_u64() {
        let compute_meter: Rc<RefCell<dyn ComputeMeter>> =
            Rc::new(RefCell::new(MockComputeMeter {
                remaining: std::u64::MAX,
            }));
        let log = Rc::new(RefCell::new(vec![]));
        let logger: Rc<RefCell<dyn Logger>> =
            Rc::new(RefCell::new(MockLogger { log: log.clone() }));
        let mut syscall_sol_log_u64 = SyscallLogU64 {
            cost: 0,
            compute_meter,
            logger,
        };
        let memory_mapping = MemoryMapping::new_from_regions(vec![]);

        syscall_sol_log_u64
            .call(1, 2, 3, 4, 5, &memory_mapping)
            .unwrap();

        assert_eq!(log.borrow().len(), 1);
        assert_eq!(log.borrow()[0], "Program log: 0x1, 0x2, 0x3, 0x4, 0x5");
    }

    #[test]
    fn test_syscall_sol_pubkey() {
        let pubkey = Pubkey::from_str("MoqiU1vryuCGQSxFKA1SZ316JdLEFFhoAu6cKUNk7dN").unwrap();
        let addr = &pubkey.as_ref()[0] as *const _ as u64;

        let compute_meter: Rc<RefCell<dyn ComputeMeter>> =
            Rc::new(RefCell::new(MockComputeMeter { remaining: 2 }));
        let log = Rc::new(RefCell::new(vec![]));
        let logger: Rc<RefCell<dyn Logger>> =
            Rc::new(RefCell::new(MockLogger { log: log.clone() }));
        let mut syscall_sol_pubkey = SyscallLogPubkey {
            cost: 1,
            compute_meter,
            logger,
            loader_id: &bpf_loader::id(),
        };
        let memory_mapping = MemoryMapping::new_from_regions(vec![MemoryRegion {
            host_addr: addr,
            vm_addr: 100,
            len: 32,
            is_writable: false,
        }]);

        syscall_sol_pubkey
            .call(100, 0, 0, 0, 0, &memory_mapping)
            .unwrap();
        assert_eq!(log.borrow().len(), 1);
        assert_eq!(
            log.borrow()[0],
            "Program log: MoqiU1vryuCGQSxFKA1SZ316JdLEFFhoAu6cKUNk7dN"
        );

        assert_access_violation!(
            syscall_sol_pubkey.call(
                101, // AccessViolation
                32,
                0,
                0,
                0,
                &memory_mapping,
            ),
            101,
            32
        );

        assert_eq!(
            Err(EbpfError::UserError(BPFError::SyscallError(
                SyscallError::InstructionError(InstructionError::ComputationalBudgetExceeded)
            ))),
            syscall_sol_pubkey.call(100, 32, 0, 0, 0, &memory_mapping)
        );
    }

    #[test]
    fn test_syscall_sol_alloc_free() {
        // large alloc
        {
            let heap = vec![0_u8; 100];
            let memory_mapping =
                MemoryMapping::new_from_regions(vec![MemoryRegion::new_from_slice(
                    &heap,
                    MM_HEAP_START,
                    true,
                )]);
            let mut syscall = SyscallAllocFree {
                aligned: true,
                allocator: BPFAllocator::new(heap, MM_HEAP_START),
            };
            assert_ne!(syscall.call(100, 0, 0, 0, 0, &memory_mapping).unwrap(), 0);
            assert_eq!(syscall.call(100, 0, 0, 0, 0, &memory_mapping).unwrap(), 0);
            assert_eq!(
                syscall.call(u64::MAX, 0, 0, 0, 0, &memory_mapping).unwrap(),
                0
            );
        }
        // many small unaligned allocs
        {
            let heap = vec![0_u8; 100];
            let memory_mapping =
                MemoryMapping::new_from_regions(vec![MemoryRegion::new_from_slice(
                    &heap,
                    MM_HEAP_START,
                    true,
                )]);
            let mut syscall = SyscallAllocFree {
                aligned: false,
                allocator: BPFAllocator::new(heap, MM_HEAP_START),
            };
            for _ in 0..100 {
                assert_ne!(syscall.call(1, 0, 0, 0, 0, &memory_mapping).unwrap(), 0);
            }
            assert_eq!(syscall.call(100, 0, 0, 0, 0, &memory_mapping).unwrap(), 0);
        }
        // many small aligned allocs
        {
            let heap = vec![0_u8; 100];
            let memory_mapping =
                MemoryMapping::new_from_regions(vec![MemoryRegion::new_from_slice(
                    &heap,
                    MM_HEAP_START,
                    true,
                )]);
            let mut syscall = SyscallAllocFree {
                aligned: true,
                allocator: BPFAllocator::new(heap, MM_HEAP_START),
            };
            for _ in 0..12 {
                assert_ne!(syscall.call(1, 0, 0, 0, 0, &memory_mapping).unwrap(), 0);
            }
            assert_eq!(syscall.call(100, 0, 0, 0, 0, &memory_mapping).unwrap(), 0);
        }
        // aligned allocs

        fn check_alignment<T>() {
            let heap = vec![0_u8; 100];
            let memory_mapping =
                MemoryMapping::new_from_regions(vec![MemoryRegion::new_from_slice(
                    &heap,
                    MM_HEAP_START,
                    true,
                )]);
            let mut syscall = SyscallAllocFree {
                aligned: true,
                allocator: BPFAllocator::new(heap, MM_HEAP_START),
            };
            let address = syscall
                .call(size_of::<u8>() as u64, 0, 0, 0, 0, &memory_mapping)
                .unwrap();
            assert_ne!(address, 0);
            assert_eq!((address as *const u8).align_offset(align_of::<u8>()), 0);
        }
        check_alignment::<u8>();
        check_alignment::<u16>();
        check_alignment::<u32>();
        check_alignment::<u64>();
        check_alignment::<u128>();
    }

    #[test]
    fn test_syscall_sha256() {
        let bytes1 = "Gaggablaghblagh!";
        let bytes2 = "flurbos";

        struct MockSlice {
            pub addr: u64,
            pub len: usize,
        }
        let mock_slice1 = MockSlice {
            addr: 4096,
            len: bytes1.len(),
        };
        let mock_slice2 = MockSlice {
            addr: 8192,
            len: bytes2.len(),
        };
        let bytes_to_hash = [mock_slice1, mock_slice2]; // TODO
        let hash_result = [0; HASH_BYTES];
        let ro_len = bytes_to_hash.len() as u64;
        let ro_va = 96;
        let rw_va = 192;
        let memory_mapping = MemoryMapping::new_from_regions(vec![
            MemoryRegion {
                host_addr: bytes1.as_ptr() as *const _ as u64,
                vm_addr: 4096,
                len: bytes1.len() as u64,
                is_writable: false,
            },
            MemoryRegion {
                host_addr: bytes2.as_ptr() as *const _ as u64,
                vm_addr: 8192,
                len: bytes2.len() as u64,
                is_writable: false,
            },
            MemoryRegion {
                host_addr: bytes_to_hash.as_ptr() as *const _ as u64,
                vm_addr: 96,
                len: 32,
                is_writable: false,
            },
            MemoryRegion {
                host_addr: hash_result.as_ptr() as *const _ as u64,
                vm_addr: rw_va,
                len: HASH_BYTES as u64,
                is_writable: true,
            },
        ]);
        let compute_meter: Rc<RefCell<dyn ComputeMeter>> =
            Rc::new(RefCell::new(MockComputeMeter {
                remaining: (bytes1.len() + bytes2.len()) as u64,
            }));
        let mut syscall = SyscallSha256 {
            sha256_base_cost: 0,
            sha256_byte_cost: 2,
            compute_meter,
            loader_id: &bpf_loader_deprecated::id(),
        };

        syscall
            .call(ro_va, ro_len, rw_va, 0, 0, &memory_mapping)
            .unwrap();

        let hash_local = hashv(&[bytes1.as_ref(), bytes2.as_ref()]).to_bytes();
        assert_eq!(hash_result, hash_local);

        assert_access_violation!(
            syscall.call(
                ro_va - 1, // AccessViolation
                ro_len,
                rw_va,
                0,
                0,
                &memory_mapping
            ),
            ro_va - 1,
            ro_len
        );
        assert_access_violation!(
            syscall.call(
                ro_va,
                ro_len + 1, // AccessViolation
                rw_va,
                0,
                0,
                &memory_mapping
            ),
            ro_va,
            ro_len + 1
        );
        assert_access_violation!(
            syscall.call(
                ro_va,
                ro_len,
                rw_va - 1, // AccessViolation
                0,
                0,
                &memory_mapping
            ),
            rw_va - 1,
            HASH_BYTES as u64
        );

        assert_eq!(
            Err(EbpfError::UserError(BPFError::SyscallError(
                SyscallError::InstructionError(InstructionError::ComputationalBudgetExceeded)
            ))),
            syscall.call(ro_va, ro_len, rw_va, 0, 0, &memory_mapping)
        );
    }
}