Don't run the GC in the middle of long operations

Author: Fidget-Spinner

Include/internal/pycore_opcode_metadata.h

@@ -207,6 +207,9 @@ extern int _PyUnicodeError_GetParams(
     Py_ssize_t *slen,
     int as_bytes);
 
+
+int _PyErr_CheckSignalsWithoutGC(void);
+
 #ifdef __cplusplus
 }
 #endif

Include/internal/pycore_pyerrors.h

@@ -1788,6 +1788,26 @@ PyErr_CheckSignals(void)
     return _PyErr_CheckSignalsTstate(tstate);
 }
 
+/* Same as PyErr_CheckSignals but does not run the GC.
+   This can be safely used if you are certain that no allocation
+   is done inside the long-running C code.
+ */
+int
+_PyErr_CheckSignalsWithoutGC(void)
+{
+    PyThreadState *tstate = _PyThreadState_GET();
+
+#if defined(Py_REMOTE_DEBUG) && defined(Py_SUPPORTS_REMOTE_DEBUG)
+    _PyRunRemoteDebugger(tstate);
+#endif
+
+    if (!_Py_ThreadCanHandleSignals(tstate->interp)) {
+        return 0;
+    }
+
+    return _PyErr_CheckSignalsTstate(tstate);
+}
+
 
 /* Declared in cpython/pyerrors.h */
 int

Include/internal/pycore_uop_metadata.h

@@ -113,7 +113,7 @@ maybe_small_long(PyLongObject *v)
 
 #define SIGCHECK(PyTryBlock)                    \
     do {                                        \
-        if (PyErr_CheckSignals()) PyTryBlock    \
+        if (_PyErr_CheckSignalsWithoutGC()) PyTryBlock    \
     } while(0)
 
 /* Normalize (remove leading zeros from) an int object.

Modules/signalmodule.c

@@ -635,6 +635,9 @@ def has_error_without_pop(op: parser.CodeDef) -> bool:
     "_PyLong_IsNegative",
     "_PyLong_IsNonNegativeCompact",
     "_PyLong_IsZero",
+    "_PyLong_Add",
+    "_PyLong_Multiply",
+    "_PyLong_Subtract",
     "_PyManagedDictPointer_IsValues",
     "_PyObject_GC_IS_SHARED",
     "_PyObject_GC_IS_TRACKED",