Merge branch 'main' into gh-102509

python · markshannon · Jul 28, 2023 · Mar 7, 2023 · Mar 13, 2023 · Jul 28, 2023
commit d1016a6dc7f9cd02cde6e8658fd92532b0dc9439
diff --git a/Objects/longobject.c b/Objects/longobject.c
@@ -161,11 +161,31 @@ _PyLong_New(Py_ssize_t size)
         PyErr_NoMemory();
         return NULL;
     }
-    _PyObject_InitVar((PyVarObject*)result, &PyLong_Type, size);
+    _PyLong_SetSignAndDigitCount(result, size != 0, size);
+    _PyObject_Init((PyObject*)result, &PyLong_Type);
+    /* The digit has to be initialized explicitly to avoid
+     * use-of-uninitialized-value. */
     result->long_value.ob_digit[0] = 0;
     return result;
 }
 
+PyLongObject *
+_PyLong_FromDigits(int negative, Py_ssize_t digit_count, digit *digits)
+{
+    assert(digit_count >= 0);
+    if (digit_count == 0) {
+        return (PyLongObject *)Py_NewRef(_PyLong_GetZero());
+    }
+    PyLongObject *result = _PyLong_New(digit_count);
+    if (result == NULL) {
+        PyErr_NoMemory();
+        return NULL;
+    }
+    _PyLong_SetSignAndDigitCount(result, negative?-1:1, digit_count);
+    memcpy(result->long_value.ob_digit, digits, digit_count * sizeof(digit));
+    return result;
+}
+
 PyObject *
 _PyLong_Copy(PyLongObject *src)
 {