Skip to content

Commit

Permalink
bpo-39829: Fix __len__() is called twice in list() constructor (pyt…
Browse files Browse the repository at this point in the history
…honGH-31816)

(cherry picked from commit 2153daf)

Co-authored-by: Crowthebird <[email protected]>
  • Loading branch information
thatbirdguythatuknownot authored and miss-islington committed May 18, 2022
1 parent add8820 commit 77d9d68
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 17 deletions.
1 change: 1 addition & 0 deletions Misc/ACKS
Original file line number Diff line number Diff line change
Expand Up @@ -1332,6 +1332,7 @@ William Park
Claude Paroz
Heikki Partanen
Harri Pasanen
Jeremiah Gabriel Pascual
Gaël Pasgrimaud
Feanil Patel
Ashish Nitin Patil
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
Removed the ``__len__()`` call when initializing a list and moved initializing to ``list_extend``. Patch by Jeremiah Pascual.
29 changes: 12 additions & 17 deletions Objects/listobject.c
Original file line number Diff line number Diff line change
Expand Up @@ -863,7 +863,6 @@ list_extend(PyListObject *self, PyObject *iterable)
PyObject *it; /* iter(v) */
Py_ssize_t m; /* size of self */
Py_ssize_t n; /* guess for size of iterable */
Py_ssize_t mn; /* m + n */
Py_ssize_t i;
PyObject *(*iternext)(PyObject *);

Expand All @@ -887,7 +886,13 @@ list_extend(PyListObject *self, PyObject *iterable)
/* It should not be possible to allocate a list large enough to cause
an overflow on any relevant platform */
assert(m < PY_SSIZE_T_MAX - n);
if (list_resize(self, m + n) < 0) {
if (self->ob_item == NULL) {
if (list_preallocate_exact(self, n) < 0) {
return NULL;
}
Py_SET_SIZE(self, n);
}
else if (list_resize(self, m + n) < 0) {
Py_DECREF(iterable);
return NULL;
}
Expand Down Expand Up @@ -926,10 +931,13 @@ list_extend(PyListObject *self, PyObject *iterable)
* eventually run out of memory during the loop.
*/
}
else if (self->ob_item == NULL) {
if (n && list_preallocate_exact(self, n) < 0)
goto error;
}
else {
mn = m + n;
/* Make room. */
if (list_resize(self, mn) < 0)
if (list_resize(self, m + n) < 0)
goto error;
/* Make the list sane again. */
Py_SET_SIZE(self, m);
Expand Down Expand Up @@ -2717,19 +2725,6 @@ list___init___impl(PyListObject *self, PyObject *iterable)
(void)_list_clear(self);
}
if (iterable != NULL) {
if (_PyObject_HasLen(iterable)) {
Py_ssize_t iter_len = PyObject_Size(iterable);
if (iter_len == -1) {
if (!PyErr_ExceptionMatches(PyExc_TypeError)) {
return -1;
}
PyErr_Clear();
}
if (iter_len > 0 && self->ob_item == NULL
&& list_preallocate_exact(self, iter_len)) {
return -1;
}
}
PyObject *rv = list_extend(self, iterable);
if (rv == NULL)
return -1;
Expand Down

0 comments on commit 77d9d68

Please sign in to comment.