diff options
author | Benjamin Peterson <benjamin@python.org> | 2010-07-07 22:46:00 +0000 |
---|---|---|
committer | Benjamin Peterson <benjamin@python.org> | 2010-07-07 22:46:00 +0000 |
commit | f092c7c1d7bfcd07dfe621b0b21ab4ece2434c57 (patch) | |
tree | b1cfb4e3de9ce76738cc465af0cf83978d4f9908 | |
parent | ddd46ceeb0b37cd639fa45cfe847fce668806d7a (diff) | |
download | cpython-f092c7c1d7bfcd07dfe621b0b21ab4ece2434c57.tar.gz cpython-f092c7c1d7bfcd07dfe621b0b21ab4ece2434c57.zip |
Merged revisions 82628,82630 via svnmerge from
svn+ssh://pythondev@svn.python.org/python/branches/py3k
........
r82628 | benjamin.peterson | 2010-07-07 13:44:05 -0500 (Wed, 07 Jul 2010) | 1 line
this needn't be in the loop
........
r82630 | benjamin.peterson | 2010-07-07 13:54:59 -0500 (Wed, 07 Jul 2010) | 1 line
don't ignore exceptions from PyObject_IsTrue
........
-rw-r--r-- | Lib/test/test_struct.py | 10 | ||||
-rw-r--r-- | Modules/_struct.c | 10 |
2 files changed, 16 insertions, 4 deletions
diff --git a/Lib/test/test_struct.py b/Lib/test/test_struct.py index 603289511cf..88404629e0c 100644 --- a/Lib/test/test_struct.py +++ b/Lib/test/test_struct.py @@ -480,6 +480,9 @@ class StructTest(unittest.TestCase): self.assertEqual(value, 0x12345678) def test_bool(self): + class ExplodingBool(object): + def __bool__(self): + raise IOError for prefix in tuple("<>!=")+('',): false = (), [], [], '', 0 true = [1], 'test', 5, -1, 0xffffffff+1, 0xffffffff/2 @@ -508,8 +511,11 @@ class StructTest(unittest.TestCase): self.assertFalse(prefix, msg='encoded bool is not one byte: %r' %packed) - for c in [b'\x01', b'\x7f', b'\xff', b'\x0f', b'\xf0']: - self.assertTrue(struct.unpack('>?', c)[0]) + self.assertRaises(IOError, struct.pack, prefix + '?', + ExplodingBool()) + + for c in [b'\x01', b'\x7f', b'\xff', b'\x0f', b'\xf0']: + self.assertTrue(struct.unpack('>?', c)[0]) def test_count_overflow(self): hugecount = '{}b'.format(sys.maxsize+1) diff --git a/Modules/_struct.c b/Modules/_struct.c index f629817dcb2..ba8a8ed979e 100644 --- a/Modules/_struct.c +++ b/Modules/_struct.c @@ -580,9 +580,13 @@ np_ulonglong(char *p, PyObject *v, const formatdef *f) static int np_bool(char *p, PyObject *v, const formatdef *f) { - BOOL_TYPE y; + int y; + BOOL_TYPE x; y = PyObject_IsTrue(v); - memcpy(p, (char *)&y, sizeof y); + if (y < 0) + return -1; + x = y; + memcpy(p, (char *)&x, sizeof x); return 0; } @@ -854,6 +858,8 @@ bp_bool(char *p, PyObject *v, const formatdef *f) { char y; y = PyObject_IsTrue(v); + if (y < 0) + return -1; memcpy(p, (char *)&y, sizeof y); return 0; } |