blob: 05840d492360a0cd5b455095fe9086048e9f4d1a [file] [log] [blame]
Guido van Rossum4b8c6ea2000-02-04 15:39:30 +00001"""An FTP client class and some helper functions.
2
Barry Warsaw100d81e2000-09-01 06:09:23 +00003Based on RFC 959: File Transfer Protocol (FTP), by J. Postel and J. Reynolds
Guido van Rossum1115ab21992-11-04 15:51:30 +00004
Guido van Rossumd2560b01996-05-28 23:41:25 +00005Example:
6
7>>> from ftplib import FTP
8>>> ftp = FTP('ftp.python.org') # connect to host, default port
Guido van Rossum24a64342001-12-28 20:54:55 +00009>>> ftp.login() # default, i.e.: user anonymous, passwd anonymous@
Guido van Rossum2f3941d1997-10-07 14:49:56 +000010'230 Guest login ok, access restrictions apply.'
Guido van Rossumd2560b01996-05-28 23:41:25 +000011>>> ftp.retrlines('LIST') # list directory contents
12total 9
13drwxr-xr-x 8 root wheel 1024 Jan 3 1994 .
14drwxr-xr-x 8 root wheel 1024 Jan 3 1994 ..
15drwxr-xr-x 2 root wheel 1024 Jan 3 1994 bin
16drwxr-xr-x 2 root wheel 1024 Jan 3 1994 etc
17d-wxrwxr-x 2 ftp wheel 1024 Sep 5 13:43 incoming
18drwxr-xr-x 2 root wheel 1024 Nov 17 1993 lib
19drwxr-xr-x 6 1094 wheel 1024 Sep 13 19:07 pub
20drwxr-xr-x 3 root wheel 1024 Jan 3 1994 usr
21-rw-r--r-- 1 root root 312 Aug 1 1994 welcome.msg
Guido van Rossum2f3941d1997-10-07 14:49:56 +000022'226 Transfer complete.'
Guido van Rossumd2560b01996-05-28 23:41:25 +000023>>> ftp.quit()
Guido van Rossum2f3941d1997-10-07 14:49:56 +000024'221 Goodbye.'
Tim Peters88869f92001-01-14 23:36:06 +000025>>>
Guido van Rossumd2560b01996-05-28 23:41:25 +000026
27A nice test that reveals some of the network dialogue would be:
28python ftplib.py -d localhost -l -p -l
Guido van Rossum4b8c6ea2000-02-04 15:39:30 +000029"""
Guido van Rossumc567c601992-11-05 22:22:37 +000030
Tim Peters88869f92001-01-14 23:36:06 +000031#
Guido van Rossum98d9fd32000-02-28 15:12:25 +000032# Changes and improvements suggested by Steve Majewski.
33# Modified by Jack to work on the mac.
34# Modified by Siebren to support docstrings and PASV.
Christian Heimes1af737c2008-01-23 08:24:23 +000035# Modified by Phil Schwartz to add storbinary and storlines callbacks.
Antoine Pitrouf988cd02009-11-17 20:21:14 +000036# Modified by Giampaolo Rodola' to add TLS support.
Guido van Rossum98d9fd32000-02-28 15:12:25 +000037#
Guido van Rossumc567c601992-11-05 22:22:37 +000038
Guido van Rossum1115ab21992-11-04 15:51:30 +000039import sys
Giampaolo Rodolà6545ed62010-08-21 20:34:37 +000040import socket
Georg Brandlf78e02b2008-06-10 17:40:04 +000041from socket import _GLOBAL_DEFAULT_TIMEOUT
Guido van Rossumb6775db1994-08-01 11:34:53 +000042
Martin Panter19e69c52015-11-14 12:46:42 +000043__all__ = ["FTP", "error_reply", "error_temp", "error_perm", "error_proto",
44 "all_errors"]
Guido van Rossum1115ab21992-11-04 15:51:30 +000045
Guido van Rossumd3166071993-05-24 14:16:22 +000046# Magic number from <socket.h>
Tim Peters88869f92001-01-14 23:36:06 +000047MSG_OOB = 0x1 # Process data out of band
Guido van Rossumd3166071993-05-24 14:16:22 +000048
49
Guido van Rossumc567c601992-11-05 22:22:37 +000050# The standard FTP server control port
Guido van Rossum1115ab21992-11-04 15:51:30 +000051FTP_PORT = 21
Serhiy Storchakac30b1782013-10-20 16:58:27 +030052# The sizehint parameter passed to readline() calls
53MAXLINE = 8192
Guido van Rossum1115ab21992-11-04 15:51:30 +000054
55
Guido van Rossum21974791992-11-06 13:34:17 +000056# Exception raised when an error or invalid response is received
Fred Drake227b1202000-08-17 05:06:49 +000057class Error(Exception): pass
Tim Peters88869f92001-01-14 23:36:06 +000058class error_reply(Error): pass # unexpected [123]xx reply
59class error_temp(Error): pass # 4xx errors
60class error_perm(Error): pass # 5xx errors
61class error_proto(Error): pass # response does not begin with [1-5]
Guido van Rossum1115ab21992-11-04 15:51:30 +000062
63
Guido van Rossum21974791992-11-06 13:34:17 +000064# All exceptions (hopefully) that may be raised here and that aren't
65# (always) programming errors on our side
Andrew Svetlovf7a17b42012-12-25 16:47:37 +020066all_errors = (Error, OSError, EOFError)
Guido van Rossum21974791992-11-06 13:34:17 +000067
68
Guido van Rossum1115ab21992-11-04 15:51:30 +000069# Line terminators (we always output CRLF, but accept any of CRLF, CR, LF)
70CRLF = '\r\n'
Benjamin Petersonbe17a112008-09-27 21:49:47 +000071B_CRLF = b'\r\n'
Guido van Rossum1115ab21992-11-04 15:51:30 +000072
Guido van Rossum1115ab21992-11-04 15:51:30 +000073# The class itself
74class FTP:
75
Tim Peters88869f92001-01-14 23:36:06 +000076 '''An FTP client class.
Guido van Rossumd2560b01996-05-28 23:41:25 +000077
Guido van Rossumd8faa362007-04-27 19:54:29 +000078 To create a connection, call the class using these arguments:
79 host, user, passwd, acct, timeout
80
81 The first four arguments are all strings, and have default value ''.
82 timeout must be numeric and defaults to None if not passed,
83 meaning that no timeout will be set on any ftp socket(s)
84 If a timeout is passed, then this is now the default timeout for all ftp
85 socket operations for this instance.
86
Tim Peters88869f92001-01-14 23:36:06 +000087 Then use self.connect() with optional host and port argument.
Guido van Rossumd2560b01996-05-28 23:41:25 +000088
Tim Peters88869f92001-01-14 23:36:06 +000089 To download a file, use ftp.retrlines('RETR ' + filename),
90 or ftp.retrbinary() with slightly different arguments.
91 To upload a file, use ftp.storlines() or ftp.storbinary(),
92 which have an open file as argument (see their definitions
93 below for details).
94 The download/upload functions first issue appropriate TYPE
95 and PORT or PASV commands.
Guido van Rossum52b89762007-07-17 20:45:57 +000096 '''
Guido van Rossumd2560b01996-05-28 23:41:25 +000097
Fred Drake9c98a422001-02-28 21:46:37 +000098 debugging = 0
99 host = ''
100 port = FTP_PORT
Serhiy Storchakac30b1782013-10-20 16:58:27 +0300101 maxline = MAXLINE
Fred Drake9c98a422001-02-28 21:46:37 +0000102 sock = None
103 file = None
104 welcome = None
105 passiveserver = 1
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000106 encoding = "latin-1"
Fred Drake9c98a422001-02-28 21:46:37 +0000107
Tim Peters88869f92001-01-14 23:36:06 +0000108 # Initialization method (called by class instantiation).
109 # Initialize host to localhost, port to standard ftp port
110 # Optional arguments are host (for connect()),
111 # and user, passwd, acct (for login())
Georg Brandlf78e02b2008-06-10 17:40:04 +0000112 def __init__(self, host='', user='', passwd='', acct='',
Giampaolo Rodolà396ff062011-02-28 19:19:51 +0000113 timeout=_GLOBAL_DEFAULT_TIMEOUT, source_address=None):
114 self.source_address = source_address
Guido van Rossumd8faa362007-04-27 19:54:29 +0000115 self.timeout = timeout
Tim Peters88869f92001-01-14 23:36:06 +0000116 if host:
Fred Drake9c98a422001-02-28 21:46:37 +0000117 self.connect(host)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000118 if user:
119 self.login(user, passwd, acct)
Guido van Rossum52fc1f61993-06-17 12:38:10 +0000120
Giampaolo Rodolàbd576b72010-05-10 14:53:29 +0000121 def __enter__(self):
122 return self
123
124 # Context management protocol: try to quit() if active
125 def __exit__(self, *args):
126 if self.sock is not None:
127 try:
128 self.quit()
Andrew Svetlov0832af62012-12-18 23:10:48 +0200129 except (OSError, EOFError):
Giampaolo Rodolàbd576b72010-05-10 14:53:29 +0000130 pass
131 finally:
132 if self.sock is not None:
133 self.close()
134
Giampaolo Rodolà396ff062011-02-28 19:19:51 +0000135 def connect(self, host='', port=0, timeout=-999, source_address=None):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000136 '''Connect to host. Arguments are:
Guido van Rossumd8faa362007-04-27 19:54:29 +0000137 - host: hostname to connect to (string, default previous host)
138 - port: port to connect to (integer, default previous port)
Giampaolo Rodola'11a9bd62013-04-26 15:23:43 +0200139 - timeout: the timeout to set against the ftp socket(s)
Giampaolo Rodolà396ff062011-02-28 19:19:51 +0000140 - source_address: a 2-tuple (host, port) for the socket to bind
141 to as its source address before connecting.
Guido van Rossumd8faa362007-04-27 19:54:29 +0000142 '''
143 if host != '':
144 self.host = host
145 if port > 0:
146 self.port = port
Georg Brandlf78e02b2008-06-10 17:40:04 +0000147 if timeout != -999:
Guido van Rossumd8faa362007-04-27 19:54:29 +0000148 self.timeout = timeout
Giampaolo Rodolà396ff062011-02-28 19:19:51 +0000149 if source_address is not None:
150 self.source_address = source_address
151 self.sock = socket.create_connection((self.host, self.port), self.timeout,
152 source_address=self.source_address)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000153 self.af = self.sock.family
Guido van Rossum52b89762007-07-17 20:45:57 +0000154 self.file = self.sock.makefile('r', encoding=self.encoding)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000155 self.welcome = self.getresp()
156 return self.welcome
Guido van Rossum1115ab21992-11-04 15:51:30 +0000157
Tim Peters88869f92001-01-14 23:36:06 +0000158 def getwelcome(self):
159 '''Get the welcome message from the server.
160 (this is read and squirreled away by connect())'''
161 if self.debugging:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000162 print('*welcome*', self.sanitize(self.welcome))
Tim Peters88869f92001-01-14 23:36:06 +0000163 return self.welcome
Guido van Rossum1115ab21992-11-04 15:51:30 +0000164
Tim Peters88869f92001-01-14 23:36:06 +0000165 def set_debuglevel(self, level):
166 '''Set the debugging level.
167 The required argument level means:
168 0: no debugging output (default)
169 1: print commands and responses but not body text etc.
170 2: also print raw lines read and sent before stripping CR/LF'''
171 self.debugging = level
172 debug = set_debuglevel
Guido van Rossum1115ab21992-11-04 15:51:30 +0000173
Tim Peters88869f92001-01-14 23:36:06 +0000174 def set_pasv(self, val):
175 '''Use passive or active mode for data transfers.
176 With a false argument, use the normal PORT mode,
177 With a true argument, use the PASV command.'''
178 self.passiveserver = val
Guido van Rossumd2560b01996-05-28 23:41:25 +0000179
Tim Peters88869f92001-01-14 23:36:06 +0000180 # Internal: "sanitize" a string for printing
181 def sanitize(self, s):
Florent Xicluna2bb96f52011-10-23 22:11:00 +0200182 if s[:5] in {'pass ', 'PASS '}:
183 i = len(s.rstrip('\r\n'))
Tim Peters88869f92001-01-14 23:36:06 +0000184 s = s[:5] + '*'*(i-5) + s[i:]
Walter Dörwald70a6b492004-02-12 17:35:32 +0000185 return repr(s)
Guido van Rossumebaf1041995-05-05 15:54:14 +0000186
Tim Peters88869f92001-01-14 23:36:06 +0000187 # Internal: send one line to the server, appending CRLF
188 def putline(self, line):
Dong-hee Na2b1e6e92017-07-23 02:20:22 +0900189 if '\r' in line or '\n' in line:
190 raise ValueError('an illegal newline character should not be contained')
Tim Peters88869f92001-01-14 23:36:06 +0000191 line = line + CRLF
Giampaolo Rodola'94c56a12013-03-07 13:51:20 +0100192 if self.debugging > 1:
193 print('*put*', self.sanitize(line))
Guido van Rossum52b89762007-07-17 20:45:57 +0000194 self.sock.sendall(line.encode(self.encoding))
Guido van Rossum1115ab21992-11-04 15:51:30 +0000195
Tim Peters88869f92001-01-14 23:36:06 +0000196 # Internal: send one command to the server (through putline())
197 def putcmd(self, line):
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000198 if self.debugging: print('*cmd*', self.sanitize(line))
Tim Peters88869f92001-01-14 23:36:06 +0000199 self.putline(line)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000200
Tim Peters88869f92001-01-14 23:36:06 +0000201 # Internal: return one line from the server, stripping CRLF.
202 # Raise EOFError if the connection is closed
203 def getline(self):
Serhiy Storchakac30b1782013-10-20 16:58:27 +0300204 line = self.file.readline(self.maxline + 1)
205 if len(line) > self.maxline:
206 raise Error("got more than %d bytes" % self.maxline)
Tim Peters88869f92001-01-14 23:36:06 +0000207 if self.debugging > 1:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000208 print('*get*', self.sanitize(line))
Giampaolo Rodola'94c56a12013-03-07 13:51:20 +0100209 if not line:
210 raise EOFError
211 if line[-2:] == CRLF:
212 line = line[:-2]
213 elif line[-1:] in CRLF:
214 line = line[:-1]
Tim Peters88869f92001-01-14 23:36:06 +0000215 return line
Guido van Rossum1115ab21992-11-04 15:51:30 +0000216
Tim Peters88869f92001-01-14 23:36:06 +0000217 # Internal: get a response from the server, which may possibly
218 # consist of multiple lines. Return a single string with no
219 # trailing CRLF. If the response consists of multiple lines,
220 # these are separated by '\n' characters in the string
221 def getmultiline(self):
222 line = self.getline()
223 if line[3:4] == '-':
224 code = line[:3]
225 while 1:
226 nextline = self.getline()
227 line = line + ('\n' + nextline)
228 if nextline[:3] == code and \
229 nextline[3:4] != '-':
230 break
231 return line
Guido van Rossum1115ab21992-11-04 15:51:30 +0000232
Tim Peters88869f92001-01-14 23:36:06 +0000233 # Internal: get a response from the server.
234 # Raise various errors if the response indicates an error
235 def getresp(self):
236 resp = self.getmultiline()
Giampaolo Rodola'94c56a12013-03-07 13:51:20 +0100237 if self.debugging:
238 print('*resp*', self.sanitize(resp))
Tim Peters88869f92001-01-14 23:36:06 +0000239 self.lastresp = resp[:3]
240 c = resp[:1]
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000241 if c in {'1', '2', '3'}:
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000242 return resp
Tim Peters88869f92001-01-14 23:36:06 +0000243 if c == '4':
Collin Winterce36ad82007-08-30 01:19:48 +0000244 raise error_temp(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000245 if c == '5':
Collin Winterce36ad82007-08-30 01:19:48 +0000246 raise error_perm(resp)
247 raise error_proto(resp)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000248
Tim Peters88869f92001-01-14 23:36:06 +0000249 def voidresp(self):
250 """Expect a response beginning with '2'."""
251 resp = self.getresp()
Benjamin Petersond23f8222009-04-05 19:13:16 +0000252 if resp[:1] != '2':
Collin Winterce36ad82007-08-30 01:19:48 +0000253 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000254 return resp
Guido van Rossumc567c601992-11-05 22:22:37 +0000255
Tim Peters88869f92001-01-14 23:36:06 +0000256 def abort(self):
257 '''Abort a file transfer. Uses out-of-band data.
258 This does not follow the procedure from the RFC to send Telnet
259 IP and Synch; that doesn't seem to work with the servers I've
260 tried. Instead, just send the ABOR command as OOB data.'''
Giampaolo Rodola'0b5c21f2011-05-07 19:03:47 +0200261 line = b'ABOR' + B_CRLF
Giampaolo Rodola'94c56a12013-03-07 13:51:20 +0100262 if self.debugging > 1:
263 print('*put urgent*', self.sanitize(line))
Martin v. Löwise12454f2002-02-16 23:06:19 +0000264 self.sock.sendall(line, MSG_OOB)
Tim Peters88869f92001-01-14 23:36:06 +0000265 resp = self.getmultiline()
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000266 if resp[:3] not in {'426', '225', '226'}:
Collin Winterce36ad82007-08-30 01:19:48 +0000267 raise error_proto(resp)
Giampaolo Rodola'0b5c21f2011-05-07 19:03:47 +0200268 return resp
Guido van Rossumd3166071993-05-24 14:16:22 +0000269
Tim Peters88869f92001-01-14 23:36:06 +0000270 def sendcmd(self, cmd):
271 '''Send a command and return the response.'''
272 self.putcmd(cmd)
273 return self.getresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000274
Tim Peters88869f92001-01-14 23:36:06 +0000275 def voidcmd(self, cmd):
276 """Send a command and expect a response beginning with '2'."""
277 self.putcmd(cmd)
278 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000279
Tim Peters88869f92001-01-14 23:36:06 +0000280 def sendport(self, host, port):
281 '''Send a PORT command with the current host and the given
282 port number.
283 '''
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000284 hbytes = host.split('.')
Benjamin Peterson3a53fbb2008-09-27 22:04:16 +0000285 pbytes = [repr(port//256), repr(port%256)]
Tim Peters88869f92001-01-14 23:36:06 +0000286 bytes = hbytes + pbytes
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000287 cmd = 'PORT ' + ','.join(bytes)
Tim Peters88869f92001-01-14 23:36:06 +0000288 return self.voidcmd(cmd)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000289
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000290 def sendeprt(self, host, port):
Martin Panter7462b6492015-11-02 03:37:02 +0000291 '''Send an EPRT command with the current host and the given port number.'''
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000292 af = 0
293 if self.af == socket.AF_INET:
294 af = 1
295 if self.af == socket.AF_INET6:
296 af = 2
297 if af == 0:
Collin Winterce36ad82007-08-30 01:19:48 +0000298 raise error_proto('unsupported address family')
Walter Dörwald70a6b492004-02-12 17:35:32 +0000299 fields = ['', repr(af), host, repr(port), '']
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000300 cmd = 'EPRT ' + '|'.join(fields)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000301 return self.voidcmd(cmd)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000302
Tim Peters88869f92001-01-14 23:36:06 +0000303 def makeport(self):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000304 '''Create a new socket and send a PORT command for it.'''
Giampaolo Rodola'fc327882012-12-17 14:20:27 +0100305 err = None
Martin v. Löwis322c0d12001-10-07 08:53:32 +0000306 sock = None
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000307 for res in socket.getaddrinfo(None, 0, self.af, socket.SOCK_STREAM, 0, socket.AI_PASSIVE):
308 af, socktype, proto, canonname, sa = res
309 try:
310 sock = socket.socket(af, socktype, proto)
311 sock.bind(sa)
Andrew Svetlov0832af62012-12-18 23:10:48 +0200312 except OSError as _:
Giampaolo Rodola'57f34872012-12-17 20:46:16 +0100313 err = _
Martin v. Löwis322c0d12001-10-07 08:53:32 +0000314 if sock:
315 sock.close()
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000316 sock = None
317 continue
318 break
Giampaolo Rodola'fc327882012-12-17 14:20:27 +0100319 if sock is None:
320 if err is not None:
321 raise err
322 else:
Andrew Svetlov0832af62012-12-18 23:10:48 +0200323 raise OSError("getaddrinfo returns an empty list")
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000324 sock.listen(1)
325 port = sock.getsockname()[1] # Get proper port
326 host = self.sock.getsockname()[0] # Get proper host
327 if self.af == socket.AF_INET:
328 resp = self.sendport(host, port)
329 else:
330 resp = self.sendeprt(host, port)
Giampaolo Rodolà5fb313b2010-04-19 22:05:54 +0000331 if self.timeout is not _GLOBAL_DEFAULT_TIMEOUT:
332 sock.settimeout(self.timeout)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000333 return sock
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000334
335 def makepasv(self):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000336 if self.af == socket.AF_INET:
337 host, port = parse227(self.sendcmd('PASV'))
338 else:
339 host, port = parse229(self.sendcmd('EPSV'), self.sock.getpeername())
340 return host, port
Guido van Rossum1115ab21992-11-04 15:51:30 +0000341
Tim Peters88869f92001-01-14 23:36:06 +0000342 def ntransfercmd(self, cmd, rest=None):
343 """Initiate a transfer over the data connection.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000344
Tim Peters88869f92001-01-14 23:36:06 +0000345 If the transfer is active, send a port command and the
346 transfer command, and accept the connection. If the server is
347 passive, send a pasv command, connect to it, and start the
348 transfer command. Either way, return the socket for the
349 connection and the expected size of the transfer. The
350 expected size may be None if it could not be determined.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000351
Tim Peters88869f92001-01-14 23:36:06 +0000352 Optional `rest' argument can be a string that is sent as the
Christian Heimes1af737c2008-01-23 08:24:23 +0000353 argument to a REST command. This is essentially a server
Tim Peters88869f92001-01-14 23:36:06 +0000354 marker used to tell the server to skip over any data up to the
355 given marker.
356 """
357 size = None
358 if self.passiveserver:
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000359 host, port = self.makepasv()
Giampaolo Rodolà396ff062011-02-28 19:19:51 +0000360 conn = socket.create_connection((host, port), self.timeout,
361 source_address=self.source_address)
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +0200362 try:
363 if rest is not None:
364 self.sendcmd("REST %s" % rest)
365 resp = self.sendcmd(cmd)
366 # Some servers apparently send a 200 reply to
367 # a LIST or STOR command, before the 150 reply
368 # (and way before the 226 reply). This seems to
369 # be in violation of the protocol (which only allows
370 # 1xx or error messages for LIST), so we just discard
371 # this response.
372 if resp[0] == '2':
373 resp = self.getresp()
374 if resp[0] != '1':
375 raise error_reply(resp)
376 except:
377 conn.close()
378 raise
Tim Peters88869f92001-01-14 23:36:06 +0000379 else:
Giampaolo Rodola'836e9aa2011-12-10 21:25:04 +0100380 with self.makeport() as sock:
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +0200381 if rest is not None:
382 self.sendcmd("REST %s" % rest)
383 resp = self.sendcmd(cmd)
384 # See above.
385 if resp[0] == '2':
386 resp = self.getresp()
387 if resp[0] != '1':
388 raise error_reply(resp)
389 conn, sockaddr = sock.accept()
390 if self.timeout is not _GLOBAL_DEFAULT_TIMEOUT:
391 conn.settimeout(self.timeout)
Tim Peters88869f92001-01-14 23:36:06 +0000392 if resp[:3] == '150':
393 # this is conditional in case we received a 125
394 size = parse150(resp)
395 return conn, size
Fred Drake4de02d91997-01-10 18:26:09 +0000396
Tim Peters88869f92001-01-14 23:36:06 +0000397 def transfercmd(self, cmd, rest=None):
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000398 """Like ntransfercmd() but returns only the socket."""
Tim Peters88869f92001-01-14 23:36:06 +0000399 return self.ntransfercmd(cmd, rest)[0]
Guido van Rossumc567c601992-11-05 22:22:37 +0000400
Tim Peters88869f92001-01-14 23:36:06 +0000401 def login(self, user = '', passwd = '', acct = ''):
402 '''Login, default anonymous.'''
Giampaolo Rodola'94c56a12013-03-07 13:51:20 +0100403 if not user:
404 user = 'anonymous'
405 if not passwd:
406 passwd = ''
407 if not acct:
408 acct = ''
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000409 if user == 'anonymous' and passwd in {'', '-'}:
Tim Peterse4418602002-02-16 07:34:19 +0000410 # If there is no anonymous ftp password specified
411 # then we'll just use anonymous@
412 # We don't send any other thing because:
413 # - We want to remain anonymous
414 # - We want to stop SPAM
415 # - We don't want to let ftp sites to discriminate by the user,
416 # host or country.
Guido van Rossumc33e0772001-12-28 20:54:28 +0000417 passwd = passwd + 'anonymous@'
Tim Peters88869f92001-01-14 23:36:06 +0000418 resp = self.sendcmd('USER ' + user)
Giampaolo Rodola'94c56a12013-03-07 13:51:20 +0100419 if resp[0] == '3':
420 resp = self.sendcmd('PASS ' + passwd)
421 if resp[0] == '3':
422 resp = self.sendcmd('ACCT ' + acct)
Tim Peters88869f92001-01-14 23:36:06 +0000423 if resp[0] != '2':
Collin Winterce36ad82007-08-30 01:19:48 +0000424 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000425 return resp
Guido van Rossumc567c601992-11-05 22:22:37 +0000426
Tim Peters88869f92001-01-14 23:36:06 +0000427 def retrbinary(self, cmd, callback, blocksize=8192, rest=None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000428 """Retrieve data in binary mode. A new port is created for you.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000429
Christian Heimes1af737c2008-01-23 08:24:23 +0000430 Args:
431 cmd: A RETR command.
432 callback: A single parameter callable to be called on each
433 block of data read.
434 blocksize: The maximum number of bytes to read from the
435 socket at one time. [default: 8192]
436 rest: Passed to transfercmd(). [default: None]
Guido van Rossum1115ab21992-11-04 15:51:30 +0000437
Christian Heimes1af737c2008-01-23 08:24:23 +0000438 Returns:
439 The response code.
Tim Peters88869f92001-01-14 23:36:06 +0000440 """
441 self.voidcmd('TYPE I')
Benjamin Petersona8071372010-10-31 18:21:16 +0000442 with self.transfercmd(cmd, rest) as conn:
443 while 1:
444 data = conn.recv(blocksize)
445 if not data:
446 break
447 callback(data)
Giampaolo Rodola'3ff5e752013-03-07 14:03:27 +0100448 # shutdown ssl layer
Giampaolo Rodola'a77d7a52013-03-18 09:59:15 +0100449 if _SSLSocket is not None and isinstance(conn, _SSLSocket):
Giampaolo Rodola'3ff5e752013-03-07 14:03:27 +0100450 conn.unwrap()
Tim Peters88869f92001-01-14 23:36:06 +0000451 return self.voidresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000452
Tim Peters88869f92001-01-14 23:36:06 +0000453 def retrlines(self, cmd, callback = None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000454 """Retrieve data in line mode. A new port is created for you.
455
456 Args:
Giampaolo Rodola'd78def92011-05-06 19:49:08 +0200457 cmd: A RETR, LIST, or NLST command.
Christian Heimes1af737c2008-01-23 08:24:23 +0000458 callback: An optional single parameter callable that is called
459 for each line with the trailing CRLF stripped.
460 [default: print_line()]
461
462 Returns:
463 The response code.
464 """
Giampaolo Rodola'94c56a12013-03-07 13:51:20 +0100465 if callback is None:
466 callback = print_line
Tim Peters88869f92001-01-14 23:36:06 +0000467 resp = self.sendcmd('TYPE A')
Benjamin Petersona8071372010-10-31 18:21:16 +0000468 with self.transfercmd(cmd) as conn, \
469 conn.makefile('r', encoding=self.encoding) as fp:
470 while 1:
Serhiy Storchakac30b1782013-10-20 16:58:27 +0300471 line = fp.readline(self.maxline + 1)
472 if len(line) > self.maxline:
473 raise Error("got more than %d bytes" % self.maxline)
Giampaolo Rodola'94c56a12013-03-07 13:51:20 +0100474 if self.debugging > 2:
475 print('*retr*', repr(line))
Benjamin Petersona8071372010-10-31 18:21:16 +0000476 if not line:
477 break
478 if line[-2:] == CRLF:
479 line = line[:-2]
480 elif line[-1:] == '\n':
481 line = line[:-1]
482 callback(line)
Giampaolo Rodola'3ff5e752013-03-07 14:03:27 +0100483 # shutdown ssl layer
Giampaolo Rodola'a77d7a52013-03-18 09:59:15 +0100484 if _SSLSocket is not None and isinstance(conn, _SSLSocket):
Giampaolo Rodola'3ff5e752013-03-07 14:03:27 +0100485 conn.unwrap()
Tim Peters88869f92001-01-14 23:36:06 +0000486 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000487
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000488 def storbinary(self, cmd, fp, blocksize=8192, callback=None, rest=None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000489 """Store a file in binary mode. A new port is created for you.
490
491 Args:
492 cmd: A STOR command.
493 fp: A file-like object with a read(num_bytes) method.
494 blocksize: The maximum data size to read from fp and send over
495 the connection at once. [default: 8192]
496 callback: An optional single parameter callable that is called on
Terry Jan Reedyc30b7b12013-03-11 17:57:08 -0400497 each block of data after it is sent. [default: None]
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000498 rest: Passed to transfercmd(). [default: None]
Christian Heimes1af737c2008-01-23 08:24:23 +0000499
500 Returns:
501 The response code.
502 """
Tim Peters88869f92001-01-14 23:36:06 +0000503 self.voidcmd('TYPE I')
Benjamin Petersona8071372010-10-31 18:21:16 +0000504 with self.transfercmd(cmd, rest) as conn:
505 while 1:
506 buf = fp.read(blocksize)
Giampaolo Rodola'94c56a12013-03-07 13:51:20 +0100507 if not buf:
508 break
Benjamin Petersona8071372010-10-31 18:21:16 +0000509 conn.sendall(buf)
Giampaolo Rodola'94c56a12013-03-07 13:51:20 +0100510 if callback:
511 callback(buf)
Giampaolo Rodola'3ff5e752013-03-07 14:03:27 +0100512 # shutdown ssl layer
Giampaolo Rodola'a77d7a52013-03-18 09:59:15 +0100513 if _SSLSocket is not None and isinstance(conn, _SSLSocket):
Giampaolo Rodola'3ff5e752013-03-07 14:03:27 +0100514 conn.unwrap()
Tim Peters88869f92001-01-14 23:36:06 +0000515 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000516
Christian Heimes1af737c2008-01-23 08:24:23 +0000517 def storlines(self, cmd, fp, callback=None):
518 """Store a file in line mode. A new port is created for you.
519
520 Args:
521 cmd: A STOR command.
522 fp: A file-like object with a readline() method.
523 callback: An optional single parameter callable that is called on
Terry Jan Reedyc30b7b12013-03-11 17:57:08 -0400524 each line after it is sent. [default: None]
Christian Heimes1af737c2008-01-23 08:24:23 +0000525
526 Returns:
527 The response code.
528 """
Tim Peters88869f92001-01-14 23:36:06 +0000529 self.voidcmd('TYPE A')
Benjamin Petersona8071372010-10-31 18:21:16 +0000530 with self.transfercmd(cmd) as conn:
531 while 1:
Serhiy Storchakac30b1782013-10-20 16:58:27 +0300532 buf = fp.readline(self.maxline + 1)
533 if len(buf) > self.maxline:
534 raise Error("got more than %d bytes" % self.maxline)
Giampaolo Rodola'94c56a12013-03-07 13:51:20 +0100535 if not buf:
536 break
Benjamin Petersona8071372010-10-31 18:21:16 +0000537 if buf[-2:] != B_CRLF:
538 if buf[-1] in B_CRLF: buf = buf[:-1]
539 buf = buf + B_CRLF
540 conn.sendall(buf)
Giampaolo Rodola'94c56a12013-03-07 13:51:20 +0100541 if callback:
542 callback(buf)
Giampaolo Rodola'3ff5e752013-03-07 14:03:27 +0100543 # shutdown ssl layer
Giampaolo Rodola'a77d7a52013-03-18 09:59:15 +0100544 if _SSLSocket is not None and isinstance(conn, _SSLSocket):
Giampaolo Rodola'3ff5e752013-03-07 14:03:27 +0100545 conn.unwrap()
Tim Peters88869f92001-01-14 23:36:06 +0000546 return self.voidresp()
Guido van Rossum0eaa74b1996-01-25 18:37:21 +0000547
Tim Peters88869f92001-01-14 23:36:06 +0000548 def acct(self, password):
549 '''Send new account name.'''
550 cmd = 'ACCT ' + password
551 return self.voidcmd(cmd)
Guido van Rossumc567c601992-11-05 22:22:37 +0000552
Tim Peters88869f92001-01-14 23:36:06 +0000553 def nlst(self, *args):
554 '''Return a list of files in a given directory (default the current).'''
555 cmd = 'NLST'
556 for arg in args:
557 cmd = cmd + (' ' + arg)
558 files = []
559 self.retrlines(cmd, files.append)
560 return files
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000561
Tim Peters88869f92001-01-14 23:36:06 +0000562 def dir(self, *args):
563 '''List a directory in long form.
564 By default list current directory to stdout.
565 Optional last argument is callback function; all
566 non-empty arguments before it are concatenated to the
567 LIST command. (This *should* only be used for a pathname.)'''
568 cmd = 'LIST'
569 func = None
570 if args[-1:] and type(args[-1]) != type(''):
571 args, func = args[:-1], args[-1]
572 for arg in args:
573 if arg:
574 cmd = cmd + (' ' + arg)
575 self.retrlines(cmd, func)
Guido van Rossumc567c601992-11-05 22:22:37 +0000576
Giampaolo Rodola'd78def92011-05-06 19:49:08 +0200577 def mlsd(self, path="", facts=[]):
578 '''List a directory in a standardized format by using MLSD
579 command (RFC-3659). If path is omitted the current directory
580 is assumed. "facts" is a list of strings representing the type
581 of information desired (e.g. ["type", "size", "perm"]).
582
583 Return a generator object yielding a tuple of two elements
584 for every file found in path.
585 First element is the file name, the second one is a dictionary
586 including a variable number of "facts" depending on the server
587 and whether "facts" argument has been provided.
588 '''
589 if facts:
590 self.sendcmd("OPTS MLST " + ";".join(facts) + ";")
591 if path:
592 cmd = "MLSD %s" % path
593 else:
594 cmd = "MLSD"
595 lines = []
596 self.retrlines(cmd, lines.append)
597 for line in lines:
598 facts_found, _, name = line.rstrip(CRLF).partition(' ')
599 entry = {}
600 for fact in facts_found[:-1].split(";"):
601 key, _, value = fact.partition("=")
602 entry[key.lower()] = value
603 yield (name, entry)
604
Tim Peters88869f92001-01-14 23:36:06 +0000605 def rename(self, fromname, toname):
606 '''Rename a file.'''
607 resp = self.sendcmd('RNFR ' + fromname)
608 if resp[0] != '3':
Collin Winterce36ad82007-08-30 01:19:48 +0000609 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000610 return self.voidcmd('RNTO ' + toname)
Guido van Rossuma61bdeb1995-10-11 17:36:31 +0000611
Tim Peters88869f92001-01-14 23:36:06 +0000612 def delete(self, filename):
613 '''Delete a file.'''
614 resp = self.sendcmd('DELE ' + filename)
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000615 if resp[:3] in {'250', '200'}:
Tim Peters88869f92001-01-14 23:36:06 +0000616 return resp
Tim Peters88869f92001-01-14 23:36:06 +0000617 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000618 raise error_reply(resp)
Guido van Rossum02cf5821993-05-17 08:00:02 +0000619
Tim Peters88869f92001-01-14 23:36:06 +0000620 def cwd(self, dirname):
621 '''Change to a directory.'''
622 if dirname == '..':
623 try:
624 return self.voidcmd('CDUP')
Guido van Rossumb940e112007-01-10 16:19:56 +0000625 except error_perm as msg:
Martin v. Löwisb5255112002-03-10 15:59:58 +0000626 if msg.args[0][:3] != '500':
627 raise
Tim Peters88869f92001-01-14 23:36:06 +0000628 elif dirname == '':
629 dirname = '.' # does nothing, but could return error
630 cmd = 'CWD ' + dirname
631 return self.voidcmd(cmd)
Guido van Rossum17ed1ae1993-06-01 13:21:04 +0000632
Tim Peters88869f92001-01-14 23:36:06 +0000633 def size(self, filename):
634 '''Retrieve the size of a file.'''
Christian Heimes1af737c2008-01-23 08:24:23 +0000635 # The SIZE command is defined in RFC-3659
Tim Peters88869f92001-01-14 23:36:06 +0000636 resp = self.sendcmd('SIZE ' + filename)
637 if resp[:3] == '213':
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000638 s = resp[3:].strip()
Florent Xicluna2bb96f52011-10-23 22:11:00 +0200639 return int(s)
Guido van Rossumc567c601992-11-05 22:22:37 +0000640
Tim Peters88869f92001-01-14 23:36:06 +0000641 def mkd(self, dirname):
642 '''Make a directory, return its full pathname.'''
Giampaolo Rodolàbbc47822010-08-23 22:10:32 +0000643 resp = self.voidcmd('MKD ' + dirname)
644 # fix around non-compliant implementations such as IIS shipped
645 # with Windows server 2003
646 if not resp.startswith('257'):
647 return ''
Tim Peters88869f92001-01-14 23:36:06 +0000648 return parse257(resp)
Guido van Rossum98245091998-02-19 21:15:44 +0000649
Tim Peters88869f92001-01-14 23:36:06 +0000650 def rmd(self, dirname):
651 '''Remove a directory.'''
652 return self.voidcmd('RMD ' + dirname)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000653
Tim Peters88869f92001-01-14 23:36:06 +0000654 def pwd(self):
655 '''Return current working directory.'''
Giampaolo Rodolàbbc47822010-08-23 22:10:32 +0000656 resp = self.voidcmd('PWD')
657 # fix around non-compliant implementations such as IIS shipped
658 # with Windows server 2003
659 if not resp.startswith('257'):
660 return ''
Tim Peters88869f92001-01-14 23:36:06 +0000661 return parse257(resp)
Guido van Rossum17ed1ae1993-06-01 13:21:04 +0000662
Tim Peters88869f92001-01-14 23:36:06 +0000663 def quit(self):
664 '''Quit, and close the connection.'''
665 resp = self.voidcmd('QUIT')
666 self.close()
667 return resp
668
669 def close(self):
670 '''Close the connection without assuming anything about it.'''
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300671 try:
672 file = self.file
673 self.file = None
674 if file is not None:
675 file.close()
676 finally:
677 sock = self.sock
678 self.sock = None
679 if sock is not None:
680 sock.close()
Guido van Rossumc567c601992-11-05 22:22:37 +0000681
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000682try:
683 import ssl
Brett Cannoncd171c82013-07-04 17:43:24 -0400684except ImportError:
Giampaolo Rodola'3ff5e752013-03-07 14:03:27 +0100685 _SSLSocket = None
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000686else:
Giampaolo Rodola'3ff5e752013-03-07 14:03:27 +0100687 _SSLSocket = ssl.SSLSocket
688
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000689 class FTP_TLS(FTP):
690 '''A FTP subclass which adds TLS support to FTP as described
691 in RFC-4217.
692
693 Connect as usual to port 21 implicitly securing the FTP control
694 connection before authenticating.
695
696 Securing the data connection requires user to explicitly ask
697 for it by calling prot_p() method.
698
699 Usage example:
700 >>> from ftplib import FTP_TLS
701 >>> ftps = FTP_TLS('ftp.python.org')
Ezio Melotti4969f702011-03-15 05:59:46 +0200702 >>> ftps.login() # login anonymously previously securing control channel
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000703 '230 Guest login ok, access restrictions apply.'
704 >>> ftps.prot_p() # switch to secure data connection
705 '200 Protection level set to P'
706 >>> ftps.retrlines('LIST') # list directory content securely
707 total 9
708 drwxr-xr-x 8 root wheel 1024 Jan 3 1994 .
709 drwxr-xr-x 8 root wheel 1024 Jan 3 1994 ..
710 drwxr-xr-x 2 root wheel 1024 Jan 3 1994 bin
711 drwxr-xr-x 2 root wheel 1024 Jan 3 1994 etc
712 d-wxrwxr-x 2 ftp wheel 1024 Sep 5 13:43 incoming
713 drwxr-xr-x 2 root wheel 1024 Nov 17 1993 lib
714 drwxr-xr-x 6 1094 wheel 1024 Sep 13 19:07 pub
715 drwxr-xr-x 3 root wheel 1024 Jan 3 1994 usr
716 -rw-r--r-- 1 root root 312 Aug 1 1994 welcome.msg
717 '226 Transfer complete.'
718 >>> ftps.quit()
719 '221 Goodbye.'
720 >>>
721 '''
Christian Heimesa170fa12017-09-15 20:27:30 +0200722 ssl_version = ssl.PROTOCOL_TLS_CLIENT
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000723
724 def __init__(self, host='', user='', passwd='', acct='', keyfile=None,
Giampaolo Rodolàa67299e2010-05-26 18:06:04 +0000725 certfile=None, context=None,
Giampaolo Rodolà396ff062011-02-28 19:19:51 +0000726 timeout=_GLOBAL_DEFAULT_TIMEOUT, source_address=None):
Giampaolo Rodolàa67299e2010-05-26 18:06:04 +0000727 if context is not None and keyfile is not None:
728 raise ValueError("context and keyfile arguments are mutually "
729 "exclusive")
730 if context is not None and certfile is not None:
731 raise ValueError("context and certfile arguments are mutually "
732 "exclusive")
Christian Heimesd0486372016-09-10 23:23:33 +0200733 if keyfile is not None or certfile is not None:
734 import warnings
735 warnings.warn("keyfile and certfile are deprecated, use a"
736 "custom context instead", DeprecationWarning, 2)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000737 self.keyfile = keyfile
738 self.certfile = certfile
Christian Heimes67986f92013-11-23 22:43:47 +0100739 if context is None:
740 context = ssl._create_stdlib_context(self.ssl_version,
741 certfile=certfile,
742 keyfile=keyfile)
Giampaolo Rodolàa67299e2010-05-26 18:06:04 +0000743 self.context = context
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000744 self._prot_p = False
Giampaolo Rodolà396ff062011-02-28 19:19:51 +0000745 FTP.__init__(self, host, user, passwd, acct, timeout, source_address)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000746
747 def login(self, user='', passwd='', acct='', secure=True):
748 if secure and not isinstance(self.sock, ssl.SSLSocket):
749 self.auth()
750 return FTP.login(self, user, passwd, acct)
751
752 def auth(self):
753 '''Set up secure control connection by using TLS/SSL.'''
754 if isinstance(self.sock, ssl.SSLSocket):
755 raise ValueError("Already using TLS")
Christian Heimesa170fa12017-09-15 20:27:30 +0200756 if self.ssl_version >= ssl.PROTOCOL_TLS:
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000757 resp = self.voidcmd('AUTH TLS')
758 else:
759 resp = self.voidcmd('AUTH SSL')
Christian Heimese5b5edf2013-12-02 02:56:02 +0100760 self.sock = self.context.wrap_socket(self.sock,
Benjamin Peterson7243b572014-11-23 17:04:34 -0600761 server_hostname=self.host)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000762 self.file = self.sock.makefile(mode='r', encoding=self.encoding)
763 return resp
764
Giampaolo Rodola'096dcb12011-06-27 11:17:51 +0200765 def ccc(self):
766 '''Switch back to a clear-text control connection.'''
767 if not isinstance(self.sock, ssl.SSLSocket):
768 raise ValueError("not using TLS")
769 resp = self.voidcmd('CCC')
770 self.sock = self.sock.unwrap()
771 return resp
772
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000773 def prot_p(self):
774 '''Set up secure data connection.'''
775 # PROT defines whether or not the data channel is to be protected.
776 # Though RFC-2228 defines four possible protection levels,
777 # RFC-4217 only recommends two, Clear and Private.
778 # Clear (PROT C) means that no security is to be used on the
779 # data-channel, Private (PROT P) means that the data-channel
780 # should be protected by TLS.
781 # PBSZ command MUST still be issued, but must have a parameter of
782 # '0' to indicate that no buffering is taking place and the data
783 # connection should not be encapsulated.
784 self.voidcmd('PBSZ 0')
785 resp = self.voidcmd('PROT P')
786 self._prot_p = True
787 return resp
788
789 def prot_c(self):
790 '''Set up clear text data connection.'''
791 resp = self.voidcmd('PROT C')
792 self._prot_p = False
793 return resp
794
795 # --- Overridden FTP methods
796
797 def ntransfercmd(self, cmd, rest=None):
798 conn, size = FTP.ntransfercmd(self, cmd, rest)
799 if self._prot_p:
Christian Heimese5b5edf2013-12-02 02:56:02 +0100800 conn = self.context.wrap_socket(conn,
Benjamin Peterson7243b572014-11-23 17:04:34 -0600801 server_hostname=self.host)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000802 return conn, size
803
Giampaolo Rodola'24befa82011-05-07 19:09:34 +0200804 def abort(self):
805 # overridden as we can't pass MSG_OOB flag to sendall()
806 line = b'ABOR' + B_CRLF
807 self.sock.sendall(line)
808 resp = self.getmultiline()
809 if resp[:3] not in {'426', '225', '226'}:
810 raise error_proto(resp)
811 return resp
812
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000813 __all__.append('FTP_TLS')
Andrew Svetlovf7a17b42012-12-25 16:47:37 +0200814 all_errors = (Error, OSError, EOFError, ssl.SSLError)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000815
816
Guido van Rossumacfb82a1997-10-22 20:49:52 +0000817_150_re = None
Fred Drake4de02d91997-01-10 18:26:09 +0000818
819def parse150(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000820 '''Parse the '150' response for a RETR request.
821 Returns the expected transfer size or None; size is not guaranteed to
822 be present in the 150 message.
823 '''
824 if resp[:3] != '150':
Collin Winterce36ad82007-08-30 01:19:48 +0000825 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000826 global _150_re
827 if _150_re is None:
828 import re
Antoine Pitroufd036452008-08-19 17:56:33 +0000829 _150_re = re.compile(
R David Murray44b548d2016-09-08 13:59:53 -0400830 r"150 .* \((\d+) bytes\)", re.IGNORECASE | re.ASCII)
Tim Peters88869f92001-01-14 23:36:06 +0000831 m = _150_re.match(resp)
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000832 if not m:
833 return None
Florent Xicluna2bb96f52011-10-23 22:11:00 +0200834 return int(m.group(1))
Fred Drake4de02d91997-01-10 18:26:09 +0000835
836
Guido van Rossum70297d32001-08-17 17:24:29 +0000837_227_re = None
838
Guido van Rossumd2560b01996-05-28 23:41:25 +0000839def parse227(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000840 '''Parse the '227' response for a PASV request.
841 Raises error_proto if it does not contain '(h1,h2,h3,h4,p1,p2)'
842 Return ('host.addr.as.numbers', port#) tuple.'''
Guido van Rossumd2560b01996-05-28 23:41:25 +0000843
Tim Peters88869f92001-01-14 23:36:06 +0000844 if resp[:3] != '227':
Collin Winterce36ad82007-08-30 01:19:48 +0000845 raise error_reply(resp)
Guido van Rossum70297d32001-08-17 17:24:29 +0000846 global _227_re
847 if _227_re is None:
848 import re
Antoine Pitroufd036452008-08-19 17:56:33 +0000849 _227_re = re.compile(r'(\d+),(\d+),(\d+),(\d+),(\d+),(\d+)', re.ASCII)
Guido van Rossum70297d32001-08-17 17:24:29 +0000850 m = _227_re.search(resp)
851 if not m:
Collin Winterce36ad82007-08-30 01:19:48 +0000852 raise error_proto(resp)
Guido van Rossum70297d32001-08-17 17:24:29 +0000853 numbers = m.groups()
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000854 host = '.'.join(numbers[:4])
855 port = (int(numbers[4]) << 8) + int(numbers[5])
Tim Peters88869f92001-01-14 23:36:06 +0000856 return host, port
Guido van Rossumd2560b01996-05-28 23:41:25 +0000857
858
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000859def parse229(resp, peer):
Martin Panter7462b6492015-11-02 03:37:02 +0000860 '''Parse the '229' response for an EPSV request.
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000861 Raises error_proto if it does not contain '(|||port|)'
862 Return ('host.addr.as.numbers', port#) tuple.'''
863
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000864 if resp[:3] != '229':
Collin Winterce36ad82007-08-30 01:19:48 +0000865 raise error_reply(resp)
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000866 left = resp.find('(')
Collin Winterce36ad82007-08-30 01:19:48 +0000867 if left < 0: raise error_proto(resp)
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000868 right = resp.find(')', left + 1)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000869 if right < 0:
Collin Winterce36ad82007-08-30 01:19:48 +0000870 raise error_proto(resp) # should contain '(|||port|)'
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000871 if resp[left + 1] != resp[right - 1]:
Collin Winterce36ad82007-08-30 01:19:48 +0000872 raise error_proto(resp)
Walter Dörwalda401ae42002-06-03 10:41:45 +0000873 parts = resp[left + 1:right].split(resp[left+1])
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000874 if len(parts) != 5:
Collin Winterce36ad82007-08-30 01:19:48 +0000875 raise error_proto(resp)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000876 host = peer[0]
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000877 port = int(parts[3])
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000878 return host, port
879
880
Guido van Rossumc567c601992-11-05 22:22:37 +0000881def parse257(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000882 '''Parse the '257' response for a MKD or PWD request.
883 This is a response to a MKD or PWD request: a directory name.
884 Returns the directoryname in the 257 reply.'''
Guido van Rossumd2560b01996-05-28 23:41:25 +0000885
Tim Peters88869f92001-01-14 23:36:06 +0000886 if resp[:3] != '257':
Collin Winterce36ad82007-08-30 01:19:48 +0000887 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000888 if resp[3:5] != ' "':
889 return '' # Not compliant to RFC 959, but UNIX ftpd does this
890 dirname = ''
891 i = 5
892 n = len(resp)
893 while i < n:
894 c = resp[i]
895 i = i+1
896 if c == '"':
897 if i >= n or resp[i] != '"':
898 break
899 i = i+1
900 dirname = dirname + c
901 return dirname
Guido van Rossum1115ab21992-11-04 15:51:30 +0000902
Guido van Rossum2f3941d1997-10-07 14:49:56 +0000903
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000904def print_line(line):
Tim Peters88869f92001-01-14 23:36:06 +0000905 '''Default retrlines callback to print a line.'''
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000906 print(line)
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000907
Guido van Rossum2f3941d1997-10-07 14:49:56 +0000908
Guido van Rossumd2560b01996-05-28 23:41:25 +0000909def ftpcp(source, sourcename, target, targetname = '', type = 'I'):
Tim Peters88869f92001-01-14 23:36:06 +0000910 '''Copy file from one FTP-instance to another.'''
Giampaolo Rodola'94c56a12013-03-07 13:51:20 +0100911 if not targetname:
912 targetname = sourcename
Tim Peters88869f92001-01-14 23:36:06 +0000913 type = 'TYPE ' + type
914 source.voidcmd(type)
915 target.voidcmd(type)
916 sourcehost, sourceport = parse227(source.sendcmd('PASV'))
917 target.sendport(sourcehost, sourceport)
918 # RFC 959: the user must "listen" [...] BEFORE sending the
919 # transfer request.
920 # So: STOR before RETR, because here the target is a "user".
921 treply = target.sendcmd('STOR ' + targetname)
Giampaolo Rodola'94c56a12013-03-07 13:51:20 +0100922 if treply[:3] not in {'125', '150'}:
923 raise error_proto # RFC 959
Tim Peters88869f92001-01-14 23:36:06 +0000924 sreply = source.sendcmd('RETR ' + sourcename)
Giampaolo Rodola'94c56a12013-03-07 13:51:20 +0100925 if sreply[:3] not in {'125', '150'}:
926 raise error_proto # RFC 959
Tim Peters88869f92001-01-14 23:36:06 +0000927 source.voidresp()
928 target.voidresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000929
Tim Peters88869f92001-01-14 23:36:06 +0000930
Guido van Rossum1115ab21992-11-04 15:51:30 +0000931def test():
Tim Peters88869f92001-01-14 23:36:06 +0000932 '''Test program.
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000933 Usage: ftp [-d] [-r[file]] host [-l[dir]] [-d[dir]] [-p] [file] ...
934
935 -d dir
936 -l list
937 -p password
938 '''
939
940 if len(sys.argv) < 2:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000941 print(test.__doc__)
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000942 sys.exit(0)
Guido van Rossumd2560b01996-05-28 23:41:25 +0000943
Berker Peksag8f791d32014-11-01 10:45:57 +0200944 import netrc
945
Tim Peters88869f92001-01-14 23:36:06 +0000946 debugging = 0
947 rcfile = None
948 while sys.argv[1] == '-d':
949 debugging = debugging+1
950 del sys.argv[1]
951 if sys.argv[1][:2] == '-r':
952 # get name of alternate ~/.netrc file:
953 rcfile = sys.argv[1][2:]
954 del sys.argv[1]
955 host = sys.argv[1]
956 ftp = FTP(host)
957 ftp.set_debuglevel(debugging)
958 userid = passwd = acct = ''
959 try:
Berker Peksag8f791d32014-11-01 10:45:57 +0200960 netrcobj = netrc.netrc(rcfile)
Andrew Svetlovf7a17b42012-12-25 16:47:37 +0200961 except OSError:
Tim Peters88869f92001-01-14 23:36:06 +0000962 if rcfile is not None:
963 sys.stderr.write("Could not open account file"
964 " -- using anonymous login.")
965 else:
966 try:
Berker Peksag8f791d32014-11-01 10:45:57 +0200967 userid, acct, passwd = netrcobj.authenticators(host)
Tim Peters88869f92001-01-14 23:36:06 +0000968 except KeyError:
969 # no account for host
970 sys.stderr.write(
971 "No account -- using anonymous login.")
972 ftp.login(userid, passwd, acct)
973 for file in sys.argv[2:]:
974 if file[:2] == '-l':
975 ftp.dir(file[2:])
976 elif file[:2] == '-d':
977 cmd = 'CWD'
978 if file[2:]: cmd = cmd + ' ' + file[2:]
979 resp = ftp.sendcmd(cmd)
980 elif file == '-p':
981 ftp.set_pasv(not ftp.passiveserver)
982 else:
983 ftp.retrbinary('RETR ' + file, \
984 sys.stdout.write, 1024)
985 ftp.quit()
Guido van Rossum221ec0b1995-08-04 04:39:30 +0000986
987
988if __name__ == '__main__':
Tim Peters88869f92001-01-14 23:36:06 +0000989 test()