Prevent YIsZero error in v2 price calculation

darkforestry · May 8, 2023 · 2b787df · 2b787df
1 parent f85831c
commit 2b787df
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/amm/uniswap_v2/mod.rs b/src/amm/uniswap_v2/mod.rs
@@ -337,9 +337,17 @@ impl UniswapV2Pool {
         };
 
         if base_token == self.token_a {
-            div_uu(r_1, r_0)
+            if r_0.is_zero() {
+                Ok(U128_0X10000000000000000)
+            } else {
+                div_uu(r_1, r_0)
+            }
         } else {
-            div_uu(r_0, r_1)
+            if r_1.is_zero() {
+                Ok(U128_0X10000000000000000)
+            } else {
+                div_uu(r_0, r_1)
+            }
         }
     }