Merge pull request #173 from pessimistic-io/math-updradable

added OZ upgradeable versions to whitelist

Author: Yhtiyar

slitherin/detectors/dubious_typecast.py

@@ -71,7 +71,12 @@ class DubiousTypecast(AbstractDetector):
     )
     WIKI_RECOMMENDATION = "Use clear constants"
 
-    WHITELIST = ["SafeCast", "SignedMath"]  # OZ
+    WHITELIST = [
+        "SafeCast",
+        "SignedMath",
+        "SafeCastUpgradeable",
+        "SignedMathUpgradeable",
+    ]  # OZ
 
     def analyze_irs(self, irs: List[Operation]) -> List[Tuple[str, str]]:
         results = []

slitherin/detectors/magic_number.py

@@ -26,7 +26,7 @@ class MagicNumber(AbstractDetector):
 
     EXCEPTION = {"0", "1", "2", "1000", "1e18"}
     used_count = defaultdict(lambda: {"count": 0, "nodes": []})
-    WHITELIST = ["SafeCast", "Math"]
+    WHITELIST = ["SafeCast", "Math", "MathUpgradeable", "SafeCastUpgradeable"]
 
     def _check_if_pow_10(self, str: str) -> bool:
         reg = re.fullmatch(r"^10*$|^10*e\d+$", str)  # 1(0..) or 1(0..)eX