1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

use crate::{types::RawHandle, vm_err_msg, vm_hooks::VMHooksHandlerSource};
use num_traits::Zero;

use super::VMHooksManagedTypes;

pub trait VMHooksCallValue: VMHooksHandlerSource + VMHooksManagedTypes {
    fn check_not_payable(&self) {
        if self.input_ref().moax_value > num_bigint::BigUint::zero() {
            self.vm_error(vm_err_msg::NON_PAYABLE_FUNC_MOAX);
        }
        if self.dct_num_transfers() > 0 {
            self.vm_error(vm_err_msg::NON_PAYABLE_FUNC_DCT);
        }
    }

    fn load_moax_value(&self, dest: RawHandle) {
        let value = self.input_ref().received_moax().clone();
        self.m_types_lock().bi_overwrite(dest, value.into());
    }

    fn load_all_dct_transfers(&self, dest_handle: RawHandle) {
        let transfers = self.input_ref().received_dct();
        self.m_types_lock()
            .mb_set_vec_of_dct_payments(dest_handle, transfers);
    }

    fn dct_num_transfers(&self) -> usize {
        self.input_ref().received_dct().len()
    }
}