pythongh-113841: fix possible undefined division by 0 in _Py_c_pow() (p…

…ythonGH-127211) `x**y == 1/x**-y ` thus changing `/=` to `*=` by negating the exponent. (cherry picked from commit f7bb658) Co-authored-by: Sergey B Kirpichev <[email protected]>
miss-islington · Nov 24, 2024 · 87656c8 · 87656c8
1 parent 69472a5
commit 87656c8
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/Lib/test/test_complex.py b/Lib/test/test_complex.py
@@ -301,6 +301,11 @@ def test_pow(self):
                     except OverflowError:
                         pass
 
+        # gh-113841: possible undefined division by 0 in _Py_c_pow()
+        x, y = 9j, 33j**3
+        with self.assertRaises(OverflowError):
+            x**y
+
     def test_pow_with_small_integer_exponents(self):
         # Check that small integer exponents are handled identically
         # regardless of their type.

diff --git a/Misc/NEWS.d/next/Core_and_Builtins/2024-11-24-07-01-28.gh-issue-113841.WFg-Bu.rst b/Misc/NEWS.d/next/Core_and_Builtins/2024-11-24-07-01-28.gh-issue-113841.WFg-Bu.rst
@@ -0,0 +1,2 @@
+Fix possible undefined behavior division by zero in :class:`complex`'s
+:c:func:`_Py_c_pow`.
diff --git a/Objects/complexobject.c b/Objects/complexobject.c
@@ -147,7 +147,7 @@ _Py_c_pow(Py_complex a, Py_complex b)
         at = atan2(a.imag, a.real);
         phase = at*b.real;
         if (b.imag != 0.0) {
-            len /= exp(at*b.imag);
+            len *= exp(-at*b.imag);
             phase += b.imag*log(vabs);
         }
         r.real = len*cos(phase);