Custom proof size limit

aleph-client/src/contract/mod.rs

@@ -61,12 +61,15 @@ use crate::{
 
 /// Default gas limit, which allows up to 25% of block execution time.
 pub const DEFAULT_MAX_GAS: u64 = 250_000_000_000u64;
+/// Default proof size limit, which allows up to 25% of block execution time.
+pub const DEFAULT_MAX_PROOF_SIZE: u64 = 250_000_000_000u64;
 
 /// Represents a contract instantiated on the chain.
 pub struct ContractInstance {
     address: AccountId,
     transcoder: ContractMessageTranscoder,
     max_gas_override: Option<u64>,
+    max_proof_size_override: Option<u64>,
 }
 
 impl ContractInstance {
@@ -76,6 +79,7 @@ impl ContractInstance {
             address,
             transcoder: ContractMessageTranscoder::load(metadata_path)?,
             max_gas_override: None,
+            max_proof_size_override: None,
         })
     }
 
@@ -85,6 +89,12 @@ impl ContractInstance {
         self.max_gas_override = limit_override;
     }
 
+    /// From now on, the contract instance will use `limit_override` as the proof size limit for all
+    /// contract calls. If `limit_override` is `None`, then [DEFAULT_MAX_PROOF_SIZE] will be used.
+    pub fn override_proof_size_limit(&mut self, limit_override: Option<u64>) {
+        self.max_proof_size_override = limit_override;
+    }
+
     /// The address of this contract instance.
     pub fn address(&self) -> &AccountId {
         &self.address
@@ -178,7 +188,9 @@ impl ContractInstance {
             value,
             Weight {
                 ref_time: self.max_gas_override.unwrap_or(DEFAULT_MAX_GAS),
-                proof_size: self.max_gas_override.unwrap_or(DEFAULT_MAX_GAS),
+                proof_size: self
+                    .max_proof_size_override
+                    .unwrap_or(DEFAULT_MAX_PROOF_SIZE),
             },
             None,
             data,