fix: clean up heap allowance and balances

rs/ledger_suite/common/ledger_core/src/approvals.rs

@@ -66,6 +66,8 @@ pub trait AllowancesData {
     fn len_expirations(&self) -> usize;
 
     fn clear_arrivals(&mut self);
+
+    fn clear_all(&mut self);
 }
 
 #[derive(Debug, Deserialize, Serialize)]
@@ -159,6 +161,12 @@ where
     fn clear_arrivals(&mut self) {
         self.arrival_queue.clear();
     }
+
+    fn clear_all(&mut self) {
+        self.allowances.clear();
+        self.expiration_queue.clear();
+        self.arrival_queue.clear();
+    }
 }
 
 #[derive(Clone, Eq, PartialEq, Debug, Deserialize, Serialize)]

rs/ledger_suite/icp/ledger/src/lib.rs

@@ -732,4 +732,8 @@ impl AllowancesData for StableAllowancesData {
     fn clear_arrivals(&mut self) {
         panic!("The method `clear_arrivals` should not be called for StableAllowancesData")
     }
+
+    fn clear_all(&mut self) {
+        panic!("The method `clear_all` should not be called for StableAllowancesData")
+    }
 }

rs/ledger_suite/icrc1/ledger/src/lib.rs

@@ -727,6 +727,11 @@ impl Ledger {
     pub fn copy_token_pool(&mut self) {
         self.stable_balances.token_pool = self.balances.token_pool;
     }
+
+    pub fn clear_heap_approvals_and_balances(&mut self) {
+        self.approvals.allowances_data.clear_all();
+        self.balances.store.clear();
+    }
 }
 
 impl LedgerContext for Ledger {
@@ -1257,6 +1262,10 @@ impl AllowancesData for StableAllowancesData {
     fn clear_arrivals(&mut self) {
         panic!("The method `clear_arrivals` should not be called for StableAllowancesData")
     }
+
+    fn clear_all(&mut self) {
+        panic!("The method `clear_all` should not be called for StableAllowancesData")
+    }
 }
 
 #[derive(Serialize, Deserialize, Debug, Default, PartialEq)]

rs/ledger_suite/icrc1/ledger/src/main.rs

@@ -259,6 +259,10 @@ fn post_upgrade(args: Option<LedgerArgument>) {
         migrate_next_part(
             MAX_INSTRUCTIONS_PER_UPGRADE.saturating_sub(pre_upgrade_instructions_consumed),
         );
+    } else {
+        Access::with_ledger_mut(|ledger| {
+            ledger.clear_heap_approvals_and_balances();
+        });
     }
 
     let end = ic_cdk::api::instruction_counter();