Set transaction size limit

Cargo.toml

@@ -108,12 +108,13 @@ metric_recorder = ["neard/metric_recorder"]
 delay_detector = ["neard/delay_detector"]
 rosetta_rpc = ["neard/rosetta_rpc"]
 nightly_protocol = ["near-primitives/nightly_protocol", "near-jsonrpc/nightly_protocol"]
-nightly_protocol_features = ["nightly_protocol", "neard/nightly_protocol_features", "protocol_feature_evm", "protocol_feature_block_header_v3", "protocol_feature_alt_bn128", "protocol_feature_access_key_nonce_range"]
+nightly_protocol_features = ["nightly_protocol", "neard/nightly_protocol_features", "protocol_feature_evm", "protocol_feature_block_header_v3", "protocol_feature_alt_bn128", "protocol_feature_access_key_nonce_range", "protocol_feature_tx_size_limit"]
 protocol_feature_forward_chunk_parts = ["neard/protocol_feature_forward_chunk_parts"]
 protocol_feature_evm = ["neard/protocol_feature_evm", "testlib/protocol_feature_evm", "runtime-params-estimator/protocol_feature_evm"]
 protocol_feature_alt_bn128 = ["neard/protocol_feature_alt_bn128", "testlib/protocol_feature_alt_bn128", "runtime-params-estimator/protocol_feature_alt_bn128"]
 protocol_feature_block_header_v3 = ["near-primitives/protocol_feature_block_header_v3", "near-chain/protocol_feature_block_header_v3", "neard/protocol_feature_block_header_v3"]
 protocol_feature_access_key_nonce_range = ["neard/protocol_feature_access_key_nonce_range"]
+protocol_feature_tx_size_limit = ["near-primitives/protocol_feature_tx_size_limit", "node-runtime/protocol_feature_tx_size_limit", "neard/protocol_feature_tx_size_limit"]
 
 # enable this to build neard with wasmer 1.0 runner
 # now if none of wasmer0_default, wasmer1_default or wasmtime_default is enabled, wasmer0 would be default

chain/chain/src/validate.rs

@@ -83,7 +83,7 @@ pub fn validate_chunk_proofs(chunk: &ShardChunk, runtime_adapter: &dyn RuntimeAd
 }
 
 /// Validates that the given transactions are in proper valid order.
-/// See https://nomicon.io/BlockchainLayer/Transactions.html#transaction-ordering
+/// See https://nomicon.io/ChainSpec/Transactions.html#transaction-ordering
 pub fn validate_transactions_order(transactions: &[SignedTransaction]) -> bool {
     let mut nonces: HashMap<(&AccountId, &PublicKey), Nonce> = HashMap::new();
     let mut batches: HashMap<(&AccountId, &PublicKey), usize> = HashMap::new();

core/primitives-core/Cargo.toml

@@ -27,3 +27,4 @@ default = []
 costs_counting = []
 protocol_feature_evm = []
 protocol_feature_alt_bn128 = []
+protocol_feature_tx_size_limit = []

core/primitives-core/src/config.rs

@@ -68,6 +68,9 @@ pub struct VMLimitConfig {
     pub max_length_returned_data: u64,
     /// Max contract size
     pub max_contract_size: u64,
+    /// Max transaction size
+    #[cfg(feature = "protocol_feature_tx_size_limit")]
+    pub max_transaction_size: u64,
     /// Max storage key size
     pub max_length_storage_key: u64,
     /// Max storage value size
@@ -149,6 +152,8 @@ impl Default for VMLimitConfig {
             max_arguments_length: 4 * 2u64.pow(20), // 4 Mib
             max_length_returned_data: 4 * 2u64.pow(20), // 4 Mib
             max_contract_size: 4 * 2u64.pow(20),    // 4 Mib,
+            #[cfg(feature = "protocol_feature_tx_size_limit")]
+            max_transaction_size: 4 * 2u64.pow(20), // 4 Mib
 
             max_length_storage_key: 4 * 2u64.pow(20), // 4 Mib
             max_length_storage_value: 4 * 2u64.pow(20), // 4 Mib

core/primitives/Cargo.toml

@@ -49,7 +49,9 @@ protocol_feature_evm = ["near-primitives-core/protocol_feature_evm"]
 protocol_feature_block_header_v3 = []
 protocol_feature_alt_bn128 = ["near-primitives-core/protocol_feature_alt_bn128", "near-vm-errors/protocol_feature_alt_bn128"]
 protocol_feature_access_key_nonce_range = []
-nightly_protocol_features = ["nightly_protocol", "protocol_feature_forward_chunk_parts", "protocol_feature_rectify_inflation", "protocol_feature_evm", "protocol_feature_block_header_v3", "protocol_feature_access_key_nonce_range", "protocol_feature_alt_bn128"]
+protocol_feature_tx_size_limit = ["near-primitives-core/protocol_feature_tx_size_limit"]
+
+nightly_protocol_features = ["nightly_protocol", "protocol_feature_forward_chunk_parts", "protocol_feature_rectify_inflation", "protocol_feature_evm", "protocol_feature_block_header_v3", "protocol_feature_access_key_nonce_range", "protocol_feature_alt_bn128", "protocol_feature_tx_size_limit"]
 nightly_protocol = []
 
 

core/primitives/src/errors.rs

@@ -199,6 +199,9 @@ pub enum ActionsValidationError {
     UnsuitableStakingKey { public_key: PublicKey },
     /// The attached amount of gas in a FunctionCall action has to be a positive number.
     FunctionCallZeroAttachedGas,
+    /// Transaction size exceeded the limit.
+    #[cfg(feature = "protocol_feature_tx_size_limit")]
+    TransactionSizeExceeded { size: u64, limit: u64 },
 }
 
 /// Describes the error for validating a receipt.
@@ -313,6 +316,12 @@ impl Display for ActionsValidationError {
                 f,
                 "The attached amount of gas in a FunctionCall action has to be a positive number",
             ),
+            #[cfg(feature = "protocol_feature_tx_size_limit")]
+            ActionsValidationError::TransactionSizeExceeded { size, limit } => write!(
+                f,
+                "Total transaction size {} exceeded the limit {}",
+                size, limit
+            ),
         }
     }
 }

core/primitives/src/transaction.rs

@@ -72,6 +72,14 @@ impl Action {
             _ => 0,
         }
     }
+    #[cfg(feature = "protocol_feature_tx_size_limit")]
+    pub fn get_size(&self) -> usize {
+        match self {
+            Action::FunctionCall(a) => a.args.len(),
+            Action::DeployContract(a) => a.code.len(),
+            _ => 0,
+        }
+    }
 }
 
 /// Create account action

core/primitives/src/version.rs

@@ -94,6 +94,8 @@ pub enum ProtocolFeature {
     AltBn128,
     #[cfg(feature = "protocol_feature_access_key_nonce_range")]
     AccessKeyNonceRange,
+    #[cfg(feature = "protocol_feature_tx_size_limit")]
+    TransactionSizeLimit,
 }
 
 /// Current latest stable version of the protocol.
@@ -102,7 +104,7 @@ pub const PROTOCOL_VERSION: ProtocolVersion = 42;
 
 /// Current latest nightly version of the protocol.
 #[cfg(feature = "nightly_protocol")]
-pub const PROTOCOL_VERSION: ProtocolVersion = 106;
+pub const PROTOCOL_VERSION: ProtocolVersion = 107;
 
 lazy_static! {
     static ref STABLE_PROTOCOL_FEATURES_TO_VERSION_MAPPING: HashMap<ProtocolFeature, ProtocolVersion> =
@@ -142,6 +144,8 @@ lazy_static! {
             (ProtocolFeature::AltBn128, 105),
             #[cfg(feature = "protocol_feature_access_key_nonce_range")]
             (ProtocolFeature::AccessKeyNonceRange, 106),
+            #[cfg(feature = "protocol_feature_tx_size_limit")]
+            (ProtocolFeature::TransactionSizeLimit, 107),
         ]
         .into_iter()
         .collect();

neard/Cargo.toml

@@ -72,7 +72,8 @@ protocol_feature_evm = ["near-primitives/protocol_feature_evm", "node-runtime/pr
 protocol_feature_alt_bn128 = ["near-primitives/protocol_feature_alt_bn128", "node-runtime/protocol_feature_alt_bn128"]
 protocol_feature_block_header_v3 = ["near-epoch-manager/protocol_feature_block_header_v3", "near-primitives/protocol_feature_block_header_v3", "near-chain/protocol_feature_block_header_v3", "near-client/protocol_feature_block_header_v3"]
 protocol_feature_access_key_nonce_range = ["near-primitives/protocol_feature_access_key_nonce_range", "node-runtime/protocol_feature_access_key_nonce_range", "near-client/protocol_feature_access_key_nonce_range"]
-nightly_protocol_features = ["nightly_protocol", "near-primitives/nightly_protocol_features", "near-client/nightly_protocol_features", "near-epoch-manager/nightly_protocol_features", "near-store/nightly_protocol_features", "protocol_feature_forward_chunk_parts", "protocol_feature_rectify_inflation", "protocol_feature_evm", "protocol_feature_block_header_v3", "protocol_feature_alt_bn128", "protocol_feature_access_key_nonce_range"]
+protocol_feature_tx_size_limit = ["near-primitives/protocol_feature_tx_size_limit", "node-runtime/protocol_feature_tx_size_limit"]
+nightly_protocol_features = ["nightly_protocol", "near-primitives/nightly_protocol_features", "near-client/nightly_protocol_features", "near-epoch-manager/nightly_protocol_features", "near-store/nightly_protocol_features", "protocol_feature_forward_chunk_parts", "protocol_feature_rectify_inflation", "protocol_feature_evm", "protocol_feature_block_header_v3", "protocol_feature_alt_bn128", "protocol_feature_access_key_nonce_range", "protocol_feature_tx_size_limit"]
 nightly_protocol = ["near-primitives/nightly_protocol", "near-jsonrpc/nightly_protocol"]
 
 [[bin]]

runtime/runtime/Cargo.toml

@@ -52,6 +52,7 @@ protocol_feature_alt_bn128 = [
     "near-vm-runner/protocol_feature_alt_bn128",
     "near-vm-errors/protocol_feature_alt_bn128",
 ]
+protocol_feature_tx_size_limit = []
 
 [dev-dependencies]
 tempfile = "3"

runtime/runtime/src/config.rs

@@ -60,6 +60,11 @@ pub fn safe_add_balance(a: Balance, b: Balance) -> Result<Balance, IntegerOverfl
     a.checked_add(b).ok_or_else(|| IntegerOverflowError {})
 }
 
+#[cfg(feature = "protocol_feature_tx_size_limit")]
+pub fn safe_add_usize(a: usize, b: usize) -> Result<usize, IntegerOverflowError> {
+    a.checked_add(b).ok_or_else(|| IntegerOverflowError {})
+}
+
 #[macro_export]
 macro_rules! safe_add_balance_apply {
     ($x: expr) => {$x};
@@ -269,6 +274,12 @@ pub fn total_prepaid_gas(actions: &[Action]) -> Result<Gas, IntegerOverflowError
     actions.iter().try_fold(0, |acc, action| safe_add_gas(acc, action.get_prepaid_gas()))
 }
 
+/// Get the total size of given actions.
+#[cfg(feature = "protocol_feature_tx_size_limit")]
+pub fn total_size(actions: &[Action]) -> Result<usize, IntegerOverflowError> {
+    actions.iter().try_fold(0, |acc, action| safe_add_usize(acc, action.get_size()))
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;

runtime/runtime/src/verifier.rs

@@ -20,6 +20,8 @@ use near_store::{
     get_access_key, get_account, set_access_key, set_account, StorageError, TrieUpdate,
 };
 
+#[cfg(feature = "protocol_feature_tx_size_limit")]
+use crate::config::total_size;
 use crate::config::{total_prepaid_gas, tx_cost, TransactionCost};
 use crate::VerificationResult;
 use near_primitives::checked_feature;
@@ -326,6 +328,18 @@ pub(crate) fn validate_actions(
         });
     }
 
+    #[cfg(feature = "protocol_feature_tx_size_limit")]
+    {
+        let total_size =
+            total_size(actions).map_err(|_| ActionsValidationError::IntegerOverflow)? as u64;
+        if total_size > limit_config.max_transaction_size {
+            return Err(ActionsValidationError::TransactionSizeExceeded {
+                size: total_size,
+                limit: limit_config.max_transaction_size,
+            });
+        }
+    }
+
     Ok(())
 }
 
@@ -1445,6 +1459,48 @@ mod tests {
         );
     }
 
+    #[test]
+    #[cfg(feature = "protocol_feature_tx_size_limit")]
+    fn test_validate_actions_checking_tx_size_limit() {
+        let mut limit_config = VMLimitConfig::default();
+        let contract_size = 5;
+        let args_size = 3;
+
+        limit_config.max_transaction_size = 3;
+        assert_eq!(
+            validate_actions(
+                &limit_config,
+                &vec![Action::DeployContract(DeployContractAction {
+                    code: vec![1; contract_size as usize]
+                }),]
+            )
+            .expect_err("Expected an error"),
+            ActionsValidationError::TransactionSizeExceeded {
+                size: contract_size,
+                limit: limit_config.max_transaction_size
+            },
+        );
+
+        limit_config.max_transaction_size = 10;
+        assert_eq!(
+            validate_actions(
+                &limit_config,
+                &vec![
+                    Action::DeployContract(DeployContractAction {
+                        code: vec![1; contract_size as usize]
+                    }),
+                    Action::FunctionCall(FunctionCallAction {
+                        method_name: "test-method".to_string(),
+                        args: vec![0; args_size as usize],
+                        gas: 100,
+                        deposit: 0
+                    })
+                ]
+            ),
+            Ok(()),
+        );
+    }
+
     // Individual actions
 
     #[test]

test-utils/testlib/Cargo.toml

@@ -54,3 +54,4 @@ protocol_feature_alt_bn128 = [
     "near-vm-errors/protocol_feature_alt_bn128",
 ]
 protocol_feature_evm = ["near-evm-runner/protocol_feature_evm", "near-primitives/protocol_feature_evm", "neard/protocol_feature_evm", "node-runtime/protocol_feature_evm", "near-chain-configs/protocol_feature_evm", "near-chain/protocol_feature_evm"]
+protocol_feature_tx_size_limit = ["near-primitives/protocol_feature_tx_size_limit", "node-runtime/protocol_feature_tx_size_limit"]