mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-04 00:48:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

#1745035: add limits for command and data size to smtpd; patch by Savio Sena.

Browse source
This commit is contained in:
Georg Brandl 2010-12-03 07:38:22 +00:00
parent 106a54d764
commit 1e5c5f8f7d
3 changed files with 45 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

18
Lib/test/test_smtpd.py
View file

@ -121,6 +121,24 @@ class SMTPDChannelTest(TestCase):
self.assertEqual(self.channel.socket.last,
b'451 Internal confusion\r\n')
def test_command_too_long(self):
self.write_line(b'MAIL from ' +
b'a' * self.channel.command_size_limit +
b'@example')
self.assertEqual(self.channel.socket.last,
b'500 Error: line too long\r\n')
def test_data_too_long(self):
# Small hack. Setting limit to 2K octets here will save us some time.
self.channel.data_size_limit = 2048
self.write_line(b'MAIL From:eggs@example')
self.write_line(b'RCPT To:spam@example')
self.write_line(b'DATA')
self.write_line(b'A' * self.channel.data_size_limit +
b'A\r\n.')
self.assertEqual(self.channel.socket.last,
b'552 Error: Too much mail data\r\n')
def test_need_MAIL(self):
self.write_line(b'RCPT to:spam@example')
self.assertEqual(self.channel.socket.last,