ref(erc20): panic in _update

OpenZeppelin · Apr 9, 2024 · a586a3a · a586a3a
1 parent 1da8c70
commit a586a3a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/contracts/src/erc20/mod.rs b/contracts/src/erc20/mod.rs
@@ -327,7 +327,11 @@ impl ERC20 {
             // `_total_supply` never overflows.
             // TODO: Think about SafeMath library
             let total_supply = self.total_supply();
-            self._total_supply.set(total_supply + value);
+            self._total_supply.set(
+                total_supply
+                    .checked_add(value)
+                    .expect("Should not exceed U256::MAX for `_total_supply"),
+            );
         } else {
             let from_balance = self._balances.get(from);
             if from_balance < value {