bpo-31577: Fix a crash in os.utime() in case of a bad ns argument. (GH-3752)
This commit is contained in:
parent
e502451781
commit
0bd1a2dcfd
|
@ -635,6 +635,22 @@ class UtimeTests(unittest.TestCase):
|
|||
with self.assertRaises(ValueError):
|
||||
os.utime(self.fname, (5, 5), ns=(5, 5))
|
||||
|
||||
@support.cpython_only
|
||||
def test_issue31577(self):
|
||||
# The interpreter shouldn't crash in case utime() received a bad
|
||||
# ns argument.
|
||||
def get_bad_int(divmod_ret_val):
|
||||
class BadInt:
|
||||
def __divmod__(*args):
|
||||
return divmod_ret_val
|
||||
return BadInt()
|
||||
with self.assertRaises(TypeError):
|
||||
os.utime(self.fname, ns=(get_bad_int(42), 1))
|
||||
with self.assertRaises(TypeError):
|
||||
os.utime(self.fname, ns=(get_bad_int(()), 1))
|
||||
with self.assertRaises(TypeError):
|
||||
os.utime(self.fname, ns=(get_bad_int((1, 2, 3)), 1))
|
||||
|
||||
|
||||
from test import mapping_tests
|
||||
|
||||
|
|
|
@ -0,0 +1,2 @@
|
|||
Fix a crash in `os.utime()` in case of a bad ns argument. Patch by Oren
|
||||
Milman.
|
|
@ -4630,6 +4630,12 @@ split_py_long_to_s_and_ns(PyObject *py_long, time_t *s, long *ns)
|
|||
divmod = PyNumber_Divmod(py_long, billion);
|
||||
if (!divmod)
|
||||
goto exit;
|
||||
if (!PyTuple_Check(divmod) || PyTuple_GET_SIZE(divmod) != 2) {
|
||||
PyErr_Format(PyExc_TypeError,
|
||||
"%.200s.__divmod__() must return a 2-tuple, not %.200s",
|
||||
Py_TYPE(py_long)->tp_name, Py_TYPE(divmod)->tp_name);
|
||||
goto exit;
|
||||
}
|
||||
*s = _PyLong_AsTime_t(PyTuple_GET_ITEM(divmod, 0));
|
||||
if ((*s == -1) && PyErr_Occurred())
|
||||
goto exit;
|
||||
|
|
Loading…
Reference in New Issue