ckb-script 1.1.0

CKB component to run the type/lock scripts
Documentation 
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

use crate::{
    cost_model::transferred_byte_cycles,
    syscalls::{
        LOAD_TRANSACTION_SYSCALL_NUMBER, LOAD_TX_HASH_SYSCALL_NUMBER, SUCCESS, utils::store_data,
    },
    types::SgData,
};
use ckb_types::{core::cell::ResolvedTransaction, prelude::*};
use ckb_vm::{
    Error as VMError, Register, SupportMachine, Syscalls,
    registers::{A0, A7},
};
use std::sync::Arc;

#[derive(Debug)]
pub struct LoadTx {
    rtx: Arc<ResolvedTransaction>,
}

impl LoadTx {
    pub fn new<DL>(sg_data: &SgData<DL>) -> LoadTx {
        LoadTx {
            rtx: Arc::clone(&sg_data.rtx),
        }
    }
}

impl<Mac: SupportMachine> Syscalls<Mac> for LoadTx {
    fn initialize(&mut self, _machine: &mut Mac) -> Result<(), VMError> {
        Ok(())
    }

    fn ecall(&mut self, machine: &mut Mac) -> Result<bool, VMError> {
        let wrote_size = match machine.registers()[A7].to_u64() {
            LOAD_TX_HASH_SYSCALL_NUMBER => {
                store_data(machine, self.rtx.transaction.hash().as_slice())?
            }
            LOAD_TRANSACTION_SYSCALL_NUMBER => {
                store_data(machine, self.rtx.transaction.data().as_slice())?
            }
            _ => return Ok(false),
        };

        machine.add_cycles_no_checking(transferred_byte_cycles(wrote_size))?;
        machine.set_register(A0, Mac::REG::from_u8(SUCCESS));
        Ok(true)
    }
}