Add new constructor declare v2

src/lib.rs

@@ -198,6 +198,7 @@ mod test {
     use std::path::PathBuf;
 
     use crate::core::contract_address::{compute_deprecated_class_hash, compute_sierra_class_hash};
+    use crate::definitions::constants::INITIAL_GAS_COST;
     use crate::definitions::{
         block_context::StarknetChainId,
         constants::{
@@ -836,7 +837,7 @@ mod test {
             tx_type: TransactionType::Declare,
             validate_entry_point_selector: VALIDATE_DECLARE_ENTRY_POINT_SELECTOR.clone(),
             version: 1.into(),
-            max_fee: 2,
+            max_fee: INITIAL_GAS_COST,
             signature: vec![],
             nonce: 0.into(),
             hash_value: 0.into(),
@@ -998,4 +999,18 @@ mod test {
             [(0, 1224), (0, 0)]
         );
     }
+
+    #[test]
+    fn test_declare_v2_with_invalid_compiled_class_hash() {
+        let (block_context, mut state) = create_account_tx_test_state().unwrap();
+        let mut declare_v2 = declarev2_tx();
+        declare_v2.compiled_class_hash = Felt252::from(1);
+        let declare_tx = Transaction::DeclareV2(Box::new(declare_v2));
+
+        let err = declare_tx
+            .execute(&mut state, &block_context, INITIAL_GAS_COST)
+            .unwrap_err();
+
+        assert_eq!(err.to_string(), "Invalid compiled class, expected class hash: \"1948962768849191111780391610229754715773924969841143100991524171924131413970\", but received: \"1\"".to_string());
+    }
 }

src/testing/mod.rs

@@ -41,7 +41,7 @@ lazy_static! {
     pub static ref TEST_CLASS_HASH: Felt252 = felt_str!("272");
     pub static ref TEST_EMPTY_CONTRACT_CLASS_HASH: Felt252 = felt_str!("274");
     pub static ref TEST_ERC20_CONTRACT_CLASS_HASH: Felt252 = felt_str!("4112");
-    pub static ref TEST_FIB_COMPILED_CONTRACT_CLASS_HASH: Felt252 = felt_str!("27727");
+    pub static ref TEST_FIB_COMPILED_CONTRACT_CLASS_HASH: Felt252 = felt_str!("1948962768849191111780391610229754715773924969841143100991524171924131413970");
 
     // Storage keys.
     pub static ref TEST_ERC20_ACCOUNT_BALANCE_KEY: Felt252 =

src/transaction/declare_v2.rs

@@ -1,5 +1,5 @@
 use super::{verify_version, Transaction};
-use crate::core::contract_address::compute_sierra_class_hash;
+use crate::core::contract_address::{compute_casm_class_hash, compute_sierra_class_hash};
 use crate::definitions::constants::QUERY_VERSION_BASE;
 use crate::services::api::contract_classes::deprecated_contract_class::EntryPointType;
 
@@ -45,7 +45,7 @@ pub struct DeclareV2 {
     pub sierra_contract_class: SierraContractClass,
     pub sierra_class_hash: Felt252,
     pub hash_value: Felt252,
-    pub casm_class: once_cell::unsync::OnceCell<CasmContractClass>,
+    pub casm_class: Option<CasmContractClass>,
     pub skip_validate: bool,
     pub skip_execute: bool,
     pub skip_fee_transfer: bool,
@@ -56,6 +56,7 @@ impl DeclareV2 {
     /// It will calculate the sierra class hash and the transaction hash.
     /// ## Parameters:
     /// - sierra_contract_class: The sierra contract class of the contract to declare
+    /// - casm_contract_class: The casm contract class of the contract to declare. This is optional.
     /// - compiled_class_hash: the class hash of the contract compiled with Cairo1 or newer.
     /// - chain_id: Id of the network where is going to be declare, those can be: Mainnet, Testnet.
     /// - sender_address: The address of the account declaring the contract.
@@ -66,6 +67,7 @@ impl DeclareV2 {
     #[allow(clippy::too_many_arguments)]
     pub fn new(
         sierra_contract_class: &SierraContractClass,
+        casm_contract_class: Option<CasmContractClass>,
         compiled_class_hash: Felt252,
         chain_id: Felt252,
         sender_address: Address,
@@ -89,6 +91,7 @@ impl DeclareV2 {
         Self::new_with_sierra_class_hash_and_tx_hash(
             sierra_contract_class,
             sierra_class_hash,
+            casm_contract_class,
             compiled_class_hash,
             sender_address,
             max_fee,
@@ -103,6 +106,7 @@ impl DeclareV2 {
     /// ## Parameters:
     /// - sierra_contract_class: The sierra contract class of the contract to declare
     /// - sierra_class_hash: The precomputed hash for the sierra contract
+    /// - casm_contract_class: The casm contract class of the contract to declare. This is optional.
     /// - compiled_class_hash: the class hash of the contract compiled with Cairo1 or newer.
     /// - sender_address: The address of the account declaring the contract.
     /// - max_fee: refers to max amount of fee that a declare takes.
@@ -116,6 +120,7 @@ impl DeclareV2 {
     pub fn new_with_sierra_class_hash_and_tx_hash(
         sierra_contract_class: &SierraContractClass,
         sierra_class_hash: Felt252,
+        casm_contract_class: Option<CasmContractClass>,
         compiled_class_hash: Felt252,
         sender_address: Address,
         max_fee: u128,
@@ -138,7 +143,7 @@ impl DeclareV2 {
             nonce,
             compiled_class_hash,
             hash_value,
-            casm_class: Default::default(),
+            casm_class: casm_contract_class,
             skip_execute: false,
             skip_validate: false,
             skip_fee_transfer: false,
@@ -156,7 +161,8 @@ impl DeclareV2 {
 
     // creates a new instance of a declare but without the computation of the transaction hash.
     /// ## Parameters:
-    /// - sierra_contract_class: The sierra contract class of the contract to declare
+    /// - sierra_contract_class: The sierra contract class of the contract to declare.
+    /// - casm_contract_class: The casm contract class of the contract to declare. This is optional.
     /// - compiled_class_hash: the class hash of the contract compiled with Cairo1 or newer.
     /// - sender_address: The address of the account declaring the contract.
     /// - max_fee: refers to max amount of fee that a declare takes.
@@ -167,6 +173,7 @@ impl DeclareV2 {
     #[allow(clippy::too_many_arguments)]
     pub fn new_with_tx_hash(
         sierra_contract_class: &SierraContractClass,
+        casm_contract_class: Option<CasmContractClass>,
         compiled_class_hash: Felt252,
         sender_address: Address,
         max_fee: u128,
@@ -180,6 +187,7 @@ impl DeclareV2 {
         Self::new_with_sierra_class_hash_and_tx_hash(
             sierra_contract_class,
             sierra_class_hash,
+            casm_contract_class,
             compiled_class_hash,
             sender_address,
             max_fee,
@@ -194,6 +202,7 @@ impl DeclareV2 {
     /// ## Parameters:
     /// - sierra_contract_class: The sierra contract class of the contract to declare
     /// - sierra_class_hash: The precomputed hash for the sierra contract
+    /// - casm_contract_class: The casm contract class of the contract to declare. This is optional.
     /// - compiled_class_hash: the class hash of the contract compiled with Cairo1 or newer.
     /// - chain_id: Id of the network where is going to be declare, those can be: Mainnet, Testnet.
     /// - sender_address: The address of the account declaring the contract.
@@ -205,6 +214,7 @@ impl DeclareV2 {
     pub fn new_with_sierra_class_hash(
         sierra_contract_class: &SierraContractClass,
         sierra_class_hash: Felt252,
+        casm_contract_class: Option<CasmContractClass>,
         compiled_class_hash: Felt252,
         chain_id: Felt252,
         sender_address: Address,
@@ -226,6 +236,7 @@ impl DeclareV2 {
         Self::new_with_sierra_class_hash_and_tx_hash(
             sierra_contract_class,
             sierra_class_hash,
+            casm_contract_class,
             compiled_class_hash,
             sender_address,
             max_fee,
@@ -374,15 +385,23 @@ impl DeclareV2 {
         &self,
         state: &mut S,
     ) -> Result<(), TransactionError> {
-        let casm_class = self
-            .casm_class
-            .get_or_try_init(|| {
+        let casm_class = match &self.casm_class {
+            None => {
                 CasmContractClass::from_contract_class(self.sierra_contract_class.clone(), true)
-            })
-            .map_err(|e| TransactionError::SierraCompileError(e.to_string()))?;
+                    .map_err(|e| TransactionError::SierraCompileError(e.to_string()))?
+            }
+            Some(casm_contract_class) => casm_contract_class.clone(),
+        };
 
+        let casm_class_hash = compute_casm_class_hash(&casm_class)?;
+        if casm_class_hash != self.compiled_class_hash {
+            return Err(TransactionError::InvalidCompiledClassHash(
+                casm_class_hash.to_string(),
+                self.compiled_class_hash.to_string(),
+            ));
+        }
         state.set_compiled_class_hash(&self.sierra_class_hash, &self.compiled_class_hash)?;
-        state.set_compiled_class(&self.compiled_class_hash, casm_class.clone())?;
+        state.set_compiled_class(&self.compiled_class_hash, casm_class)?;
 
         Ok(())
     }
@@ -453,7 +472,7 @@ mod tests {
     use std::{collections::HashMap, fs::File, io::BufReader, path::PathBuf};
 
     use super::DeclareV2;
-    use crate::core::contract_address::compute_sierra_class_hash;
+    use crate::core::contract_address::{compute_casm_class_hash, compute_sierra_class_hash};
     use crate::definitions::constants::QUERY_VERSION_BASE;
     use crate::services::api::contract_classes::compiled_class::CompiledClass;
     use crate::state::state_api::StateReader;
@@ -466,7 +485,7 @@ mod tests {
     use num_traits::{One, Zero};
 
     #[test]
-    fn create_declare_v2_test() {
+    fn create_declare_v2_without_casm_contract_class_test() {
         // read file to create sierra contract class
         let version;
         let path;
@@ -487,12 +506,85 @@ mod tests {
         let sierra_contract_class: cairo_lang_starknet::contract_class::ContractClass =
             serde_json::from_reader(reader).unwrap();
         let sender_address = Address(1.into());
+        let casm_class =
+            CasmContractClass::from_contract_class(sierra_contract_class.clone(), true).unwrap();
+        let casm_class_hash = compute_casm_class_hash(&casm_class).unwrap();
 
         // create internal declare v2
 
         let internal_declare = DeclareV2::new_with_tx_hash(
             &sierra_contract_class,
+            None,
+            casm_class_hash,
+            sender_address,
+            0,
+            version,
+            [1.into()].to_vec(),
+            Felt252::zero(),
             Felt252::one(),
+        )
+        .unwrap();
+
+        // crate state to store casm contract class
+        let casm_contract_class_cache = HashMap::new();
+        let state_reader = InMemoryStateReader::default();
+        let mut state = CachedState::new(state_reader, None, Some(casm_contract_class_cache));
+
+        // call compile and store
+        assert!(internal_declare
+            .compile_and_store_casm_class(&mut state)
+            .is_ok());
+
+        // test we  can retreive the data
+        let expected_casm_class = CasmContractClass::from_contract_class(
+            internal_declare.sierra_contract_class.clone(),
+            true,
+        )
+        .unwrap();
+
+        let casm_class = match state
+            .get_contract_class(&internal_declare.compiled_class_hash.to_be_bytes())
+            .unwrap()
+        {
+            CompiledClass::Casm(casm) => *casm,
+            _ => unreachable!(),
+        };
+
+        assert_eq!(expected_casm_class, casm_class);
+    }
+
+    #[test]
+    fn create_declare_v2_with_casm_contract_class_test() {
+        // read file to create sierra contract class
+        let version;
+        let path;
+        #[cfg(not(feature = "cairo_1_tests"))]
+        {
+            version = Felt252::from(2);
+            path = PathBuf::from("starknet_programs/cairo2/fibonacci.sierra");
+        }
+
+        #[cfg(feature = "cairo_1_tests")]
+        {
+            version = Felt252::from(1);
+            path = PathBuf::from("starknet_programs/cairo1/fibonacci.sierra");
+        }
+
+        let file = File::open(path).unwrap();
+        let reader = BufReader::new(file);
+        let sierra_contract_class: cairo_lang_starknet::contract_class::ContractClass =
+            serde_json::from_reader(reader).unwrap();
+        let sender_address = Address(1.into());
+        let casm_class =
+            CasmContractClass::from_contract_class(sierra_contract_class.clone(), true).unwrap();
+        let casm_class_hash = compute_casm_class_hash(&casm_class).unwrap();
+
+        // create internal declare v2
+
+        let internal_declare = DeclareV2::new_with_tx_hash(
+            &sierra_contract_class,
+            Some(casm_class),
+            casm_class_hash,
             sender_address,
             0,
             version,
@@ -553,12 +645,17 @@ mod tests {
             serde_json::from_reader(reader).unwrap();
         let sierra_class_hash = compute_sierra_class_hash(&sierra_contract_class).unwrap();
         let sender_address = Address(1.into());
+        let casm_class =
+            CasmContractClass::from_contract_class(sierra_contract_class.clone(), true).unwrap();
+        let casm_class_hash = compute_casm_class_hash(&casm_class).unwrap();
 
         // create internal declare v2
 
-        let internal_declare = DeclareV2::new_with_tx_hash(
+        let internal_declare = DeclareV2::new_with_sierra_class_hash_and_tx_hash(
             &sierra_contract_class,
             sierra_class_hash,
+            Some(casm_class),
+            casm_class_hash,
             sender_address,
             0,
             version,

src/transaction/error.rs

@@ -135,4 +135,6 @@ pub enum TransactionError {
     CallInfoIsNone,
     #[error("Unsupported version {0:?}")]
     UnsupportedVersion(String),
+    #[error("Invalid compiled class, expected class hash: {0:?}, but received: {1:?}")]
+    InvalidCompiledClassHash(String, String),
 }

starknet_programs/cairo1/fibonacci_dispatcher.cairo

@@ -42,7 +42,8 @@ mod Dispatcher {
         class_hash: felt252, selector: felt252, a: felt252, b: felt252, n: felt252
     ) -> felt252 {
         FibonacciLibraryDispatcher {
-            class_hash: starknet::class_hash_const::<27727>(),
+            // THIS VALUE IS THE HASH OF THE FIBONACCI CASM CLASS HASH. THE SAME AS THE CONSTANT: TEST_FIB_COMPILED_CONTRACT_CLASS_HASH
+            class_hash: starknet::class_hash_const::<1948962768849191111780391610229754715773924969841143100991524171924131413970>(),
             selector
         }.fib(a, b, n)
     }

starknet_programs/cairo2/fibonacci_dispatcher.cairo

@@ -51,7 +51,8 @@ mod Dispatcher {
             self: @ContractState, class_hash: felt252, selector: felt252, a: felt252, b: felt252, n: felt252
         ) -> felt252 {
             FibonacciLibraryDispatcher {
-                class_hash: starknet::class_hash_const::<27727>(),
+                // THIS VALUE IS THE HASH OF THE FIBONACCI CASM CLASS HASH.
+                class_hash: starknet::class_hash_const::<2889767417435368609058888822622483550637539736178264636938129582300971548553>(),
                 selector
             }.fib(a, b, n)
         }