add div by zero check

elys-network · Dec 9, 2024 · 4d6a756 · 4d6a756
1 parent 72b5aa5
commit 4d6a756
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/x/amm/types/pool_calc_join_pool_no_swap_shares.go b/x/amm/types/pool_calc_join_pool_no_swap_shares.go
@@ -31,6 +31,10 @@ func MaximalExactRatioJoin(p *Pool, tokensIn sdk.Coins) (numShares sdkmath.Int,
 		// Note: QuoInt implements floor division, unlike Quo
 		// This is because it calls the native golang routine big.Int.Quo
 		// https://pkg.go.dev/math/big#Int.Quo
+		// Division by zero check
+		if poolLiquidity.AmountOfNoDenomValidation(coin.Denom).IsZero() {
+			return numShares, remCoins, errors.New("pool liquidity is zero for denom: " + coin.Denom)
+		}
 		shareRatio := sdkmath.LegacyNewDecFromBigInt(coin.Amount.BigInt()).QuoInt(poolLiquidity.AmountOfNoDenomValidation(coin.Denom))
 		if shareRatio.LT(minShareRatio) {
 			minShareRatio = shareRatio