From 9622e9a4ed8bee47333aeb5112321a30857a0018 Mon Sep 17 00:00:00 2001 From: Arni Hod Date: Sun, 7 Jul 2024 12:15:38 +0300 Subject: [PATCH] chore: create struct for gateway compiler --- crates/gateway/src/compilation.rs | 113 ++++++++++-------- crates/gateway/src/compilation_config.rs | 15 +++ crates/gateway/src/compilation_test.rs | 9 +- crates/gateway/src/config.rs | 3 + crates/gateway/src/gateway.rs | 7 +- crates/gateway/src/gateway_test.rs | 8 +- crates/gateway/src/lib.rs | 1 + .../stateful_transaction_validator_test.rs | 7 +- .../src/integration_test_utils.rs | 8 +- 9 files changed, 109 insertions(+), 62 deletions(-) create mode 100644 crates/gateway/src/compilation_config.rs diff --git a/crates/gateway/src/compilation.rs b/crates/gateway/src/compilation.rs index 5381ce44..58ec1869 100644 --- a/crates/gateway/src/compilation.rs +++ b/crates/gateway/src/compilation.rs @@ -11,6 +11,7 @@ use starknet_sierra_compile::compile::compile_sierra_to_casm; use starknet_sierra_compile::errors::CompilationUtilError; use starknet_sierra_compile::utils::into_contract_class_for_compilation; +use crate::compilation_config::GatewayCompilerConfig; use crate::errors::{GatewayError, GatewayResult}; use crate::utils::is_subsequence; @@ -18,44 +19,72 @@ use crate::utils::is_subsequence; #[path = "compilation_test.rs"] mod compilation_test; -/// Formats the contract class for compilation, compiles it, and returns the compiled contract class -/// wrapped in a [`ClassInfo`]. -/// Assumes the contract class is of a Sierra program which is compiled to Casm. -pub fn compile_contract_class(declare_tx: &RPCDeclareTransaction) -> GatewayResult { - let RPCDeclareTransaction::V3(tx) = declare_tx; - let starknet_api_contract_class = &tx.contract_class; - let cairo_lang_contract_class = - into_contract_class_for_compilation(starknet_api_contract_class); +pub struct GatewayCompiler { + pub config: GatewayCompilerConfig, +} + +impl GatewayCompiler { + /// Formats the contract class for compilation, compiles it, and returns the compiled contract + /// class wrapped in a [`ClassInfo`]. + /// Assumes the contract class is of a Sierra program which is compiled to Casm. + pub fn compile_contract_class( + &self, + declare_tx: &RPCDeclareTransaction, + ) -> GatewayResult { + let RPCDeclareTransaction::V3(tx) = declare_tx; + let starknet_api_contract_class = &tx.contract_class; + let cairo_lang_contract_class = + into_contract_class_for_compilation(starknet_api_contract_class); - // Compile Sierra to Casm. - let catch_unwind_result = - panic::catch_unwind(|| compile_sierra_to_casm(cairo_lang_contract_class)); - let casm_contract_class = match catch_unwind_result { - Ok(compilation_result) => compilation_result?, - Err(_) => { - // TODO(Arni): Log the panic. - return Err(GatewayError::CompilationError(CompilationUtilError::CompilationPanic)); + // Compile Sierra to Casm. + let catch_unwind_result = + panic::catch_unwind(|| compile_sierra_to_casm(cairo_lang_contract_class)); + let casm_contract_class = match catch_unwind_result { + Ok(compilation_result) => compilation_result?, + Err(_) => { + // TODO(Arni): Log the panic. + return Err(GatewayError::CompilationError(CompilationUtilError::CompilationPanic)); + } + }; + self.validate_casm_class(&casm_contract_class)?; + + let hash_result = CompiledClassHash(casm_contract_class.compiled_class_hash()); + if hash_result != tx.compiled_class_hash { + return Err(GatewayError::CompiledClassHashMismatch { + supplied: tx.compiled_class_hash, + hash_result, + }); } - }; - validate_casm_class(&casm_contract_class)?; - let hash_result = CompiledClassHash(casm_contract_class.compiled_class_hash()); - if hash_result != tx.compiled_class_hash { - return Err(GatewayError::CompiledClassHashMismatch { - supplied: tx.compiled_class_hash, - hash_result, - }); + // Convert Casm contract class to Starknet contract class directly. + let blockifier_contract_class = + ContractClass::V1(ContractClassV1::try_from(casm_contract_class)?); + let class_info = ClassInfo::new( + &blockifier_contract_class, + starknet_api_contract_class.sierra_program.len(), + starknet_api_contract_class.abi.len(), + )?; + Ok(class_info) } - // Convert Casm contract class to Starknet contract class directly. - let blockifier_contract_class = - ContractClass::V1(ContractClassV1::try_from(casm_contract_class)?); - let class_info = ClassInfo::new( - &blockifier_contract_class, - starknet_api_contract_class.sierra_program.len(), - starknet_api_contract_class.abi.len(), - )?; - Ok(class_info) + // TODO(Arni): Add test. + fn validate_casm_class(&self, contract_class: &CasmContractClass) -> Result<(), GatewayError> { + let CasmContractEntryPoints { external, l1_handler, constructor } = + &contract_class.entry_points_by_type; + let entry_points_iterator = + external.iter().chain(l1_handler.iter()).chain(constructor.iter()); + + for entry_point in entry_points_iterator { + let builtins = &entry_point.builtins; + if !is_subsequence(builtins, supported_builtins()) { + return Err(GatewayError::UnsupportedBuiltins { + builtins: builtins.clone(), + supported_builtins: supported_builtins().to_vec(), + }); + } + } + Ok(()) + } } // TODO(Arni): Add to a config. @@ -70,21 +99,3 @@ fn supported_builtins() -> &'static Vec { SUPPORTED_BUILTIN_NAMES.iter().map(|builtin| builtin.to_string()).collect::>() }) } - -// TODO(Arni): Add test. -fn validate_casm_class(contract_class: &CasmContractClass) -> Result<(), GatewayError> { - let CasmContractEntryPoints { external, l1_handler, constructor } = - &contract_class.entry_points_by_type; - let entry_points_iterator = external.iter().chain(l1_handler.iter()).chain(constructor.iter()); - - for entry_point in entry_points_iterator { - let builtins = &entry_point.builtins; - if !is_subsequence(builtins, supported_builtins()) { - return Err(GatewayError::UnsupportedBuiltins { - builtins: builtins.clone(), - supported_builtins: supported_builtins().to_vec(), - }); - } - } - Ok(()) -} diff --git a/crates/gateway/src/compilation_config.rs b/crates/gateway/src/compilation_config.rs new file mode 100644 index 00000000..155b81cd --- /dev/null +++ b/crates/gateway/src/compilation_config.rs @@ -0,0 +1,15 @@ +use std::collections::BTreeMap; + +use papyrus_config::dumping::SerializeConfig; +use papyrus_config::{ParamPath, SerializedParam}; +use serde::{Deserialize, Serialize}; +use validator::Validate; + +#[derive(Clone, Debug, Default, Serialize, Deserialize, Validate, PartialEq)] +pub struct GatewayCompilerConfig {} + +impl SerializeConfig for GatewayCompilerConfig { + fn dump(&self) -> BTreeMap { + BTreeMap::new() + } +} diff --git a/crates/gateway/src/compilation_test.rs b/crates/gateway/src/compilation_test.rs index ea0336de..7f5523fd 100644 --- a/crates/gateway/src/compilation_test.rs +++ b/crates/gateway/src/compilation_test.rs @@ -6,7 +6,7 @@ use starknet_api::rpc_transaction::{RPCDeclareTransaction, RPCTransaction}; use starknet_sierra_compile::errors::CompilationUtilError; use test_utils::starknet_api_test_utils::declare_tx; -use crate::compilation::compile_contract_class; +use crate::compilation::GatewayCompiler; use crate::errors::GatewayError; #[test] @@ -21,7 +21,7 @@ fn test_compile_contract_class_compiled_class_hash_missmatch() { tx.compiled_class_hash = supplied_hash; let declare_tx = RPCDeclareTransaction::V3(tx); - let result = compile_contract_class(&declare_tx); + let result = GatewayCompiler { config: Default::default() }.compile_contract_class(&declare_tx); assert_matches!( result.unwrap_err(), GatewayError::CompiledClassHashMismatch { supplied, hash_result } @@ -39,7 +39,7 @@ fn test_compile_contract_class_bad_sierra() { tx.contract_class.sierra_program = tx.contract_class.sierra_program[..100].to_vec(); let declare_tx = RPCDeclareTransaction::V3(tx); - let result = compile_contract_class(&declare_tx); + let result = GatewayCompiler { config: Default::default() }.compile_contract_class(&declare_tx); assert_matches!( result.unwrap_err(), GatewayError::CompilationError(CompilationUtilError::AllowedLibfuncsError( @@ -57,7 +57,8 @@ fn test_compile_contract_class() { let RPCDeclareTransaction::V3(declare_tx_v3) = &declare_tx; let contract_class = &declare_tx_v3.contract_class; - let class_info = compile_contract_class(&declare_tx).unwrap(); + let class_info = + GatewayCompiler { config: Default::default() }.compile_contract_class(&declare_tx).unwrap(); assert_matches!(class_info.contract_class(), ContractClass::V1(_)); assert_eq!(class_info.sierra_program_length(), contract_class.sierra_program.len()); assert_eq!(class_info.abi_length(), contract_class.abi.len()); diff --git a/crates/gateway/src/config.rs b/crates/gateway/src/config.rs index 9add4d86..215aacf0 100644 --- a/crates/gateway/src/config.rs +++ b/crates/gateway/src/config.rs @@ -8,6 +8,7 @@ use serde::{Deserialize, Serialize}; use starknet_api::core::{ChainId, ContractAddress, Nonce}; use validator::Validate; +use crate::compilation_config::GatewayCompilerConfig; use crate::compiler_version::VersionId; #[derive(Clone, Debug, Default, Serialize, Deserialize, Validate, PartialEq)] @@ -15,6 +16,7 @@ pub struct GatewayConfig { pub network_config: GatewayNetworkConfig, pub stateless_tx_validator_config: StatelessTransactionValidatorConfig, pub stateful_tx_validator_config: StatefulTransactionValidatorConfig, + pub gateway_compiler_config: GatewayCompilerConfig, } impl SerializeConfig for GatewayConfig { @@ -29,6 +31,7 @@ impl SerializeConfig for GatewayConfig { self.stateful_tx_validator_config.dump(), "stateful_tx_validator_config", ), + append_sub_config_name(self.gateway_compiler_config.dump(), "gateway_compiler_config"), ] .into_iter() .flatten() diff --git a/crates/gateway/src/gateway.rs b/crates/gateway/src/gateway.rs index d85cf007..f7638827 100644 --- a/crates/gateway/src/gateway.rs +++ b/crates/gateway/src/gateway.rs @@ -12,7 +12,7 @@ use starknet_mempool_infra::component_runner::{ComponentRunner, ComponentStartEr use starknet_mempool_types::communication::SharedMempoolClient; use starknet_mempool_types::mempool_types::{Account, MempoolInput}; -use crate::compilation::compile_contract_class; +use crate::compilation::GatewayCompiler; use crate::config::{GatewayConfig, GatewayNetworkConfig, RpcStateReaderConfig}; use crate::errors::{GatewayError, GatewayResult, GatewayRunError}; use crate::rpc_state_reader::RpcStateReaderFactory; @@ -119,7 +119,10 @@ fn process_tx( // Compile Sierra to Casm. let optional_class_info = match &tx { - RPCTransaction::Declare(declare_tx) => Some(compile_contract_class(declare_tx)?), + RPCTransaction::Declare(declare_tx) => { + let gateway_compiler = GatewayCompiler { config: Default::default() }; + Some(gateway_compiler.compile_contract_class(declare_tx)?) + } _ => None, }; diff --git a/crates/gateway/src/gateway_test.rs b/crates/gateway/src/gateway_test.rs index 25f15d95..1469abd0 100644 --- a/crates/gateway/src/gateway_test.rs +++ b/crates/gateway/src/gateway_test.rs @@ -18,7 +18,7 @@ use tokio::sync::mpsc::channel; use tokio::task; use crate::config::{StatefulTransactionValidatorConfig, StatelessTransactionValidatorConfig}; -use crate::gateway::{add_tx, compile_contract_class, AppState, SharedMempoolClient}; +use crate::gateway::{add_tx, AppState, GatewayCompiler, SharedMempoolClient}; use crate::state_reader_test_utils::{ local_test_state_reader_factory, local_test_state_reader_factory_for_deploy_account, TestStateReaderFactory, @@ -110,7 +110,11 @@ async fn to_bytes(res: Response) -> Bytes { fn calculate_hash(external_tx: &RPCTransaction) -> TransactionHash { let optional_class_info = match &external_tx { - RPCTransaction::Declare(declare_tx) => Some(compile_contract_class(declare_tx).unwrap()), + RPCTransaction::Declare(declare_tx) => Some( + GatewayCompiler { config: Default::default() } + .compile_contract_class(declare_tx) + .unwrap(), + ), _ => None, }; diff --git a/crates/gateway/src/lib.rs b/crates/gateway/src/lib.rs index 59f79bf8..b9f60e70 100644 --- a/crates/gateway/src/lib.rs +++ b/crates/gateway/src/lib.rs @@ -1,5 +1,6 @@ pub mod communication; pub mod compilation; +pub mod compilation_config; pub mod compiler_version; pub mod config; pub mod errors; diff --git a/crates/gateway/src/stateful_transaction_validator_test.rs b/crates/gateway/src/stateful_transaction_validator_test.rs index 6349f8a1..4d5688a3 100644 --- a/crates/gateway/src/stateful_transaction_validator_test.rs +++ b/crates/gateway/src/stateful_transaction_validator_test.rs @@ -12,7 +12,7 @@ use test_utils::starknet_api_test_utils::{ VALID_L1_GAS_MAX_PRICE_PER_UNIT, }; -use crate::compilation::compile_contract_class; +use crate::compilation::GatewayCompiler; use crate::config::StatefulTransactionValidatorConfig; use crate::errors::{StatefulTransactionValidatorError, StatefulTransactionValidatorResult}; use crate::state_reader_test_utils::{ @@ -81,7 +81,10 @@ fn test_stateful_tx_validator( }, }; let optional_class_info = match &external_tx { - RPCTransaction::Declare(declare_tx) => Some(compile_contract_class(declare_tx).unwrap()), + RPCTransaction::Declare(declare_tx) => { + let gateway_compiler = GatewayCompiler { config: Default::default() }; + Some(gateway_compiler.compile_contract_class(declare_tx).unwrap()) + } _ => None, }; diff --git a/crates/tests-integration/src/integration_test_utils.rs b/crates/tests-integration/src/integration_test_utils.rs index 2d6147fe..bf1b70a8 100644 --- a/crates/tests-integration/src/integration_test_utils.rs +++ b/crates/tests-integration/src/integration_test_utils.rs @@ -24,8 +24,14 @@ async fn create_gateway_config() -> GatewayConfig { let socket = get_available_socket().await; let network_config = GatewayNetworkConfig { ip: socket.ip(), port: socket.port() }; let stateful_tx_validator_config = StatefulTransactionValidatorConfig::create_for_testing(); + let gateway_compiler_config = Default::default(); - GatewayConfig { network_config, stateless_tx_validator_config, stateful_tx_validator_config } + GatewayConfig { + network_config, + stateless_tx_validator_config, + stateful_tx_validator_config, + gateway_compiler_config, + } } pub async fn create_config(rpc_server_addr: SocketAddr) -> MempoolNodeConfig {