Fix passing backend FPV objects to math.isnan and math.isinf

angr · Nov 18, 2024 · b8a291e · b8a291e
1 parent c7d8d82
commit b8a291e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/claripy/backends/backend_concrete/fp.py b/claripy/backends/backend_concrete/fp.py
@@ -403,11 +403,11 @@ def fpIsNaN(x):
     """
     Checks whether the argument is a floating point NaN.
     """
-    return math.isnan(x)
+    return math.isnan(x.value)
 
 
 def fpIsInf(x):
     """
     Checks whether the argument is a floating point infinity.
     """
-    return math.isinf(x)
+    return math.isinf(x.value)
diff --git a/tests/test_fp.py b/tests/test_fp.py
@@ -67,6 +67,12 @@ def test_fp_precision_loss(self):
         assert edd != edd2
         assert edd2 == 0x4237B4C7C0000000
 
+    def test_concrete_isnan(self):
+        assert claripy.FPV(0, claripy.FSORT_FLOAT).isNaN() is claripy.false()
+
+    def test_concrete_isinf(self):
+        assert claripy.FPV(0, claripy.FSORT_FLOAT).isInf() is claripy.false()
+
 
 if __name__ == "__main__":
     unittest.main()