force panic in Add operator

CosmWasm · Mar 31, 2021 · 6f81c24 · 6f81c24
1 parent e005f15
commit 6f81c24
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/packages/std/src/math.rs b/packages/std/src/math.rs
@@ -324,27 +324,27 @@ impl ops::Add<Uint128> for Uint128 {
     type Output = Self;
 
     fn add(self, rhs: Self) -> Self {
-        Uint128(self.u128() + rhs.u128())
+        Uint128(self.u128().checked_add(rhs.u128()).unwrap())
     }
 }
 
 impl<'a> ops::Add<&'a Uint128> for Uint128 {
     type Output = Self;
 
     fn add(self, rhs: &'a Uint128) -> Self {
-        Uint128(self.u128() + rhs.u128())
+        Uint128(self.u128().checked_add(rhs.u128()).unwrap())
     }
 }
 
 impl ops::AddAssign<Uint128> for Uint128 {
     fn add_assign(&mut self, rhs: Uint128) {
-        self.0 += rhs.u128();
+        self.0 = self.0.checked_add(rhs.u128()).unwrap();
     }
 }
 
 impl<'a> ops::AddAssign<&'a Uint128> for Uint128 {
     fn add_assign(&mut self, rhs: &'a Uint128) {
-        self.0 += rhs.u128();
+        self.0 = self.0.checked_add(rhs.u128()).unwrap();
     }
 }