Support higher tx versions in Account

.github/workflows/test.yml

@@ -16,7 +16,7 @@ jobs:
     - uses: actions/checkout@v3
     - uses: software-mansion/setup-scarb@v1
       with:
-        scarb-version: "2.3.1"
+        scarb-version: "2.4.1"
     - name: Markdown lint
       uses: DavidAnson/markdownlint-cli2-action@5b7c9f74fec47e6b15667b2cc23c63dff11e449e # v9
       with:

CHANGELOG.md

@@ -18,3 +18,4 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 - Use ComponentState in tests (#836)
 - Docsite navbar (#838)
+- Support higher tx versions in Account (#858)

Scarb.toml

@@ -1,7 +1,7 @@
 [package]
 name = "openzeppelin"
 version = "0.8.0"
-cairo-version = "2.3.1"
+cairo-version = "2.4.1"
 authors = ["OpenZeppelin Community <maintainers@openzeppelin.org>"]
 description = "OpenZeppelin Contracts written in Cairo for StarkNet, a decentralized ZK Rollup"
 documentation = "https://docs.openzeppelin.com/contracts-cairo"
@@ -11,7 +11,7 @@ license-file = "LICENSE"
 keywords = ["openzeppelin", "starknet", "cairo", "contracts", "security", "standards"]
 
 [dependencies]
-starknet = "2.3.1"
+starknet = "2.4.1"
 
 [lib]
 

src/account.cairo

@@ -1,6 +1,7 @@
 mod account;
 mod dual_account;
 mod interface;
+mod utils;
 
 use account::AccountComponent;
 use interface::AccountABIDispatcher;

src/account/account.cairo

@@ -1,5 +1,5 @@
 // SPDX-License-Identifier: MIT
-// OpenZeppelin Contracts for Cairo v0.8.0 (account/account.cairo)
+// OpenZeppelin Contracts for Cairo vX.Y.Z (account/account.cairo)
 
 /// # Account Component
 ///
@@ -8,17 +8,14 @@
 mod AccountComponent {
     use ecdsa::check_ecdsa_signature;
     use openzeppelin::account::interface;
+    use openzeppelin::account::utils::TRANSACTION_VERSION;
     use openzeppelin::introspection::src5::SRC5Component::InternalTrait as SRC5InternalTrait;
     use openzeppelin::introspection::src5::SRC5Component;
     use starknet::account::Call;
     use starknet::get_caller_address;
     use starknet::get_contract_address;
     use starknet::get_tx_info;
 
-    const TRANSACTION_VERSION: felt252 = 1;
-    // 2**128 + TRANSACTION_VERSION
-    const QUERY_VERSION: felt252 = 0x100000000000000000000000000000001;
-
     #[storage]
     struct Storage {
         Account_public_key: felt252
@@ -59,8 +56,7 @@ mod AccountComponent {
         ///
         /// Requirements:
         ///
-        /// - The transaction version must be `TRANSACTION_VERSION` for actual transactions.
-        /// For simulations, the version must be `QUERY_VERSION`.
+        /// - The transaction version must be greater than or equal to `TRANSACTION_VERSION`.
         fn __execute__(
             self: @ComponentState<TContractState>, mut calls: Array<Call>
         ) -> Array<Span<felt252>> {
@@ -71,10 +67,8 @@ mod AccountComponent {
 
             // Check tx version
             let tx_info = get_tx_info().unbox();
-            let version = tx_info.version;
-            if version != TRANSACTION_VERSION {
-                assert(version == QUERY_VERSION, Errors::INVALID_TX_VERSION);
-            }
+            let tx_version: u256 = tx_info.version.into();
+            assert(TRANSACTION_VERSION <= tx_version, Errors::INVALID_TX_VERSION);
 
             _execute_calls(calls)
         }

src/account/utils.cairo

@@ -0,0 +1,6 @@
+// SPDX-License-Identifier: MIT
+// OpenZeppelin Contracts for Cairo vX.Y.Z (account/utils.cairo)
+
+const TRANSACTION_VERSION: u256 = 1;
+// 2**128 + TRANSACTION_VERSION
+const QUERY_VERSION: u256 = 0x100000000000000000000000000000001;

src/tests/account/test_account.cairo

@@ -1,14 +1,15 @@
 use openzeppelin::account::AccountComponent::{InternalTrait, SRC6CamelOnlyImpl};
 use openzeppelin::account::AccountComponent::{OwnerAdded, OwnerRemoved};
 use openzeppelin::account::AccountComponent::{PublicKeyCamelImpl, PublicKeyImpl};
-use openzeppelin::account::AccountComponent::{TRANSACTION_VERSION, QUERY_VERSION};
 use openzeppelin::account::AccountComponent;
 use openzeppelin::account::interface::{ISRC6, ISRC6_ID};
 use openzeppelin::account::{AccountABIDispatcherTrait, AccountABIDispatcher};
 use openzeppelin::introspection::interface::{ISRC5, ISRC5_ID};
 use openzeppelin::tests::mocks::account_mocks::DualCaseAccountMock;
 use openzeppelin::tests::mocks::erc20_mocks::DualCaseERC20Mock;
-use openzeppelin::tests::utils::constants::{PUBKEY, NEW_PUBKEY, SALT, ZERO};
+use openzeppelin::tests::utils::constants::{
+    PUBKEY, NEW_PUBKEY, SALT, ZERO, QUERY_VERSION, TRANSACTION_VERSION
+};
 use openzeppelin::tests::utils;
 use openzeppelin::token::erc20::interface::{IERC20DispatcherTrait, IERC20Dispatcher};
 use openzeppelin::utils::selectors;
@@ -71,7 +72,7 @@ fn setup() -> ComponentState {
 }
 
 fn setup_dispatcher(data: Option<@SignedTransactionData>) -> AccountABIDispatcher {
-    testing::set_version(TRANSACTION_VERSION);
+    testing::set_version(TRANSACTION_VERSION());
 
     let mut calldata = array![];
     if data.is_some() {
@@ -287,17 +288,22 @@ fn test_execute() {
     test_execute_with_version(Option::None(()));
 }
 
+#[test]
+fn test_execute_future_version() {
+    test_execute_with_version(Option::Some(TRANSACTION_VERSION() + 1));
+}
+
 #[test]
 #[available_gas(2000000)]
 fn test_execute_query_version() {
-    test_execute_with_version(Option::Some(QUERY_VERSION));
+    test_execute_with_version(Option::Some(QUERY_VERSION()));
 }
 
 #[test]
 #[available_gas(2000000)]
 #[should_panic(expected: ('Account: invalid tx version', 'ENTRYPOINT_FAILED'))]
 fn test_execute_invalid_version() {
-    test_execute_with_version(Option::Some(TRANSACTION_VERSION - 1));
+    test_execute_with_version(Option::Some(TRANSACTION_VERSION() - 1));
 }
 
 #[test]

src/tests/presets/test_account.cairo

@@ -1,13 +1,14 @@
 use openzeppelin::account::AccountComponent::{OwnerAdded, OwnerRemoved};
-use openzeppelin::account::AccountComponent::{TRANSACTION_VERSION, QUERY_VERSION};
 use openzeppelin::account::interface::ISRC6_ID;
 use openzeppelin::account::{AccountABIDispatcherTrait, AccountABIDispatcher};
 use openzeppelin::introspection::interface::ISRC5_ID;
 use openzeppelin::presets::Account;
 use openzeppelin::tests::account::test_account::{
     deploy_erc20, SIGNED_TX_DATA, SignedTransactionData
 };
-use openzeppelin::tests::utils::constants::{PUBKEY, NEW_PUBKEY, SALT, ZERO, RECIPIENT};
+use openzeppelin::tests::utils::constants::{
+    PUBKEY, NEW_PUBKEY, SALT, ZERO, RECIPIENT, QUERY_VERSION, TRANSACTION_VERSION
+};
 use openzeppelin::tests::utils;
 use openzeppelin::token::erc20::interface::{IERC20DispatcherTrait, IERC20Dispatcher};
 use openzeppelin::utils::selectors;
@@ -34,7 +35,7 @@ fn setup_dispatcher() -> AccountABIDispatcher {
 }
 
 fn setup_dispatcher_with_data(data: Option<@SignedTransactionData>) -> AccountABIDispatcher {
-    testing::set_version(TRANSACTION_VERSION);
+    testing::set_version(TRANSACTION_VERSION());
 
     let mut calldata = array![];
     if data.is_some() {
@@ -328,17 +329,22 @@ fn test_execute() {
     test_execute_with_version(Option::None(()));
 }
 
+#[test]
+fn test_execute_future_version() {
+    test_execute_with_version(Option::Some(TRANSACTION_VERSION() + 1));
+}
+
 #[test]
 #[available_gas(2000000)]
 fn test_execute_query_version() {
-    test_execute_with_version(Option::Some(QUERY_VERSION));
+    test_execute_with_version(Option::Some(QUERY_VERSION()));
 }
 
 #[test]
 #[available_gas(2000000)]
 #[should_panic(expected: ('Account: invalid tx version', 'ENTRYPOINT_FAILED'))]
 fn test_execute_invalid_version() {
-    test_execute_with_version(Option::Some(TRANSACTION_VERSION - 1));
+    test_execute_with_version(Option::Some(TRANSACTION_VERSION() - 1));
 }
 
 #[test]

src/tests/utils/constants.cairo

@@ -1,3 +1,4 @@
+use openzeppelin::account::utils;
 use starknet::ClassHash;
 use starknet::ContractAddress;
 use starknet::class_hash_const;
@@ -75,3 +76,11 @@ fn DATA(success: bool) -> Span<felt252> {
     }
     data.span()
 }
+
+fn TRANSACTION_VERSION() -> felt252 {
+    utils::TRANSACTION_VERSION.try_into().unwrap()
+}
+
+fn QUERY_VERSION() -> felt252 {
+    utils::QUERY_VERSION.try_into().unwrap()
+}