fix checkpoint bug

Author: megakabi

crates/context/interface/src/journaled_state.rs

@@ -252,6 +252,8 @@ pub struct JournalCheckpoint {
     pub log_i: usize,
     /// Checkpoint to where on revert we will go back to and revert other journal entries.
     pub journal_i: usize,
+    /// Refund value at the time of checkpoint creation.
+    pub refund: i64,
 }
 
 /// State load information that contains the data and if the account or storage is cold loaded

crates/context/src/journal/inner.rs

@@ -461,6 +461,7 @@ impl<ENTRY: JournalEntryTr> JournalInner<ENTRY> {
         let checkpoint = JournalCheckpoint {
             log_i: self.logs.len(),
             journal_i: self.journal.len(),
+            refund: self.refund,
         };
         self.depth += 1;
         checkpoint
@@ -481,6 +482,9 @@ impl<ENTRY: JournalEntryTr> JournalInner<ENTRY> {
         self.depth -= 1;
         self.logs.truncate(checkpoint.log_i);
 
+        // Restore refund to checkpoint value
+        self.refund = checkpoint.refund;
+
         // iterate over last N journals sets and revert our global state
         self.journal
             .drain(checkpoint.journal_i..)