tighten deallocate truncation of stack (#1686)

2023-01-12 23:46:57 -07:00
parent 84583da5b8
commit 3a4aa2a541
1 changed files with 5 additions and 2 deletions
--- a/src/machine/machine_state_impl.rs
+++ b/src/machine/machine_state_impl.rs
@@ -2784,8 +2784,11 @@ impl MachineState {
        self.cp = frame.prelude.cp;
        self.e = frame.prelude.e;

-        if e > self.b {
-            self.stack.truncate(e);
+        if self.e > self.b {
+            let frame = self.stack.index_and_frame(self.e);
+            let size = AndFrame::size_of(frame.prelude.univ_prelude.num_cells);
+
+            self.stack.truncate(self.e + size);
        }

        self.p += 1;