fix: compute recv record count from max_recv (tlsnotary#743)

* fix: compute recv record count from max_recv

* pad after check

* fix: add `max_recv` to mpc-tls integration test

---------

Co-authored-by: th4s <[email protected]>

Author: sinui0

crates/mpc-tls/src/config.rs

@@ -27,6 +27,9 @@ pub struct Config {
     /// the TLS connection is active. Data which can be decrypted after the TLS
     /// connection will be decrypted for free.
     pub(crate) max_recv_online: usize,
+    /// Maximum number of received bytes.
+    #[allow(unused)]
+    pub(crate) max_recv: usize,
 }
 
 impl Config {
@@ -44,24 +47,35 @@ impl ConfigBuilder {
             + self
                 .max_sent
                 .ok_or(ConfigBuilderError::UninitializedField("max_sent"))?;
-        let max_recv_online = MIN_RECV
-            + self
-                .max_recv_online
-                .ok_or(ConfigBuilderError::UninitializedField("max_recv_online"))?;
+        let mut max_recv_online = self
+            .max_recv_online
+            .ok_or(ConfigBuilderError::UninitializedField("max_recv_online"))?;
+        let max_recv = self
+            .max_recv
+            .ok_or(ConfigBuilderError::UninitializedField("max_recv"))?;
+
+        if max_recv_online > max_recv {
+            return Err(ConfigBuilderError::ValidationError(
+                "max_recv_online must be less than or equal to max_recv".to_string(),
+            ));
+        }
+
+        max_recv_online += MIN_RECV;
 
         let max_sent_records = self
             .max_sent_records
             .unwrap_or_else(|| MIN_SENT_RECORDS + max_sent.div_ceil(16384));
         let max_recv_records = self
             .max_recv_records
-            .unwrap_or_else(|| MIN_RECV_RECORDS + max_recv_online.div_ceil(16384));
+            .unwrap_or_else(|| MIN_RECV_RECORDS + max_recv.div_ceil(16384));
 
         Ok(Config {
             defer_decryption,
             max_sent_records,
             max_sent,
             max_recv_records,
             max_recv_online,
+            max_recv,
         })
     }
 }

crates/mpc-tls/tests/test.rs

@@ -28,6 +28,7 @@ async fn mpc_tls_test() {
         .defer_decryption(false)
         .max_sent(1 << 13)
         .max_recv_online(1 << 13)
+        .max_recv(1 << 13)
         .build()
         .unwrap();
 

crates/prover/src/config.rs

@@ -53,6 +53,7 @@ impl ProverConfig {
             .defer_decryption(self.defer_decryption_from_start)
             .max_sent(self.protocol_config.max_sent_data())
             .max_recv_online(self.protocol_config.max_recv_data_online())
+            .max_recv(self.protocol_config.max_recv_data())
             .build()
             .unwrap()
     }

crates/verifier/src/config.rs

@@ -46,6 +46,7 @@ impl VerifierConfig {
         Config::builder()
             .max_sent(protocol_config.max_sent_data())
             .max_recv_online(protocol_config.max_recv_data_online())
+            .max_recv(protocol_config.max_recv_data())
             .build()
             .unwrap()
     }