ckb-testtool 1.1.1

use crate::tx_verifier::OutputsDataVerifier;
use ckb_chain_spec::consensus::{ConsensusBuilder, TYPE_ID_CODE_HASH};
use ckb_error::Error as CKBError;
use ckb_mock_tx_types::{MockCellDep, MockInfo, MockInput, MockTransaction, ReprMockTransaction};
#[cfg(feature = "native-simulator")]
use ckb_script::ScriptError;
use ckb_script::{TransactionScriptsVerifier, TxVerifyEnv};
use ckb_traits::{CellDataProvider, ExtensionProvider, HeaderProvider};
use ckb_types::{
    bytes::Bytes,
    core::{
        Capacity, Cycle, DepType, EpochExt, HeaderBuilder, HeaderView, ScriptHashType,
        TransactionInfo, TransactionView,
        cell::{CellMetaBuilder, ResolvedTransaction},
        hardfork::{CKB2021, CKB2023, HardForks},
    },
    packed::{Byte32, CellDep, CellDepBuilder, CellOutput, OutPoint, OutPointVec, Script},
    prelude::*,
};
use rand::{Rng, SeedableRng, rngs::StdRng, thread_rng};
use std::collections::HashMap;
use std::env;
use std::path::PathBuf;
use std::sync::{Arc, Mutex};

/// Return a random hash.
pub fn random_hash() -> Byte32 {
    let mut rng = thread_rng();
    let mut buf = [0u8; 32];
    rng.fill(&mut buf);
    buf.pack()
}

/// Return a random OutPoint.
pub fn random_out_point() -> OutPoint {
    OutPoint::new_builder().tx_hash(random_hash()).build()
}

/// Return a random Type ID Script.
pub fn random_type_id_script() -> Script {
    let args = random_hash().as_bytes();
    debug_assert_eq!(args.len(), 32);
    Script::new_builder()
        .code_hash(TYPE_ID_CODE_HASH.pack())
        .hash_type(ScriptHashType::Type)
        .args(args.pack())
        .build()
}

/// A single debug message. By setting context.capture_debug, you can capture debug syscalls issued by your script.
#[derive(Debug, Clone, Eq, PartialEq)]
pub struct Message {
    /// The current script hash.
    pub id: Byte32,
    /// Payload.
    pub message: String,
}

/// Verification Context.
#[derive(Clone)]
pub struct Context {
    pub cells: HashMap<OutPoint, (CellOutput, Bytes)>,
    pub transaction_infos: HashMap<OutPoint, TransactionInfo>,
    pub headers: HashMap<Byte32, HeaderView>,
    pub epoches: HashMap<Byte32, EpochExt>,
    pub block_extensions: HashMap<Byte32, Bytes>,
    pub cells_by_data_hash: HashMap<Byte32, OutPoint>,
    pub cells_by_type_hash: HashMap<Byte32, OutPoint>,
    deterministic_rng: bool,
    capture_debug: bool,
    captured_messages: Arc<Mutex<Vec<Message>>>,
    contracts_dirs: Vec<PathBuf>,
    #[cfg(feature = "native-simulator")]
    simulator_binaries: HashMap<Byte32, PathBuf>,
    #[cfg(feature = "native-simulator")]
    simulator_bin_name: String,
}

impl Default for Context {
    fn default() -> Self {
        // The search directory for scripts is $TOP/build/$MODE. You can change this path by setting the $TOP and $MODE
        // environment variables. If you do nothing, the default path is ../build/release.
        let mut contracts_dir = env::var("TOP").map(PathBuf::from).unwrap_or_default();
        contracts_dir.push("build");
        if !contracts_dir.exists() {
            contracts_dir.pop();
            contracts_dir.push("../build");
        }
        let contracts_dir = contracts_dir.join(env::var("MODE").unwrap_or("release".to_string()));

        Self {
            cells: Default::default(),
            transaction_infos: Default::default(),
            headers: Default::default(),
            epoches: Default::default(),
            block_extensions: Default::default(),
            cells_by_data_hash: Default::default(),
            cells_by_type_hash: Default::default(),
            deterministic_rng: false,
            capture_debug: Default::default(),
            captured_messages: Default::default(),
            contracts_dirs: vec![contracts_dir],
            #[cfg(feature = "native-simulator")]
            simulator_binaries: Default::default(),
            #[cfg(feature = "native-simulator")]
            simulator_bin_name: "lib<contract>_sim".to_string(),
        }
    }
}

impl Context {
    /// Create a new context with a deterministic random number generator, which can be used to generate deterministic
    /// out point of deployed contract.
    pub fn new_with_deterministic_rng() -> Self {
        Self {
            deterministic_rng: true,
            ..Default::default()
        }
    }

    /// Add new script search paths. Note that this does not replace the default search paths.
    pub fn add_contract_dir(&mut self, path: &str) {
        self.contracts_dirs.push(path.into());
    }

    /// Deploy a cell.
    pub fn deploy_cell(&mut self, data: Bytes) -> OutPoint {
        let data_hash = CellOutput::calc_data_hash(&data);
        if let Some(out_point) = self.cells_by_data_hash.get(&data_hash) {
            // Contract has already been deployed.
            return out_point.to_owned();
        }
        let (out_point, type_id_script) = if self.deterministic_rng {
            let mut rng = StdRng::from_seed(data_hash.as_slice().try_into().unwrap());
            let mut tx_hash = [0u8; 32];
            rng.fill(&mut tx_hash);
            let mut script_args = [0u8; 32];
            rng.fill(&mut script_args);
            (
                OutPoint::new_builder().tx_hash(tx_hash.pack()).build(),
                Script::new_builder()
                    .code_hash(TYPE_ID_CODE_HASH.pack())
                    .hash_type(ScriptHashType::Type)
                    .args(script_args.as_slice().pack())
                    .build(),
            )
        } else {
            (random_out_point(), random_type_id_script())
        };
        let type_id_hash = type_id_script.calc_script_hash();
        let cell = {
            let cell = CellOutput::new_builder()
                .type_(Some(type_id_script).pack())
                .build();
            let occupied_capacity = cell
                .occupied_capacity(Capacity::bytes(data.len()).expect("data occupied capacity"))
                .expect("cell capacity");
            cell.as_builder().capacity(occupied_capacity.pack()).build()
        };
        self.cells.insert(out_point.clone(), (cell, data));
        self.cells_by_data_hash.insert(data_hash, out_point.clone());
        self.cells_by_type_hash
            .insert(type_id_hash, out_point.clone());
        out_point
    }

    /// Deploy a cell by filename. It provides the same functionality as the deploy_cell function, but looks for data
    /// in the file system.
    pub fn deploy_cell_by_name(&mut self, filename: &str) -> OutPoint {
        let path = self.get_contract_path(filename).expect("get contract path");
        let data = std::fs::read(&path).unwrap_or_else(|_| panic!("read local file: {:?}", path));

        #[cfg(feature = "native-simulator")]
        {
            let native_path = self.get_native_simulator_path(filename);
            if native_path.is_some() {
                let code_hash = CellOutput::calc_data_hash(&data);
                self.simulator_binaries
                    .insert(code_hash, native_path.unwrap());
            }
        }

        self.deploy_cell(data.into())
    }

    /// Get the full path of the specified script.
    fn get_contract_path(&self, filename: &str) -> Option<PathBuf> {
        for dir in &self.contracts_dirs {
            let path = dir.join(filename);
            if path.is_file() {
                return Some(path);
            }
        }
        None
    }

    #[cfg(feature = "native-simulator")]
    /// Get the full path of the specified script. Only useful in simulator mode.
    fn get_native_simulator_path(&self, filename: &str) -> Option<PathBuf> {
        let cdylib_name = format!(
            "{}.{}",
            self.simulator_bin_name
                .replace("<contract>", &filename.replace("-", "_")),
            std::env::consts::DLL_EXTENSION
        );
        for dir in &self.contracts_dirs {
            let path = dir.join(&cdylib_name);
            if path.is_file() {
                return Some(path);
            }
        }
        None
    }

    /// Insert a block header into context. Afterwards, the header can be retrieved by its hash.
    pub fn insert_header(&mut self, header: HeaderView) {
        self.headers.insert(header.hash(), header);
    }

    /// Link a cell with a block to make the load_header_by_cell syscalls works.
    pub fn link_cell_with_block(
        &mut self,
        out_point: OutPoint,
        block_hash: Byte32,
        tx_index: usize,
    ) {
        let header = self
            .headers
            .get(&block_hash)
            .expect("can't find the header");
        self.transaction_infos.insert(
            out_point,
            TransactionInfo::new(header.number(), header.epoch(), block_hash, tx_index),
        );
    }

    /// Get the out-point of a cell by data_hash. The cell must has deployed to this context.
    pub fn get_cell_by_data_hash(&self, data_hash: &Byte32) -> Option<OutPoint> {
        self.cells_by_data_hash.get(data_hash).cloned()
    }

    /// Create a cell with data.
    pub fn create_cell(&mut self, cell: CellOutput, data: Bytes) -> OutPoint {
        let out_point = random_out_point();
        self.create_cell_with_out_point(out_point.clone(), cell, data);
        out_point
    }

    /// Create cell with specified out-point and cell data.
    pub fn create_cell_with_out_point(
        &mut self,
        out_point: OutPoint,
        cell: CellOutput,
        data: Bytes,
    ) {
        let data_hash = CellOutput::calc_data_hash(&data);
        self.cells_by_data_hash.insert(data_hash, out_point.clone());
        if let Some(_type) = cell.type_().to_opt() {
            let type_hash = _type.calc_script_hash();
            self.cells_by_type_hash.insert(type_hash, out_point.clone());
        }
        self.cells.insert(out_point, (cell, data));
    }

    /// Get cell output and data by out-point.
    pub fn get_cell(&self, out_point: &OutPoint) -> Option<(CellOutput, Bytes)> {
        self.cells.get(out_point).cloned()
    }

    /// Build script with out_point, hash_type, args. Return none if the out-point is not exist.
    pub fn build_script_with_hash_type(
        &mut self,
        out_point: &OutPoint,
        hash_type: ScriptHashType,
        args: Bytes,
    ) -> Option<Script> {
        let (cell, contract_data) = self.cells.get(out_point)?;
        let code_hash = match hash_type {
            ScriptHashType::Data | ScriptHashType::Data1 | ScriptHashType::Data2 => {
                CellOutput::calc_data_hash(contract_data)
            }
            ScriptHashType::Type => cell
                .type_()
                .to_opt()
                .expect("get cell's type hash")
                .calc_script_hash(),
            _ => unreachable!(),
        };
        Some(
            Script::new_builder()
                .code_hash(code_hash)
                .hash_type(hash_type)
                .args(args.pack())
                .build(),
        )
    }

    /// Build script with out_point, args and hash_type(ScriptHashType::Type). Return none if the out-point is not
    /// exist.
    pub fn build_script(&mut self, out_point: &OutPoint, args: Bytes) -> Option<Script> {
        self.build_script_with_hash_type(out_point, ScriptHashType::Type, args)
    }

    /// Find cell dep for the specified script.
    fn find_cell_dep_for_script(&self, script: &Script) -> CellDep {
        let out_point = match ScriptHashType::try_from(u8::from(script.hash_type()))
            .expect("invalid script hash type")
        {
            ScriptHashType::Data | ScriptHashType::Data1 | ScriptHashType::Data2 => self
                .get_cell_by_data_hash(&script.code_hash())
                .expect("find contract out point by data_hash"),
            ScriptHashType::Type => self
                .cells_by_type_hash
                .get(&script.code_hash())
                .cloned()
                .expect("find contract out point by type_hash"),
            _ => unreachable!(),
        };

        CellDep::new_builder()
            .out_point(out_point)
            .dep_type(DepType::Code)
            .build()
    }

    /// Complete cell deps for a transaction.
    /// This function searches context cells; generate cell dep for referenced scripts.
    pub fn complete_tx(&mut self, tx: TransactionView) -> TransactionView {
        let mut cell_deps: Vec<CellDep> = Vec::new();

        for cell_dep in tx.cell_deps_iter() {
            cell_deps.push(cell_dep);
        }

        for i in tx.input_pts_iter() {
            if let Some((cell, _data)) = self.cells.get(&i) {
                let dep = self.find_cell_dep_for_script(&cell.lock());
                if !cell_deps.contains(&dep) {
                    cell_deps.push(dep);
                }
                if let Some(script) = cell.type_().to_opt() {
                    let dep = self.find_cell_dep_for_script(&script);
                    if !cell_deps.contains(&dep) {
                        cell_deps.push(dep);
                    }
                }
            }
        }

        for (cell, _data) in tx.outputs_with_data_iter() {
            if let Some(script) = cell.type_().to_opt() {
                let dep = self.find_cell_dep_for_script(&script);
                if !cell_deps.contains(&dep) {
                    cell_deps.push(dep);
                }
            }
        }

        tx.as_advanced_builder()
            .set_cell_deps(Vec::new())
            .cell_deps(cell_deps.pack())
            .build()
    }

    /// Build transaction with resolved input cells.
    fn build_resolved_tx(&self, tx: &TransactionView) -> ResolvedTransaction {
        let input_cells = tx
            .inputs()
            .into_iter()
            .map(|input| {
                let previous_out_point = input.previous_output();
                let (input_output, input_data) = self.cells.get(&previous_out_point).unwrap();
                let tx_info_opt = self.transaction_infos.get(&previous_out_point);
                let mut b = CellMetaBuilder::from_cell_output(
                    input_output.to_owned(),
                    input_data.to_vec().into(),
                )
                .out_point(previous_out_point);
                if let Some(tx_info) = tx_info_opt {
                    b = b.transaction_info(tx_info.to_owned());
                }
                b.build()
            })
            .collect();
        let mut resolved_cell_deps = vec![];
        let mut resolved_dep_groups = vec![];
        tx.cell_deps().into_iter().for_each(|cell_dep| {
            let mut out_points = vec![];
            if cell_dep.dep_type() == DepType::DepGroup.into() {
                let (dep_group_output, dep_group_data) =
                    self.cells.get(&cell_dep.out_point()).unwrap();
                let dep_group_tx_info_opt = self.transaction_infos.get(&cell_dep.out_point());
                let mut b = CellMetaBuilder::from_cell_output(
                    dep_group_output.to_owned(),
                    dep_group_data.to_vec().into(),
                )
                .out_point(cell_dep.out_point());
                if let Some(tx_info) = dep_group_tx_info_opt {
                    b = b.transaction_info(tx_info.to_owned());
                }
                resolved_dep_groups.push(b.build());

                let sub_out_points =
                    OutPointVec::from_slice(dep_group_data).expect("Parsing dep group error!");
                out_points.extend(sub_out_points);
            } else {
                out_points.push(cell_dep.out_point());
            }

            for out_point in out_points {
                let (dep_output, dep_data) = self.cells.get(&out_point).unwrap();
                let tx_info_opt = self.transaction_infos.get(&out_point);
                let mut b = CellMetaBuilder::from_cell_output(
                    dep_output.to_owned(),
                    dep_data.to_vec().into(),
                )
                .out_point(out_point);
                if let Some(tx_info) = tx_info_opt {
                    b = b.transaction_info(tx_info.to_owned());
                }
                resolved_cell_deps.push(b.build());
            }
        });
        ResolvedTransaction {
            transaction: tx.clone(),
            resolved_cell_deps,
            resolved_inputs: input_cells,
            resolved_dep_groups,
        }
    }

    /// Check format and consensus rules.
    fn verify_tx_consensus(&self, tx: &TransactionView) -> Result<(), CKBError> {
        OutputsDataVerifier::new(tx).verify()?;
        Ok(())
    }

    /// Return capture_debug flag.
    pub fn capture_debug(&self) -> bool {
        self.capture_debug
    }

    /// Capture debug output, default value is false.
    pub fn set_capture_debug(&mut self, capture_debug: bool) {
        self.capture_debug = capture_debug;
    }

    /// Return captured messages.
    pub fn captured_messages(&self) -> Vec<Message> {
        self.captured_messages.lock().unwrap().clone()
    }

    /// Verify the transaction in CKB-VM.
    pub fn verify_tx(&self, tx: &TransactionView, max_cycles: u64) -> Result<Cycle, CKBError> {
        self.verify_tx_consensus(tx)?;
        let resolved_tx = self.build_resolved_tx(tx);
        let consensus = ConsensusBuilder::default()
            .hardfork_switch(HardForks {
                ckb2021: CKB2021::new_dev_default(),
                ckb2023: CKB2023::new_dev_default(),
            })
            .build();
        let tip = HeaderBuilder::default().number(0).build();
        let tx_verify_env = TxVerifyEnv::new_submit(&tip);
        let verifier = if self.capture_debug {
            let captured_messages = self.captured_messages.clone();
            TransactionScriptsVerifier::new_with_debug_printer(
                Arc::new(resolved_tx),
                self.clone(),
                Arc::new(consensus),
                Arc::new(tx_verify_env),
                Arc::new(move |id, message| {
                    //
                    let msg = Message {
                        id: id.clone(),
                        message: message.to_string(),
                    };
                    captured_messages.lock().unwrap().push(msg);
                }),
            )
        } else {
            TransactionScriptsVerifier::new_with_debug_printer(
                Arc::new(resolved_tx),
                self.clone(),
                Arc::new(consensus),
                Arc::new(tx_verify_env),
                Arc::new(|_id, msg| {
                    println!("[contract debug] {}", msg);
                }),
            )
        };

        #[cfg(feature = "native-simulator")]
        {
            self.native_simulator_verify(tx, verifier, max_cycles)
        }
        #[cfg(not(feature = "native-simulator"))]
        verifier.verify(max_cycles)
    }

    #[cfg(feature = "native-simulator")]
    /// Verify the transaction in simulator mode.
    fn native_simulator_verify<DL>(
        &self,
        tx: &TransactionView,
        verifier: TransactionScriptsVerifier<DL>,
        max_cycles: u64,
    ) -> Result<Cycle, CKBError>
    where
        DL: CellDataProvider + HeaderProvider + ExtensionProvider + Send + Sync + Clone + 'static,
    {
        let mut cycles: Cycle = 0;

        for (hash, group) in verifier.groups() {
            let code_hash = if group.script.hash_type() == ScriptHashType::Type.into() {
                let code_hash = group.script.code_hash();
                let out_point = match self.cells_by_type_hash.get(&code_hash) {
                    Some(out_point) => out_point,
                    None => panic!("unknow code hash(ScriptHashType::Type)"),
                };

                match self.cells.get(out_point) {
                    Some((_cell, bin)) => CellOutput::calc_data_hash(bin),
                    None => panic!("unknow code hash(ScriptHashType::Type) in deps"),
                }
            } else {
                group.script.code_hash()
            };

            let use_cycles = match self.simulator_binaries.get(&code_hash) {
                Some(sim_path) => self
                    .run_simulator(sim_path, tx, group)
                    .map_err(|e| e.source(group))?,
                None => {
                    group.script.code_hash();
                    verifier
                        .verify_single(group.group_type, hash, max_cycles)
                        .map_err(|e| e.source(group))?
                }
            };
            let r = cycles.overflowing_add(use_cycles);
            assert!(!r.1, "cycles overflow");
            cycles = r.0;
        }
        Ok(cycles)
    }

    #[cfg(feature = "native-simulator")]
    /// Run simulator.
    fn run_simulator(
        &self,
        sim_path: &PathBuf,
        tx: &TransactionView,
        group: &ckb_script::ScriptGroup,
    ) -> Result<u64, ScriptError> {
        println!(
            "run native-simulator: {}",
            sim_path.file_name().unwrap().to_str().unwrap()
        );
        let tmp_dir = if !self.simulator_binaries.is_empty() {
            let tmp_dir = std::env::temp_dir().join("ckb-simulator-debugger");
            if !tmp_dir.exists() {
                std::fs::create_dir(tmp_dir.clone())
                    .expect("create tmp dir: ckb-simulator-debugger");
            }
            let tx_file: PathBuf = tmp_dir.join("ckb_running_tx.json");
            let dump_tx = self.dump_tx(&tx).unwrap();

            let tx_json = serde_json::to_string(&dump_tx).expect("dump tx to string");
            std::fs::write(&tx_file, tx_json).expect("write setup");

            unsafe {
                std::env::set_var("CKB_TX_FILE", tx_file.to_str().unwrap());
            }
            Some(tmp_dir)
        } else {
            None
        };
        let running_setup = tmp_dir.as_ref().unwrap().join("ckb_running_setup.json");

        let mut native_binaries = self
            .simulator_binaries
            .iter()
            .map(|(code_hash, path)| {
                let buf = vec![
                    code_hash.as_bytes().to_vec(),
                    vec![0xff],
                    0u32.to_le_bytes().to_vec(),
                    0u32.to_le_bytes().to_vec(),
                ]
                .concat();

                format!(
                    "\"0x{}\" : \"{}\",",
                    faster_hex::hex_string(&buf),
                    path.to_str().unwrap()
                )
            })
            .collect::<Vec<String>>()
            .concat();
        if !native_binaries.is_empty() {
            native_binaries.pop();
        }

        let native_binaries = format!("{{ {} }}", native_binaries);

        // For type scripts that only appear on outputs (e.g., genesis transactions),
        // input_indices is empty. In that case, use output_indices[0] and set is_output to true.
        let (is_output, script_index) = if !group.input_indices.is_empty() {
            (false, group.input_indices[0])
        } else if !group.output_indices.is_empty() {
            (true, group.output_indices[0])
        } else {
            panic!("script group has no inputs or outputs");
        };

        let setup = format!(
            "{{\"is_lock_script\": {}, \"is_output\": {}, \"script_index\": {}, \"vm_version\": {}, \"native_binaries\": {}, \"run_type\": \"DynamicLib\" }}",
            group.group_type == ckb_script::ScriptGroupType::Lock,
            is_output,
            script_index,
            2,
            native_binaries
        );
        std::fs::write(&running_setup, setup).expect("write setup");
        unsafe {
            std::env::set_var("CKB_RUNNING_SETUP", running_setup.to_str().unwrap());
        }

        type CkbMainFunc<'a> =
            libloading::Symbol<'a, unsafe extern "C" fn(argc: i32, argv: *const *const i8) -> i8>;
        type SetScriptInfo<'a> = libloading::Symbol<
            'a,
            unsafe extern "C" fn(ptr: *const std::ffi::c_void, tx_ctx_id: u64, vm_ctx_id: u64),
        >;

        // ckb_x64_simulator::run_native_simulator(sim_path);
        unsafe {
            let lib = libloading::Library::new(sim_path).expect("Load library");

            let func: SetScriptInfo = lib
                .get(b"__set_script_info")
                .expect("load function : __update_spawn_info");
            func(std::ptr::null(), 0, 0);

            let func: CkbMainFunc = lib
                .get(b"__ckb_std_main")
                .expect("load function : __ckb_std_main");
            let argv = vec![];
            let result = func(0, argv.as_ptr());
            if result != 0 {
                return Err(ScriptError::validation_failure(&group.script, result));
            }
        }
        Ok(0)
    }

    #[cfg(feature = "native-simulator")]
    /// Add code_hash and script binary to simulator.
    pub fn set_simulator(&mut self, code_hash: Byte32, path: &str) {
        let path = PathBuf::from(path);
        assert!(path.is_file());
        self.simulator_binaries.insert(code_hash, path);
    }

    /// Dump the transaction in mock transaction format, so we can offload it to ckb debugger.
    pub fn dump_tx(&self, tx: &TransactionView) -> Result<ReprMockTransaction, CKBError> {
        let rtx = self.build_resolved_tx(tx);
        let mut inputs = Vec::with_capacity(rtx.resolved_inputs.len());
        // We are doing it this way so we can keep original since value is available
        for (i, input) in rtx.resolved_inputs.iter().enumerate() {
            inputs.push(MockInput {
                input: rtx.transaction.inputs().get(i).unwrap(),
                output: input.cell_output.clone(),
                data: input.mem_cell_data.clone().unwrap(),
                header: input.transaction_info.clone().map(|info| info.block_hash),
            });
        }
        // MockTransaction keeps both types of cell deps in a single array, the order does
        // not really matter for now
        let mut cell_deps =
            Vec::with_capacity(rtx.resolved_cell_deps.len() + rtx.resolved_dep_groups.len());
        for dep in rtx.resolved_cell_deps.iter() {
            cell_deps.push(MockCellDep {
                cell_dep: CellDepBuilder::default()
                    .out_point(dep.out_point.clone())
                    .dep_type(DepType::Code)
                    .build(),
                output: dep.cell_output.clone(),
                data: dep.mem_cell_data.clone().unwrap(),
                header: dep.transaction_info.clone().map(|info| info.block_hash),
            });
        }
        for dep in rtx.resolved_dep_groups.iter() {
            cell_deps.push(MockCellDep {
                cell_dep: CellDepBuilder::default()
                    .out_point(dep.out_point.clone())
                    .dep_type(DepType::DepGroup)
                    .build(),
                output: dep.cell_output.clone(),
                data: dep.mem_cell_data.clone().unwrap(),
                header: dep.transaction_info.clone().map(|info| info.block_hash),
            });
        }
        let mut header_deps = Vec::with_capacity(rtx.transaction.header_deps().len());
        let mut extensions = Vec::new();
        for header_hash in rtx.transaction.header_deps_iter() {
            header_deps.push(self.get_header(&header_hash).unwrap());
            if let Some(extension) = self.get_block_extension(&header_hash) {
                extensions.push((header_hash, extension.unpack()));
            }
        }
        Ok(MockTransaction {
            mock_info: MockInfo {
                inputs,
                cell_deps,
                header_deps,
                extensions,
            },
            tx: rtx.transaction.data(),
        }
        .into())
    }
}

impl CellDataProvider for Context {
    fn get_cell_data(&self, out_point: &OutPoint) -> Option<Bytes> {
        self.cells
            .get(out_point)
            .map(|(_, data)| Bytes::from(data.to_vec()))
    }

    fn get_cell_data_hash(&self, out_point: &OutPoint) -> Option<Byte32> {
        self.cells
            .get(out_point)
            .map(|(_, data)| CellOutput::calc_data_hash(data))
    }
}

impl HeaderProvider for Context {
    fn get_header(&self, block_hash: &Byte32) -> Option<HeaderView> {
        self.headers.get(block_hash).cloned()
    }
}

impl ExtensionProvider for Context {
    fn get_block_extension(
        &self,
        hash: &ckb_types::packed::Byte32,
    ) -> Option<ckb_types::packed::Bytes> {
        self.block_extensions.get(hash).map(|b| b.pack())
    }
}