use super::utils::{pad_g1_point, remove_g1_padding};
use crate::{
bls12_381_const::{G1_ADD_ADDRESS, G1_ADD_BASE_GAS_FEE, G1_ADD_INPUT_LENGTH, PADDED_G1_LENGTH},
crypto, eth_precompile_fn, EthPrecompileOutput, EthPrecompileResult, Precompile,
PrecompileHalt, PrecompileId,
};
eth_precompile_fn!(g1_add_precompile, g1_add);
pub const PRECOMPILE: Precompile =
Precompile::new(PrecompileId::Bls12G1Add, G1_ADD_ADDRESS, g1_add_precompile);
pub fn g1_add(input: &[u8], gas_limit: u64) -> EthPrecompileResult {
if G1_ADD_BASE_GAS_FEE > gas_limit {
return Err(PrecompileHalt::OutOfGas);
}
if input.len() != G1_ADD_INPUT_LENGTH {
return Err(PrecompileHalt::Bls12381G1AddInputLength);
}
let [a_x, a_y] = remove_g1_padding(&input[..PADDED_G1_LENGTH])?;
let [b_x, b_y] = remove_g1_padding(&input[PADDED_G1_LENGTH..])?;
let a = (*a_x, *a_y);
let b = (*b_x, *b_y);
let unpadded_result = crypto().bls12_381_g1_add(a, b)?;
let padded_result = pad_g1_point(&unpadded_result);
Ok(EthPrecompileOutput::new(
G1_ADD_BASE_GAS_FEE,
padded_result.into(),
))
}