From 87656c869c3b0555136136476b277f4407e63c74 Mon Sep 17 00:00:00 2001 From: Sergey B Kirpichev Date: Sun, 24 Nov 2024 10:37:37 +0300 Subject: [PATCH] gh-113841: fix possible undefined division by 0 in _Py_c_pow() (GH-127211) `x**y == 1/x**-y ` thus changing `/=` to `*=` by negating the exponent. (cherry picked from commit f7bb658124aba74be4c13f498bf46cfded710ef9) Co-authored-by: Sergey B Kirpichev --- Lib/test/test_complex.py | 5 +++++ .../2024-11-24-07-01-28.gh-issue-113841.WFg-Bu.rst | 2 ++ Objects/complexobject.c | 2 +- 3 files changed, 8 insertions(+), 1 deletion(-) create mode 100644 Misc/NEWS.d/next/Core_and_Builtins/2024-11-24-07-01-28.gh-issue-113841.WFg-Bu.rst diff --git a/Lib/test/test_complex.py b/Lib/test/test_complex.py index d5e58e3c6bc59a..a991040d811afb 100644 --- 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 new file mode 100644 index 00000000000000..2b07fdfcc6b527 --- /dev/null +++ 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 index 1aa3960a1c63f9..502b4a9a96119e 100644 --- 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);