clear transient storage at end tx

privacy-scaling-explorations · Apr 4, 2024 · c65544e · c65544e
1 parent d28a696
commit c65544e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/bus-mapping/src/circuit_input_builder.rs b/bus-mapping/src/circuit_input_builder.rs
@@ -460,6 +460,7 @@ impl<'a, C: CircuitsParams> CircuitInputBuilder<C> {
             // Generate EndTx step
             let end_tx_step =
                 gen_associated_steps(&mut self.state_ref(&mut tx, &mut tx_ctx), ExecState::EndTx)?;
+            self.sdb.clear_transient_storage();
             tx.steps_mut().push(end_tx_step.clone());
             (end_tx_step, last_call)
         } else if self.feature_config.invalid_tx {

diff --git a/bus-mapping/src/state_db.rs b/bus-mapping/src/state_db.rs
@@ -332,6 +332,11 @@ impl StateDB {
         }
         self.refund = 0;
     }
+
+    /// Clear transient storage.
+    pub fn clear_transient_storage(&mut self) {
+        self.transient_storage = HashMap::new();
+    }
 }
 
 #[cfg(test)]