diff options
| author | Benjamin Peterson <benjamin@python.org> | 2014-01-14 03:59:38 (GMT) |
|---|---|---|
| committer | Benjamin Peterson <benjamin@python.org> | 2014-01-14 03:59:38 (GMT) |
| commit | 28cf368c1baba3db1f01010e921f63017af74c8f (patch) | |
| tree | 3aa9a09401c90a5be07b28da328d7f5d61c5c5d4 /Lib/test/test_socket.py | |
| parent | aec3065b997e13df7ddaebc39d5be69028535bf8 (diff) | |
| download | cpython-28cf368c1baba3db1f01010e921f63017af74c8f.zip cpython-28cf368c1baba3db1f01010e921f63017af74c8f.tar.gz cpython-28cf368c1baba3db1f01010e921f63017af74c8f.tar.bz2 | |
complain when nbytes > buflen to fix possible buffer overflow (closes #20246)
Diffstat (limited to 'Lib/test/test_socket.py')
| -rw-r--r-- | Lib/test/test_socket.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/Lib/test/test_socket.py b/Lib/test/test_socket.py index 4a60be3..c7ad121 100644 --- a/Lib/test/test_socket.py +++ b/Lib/test/test_socket.py @@ -1620,6 +1620,16 @@ class BufferIOTest(SocketConnectedTest): _testRecvFromIntoMemoryview = _testRecvFromIntoArray + def testRecvFromIntoSmallBuffer(self): + # See issue #20246. + buf = bytearray(8) + self.assertRaises(ValueError, self.cli_conn.recvfrom_into, buf, 1024) + + def _testRecvFromIntoSmallBuffer(self): + with test_support.check_py3k_warnings(): + buf = buffer(MSG*2048) + self.serv_conn.send(buf) + TIPC_STYPE = 2000 TIPC_LOWER = 200 |