added OZ upgradeable versions to whitelist

pessimistic-io · Jun 19, 2024 · 391e8d2 · 391e8d2
1 parent f3324ab
commit 391e8d2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/slitherin/detectors/dubious_typecast.py b/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 = []

diff --git a/slitherin/detectors/magic_number.py b/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