diff options
author | Giampaolo Rodola' <g.rodola@gmail.com> | 2011-05-07 17:09:34 (GMT) |
---|---|---|
committer | Giampaolo Rodola' <g.rodola@gmail.com> | 2011-05-07 17:09:34 (GMT) |
commit | 24befa87dcddd95005c82a1c42553856de7dcd08 (patch) | |
tree | 0d45305c10d52775b78d611f0ef897a15e030988 /Lib | |
parent | 4c1aebd88bc71fcbca26279bd21f971f93acb641 (diff) | |
parent | 0b5c21f9c96e6a0734dd7bcbdeec05500a7baf70 (diff) | |
download | cpython-24befa87dcddd95005c82a1c42553856de7dcd08.zip cpython-24befa87dcddd95005c82a1c42553856de7dcd08.tar.gz cpython-24befa87dcddd95005c82a1c42553856de7dcd08.tar.bz2 |
merge with 3.1
Diffstat (limited to 'Lib')
-rw-r--r-- | Lib/ftplib.py | 12 | ||||
-rw-r--r-- | Lib/test/test_ftplib.py | 8 |
2 files changed, 19 insertions, 1 deletions
diff --git a/Lib/ftplib.py b/Lib/ftplib.py index 7c39887..be1797f 100644 --- a/Lib/ftplib.py +++ b/Lib/ftplib.py @@ -241,12 +241,13 @@ class FTP: This does not follow the procedure from the RFC to send Telnet IP and Synch; that doesn't seem to work with the servers I've tried. Instead, just send the ABOR command as OOB data.''' - line = 'ABOR' + CRLF + line = b'ABOR' + B_CRLF if self.debugging > 1: print('*put urgent*', self.sanitize(line)) self.sock.sendall(line, MSG_OOB) resp = self.getmultiline() if resp[:3] not in {'426', '225', '226'}: raise error_proto(resp) + return resp def sendcmd(self, cmd): '''Send a command and return the response.''' @@ -781,6 +782,15 @@ else: conn.close() return self.voidresp() + def abort(self): + # overridden as we can't pass MSG_OOB flag to sendall() + line = b'ABOR' + B_CRLF + self.sock.sendall(line) + resp = self.getmultiline() + if resp[:3] not in {'426', '225', '226'}: + raise error_proto(resp) + return resp + __all__.append('FTP_TLS') all_errors = (Error, IOError, EOFError, ssl.SSLError) diff --git a/Lib/test/test_ftplib.py b/Lib/test/test_ftplib.py index 9d2eab7..5b69b0a 100644 --- a/Lib/test/test_ftplib.py +++ b/Lib/test/test_ftplib.py @@ -61,6 +61,8 @@ class DummyFTPHandler(asynchat.async_chat): def __init__(self, conn): asynchat.async_chat.__init__(self, conn) + # tells the socket to handle urgent data inline (ABOR command) + self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_OOBINLINE, 1) self.set_terminator(b"\r\n") self.in_buffer = [] self.dtp = None @@ -181,6 +183,9 @@ class DummyFTPHandler(asynchat.async_chat): self.push('221 quit ok') self.close() + def cmd_abor(self, arg): + self.push('226 abor ok') + def cmd_stor(self, arg): self.push('125 stor ok') @@ -491,6 +496,9 @@ class TestFTPClass(TestCase): # Ensure the connection gets closed; sock attribute should be None self.assertEqual(self.client.sock, None) + def test_abort(self): + self.client.abort() + def test_retrbinary(self): def callback(data): received.append(data.decode('ascii')) |