feat: use kzg-rs for kzg point evaluation

crates/interpreter/Cargo.toml

@@ -81,3 +81,5 @@ optional_eip3607 = ["revm-primitives/optional_eip3607"]
 optional_gas_refund = ["revm-primitives/optional_gas_refund"]
 optional_no_base_fee = ["revm-primitives/optional_no_base_fee"]
 optional_beneficiary_reward = ["revm-primitives/optional_beneficiary_reward"]
+
+kzg-rs = ["revm-primitives/kzg-rs"]

crates/precompile/Cargo.toml

@@ -47,6 +47,11 @@ bn = { package = "substrate-bn", version = "0.6", default-features = false }
 # KZG point evaluation precompile
 c-kzg = { version = "1.0.2", default-features = false, optional = true }
 
+# Optionally use `kzg-rs` for a pure Rust implementation of KZG point evaluation.
+kzg-rs = { version = "0.1", default-features = false, features = [
+    'cache',
+], optional = true }
+
 # BLS12-381 precompiles
 blst = { version = "0.3.12", optional = true }
 
@@ -98,6 +103,8 @@ secp256r1 = ["dep:p256"]
 
 # Enables the KZG point evaluation precompile.
 c-kzg = ["dep:c-kzg", "revm-primitives/c-kzg"]
+kzg-rs = ["dep:kzg-rs", "revm-primitives/kzg-rs"]
+
 portable = ["revm-primitives/portable", "c-kzg?/portable"]
 
 # Use `secp256k1` as a faster alternative to `k256`.

crates/precompile/src/kzg_point_evaluation.rs

@@ -1,5 +1,11 @@
 use crate::{Address, Error, Precompile, PrecompileResult, PrecompileWithAddress};
-use c_kzg::{Bytes32, Bytes48, KzgProof, KzgSettings};
+cfg_if::cfg_if! {
+    if #[cfg(feature = "c-kzg")] {
+        use c_kzg::{Bytes32, Bytes48, KzgProof, KzgSettings};
+    } else if #[cfg(feature = "kzg-rs")] {
+        use kzg_rs::{Bytes32, Bytes48, KzgProof, KzgSettings};
+    }
+}
 use revm_primitives::{hex_literal::hex, Bytes, Env, PrecompileOutput};
 use sha2::{Digest, Sha256};
 

crates/precompile/src/lib.rs

@@ -15,7 +15,7 @@ pub mod bn128;
 pub mod fatal_precompile;
 pub mod hash;
 pub mod identity;
-#[cfg(feature = "c-kzg")]
+#[cfg(any(feature = "c-kzg", feature = "kzg-rs"))]
 pub mod kzg_point_evaluation;
 pub mod modexp;
 pub mod secp256k1;
@@ -25,6 +25,9 @@ pub mod utilities;
 
 pub use fatal_precompile::fatal_precompile;
 
+#[cfg(all(feature = "c-kzg", feature = "kzg-rs"))]
+// silence kzg-rs lint as c-kzg will be used as default if both are enabled.
+use kzg_rs as _;
 pub use primitives::{
     precompile::{PrecompileError as Error, *},
     Address, Bytes, HashMap, HashSet, Log, B256,
@@ -142,7 +145,7 @@ impl Precompiles {
 
             // EIP-4844: Shard Blob Transactions
             cfg_if! {
-                if #[cfg(feature = "c-kzg")] {
+                if #[cfg(any(feature = "c-kzg", feature = "kzg-rs"))] {
                     let precompile = kzg_point_evaluation::POINT_EVALUATION.clone();
                 } else {
                     // TODO move constants to separate file.

crates/primitives/Cargo.toml

@@ -22,7 +22,7 @@ rust_2018_idioms = "deny"
 all = "warn"
 
 [dependencies]
-alloy-eips = { version = "0.1", default-features = false, features = ["k256"]}
+alloy-eips = { version = "0.1", default-features = false, features = ["k256"] }
 alloy-primitives = { version = "0.7.2", default-features = false, features = [
     "rlp",
 ] }
@@ -35,6 +35,11 @@ bitflags = { version = "2.5.0", default-features = false }
 c-kzg = { version = "1.0.2", default-features = false, optional = true }
 once_cell = { version = "1.19", default-features = false, optional = true }
 
+# Optionally use `kzg-rs` for a pure Rust implementation of KZG.
+kzg-rs = { version = "0.1", default-features = false, features = [
+    'cache',
+], optional = true }
+
 # utility
 enumn = "0.1"
 derive_more = { version = "0.99", optional = true }
@@ -70,12 +75,13 @@ serde = [
     "bitvec/serde",
     "bitflags/serde",
     "c-kzg?/serde",
+    "kzg-rs?/serde",
 ]
 arbitrary = [
     "std",
     "alloy-eips/arbitrary",
     "alloy-primitives/arbitrary",
-    "bitflags/arbitrary"
+    "bitflags/arbitrary",
 ]
 asm-keccak = ["alloy-primitives/asm-keccak"]
 portable = ["c-kzg?/portable"]
@@ -105,3 +111,4 @@ rand = ["alloy-primitives/rand"]
 
 # See comments in `revm-precompile`
 c-kzg = ["dep:c-kzg", "dep:once_cell", "dep:derive_more"]
+kzg-rs = ["dep:kzg-rs", "dep:once_cell", "dep:derive_more"]

crates/primitives/src/env.rs

@@ -275,7 +275,7 @@ pub struct CfgEnv {
     /// Chain ID is introduced EIP-155
     pub chain_id: u64,
     /// KZG Settings for point evaluation precompile. By default, this is loaded from the ethereum mainnet trusted setup.
-    #[cfg(feature = "c-kzg")]
+    #[cfg(any(feature = "c-kzg", feature = "kzg-rs"))]
     #[cfg_attr(feature = "serde", serde(skip))]
     pub kzg_settings: crate::kzg::EnvKzgSettings,
     /// Bytecode that is created with CREATE/CREATE2 is by default analysed and jumptable is created.

crates/primitives/src/kzg.rs

@@ -1,7 +1,14 @@
 mod env_settings;
 mod trusted_setup_points;
 
-pub use c_kzg::KzgSettings;
+cfg_if::cfg_if! {
+    if #[cfg(feature = "c-kzg")] {
+        pub use c_kzg::KzgSettings;
+    } else if #[cfg(feature = "kzg-rs")] {
+        pub use kzg_rs::KzgSettings;
+    }
+}
+
 pub use env_settings::EnvKzgSettings;
 pub use trusted_setup_points::{
     parse_kzg_trusted_setup, G1Points, G2Points, KzgErrors, BYTES_PER_G1_POINT, BYTES_PER_G2_POINT,

crates/primitives/src/kzg/env_settings.rs

@@ -5,33 +5,39 @@ use super::{
 use once_cell::race::OnceBox;
 use std::{boxed::Box, sync::Arc};
 
-/// KZG Settings that allow us to specify a custom trusted setup.
-/// or use hardcoded default settings.
-#[derive(Debug, Clone, Default, PartialEq, Eq, Hash)]
-pub enum EnvKzgSettings {
-    /// Default mainnet trusted setup
-    #[default]
-    Default,
-    /// Custom trusted setup.
-    Custom(Arc<c_kzg::KzgSettings>),
-}
+cfg_if::cfg_if! {
+    if #[cfg(feature = "c-kzg")] {
+        /// KZG Settings that allow us to specify a custom trusted setup.
+        /// or use hardcoded default settings.
+        #[derive(Debug, Clone, Default, PartialEq, Eq )]
+        pub enum EnvKzgSettings {
+            /// Default mainnet trusted setup
+            #[default]
+            Default,
+            /// Custom trusted setup.
+            Custom(Arc<c_kzg::KzgSettings>),
+        }
 
-impl EnvKzgSettings {
-    /// Return set KZG settings.
-    ///
-    /// In will initialize the default settings if it is not already loaded.
-    pub fn get(&self) -> &KzgSettings {
-        match self {
-            Self::Default => {
-                static DEFAULT: OnceBox<KzgSettings> = OnceBox::new();
-                DEFAULT.get_or_init(|| {
-                    let settings =
-                        KzgSettings::load_trusted_setup(G1_POINTS.as_ref(), G2_POINTS.as_ref())
-                            .expect("failed to load default trusted setup");
-                    Box::new(settings)
-                })
+        impl EnvKzgSettings {
+            /// Return set KZG settings.
+            ///
+            /// In will initialize the default settings if it is not already loaded.
+            pub fn get(&self) -> &KzgSettings {
+                match self {
+                    Self::Default => {
+                        static DEFAULT: OnceBox<KzgSettings> = OnceBox::new();
+                        DEFAULT.get_or_init(|| {
+                            let settings =
+                                KzgSettings::load_trusted_setup(G1_POINTS.as_ref(), G2_POINTS.as_ref())
+                                    .expect("failed to load default trusted setup");
+                            Box::new(settings)
+                        })
+                    }
+                    Self::Custom(settings) => settings,
+                }
             }
-            Self::Custom(settings) => settings,
         }
+    } else if #[cfg(feature = "kzg-rs")] {
+        pub use kzg_rs::EnvKzgSettings;
     }
 }

crates/primitives/src/kzg/trusted_setup_points.rs

@@ -2,7 +2,13 @@ use core::fmt;
 use derive_more::{AsMut, AsRef, Deref, DerefMut};
 use std::boxed::Box;
 
-pub use c_kzg::{BYTES_PER_G1_POINT, BYTES_PER_G2_POINT};
+cfg_if::cfg_if! {
+    if #[cfg(feature = "c-kzg")] {
+        pub use c_kzg::{BYTES_PER_G1_POINT, BYTES_PER_G2_POINT};
+    } else if #[cfg(feature = "kzg-rs")] {
+        pub use kzg_rs::{BYTES_PER_G1_POINT, BYTES_PER_G2_POINT};
+    }
+}
 
 /// Number of G1 Points.
 pub const NUM_G1_POINTS: usize = 4096;

crates/primitives/src/lib.rs

@@ -12,7 +12,7 @@ mod constants;
 pub mod db;
 pub mod env;
 
-#[cfg(feature = "c-kzg")]
+#[cfg(any(feature = "c-kzg", feature = "kzg-rs"))]
 pub mod kzg;
 pub mod precompile;
 pub mod result;
@@ -38,10 +38,14 @@ cfg_if::cfg_if! {
     }
 }
 
-#[cfg(feature = "c-kzg")]
+#[cfg(any(feature = "c-kzg", feature = "kzg-rs"))]
 pub use kzg::{EnvKzgSettings, KzgSettings};
 pub use precompile::*;
 pub use result::*;
 pub use specification::*;
 pub use state::*;
 pub use utilities::*;
+
+#[cfg(all(feature = "c-kzg", feature = "kzg-rs"))]
+// silence kzg-rs lint as c-kzg will be used as default if both are enabled 
+use kzg_rs as _;

crates/revm/Cargo.toml

@@ -126,6 +126,7 @@ optional_beneficiary_reward = ["revm-interpreter/optional_beneficiary_reward"]
 # See comments in `revm-precompile`
 secp256k1 = ["revm-precompile/secp256k1"]
 c-kzg = ["revm-precompile/c-kzg"]
+kzg-rs = ["revm-precompile/kzg-rs"]
 blst = ["revm-precompile/blst"]
 
 [[example]]