diff options
author | Miss Islington (bot) <31488909+miss-islington@users.noreply.github.com> | 2017-10-23 15:20:13 (GMT) |
---|---|---|
committer | Serhiy Storchaka <storchaka@gmail.com> | 2017-10-23 15:20:13 (GMT) |
commit | 6e45d7b90accbbdfef353c41ab0a78a3e4742803 (patch) | |
tree | 12d365a7f7ecf144b54677b11fb72e6e0e3f0b90 /Modules | |
parent | aaf6a3dbbdb9754f98d480b468adfcae0f66e3a2 (diff) | |
download | cpython-6e45d7b90accbbdfef353c41ab0a78a3e4742803.zip cpython-6e45d7b90accbbdfef353c41ab0a78a3e4742803.tar.gz cpython-6e45d7b90accbbdfef353c41ab0a78a3e4742803.tar.bz2 |
bpo-31752: Fix possible crash in timedelta constructor called with custom integers. (GH-3947) (#4086)
Bad remainder in divmod() in intermediate calculations caused an assertion failure.
(cherry picked from commit 4ffd4653a7ec9c97775472276cf5e159e2366bb2)
Diffstat (limited to 'Modules')
-rw-r--r-- | Modules/_datetimemodule.c | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/Modules/_datetimemodule.c b/Modules/_datetimemodule.c index d44bb3e..fe92b93 100644 --- a/Modules/_datetimemodule.c +++ b/Modules/_datetimemodule.c @@ -1533,6 +1533,7 @@ delta_to_microseconds(PyDateTime_Delta *self) if (x2 == NULL) goto Done; result = PyNumber_Add(x1, x2); + assert(result == NULL || PyLong_CheckExact(result)); Done: Py_XDECREF(x1); @@ -1555,6 +1556,7 @@ microseconds_to_delta_ex(PyObject *pyus, PyTypeObject *type) PyObject *num = NULL; PyObject *result = NULL; + assert(PyLong_CheckExact(pyus)); tuple = PyNumber_Divmod(pyus, us_per_second); if (tuple == NULL) goto Done; @@ -2108,11 +2110,13 @@ accum(const char* tag, PyObject *sofar, PyObject *num, PyObject *factor, assert(num != NULL); if (PyLong_Check(num)) { - prod = PyNumber_Multiply(num, factor); + prod = PyNumber_Multiply(factor, num); if (prod == NULL) return NULL; + assert(PyLong_CheckExact(prod)); sum = PyNumber_Add(sofar, prod); Py_DECREF(prod); + assert(sum == NULL || PyLong_CheckExact(sum)); return sum; } @@ -2155,7 +2159,7 @@ accum(const char* tag, PyObject *sofar, PyObject *num, PyObject *factor, * fractional part requires float arithmetic, and may * lose a little info. */ - assert(PyLong_Check(factor)); + assert(PyLong_CheckExact(factor)); dnum = PyLong_AsDouble(factor); dnum *= fracpart; @@ -2170,6 +2174,7 @@ accum(const char* tag, PyObject *sofar, PyObject *num, PyObject *factor, Py_DECREF(sum); Py_DECREF(x); *leftover += fracpart; + assert(y == NULL || PyLong_CheckExact(y)); return y; } |