From: Victor Stinner <vstinner@python.org>
Date: Sat, 23 Mar 2024 12:01:20 +0000 (+0100)
Subject: gh-117008: Fix functools test_recursive_pickle() (#117009)
X-Git-Tag: v3.13.0a6~176
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=9967b568edd2e35b0415c14c7242f3ca2c0dc03d;p=thirdparty%2FPython%2Fcpython.git

gh-117008: Fix functools test_recursive_pickle() (#117009)

Use support.infinite_recursion() in test_recursive_pickle() of
test_functools to prevent a stack overflow on "ARM64 Windows
Non-Debug" buildbot.

Lower Py_C_RECURSION_LIMIT to 1,000 frames on Windows ARM64.
---

diff --git a/Include/cpython/pystate.h b/Include/cpython/pystate.h
index f1540fde9259..7fb6b1763921 100644
--- a/Include/cpython/pystate.h
+++ b/Include/cpython/pystate.h
@@ -209,6 +209,8 @@ struct _ts {
 #  define Py_C_RECURSION_LIMIT 500
 #elif defined(__s390x__)
 #  define Py_C_RECURSION_LIMIT 800
+#elif defined(_WIN32) && defined(_M_ARM64)
+#  define Py_C_RECURSION_LIMIT 1000
 #elif defined(_WIN32)
 #  define Py_C_RECURSION_LIMIT 3000
 #elif defined(__ANDROID__)
diff --git a/Lib/test/test_functools.py b/Lib/test/test_functools.py
index 1a6d8afe6ed6..3ba4929dd1b1 100644
--- a/Lib/test/test_functools.py
+++ b/Lib/test/test_functools.py
@@ -334,8 +334,10 @@ class TestPartial:
             f.__setstate__((f, (), {}, {}))
             try:
                 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
-                    with self.assertRaises(RecursionError):
-                        pickle.dumps(f, proto)
+                    # gh-117008: Small limit since pickle uses C stack memory
+                    with support.infinite_recursion(100):
+                        with self.assertRaises(RecursionError):
+                            pickle.dumps(f, proto)
             finally:
                 f.__setstate__((capture, (), {}, {}))