Inline PyLong_AsInt64() to avoid the exception

vstinner · vstinner · commit d70a1214a675 · 2024-09-17T10:55:42.000+02:00
diff --git a/Objects/longobject.c b/Objects/longobject.c
@@ -6797,17 +6797,22 @@ PyLong_Export(PyObject *obj, PyLongExport *export_long)
         PyErr_Format(PyExc_TypeError, "expect int, got %T", obj);
         return -1;
     }
+
     int64_t value;
-    if (PyLong_AsInt64(obj, &value) >= 0) {
+    int flags = Py_ASNATIVEBYTES_NATIVE_ENDIAN;
+    Py_ssize_t bytes = PyLong_AsNativeBytes(obj, &value, sizeof(value), flags);
+    if (bytes < 0) {
+        return -1;
+    }
+
+    if ((size_t)bytes <= sizeof(value)) {
         export_long->value = value;
         export_long->negative = 0;
         export_long->ndigits = 0;
         export_long->digits = 0;
         export_long->_reserved = 0;
     }
     else {
-        PyErr_Clear();
-
         PyLongObject *self = (PyLongObject*)obj;
         export_long->value = 0;
         export_long->negative = _PyLong_IsNegative(self);
