use crate::syscalls::CURRENT_CYCLES;
use ckb_vm::{
registers::{A0, A7},
Error as VMError, Register, SupportMachine, Syscalls,
};
#[derive(Debug, Default)]
pub struct CurrentCycles {
base: u64,
}
impl CurrentCycles {
pub fn new(base: u64) -> Self {
Self { base }
}
}
impl<Mac: SupportMachine> Syscalls<Mac> for CurrentCycles {
fn initialize(&mut self, _machine: &mut Mac) -> Result<(), VMError> {
Ok(())
}
fn ecall(&mut self, machine: &mut Mac) -> Result<bool, VMError> {
if machine.registers()[A7].to_u64() != CURRENT_CYCLES {
return Ok(false);
}
let cycles = self
.base
.checked_add(machine.cycles())
.ok_or(VMError::CyclesOverflow)?;
machine.set_register(A0, Mac::REG::from_u64(cycles));
Ok(true)
}
}