From cac4f4b1b18b574fb3790446af3ca8cf8910d4e7 Mon Sep 17 00:00:00 2001
From: Hardhat Chad <hardhatchad@gmail.com>
Date: Thu, 25 Jul 2024 20:30:25 +0000
Subject: [PATCH] div by zero check

---
 program/src/mine.rs | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/program/src/mine.rs b/program/src/mine.rs
index ea7673a..8c03516 100644
--- a/program/src/mine.rs
+++ b/program/src/mine.rs
@@ -117,16 +117,16 @@ pub fn process_mine<'a, 'info>(
     // if the miner's last stake deposit was more than one minute ago.
     let mut bus_data = bus_info.data.borrow_mut();
     let bus = Bus::try_from_bytes_mut(&mut bus_data)?;
-    if config.top_balance.gt(&0)
-        && proof.balance.gt(&0)
-        && proof.last_stake_at.saturating_add(ONE_MINUTE).lt(&t)
-    {
-        let staking_reward = (reward as u128)
-            .checked_mul(proof.balance.min(config.top_balance) as u128)
-            .unwrap()
-            .checked_div(config.top_balance as u128)
-            .unwrap() as u64;
-        reward = reward.checked_add(staking_reward).unwrap();
+    if proof.balance.gt(&0) && proof.last_stake_at.saturating_add(ONE_MINUTE).lt(&t) {
+        // Update staking reward
+        if config.top_balance.gt(&0) {
+            let staking_reward = (reward as u128)
+                .checked_mul(proof.balance.min(config.top_balance) as u128)
+                .unwrap()
+                .checked_div(config.top_balance as u128)
+                .unwrap() as u64;
+            reward = reward.checked_add(staking_reward).unwrap();
+        }
 
         // Update bus stake tracker if stake is active
         if proof.balance.gt(&bus.top_balance) {