diff options
| -rw-r--r-- | Lib/test/test_audioop.py | 4 | ||||
| -rw-r--r-- | Modules/audioop.c | 2 |
2 files changed, 5 insertions, 1 deletions
diff --git a/Lib/test/test_audioop.py b/Lib/test/test_audioop.py index b94be91..8d4f30b 100644 --- a/Lib/test/test_audioop.py +++ b/Lib/test/test_audioop.py @@ -161,6 +161,10 @@ class TestAudioop(unittest.TestCase): self.assertEqual(audioop.getsample(data[1], 2, i), i) self.assertEqual(audioop.getsample(data[2], 4, i), i) + def test_negavitelen(self): + # from issue 3306, previously it segfaulted + self.assertRaises(audioop.error, + audioop.findmax, ''.join( chr(x) for x in xrange(256)), -2392392) def test_main(): run_unittest(TestAudioop) diff --git a/Modules/audioop.c b/Modules/audioop.c index 31e3fa4..767cae6 100644 --- a/Modules/audioop.c +++ b/Modules/audioop.c @@ -575,7 +575,7 @@ audioop_findmax(PyObject *self, PyObject *args) } len1 >>= 1; - if ( len1 < len2 ) { + if ( len2 < 0 || len1 < len2 ) { PyErr_SetString(AudioopError, "Input sample should be longer"); return 0; } |