use super::calc_linear_cost_u32;
use crate::{
crypto, Precompile, PrecompileError, PrecompileId, PrecompileOutput, PrecompileResult,
};
pub const SHA256: Precompile =
Precompile::new(PrecompileId::Sha256, crate::u64_to_address(2), sha256_run);
pub const RIPEMD160: Precompile = Precompile::new(
PrecompileId::Ripemd160,
crate::u64_to_address(3),
ripemd160_run,
);
pub fn sha256_run(input: &[u8], gas_limit: u64) -> PrecompileResult {
let cost = calc_linear_cost_u32(input.len(), 60, 12);
if cost > gas_limit {
Err(PrecompileError::OutOfGas)
} else {
let output = crypto().sha256(input);
Ok(PrecompileOutput::new(cost, output.to_vec().into()))
}
}
pub fn ripemd160_run(input: &[u8], gas_limit: u64) -> PrecompileResult {
let gas_used = calc_linear_cost_u32(input.len(), 600, 120);
if gas_used > gas_limit {
Err(PrecompileError::OutOfGas)
} else {
let output = crypto().ripemd160(input);
Ok(PrecompileOutput::new(gas_used, output.to_vec().into()))
}
}