add before_swap_check validation in reverse swap and swap_exact_amoun…

…t_out
astroport-fi · Jan 3, 2024 · caaafa9 · caaafa9
1 parent 19bfecb
commit caaafa9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/contracts/pair_concentrated/src/queries.rs b/contracts/pair_concentrated/src/queries.rs
@@ -275,6 +275,8 @@ pub fn query_reverse_simulation(
 
     let pools = query_pools(deps.querier, &env.contract.address, &config, &precisions)?;
 
+    before_swap_check(&pools, ask_asset_dec.amount)?;
+
     let (ask_ind, _) = pools
         .iter()
         .find_position(|asset| asset.info == ask_asset.info)

diff --git a/contracts/pair_concentrated/src/sudo.rs b/contracts/pair_concentrated/src/sudo.rs
@@ -95,6 +95,16 @@ fn swap_exact_amount_out(
         .pair_info
         .query_pools(&deps.querier, &env.contract.address)?;
 
+    // Before swap checks
+    ensure!(
+        !token_out.amount.is_zero(),
+        StdError::generic_err("Token out amount must not be zero")
+    );
+    ensure!(
+        pools.iter().all(|a| !a.amount.is_zero()),
+        StdError::generic_err("One of the pools is empty")
+    );
+
     let ask_ind = pools
         .iter()
         .position(|asset| asset.info == ask_asset.info)