Finalize VM first

tlsnotary · Nov 19, 2023 · 019a4d9 · 019a4d9
1 parent 2250be4
commit 019a4d9
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tlsn/tlsn-prover/src/tls/prove.rs b/tlsn/tlsn-prover/src/tls/prove.rs
@@ -162,19 +162,19 @@ impl Prover<ProveState> {
         let mut verify_fut = Box::pin(async move {
             let mut channel = verify_mux.get_channel("finalize").await?;
 
+            _ = vm
+                .finalize()
+                .await
+                .map_err(|e| ProverError::MpcError(Box::new(e)))?
+                .expect("encoder seed returned");
+
             // This is a temporary approach until a maliciously secure share conversion protocol is implemented.
             // The prover is essentially revealing the TLS MAC key. In some exotic scenarios this allows a malicious
             // TLS verifier to modify the prover's request.
             gf2.reveal()
                 .await
                 .map_err(|e| ProverError::MpcError(Box::new(e)))?;
 
-            _ = vm
-                .finalize()
-                .await
-                .map_err(|e| ProverError::MpcError(Box::new(e)))?
-                .expect("encoder seed returned");
-
             // Send session_info to the verifier
             channel.send(TlsnMessage::SessionInfo(session_info)).await?;